Skip to content

Commit ed79d06

Browse files
add tests and move to folders
1 parent ec5df78 commit ed79d06

9 files changed

Lines changed: 138 additions & 7 deletions

MC/config/PWGGAJE/external/generator/generator_pythia8_box_parametrisedModel_pythia6Fragmentation.C renamed to MC/config/PWGGAJE/external/generator/parametrisedJetModel/generator_pythia8_box_parametrisedModel_pythia6Fragmentation.C

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@
1515
#include "Pythia8/Pythia.h"
1616
#include <rapidjson/document.h>
1717
#include <rapidjson/filereadstream.h>
18-
// #include "CCDB/CcdbApi.h"
1918

2019
R__LOAD_LIBRARY(libpythia6)
2120
R__LOAD_LIBRARY(libGeneratorParam)
@@ -27,7 +26,7 @@ using namespace Pythia8;
2726
// #include "SimulationDataFormat/MCEventHeader.h"
2827

2928
// Input to simulation:
30-
// inputFilePathName file is expected to be a json file with the structure so:
29+
// inputFilePathName file is expected to be a json file with the structure like so:
3130
// {
3231
// "simLog": false,
3332
// "sglGenRAA": 1,
@@ -67,7 +66,7 @@ public:
6766
// understands it's not a ROOT file), thus one
6867
// cannot use TFile::Cp()
6968
if (gSystem->Exec(aliencp.Data()) != 0) {
70-
cout << "Error: Model file " << inputFilePathName << " does not exist!" << endl;
69+
cout << "Error: Sim parameters .json file " << inputFilePathName << " does not exist!" << endl;
7170
exit(1);
7271
}
7372
// open the file

