Skip to content

Commit 4d9070c

Browse files
fuchuncuifuchuncui
authored
Add files via upload
1 parent 8eb223d commit 4d9070c

File tree

1 file changed

+55
-32
lines changed

1 file changed

+55
-32
lines changed

‎PWGCF/Flow/Tasks/flowGfwOmegaXi.cxx‎

Lines changed: 55 additions & 32 deletions
Original file line numberDiff line numberDiff line change
@@ -230,6 +230,29 @@ struct FlowGfwOmegaXi {
230230
std::vector<int> runNumbers;
231231
std::map<int, std::vector<std::shared_ptr<TH1>>> th1sList;
232232
std::map<int, std::vector<std::shared_ptr<TH3>>> th3sList;
233+
enum OutputTH1Names {
234+
// here are TProfiles for vn-pt correlations that are not implemented in GFW
235+
hPhi = 0,
236+
hPhicorr,
237+
hPhiK0s,
238+
hPhiLambda,
239+
hPhiXi,
240+
hPhiOmega,
241+
hPhiK0scorr,
242+
hPhiLambdacorr,
243+
hPhiXicorr,
244+
hPhiOmegacorr,
245+
kCount_TH1Names
246+
};
247+
248+
enum OutputTH3Names {
249+
hPhiEtaVtxz = 0,
250+
hPhiEtaVtxzK0s,
251+
hPhiEtaVtxzLambda,
252+
hPhiEtaVtxzXi,
253+
hPhiEtaVtxzOmega,
254+
kCount_TH3Names
255+
};
233256

234257
std::vector<GFWWeights*> mAcceptance;
235258
std::vector<TH1D*> mEfficiency;
@@ -337,25 +360,25 @@ struct FlowGfwOmegaXi {
337360
if (cfgOutputrunbyrun) {
338361
runNumbers = cfgRunNumbers;
339362
for (const auto& runNumber : runNumbers) {
340-
std::vector<std::shared_ptr<TH1>> histosPhi;
341-
histosPhi.push_back(registry.add<TH1>(Form("%d/hPhi", runNumber), "", {HistType::kTH1D, {cfgaxisPhi}}));
342-
histosPhi.push_back(registry.add<TH1>(Form("%d/hPhicorr", runNumber), "", {HistType::kTH1D, {cfgaxisPhi}}));
343-
histosPhi.push_back(registry.add<TH1>(Form("%d/hPhiK0s", runNumber), "", {HistType::kTH1D, {cfgaxisPhi}}));
344-
histosPhi.push_back(registry.add<TH1>(Form("%d/hPhiLambda", runNumber), "", {HistType::kTH1D, {cfgaxisPhi}}));
345-
histosPhi.push_back(registry.add<TH1>(Form("%d/hPhiXi", runNumber), "", {HistType::kTH1D, {cfgaxisPhi}}));
346-
histosPhi.push_back(registry.add<TH1>(Form("%d/hPhiOmega", runNumber), "", {HistType::kTH1D, {cfgaxisPhi}}));
347-
histosPhi.push_back(registry.add<TH1>(Form("%d/hPhiK0scorr", runNumber), "", {HistType::kTH1D, {cfgaxisPhi}}));
348-
histosPhi.push_back(registry.add<TH1>(Form("%d/hPhiLambdacorr", runNumber), "", {HistType::kTH1D, {cfgaxisPhi}}));
349-
histosPhi.push_back(registry.add<TH1>(Form("%d/hPhiXicorr", runNumber), "", {HistType::kTH1D, {cfgaxisPhi}}));
350-
histosPhi.push_back(registry.add<TH1>(Form("%d/hPhiOmegacorr", runNumber), "", {HistType::kTH1D, {cfgaxisPhi}}));
363+
std::vector<std::shared_ptr<TH1>> histosPhi(kCount_TH1Names);
364+
histosPhi[hPhi] = registry.add<TH1>(Form("%d/hPhi", runNumber), "", {HistType::kTH1D, {cfgaxisPhi}});
365+
histosPhi[hPhicorr] = registry.add<TH1>(Form("%d/hPhicorr", runNumber), "", {HistType::kTH1D, {cfgaxisPhi}});
366+
histosPhi[hPhiK0s] = registry.add<TH1>(Form("%d/hPhiK0s", runNumber), "", {HistType::kTH1D, {cfgaxisPhi}});
367+
histosPhi[hPhiLambda] = registry.add<TH1>(Form("%d/hPhiLambda", runNumber), "", {HistType::kTH1D, {cfgaxisPhi}});
368+
histosPhi[hPhiXi] = registry.add<TH1>(Form("%d/hPhiXi", runNumber), "", {HistType::kTH1D, {cfgaxisPhi}});
369+
histosPhi[hPhiOmega] = registry.add<TH1>(Form("%d/hPhiOmega", runNumber), "", {HistType::kTH1D, {cfgaxisPhi}});
370+
histosPhi[hPhiK0scorr] = registry.add<TH1>(Form("%d/hPhiK0scorr", runNumber), "", {HistType::kTH1D, {cfgaxisPhi}});
371+
histosPhi[hPhiLambdacorr] = registry.add<TH1>(Form("%d/hPhiLambdacorr", runNumber), "", {HistType::kTH1D, {cfgaxisPhi}});
372+
histosPhi[hPhiXicorr] = registry.add<TH1>(Form("%d/hPhiXicorr", runNumber), "", {HistType::kTH1D, {cfgaxisPhi}});
373+
histosPhi[hPhiOmegacorr] = registry.add<TH1>(Form("%d/hPhiOmegacorr", runNumber), "", {HistType::kTH1D, {cfgaxisPhi}});
351374
th1sList.insert(std::make_pair(runNumber, histosPhi));
352375

353-
std::vector<std::shared_ptr<TH3>> nuaTH3;
354-
nuaTH3.push_back(registry.add<TH3>(Form("%d/hPhiEtaVtxz", runNumber), ";#varphi;#eta;v_{z}", {HistType::kTH3D, {cfgaxisPhi, {64, -1.6, 1.6}, cfgaxisVertex}}));
355-
nuaTH3.push_back(registry.add<TH3>(Form("%d/hPhiEtaVtxzK0s", runNumber), ";#varphi;#eta;v_{z}", {HistType::kTH3D, {cfgaxisPhi, {64, -1.6, 1.6}, cfgaxisVertex}}));
356-
nuaTH3.push_back(registry.add<TH3>(Form("%d/hPhiEtaVtxzLambda", runNumber), ";#varphi;#eta;v_{z}", {HistType::kTH3D, {cfgaxisPhi, {64, -1.6, 1.6}, cfgaxisVertex}}));
357-
nuaTH3.push_back(registry.add<TH3>(Form("%d/hPhiEtaVtxzXi", runNumber), ";#varphi;#eta;v_{z}", {HistType::kTH3D, {cfgaxisPhi, {64, -1.6, 1.6}, cfgaxisVertex}}));
358-
nuaTH3.push_back(registry.add<TH3>(Form("%d/hPhiEtaVtxzOmega", runNumber), ";#varphi;#eta;v_{z}", {HistType::kTH3D, {cfgaxisPhi, {64, -1.6, 1.6}, cfgaxisVertex}}));
376+
std::vector<std::shared_ptr<TH3>> nuaTH3(kCount_TH3Names);
377+
nuaTH3[hPhiEtaVtxz] = registry.add<TH3>(Form("%d/hPhiEtaVtxz", runNumber), ";#varphi;#eta;v_{z}", {HistType::kTH3D, {cfgaxisPhi, {64, -1.6, 1.6}, cfgaxisVertex}});
378+
nuaTH3[hPhiEtaVtxzK0s] = registry.add<TH3>(Form("%d/hPhiEtaVtxzK0s", runNumber), ";#varphi;#eta;v_{z}", {HistType::kTH3D, {cfgaxisPhi, {64, -1.6, 1.6}, cfgaxisVertex}});
379+
nuaTH3[hPhiEtaVtxzLambda] = registry.add<TH3>(Form("%d/hPhiEtaVtxzLambda", runNumber), ";#varphi;#eta;v_{z}", {HistType::kTH3D, {cfgaxisPhi, {64, -1.6, 1.6}, cfgaxisVertex}});
380+
nuaTH3[hPhiEtaVtxzXi] = registry.add<TH3>(Form("%d/hPhiEtaVtxzXi", runNumber), ";#varphi;#eta;v_{z}", {HistType::kTH3D, {cfgaxisPhi, {64, -1.6, 1.6}, cfgaxisVertex}});
381+
nuaTH3[hPhiEtaVtxzOmega] = registry.add<TH3>(Form("%d/hPhiEtaVtxzOmega", runNumber), ";#varphi;#eta;v_{z}", {HistType::kTH3D, {cfgaxisPhi, {64, -1.6, 1.6}, cfgaxisVertex}});
359382
th3sList.insert(std::make_pair(runNumber, nuaTH3));
360383
}
361384
}
@@ -992,9 +1015,9 @@ struct FlowGfwOmegaXi {
9921015
fWeightsREF->fill(track.phi(), track.eta(), vtxz, track.pt(), cent, 0);
9931016

9941017
if (cfgOutputrunbyrun) {
995-
th1sList[runNumber][0]->Fill(track.phi());
996-
th1sList[runNumber][1]->Fill(track.phi(), wacc);
997-
th3sList[runNumber][0]->Fill(track.phi(), track.eta(), vtxz);
1018+
th1sList[runNumber][hPhi]->Fill(track.phi());
1019+
th1sList[runNumber][hPhicorr]->Fill(track.phi(), wacc);
1020+
th3sList[runNumber][hPhiEtaVtxz]->Fill(track.phi(), track.eta(), vtxz);
9981021
}
9991022
}
10001023
if (cfgDoLocDenCorr) {
@@ -1123,9 +1146,9 @@ struct FlowGfwOmegaXi {
11231146
if (cfgOutputNUAWeights)
11241147
fWeightsK0s->fill(v0.phi(), v0.eta(), vtxz, v0.pt(), cent, 0);
11251148
if (cfgOutputrunbyrun) {
1126-
th1sList[runNumber][2]->Fill(v0.phi());
1127-
th1sList[runNumber][6]->Fill(v0.phi(), wacc);
1128-
th3sList[runNumber][1]->Fill(v0.phi(), v0.eta(), vtxz);
1149+
th1sList[runNumber][hPhiK0s]->Fill(v0.phi());
1150+
th1sList[runNumber][hPhiK0scorr]->Fill(v0.phi(), wacc);
1151+
th3sList[runNumber][hPhiEtaVtxzK0s]->Fill(v0.phi(), v0.eta(), vtxz);
11291152
}
11301153
}
11311154
if (isLambda) {
@@ -1147,9 +1170,9 @@ struct FlowGfwOmegaXi {
11471170
if (cfgOutputNUAWeights)
11481171
fWeightsLambda->fill(v0.phi(), v0.eta(), vtxz, v0.pt(), cent, 0);
11491172
if (cfgOutputrunbyrun) {
1150-
th1sList[runNumber][3]->Fill(v0.phi());
1151-
th1sList[runNumber][7]->Fill(v0.phi(), wacc);
1152-
th3sList[runNumber][2]->Fill(v0.phi(), v0.eta(), vtxz);
1173+
th1sList[runNumber][hPhiLambda]->Fill(v0.phi());
1174+
th1sList[runNumber][hPhiLambdacorr]->Fill(v0.phi(), wacc);
1175+
th3sList[runNumber][hPhiEtaVtxzLambda]->Fill(v0.phi(), v0.eta(), vtxz);
11531176
}
11541177
}
11551178
}
@@ -1317,9 +1340,9 @@ struct FlowGfwOmegaXi {
13171340
if (cfgOutputNUAWeights)
13181341
fWeightsOmega->fill(casc.phi(), casc.eta(), vtxz, casc.pt(), cent, 0);
13191342
if (cfgOutputrunbyrun) {
1320-
th1sList[runNumber][5]->Fill(casc.phi());
1321-
th1sList[runNumber][9]->Fill(casc.phi(), wacc);
1322-
th3sList[runNumber][4]->Fill(casc.phi(), casc.eta(), vtxz);
1343+
th1sList[runNumber][hPhiOmega]->Fill(casc.phi());
1344+
th1sList[runNumber][hPhiOmegacorr]->Fill(casc.phi(), wacc);
1345+
th3sList[runNumber][hPhiEtaVtxzOmega]->Fill(casc.phi(), casc.eta(), vtxz);
13231346
}
13241347
}
13251348
if (isXi) {
@@ -1343,9 +1366,9 @@ struct FlowGfwOmegaXi {
13431366
if (cfgOutputNUAWeights)
13441367
fWeightsXi->fill(casc.phi(), casc.eta(), vtxz, casc.pt(), cent, 0);
13451368
if (cfgOutputrunbyrun) {
1346-
th1sList[runNumber][4]->Fill(casc.phi());
1347-
th1sList[runNumber][8]->Fill(casc.phi(), wacc);
1348-
th3sList[runNumber][3]->Fill(casc.phi(), casc.eta(), vtxz);
1369+
th1sList[runNumber][hPhiXi]->Fill(casc.phi());
1370+
th1sList[runNumber][hPhiXicorr]->Fill(casc.phi(), wacc);
1371+
th3sList[runNumber][hPhiEtaVtxzXi]->Fill(casc.phi(), casc.eta(), vtxz);
13491372
}
13501373
}
13511374
}

0 commit comments

Comments
 (0)