Skip to content

Commit 74c9a01

Browse files
wuctlbywuctlby
committed
add .c test marco for pp reference and update the corresponding marcos for PbPb and pp
1 parent 475200a commit 74c9a01

File tree

3 files changed

+319
-14
lines changed

3 files changed

+319
-14
lines changed

‎MC/config/PWGHF/ini/tests/GeneratorHF_D2H_ccbar_and_bbbar_PbPb.C‎

Lines changed: 62 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -8,13 +8,68 @@ int External() {
88

99
std::vector<int> checkPdgHadron{411, 421, 431, 4122, 4132, 4232, 4332};
1010
std::map<int, std::vector<std::vector<int>>> checkHadronDecays{ // sorted pdg of daughters
11-
{411, {{-321, 211, 211}, {-313, 211}, {211, 311}, {211, 333}}}, // D+
12-
{421, {{-321, 211}, {-321, 111, 211}}}, // D0
13-
{431, {{211, 333}, {-313, 321}}}, // Ds+
14-
{4122, {{-313, 2212}, {-321, 2224}, {211, 102134}, {-321, 211, 2212}, {311, 2212}}}, // Lc+
15-
{4132, {{211, 3312}}}, // Xic0
16-
{4232, {{-313, 2212}, {-321, 3324}, {211, 211, 3312}, {-321, 211, 2212}}}, // Xic+
17-
{4332, {{211, 3334}}} // Omegac+
11+
{411, {
12+
{-321, 211, 211}, // K- π+ π+ (non-resonant)
13+
{-10311, 211}, // antiK*0(1430) π+
14+
{-313, 211}, // K*0(892) π+
15+
{-321, 211, 211, 111}, // K- π+ π+ π0
16+
{333, 211}, // φ π+ (needed for signal)
17+
{-313, 321}, // K*0(892) K+
18+
{-10311, 321}, // K*0(1430) K+
19+
{-321, 321, 211}, // K- K+ π+ (non-resonant)
20+
{113, 211}, // ρ0 π+
21+
{225, 211}, // f2(1270) π+
22+
{-211, 211, 211} // π- π+ π+ (non-resonant)
23+
}},
24+
{421, {
25+
{-321, 211}, // K- π+ (non-resonant)
26+
{-321, 111, 211}, // K- π+ π0
27+
{213, -321}, // ρ+ K-
28+
{-313, 111}, // antiK*0(892) π0
29+
{-323, 211}, // K*-(892) π+
30+
{-211, 211}, // π- π+
31+
{213, -211}, // ρ+ π-
32+
{-211, 211, 111}, // π- π+ π0
33+
{-321, 321} // K- K+
34+
}},
35+
{431, {
36+
{211, 333}, // φ π+
37+
{-313, 321}, // antiK*(892) K+
38+
{333, 213}, // φ ρ
39+
{113, 211}, // ρ π+
40+
{225, 211}, // f2(1270) π+
41+
{-211, 211, 211}, // π- π+ π+ (s-wave)
42+
{313, 211}, // K*(892)0 π+
43+
{10221, 321}, // f0(1370) K+
44+
{113, 321}, // ρ0 K+
45+
{-211, 321, 211}, // π- K+ π+ (non-resonant)
46+
{221, 211} // η π+
47+
}},
48+
{4122, {
49+
{2212, -321, 211}, // p K- π+ (non-resonant)
50+
{2212, -313}, // p K*0(892)
51+
{2224, -321}, // Δ++ K-
52+
{102134, 211}, // Λ(1520) K-
53+
{2212, 311}, // p K0S
54+
{2212, -321, 211, 111}, // p K- π+ π0
55+
{2212, -211, 211}, // p π- π+
56+
{2212, 333} // p φ
57+
}},
58+
{4232, {
59+
{2122, -321, 211}, // p K- π+
60+
{2212, -313}, // p K*0(892)
61+
{3312, 211, 211}, // Ξ- π+ π+
62+
{2212, 333}, // p φ
63+
{3222, -211, 211} // Σ+ π- π+
64+
{3324, 211} // Ξ*0 π+
65+
}},
66+
{4132, {
67+
{3312, 211} // Xic0 -> Xi- π+
68+
}},
69+
{4332, {
70+
{3334, 211}, // Omegac+ -> Omega- π+
71+
{3312, 211} // Omegac+ -> Xi- π+
72+
}}
1873
};
1974

2075
TFile file(path.c_str(), "READ");

‎MC/config/PWGHF/ini/tests/GeneratorHF_D2H_ccbar_and_bbbar_gap5_Mode2.C‎

Lines changed: 68 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -7,13 +7,74 @@ int External() {
77

88
std::vector<int> checkPdgHadron{411, 421, 431, 4122, 4132, 4232, 4332};
99
std::map<int, std::vector<std::vector<int>>> checkHadronDecays{ // sorted pdg of daughters
10-
{411, {{-321, 211, 211}, {-313, 211}, {211, 311}, {211, 333}}}, // D+
11-
{421, {{-321, 211}, {-321, 111, 211}}}, // D0
12-
{431, {{211, 333}, {-313, 321}}}, // Ds+
13-
{4122, {{-313, 2212}, {-321, 2224}, {211, 102134}, {-321, 211, 2212}, {311, 2212}}}, // Lc+
14-
{4132, {{211, 3312}}}, // Xic0
15-
{4232, {{-313, 2212}, {-321, 3324}, {211, 211, 3312}, {-321, 211, 2212}}}, // Xic+
16-
{4332, {{211, 3334}}} // Omegac+
10+
{421, {
11+
{-321, 211}, // D0 -> K-, pi+
12+
{-321, 211, 111}, // D0 -> K-, pi+, pi0
13+
{213, -321}, // D0 -> rho(770)+, K-
14+
{-313, 111}, // D0 -> Kbar^*(892)0, pi0
15+
{-323, 211}, // D0 -> K^*(892)-, pi+
16+
{-211, 211}, // D0 -> pi-, pi+
17+
{213, -211}, // D0 -> rho(770)+, pi-
18+
{-211, 211, 111}, // D0 -> pi-, pi+, pi0
19+
{-321, 321}, // D0 -> K-, K+
20+
}},
21+
22+
{411, {
23+
{-321, 211, 211}, // D+ -> K-, pi+, pi+
24+
{-10311, 211}, // D+ -> Kbar0^*(1430)0, pi+
25+
{-313, 211}, // D+ -> Kbar^*(892)0, pi+
26+
{-321, 211, 211, 111}, // D+ -> K-, pi+, pi+, pi0
27+
{333, 211}, // D+ -> phi(1020)0, pi+
28+
{-313, 321}, // D+ -> Kbar^*(892)0, K+
29+
{-10311, 321}, // D+ -> Kbar0^*(1430)0, K+
30+
{-321, 321, 211}, // D+ -> K-, K+, pi+
31+
{113, 211}, // D+ -> rho(770)0, pi+
32+
{225, 211}, // D+ -> f2(1270)0, pi+
33+
{-211, 211, 211}, // D+ -> pi-, pi+, pi+
34+
}},
35+
36+
{431, {
37+
{333, 211}, // Ds+ -> phi(1020)0, pi+
38+
{-313, 321}, // Ds+ -> Kbar^*(892)0, K+
39+
{333, 213}, // Ds+ -> phi(1020)0, rho(770)+
40+
{113, 211}, // Ds+ -> rho(770)0, pi+
41+
{225, 211}, // Ds+ -> f2(1270)0, pi+
42+
{-211, 211, 211}, // Ds+ -> pi-, pi+, pi+
43+
{313, 211}, // Ds+ -> K^*(892)0, pi+
44+
{10221, 321}, // Ds+ -> f0(1370)0, K+
45+
{113, 321}, // Ds+ -> rho(770)0, K+
46+
{-211, 321, 211}, // Ds+ -> pi-, K+, pi+
47+
{221, 211}, // Ds+ -> eta, pi+
48+
}},
49+
50+
{4122, {
51+
{2212, -321, 211}, // Lambdac+ -> p, K-, pi+
52+
{2212, -313}, // Lambdac+ -> p, Kbar^*(892)0
53+
{2224, -321}, // Lambdac+ -> Delta(1232)++, K-
54+
{102134, 211}, // Lambdac+ -> 102134, pi+
55+
{2212, 311}, // Lambdac+ -> p, K0
56+
{2212, -321, 211, 111}, // Lambdac+ -> p, K-, pi+, pi0
57+
{2212, -211, 211}, // Lambdac+ -> p, pi-, pi+
58+
{2212, 333}, // Lambdac+ -> p, phi(1020)0
59+
}},
60+
61+
{4232, {
62+
{2212, -321, 211}, // Xic+ -> p, K-, pi+
63+
{2212, -313}, // Xic+ -> p, Kbar^*(892)0
64+
{3312, 211, 211}, // Xic+ -> Xi-, pi+, pi+
65+
{2212, 333}, // Xic+ -> p, phi(1020)0
66+
{3222, -211, 211}, // Xic+ -> Sigma+, pi-, pi+
67+
{3324, 211}, // Xic+ -> Xi(1530)0, pi+
68+
}},
69+
70+
{4132, {
71+
{3312, 211}, // Xic0 -> Xi-, pi+
72+
}},
73+
74+
{4332, {
75+
{3334, 211}, // Omegac0 -> Omega-, pi+
76+
{3312, 211}, // Omegac0 -> Xi-, pi+
77+
}},
1778
};
1879

1980
TFile file(path.c_str(), "READ");

‎MC/config/PWGHF/ini/tests/GeneratorHF_D2H_ccbar_and_bbbar_gap5_Mode2_pp_ref.C‎

Lines changed: 189 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,189 @@
1+
int External() {
2+
std::string path{"/home/wuct/ALICE/mc_production/test/standard/ppref/test/tf1/sgn_Kine.root"};
3+
4+
int checkPdgQuarkOne{4};
5+
int checkPdgQuarkTwo{5};
6+
float ratioTrigger = 1./5; // one event triggered out of 5
7+
8+
std::vector<int> checkPdgHadron{411, 421, 431, 4122, 4132, 4232, 4332};
9+
std::map<int, std::vector<std::vector<int>>> checkHadronDecays{ // sorted pdg of daughters
10+
{421, {
11+
{-321, 211}, // D0 -> K-, pi+
12+
{-321, 211, 111}, // D0 -> K-, pi+, pi0
13+
{213, -321}, // D0 -> rho(770)+, K-
14+
{-313, 111}, // D0 -> Kbar^*(892)0, pi0
15+
{-323, 211}, // D0 -> K^*(892)-, pi+
16+
{-211, 211}, // D0 -> pi-, pi+
17+
{213, -211}, // D0 -> rho(770)+, pi-
18+
{-211, 211, 111}, // D0 -> pi-, pi+, pi0
19+
{-321, 321}, // D0 -> K-, K+
20+
}},
21+
22+
{411, {
23+
{-321, 211, 211}, // D+ -> K-, pi+, pi+
24+
{-10311, 211}, // D+ -> Kbar0^*(1430)0, pi+
25+
{-313, 211}, // D+ -> Kbar^*(892)0, pi+
26+
{-321, 211, 211, 111}, // D+ -> K-, pi+, pi+, pi0
27+
{333, 211}, // D+ -> phi(1020)0, pi+
28+
{-313, 321}, // D+ -> Kbar^*(892)0, K+
29+
{-10311, 321}, // D+ -> Kbar0^*(1430)0, K+
30+
{-321, 321, 211}, // D+ -> K-, K+, pi+
31+
{113, 211}, // D+ -> rho(770)0, pi+
32+
{225, 211}, // D+ -> f2(1270)0, pi+
33+
{-211, 211, 211}, // D+ -> pi-, pi+, pi+
34+
}},
35+
36+
{431, {
37+
{333, 211}, // Ds+ -> phi(1020)0, pi+
38+
{-313, 321}, // Ds+ -> Kbar^*(892)0, K+
39+
{333, 213}, // Ds+ -> phi(1020)0, rho(770)+
40+
{113, 211}, // Ds+ -> rho(770)0, pi+
41+
{225, 211}, // Ds+ -> f2(1270)0, pi+
42+
{-211, 211, 211}, // Ds+ -> pi-, pi+, pi+
43+
{313, 211}, // Ds+ -> K^*(892)0, pi+
44+
{10221, 321}, // Ds+ -> f0(1370)0, K+
45+
{113, 321}, // Ds+ -> rho(770)0, K+
46+
{-211, 321, 211}, // Ds+ -> pi-, K+, pi+
47+
{221, 211}, // Ds+ -> eta, pi+
48+
}},
49+
50+
{4122, {
51+
{2212, -321, 211}, // Lambdac+ -> p, K-, pi+
52+
{2212, -313}, // Lambdac+ -> p, Kbar^*(892)0
53+
{2224, -321}, // Lambdac+ -> Delta(1232)++, K-
54+
{102134, 211}, // Lambdac+ -> 102134, pi+
55+
{2212, 311}, // Lambdac+ -> p, K0
56+
{2212, -321, 211, 111}, // Lambdac+ -> p, K-, pi+, pi0
57+
{2212, -211, 211}, // Lambdac+ -> p, pi-, pi+
58+
{2212, 333}, // Lambdac+ -> p, phi(1020)0
59+
}},
60+
61+
{4232, {
62+
{2212, -321, 211}, // Xic+ -> p, K-, pi+
63+
{2212, -313}, // Xic+ -> p, Kbar^*(892)0
64+
{3312, 211, 211}, // Xic+ -> Xi-, pi+, pi+
65+
{2212, 333}, // Xic+ -> p, phi(1020)0
66+
{3222, -211, 211}, // Xic+ -> Sigma+, pi-, pi+
67+
{3324, 211}, // Xic+ -> Xi(1530)0, pi+
68+
}},
69+
70+
{4132, {
71+
{3312, 211}, // Xic0 -> Xi-, pi+
72+
}},
73+
74+
{4332, {
75+
{3334, 211}, // Omegac0 -> Omega-, pi+
76+
{3312, 211}, // Omegac0 -> Xi-, pi+
77+
}},
78+
};
79+
80+
TFile file(path.c_str(), "READ");
81+
if (file.IsZombie()) {
82+
std::cerr << "Cannot open ROOT file " << path << "\n";
83+
return 1;
84+
}
85+
86+
auto tree = (TTree *)file.Get("o2sim");
87+
std::vector<o2::MCTrack> *tracks{};
88+
tree->SetBranchAddress("MCTrack", &tracks);
89+
o2::dataformats::MCEventHeader *eventHeader = nullptr;
90+
tree->SetBranchAddress("MCEventHeader.", &eventHeader);
91+
92+
int nEventsMB{}, nEventsInjOne{}, nEventsInjTwo{};
93+
int nQuarksOne{}, nQuarksTwo{}, nSignals{}, nSignalGoodDecay{};
94+
auto nEvents = tree->GetEntries();
95+
96+
for (int i = 0; i < nEvents; i++) {
97+
tree->GetEntry(i);
98+
99+
// check subgenerator information
100+
if (eventHeader->hasInfo(o2::mcgenid::GeneratorProperty::SUBGENERATORID)) {
101+
bool isValid = false;
102+
int subGeneratorId = eventHeader->getInfo<int>(o2::mcgenid::GeneratorProperty::SUBGENERATORID, isValid);
103+
if (subGeneratorId == 0) {
104+
nEventsMB++;
105+
} else if (subGeneratorId == checkPdgQuarkOne) {
106+
nEventsInjOne++;
107+
} else if (subGeneratorId == checkPdgQuarkTwo) {
108+
nEventsInjTwo++;
109+
}
110+
}
111+
112+
for (auto &track : *tracks) {
113+
auto pdg = track.GetPdgCode();
114+
if (std::abs(pdg) == checkPdgQuarkOne) {
115+
nQuarksOne++;
116+
continue;
117+
}
118+
if (std::abs(pdg) == checkPdgQuarkTwo) {
119+
nQuarksTwo++;
120+
continue;
121+
}
122+
if (std::find(checkPdgHadron.begin(), checkPdgHadron.end(), std::abs(pdg)) != checkPdgHadron.end()) { // found signal
123+
nSignals++; // count signal PDG
124+
125+
std::vector<int> pdgsDecay{};
126+
std::vector<int> pdgsDecayAntiPart{};
127+
for (int j{track.getFirstDaughterTrackId()}; j <= track.getLastDaughterTrackId(); ++j) {
128+
auto pdgDau = tracks->at(j).GetPdgCode();
129+
pdgsDecay.push_back(pdgDau);
130+
if (pdgDau != 333) { // phi is antiparticle of itself
131+
pdgsDecayAntiPart.push_back(-pdgDau);
132+
} else {
133+
pdgsDecayAntiPart.push_back(pdgDau);
134+
}
135+
}
136+
137+
std::sort(pdgsDecay.begin(), pdgsDecay.end());
138+
std::sort(pdgsDecayAntiPart.begin(), pdgsDecayAntiPart.end());
139+
140+
for (auto &decay : checkHadronDecays[std::abs(pdg)]) {
141+
if (pdgsDecay == decay || pdgsDecayAntiPart == decay) {
142+
nSignalGoodDecay++;
143+
break;
144+
}
145+
}
146+
}
147+
}
148+
}
149+
150+
std::cout << "--------------------------------\n";
151+
std::cout << "# Events: " << nEvents << "\n";
152+
std::cout << "# MB events: " << nEventsMB << "\n";
153+
std::cout << Form("# events injected with %d quark pair: ", checkPdgQuarkOne) << nEventsInjOne << "\n";
154+
std::cout << Form("# events injected with %d quark pair: ", checkPdgQuarkTwo) << nEventsInjTwo << "\n";
155+
std::cout << Form("# %d (anti)quarks: ", checkPdgQuarkOne) << nQuarksOne << "\n";
156+
std::cout << Form("# %d (anti)quarks: ", checkPdgQuarkTwo) << nQuarksTwo << "\n";
157+
std::cout <<"# signal hadrons: " << nSignals << "\n";
158+
std::cout <<"# signal hadrons decaying in the correct channel: " << nSignalGoodDecay << "\n";
159+
160+
if (nEventsMB < nEvents * (1 - ratioTrigger) * 0.95 || nEventsMB > nEvents * (1 - ratioTrigger) * 1.05) { // we put some tolerance since the number of generated events is small
161+
std::cerr << "Number of generated MB events different than expected\n";
162+
return 1;
163+
}
164+
if (nEventsInjOne < nEvents * ratioTrigger * 0.5 * 0.95 || nEventsInjOne > nEvents * ratioTrigger * 0.5 * 1.05) {
165+
std::cerr << "Number of generated events injected with " << checkPdgQuarkOne << " different than expected\n";
166+
return 1;
167+
}
168+
if (nEventsInjTwo < nEvents * ratioTrigger * 0.5 * 0.95 || nEventsInjTwo > nEvents * ratioTrigger * 0.5 * 1.05) {
169+
std::cerr << "Number of generated events injected with " << checkPdgQuarkTwo << " different than expected\n";
170+
return 1;
171+
}
172+
173+
if (nQuarksOne < nEvents * ratioTrigger) { // we expect anyway more because the same quark is repeated several time, after each gluon radiation
174+
std::cerr << "Number of generated (anti)quarks " << checkPdgQuarkOne << " lower than expected\n";
175+
return 1;
176+
}
177+
if (nQuarksTwo < nEvents * ratioTrigger) { // we expect anyway more because the same quark is repeated several time, after each gluon radiation
178+
std::cerr << "Number of generated (anti)quarks " << checkPdgQuarkTwo << " lower than expected\n";
179+
return 1;
180+
}
181+
182+
float fracForcedDecays = float(nSignalGoodDecay) / nSignals;
183+
if (fracForcedDecays < 0.9) { // we put some tolerance (e.g. due to oscillations which might change the final state)
184+
std::cerr << "Fraction of signals decaying into the correct channel " << fracForcedDecays << " lower than expected\n";
185+
return 1;
186+
}
187+
188+
return 0;
189+
}

0 commit comments

Comments
 (0)