Change ROOTFrame{Writer,Reader} to ROOT{Writer,Reader} when possible

key4hep · Feb 6, 2024 · e8b94b8 · e8b94b8
1 parent 9fc0b00
commit e8b94b8
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 0 deletions.
diff --git a/test/hepmc/edm4hep_testhepmc.cc b/test/hepmc/edm4hep_testhepmc.cc
@@ -17,7 +17,11 @@
 #include "HepPDT/ParticleID.hh"
 
 #include "podio/Frame.h"
+#if __has_include("podio/ROOTWriter.h")
+#include "podio/ROOTWriter.h"
+#else
 #include "podio/ROOTFrameWriter.h"
+#endif
 
 #include "edm4hep/MCParticleCollection.h"
 
@@ -154,7 +158,11 @@ int main() {
   auto event = podio::Frame();
   event.put(std::move(edm_particle_collection), "TestParticles2");
 
+#if __has_include("podio/ROOTWriter.h")
+  auto writer = podio::ROOTWriter("edm4hep_testhepmc.root");
+#else
   auto writer = podio::ROOTFrameWriter("edm4hep_testhepmc.root");
+#endif
   writer.writeFrame(event, "events");
 
   // after all events

diff --git a/test/read_events.cc b/test/read_events.cc
@@ -1,8 +1,16 @@
 #include "read_events.h"
+#if __has_include("podio/ROOTReader.h")
+#include "podio/ROOTReader.h"
+#else
 #include "podio/ROOTFrameReader.h"
+#endif
 
 int main() {
+#if __has_include("podio/ROOTReader.h")
+  read_events<podio::ROOTReader>("edm4hep_events.root");
+#else
   read_events<podio::ROOTFrameReader>("edm4hep_events.root");
+#endif
 
   return 0;
 }
diff --git a/test/write_events.cc b/test/write_events.cc
@@ -1,7 +1,15 @@
 #include "write_events.h"
+#if __has_include("podio/ROOTWriter.h")
+#include "podio/ROOTWriter.h"
+#else
 #include "podio/ROOTFrameWriter.h"
+#endif
 
 int main(int, char*[]) {
 
+#if __has_include("podio/ROOTWriter.h")
+  write<podio::ROOTWriter>("edm4hep_events.root");
+#else
   write<podio::ROOTFrameWriter>("edm4hep_events.root");
+#endif
 }
diff --git a/tools/src/edm4hep2json.cxx b/tools/src/edm4hep2json.cxx
@@ -5,7 +5,11 @@
 #include "TFile.h"
 
 // podio
+#if __has_include("podio/ROOTReader.h")
+#include "podio/ROOTReader.h"
+#else
 #include "podio/ROOTFrameReader.h"
+#endif
 #include "podio/ROOTLegacyReader.h"
 
 // std
@@ -131,8 +135,13 @@ int main(int argc, char** argv) {
     return read_frames<podio::ROOTLegacyReader>(inFilePath, outFilePath, requestedCollections, requestedEvents,
                                                 frameName, nEventsMax, verboser);
   } else {
+#if __has_include("podio/ROOTReader.h")
+    return read_frames<podio::ROOTReader>(inFilePath, outFilePath, requestedCollections, requestedEvents,
+                                               frameName, nEventsMax, verboser);
+#else
     return read_frames<podio::ROOTFrameReader>(inFilePath, outFilePath, requestedCollections, requestedEvents,
                                                frameName, nEventsMax, verboser);
+#endif
   }
 
   return EXIT_SUCCESS;