Skip to content

Commit d5da4aa

Browse files
committed
Add 900 tester
1 parent 5bcdbe7 commit d5da4aa

3 files changed

Lines changed: 87 additions & 29 deletions

File tree

Lines changed: 58 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,58 @@
1+
int External()
2+
{
3+
std::string path{"o2sim_Kine.root"};
4+
int numberOfInjectedSignalsPerEvent{1};
5+
std::vector<int> injectedPDGs = {
6+
3334,
7+
-3334,
8+
3312,
9+
-3312};
10+
11+
auto nInjection = injectedPDGs.size();
12+
13+
TFile file(path.c_str(), "READ");
14+
if (file.IsZombie()) {
15+
std::cerr << "Cannot open ROOT file " << path << "\n";
16+
return 1;
17+
}
18+
19+
auto tree = (TTree*)file.Get("o2sim");
20+
if (!tree) {
21+
std::cerr << "Cannot find tree o2sim in file " << path << "\n";
22+
return 1;
23+
}
24+
std::vector<o2::MCTrack>* tracks{};
25+
tree->SetBranchAddress("MCTrack", &tracks);
26+
27+
std::vector<int> nSignal;
28+
for (int i = 0; i < nInjection; i++) {
29+
nSignal.push_back(0);
30+
}
31+
32+
auto nEvents = tree->GetEntries();
33+
for (int i = 0; i < nEvents; i++) {
34+
auto check = tree->GetEntry(i);
35+
for (int idxMCTrack = 0; idxMCTrack < tracks->size(); ++idxMCTrack) {
36+
auto track = tracks->at(idxMCTrack);
37+
auto pdg = track.GetPdgCode();
38+
auto it = std::find(injectedPDGs.begin(), injectedPDGs.end(), pdg);
39+
int index = std::distance(injectedPDGs.begin(), it); // index of injected PDG
40+
if (it != injectedPDGs.end()) // found
41+
{
42+
// count signal PDG
43+
nSignal[index]++;
44+
}
45+
}
46+
}
47+
std::cout << "--------------------------------\n";
48+
std::cout << "# Events: " << nEvents << "\n";
49+
for (int i = 0; i < nInjection; i++) {
50+
std::cout << "# Injected nuclei \n";
51+
std::cout << injectedPDGs[i] << ": " << nSignal[i] << "\n";
52+
if (nSignal[i] == 0) {
53+
std::cerr << "No generated: " << injectedPDGs[i] << "\n";
54+
return 1; // At least one of the injected particles should be generated
55+
}
56+
}
57+
return 0;
58+
}
Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
1-
# PDG N ptMin ptMax genDecayed
2-
1000010020 10 0.2 10
3-
-1000010020 10 0.2 10
4-
1000010030 10 0.2 10
5-
-1000010030 10 0.2 10
6-
1000020030 10 0.2 10
7-
-1000020030 10 0.2 10
8-
1000020040 10 0.2 10
9-
-1000020040 10 0.2 10
1+
# PDG N ptMin ptMax etaMin etaMax genDecayed
2+
1000010020 10 0.2 10 -1.2 1.2
3+
-1000010020 10 0.2 10 -1.2 1.2
4+
1000010030 10 0.2 10 -1.2 1.2
5+
-1000010030 10 0.2 10 -1.2 1.2
6+
1000020030 10 0.2 10 -1.2 1.2
7+
-1000020030 10 0.2 10 -1.2 1.2
8+
1000020040 10 0.2 10 -1.2 1.2
9+
-1000020040 10 0.2 10 -1.2 1.2
Lines changed: 20 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -1,20 +1,20 @@
1-
# PDG N ptMin ptMax DecayByGenerator
2-
313 10 0.2 10 genDecayed
3-
-313 10 0.2 10 genDecayed
4-
323 10 0.2 10 genDecayed
5-
-323 10 0.2 10 genDecayed
6-
333 10 0.2 10 genDecayed
7-
9010221 10 0.2 10 genDecayed
8-
113 10 0.2 10 genDecayed
9-
213 10 0.2 10 genDecayed
10-
-213 10 0.2 10 genDecayed
11-
3114 10 0.2 10 genDecayed
12-
-3114 10 0.2 10 genDecayed
13-
3224 10 0.2 10 genDecayed
14-
-3224 10 0.2 10 genDecayed
15-
3124 10 0.2 10 genDecayed
16-
-3124 10 0.2 10 genDecayed
17-
3324 10 0.2 10 genDecayed
18-
-3324 10 0.2 10 genDecayed
19-
10323 10 0.2 10 genDecayed
20-
-10323 10 0.2 10 genDecayed
1+
# PDG N ptMin ptMax etaMin etaMax DecayByGenerator
2+
313 10 0.2 10 -1.2 1.2 genDecayed
3+
-313 10 0.2 10 -1.2 1.2 genDecayed
4+
323 10 0.2 10 -1.2 1.2 genDecayed
5+
-323 10 0.2 10 -1.2 1.2 genDecayed
6+
333 10 0.2 10 -1.2 1.2 genDecayed
7+
9010221 10 0.2 10 -1.2 1.2 genDecayed
8+
113 10 0.2 10 -1.2 1.2 genDecayed
9+
213 10 0.2 10 -1.2 1.2 genDecayed
10+
-213 10 0.2 10 -1.2 1.2 genDecayed
11+
3114 10 0.2 10 -1.2 1.2 genDecayed
12+
-3114 10 0.2 10 -1.2 1.2 genDecayed
13+
3224 10 0.2 10 -1.2 1.2 genDecayed
14+
-3224 10 0.2 10 -1.2 1.2 genDecayed
15+
3124 10 0.2 10 -1.2 1.2 genDecayed
16+
-3124 10 0.2 10 -1.2 1.2 genDecayed
17+
3324 10 0.2 10 -1.2 1.2 genDecayed
18+
-3324 10 0.2 10 -1.2 1.2 genDecayed
19+
10323 10 0.2 10 -1.2 1.2 genDecayed
20+
-10323 10 0.2 10 -1.2 1.2 genDecayed

0 commit comments

Comments
 (0)