1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
#include <stdio.h>
#include <iostream>
#include <vector>
#include <chrono>
#include "TFile.h"
#include "TTree.h"
#include "TBranch.h"
#include "TInterpreter.h"
void read_jagged2_root() {
gInterpreter->GenerateDictionary("vector<vector<float> >", "vector");
std::vector<std::vector<float>>* data;
data = nullptr;
TFile* f = new TFile("/home/jpivarski/storage/data/chep-2021-jagged-jagged-jagged/zlib9-jagged2.root");
TTree* t;
f->GetObject("tree", t);
TBranch* b;
b = nullptr;
t->SetBranchAddress("branch", &data, &b);
auto begin_time = std::chrono::high_resolution_clock::now();
int64_t num_entries = b->GetEntries();
int64_t num_baskets = b->GetWriteBasket();
Long64_t* basket_starts = b->GetBasketEntry();
for (int64_t basketid = 0; basketid < num_baskets; basketid++) {
int64_t start = basket_starts[basketid];
int64_t stop = num_entries;
if (basketid + 1 < num_baskets) {
stop = basket_starts[basketid + 1];
}
std::vector<int32_t> offsets1;
std::vector<int32_t> offsets0;
std::vector<float> content;
offsets1.reserve(1024);
offsets0.reserve(1024);
content.reserve(1024);
offsets1.push_back(0);
offsets0.push_back(0);
for (int64_t i = start; i < stop; i++) {
b->GetEntry(i);
offsets1.push_back(offsets1.back() + data->size());
for (auto y : *data) {
offsets0.push_back(offsets0.back() + y.size());
std::copy(y.begin(), y.end(), std::back_inserter(content));
}
}
}
auto end_time = std::chrono::high_resolution_clock::now();
int64_t count_nanoseconds = std::chrono::duration_cast<std::chrono::nanoseconds>(
end_time - begin_time
).count();
std::cout << "ROOT zlib9-jagged2.root " << num_entries << " entries " << (count_nanoseconds / 1e9) << " seconds" << std::endl;
}
|
