[ntuple] add integration test for streamer field evolution

Author: jblomer

roottest/root/ntuple/streamerfield/CMakeLists.txt

@@ -0,0 +1,46 @@
+ROOTTEST_GENERATE_DICTIONARY(
+  event_v2_dict
+  ${CMAKE_CURRENT_SOURCE_DIR}/Event_v2.hxx
+  LINKDEF ${CMAKE_CURRENT_SOURCE_DIR}/Event_v2_LinkDef.h
+  NO_ROOTMAP NO_CXXMODULE
+  FIXTURES_SETUP generated_event_v2_dictionary
+)
+
+ROOTTEST_GENERATE_EXECUTABLE(
+  write_event
+  LIBRARIES Core RIO ROOTNTuple
+  FIXTURES_REQUIRED generated_event_v2_dictionary
+  FIXTURES_SETUP write_event_excutable)
+
+target_sources(
+  write_event
+  PRIVATE write_event.cxx event_v2_dict.cxx
+)
+
+ROOTTEST_ADD_TEST(write_event
+                  EXEC ./write_event
+                  FIXTURES_REQUIRED write_event_excutable
+                  FIXTURES_SETUP written_event)
+
+ROOTTEST_GENERATE_DICTIONARY(
+  event_v3_dict
+  ${CMAKE_CURRENT_SOURCE_DIR}/Event_v3.hxx
+  LINKDEF ${CMAKE_CURRENT_SOURCE_DIR}/Event_v3_LinkDef.h
+  NO_ROOTMAP NO_CXXMODULE
+  FIXTURES_SETUP generated_event_v3_dictionary
+)
+
+ROOTTEST_GENERATE_EXECUTABLE(
+  read_event
+  LIBRARIES Core RIO ROOTNTuple
+  FIXTURES_REQUIRED generated_event_v3_dictionary
+  FIXTURES_SETUP read_event_executable)
+
+target_sources(
+  read_event
+  PRIVATE read_event.cxx event_v3_dict.cxx
+)
+
+ROOTTEST_ADD_TEST(read_event
+                  EXEC ./read_event
+                  FIXTURES_REQUIRED read_event_executable written_event)

roottest/root/ntuple/streamerfield/Event_v2.hxx

@@ -0,0 +1,36 @@
+#ifndef EVENT_V2_H
+#define EVENT_V2_H
+
+#include <RtypesCore.h>
+
+#include <memory>
+
+struct StreamerBase {
+   int fBase = 0;
+   virtual ~StreamerBase() = default;
+
+   ClassDef(StreamerBase, 2)
+};
+
+struct StreamerDerived : public StreamerBase {
+   int fFirst = 1;
+   int fSecond = 2;
+   virtual ~StreamerDerived() = default;
+
+   ClassDef(StreamerDerived, 2)
+};
+
+struct StreamerContainer {
+   std::unique_ptr<StreamerBase> fPtr;
+
+   ClassDefNV(StreamerContainer, 2)
+};
+
+struct Event {
+   int fX = 137;
+   StreamerContainer fField;
+
+   ClassDefNV(Event, 2)
+};
+
+#endif // EVENT_V2_H

roottest/root/ntuple/streamerfield/Event_v2_LinkDef.h

@@ -0,0 +1,8 @@
+#ifdef __ROOTCLING__
+
+#pragma link C++ class StreamerBase+;
+#pragma link C++ class StreamerDerived+;
+#pragma link C++ options = rntupleStreamerMode(true) class StreamerContainer+;
+#pragma link C++ class Event+;
+
+#endif

roottest/root/ntuple/streamerfield/Event_v3.hxx

@@ -0,0 +1,36 @@
+#ifndef EVENT_V3_H
+#define EVENT_V3_H
+
+#include <RtypesCore.h>
+
+#include <memory>
+
+struct StreamerBase {
+   int fBase = 0;
+   virtual ~StreamerBase() = default;
+
+   ClassDef(StreamerBase, 2)
+};
+
+struct StreamerDerived : public StreamerBase {
+   int fSecond = 2;
+   int fFirst = 1;
+   virtual ~StreamerDerived() = default;
+
+   ClassDef(StreamerDerived, 3)
+};
+
+struct StreamerContainer {
+   std::unique_ptr<StreamerBase> fPtr;
+
+   ClassDefNV(StreamerContainer, 2)
+};
+
+struct Event {
+   StreamerContainer fField;
+   int fY = 42;
+
+   ClassDefNV(Event, 3)
+};
+
+#endif // EVENT_V3_H

roottest/root/ntuple/streamerfield/Event_v3_LinkDef.h

@@ -0,0 +1,12 @@
+#ifdef __ROOTCLING__
+
+#pragma link C++ class StreamerBase+;
+#pragma link C++ class StreamerDerived+;
+#pragma link C++ options = rntupleStreamerMode(true) class StreamerContainer+;
+#pragma link C++ class Event+;
+
+#pragma read sourceClass = "StreamerBase" source = "int fBase;" version = "[2]" \
+    targetClass = "StreamerBase" target = "fBase" \
+    code = "{ fBase = onfile.fBase + 1000; }"
+
+#endif

roottest/root/ntuple/streamerfield/read_event.cxx

@@ -0,0 +1,34 @@
+#include <ROOT/RNTupleReader.hxx>
+
+#include "Event_v3.hxx"
+
+#include <cstdio>
+
+int main()
+{
+   auto reader = ROOT::RNTupleReader::Open("ntpl", "root_test_streamerfield.root");
+
+   auto ptrEvent = reader->GetModel().GetDefaultEntry().GetPtr<Event>("event");
+
+   reader->LoadEntry(0);
+
+   if (!dynamic_cast<const ROOT::RStreamerField *>(&reader->GetModel().GetConstField("event.fField")))
+      return 10;
+
+   if (!ptrEvent->fField.fPtr)
+      return 1;
+   auto derived = dynamic_cast<StreamerDerived *>(ptrEvent->fField.fPtr.get());
+   if (!derived)
+      return 2;
+   if (derived->fBase != 1000)
+      return 3;
+   if (derived->fFirst != 1)
+      return 4;
+   if (derived->fSecond != 2)
+      return 5;
+   if (ptrEvent->fY != 42)
+      return 6;
+
+   std::remove("root_test_streamerfield.root");
+   return 0;
+}

roottest/root/ntuple/streamerfield/write_event.cxx

@@ -0,0 +1,18 @@
+#include <ROOT/RNTupleModel.hxx>
+#include <ROOT/RNTupleWriter.hxx>
+
+#include "Event_v2.hxx"
+
+#include <memory>
+
+int main()
+{
+   auto model = ROOT::RNTupleModel::Create();
+   auto field = model->MakeField<Event>("event");
+   auto writer = ROOT::RNTupleWriter::Recreate(std::move(model), "ntpl", "root_test_streamerfield.root");
+
+   field->fField.fPtr = std::unique_ptr<StreamerBase>(new StreamerDerived());
+   writer->Fill();
+
+   return 0;
+}