Add a header file including all the collections

python/podio_gen/cpp_generator.py

@@ -88,6 +88,7 @@ def post_process(self, _):
         if "ROOT" in self.io_handlers:
             self._prepare_iorules()
             self._create_selection_xml()
+        self._write_all_collections_header()
         self._write_cmake_lists_file()
 
     def do_process_component(self, name, component):
@@ -485,6 +486,22 @@ def _write_list(name, target_folder, files, comment):
             self.any_changes,
         )
 
+    def _write_all_collections_header(self):
+        """Write a header file that includes all collection headers"""
+
+        collection_files = (x.split("::")[-1] + "Collection.h" for x in self.datamodel.datatypes)
+        self._write_file(
+            os.path.join(self.install_dir, self.package_name, f"{self.package_name}.h"),
+            self._eval_template(
+                "AllCollections.h.jinja2",
+                {
+                    "includes": collection_files,
+                    "incfolder": self.incfolder,
+                    "package_name": self.package_name,
+                },
+            ),
+        )
+
     def _write_edm_def_file(self):
         """Write the edm definition to a compile time string"""
         model_encoder = DataModelJSONEncoder()

python/templates/AllCollections.h.jinja2

@@ -0,0 +1,10 @@
+// AUTOMATICALLY GENERATED FILE - DO NOT EDIT
+
+#ifndef {{ package_name.upper() }}_{{ package_name }}_H
+#define {{ package_name.upper() }}_{{ package_name }}_H
+
+{% for name in includes %}
+#include "{{ incfolder }}{{ name }}"
+{% endfor %}
+
+#endif // {{ package_name.upper() }}_{{ package_name }}_H

python/templates/CMakeLists.txt

@@ -1,6 +1,7 @@
 set(PODIO_TEMPLATES
   ${CMAKE_CURRENT_LIST_DIR}/Collection.cc.jinja2
   ${CMAKE_CURRENT_LIST_DIR}/Collection.h.jinja2
+  ${CMAKE_CURRENT_LIST_DIR}/AllCollections.h.jinja2
   ${CMAKE_CURRENT_LIST_DIR}/CollectionData.cc.jinja2
   ${CMAKE_CURRENT_LIST_DIR}/CollectionData.h.jinja2
   ${CMAKE_CURRENT_LIST_DIR}/Component.h.jinja2