MC/config/PWGGAJE/ini/parametrisedJetModel/GeneratorPythia8Box_parametrisedModel_pythia6Fragmentation_PbPb_cent0010_noBkg.ini

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
### bkg production using pythia8 box generator, to embed jet-jet production into
22
[GeneratorExternal]
3-
fileName = ${O2DPG_ROOT}/MC/config/PWGGAJE/external/generator/generator_pythia8_box_parametrisedModel_pythia6Fragmentation.C
3+
fileName = ${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGGAJE/external/generator/parametrisedJetModel/generator_pythia8_box_parametrisedModel_pythia6Fragmentation.C
44
funcName = generateParametrisedJetModel("/alice/cern.ch/user/a/alandou/Analysis/PWGJE/simSettings/combinatorialBkgClosure/", "parametrisedModel_PbPb_5p36TeV_cent0010.json", false)
55

66
[GeneratorPythia8]

MC/config/PWGGAJE/ini/parametrisedJetModel/GeneratorPythia8Box_parametrisedModel_pythia6Fragmentation_PbPb_cent0010_withBkg.ini

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
### bkg production using pythia8 box generator, to embed jet-jet production into
22
[GeneratorExternal]
3-
fileName = ${O2DPG_ROOT}/MC/config/PWGGAJE/external/generator/generator_pythia8_box_parametrisedModel_pythia6Fragmentation.C
3+
fileName = ${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGGAJE/external/generator/parametrisedJetModel/generator_pythia8_box_parametrisedModel_pythia6Fragmentation.C
44
funcName = generateParametrisedJetModel("/alice/cern.ch/user/a/alandou/Analysis/PWGJE/simSettings/combinatorialBkgClosure/", "parametrisedModel_PbPb_5p36TeV_cent0010.json", true)
55

66
[GeneratorPythia8]

MC/config/PWGGAJE/ini/parametrisedJetModel/GeneratorPythia8Box_parametrisedModel_pythia6Fragmentation_PbPb_cent5080_noBkg.ini

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
### bkg production using pythia8 box generator, to embed jet-jet production into
22
[GeneratorExternal]
3-
fileName = ${O2DPG_ROOT}/MC/config/PWGGAJE/external/generator/generator_pythia8_box_parametrisedModel_pythia6Fragmentation.C
3+
fileName = ${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGGAJE/external/generator/parametrisedJetModel/generator_pythia8_box_parametrisedModel_pythia6Fragmentation.C
44
funcName = generateParametrisedJetModel("/alice/cern.ch/user/a/alandou/Analysis/PWGJE/simSettings/combinatorialBkgClosure/", "parametrisedModel_PbPb_5p36TeV_cent5080.json", false)
55

66
[GeneratorPythia8]

MC/config/PWGGAJE/ini/parametrisedJetModel/GeneratorPythia8Box_parametrisedModel_pythia6Fragmentation_PbPb_cent5080_withBkg.ini

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
### bkg production using pythia8 box generator, to embed jet-jet production into
22
[GeneratorExternal]
3-
fileName = ${O2DPG_ROOT}/MC/config/PWGGAJE/external/generator/generator_pythia8_box_parametrisedModel_pythia6Fragmentation.C
3+
fileName = ${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGGAJE/external/generator/parametrisedJetModel/generator_pythia8_box_parametrisedModel_pythia6Fragmentation.C
44
funcName = generateParametrisedJetModel("/alice/cern.ch/user/a/alandou/Analysis/PWGJE/simSettings/combinatorialBkgClosure/", "parametrisedModel_PbPb_5p36TeV_cent5080.json", true)
55

66
[GeneratorPythia8]
Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
int External() {
2+
std::string path{"o2sim_Kine.root"};
3+
4+
TFile file(path.c_str(), "READ");
5+
if (file.IsZombie()) {
6+
std::cerr << "Cannot open ROOT file " << path << "\n";
7+
return 1;
8+
}
9+
10+
auto tree = (TTree *)file.Get("o2sim");
11+
std::vector<o2::MCTrack> *tracks{};
12+
tree->SetBranchAddress("MCTrack", &tracks);
13+
o2::dataformats::MCEventHeader *eventHeader = nullptr;
14+
tree->SetBranchAddress("MCEventHeader.", &eventHeader);
15+
16+
int sumTracks{};
17+
auto nEvents = tree->GetEntries();
18+
19+
for (int i = 0; i < nEvents; i++) {
20+
tree->GetEntry(i);
21+
sumTracks += tracks->size();
22+
}
23+
24+
std::cout << "--------------------------------\n";
25+
std::cout << "# Events: " << nEvents << "\n";
26+
std::cout << "# tracks summed over all events (jet-jet + MB): " << sumTracks << "\n";
27+
28+
if(sumTracks < 1) {
29+
std::cerr << "No tracks in simulated events\n";
30+
return 1;
31+
}
32+
return 0;
33+
}
Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
int External() {
2+
std::string path{"o2sim_Kine.root"};
3+
4+
TFile file(path.c_str(), "READ");
5+
if (file.IsZombie()) {
6+
std::cerr << "Cannot open ROOT file " << path << "\n";
7+
return 1;
8+
}
9+
10+
auto tree = (TTree *)file.Get("o2sim");
11+
std::vector<o2::MCTrack> *tracks{};
12+
tree->SetBranchAddress("MCTrack", &tracks);
13+
o2::dataformats::MCEventHeader *eventHeader = nullptr;
14+
tree->SetBranchAddress("MCEventHeader.", &eventHeader);
15+
16+
int sumTracks{};
17+
auto nEvents = tree->GetEntries();
18+
19+
for (int i = 0; i < nEvents; i++) {
20+
tree->GetEntry(i);
21+
sumTracks += tracks->size();
22+
}
23+
24+
std::cout << "--------------------------------\n";
25+
std::cout << "# Events: " << nEvents << "\n";
26+
std::cout << "# tracks summed over all events (jet-jet + MB): " << sumTracks << "\n";
27+
28+
if(sumTracks < 1) {
29+
std::cerr << "No tracks in simulated events\n";
30+
return 1;
31+
}
32+
return 0;
33+
}
Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
int External() {
2+
std::string path{"o2sim_Kine.root"};
3+
4+
TFile file(path.c_str(), "READ");
5+
if (file.IsZombie()) {
6+
std::cerr << "Cannot open ROOT file " << path << "\n";
7+
return 1;
8+
}
9+
10+
auto tree = (TTree *)file.Get("o2sim");
11+
std::vector<o2::MCTrack> *tracks{};
12+
tree->SetBranchAddress("MCTrack", &tracks);
13+
o2::dataformats::MCEventHeader *eventHeader = nullptr;
14+
tree->SetBranchAddress("MCEventHeader.", &eventHeader);
15+
16+
int sumTracks{};
17+
auto nEvents = tree->GetEntries();
18+
19+
for (int i = 0; i < nEvents; i++) {
20+
tree->GetEntry(i);
21+
sumTracks += tracks->size();
22+
}
23+
24+
std::cout << "--------------------------------\n";
25+
std::cout << "# Events: " << nEvents << "\n";
26+
std::cout << "# tracks summed over all events (jet-jet + MB): " << sumTracks << "\n";
27+
28+
if(sumTracks < 1) {
29+
std::cerr << "No tracks in simulated events\n";
30+
return 1;
31+
}
32+
return 0;
33+
}
Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
int External() {
2+
std::string path{"o2sim_Kine.root"};
3+
4+
TFile file(path.c_str(), "READ");
5+
if (file.IsZombie()) {
6+
std::cerr << "Cannot open ROOT file " << path << "\n";
7+
return 1;
8+
}
9+
10+
auto tree = (TTree *)file.Get("o2sim");
11+
std::vector<o2::MCTrack> *tracks{};
12+
tree->SetBranchAddress("MCTrack", &tracks);
13+
o2::dataformats::MCEventHeader *eventHeader = nullptr;
14+
tree->SetBranchAddress("MCEventHeader.", &eventHeader);
15+
16+
int sumTracks{};
17+
auto nEvents = tree->GetEntries();
18+
19+
for (int i = 0; i < nEvents; i++) {
20+
tree->GetEntry(i);
21+
sumTracks += tracks->size();
22+
}
23+
24+
std::cout << "--------------------------------\n";
25+
std::cout << "# Events: " << nEvents << "\n";
26+
std::cout << "# tracks summed over all events (jet-jet + MB): " << sumTracks << "\n";
27+
28+
if(sumTracks < 1) {
29+
std::cerr << "No tracks in simulated events\n";
30+
return 1;
31+
}
32+
return 0;
33+
}

0 commit comments

Comments
 (0)