Skip to content

Commit fb92fad

Browse files
zhengqi-wzhengqi-w
committed
update with FT0Cq2
1 parent 3703f7a commit fb92fad

File tree

1 file changed

+50
-11
lines changed

1 file changed

+50
-11
lines changed

‎PWGCF/Flow/Tasks/flowEsePHe3.cxx‎

Lines changed: 50 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,7 @@ DECLARE_SOA_COLUMN(EseCentFT0C, eseCentFT0C, float);
6464
DECLARE_SOA_COLUMN(EsePsi2FT0C, esePsi2FT0C, float);
6565
DECLARE_SOA_COLUMN(Eseq2Tar, eseq2Tar, float);
6666
DECLARE_SOA_COLUMN(Eseq2Ref, eseq2Ref, float);
67+
DECLARE_SOA_COLUMN(Eseq2FT0C, eseq2FT0C, float);
6768
DECLARE_SOA_COLUMN(EseTarSign, eseTarSign, int8_t);
6869
DECLARE_SOA_COLUMN(EseTarTPCInnerParam, eseTarTPCInnerParam, float);
6970
DECLARE_SOA_COLUMN(EseTarTPCSignal, eseTarTPCSignal, float);
@@ -88,6 +89,7 @@ DECLARE_SOA_TABLE(ESETable, "AOD", "ESETable",
8889
ese_var_table::EsePsi2FT0C,
8990
ese_var_table::Eseq2Tar,
9091
ese_var_table::Eseq2Ref,
92+
ese_var_table::Eseq2FT0C,
9193
ese_var_table::EseTarSign,
9294
ese_var_table::EseTarTPCInnerParam,
9395
ese_var_table::EseTarTPCSignal,
@@ -112,6 +114,7 @@ struct ESECandidate {
112114
float psi2FT0C;
113115
float q2Tar;
114116
float q2Ref;
117+
float q2FT0C;
115118
int8_t signTar;
116119
float tpcInnerParamTar;
117120
float tpcSignalTar;
@@ -297,16 +300,19 @@ std::shared_ptr<TH1> hPIDQATar1D[12];
297300
std::shared_ptr<TH2> hPIDQATar2D[4];
298301
std::shared_ptr<TH3> hPIDQATar3D[3];
299302
std::shared_ptr<TProfile3D> hv2Tar[2];
300-
std::shared_ptr<TH1> hESEQATar1D[4];
301-
std::shared_ptr<TH2> hESEQATar2D;
303+
std::shared_ptr<TH1> hESEQATar1D[8];
304+
std::shared_ptr<TH2> hESEQATar2D[2];
302305
std::shared_ptr<THnSparse> hESETar;
303306
// Ref ptr
304307
std::shared_ptr<TH1> hPIDQARef1D[12];
305308
std::shared_ptr<TH2> hPIDQARef2D[4];
306309
std::shared_ptr<TH3> hPIDQARef3D[3];
307310
std::shared_ptr<TProfile3D> hv2Ref[2];
308-
std::shared_ptr<TH1> hESEQARef1D[4];
309-
std::shared_ptr<TH2> hESEQARef2D;
311+
std::shared_ptr<TH1> hESEQARef1D[8];
312+
std::shared_ptr<TH2> hESEQARef2D[2];
313+
// FT0C general q2 QA
314+
std::shared_ptr<TH1> hESEQAFT0C1D[2];
315+
std::shared_ptr<TH2> hESEQAFT0C2D[2];
310316
} // namespace ese_parameters
311317

312318
using TracksPIDFull = soa::Join<aod::Tracks, aod::TracksExtra, aod::TrackSelection, aod::TracksDCA, aod::TrackSelectionExtension, aod::pidTPCFullPi, aod::pidTPCFullKa, aod::pidTPCFullPr, aod::pidTOFFullPi, aod::pidTOFFullKa, aod::pidTOFFullPr, aod::pidTOFFullDe, aod::pidTOFFullTr, aod::pidTOFFullHe, aod::pidTOFFullAl>;
@@ -396,6 +402,7 @@ struct FlowEsePHe3 {
396402
ConfigurableAxis cfgaxisCent{"cfgaxisCent", {90, 0, 90}, ""};
397403
ConfigurableAxis cfgaxisq2Tar{"cfgaxisq2Tar", {100, 0, 2}, "Binning for q_{2} traget particle"};
398404
ConfigurableAxis cfgaxisq2Ref{"cfgaxisq2Ref", {120, 0, 12}, "Binning for q_{2} reference particle"};
405+
ConfigurableAxis cfgaxisq2FT0C{"cfgaxisq2FT0C", {900, 0, 900}, "Binning for q_{2} FT0C"};
399406
ConfigurableAxis cfgq2NumeratorTar{"cfgq2NumeratorTar", {20, 0, 2}, "q2 Numerator bin for tar particle"};
400407
ConfigurableAxis cfgq2NumeratorRef{"cfgq2NumeratorRef", {100, 0, 10}, "q2 Numerator bin for ref particle"};
401408
ConfigurableAxis cfgq2DenominatorTar{"cfgq2DenominatorTar", {20, 0, 2}, "q2 Denominator bin for tar particle"};
@@ -962,6 +969,7 @@ struct FlowEsePHe3 {
962969
float psi2 = helperEP.GetEventPlane(collision.qvecRe()[detInd + 3], collision.qvecIm()[detInd + 3], 2);
963970
float q2Tarinit{0.f};
964971
float q2Refinit{0.f};
972+
float q2FT0Cinit{0.f};
965973
for (const auto& track : tracks) { // loop on tracks
966974
if (!trackSelBasic(track)) {
967975
continue;
@@ -1016,7 +1024,7 @@ struct FlowEsePHe3 {
10161024
}
10171025
}
10181026
ese_parameters::eseCandidates.emplace_back(ESECandidate{
1019-
collision.posZ(), collision.centFT0C(), psi2, q2Tarinit, q2Refinit, static_cast<int8_t>(track.sign()), correctedTpcInnerParam, track.tpcSignal(), ese_parameters::Charges[poiTar] * track.pt(), track.eta(), track.phi(),
1027+
collision.posZ(), collision.centFT0C(), psi2, q2Tarinit, q2Refinit, q2FT0Cinit, static_cast<int8_t>(track.sign()), correctedTpcInnerParam, track.tpcSignal(), ese_parameters::Charges[poiTar] * track.pt(), track.eta(), track.phi(),
10201028
track.dcaXY(), track.dcaZ(), static_cast<uint8_t>(track.tpcNClsFound()), track.itsNCls(), track.tpcChi2NCl(), track.itsChi2NCl(),
10211029
nSigmaTPCTar, nSigmaTOFTar, nSigmaITSTar, track.itsClusterSizes()});
10221030
}
@@ -1245,19 +1253,33 @@ struct FlowEsePHe3 {
12451253
ese_parameters::hESEQATar1D[1] = histsESE.add<TH1>(Form("ESE/ESEQA/hist_%sq2", cfgTarName.value.c_str()), ";#it{q}_{2};counts", HistType::kTH1F, {cfgaxisq2Tar});
12461254
ese_parameters::hESEQATar1D[2] = histsESE.add<TH1>(Form("ESE/ESEQA/hist_%sq2Numerator", cfgTarName.value.c_str()), ";#it{q}_{2} numerator;counts", HistType::kTH1F, {cfgq2NumeratorTar});
12471255
ese_parameters::hESEQATar1D[3] = histsESE.add<TH1>(Form("ESE/ESEQA/hist_%sq2Denominator", cfgTarName.value.c_str()), ";#it{q}_{2} denominator;counts", HistType::kTH1F, {cfgq2DenominatorTar});
1248-
ese_parameters::hESEQATar2D = histsESE.add<TH2>(Form("ESE/ESEQA/hist_%sq2_Cent", cfgTarName.value.c_str()), ";#it{q}_{2};Centrality (%)", HistType::kTH2F, {cfgaxisq2Tar, cfgaxisCent});
1256+
ese_parameters::hESEQATar1D[4] = histsESE.add<TH1>(Form("ESE/ESEQA/hist_%sNum_Survived", cfgTarName.value.c_str()), ";Num_{Proton}/Event;counts", HistType::kTH1F, {{100, 0, 100}});
1257+
ese_parameters::hESEQATar1D[5] = histsESE.add<TH1>(Form("ESE/ESEQA/hist_%sq2_Survived", cfgTarName.value.c_str()), ";#it{q}_{2};counts", HistType::kTH1F, {cfgaxisq2Tar});
1258+
ese_parameters::hESEQATar1D[6] = histsESE.add<TH1>(Form("ESE/ESEQA/hist_%sq2Numerator_Survived", cfgTarName.value.c_str()), ";#it{q}_{2} numerator;counts", HistType::kTH1F, {cfgq2NumeratorTar});
1259+
ese_parameters::hESEQATar1D[7] = histsESE.add<TH1>(Form("ESE/ESEQA/hist_%sq2Denominator_Survived", cfgTarName.value.c_str()), ";#it{q}_{2} denominator;counts", HistType::kTH1F, {cfgq2DenominatorTar});
1260+
ese_parameters::hESEQATar2D[0] = histsESE.add<TH2>(Form("ESE/ESEQA/hist_%sq2_Cent", cfgTarName.value.c_str()), ";#it{q}_{2};Centrality (%)", HistType::kTH2F, {cfgaxisq2Tar, cfgaxisCent});
1261+
ese_parameters::hESEQATar2D[1] = histsESE.add<TH2>(Form("ESE/ESEQA/hist_%sq2_Cent_Survived", cfgTarName.value.c_str()), ";#it{q}_{2};Centrality (%)", HistType::kTH2F, {cfgaxisq2Tar, cfgaxisCent});
12491262
ese_parameters::hESEQARef1D[0] = histsESE.add<TH1>(Form("ESE/ESEQA/hist_%sNum", cfgRefName.value.c_str()), ";Num_{He3}/Event;counts", HistType::kTH1F, {{10, 0, 10}});
12501263
ese_parameters::hESEQARef1D[1] = histsESE.add<TH1>(Form("ESE/ESEQA/hist_%sq2", cfgRefName.value.c_str()), ";#it{q}_{2};counts", HistType::kTH1F, {cfgaxisq2Ref});
12511264
ese_parameters::hESEQARef1D[2] = histsESE.add<TH1>(Form("ESE/ESEQA/hist_%sq2Numerator", cfgRefName.value.c_str()), ";#it{q}_{2} numerator;counts", HistType::kTH1F, {cfgq2NumeratorRef});
12521265
ese_parameters::hESEQARef1D[3] = histsESE.add<TH1>(Form("ESE/ESEQA/hist_%sq2Denominator", cfgRefName.value.c_str()), ";#it{q}_{2} denominator;counts", HistType::kTH1F, {cfgq2DenominatorRef});
1253-
ese_parameters::hESEQARef2D = histsESE.add<TH2>(Form("ESE/ESEQA/hist_%sq2_Cent", cfgRefName.value.c_str()), ";#it{q}_{2};Centrality (%)", HistType::kTH2F, {cfgaxisq2Ref, cfgaxisCent});
1266+
ese_parameters::hESEQARef1D[4] = histsESE.add<TH1>(Form("ESE/ESEQA/hist_%sNum_Survived", cfgRefName.value.c_str()), ";Num_{He3}/Event;counts", HistType::kTH1F, {{10, 0, 10}});
1267+
ese_parameters::hESEQARef1D[5] = histsESE.add<TH1>(Form("ESE/ESEQA/hist_%sq2_Survived", cfgRefName.value.c_str()), ";#it{q}_{2};counts", HistType::kTH1F, {cfgaxisq2Ref});
1268+
ese_parameters::hESEQARef1D[6] = histsESE.add<TH1>(Form("ESE/ESEQA/hist_%sq2Numerator_Survived", cfgRefName.value.c_str()), ";#it{q}_{2} numerator;counts", HistType::kTH1F, {cfgq2NumeratorRef});
1269+
ese_parameters::hESEQARef1D[7] = histsESE.add<TH1>(Form("ESE/ESEQA/hist_%sq2Denominator_Survived", cfgRefName.value.c_str()), ";#it{q}_{2} denominator;counts", HistType::kTH1F, {cfgq2DenominatorRef});
1270+
ese_parameters::hESEQARef2D[0] = histsESE.add<TH2>(Form("ESE/ESEQA/hist_%sq2_Cent", cfgRefName.value.c_str()), ";#it{q}_{2};Centrality (%)", HistType::kTH2F, {cfgaxisq2Ref, cfgaxisCent});
1271+
ese_parameters::hESEQARef2D[1] = histsESE.add<TH2>(Form("ESE/ESEQA/hist_%sq2_Cent_Survived", cfgRefName.value.c_str()), ";#it{q}_{2};Centrality (%)", HistType::kTH2F, {cfgaxisq2Ref, cfgaxisCent});
1272+
ese_parameters::hESEQAFT0C1D[0] = histsESE.add<TH1>("ESE/ESEQA/hist_q2FT0C", ";#it{q}_{2} (FT0C)", HistType::kTH1F, {cfgaxisq2FT0C});
1273+
ese_parameters::hESEQAFT0C1D[1] = histsESE.add<TH1>("ESE/ESEQA/hist_q2FT0C_Survived", ";#it{q}_{2} (FT0C)", HistType::kTH1F, {cfgaxisq2FT0C});
1274+
ese_parameters::hESEQAFT0C2D[0] = histsESE.add<TH2>("ESE/ESEQA/hist_q2FT0C_Cent", ";#it{q}_{2} (FT0C);Centrality (%)", HistType::kTH2F, {cfgaxisq2FT0C, cfgaxisCent});
1275+
ese_parameters::hESEQAFT0C2D[1] = histsESE.add<TH2>("ESE/ESEQA/hist_q2FT0C_Cent_Survived", ";#it{q}_{2} (FT0C);Centrality (%)", HistType::kTH2F, {cfgaxisq2FT0C, cfgaxisCent});
12541276
}
12551277
if (cfgOpenESE) {
12561278
ese_parameters::hESETar = histsESE.add<THnSparse>(Form("ESE/ESE/histESE_%s", cfgTarName.value.c_str()), ";#it{p}_{T};Centrality (%);#it{q}_{2};cos(#phi-#Psi_{2});Charge", HistType::kTHnSparseF, {cfgaxispt, cfgaxisCent, cfgaxisq2Ref, axisCos, axisCharge});
12571279
}
12581280
}
12591281

1260-
void process(soa::Filtered<soa::Join<aod::Collisions, aod::EvSels, aod::CentFT0Cs, aod::Mults, aod::Qvectors>>::iterator const& collision, TracksPIDFull const& tracks)
1282+
void process(soa::Filtered<soa::Join<aod::Collisions, aod::EvSels, aod::CentFT0Cs, aod::Mults, aod::Qvectors, aod::QvectorFT0CVecs>>::iterator const& collision, TracksPIDFull const& tracks)
12611283
{
12621284
ese_parameters::eseCandidates.clear();
12631285
ese_parameters::eseReferences.clear();
@@ -1269,6 +1291,7 @@ struct FlowEsePHe3 {
12691291
return;
12701292
if (cfgOpenFullEventQA)
12711293
fillEventQAhistAf(collision, multTrk, centrality);
1294+
float q2FT0C{std::sqrt(collision.qvecFT0CReVec()[0] * collision.qvecFT0CReVec()[0] + collision.qvecFT0CImVec()[0] * collision.qvecFT0CImVec()[0]) * std::sqrt(collision.sumAmplFT0C())};
12721295
float q2Tarx{0.};
12731296
float q2Tary{0.};
12741297
float q2Refx{0.};
@@ -1284,12 +1307,14 @@ struct FlowEsePHe3 {
12841307
ese_parameters::hESEQATar1D[1]->Fill(q2Tar);
12851308
ese_parameters::hESEQATar1D[2]->Fill(std::hypot(q2Tarx, q2Tary));
12861309
ese_parameters::hESEQATar1D[3]->Fill(std::sqrt(static_cast<float>(multiTar)));
1287-
ese_parameters::hESEQATar2D->Fill(q2Tar, centrality);
1310+
ese_parameters::hESEQATar2D[0]->Fill(q2Tar, centrality);
12881311
ese_parameters::hESEQARef1D[0]->Fill(multiRef);
12891312
ese_parameters::hESEQARef1D[1]->Fill(q2Ref);
1290-
ese_parameters::hESEQARef2D->Fill(q2Ref, centrality);
12911313
ese_parameters::hESEQARef1D[2]->Fill(std::hypot(q2Refx, q2Refy));
12921314
ese_parameters::hESEQARef1D[3]->Fill(std::sqrt(static_cast<float>(multiRef)));
1315+
ese_parameters::hESEQARef2D[0]->Fill(q2Ref, centrality);
1316+
ese_parameters::hESEQAFT0C1D[0]->Fill(q2FT0C);
1317+
ese_parameters::hESEQAFT0C2D[0]->Fill(q2FT0C, centrality);
12931318
}
12941319
if (cfgOpenv2Ref) {
12951320
for (const auto& c : ese_parameters::eseReferences) {
@@ -1302,8 +1327,22 @@ struct FlowEsePHe3 {
13021327
}
13031328
if (multiTar == 0)
13041329
return;
1330+
if (cfgOpenESEQA) {
1331+
ese_parameters::hESEQATar1D[4]->Fill(multiTar);
1332+
ese_parameters::hESEQATar1D[5]->Fill(q2Tar);
1333+
ese_parameters::hESEQATar1D[6]->Fill(std::hypot(q2Tarx, q2Tary));
1334+
ese_parameters::hESEQATar1D[7]->Fill(std::sqrt(static_cast<float>(multiTar)));
1335+
ese_parameters::hESEQATar2D[1]->Fill(q2Tar, centrality);
1336+
ese_parameters::hESEQARef1D[4]->Fill(multiRef);
1337+
ese_parameters::hESEQARef1D[5]->Fill(q2Ref);
1338+
ese_parameters::hESEQARef1D[6]->Fill(std::hypot(q2Refx, q2Refy));
1339+
ese_parameters::hESEQARef1D[7]->Fill(std::sqrt(static_cast<float>(multiRef)));
1340+
ese_parameters::hESEQARef2D[1]->Fill(q2Ref, centrality);
1341+
ese_parameters::hESEQAFT0C1D[1]->Fill(q2FT0C);
1342+
ese_parameters::hESEQAFT0C2D[1]->Fill(q2FT0C, centrality);
1343+
}
13051344
for (const auto& c : ese_parameters::eseCandidates) {
1306-
eseTable(c.vtz, c.centFT0C, c.psi2FT0C, q2Tar, q2Ref, c.signTar, c.tpcInnerParamTar, c.tpcSignalTar, c.ptTar, c.etaTar, c.phiTar, c.dcaXYTar, c.dcaZTar, c.tpcNclsTar, c.itsNclsTar, c.tpcChi2NDFTar, c.itsChi2NDFTar, c.tpcNSigmaTar, c.tofNSigmaTar, c.itsNSigmaTar, c.itsClusSizeTar);
1345+
eseTable(c.vtz, c.centFT0C, c.psi2FT0C, q2Tar, q2Ref, q2FT0C, c.signTar, c.tpcInnerParamTar, c.tpcSignalTar, c.ptTar, c.etaTar, c.phiTar, c.dcaXYTar, c.dcaZTar, c.tpcNclsTar, c.itsNclsTar, c.tpcChi2NDFTar, c.itsChi2NDFTar, c.tpcNSigmaTar, c.tofNSigmaTar, c.itsNSigmaTar, c.itsClusSizeTar);
13071346
if (cfgOpenESE) {
13081347
ese_parameters::hESETar->Fill(c.ptTar, c.centFT0C, q2Ref, std::cos(2 * (c.phiTar - c.psi2FT0C)), c.signTar);
13091348
}

0 commit comments

Comments
 (0)