Skip to content

Commit 458ff61

Browse files
authored
add generator for NeNe based on EPOS4 (#2322)
1 parent afdcafc commit 458ff61

File tree

2 files changed

+90
-0
lines changed

2 files changed

+90
-0
lines changed
Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
#NEV_TEST> 10
2+
[GeneratorExternal]
3+
fileName=${O2DPG_MC_CONFIG_ROOT}/MC/config/examples/epos4/generator_EPOS4.C
4+
funcName=generateEPOS4("${O2DPG_MC_CONFIG_ROOT}/MC/config/examples/epos4/generator/NeNe_536TeV_EPOS4.optns", 2147483647)
5+
6+
[GeneratorFileOrCmd]
7+
cmd=${O2DPG_MC_CONFIG_ROOT}/MC/config/examples/epos4/epos.sh
8+
bMaxSwitch=none
9+
10+
# Set to version 2 if EPOS4.0.0 is used
11+
[HepMC]
12+
version=3
Lines changed: 78 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,78 @@
1+
int External()
2+
{
3+
std::string path{"o2sim_Kine.root"};
4+
5+
// Check that file exists, can be opened and has the correct tree
6+
TFile file(path.c_str(), "READ");
7+
if (file.IsZombie())
8+
{
9+
std::cerr << "Cannot open ROOT file " << path << "\n";
10+
return 1;
11+
}
12+
13+
auto tree = (TTree*)file.Get("o2sim");
14+
if (!tree)
15+
{
16+
std::cerr << "Cannot find tree o2sim in file " << path << "\n";
17+
return 1;
18+
}
19+
20+
std::vector<o2::MCTrack>* tracks{};
21+
tree->SetBranchAddress("MCTrack", &tracks);
22+
23+
// Check if all events are filled
24+
auto nEvents = tree->GetEntries();
25+
for (Long64_t i = 0; i < nEvents; ++i)
26+
{
27+
tree->GetEntry(i);
28+
if (tracks->empty())
29+
{
30+
std::cerr << "Empty entry found at event " << i << "\n";
31+
return 1;
32+
}
33+
}
34+
35+
// Check if there is 1 event, as customarily set in the ini file
36+
// Heavy-ion collisions with hydro and hadronic cascade are very slow to simulate
37+
if (nEvents != 10)
38+
{
39+
std::cerr << "Expected 10 events, got " << nEvents << "\n";
40+
return 1;
41+
}
42+
43+
// ---- Neon-Neon parameters ----
44+
constexpr int kNeonPDG = 1000100200; // Ne-20 ion
45+
constexpr double kEnucleon = 5360.; // GeV per nucleon
46+
constexpr int kA = 20; // Neon mass number
47+
constexpr double kNeonEnergy = kA * kEnucleon / 2.0; // beam energy in GeV
48+
49+
// Check if each event has two neon ions at expected energy
50+
for (int i = 0; i < nEvents; i++)
51+
{
52+
tree->GetEntry(i);
53+
int count = 0;
54+
55+
for (int idxMCTrack = 0; idxMCTrack < tracks->size(); ++idxMCTrack)
56+
{
57+
auto track = tracks->at(idxMCTrack);
58+
double energy = track.GetEnergy();
59+
60+
// 50 MeV tolerance (floating point safety)
61+
if (std::abs(energy - kNeonEnergy) < 5e-2 &&
62+
track.GetPdgCode() == kNeonPDG)
63+
{
64+
count++;
65+
}
66+
}
67+
68+
if (count < 2)
69+
{
70+
std::cerr << "Event " << i
71+
<< " has less than 2 neon ions at "
72+
<< kNeonEnergy << " GeV\n";
73+
return 1;
74+
}
75+
}
76+
77+
return 0;
78+
}

0 commit comments

Comments
 (0)