Skip to content

Commit e1875e6

Browse files
sarjeetagamisarjeeta gami
andauthored
[PWGLF] added scalar product method (#9210)
Co-authored-by: sarjeeta gami <sarjeetagami@sarjeetas-MacBook-Air.local>
1 parent e01a9ea commit e1875e6

File tree

1 file changed

+56
-7
lines changed

1 file changed

+56
-7
lines changed

PWGLF/Tasks/Resonances/kstarpbpb.cxx

Lines changed: 56 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -115,6 +115,7 @@ struct kstarpbpb {
115115
Configurable<bool> like{"like", true, "fill rotation"};
116116
Configurable<bool> fillOccupancy{"fillOccupancy", false, "fill Occupancy"};
117117
Configurable<int> cfgOccupancyCut{"cfgOccupancyCut", 500, "Occupancy cut"};
118+
Configurable<bool> useSP{"useSP", true, "useSP"};
118119

119120
Filter collisionFilter = nabs(aod::collision::posZ) < cfgCutVertex;
120121
Filter centralityFilter = nabs(aod::cent::centFT0C) < cfgCutCentrality;
@@ -171,6 +172,9 @@ struct kstarpbpb {
171172
histos.add("ResFT0CTPC", "ResFT0CTPC", kTH2F, {centAxis, resAxis});
172173
histos.add("ResFT0CFT0A", "ResFT0CFT0A", kTH2F, {centAxis, resAxis});
173174
histos.add("ResFT0ATPC", "ResFT0ATPC", kTH2F, {centAxis, resAxis});
175+
histos.add("ResFT0CTPCSP", "ResFT0CTPCSP", kTH2F, {centAxis, resAxis});
176+
histos.add("ResFT0CFT0ASP", "ResFT0CFT0ASP", kTH2F, {centAxis, resAxis});
177+
histos.add("ResFT0ATPCSP", "ResFT0ATPCSP", kTH2F, {centAxis, resAxis});
174178
}
175179
if (additionalQAplots) {
176180
// DCA QA
@@ -387,6 +391,7 @@ struct kstarpbpb {
387391
ConfigurableAxis axisVertex{"axisVertex", {20, -10, 10}, "vertex axis for bin"};
388392
ConfigurableAxis axisMultiplicityClass{"axisMultiplicityClass", {20, 0, 100}, "multiplicity percentile for bin"};
389393
ConfigurableAxis axisEPAngle{"axisEPAngle", {9, -TMath::Pi() / 2, TMath::Pi() / 2}, "event plane angle"};
394+
double v2, v2Rot;
390395

391396
using BinningTypeVertexContributor = ColumnBinningPolicy<aod::collision::PosZ, aod::cent::CentFT0C, aod::epcalibrationtable::PsiFT0C>;
392397
ROOT::Math::PxPyPzMVector KstarMother, daughter1, daughter2, kaonrot, kstarrot;
@@ -402,6 +407,9 @@ struct kstarpbpb {
402407
auto psiFT0C = collision.psiFT0C();
403408
auto psiFT0A = collision.psiFT0A();
404409
auto psiTPC = collision.psiTPC();
410+
auto QFT0C = collision.qFT0C();
411+
auto QFT0A = collision.qFT0A();
412+
auto QTPC = collision.qTPC();
405413
if (fillOccupancy && occupancy >= cfgOccupancyCut) {
406414
return;
407415
}
@@ -416,6 +424,9 @@ struct kstarpbpb {
416424
histos.fill(HIST("ResFT0CTPC"), centrality, TMath::Cos(2.0 * (psiFT0C - psiTPC)));
417425
histos.fill(HIST("ResFT0CFT0A"), centrality, TMath::Cos(2.0 * (psiFT0C - psiFT0A)));
418426
histos.fill(HIST("ResFT0ATPC"), centrality, TMath::Cos(2.0 * (psiTPC - psiFT0A)));
427+
histos.fill(HIST("ResFT0CTPCSP"), centrality, QFT0C * QTPC * TMath::Cos(2.0 * (psiFT0C - psiTPC)));
428+
histos.fill(HIST("ResFT0CFT0ASP"), centrality, QFT0C * QFT0A * TMath::Cos(2.0 * (psiFT0C - psiFT0A)));
429+
histos.fill(HIST("ResFT0ATPCSP"), centrality, QTPC * QFT0A * TMath::Cos(2.0 * (psiTPC - psiFT0A)));
419430
histos.fill(HIST("hCentrality"), centrality);
420431
histos.fill(HIST("hOccupancy"), occupancy);
421432
histos.fill(HIST("hVtxZ"), collision.posZ());
@@ -461,7 +472,12 @@ struct kstarpbpb {
461472
continue;
462473
}
463474
auto phiminuspsi = GetPhiInRange(KstarMother.Phi() - psiFT0C);
464-
auto v2 = TMath::Cos(2.0 * phiminuspsi);
475+
if (useSP) {
476+
v2 = TMath::Cos(2.0 * phiminuspsi) * QFT0C;
477+
}
478+
if (!useSP) {
479+
v2 = TMath::Cos(2.0 * phiminuspsi);
480+
}
465481
// unlike sign
466482
if (track1.sign() * track2.sign() < 0) {
467483
histos.fill(HIST("hSparseV2SASameEvent_V2"), KstarMother.M(), KstarMother.Pt(), v2, centrality);
@@ -480,7 +496,12 @@ struct kstarpbpb {
480496
continue;
481497
}
482498
auto phiminuspsiRot = GetPhiInRange(kstarrot.Phi() - psiFT0C);
483-
auto v2Rot = TMath::Cos(2.0 * phiminuspsiRot);
499+
if (useSP) {
500+
v2Rot = TMath::Cos(2.0 * phiminuspsiRot) * QFT0C;
501+
}
502+
if (!useSP) {
503+
v2Rot = TMath::Cos(2.0 * phiminuspsiRot);
504+
}
484505
histos.fill(HIST("hSparseV2SASameEventRotational_V2"), kstarrot.M(), kstarrot.Pt(), v2Rot, centrality);
485506
}
486507
}
@@ -506,6 +527,7 @@ struct kstarpbpb {
506527
}
507528
auto centrality = collision.centFT0C();
508529
auto multTPC = collision.multNTracksPV();
530+
auto QFT0C = collision.qFT0C();
509531
if (!collision.triggereventep()) {
510532
return;
511533
}
@@ -652,7 +674,12 @@ struct kstarpbpb {
652674
continue;
653675
}
654676
auto phiminuspsi = GetPhiInRange(KstarMother.Phi() - psiFT0C);
655-
auto v2 = TMath::Cos(2.0 * phiminuspsi);
677+
if (useSP) {
678+
v2 = TMath::Cos(2.0 * phiminuspsi) * QFT0C;
679+
}
680+
if (!useSP) {
681+
v2 = TMath::Cos(2.0 * phiminuspsi);
682+
}
656683
histos.fill(HIST("hSparseV2SASameEvent_V2"), KstarMother.M(), KstarMother.Pt(), v2, centrality);
657684
if (fillRotation) {
658685
for (int nrotbkg = 0; nrotbkg < nBkgRotations; nrotbkg++) {
@@ -679,7 +706,12 @@ struct kstarpbpb {
679706
continue;
680707
}
681708
auto phiminuspsiRot = GetPhiInRange(kstarrot.Phi() - psiFT0C);
682-
auto v2Rot = TMath::Cos(2.0 * phiminuspsiRot);
709+
if (useSP) {
710+
v2Rot = TMath::Cos(2.0 * phiminuspsiRot) * QFT0C;
711+
}
712+
if (!useSP) {
713+
v2Rot = TMath::Cos(2.0 * phiminuspsiRot);
714+
}
683715
histos.fill(HIST("hSparseV2SASameEventRotational_V2"), kstarrot.M(), kstarrot.Pt(), v2Rot, centrality);
684716
}
685717
}
@@ -693,6 +725,7 @@ struct kstarpbpb {
693725
return;
694726
}
695727
auto centrality = collision.centFT0C();
728+
auto QFT0C = collision.qFT0C();
696729
if (!collision.triggereventep()) {
697730
return;
698731
}
@@ -786,7 +819,12 @@ struct kstarpbpb {
786819
continue;
787820
}
788821
auto phiminuspsi = GetPhiInRange(KstarMother.Phi() - psiFT0C);
789-
auto v2 = TMath::Cos(2.0 * phiminuspsi);
822+
if (useSP) {
823+
v2 = TMath::Cos(2.0 * phiminuspsi) * QFT0C;
824+
}
825+
if (!useSP) {
826+
v2 = TMath::Cos(2.0 * phiminuspsi);
827+
}
790828
histos.fill(HIST("hSparseV2SAlikeEventNN_V2"), KstarMother.M(), KstarMother.Pt(), v2, centrality);
791829
}
792830
} else if (track1pion && track2kaon) {
@@ -798,7 +836,12 @@ struct kstarpbpb {
798836
continue;
799837
}
800838
auto phiminuspsi = GetPhiInRange(KstarMother.Phi() - psiFT0C);
801-
auto v2 = TMath::Cos(2.0 * phiminuspsi);
839+
if (useSP) {
840+
v2 = TMath::Cos(2.0 * phiminuspsi) * QFT0C;
841+
}
842+
if (!useSP) {
843+
v2 = TMath::Cos(2.0 * phiminuspsi);
844+
}
802845
histos.fill(HIST("hSparseV2SAlikeEventPP_V2"), KstarMother.M(), KstarMother.Pt(), v2, centrality);
803846
}
804847
}
@@ -842,6 +885,7 @@ struct kstarpbpb {
842885
auto centrality = collision1.centFT0C();
843886
auto centrality2 = collision2.centFT0C();
844887
auto psiFT0C = collision1.psiFT0C();
888+
auto QFT0C = collision1.qFT0C();
845889
bool track1pion = false;
846890
bool track1kaon = false;
847891
bool track2pion = false;
@@ -923,7 +967,12 @@ struct kstarpbpb {
923967
continue;
924968
}
925969
auto phiminuspsi = GetPhiInRange(KstarMother.Phi() - psiFT0C);
926-
auto v2 = TMath::Cos(2.0 * phiminuspsi);
970+
if (useSP) {
971+
v2 = TMath::Cos(2.0 * phiminuspsi) * QFT0C;
972+
}
973+
if (!useSP) {
974+
v2 = TMath::Cos(2.0 * phiminuspsi);
975+
}
927976
histos.fill(HIST("hSparseV2SAMixedEvent_V2"), KstarMother.M(), KstarMother.Pt(), v2, centrality);
928977
}
929978
}

0 commit comments

Comments
 (0)