Skip to content
27 changes: 17 additions & 10 deletions PWGLF/Tasks/Resonances/f1protoncorrelation.cxx
Original file line number Diff line number Diff line change
Expand Up @@ -292,9 +292,9 @@ struct f1protoncorrelation {
if (f1track.f1d1TOFHit() != 1) {
return (std::abs(nsTPC) < cutNoTOF);
}

const float comb = std::sqrt(nsTPC * nsTPC + nsTOF * nsTOF);
return (comb < cutWithTOF);
return (std::abs(nsTPC) < cutNoTOF && std::abs(nsTOF) < cutWithTOF);
// const float comb = std::sqrt(nsTPC * nsTPC + nsTOF * nsTOF);
// return (comb < cutWithTOF);
}
inline bool passKaonPID(int pidMode,
const aod::F1Tracks::iterator& f1track,
Expand Down Expand Up @@ -340,10 +340,10 @@ struct f1protoncorrelation {
}
return true;
}

return (std::abs(nsTPC) < cutNoTOFBase && std::abs(nsTOF) < cutWithTOF);
// --- TOF available: circular cut in (TPC,TOF) nσ plane
const float comb = std::sqrt(nsTPC * nsTPC + nsTOF * nsTOF);
return (comb < cutWithTOF);
// const float comb = std::sqrt(nsTPC * nsTPC + nsTOF * nsTOF);
// return (comb < cutWithTOF);
}
inline bool passProtonPID(int pidMode,
const aod::ProtonTracks::iterator& ptrack,
Expand Down Expand Up @@ -379,8 +379,9 @@ struct f1protoncorrelation {
// circular cut in (TPC,TOF)
const float nsTPC = ptrack.protonNsigmaTPC();
const float nsTOF = ptrack.protonNsigmaTOF();
const float comb = std::sqrt(nsTPC * nsTPC + nsTOF * nsTOF);
return (comb < cutCircle);
return (std::abs(nsTPC) < cutTPC && std::abs(nsTOF) < cutCircle);
// const float comb = std::sqrt(nsTPC * nsTPC + nsTOF * nsTOF);
// return (comb < cutCircle);
}

// Initialize the ananlysis task
Expand Down Expand Up @@ -610,7 +611,7 @@ struct f1protoncorrelation {
histos.fill(HIST("hNsigmaProtonTPC"), protontrack.protonNsigmaTPC(), protontrack.protonNsigmaTOF(), Proton.Pt());
}
histos.fill(HIST("h2SameEventPtCorrelation"), relative_momentum, F1.Pt(), Proton.Pt());

auto mT = getmT(F1, Proton);
if (f1track.f1SignalStat() > 0) {
// check charge
float pairCharge = f1track.f1SignalStat() * protontrack.protonCharge();
Expand All @@ -625,7 +626,9 @@ struct f1protoncorrelation {
histos.fill(HIST("hPhaseSpaceProtonKaonSame"), Proton.Eta() - Kaon.Eta(), PhiAtSpecificRadiiTPC(Proton, Kaon, protontrack.protonCharge(), kaonCharge, bz, bz), relative_momentum); // Phase Space Proton kaon
if (pionCharge == protontrack.protonCharge())
histos.fill(HIST("hPhaseSpaceProtonPionSame"), Proton.Eta() - Pion.Eta(), PhiAtSpecificRadiiTPC(Proton, Pion, protontrack.protonCharge(), pionCharge, bz, bz), relative_momentum); // Phase Space Proton Pionsyst
histos.fill(HIST("h2SameEventInvariantMassUnlike_mass"), relative_momentum, F1.Pt(), F1.M(), pairCharge, collision.numContrib()); // F1 sign = 1 unlike, F1 sign = -1 like

histos.fill(HIST("h2SameEventInvariantMassUnlike_mass"), relative_momentum, F1.Pt(), F1.M(), pairCharge, collision.numContrib()); // F1 sign = 1 unlike, F1 sign = -1 like
histos.fill(HIST("h2SameEventInvariantMassUnlike_mass_SYS"), 0, relative_momentum, mT, F1.M(), collision.numContrib());
if (fillSparse) {
histos.fill(HIST("SEMassUnlike"), F1.M(), F1.Pt(), Proton.Pt(), relative_momentum, combinedTPC, pairCharge);
}
Expand All @@ -651,6 +654,7 @@ struct f1protoncorrelation {
}
}
if (f1track.f1SignalStat() == -1) {
histos.fill(HIST("h2SameEventInvariantMassLike_mass_SYS"), 0, relative_momentum, mT, F1.M(), collision.numContrib());
histos.fill(HIST("h2SameEventInvariantMassLike_mass"), relative_momentum, F1.Pt(), F1.M(), protontrack.protonCharge(), collision.numContrib());
if (fillSparse) {
histos.fill(HIST("SEMassLike"), F1.M(), F1.Pt(), Proton.Pt(), relative_momentum, combinedTPC, protontrack.protonCharge());
Expand Down Expand Up @@ -896,6 +900,7 @@ struct f1protoncorrelation {
continue;
}
auto relative_momentum = getkstar(F1, Proton);
auto mT = getmT(F1, Proton);
if (t1.f1SignalStat() > 0) {
float pairCharge = t1.f1SignalStat() * t2.protonCharge();
int f1Charge = t1.f1SignalStat();
Expand All @@ -905,6 +910,7 @@ struct f1protoncorrelation {
pionCharge = 1;
kaonCharge = -1;
}
histos.fill(HIST("h2MixEventInvariantMassUnlike_mass_SYS"), 0, relative_momentum, mT, F1.M(), collision1.numContrib());
histos.fill(HIST("h2MixEventInvariantMassUnlike_mass"), relative_momentum, F1.Pt(), F1.M(), pairCharge, collision1.numContrib()); // F1 sign = 1 unlike, F1 sign = -1 like
histos.fill(HIST("hPhaseSpaceProtonKaonMix"), Proton.Eta() - Kaon.Eta(), PhiAtSpecificRadiiTPC(Proton, Kaon, t2.protonCharge(), kaonCharge, bz, bz2), relative_momentum); // Phase Space Proton kaon
histos.fill(HIST("hPhaseSpaceProtonPionMix"), Proton.Eta() - Pion.Eta(), PhiAtSpecificRadiiTPC(Proton, Pion, t2.protonCharge(), pionCharge, bz, bz2), relative_momentum); // Phase Space Proton Pion
Expand Down Expand Up @@ -936,6 +942,7 @@ struct f1protoncorrelation {
}
}
if (t1.f1SignalStat() == -1) {
histos.fill(HIST("h2MixEventInvariantMassLike_mass_SYS"), 0, relative_momentum, mT, F1.M(), collision1.numContrib());
histos.fill(HIST("h2MixEventInvariantMassLike_mass"), relative_momentum, F1.Pt(), F1.M(), t2.protonCharge(), collision1.numContrib());
if (fillSparse) {
histos.fill(HIST("MEMassLike"), F1.M(), F1.Pt(), Proton.Pt(), relative_momentum, combinedTPC, t2.protonCharge());
Expand Down
Loading