Feature/lean kernel exposure

.bumpversion.cfg

@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 0.8.0.dev7
+current_version = 0.8.0.dev9
 commit = True
 tag = True
 parse = (?P<major>\d+)\.(?P<minor>\d+)\.(?P<patch>\d+)(\.(?P<release>[a-z]+)(?P<dev>\d+))?

tudatpy/CMakeLists.txt

@@ -162,10 +162,13 @@ install(TARGETS kernel
 install(FILES ${TUDATPY_PYTHON_FILES} "${CMAKE_CURRENT_BINARY_DIR}/_version.py"
         DESTINATION ${TUDATPY_INSTALL_PATH})
 
-# Install hybrid C++/Python modules
 macro(INSTALL_HYBRID_CPP_PYTHON_MODULE module_dir_name)
-    install(DIRECTORY ${module_dir_name}/ DESTINATION "${TUDATPY_INSTALL_PATH}/${module_dir_name}")
-    file(COPY ${module_dir_name}/ DESTINATION ${CMAKE_CURRENT_BINARY_DIR}/${module_dir_name}/)
+    if(EXISTS ${CMAKE_CURRENT_SOURCE_DIR}/${module_dir_name})
+        install(DIRECTORY ${module_dir_name}/ DESTINATION "${TUDATPY_INSTALL_PATH}/${module_dir_name}")
+        file(COPY ${module_dir_name}/ DESTINATION ${CMAKE_CURRENT_BINARY_DIR}/${module_dir_name}/)
+    else()
+        message(WARNING "Not installing hybrid C++/Python module ${module_dir_name} because it cannot be found in ${CMAKE_CURRENT_SOURCE_DIR}")
+    endif()
 endmacro()
 
 # Hybrid modules
@@ -183,6 +186,44 @@ INSTALL_HYBRID_CPP_PYTHON_MODULE(data)
 INSTALL_HYBRID_CPP_PYTHON_MODULE(util)
 INSTALL_HYBRID_CPP_PYTHON_MODULE(io)
 
+#################################################
+# Generate stubs to enable kernel autocompletions
+#################################################
+
+# Find stubgen executable
+find_program(Stubgen_Executable NAMES stubgen)
+if(NOT Stubgen_Executable)
+    message(FATAL_ERROR "Could not find stubgen. Please make sure it's installed in the active conda environment.")
+endif()
+
+# Generate stubs
+add_custom_command(TARGET kernel POST_BUILD
+    COMMAND ${Stubgen_Executable} -p tudatpy.kernel -o . > /dev/null 2>&1
+    WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/../
+    COMMENT "Generating stubs..."
+)
+
+# Move stubs into tudatpy
+# - TODO: replace by install directory just like tudatpy
+# - TODO: turn into macro that can becalled to include stubs from particular kernel submodules, just like tudatpy python modules are selected for install
+add_custom_command(TARGET kernel POST_BUILD
+    COMMAND ${CMAKE_COMMAND} -E copy_directory ${CMAKE_CURRENT_BINARY_DIR}/kernel ${CMAKE_CURRENT_BINARY_DIR} > /dev/null 2>&1
+    COMMENT "Moving stubs to tudatpy..."
+)
+
+# Remove tudatpy/__init__.pyi
+add_custom_command(TARGET kernel POST_BUILD
+    COMMAND ${CMAKE_COMMAND} -E remove "${CMAKE_CURRENT_BINARY_DIR}/__init__.pyi" > /dev/null 2>&1
+    COMMENT "Removing __init__.pyi file..."
+)
+
+# Remove directory (tudatpy/kernel) where stubgen generated the stubs
+add_custom_command(TARGET kernel POST_BUILD
+    COMMAND ${CMAKE_COMMAND} -E remove_directory "${CMAKE_CURRENT_BINARY_DIR}/kernel" > /dev/null 2>&1
+    COMMENT "Removing stub directory..."
+)
+#################################################
+
 #install(FILES test.py "${CMAKE_CURRENT_BINARY_DIR}/_version.py")
 file(COPY ../tests DESTINATION ${CMAKE_CURRENT_BINARY_DIR}/../)