Skip to content

Commit c1a76e0

Browse files
committed
add missing test macro
1 parent e712443 commit c1a76e0

1 file changed

Lines changed: 57 additions & 0 deletions

File tree

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

0 commit comments

Comments
 (0)