ctkMacroCompilePythonScript.cmake 4.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137
  1. #
  2. # Based on ParaView/VTK/Utilities/vtkTclTest2Py/CMakeLists.txt and
  3. # ParaView/VTK/Wrapping/Python/CMakeLists.txt
  4. #
  5. INCLUDE(${CTK_CMAKE_DIR}/ctkMacroParseArguments.cmake)
  6. #! \ingroup CMakeAPI
  7. MACRO(ctkMacroCompilePythonScript)
  8. ctkMacroParseArguments(MY
  9. "TARGET_NAME;SCRIPTS;RESOURCES;SOURCE_DIR;DESTINATION_DIR;INSTALL_DIR"
  10. ""
  11. ${ARGN}
  12. )
  13. FIND_PACKAGE(PythonInterp REQUIRED)
  14. FIND_PACKAGE(PythonLibs REQUIRED)
  15. # Extract python lib path
  16. get_filename_component(PYTHON_LIBRARY_PATH ${PYTHON_LIBRARY} PATH)
  17. # Sanity checks
  18. FOREACH(varname TARGET_NAME SCRIPTS DESTINATION_DIR INSTALL_DIR)
  19. IF(NOT DEFINED MY_${varname})
  20. MESSAGE(FATAL_ERROR "${varname} is mandatory")
  21. ENDIF()
  22. ENDFOREACH()
  23. IF(NOT DEFINED MY_SOURCE_DIR)
  24. SET(MY_SOURCE_DIR ${CMAKE_CURRENT_SOURCE_DIR})
  25. ENDIF()
  26. SET(copied_files)
  27. SET(copied_python_files)
  28. FOREACH(file ${MY_SCRIPTS})
  29. # Append "py" extension if needed
  30. get_filename_component(file_ext ${file} EXT)
  31. IF(NOT "${file_ext}" MATCHES "py")
  32. SET(file "${file}.py")
  33. ENDIF()
  34. SET(src "${MY_SOURCE_DIR}/${file}")
  35. SET(tgt "${MY_DESTINATION_DIR}/${file}")
  36. IF(IS_ABSOLUTE ${file})
  37. SET(src ${file})
  38. file(RELATIVE_PATH tgt_file ${CMAKE_CURRENT_BINARY_DIR} ${file})
  39. SET(tgt "${MY_DESTINATION_DIR}/${tgt_file}")
  40. ENDIF()
  41. ADD_CUSTOM_COMMAND(DEPENDS ${src}
  42. COMMAND ${CMAKE_COMMAND} -E copy ${src} ${tgt}
  43. OUTPUT ${tgt}
  44. COMMENT "Copying python script: ${file}")
  45. LIST(APPEND copied_python_files ${tgt})
  46. LIST(APPEND copied_files ${tgt})
  47. ENDFOREACH()
  48. IF(DEFINED MY_RESOURCES)
  49. FOREACH(file ${MY_RESOURCES})
  50. SET(src "${CMAKE_CURRENT_SOURCE_DIR}/${file}")
  51. SET(tgt "${MY_DESTINATION_DIR}/${file}")
  52. ADD_CUSTOM_COMMAND(DEPENDS ${src}
  53. COMMAND ${CMAKE_COMMAND} -E copy ${src} ${tgt}
  54. OUTPUT ${tgt}
  55. COMMENT "Copying python resource: ${file}")
  56. LIST(APPEND copied_files ${tgt})
  57. ENDFOREACH()
  58. ENDIF()
  59. IF(copied_files)
  60. ADD_CUSTOM_TARGET(Copy${MY_TARGET_NAME}PythonFiles
  61. ALL
  62. DEPENDS ${copied_files})
  63. ENDIF()
  64. # Byte compile the Python files.
  65. SET(compile_all_script "${CMAKE_CURRENT_BINARY_DIR}/compile_${MY_TARGET_NAME}_python_scripts.py")
  66. # Generate compile_${MY_TARGET_NAME}_python_scripts.py
  67. FILE(WRITE ${compile_all_script} "
  68. #
  69. # Generated by ctkMacroCompilePythonScript CMAKE macro
  70. #
  71. # Based on paraview/VTK/Wrapping/Python/compile_all_vtk.py.in
  72. import compileall
  73. compileall.compile_dir('@MY_DESTINATION_DIR@')
  74. file = open('@CMAKE_CURRENT_BINARY_DIR@/python_compile_@MY_TARGET_NAME@_complete', 'w')
  75. file.write('Done')
  76. ")
  77. # Configure cmake script associated with the custom command
  78. # required to properly update the library path with PYTHON_LIBRARY_PATH
  79. SET(compile_all_cmake_script "${CMAKE_CURRENT_BINARY_DIR}/compile_${MY_TARGET_NAME}_python_scripts.cmake")
  80. FILE(WRITE ${compile_all_cmake_script} "
  81. #
  82. # Generated by ctkMacroCompilePythonScript CMAKE macro
  83. #
  84. IF(WIN32)
  85. SET(ENV{PATH} \"@PYTHON_LIBRARY_PATH@;\$ENV{PATH}\")
  86. ELSEIF(APPLE)
  87. SET(ENV{DYLD_LIBRARY_PATH} \"@PYTHON_LIBRARY_PATH@:\$ENV{DYLD_LIBRARY_PATH}\")
  88. ELSE()
  89. SET(ENV{LD_LIBRARY_PATH} \"@PYTHON_LIBRARY_PATH@:\$ENV{LD_LIBRARY_PATH}\")
  90. ENDIF()
  91. EXECUTE_PROCESS(
  92. COMMAND \"@PYTHON_EXECUTABLE@\" \"@compile_all_script@\"
  93. )
  94. ")
  95. ADD_CUSTOM_COMMAND(
  96. COMMAND ${CMAKE_COMMAND} -P ${compile_all_cmake_script}
  97. DEPENDS ${copied_python_files} ${compile_all_script}
  98. OUTPUT "${CMAKE_CURRENT_BINARY_DIR}/python_compile_${MY_TARGET_NAME}_complete"
  99. COMMENT "Compiling python scripts: ${MY_TARGET_NAME}"
  100. )
  101. ADD_CUSTOM_TARGET(Compile${MY_TARGET_NAME}PythonFiles ALL
  102. DEPENDS
  103. ${CMAKE_CURRENT_BINARY_DIR}/python_compile_${MY_TARGET_NAME}_complete
  104. ${compile_all_script}
  105. )
  106. # Install python module directory
  107. INSTALL(DIRECTORY "${MY_DESTINATION_DIR}"
  108. DESTINATION "${MY_INSTALL_DIR}" COMPONENT Runtime
  109. USE_SOURCE_PERMISSIONS)
  110. ENDMACRO()