Create an EventHeader and also compare it

key4hep · Oct 5, 2023 · caff3e9 · caff3e9
1 parent 55e117c
commit caff3e9
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 5 deletions.
diff --git a/tests/edm4hep_to_lcio.cpp b/tests/edm4hep_to_lcio.cpp
@@ -14,14 +14,16 @@ int main()
   const auto edmEvent = createExampleEvent();
   const auto lcioEvent = EDM4hep2LCIOConv::convEvent(edmEvent);
 
-  // if (!compareEventHeader(lcioEvent.get(), &edmEvent)) {
-  //   return 1;
-  // }
+  if (!compareEventHeader(lcioEvent.get(), &edmEvent)) {
+    return 1;
+  }
 
   for (const auto& name : edmEvent.getAvailableCollections()) {
     const auto edmColl = edmEvent.get(name);
     const auto typeName = edmColl->getValueTypeName();
-    if (typeName == "edm4hep::CaloHitContribution" || typeName == "edm4hep::ParticleID") {
+    if (
+      typeName == "edm4hep::CaloHitContribution" || typeName == "edm4hep::ParticleID" ||
+      typeName == "edm4hep::EventHeader") {
       continue;
     }
     try {
@@ -41,7 +43,9 @@ int main()
 
   for (const auto& name : edmEvent.getAvailableCollections()) {
     const auto type = edmEvent.get(name)->getTypeName();
-    if (type == "edm4hep::CaloHitContributionCollection" || type == "edm4hep::ParticleIDCollection") {
+    if (
+      type == "edm4hep::CaloHitContributionCollection" || type == "edm4hep::ParticleIDCollection" ||
+      type == "edm4hep::EventHeaderCollection") {
       continue;
     }
     const auto* lcioColl = lcioEvent->getCollection(name);

diff --git a/tests/src/EDM4hep2LCIOUtilities.cc b/tests/src/EDM4hep2LCIOUtilities.cc
@@ -5,6 +5,7 @@
 #include "edm4hep/RawCalorimeterHitCollection.h"
 #include "edm4hep/SimCalorimeterHitCollection.h"
 #include "edm4hep/TrackerHitCollection.h"
+#include <edm4hep/EventHeaderCollection.h>
 #include <edm4hep/RawTimeSeriesCollection.h>
 #include "edm4hep/TrackCollection.h"
 #include "edm4hep/SimCalorimeterHitCollection.h"
@@ -252,10 +253,24 @@ std::pair<edm4hep::SimCalorimeterHitCollection, edm4hep::CaloHitContributionColl
   return {std::move(simcalohit_coll), std::move(contrib_coll)};
 }
 
+edm4hep::EventHeaderCollection createEventHeader()
+{
+  auto evtHeaderColl = edm4hep::EventHeaderCollection {};
+  auto evtHeader = evtHeaderColl.create();
+
+  evtHeader.setWeight(3.14f);
+  evtHeader.setEventNumber(123456789);
+  evtHeader.setRunNumber(42);
+  evtHeader.setTimeStamp(0x71AAE);
+
+  return evtHeaderColl;
+}
+
 podio::Frame createExampleEvent()
 {
   podio::Frame event;
 
+  event.put(createEventHeader(), "EventHeader");
   const auto& mcParticles =
     event.put(createMCParticles(test_config::nMCParticles, test_config::mcpParentIdcs), "mcParticles");
   const auto& caloHits = event.put(createCalorimeterHits(test_config::nCaloHits), "caloHits");

diff --git a/tests/src/EDM4hep2LCIOUtilities.h b/tests/src/EDM4hep2LCIOUtilities.h
@@ -15,6 +15,7 @@ namespace edm4hep {
   class TrackCollection;
   class SimCalorimeterHitCollection;
   class CaloHitContributionCollection;
+  class EventHeaderCollection;
 } // namespace edm4hep
 
 namespace podio {
@@ -120,6 +121,8 @@ std::pair<edm4hep::SimCalorimeterHitCollection, edm4hep::CaloHitContributionColl
   const edm4hep::MCParticleCollection& mcParticles,
   const std::vector<test_config::CaloContIdx>& link_mcparticles_idcs);
 
+edm4hep::EventHeaderCollection createEventHeader();
+
 /**
  * Create an example event that can be used to test the converter.
  *