Skip to content

Commit f7f259e

Browse files
authored
Refactor corrParticle configuration to use enum
1 parent 2497722 commit f7f259e

1 file changed

Lines changed: 12 additions & 5 deletions

File tree

PWGHF/HFC/TableProducer/correlatorXicHadrons.cxx

Lines changed: 12 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -102,6 +102,13 @@ enum class PDGChargeScale : size_t {
102102
Scale = 3u
103103
};
104104

105+
namespace CorrParticleType {
106+
enum Type : int8_t {
107+
PhysicalPrimary = 0,
108+
IdentifiedParticle = 1,
109+
V0 = 2
110+
};
111+
}
105112
// ============================================================================
106113
// HELPER FUNCTIONS
107114
// ============================================================================
@@ -406,7 +413,7 @@ struct HfCorrelatorXicHadrons {
406413
Configurable<bool> calEffEventWithCand{"calEffEventWithCand", true, "Calculate the efficiency of Xic candidate"};
407414
Configurable<float> eventFractionToAnalyze{"eventFractionToAnalyze", -1, "Fraction of events to analyze"};
408415
Configurable<int> particlePdg{"particlePdg", PDG_t::kProton, "PDG code of particle: kProton(2212), kPiPlus(211), kKPlus(321), kLambda0(3122)"};
409-
Configurable<int8_t> corrParticle{"corrParticle", 1, "put '0' for physical primary, '1' for indivisual identified particle, '2' for V0s"};
416+
Configurable<int8_t> corrParticle{"corrParticle", CorrParticleType::IdentifiedParticle, "0: physical primary, 1: individual identified particle, 2: V0s"};
410417
} cfgXicCand;
411418

412419
struct : ConfigurableGroup {
@@ -1372,7 +1379,7 @@ struct HfCorrelatorXicHadrons {
13721379
}
13731380

13741381
if constexpr (!isV0) {
1375-
if (cfgXicCand.corrParticle != 2 && !assocParticle.isGlobalTrackWoDCA()) {
1382+
if (cfgXicCand.corrParticle != CorrParticleType::V0 && !assocParticle.isGlobalTrackWoDCA()) {
13761383
continue;
13771384
}
13781385
if (cfgXicCand.pidTrkApplied) {
@@ -1493,15 +1500,15 @@ struct HfCorrelatorXicHadrons {
14931500
continue;
14941501
}
14951502

1496-
if (cfgXicCand.corrParticle == 1 && ((std::abs(particleAssoc.pdgCode()) != kElectron) && (std::abs(particleAssoc.pdgCode()) != kMuonMinus) && (std::abs(particleAssoc.pdgCode()) != kPiPlus) && (std::abs(particleAssoc.pdgCode()) != kKPlus) && (std::abs(particleAssoc.pdgCode()) != kProton))) {
1503+
if (cfgXicCand.corrParticle == CorrParticleType::IdentifiedParticle && ((std::abs(particleAssoc.pdgCode()) != kElectron) && (std::abs(particleAssoc.pdgCode()) != kMuonMinus) && (std::abs(particleAssoc.pdgCode()) != kPiPlus) && (std::abs(particleAssoc.pdgCode()) != kKPlus) && (std::abs(particleAssoc.pdgCode()) != kProton))) {
14971504
continue;
14981505
}
14991506

1500-
if (cfgXicCand.corrParticle == 2 && (std::abs(particleAssoc.pdgCode()) != cfgXicCand.particlePdg)) {
1507+
if (cfgXicCand.corrParticle == CorrParticleType::V0 && (std::abs(particleAssoc.pdgCode()) != cfgXicCand.particlePdg)) {
15011508
continue;
15021509
}
15031510

1504-
if (cfgXicCand.corrParticle == 1 && !particleAssoc.isPhysicalPrimary()) {
1511+
if (cfgXicCand.corrParticle == CorrParticleType::IdentifiedParticle && !particleAssoc.isPhysicalPrimary()) {
15051512
continue;
15061513
}
15071514

0 commit comments

Comments
 (0)