From 6571d6ba5a03f43f037ca1cd9af7e23bee2f5fe9 Mon Sep 17 00:00:00 2001
From: Wouter Wijsman <wwijsman@live.nl>
Date: Wed, 30 Oct 2024 23:25:16 +0100
Subject: [PATCH] Add add_prx_module macro to psp-cmake

---
 src/base/CreatePRX.cmake | 57 ++++++++++++++++++++++++++++++++++++++++
 src/base/Makefile.am     |  2 +-
 src/base/pspdev.cmake    |  1 +
 3 files changed, 59 insertions(+), 1 deletion(-)
 create mode 100644 src/base/CreatePRX.cmake

diff --git a/src/base/CreatePRX.cmake b/src/base/CreatePRX.cmake
new file mode 100644
index 00000000..56119b21
--- /dev/null
+++ b/src/base/CreatePRX.cmake
@@ -0,0 +1,57 @@
+# File defining macro outputting PSP-specific PRX module out of passed executable target.
+cmake_minimum_required(VERSION 3.10)
+
+macro(add_prx_module)
+  
+  set(OPTIONS_ARGS)
+  set(ONE_VALUE_ARGS
+    TARGET # CMake target for the prx module that is being output
+  )
+  set(MULTI_VALUE_ARGS
+    SRC_FILES # Source files for the module
+    EXP_FILES # Export files for the module
+  )
+
+  cmake_parse_arguments("ARG" "${OPTIONS_ARGS}" "${ONE_VALUE_ARGS}" "${MULTI_VALUE_ARGS}" ${ARGN} )
+
+  # Initialize a list to store generated .c files
+  set(GENERATED_C_FILES)
+
+  # Loop over each .exp file to set up a rule to generate a corresponding .c file
+  foreach(EXP_FILE ${ARG_EXP_FILES})
+    # Get the filename without extension
+    get_filename_component(EXP_FILE_NAME ${EXP_FILE} NAME_WE)
+
+    # Define the output .c file path
+    set(GENERATED_C_FILE ${CMAKE_BINARY_DIR}/${EXP_FILE_NAME}.c)
+
+    # Add a custom command for each .exp file to generate a .c file
+    add_custom_command(
+        OUTPUT ${GENERATED_C_FILE}
+        COMMAND psp-build-exports -b ${EXP_FILE} > ${GENERATED_C_FILE}
+        DEPENDS ${EXP_FILE}
+        COMMENT "Generating ${EXP_FILE_NAME}.c from ${EXP_FILE_NAME}.exp"
+    )
+
+    # Add the generated .c file to the list
+    list(APPEND GENERATED_C_FILES ${GENERATED_C_FILE})
+  endforeach()
+
+  add_executable(${ARG_TARGET} ${ARG_SRC_FILES} ${GENERATED_C_FILES})
+  target_link_options(${ARG_TARGET} PRIVATE -Wl,-q,-T${PSPDEV}/psp/sdk/lib/linkfile.prx -nostartfiles -Wl,-zmax-page-size=128)
+
+  add_custom_command(
+    TARGET ${ARG_TARGET}
+    POST_BUILD COMMAND
+    "$ENV{PSPDEV}/bin/psp-fixup-imports" "$<TARGET_FILE:${ARG_TARGET}>"
+    COMMENT "Calling psp-fixup-imports"
+  )
+
+  add_custom_command(
+    TARGET ${ARG_TARGET}
+    POST_BUILD COMMAND
+    "${PSPDEV}/bin/psp-prxgen" "$<TARGET_FILE:${ARG_TARGET}>"
+    "$<TARGET_FILE:${ARG_TARGET}>.prx"
+    COMMENT "Calling prxgen"
+  )
+endmacro(add_prx_module)
diff --git a/src/base/Makefile.am b/src/base/Makefile.am
index 879f06ba..c55ec7b0 100644
--- a/src/base/Makefile.am
+++ b/src/base/Makefile.am
@@ -16,6 +16,6 @@ buildmakprx_DATA = build_prx.mak linkfile.prx
 
 ## Install the cmake toolchain config to $PSPDEV/share
 buildmakcmakedir = @PSPDEV_SHAREDIR@
-buildmakcmake_DATA = pspdev.cmake CreatePBP.cmake
+buildmakcmake_DATA = pspdev.cmake CreatePBP.cmake CreatePRX.cmake
 
 EXTRA_DIST = build.mak prxspecs build_prx.mak linkfile.prx pspdev.cmake
diff --git a/src/base/pspdev.cmake b/src/base/pspdev.cmake
index af3452f8..85b60218 100644
--- a/src/base/pspdev.cmake
+++ b/src/base/pspdev.cmake
@@ -35,3 +35,4 @@ SET(PLATFORM_PSP TRUE)
 SET(PSP TRUE)
 
 include("${PSPDEV}/psp/share/CreatePBP.cmake")
+include("${PSPDEV}/psp/share/CreatePRX.cmake")