Skip to content

Commit 46a28d2

Browse files
mfagginMattia Faggin
mfaggin
and
Mattia Faggin
authored
[PWGHF] Add Y cut for reco candidates and fill data sparses with abs(Y). (AliceO2Group#8560)
Co-authored-by: Mattia Faggin <[email protected]>
1 parent 0052f3a commit 46a28d2

File tree

1 file changed

+26
-16
lines changed

1 file changed

+26
-16
lines changed

‎PWGHF/D2H/Tasks/taskCharmPolarisation.cxx‎

Lines changed: 26 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,8 @@
1616
/// \author S. Kundu (CERN) [email protected]
1717
/// \author M. Faggin (CERN) [email protected]
1818

19+
#include <vector>
20+
1921
#include "TRandom3.h"
2022
#include "Math/Vector3D.h"
2123
#include "Math/Vector4D.h"
@@ -189,6 +191,9 @@ struct TaskPolarisationCharmHadrons {
189191
Configurable<bool> activateTHnEulerPhiMonitor{"activateTHnEulerPhiMonitor", false, "Flag to switch on the monitoring THnSparse vs. Euler angle phi (Lc -> pKpi)"};
190192
ConfigurableAxis configTHnAxisEulerPhi{"configTHnAxisEulerPhi", {24, -o2::constants::math::PI, o2::constants::math::PI}, "Euler polar angle #phi"};
191193

194+
/// Application of rapidity cut for reconstructed candidates
195+
Configurable<float> rapidityCut{"rapidityCut", 999.f, "Max. value of reconstructed candidate rapidity (abs. value)"};
196+
192197
Filter filterSelectDstarCandidates = aod::hf_sel_candidate_dstar::isSelDstarToD0Pi == selectionFlagDstarToD0Pi;
193198
Filter filterSelectLcToPKPiCandidates = (aod::hf_sel_candidate_lc::isSelLcToPKPi >= selectionFlagLcToPKPi) || (aod::hf_sel_candidate_lc::isSelLcToPiKP >= selectionFlagLcToPKPi);
194199

@@ -537,18 +542,18 @@ struct TaskPolarisationCharmHadrons {
537542
if constexpr (!doMc) { // data
538543
if constexpr (withMl) { // with ML
539544
if constexpr (channel == charm_polarisation::DecayChannel::DstarToDzeroPi) { // D*+
540-
registry.fill(HIST("hHelicity"), invMassCharmHad, ptCharmHad, numPvContributors, rapCharmHad, invMassD0, cosThetaStar, outputMl[0], /*outputMl[1],*/ outputMl[2], absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate);
545+
registry.fill(HIST("hHelicity"), invMassCharmHad, ptCharmHad, numPvContributors, std::abs(rapCharmHad), invMassD0, cosThetaStar, outputMl[0], /*outputMl[1],*/ outputMl[2], absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate);
541546
} else if constexpr (channel == charm_polarisation::DecayChannel::LcToPKPi) { // Lc+
542-
registry.fill(HIST("hHelicity"), invMassCharmHad, ptCharmHad, numPvContributors, rapCharmHad, invMassKPiLc, cosThetaStar, outputMl[0], /*outputMl[1],*/ outputMl[2], absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate, charge);
547+
registry.fill(HIST("hHelicity"), invMassCharmHad, ptCharmHad, numPvContributors, std::abs(rapCharmHad), invMassKPiLc, cosThetaStar, outputMl[0], /*outputMl[1],*/ outputMl[2], absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate, charge);
543548
if (activateTHnEulerPhiMonitor) {
544549
registry.fill(HIST("hEulerPhiHelicity"), invMassCharmHad, ptCharmHad, invMassKPiLc, phiEuler, outputMl[0], /*outputMl[1],*/ outputMl[2], charge);
545550
}
546551
}
547552
} else { // without ML
548553
if constexpr (channel == charm_polarisation::DecayChannel::DstarToDzeroPi) { // D*+
549-
registry.fill(HIST("hHelicity"), invMassCharmHad, ptCharmHad, numPvContributors, rapCharmHad, invMassD0, cosThetaStar, absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate);
554+
registry.fill(HIST("hHelicity"), invMassCharmHad, ptCharmHad, numPvContributors, std::abs(rapCharmHad), invMassD0, cosThetaStar, absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate);
550555
} else if constexpr (channel == charm_polarisation::DecayChannel::LcToPKPi) { // Lc+
551-
registry.fill(HIST("hHelicity"), invMassCharmHad, ptCharmHad, numPvContributors, rapCharmHad, invMassKPiLc, cosThetaStar, absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate, charge);
556+
registry.fill(HIST("hHelicity"), invMassCharmHad, ptCharmHad, numPvContributors, std::abs(rapCharmHad), invMassKPiLc, cosThetaStar, absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate, charge);
552557
if (activateTHnEulerPhiMonitor) {
553558
registry.fill(HIST("hEulerPhiHelicity"), invMassCharmHad, ptCharmHad, invMassKPiLc, phiEuler, charge);
554559
}
@@ -601,18 +606,18 @@ struct TaskPolarisationCharmHadrons {
601606
if constexpr (!doMc) { // data
602607
if constexpr (withMl) { // with ML
603608
if constexpr (channel == charm_polarisation::DecayChannel::DstarToDzeroPi) { // D*+
604-
registry.fill(HIST("hProduction"), invMassCharmHad, ptCharmHad, numPvContributors, rapCharmHad, invMassD0, cosThetaStar, outputMl[0], /*outputMl[1],*/ outputMl[2], absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate);
609+
registry.fill(HIST("hProduction"), invMassCharmHad, ptCharmHad, numPvContributors, std::abs(rapCharmHad), invMassD0, cosThetaStar, outputMl[0], /*outputMl[1],*/ outputMl[2], absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate);
605610
} else if constexpr (channel == charm_polarisation::DecayChannel::LcToPKPi) { // Lc+
606-
registry.fill(HIST("hProduction"), invMassCharmHad, ptCharmHad, numPvContributors, rapCharmHad, invMassKPiLc, cosThetaStar, outputMl[0], /*outputMl[1],*/ outputMl[2], absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate, charge);
611+
registry.fill(HIST("hProduction"), invMassCharmHad, ptCharmHad, numPvContributors, std::abs(rapCharmHad), invMassKPiLc, cosThetaStar, outputMl[0], /*outputMl[1],*/ outputMl[2], absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate, charge);
607612
if (activateTHnEulerPhiMonitor) {
608613
registry.fill(HIST("hEulerPhiProduction"), invMassCharmHad, ptCharmHad, invMassKPiLc, phiEuler, outputMl[0], /*outputMl[1],*/ outputMl[2], charge);
609614
}
610615
}
611616
} else { // without ML
612617
if constexpr (channel == charm_polarisation::DecayChannel::DstarToDzeroPi) { // D*+
613-
registry.fill(HIST("hProduction"), invMassCharmHad, ptCharmHad, numPvContributors, rapCharmHad, invMassD0, cosThetaStar, absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate);
618+
registry.fill(HIST("hProduction"), invMassCharmHad, ptCharmHad, numPvContributors, std::abs(rapCharmHad), invMassD0, cosThetaStar, absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate);
614619
} else if constexpr (channel == charm_polarisation::DecayChannel::LcToPKPi) { // Lc+
615-
registry.fill(HIST("hProduction"), invMassCharmHad, ptCharmHad, numPvContributors, rapCharmHad, invMassKPiLc, cosThetaStar, absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate, charge);
620+
registry.fill(HIST("hProduction"), invMassCharmHad, ptCharmHad, numPvContributors, std::abs(rapCharmHad), invMassKPiLc, cosThetaStar, absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate, charge);
616621
if (activateTHnEulerPhiMonitor) {
617622
registry.fill(HIST("hEulerPhiProduction"), invMassCharmHad, ptCharmHad, invMassKPiLc, phiEuler, charge);
618623
}
@@ -665,18 +670,18 @@ struct TaskPolarisationCharmHadrons {
665670
if constexpr (!doMc) { // data
666671
if constexpr (withMl) { // with ML
667672
if constexpr (channel == charm_polarisation::DecayChannel::DstarToDzeroPi) { // D*+
668-
registry.fill(HIST("hBeam"), invMassCharmHad, ptCharmHad, numPvContributors, rapCharmHad, invMassD0, cosThetaStar, outputMl[0], /*outputMl[1],*/ outputMl[2], absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate);
673+
registry.fill(HIST("hBeam"), invMassCharmHad, ptCharmHad, numPvContributors, std::abs(rapCharmHad), invMassD0, cosThetaStar, outputMl[0], /*outputMl[1],*/ outputMl[2], absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate);
669674
} else if constexpr (channel == charm_polarisation::DecayChannel::LcToPKPi) { // Lc+
670-
registry.fill(HIST("hBeam"), invMassCharmHad, ptCharmHad, numPvContributors, rapCharmHad, invMassKPiLc, cosThetaStar, outputMl[0], /*outputMl[1],*/ outputMl[2], absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate, charge);
675+
registry.fill(HIST("hBeam"), invMassCharmHad, ptCharmHad, numPvContributors, std::abs(rapCharmHad), invMassKPiLc, cosThetaStar, outputMl[0], /*outputMl[1],*/ outputMl[2], absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate, charge);
671676
if (activateTHnEulerPhiMonitor) {
672677
registry.fill(HIST("hEulerPhiBeam"), invMassCharmHad, ptCharmHad, invMassKPiLc, phiEuler, outputMl[0], /*outputMl[1],*/ outputMl[2], charge);
673678
}
674679
}
675680
} else { // without ML
676681
if constexpr (channel == charm_polarisation::DecayChannel::DstarToDzeroPi) { // D*+
677-
registry.fill(HIST("hBeam"), invMassCharmHad, ptCharmHad, numPvContributors, rapCharmHad, invMassD0, cosThetaStar, absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate);
682+
registry.fill(HIST("hBeam"), invMassCharmHad, ptCharmHad, numPvContributors, std::abs(rapCharmHad), invMassD0, cosThetaStar, absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate);
678683
} else if constexpr (channel == charm_polarisation::DecayChannel::LcToPKPi) { // Lc+
679-
registry.fill(HIST("hBeam"), invMassCharmHad, ptCharmHad, numPvContributors, rapCharmHad, invMassKPiLc, cosThetaStar, absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate, charge);
684+
registry.fill(HIST("hBeam"), invMassCharmHad, ptCharmHad, numPvContributors, std::abs(rapCharmHad), invMassKPiLc, cosThetaStar, absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate, charge);
680685
if (activateTHnEulerPhiMonitor) {
681686
registry.fill(HIST("hEulerPhiBeam"), invMassCharmHad, ptCharmHad, invMassKPiLc, phiEuler, charge);
682687
}
@@ -729,15 +734,15 @@ struct TaskPolarisationCharmHadrons {
729734
if constexpr (!doMc) { // data
730735
if constexpr (withMl) { // with ML
731736
if constexpr (channel == charm_polarisation::DecayChannel::DstarToDzeroPi) { // D*+
732-
registry.fill(HIST("hRandom"), invMassCharmHad, ptCharmHad, numPvContributors, rapCharmHad, invMassD0, cosThetaStar, outputMl[0], /*outputMl[1],*/ outputMl[2], absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate);
737+
registry.fill(HIST("hRandom"), invMassCharmHad, ptCharmHad, numPvContributors, std::abs(rapCharmHad), invMassD0, cosThetaStar, outputMl[0], /*outputMl[1],*/ outputMl[2], absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate);
733738
} else if constexpr (channel == charm_polarisation::DecayChannel::LcToPKPi) { // Lc+
734-
registry.fill(HIST("hRandom"), invMassCharmHad, ptCharmHad, numPvContributors, rapCharmHad, invMassKPiLc, cosThetaStar, outputMl[0], /*outputMl[1],*/ outputMl[2], absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate, charge);
739+
registry.fill(HIST("hRandom"), invMassCharmHad, ptCharmHad, numPvContributors, std::abs(rapCharmHad), invMassKPiLc, cosThetaStar, outputMl[0], /*outputMl[1],*/ outputMl[2], absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate, charge);
735740
}
736741
} else { // without ML
737742
if constexpr (channel == charm_polarisation::DecayChannel::DstarToDzeroPi) { // D*+
738-
registry.fill(HIST("hRandom"), invMassCharmHad, ptCharmHad, numPvContributors, rapCharmHad, invMassD0, cosThetaStar, absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate);
743+
registry.fill(HIST("hRandom"), invMassCharmHad, ptCharmHad, numPvContributors, std::abs(rapCharmHad), invMassD0, cosThetaStar, absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate);
739744
} else if constexpr (channel == charm_polarisation::DecayChannel::LcToPKPi) { // Lc+
740-
registry.fill(HIST("hRandom"), invMassCharmHad, ptCharmHad, numPvContributors, rapCharmHad, invMassKPiLc, cosThetaStar, absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate, charge);
745+
registry.fill(HIST("hRandom"), invMassCharmHad, ptCharmHad, numPvContributors, std::abs(rapCharmHad), invMassKPiLc, cosThetaStar, absEtaMin, numItsClsMin, numTpcClsMin, isRotatedCandidate, charge);
741746
}
742747
}
743748
} else { // MC --> no distinction among channels, since rotational bkg not supported
@@ -1168,6 +1173,11 @@ struct TaskPolarisationCharmHadrons {
11681173
continue;
11691174
}
11701175

1176+
/// apply rapidity selection on the reconstructed candidate
1177+
if (std::abs(rapidity) > rapidityCut) {
1178+
continue;
1179+
}
1180+
11711181
float phiRandom = gRandom->Uniform(0.f, constants::math::TwoPI);
11721182
float thetaRandom = gRandom->Uniform(0.f, constants::math::PI);
11731183
ROOT::Math::PxPyPzMVector fourVecDau = ROOT::Math::PxPyPzMVector(pxDau, pyDau, pzDau, massDau);

0 commit comments

Comments
 (0)