21#include <ROOT/RNTuple.hxx>
43constexpr char const *kNTupleFileName =
"ntpl007_mtFill.root";
46constexpr int kNWriterThreads = 4;
49constexpr int kNEventsPerThread = 25000;
52void FillData(std::unique_ptr<REntry> entry, RNTupleWriter *ntuple) {
54 static std::mutex gLock;
56 static std::atomic<std::uint32_t> gThreadId;
57 const auto threadId = ++gThreadId;
59 auto prng = std::make_unique<TRandom3>();
62 auto id = entry->Get<std::uint32_t>(
"id");
63 auto vpx = entry->Get<std::vector<float>>(
"vpx");
64 auto vpy = entry->Get<std::vector<float>>(
"vpy");
65 auto vpz = entry->Get<std::vector<float>>(
"vpz");
67 for (
int i = 0; i < kNEventsPerThread; i++) {
73 int npx =
static_cast<int>(prng->Rndm(1) * 15);
75 for (
int j = 0; j < npx; ++j) {
80 vpx->emplace_back(px);
81 vpy->emplace_back(py);
82 vpz->emplace_back(pz);
85 std::lock_guard<std::mutex> guard(gLock);
94 auto model = RNTupleModel::Create();
95 model->MakeField<std::uint32_t>(
"id");
96 model->MakeField<std::vector<float>>(
"vpx");
97 model->MakeField<std::vector<float>>(
"vpy");
98 model->MakeField<std::vector<float>>(
"vpz");
101 auto ntuple = RNTupleWriter::Recreate(std::move(model),
"NTuple", kNTupleFileName);
103 std::vector<std::unique_ptr<REntry>> entries;
104 std::vector<std::thread> threads;
105 for (
int i = 0; i < kNWriterThreads; ++i)
106 entries.emplace_back(ntuple->CreateEntry());
107 for (
int i = 0; i < kNWriterThreads; ++i)
108 threads.emplace_back(FillData, std::move(entries[i]), ntuple.get());
109 for (
int i = 0; i < kNWriterThreads; ++i)
120 auto ntuple = RNTupleReader::Open(
"NTuple", kNTupleFileName);
122 auto viewVpx = ntuple->GetView<
float>(
"vpx._0");
126 TCanvas *
c1 =
new TCanvas(
"c2",
"Multi-Threaded Filling Example", 200, 10, 1500, 500);
130 TH1F h(
"h",
"This is the px distribution", 100, -4, 4);
133 for (
auto i : viewVpx.GetFieldRange())
139 auto nEvents = ntuple->GetNEntries();
140 auto viewId = ntuple->GetView<std::uint32_t>(
"id");
141 TH2F hFillSequence(
"",
"Entry Id vs Thread Id;Entry Sequence Number;Filling Thread",
142 100, 0, nEvents, 100, 0, kNWriterThreads);
143 for (
auto i : ntuple->GetEntryRange())
144 hFillSequence.Fill(i, viewId(i));
145 hFillSequence.DrawCopy();
#define R__LOAD_LIBRARY(LIBRARY)
R__EXTERN TStyle * gStyle
The REntry is a collection of values in an ntuple corresponding to a complete row in the data set.
The RNTupleModel encapulates the schema of an ntuple.
An RNTuple that is used to read data from storage.
An RNTuple that gets filled with entries (data) and writes them to storage.
1-D histogram with a float per channel (see TH1 documentation)}
2-D histogram with a float per channel (see TH1 documentation)}
void SetOptStat(Int_t stat=1)
The type of information printed in the histogram statistics box can be selected via the parameter mod...
void FillData(BinData &dv, const TH1 *hist, TF1 *func=nullptr)
fill the data vector from a TH1.