Skip to content

Commit 680ec06

Browse files
author
Prottay Das
committed
updated the task with acceptance corrections
1 parent 6292a2e commit 680ec06

File tree

1 file changed

+39
-29
lines changed

1 file changed

+39
-29
lines changed

PWGLF/Tasks/Strangeness/lambdapolsp.cxx

Lines changed: 39 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,8 @@ struct lambdapolsp {
7373
Configurable<bool> additionalEvSel{"additionalEvSel", false, "additionalEvSel"};
7474
Configurable<bool> additionalEvSel2{"additionalEvSel2", false, "additionalEvSel2"};
7575
Configurable<bool> additionalEvSel3{"additionalEvSel3", false, "additionalEvSel3"};
76-
Configurable<bool> correction{"correction", false, "fill histograms including corrections"};
76+
Configurable<bool> correction1{"correction1", false, "fill histograms including corrections 1"};
77+
Configurable<bool> correction2{"correction2", false, "fill histograms including corrections 2"};
7778
Configurable<bool> QA{"QA", false, "flag for QA"};
7879
Configurable<bool> mycut{"mycut", false, "select tracks based on my cuts"};
7980
Configurable<bool> tofhit{"tofhit", true, "select tracks based on tof hit"};
@@ -143,6 +144,7 @@ struct lambdapolsp {
143144
Configurable<int> spNbins{"spNbins", 2000, "Number of bins in sp"};
144145
Configurable<float> lbinsp{"lbinsp", -1.0, "lower bin value in sp histograms"};
145146
Configurable<float> hbinsp{"hbinsp", 1.0, "higher bin value in sp histograms"};
147+
Configurable<int> phiNbins{"phiNbins", 30, "Number of bins in phi"};
146148

147149
ConfigurableAxis configcentAxis{"configcentAxis", {VARIABLE_WIDTH, 0.0, 10.0, 40.0, 80.0}, "Cent V0M"};
148150
ConfigurableAxis configthnAxispT{"configthnAxisPt", {VARIABLE_WIDTH, 0.2, 0.5, 1.0, 1.5, 2.0, 2.5, 3.0, 4.0, 5.0, 6.5, 8.0, 10.0, 100.0}, "#it{p}_{T} (GeV/#it{c})"};
@@ -154,6 +156,7 @@ struct lambdapolsp {
154156
void init(o2::framework::InitContext&)
155157
{
156158
AxisSpec thnAxispT{ptNbins, lbinpt, hbinpt, "#it{p}_{T} (GeV/#it{c})"};
159+
AxisSpec thnAxisRap{100, -0.5, 0.5, "Rapidity"};
157160
AxisSpec thnAxisres{resNbins, lbinres, hbinres, "Reso"};
158161
AxisSpec thnAxisInvMass{IMNbins, lbinIM, hbinIM, "#it{M} (GeV/#it{c}^{2})"};
159162
AxisSpec thnAxisPol{PolNbins, lbinPol, hbinPol, "Sin(#phi - #psi)"};
@@ -162,7 +165,7 @@ struct lambdapolsp {
162165
AxisSpec etaAxis = {etaNbins, lbineta, hbineta, "Eta"};
163166
AxisSpec spAxis = {spNbins, lbinsp, hbinsp, "Sp"};
164167
AxisSpec qxZDCAxis = {QxyNbins, lbinQxy, hbinQxy, "Qx"};
165-
// AxisSpec psiACAxis = {120, -6.28, 6.28, "psiAC"};
168+
AxisSpec phiAxis = {phiNbins, 0.0, 6.28, "phi-phiStar"};
166169

167170
if (checkwithpub) {
168171
if (useprofile == 1) {
@@ -229,22 +232,22 @@ struct lambdapolsp {
229232
histos.add("hpCosPsiC", "hpCosPsiC", HistType::kTHnSparseF, {centAxis, thnAxisres});
230233
histos.add("hpSinPsiA", "hpSinPsiA", HistType::kTHnSparseF, {centAxis, thnAxisres});
231234
histos.add("hpSinPsiC", "hpSinPsiC", HistType::kTHnSparseF, {centAxis, thnAxisres});
232-
histos.add("hcentQxZDCA", "hcentQxZDCA", kTH2F, {{centAxis}, {qxZDCAxis}});
235+
/*histos.add("hcentQxZDCA", "hcentQxZDCA", kTH2F, {{centAxis}, {qxZDCAxis}});
233236
histos.add("hcentQyZDCA", "hcentQyZDCA", kTH2F, {{centAxis}, {qxZDCAxis}});
234237
histos.add("hcentQxZDCC", "hcentQxZDCC", kTH2F, {{centAxis}, {qxZDCAxis}});
235-
histos.add("hcentQyZDCC", "hcentQyZDCC", kTH2F, {{centAxis}, {qxZDCAxis}});
238+
histos.add("hcentQyZDCC", "hcentQyZDCC", kTH2F, {{centAxis}, {qxZDCAxis}});*/
236239
}
237-
if (!correction) {
238-
histos.add("hSparseLambdaPolA", "hSparseLambdaPolA", HistType::kTHnSparseF, {thnAxisInvMass, thnAxispT, thnAxisPol, centAxis}, true);
239-
histos.add("hSparseLambdaPolC", "hSparseLambdaPolC", HistType::kTHnSparseF, {thnAxisInvMass, thnAxispT, thnAxisPol, centAxis}, true);
240-
histos.add("hSparseAntiLambdaPolA", "hSparseAntiLambdaPolA", HistType::kTHnSparseF, {thnAxisInvMass, thnAxispT, thnAxisPol, centAxis}, true);
241-
histos.add("hSparseAntiLambdaPolC", "hSparseAntiLambdaPolC", HistType::kTHnSparseF, {thnAxisInvMass, thnAxispT, thnAxisPol, centAxis}, true);
240+
if (!correction1 && !correction2) {
241+
histos.add("hSparseLambdaPolA", "hSparseLambdaPolA", HistType::kTHnSparseF, {thnAxisInvMass, thnAxispT, etaAxis, thnAxisPol, centAxis, phiAxis}, true);
242+
histos.add("hSparseLambdaPolC", "hSparseLambdaPolC", HistType::kTHnSparseF, {thnAxisInvMass, thnAxispT, etaAxis, thnAxisPol, centAxis, phiAxis}, true);
243+
histos.add("hSparseAntiLambdaPolA", "hSparseAntiLambdaPolA", HistType::kTHnSparseF, {thnAxisInvMass, thnAxispT, etaAxis, thnAxisPol, centAxis, phiAxis}, true);
244+
histos.add("hSparseAntiLambdaPolC", "hSparseAntiLambdaPolC", HistType::kTHnSparseF, {thnAxisInvMass, thnAxispT, etaAxis, thnAxisPol, centAxis, phiAxis}, true);
242245
}
243-
if (correction) {
244-
histos.add("hSparseLambdaPolA_corr", "hSparseLambdaPolA_corr", HistType::kTHnSparseF, {thnAxisInvMass, thnAxispT, thnAxisCosThetaStar, thnAxisPol, thnAxisPol, thnAxisPol, centAxis}, true);
245-
histos.add("hSparseLambdaPolC_corr", "hSparseLambdaPolC_corr", HistType::kTHnSparseF, {thnAxisInvMass, thnAxispT, thnAxisCosThetaStar, thnAxisPol, thnAxisPol, thnAxisPol, centAxis}, true);
246-
histos.add("hSparseAntiLambdaPolA_corr", "hSparseAntiLambdaPolA_corr", HistType::kTHnSparseF, {thnAxisInvMass, thnAxispT, thnAxisCosThetaStar, thnAxisPol, thnAxisPol, thnAxisPol, centAxis}, true);
247-
histos.add("hSparseAntiLambdaPolC_corr", "hSparseAntiLambdaPolC_corr", HistType::kTHnSparseF, {thnAxisInvMass, thnAxispT, thnAxisCosThetaStar, thnAxisPol, thnAxisPol, thnAxisPol, centAxis}, true);
246+
if (correction1 || correction2) {
247+
histos.add("hSparseLambdaPolA_corr", "hSparseLambdaPolA_corr", HistType::kTHnSparseF, {thnAxisInvMass, thnAxispT, thnAxisCosThetaStar, thnAxisPol, thnAxisPol, thnAxisPol, centAxis, etaAxis}, true);
248+
histos.add("hSparseLambdaPolC_corr", "hSparseLambdaPolC_corr", HistType::kTHnSparseF, {thnAxisInvMass, thnAxispT, thnAxisCosThetaStar, thnAxisPol, thnAxisPol, thnAxisPol, centAxis, etaAxis}, true);
249+
histos.add("hSparseAntiLambdaPolA_corr", "hSparseAntiLambdaPolA_corr", HistType::kTHnSparseF, {thnAxisInvMass, thnAxispT, thnAxisCosThetaStar, thnAxisPol, thnAxisPol, thnAxisPol, centAxis, etaAxis}, true);
250+
histos.add("hSparseAntiLambdaPolC_corr", "hSparseAntiLambdaPolC_corr", HistType::kTHnSparseF, {thnAxisInvMass, thnAxispT, thnAxisCosThetaStar, thnAxisPol, thnAxisPol, thnAxisPol, centAxis, etaAxis}, true);
248251
}
249252
}
250253
}
@@ -395,8 +398,7 @@ struct lambdapolsp {
395398
return;
396399
}
397400
auto centrality = collision.centFT0C();
398-
399-
histos.fill(HIST("hCentrality0"), centrality);
401+
// histos.fill(HIST("hCentrality0"), centrality);
400402
if (!collision.triggereventsp()) {
401403
return;
402404
}
@@ -432,10 +434,10 @@ struct lambdapolsp {
432434
histos.fill(HIST("hpCosPsiC"), centrality, (TMath::Cos(GetPhiInRange(psiZDCC))));
433435
histos.fill(HIST("hpSinPsiA"), centrality, (TMath::Sin(GetPhiInRange(psiZDCA))));
434436
histos.fill(HIST("hpSinPsiC"), centrality, (TMath::Sin(GetPhiInRange(psiZDCC))));
435-
histos.fill(HIST("hcentQxZDCA"), centrality, qxZDCA);
437+
/*histos.fill(HIST("hcentQxZDCA"), centrality, qxZDCA);
436438
histos.fill(HIST("hcentQyZDCA"), centrality, qyZDCA);
437439
histos.fill(HIST("hcentQxZDCC"), centrality, qxZDCC);
438-
histos.fill(HIST("hcentQyZDCC"), centrality, qyZDCC);
440+
histos.fill(HIST("hcentQyZDCC"), centrality, qyZDCC);*/
439441
}
440442
}
441443

@@ -574,30 +576,38 @@ struct lambdapolsp {
574576
// histos.fill(HIST("hphiminuspsiC"), (phiminuspsiC));
575577
// auto cosThetaStar = eventplaneVecNorm.Dot(threeVecDauCM) / std::sqrt(threeVecDauCM.Mag2()) / std::sqrt(eventplaneVecNorm.Mag2());
576578
auto cosThetaStar = fourVecDauCM.Pz() / fourVecDauCM.P(); // A0 correction
579+
auto sinThetaStar = TMath::Sqrt(1 - (cosThetaStar * cosThetaStar));
577580
auto PolC = TMath::Sin(phiminuspsiC);
578581
auto PolA = TMath::Sin(phiminuspsiA);
579582

580583
// needed for corrections
581584
auto sinPhiStar = TMath::Sin(GetPhiInRange(phiangle));
582585
auto cosPhiStar = TMath::Cos(GetPhiInRange(phiangle));
583-
// auto sinThetaStarcosphiphiStar=sinThetaStar*TMath::Cos(2* GetPhiInRange((Lambda.Phi()-phiangle))); //A2 correction
586+
auto sinThetaStarcosphiphiStar = sinThetaStar * TMath::Cos(2 * GetPhiInRange((Lambda.Phi() - phiangle))); // A2 correction
587+
auto phi_phiStar = GetPhiInRange(Lambda.Phi() - GetPhiInRange(phiangle));
584588

585589
if (LambdaTag) {
586-
if (correction) {
587-
histos.fill(HIST("hSparseLambdaPolA_corr"), v0.mLambda(), v0.pt(), cosThetaStar, sinPhiStar, cosPhiStar, PolA, centrality);
588-
histos.fill(HIST("hSparseLambdaPolC_corr"), v0.mLambda(), v0.pt(), cosThetaStar, sinPhiStar, cosPhiStar, PolC, centrality);
590+
if (correction1) {
591+
histos.fill(HIST("hSparseLambdaPolA_corr"), v0.mLambda(), v0.pt(), cosThetaStar, sinPhiStar, cosPhiStar, PolA, centrality, v0.eta());
592+
histos.fill(HIST("hSparseLambdaPolC_corr"), v0.mLambda(), v0.pt(), cosThetaStar, sinPhiStar, cosPhiStar, PolC, centrality, v0.eta());
593+
} else if (correction2) {
594+
histos.fill(HIST("hSparseLambdaPolA_corr"), v0.mLambda(), v0.pt(), sinThetaStar, sinThetaStarcosphiphiStar, PolA, centrality, v0.eta());
595+
histos.fill(HIST("hSparseLambdaPolC_corr"), v0.mLambda(), v0.pt(), sinThetaStar, sinThetaStarcosphiphiStar, PolC, centrality, v0.eta());
589596
} else {
590-
histos.fill(HIST("hSparseLambdaPolA"), v0.mLambda(), v0.pt(), PolA, centrality);
591-
histos.fill(HIST("hSparseLambdaPolC"), v0.mLambda(), v0.pt(), PolC, centrality);
597+
histos.fill(HIST("hSparseLambdaPolA"), v0.mLambda(), v0.pt(), v0.eta(), PolA, centrality, (phi_phiStar));
598+
histos.fill(HIST("hSparseLambdaPolC"), v0.mLambda(), v0.pt(), v0.eta(), PolC, centrality, (phi_phiStar));
592599
}
593600
}
594601
if (aLambdaTag) {
595-
if (correction) {
596-
histos.fill(HIST("hSparseAntiLambdaPolA_corr"), v0.mAntiLambda(), v0.pt(), cosThetaStar, sinPhiStar, cosPhiStar, PolA, centrality);
597-
histos.fill(HIST("hSparseAntiLambdaPolC_corr"), v0.mAntiLambda(), v0.pt(), cosThetaStar, sinPhiStar, cosPhiStar, PolC, centrality);
602+
if (correction1) {
603+
histos.fill(HIST("hSparseAntiLambdaPolA_corr"), v0.mAntiLambda(), v0.pt(), cosThetaStar, sinPhiStar, cosPhiStar, PolA, centrality, v0.eta());
604+
histos.fill(HIST("hSparseAntiLambdaPolC_corr"), v0.mAntiLambda(), v0.pt(), cosThetaStar, sinPhiStar, cosPhiStar, PolC, centrality, v0.eta());
605+
} else if (correction2) {
606+
histos.fill(HIST("hSparseAntiLambdaPolA_corr"), v0.mAntiLambda(), v0.pt(), sinThetaStar, sinThetaStarcosphiphiStar, PolA, centrality, v0.eta());
607+
histos.fill(HIST("hSparseAntiLambdaPolC_corr"), v0.mAntiLambda(), v0.pt(), sinThetaStar, sinThetaStarcosphiphiStar, PolC, centrality, v0.eta());
598608
} else {
599-
histos.fill(HIST("hSparseAntiLambdaPolA"), v0.mAntiLambda(), v0.pt(), PolA, centrality);
600-
histos.fill(HIST("hSparseAntiLambdaPolC"), v0.mAntiLambda(), v0.pt(), PolC, centrality);
609+
histos.fill(HIST("hSparseAntiLambdaPolA"), v0.mAntiLambda(), v0.pt(), v0.eta(), PolA, centrality, (phi_phiStar));
610+
histos.fill(HIST("hSparseAntiLambdaPolC"), v0.mAntiLambda(), v0.pt(), v0.eta(), PolC, centrality, (phi_phiStar));
601611
}
602612
}
603613
}

0 commit comments

Comments
 (0)