Skip to content

[PWGLF] Applied rapidity cut for rotational mother #12029

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
dmallick2 merged 3 commits into from
Jul 12, 2025
Merged

[PWGLF] Applied rapidity cut for rotational mother #12029

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
2320e66
applied rapidity cut for rotational mother
Jul 12, 2025
5065351
added signal loss process function
Jul 12, 2025
71f0d52
Code optimisation
Jul 12, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
271 changes: 169 additions & 102 deletions PWGLF/Tasks/Resonances/kstarqa.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -119,8 +119,8 @@
Configurable<float> nsigmaCutCombined{"nsigmaCutCombined", 3.0, "Combined Nsigma cut"};

// Event selection configurables
Configurable<bool> timFrameEvsel{"timFrameEvsel", true, "TPC Time frame boundary cut"};
Configurable<bool> cTVXEvsel{"cTVXEvsel", true, "Triggger selection"};
// Configurable<bool> timFrameEvsel{"timFrameEvsel", true, "TPC Time frame boundary cut"};
// Configurable<bool> cTVXEvsel{"cTVXEvsel", true, "Triggger selection"};
Configurable<float> cutzvertex{"cutzvertex", 10.0f, "Accepted z-vertex range (cm)"};
// Configurable<bool> cMID{"cMID", false, "Misidentification of tracks"};

Expand Down Expand Up @@ -213,6 +213,8 @@
rEventSelection.add("events_check", "No. of events in the generated MC", kTH1I, {{20, 0, 20}});
rEventSelection.add("events_checkrec", "No. of events in the reconstructed MC", kTH1I, {{20, 0, 20}});
hInvMass.add("h1KSRecsplit", "KS meson Rec split", kTH1F, {{100, 0.0f, 10.0f}});
hInvMass.add("kstargenBeforeEvtSel", "Kstar generated before event selection", kTH1F, {ptAxis});
hInvMass.add("kstargenAfterEvtSel", "Kstar generated after event selection", kTH1F, {ptAxis});

