Make Association I/O work with python

AIDASoft · Mar 24, 2023 · 10e5243 · 10e5243
1 parent 688dd2c
commit 10e5243
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt
@@ -13,6 +13,7 @@ PODIO_GENERATE_DATAMODEL(datamodel datalayout.yaml headers sources
 
 # Use the cmake building blocks to add the different parts (conditionally)
 PODIO_ADD_DATAMODEL_CORE_LIB(TestDataModel "${headers}" "${sources}")
+target_sources(TestDataModel PRIVATE DatamodelAssociations.cc)
 find_package(nlohmann_json 3.10)
 if (nlohmann_json_FOUND)
   message(STATUS "Found compatible version of JSON library, will add JSON support to test datamodel")
@@ -22,6 +23,7 @@ endif()
 
 PODIO_ADD_ROOT_IO_DICT(TestDataModelDict TestDataModel "${headers}" src/selection.xml)
 PODIO_ADD_SIO_IO_BLOCKS(TestDataModel "${headers}" "${sources}")
+target_sources(TestDataModelSioBlocks PRIVATE DatamodelAssociations.cc)
 
 # Build the extension data model and link it against the upstream model
 PODIO_GENERATE_DATAMODEL(extension_model datalayout_extension.yaml ext_headers ext_sources

diff --git a/tests/DatamodelAssociations.cc b/tests/DatamodelAssociations.cc
@@ -0,0 +1,6 @@
+#include "podio/AssociationCollection.h"
+
+#include "datamodel/ExampleMCCollection.h"
+#include "datamodel/ExampleWithARelationCollection.h"
+
+PODIO_DECLARE_ASSOCIATION(ExampleMC, ex42::ExampleWithARelation)