|
| 1 | +#include <ROOT/RFieldBase.hxx> |
| 2 | +#include <ROOT/RNTupleModel.hxx> |
| 3 | +#include <ROOT/RNTupleReader.hxx> |
| 4 | +#include <ROOT/RNTupleWriter.hxx> |
| 5 | + |
| 6 | +#include <ROOT/TestSupport.hxx> |
| 7 | + |
| 8 | +#include <cstdio> |
| 9 | +#include <string> |
| 10 | +#include <utility> |
| 11 | + |
| 12 | +#include "StreamerBeyond.hxx" |
| 13 | +#include "gtest/gtest.h" |
| 14 | + |
| 15 | +TEST(RField, StreamerBeyond) |
| 16 | +{ |
| 17 | + ROOT::TestSupport::FileRaii fileGuard("test_ntuple_rfield_streamer_beyond.root"); |
| 18 | + |
| 19 | + { |
| 20 | + auto model = ROOT::RNTupleModel::Create(); |
| 21 | + auto f = ROOT::RFieldBase::Create("f", "StreamerBeyond").Unwrap(); |
| 22 | + EXPECT_TRUE(dynamic_cast<ROOT::RStreamerField *>(f.get())); |
| 23 | + model->AddField(std::move(f)); |
| 24 | + auto writer = ROOT::RNTupleWriter::Recreate(std::move(model), "ntpl", fileGuard.GetPath()); |
| 25 | + |
| 26 | + auto ptr = writer->GetModel().GetDefaultEntry().GetPtr<StreamerBeyond>("f"); |
| 27 | + ptr->fOne = std::vector<std::int64_t>(100000000, -1); |
| 28 | + ptr->fTwo = std::vector<std::int64_t>(100000000, -2); |
| 29 | + |
| 30 | + writer->Fill(); |
| 31 | + } |
| 32 | + |
| 33 | + auto reader = ROOT::RNTupleReader::Open("ntpl", fileGuard.GetPath()); |
| 34 | + ASSERT_EQ(1u, reader->GetNEntries()); |
| 35 | + StreamerBeyond sb; |
| 36 | + auto view = reader->GetView("f", &sb, "StreamerBeyond"); |
| 37 | + |
| 38 | + view(0); |
| 39 | + |
| 40 | + auto ptr = view.GetValue().GetPtr<StreamerBeyond>(); |
| 41 | + EXPECT_EQ(100000000u, ptr->fOne.size()); |
| 42 | + EXPECT_EQ(-1, ptr->fOne.at(1000)); |
| 43 | + EXPECT_EQ(100000000u, ptr->fTwo.size()); |
| 44 | + EXPECT_EQ(-2, ptr->fTwo.at(2000)); |
| 45 | + EXPECT_EQ(-2, ptr->fTwo.at(99999999u)); |
| 46 | +} |
0 commit comments