// Multplicity distribution
if (cQAevents) {
Expand All @@ -229,6 +231,25 @@
double massPi = o2::constants::physics::MassPiPlus;
double massKa = o2::constants::physics::MassKPlus;

template <typename Coll>
bool selectionEvent(const Coll& collision)
{
if (std::abs(collision.posZ()) > cutzvertex)
return false;
if (!collision.sel8())
return false;
if (!collision.selection_bit(aod::evsel::kIsTriggerTVX)) {
return false;
}
if (!collision.selection_bit(aod::evsel::kNoTimeFrameBorder) || !collision.selection_bit(aod::evsel::kNoITSROFrameBorder)) {
return false;
}
if (rctCut.requireRCTFlagChecker && !rctChecker(collision)) {
return false;
}
return true;
}

template <typename T>
bool selectionTrack(const T& candidate)
{
Expand Down Expand Up @@ -333,23 +354,23 @@
bool selectionPIDNew(const T& candidate, int PID)
{
if (PID == 0) {
if (candidate.pt() < 0.5 && TMath::Abs(candidate.tpcNSigmaPi()) < nsigmaCutTPCPi) {

Check failure on line 357 in PWGLF/Tasks/Resonances/kstarqa.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[root/entity] 

Replace ROOT entities with equivalents from standard C++ or from O2.
return true;
}
if (candidate.pt() >= 0.5 && TMath::Abs(candidate.tpcNSigmaPi()) < nsigmaCutTPCPi && candidate.hasTOF() && TMath::Abs(candidate.tofNSigmaPi()) < nsigmaCutTOFPi) {

Check failure on line 360 in PWGLF/Tasks/Resonances/kstarqa.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[root/entity] 

Replace ROOT entities with equivalents from standard C++ or from O2.
return true;
}
if (candidate.pt() >= 0.5 && TMath::Abs(candidate.tpcNSigmaPi()) < nsigmaCutTPCPi && !candidate.hasTOF()) {

Check failure on line 363 in PWGLF/Tasks/Resonances/kstarqa.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[root/entity] 

Replace ROOT entities with equivalents from standard C++ or from O2.
return true;
}
} else if (PID == 1) {
if (candidate.pt() < 0.5 && TMath::Abs(candidate.tpcNSigmaKa()) < nsigmaCutTPCKa) {

Check failure on line 367 in PWGLF/Tasks/Resonances/kstarqa.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[root/entity] 

Replace ROOT entities with equivalents from standard C++ or from O2.
return true;
}
if (candidate.pt() >= 0.5 && TMath::Abs(candidate.tpcNSigmaKa()) < nsigmaCutTPCKa && candidate.hasTOF() && TMath::Abs(candidate.tofNSigmaKa()) < nsigmaCutTOFKa) {

Check failure on line 370 in PWGLF/Tasks/Resonances/kstarqa.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[root/entity] 

Replace ROOT entities with equivalents from standard C++ or from O2.
return true;
}
if (candidate.pt() >= 0.5 && TMath::Abs(candidate.tpcNSigmaKa()) < nsigmaCutTPCKa && !candidate.hasTOF()) {

Check failure on line 373 in PWGLF/Tasks/Resonances/kstarqa.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[root/entity] 

Replace ROOT entities with equivalents from standard C++ or from O2.
return true;
}
}
Expand Down Expand Up @@ -451,7 +472,7 @@

using EventCandidates = soa::Filtered<soa::Join<aod::Collisions, aod::EvSels, aod::FT0Mults, aod::MultZeqs, aod::CentFT0Ms, aod::CentFT0As, aod::CentFT0Cs>>;
using TrackCandidates = soa::Filtered<soa::Join<aod::Tracks, aod::TracksExtra, aod::TracksDCA, aod::TrackSelection, aod::pidTPCFullPi, aod::pidTOFFullPi, aod::pidTPCFullKa, aod::pidTOFFullKa, aod::pidTOFbeta>>;
using EventCandidatesMC = soa::Join<aod::Collisions, aod::EvSels, aod::McCollisionLabels, aod::CentFT0Ms>;
using EventCandidatesMC = soa::Join<aod::Collisions, aod::EvSels, aod::McCollisionLabels, aod::CentFT0Ms, aod::CentFT0As, aod::CentFT0Cs>;

using TrackCandidatesMC = soa::Filtered<soa::Join<aod::Tracks, aod::TracksExtra, aod::TracksDCA, aod::TrackSelection, aod::pidTPCFullKa, aod::pidTOFFullKa, aod::pidTPCFullPi, aod::pidTOFFullPi, aod::McTrackLabels, aod::pidTOFbeta>>;

Expand All @@ -468,114 +489,122 @@
ROOT::Math::Boost boost{mother.BoostToCM()}; // boost mother to center of mass frame
fourVecDauCM = boost(daughterSelected); // boost the frame of daughter same as mother

if (std::abs(mother.Rapidity()) < 0.5) {
if (activateTHnSparseCosThStarHelicity) {
auto cosThetaStarHelicity = mother.Vect().Dot(fourVecDauCM.Vect()) / (std::sqrt(fourVecDauCM.Vect().Mag2()) * std::sqrt(mother.Vect().Mag2()));
// if (std::abs(mother.Rapidity()) < 0.5) {
if (activateTHnSparseCosThStarHelicity) {
auto cosThetaStarHelicity = mother.Vect().Dot(fourVecDauCM.Vect()) / (std::sqrt(fourVecDauCM.Vect().Mag2()) * std::sqrt(mother.Vect().Mag2()));

if (track1.sign() * track2.sign() < 0) {
if (!isMix) {
if (track1.sign() * track2.sign() < 0) {
if (!isMix) {
if (std::abs(mother.Rapidity()) < 0.5) {
hInvMass.fill(HIST("h3KstarInvMassUnlikeSign"), multiplicity, mother.Pt(), mother.M(), cosThetaStarHelicity);
}

for (int i = 0; i < cRotations; i++) {
theta2 = rn->Uniform(o2::constants::math::PI - o2::constants::math::PI / rotationalCut, o2::constants::math::PI + o2::constants::math::PI / rotationalCut);
for (int i = 0; i < cRotations; i++) {
theta2 = rn->Uniform(o2::constants::math::PI - o2::constants::math::PI / rotationalCut, o2::constants::math::PI + o2::constants::math::PI / rotationalCut);

daughterRot = ROOT::Math::PxPyPzMVector(daughter1.Px() * std::cos(theta2) - daughter1.Py() * std::sin(theta2), daughter1.Px() * std::sin(theta2) + daughter1.Py() * std::cos(theta2), daughter1.Pz(), daughter1.M());
daughterRot = ROOT::Math::PxPyPzMVector(daughter1.Px() * std::cos(theta2) - daughter1.Py() * std::sin(theta2), daughter1.Px() * std::sin(theta2) + daughter1.Py() * std::cos(theta2), daughter1.Pz(), daughter1.M());

motherRot = daughterRot + daughter2;
motherRot = daughterRot + daughter2;

ROOT::Math::Boost boost2{motherRot.BoostToCM()};
daughterRotCM = boost2(daughterRot);
ROOT::Math::Boost boost2{motherRot.BoostToCM()};
daughterRotCM = boost2(daughterRot);

auto cosThetaStarHelicityRot = motherRot.Vect().Dot(daughterRotCM.Vect()) / (std::sqrt(daughterRotCM.Vect().Mag2()) * std::sqrt(motherRot.Vect().Mag2()));
auto cosThetaStarHelicityRot = motherRot.Vect().Dot(daughterRotCM.Vect()) / (std::sqrt(daughterRotCM.Vect().Mag2()) * std::sqrt(motherRot.Vect().Mag2()));

if (calcRotational)
hInvMass.fill(HIST("h3KstarInvMassRotated"), multiplicity, motherRot.Pt(), motherRot.M(), cosThetaStarHelicityRot);
}
} else {
hInvMass.fill(HIST("h3KstarInvMassMixed"), multiplicity, mother.Pt(), mother.M(), cosThetaStarHelicity);
}
} else {
if (!isMix) {
if (calcLikeSign)
hInvMass.fill(HIST("h3KstarInvMasslikeSign"), multiplicity, mother.Pt(), mother.M(), cosThetaStarHelicity);
if (calcRotational && motherRot.Rapidity() < 0.5)
hInvMass.fill(HIST("h3KstarInvMassRotated"), multiplicity, motherRot.Pt(), motherRot.M(), cosThetaStarHelicityRot);
}
} else if (isMix && std::abs(mother.Rapidity()) < 0.5) {
hInvMass.fill(HIST("h3KstarInvMassMixed"), multiplicity, mother.Pt(), mother.M(), cosThetaStarHelicity);
}
} else {
if (!isMix) {
if (calcLikeSign && std::abs(mother.Rapidity()) < 0.5)
hInvMass.fill(HIST("h3KstarInvMasslikeSign"), multiplicity, mother.Pt(), mother.M(), cosThetaStarHelicity);
}
}

} else if (activateTHnSparseCosThStarProduction) {
normalVec = ROOT::Math::XYZVector(mother.Py(), -mother.Px(), 0.f);
auto cosThetaStarProduction = normalVec.Dot(fourVecDauCM.Vect()) / (std::sqrt(fourVecDauCM.Vect().Mag2()) * std::sqrt(normalVec.Mag2()));
} else if (activateTHnSparseCosThStarProduction) {
normalVec = ROOT::Math::XYZVector(mother.Py(), -mother.Px(), 0.f);
auto cosThetaStarProduction = normalVec.Dot(fourVecDauCM.Vect()) / (std::sqrt(fourVecDauCM.Vect().Mag2()) * std::sqrt(normalVec.Mag2()));

if (track1.sign() * track2.sign() < 0) {
if (!isMix) {
if (track1.sign() * track2.sign() < 0) {
if (!isMix) {
if (std::abs(mother.Rapidity()) < 0.5) {
hInvMass.fill(HIST("h3KstarInvMassUnlikeSign"), multiplicity, mother.Pt(), mother.M(), cosThetaStarProduction);
for (int i = 0; i < cRotations; i++) {
theta2 = rn->Uniform(0, o2::constants::math::PI);
daughterRot = ROOT::Math::PxPyPzMVector(daughter1.Px() * std::cos(theta2) - daughter1.Py() * std::sin(theta2), daughter1.Px() * std::sin(theta2) + daughter1.Py() * std::cos(theta2), daughter1.Pz(), daughter1.M());

motherRot = daughterRot + daughter2;
if (calcRotational)
hInvMass.fill(HIST("h3KstarInvMassRotated"), multiplicity, motherRot.Pt(), motherRot.M(), cosThetaStarProduction);
}
} else {
hInvMass.fill(HIST("h3KstarInvMassMixed"), multiplicity, mother.Pt(), mother.M(), cosThetaStarProduction);
}
} else {
if (!isMix) {
if (calcLikeSign)
hInvMass.fill(HIST("h3KstarInvMasslikeSign"), multiplicity, mother.Pt(), mother.M(), cosThetaStarProduction);
for (int i = 0; i < cRotations; i++) {
theta2 = rn->Uniform(0, o2::constants::math::PI);
daughterRot = ROOT::Math::PxPyPzMVector(daughter1.Px() * std::cos(theta2) - daughter1.Py() * std::sin(theta2), daughter1.Px() * std::sin(theta2) + daughter1.Py() * std::cos(theta2), daughter1.Pz(), daughter1.M());

motherRot = daughterRot + daughter2;
if (calcRotational && abs(motherRot.Rapidity()) < 0.5)

Check failure on line 541 in PWGLF/Tasks/Resonances/kstarqa.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[std-prefix] 

Use std:: prefix for names from the std namespace.
hInvMass.fill(HIST("h3KstarInvMassRotated"), multiplicity, motherRot.Pt(), motherRot.M(), cosThetaStarProduction);
}
} else if (isMix && std::abs(mother.Rapidity()) < 0.5) {
hInvMass.fill(HIST("h3KstarInvMassMixed"), multiplicity, mother.Pt(), mother.M(), cosThetaStarProduction);
}
} else if (activateTHnSparseCosThStarBeam) {
beamVec = ROOT::Math::XYZVector(0.f, 0.f, 1.f);
auto cosThetaStarBeam = beamVec.Dot(fourVecDauCM.Vect()) / std::sqrt(fourVecDauCM.Vect().Mag2());
} else {
if (!isMix) {
if (calcLikeSign && std::abs(mother.Rapidity()) < 0.5)
hInvMass.fill(HIST("h3KstarInvMasslikeSign"), multiplicity, mother.Pt(), mother.M(), cosThetaStarProduction);
}
}
} else if (activateTHnSparseCosThStarBeam) {
beamVec = ROOT::Math::XYZVector(0.f, 0.f, 1.f);
auto cosThetaStarBeam = beamVec.Dot(fourVecDauCM.Vect()) / std::sqrt(fourVecDauCM.Vect().Mag2());

if (track1.sign() * track2.sign() < 0) {
if (!isMix) {
if (track1.sign() * track2.sign() < 0) {
if (!isMix) {
if (std::abs(mother.Rapidity()) < 0.5) {
hInvMass.fill(HIST("h3KstarInvMassUnlikeSign"), multiplicity, mother.Pt(), mother.M(), cosThetaStarBeam);
for (int i = 0; i < cRotations; i++) {
theta2 = rn->Uniform(0, o2::constants::math::PI);
daughterRot = ROOT::Math::PxPyPzMVector(daughter1.Px() * std::cos(theta2) - daughter1.Py() * std::sin(theta2), daughter1.Px() * std::sin(theta2) + daughter1.Py() * std::cos(theta2), daughter1.Pz(), daughter1.M());
}
for (int i = 0; i < cRotations; i++) {
theta2 = rn->Uniform(0, o2::constants::math::PI);
daughterRot = ROOT::Math::PxPyPzMVector(daughter1.Px() * std::cos(theta2) - daughter1.Py() * std::sin(theta2), daughter1.Px() * std::sin(theta2) + daughter1.Py() * std::cos(theta2), daughter1.Pz(), daughter1.M());

motherRot = daughterRot + daughter2;
if (calcRotational)
hInvMass.fill(HIST("h3KstarInvMassRotated"), multiplicity, motherRot.Pt(), motherRot.M(), cosThetaStarBeam);
}
} else {
hInvMass.fill(HIST("h3KstarInvMassMixed"), multiplicity, mother.Pt(), mother.M(), cosThetaStarBeam);
motherRot = daughterRot + daughter2;
if (calcRotational && std::abs(motherRot.Rapidity()) < 0.5)
hInvMass.fill(HIST("h3KstarInvMassRotated"), multiplicity, motherRot.Pt(), motherRot.M(), cosThetaStarBeam);
}
} else {
if (calcLikeSign)
hInvMass.fill(HIST("h3KstarInvMasslikeSign"), multiplicity, mother.Pt(), mother.M(), cosThetaStarBeam);
} else if (isMix && std::abs(mother.Rapidity()) < 0.5) {
hInvMass.fill(HIST("h3KstarInvMassMixed"), multiplicity, mother.Pt(), mother.M(), cosThetaStarBeam);
}
} else if (activateTHnSparseCosThStarRandom) {
auto phiRandom = gRandom->Uniform(0.f, constants::math::TwoPI);
auto thetaRandom = gRandom->Uniform(0.f, constants::math::PI);
} else {
if (calcLikeSign && std::abs(mother.Rapidity()) < 0.5)
hInvMass.fill(HIST("h3KstarInvMasslikeSign"), multiplicity, mother.Pt(), mother.M(), cosThetaStarBeam);
}
} else if (activateTHnSparseCosThStarRandom) {
auto phiRandom = gRandom->Uniform(0.f, constants::math::TwoPI);
auto thetaRandom = gRandom->Uniform(0.f, constants::math::PI);

randomVec = ROOT::Math::XYZVector(std::sin(thetaRandom) * std::cos(phiRandom), std::sin(thetaRandom) * std::sin(phiRandom), std::cos(thetaRandom));
auto cosThetaStarRandom = randomVec.Dot(fourVecDauCM.Vect()) / std::sqrt(fourVecDauCM.Vect().Mag2());
randomVec = ROOT::Math::XYZVector(std::sin(thetaRandom) * std::cos(phiRandom), std::sin(thetaRandom) * std::sin(phiRandom), std::cos(thetaRandom));
auto cosThetaStarRandom = randomVec.Dot(fourVecDauCM.Vect()) / std::sqrt(fourVecDauCM.Vect().Mag2());

if (track1.sign() * track2.sign() < 0) {
if (!isMix) {
if (track1.sign() * track2.sign() < 0) {
if (!isMix) {
if (std::abs(mother.Rapidity()) < 0.5) {
hInvMass.fill(HIST("h3KstarInvMassUnlikeSign"), multiplicity, mother.Pt(), mother.M(), cosThetaStarRandom);
for (int i = 0; i < cRotations; i++) {
theta2 = rn->Uniform(0, o2::constants::math::PI);
daughterRot = ROOT::Math::PxPyPzMVector(daughter1.Px() * std::cos(theta2) - daughter1.Py() * std::sin(theta2), daughter1.Px() * std::sin(theta2) + daughter1.Py() * std::cos(theta2), daughter1.Pz(), daughter1.M());

motherRot = daughterRot + daughter2;
if (calcRotational)
hInvMass.fill(HIST("h3KstarInvMassRotated"), multiplicity, motherRot.Pt(), motherRot.M(), cosThetaStarRandom);
}
} else {
hInvMass.fill(HIST("h3KstarInvMassMixed"), multiplicity, mother.Pt(), mother.M(), cosThetaStarRandom);
}
} else {
if (!isMix) {
if (calcLikeSign)
hInvMass.fill(HIST("h3KstarInvMasslikeSign"), multiplicity, mother.Pt(), mother.M(), cosThetaStarRandom);
for (int i = 0; i < cRotations; i++) {
theta2 = rn->Uniform(0, o2::constants::math::PI);
daughterRot = ROOT::Math::PxPyPzMVector(daughter1.Px() * std::cos(theta2) - daughter1.Py() * std::sin(theta2), daughter1.Px() * std::sin(theta2) + daughter1.Py() * std::cos(theta2), daughter1.Pz(), daughter1.M());

motherRot = daughterRot + daughter2;
if (calcRotational && std::abs(motherRot.Rapidity()) < 0.5)
hInvMass.fill(HIST("h3KstarInvMassRotated"), multiplicity, motherRot.Pt(), motherRot.M(), cosThetaStarRandom);
}
} else if (isMix && std::abs(mother.Rapidity()) < 0.5) {
hInvMass.fill(HIST("h3KstarInvMassMixed"), multiplicity, mother.Pt(), mother.M(), cosThetaStarRandom);
}
} else {
if (!isMix) {
if (calcLikeSign && std::abs(mother.Rapidity()) < 0.5)
hInvMass.fill(HIST("h3KstarInvMasslikeSign"), multiplicity, mother.Pt(), mother.M(), cosThetaStarRandom);
}
}
}
// }
}

// int counter = 0;
Expand All @@ -585,23 +614,24 @@
{
rEventSelection.fill(HIST("events_check_data"), 0.5);

if (cTVXEvsel && (!collision.selection_bit(aod::evsel::kIsTriggerTVX))) {
return;
}
rEventSelection.fill(HIST("events_check_data"), 1.5);
// if (cTVXEvsel && (!collision.selection_bit(aod::evsel::kIsTriggerTVX))) {
// return;
// }
// rEventSelection.fill(HIST("events_check_data"), 1.5);

if (timFrameEvsel && (!collision.selection_bit(aod::evsel::kNoTimeFrameBorder) || !collision.selection_bit(aod::evsel::kNoITSROFrameBorder))) {
return;
}
// if (timFrameEvsel && (!collision.selection_bit(aod::evsel::kNoTimeFrameBorder) || !collision.selection_bit(aod::evsel::kNoITSROFrameBorder))) {
// return;
// }

if (!collision.sel8()) {
return;
}
rEventSelection.fill(HIST("events_check_data"), 2.5);
// if (!collision.sel8()) {
// return;
// }
// rEventSelection.fill(HIST("events_check_data"), 2.5);

if (rctCut.requireRCTFlagChecker && !rctChecker(collision)) {
if (!selectionEvent(collision)) {
return;
}

rEventSelection.fill(HIST("events_check_data"), 3.5);

multiplicity = -1;
Expand All @@ -613,7 +643,7 @@
} else if (cSelectMultEstimator == 2) {
multiplicity = collision.centFT0C();
} else {
multiplicity = collision.multFT0M();
multiplicity = collision.centFT0M();
}

// Fill the event counter
Expand Down Expand Up @@ -747,11 +777,12 @@
// Map estimator to pair and multiplicity accessor
auto runMixing = [&](auto& pair, auto multiplicityGetter) {
for (const auto& [c1, tracks1, c2, tracks2] : pair) {
if (!c1.sel8() || !c2.sel8())
continue;
// if (!c1.sel8() || !c2.sel8())
// continue;

if (rctCut.requireRCTFlagChecker && (!rctChecker(c1) || !rctChecker(c2)))
if (!selectionEvent(c1) || !selectionEvent(c2)) {
continue;
}

multiplicity = multiplicityGetter(c1);

Expand Down Expand Up @@ -797,7 +828,7 @@
int nChInel = 0;
for (const auto& mcParticle : mcParticles) {
auto pdgcode = std::abs(mcParticle.pdgCode());
if (mcParticle.isPhysicalPrimary() && (pdgcode == 211 || pdgcode == 321 || pdgcode == 2212 || pdgcode == 11 || pdgcode == 13)) {

Check failure on line 831 in PWGLF/Tasks/Resonances/kstarqa.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[pdg/explicit-code] 

Avoid hard-coded PDG codes. Use named values from PDG_t or o2::constants::physics::Pdg instead.
if (std::abs(mcParticle.eta()) < 1.0) {
nChInel = nChInel + 1;
}
Expand All @@ -815,10 +846,10 @@
continue;
}

if (timFrameEvsel && !collision.selection_bit(aod::evsel::kNoTimeFrameBorder)) {
if (!collision.selection_bit(aod::evsel::kNoTimeFrameBorder)) {
continue;
}
if (cTVXEvsel && (!collision.selection_bit(aod::evsel::kIsTriggerTVX))) {
if (!collision.selection_bit(aod::evsel::kIsTriggerTVX)) {
continue;
}
if (!collision.sel8()) {
Expand All @@ -844,7 +875,7 @@
}
rEventSelection.fill(HIST("events_check"), 5.5);

if (std::abs(mcParticle.pdgCode()) != 313) {

Check failure on line 878 in PWGLF/Tasks/Resonances/kstarqa.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[pdg/explicit-code] 

Avoid hard-coded PDG codes. Use named values from PDG_t or o2::constants::physics::Pdg instead.
continue;
}
rEventSelection.fill(HIST("events_check"), 6.5);
Expand Down Expand Up @@ -882,6 +913,42 @@
}
PROCESS_SWITCH(Kstarqa, processGen, "Process Generated", false);

void processEvtLossSigLossMC(aod::McCollisions::iterator const&, aod::McParticles const& mcParticles, const soa::SmallGroups<EventCandidatesMC>& recCollisions)
{

bool isSel = false;
// auto multiplicity1 = -999.;
for (const auto& RecCollision : recCollisions) {
if (!selectionEvent(RecCollision))
continue;

// if (cSelectMultEstimator == 0) {
// multiplicity1 = RecCollision.centFT0M();
// } else if (cSelectMultEstimator == 1) {
// multiplicity1 = RecCollision.centFT0A();
// } else if (cSelectMultEstimator == 2) {
// multiplicity1 = RecCollision.centFT0C();
// } else {
// multiplicity1 = RecCollision.centFT0M();
// }

isSel = true;
}

// Generated MC
for (const auto& mcPart : mcParticles) {
if (std::abs(mcPart.y()) >= 0.5 || std::abs(mcPart.pdgCode()) != 313)
continue;

// signal loss estimation
hInvMass.fill(HIST("kstargenBeforeEvtSel"), mcPart.pt());
if (isSel) {
hInvMass.fill(HIST("kstargenAfterEvtSel"), mcPart.pt());
}
} // end loop on gen particles
}
PROCESS_SWITCH(Kstarqa, processEvtLossSigLossMC, "Process Signal Loss, Event Loss", false);

void processRec(EventCandidatesMC::iterator const& collision, TrackCandidatesMC const& tracks, aod::McParticles const&, aod::McCollisions const& /*mcCollisions*/)
{

Expand All @@ -898,12 +965,12 @@
}
rEventSelection.fill(HIST("events_checkrec"), 2.5);

if (timFrameEvsel && !collision.selection_bit(aod::evsel::kNoTimeFrameBorder)) {
if (!collision.selection_bit(aod::evsel::kNoTimeFrameBorder)) {
return;
}
rEventSelection.fill(HIST("events_checkrec"), 3.5);

if (cTVXEvsel && (!collision.selection_bit(aod::evsel::kIsTriggerTVX))) {
if (!collision.selection_bit(aod::evsel::kIsTriggerTVX)) {
return;
}
rEventSelection.fill(HIST("events_checkrec"), 4.5);
Expand Down Expand Up @@ -1043,7 +1110,7 @@
daughter2 = ROOT::Math::PxPyPzMVector(track2.px(), track2.py(), track2.pz(), massPi);
mother = daughter1 + daughter2; // Kstar meson

hInvMass.fill(HIST("h2KstarRecpt2"), mothertrack1.pt(), multiplicity, TMath::Sqrt(mothertrack1.e() * mothertrack1.e() - mothertrack1.p() * mothertrack1.p()));

Check failure on line 1113 in PWGLF/Tasks/Resonances/kstarqa.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[root/entity] 

Replace ROOT entities with equivalents from standard C++ or from O2.

if (applyRecMotherRapidity && mother.Rapidity() >= 0) {
continue;
Expand Down
Loading