Skip to content

Commit ee7372e

Browse files
alcalivaalcaliva
committed
Add generators for light-ion systems based on EPOS4
1 parent eb80feb commit ee7372e

File tree

6 files changed

+244
-0
lines changed

6 files changed

+244
-0
lines changed

‎MC/config/examples/epos4/generator/NeNe_536TeV_EPOS4.optns‎

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,32 @@
1+
!--------------------------------------------------------------------
2+
! Neon-Neon collisions with hydro and hadronic cascade
3+
!--------------------------------------------------------------------
4+
5+
!---------------------------------------
6+
! Define run
7+
!---------------------------------------
8+
9+
application hadron !hadron-hadron, hadron-nucleus, or nucleus-nucleus
10+
set laproj 10 !projectile atomic number
11+
set maproj 20 !projectile mass number
12+
set latarg 10 !target atomic number
13+
set matarg 20 !target mass number
14+
set ecms 5360 !sqrt(s)_pp
15+
set istmax 25 !max status considered for storage
16+
17+
ftime on !string formation time non-zero
18+
!suppressed decays:
19+
nodecays
20+
110 20 2130 -2130 2230 -2230 1130 -1130 1330 -1330 2330 -2330 3331 -3331
21+
end
22+
23+
set ninicon 1 !number of initial conditions used for hydro evolution
24+
core full !core/corona activated
25+
hydro hlle !hydro activated
26+
eos x3ff !eos activated (epos standard EoS)
27+
hacas full !hadronic cascade activated (UrQMD)
28+
set nfreeze 1 !number of freeze out events per hydro event
29+
set modsho 1 !printout every modsho events
30+
set centrality 0 !0=min bias
31+
set ihepmc 2 !HepMC output enabled on stdout
32+
set nfull 10

‎MC/config/examples/epos4/generator/OO_536TeV_EPOS4.optns‎

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,32 @@
1+
!--------------------------------------------------------------------
2+
! Oxygen-Oxygen collisions with hydro and hadronic cascade
3+
!--------------------------------------------------------------------
4+
5+
!---------------------------------------
6+
! Define run
7+
!---------------------------------------
8+
9+
application hadron !hadron-hadron, hadron-nucleus, or nucleus-nucleus
10+
set laproj 8 !projectile atomic number
11+
set maproj 16 !projectile mass number
12+
set latarg 8 !target atomic number
13+
set matarg 16 !target mass number
14+
set ecms 5360 !sqrt(s)_pp
15+
set istmax 25 !max status considered for storage
16+
17+
ftime on !string formation time non-zero
18+
!suppressed decays:
19+
nodecays
20+
110 20 2130 -2130 2230 -2230 1130 -1130 1330 -1330 2330 -2330 3331 -3331
21+
end
22+
23+
set ninicon 1 !number of initial conditions used for hydro evolution
24+
core full !core/corona activated
25+
hydro hlle !hydro activated
26+
eos x3ff !eos activated (epos standard EoS)
27+
hacas full !hadronic cascade activated (UrQMD)
28+
set nfreeze 1 !number of freeze out events per hydro event
29+
set modsho 1 !printout every modsho events
30+
set centrality 0 !0=min bias
31+
set ihepmc 2 !HepMC output enabled on stdout
32+
set nfull 10

‎MC/config/examples/ini/GeneratorEPOS4NeNe536TeV.ini‎

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
#NEV_TEST> 1
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

‎MC/config/examples/ini/GeneratorEPOS4OO536TeV.ini‎

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
#NEV_TEST> 1
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/OO_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

‎MC/config/examples/ini/tests/GeneratorEPOS4NeNe536TeV.C‎

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 exactly 1 event, as set in the ini file
36+
// Heavy-ion collisions with hydro and hadronic cascade are slow to simulate
37+
if (nEvents != 1)
38+
{
39+
std::cerr << "Expected 1 event, 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; // 107200 GeV
48+
49+
// Check if each event has two neon ions at the 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 to account for floating point precision
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+
}

‎MC/config/examples/ini/tests/GeneratorEPOS4OO536TeV.C‎

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 customly set in the ini file
36+
// Heavy-ion collisions with hydro and hadronic cascade are very slow to simulate
37+
if (nEvents != 1)
38+
{
39+
std::cerr << "Expected 1 event, got " << nEvents << "\n";
40+
return 1;
41+
}
42+
43+
// ---- Oxygen-Oxygen parameters ----
44+
constexpr int kOxygenPDG = 1000080160; // O-16 ion
45+
constexpr double kEnucleon = 5360.; // GeV per nucleon
46+
constexpr int kA = 16; // Oxygen mass number
47+
constexpr double kOxygenEnergy = kA * kEnucleon; // 85760 GeV
48+
49+
// Check if each event has two oxygen 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 - kOxygenEnergy) < 5e-2 &&
62+
track.GetPdgCode() == kOxygenPDG)
63+
{
64+
count++;
65+
}
66+
}
67+
68+
if (count < 2)
69+
{
70+
std::cerr << "Event " << i
71+
<< " has less than 2 oxygen ions at "
72+
<< kOxygenEnergy << " GeV\n";
73+
return 1;
74+
}
75+
}
76+
77+
return 0;
78+
}

0 commit comments

Comments
 (0)