Adding missing function docstring and typehint; Restricting directories for cmake

Author: a-monsch

cmake/ConfigureCrownlib.cmake

@@ -2,15 +2,15 @@
 include_directories(${CMAKE_SOURCE_DIR}/src)
 include_directories(${CMAKE_SOURCE_DIR}/include)
 
-include_directories(${CMAKE_SOURCE_DIR}/analysis_configurations/*/cpp_addons/src)
-include_directories(${CMAKE_SOURCE_DIR}/analysis_configurations/*/cpp_addons/include)
+include_directories(${CMAKE_SOURCE_DIR}/analysis_configurations/${ANALYSIS}/cpp_addons/src)
+include_directories(${CMAKE_SOURCE_DIR}/analysis_configurations/${ANALYSIS}/cpp_addons/include)
 
 file(GLOB SOURCES_1 ${CMAKE_SOURCE_DIR}/src/*.cxx)
 file(GLOB SOURCES_2 ${CMAKE_SOURCE_DIR}/src/utility/*.cxx
      ${CMAKE_SOURCE_DIR}/src/RecoilCorrections/*.cxx
      ${CMAKE_SOURCE_DIR}/src/SVFit/*.cxx)
 
-file(GLOB SOURCES_3 ${CMAKE_SOURCE_DIR}/analysis_configurations/*/cpp_addons/src/*.cxx)
+file(GLOB SOURCES_3 ${CMAKE_SOURCE_DIR}/analysis_configurations/${ANALYSIS}/cpp_addons/src/*.cxx)
 
 set(SOURCES ${SOURCES_1} ${SOURCES_2} ${SOURCES_3})
 

code_generation/code_generation.py

@@ -324,11 +324,19 @@ def load_template(self, template_path: str) -> str:
             template = template_file.read()
         return template
 
-    def addon_includes(self):
+    def addon_includes(self) -> str:
+        """
+        Add the includes all .hxx files from analysis configuration folder:
+        analysis_configurations/{analysis_name}/cpp_addons/include
+        Args:
+            None
+        Returns:
+            str - the include statements for the cpp addons
+        """
         path = f"analysis_configurations/{self.analysis_name}/cpp_addons/include"
         if os.path.exists(path) and os.path.isdir(path) and os.listdir(path):
             log.info(f"Adding addons from {path}: {' '.join(os.listdir(path))}")
-            paths = "\n".join(f'#include "{os.path.join(path, item)}"' for item in os.listdir(path))
+            paths = "\n".join(f'#include "{os.path.join(path, item)}"' for item in os.listdir(path) if item.endswith(".hxx"))
             return paths
         else:
             log.info(f"No addons found in {path}")