Skip to content

Commit bd6a438

Browse files
committed
candidateCreator3Prong.cxx: Simplify masses
1 parent bda79e0 commit bd6a438

File tree

1 file changed

+9
-23
lines changed

1 file changed

+9
-23
lines changed

PWGHF/TableProducer/candidateCreator3Prong.cxx

Lines changed: 9 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -129,19 +129,9 @@ struct HfCandidateCreator3Prong {
129129
Service<o2::ccdb::BasicCCDBManager> ccdb;
130130

131131
int runNumber{0};
132-
float toMicrometers = 10000.; // from cm to µm
133-
double massP{0.};
134-
double massPi{0.};
135-
double massK{0.};
136-
double massPKPi{0.};
137-
double massPiKP{0.};
138-
double massPiKPi{0.};
139-
double massKKPi{0.};
140-
double massPiKK{0.};
141-
double massKPi{0.};
142-
double massPiK{0.};
143132
double bz{0.};
144133

134+
const float toMicrometers = 10000.; // from cm to µm
145135
constexpr static float UndefValueFloat{-999.f};
146136

147137
using FilteredHf3Prongs = soa::Filtered<aod::Hf3Prongs>;
@@ -223,10 +213,6 @@ struct HfCandidateCreator3Prong {
223213
// init HF event selection helper
224214
hfEvSel.init(registry);
225215

226-
massP = MassProton;
227-
massPi = MassPiPlus;
228-
massK = MassKPlus;
229-
230216
// Configure DCAFitterN
231217
// df.setBz(bz);
232218
df.setPropagateToPCA(propagateToPCA);
@@ -418,14 +404,14 @@ struct HfCandidateCreator3Prong {
418404
// fill histograms
419405
if (fillHistograms) {
420406
// calculate invariant mass
421-
auto arrayMomenta = std::array{pvec0, pvec1, pvec2};
422-
massPKPi = RecoDecay::m(arrayMomenta, std::array{massP, massK, massPi});
423-
massPiKP = RecoDecay::m(arrayMomenta, std::array{massPi, massK, massP});
424-
massPiKPi = RecoDecay::m(arrayMomenta, std::array{massPi, massK, massPi});
425-
massKKPi = RecoDecay::m(arrayMomenta, std::array{massK, massK, massPi});
426-
massPiKK = RecoDecay::m(arrayMomenta, std::array{massPi, massK, massK});
427-
massKPi = RecoDecay::m(std::array{arrayMomenta.at(1), arrayMomenta.at(2)}, std::array{massK, massPi});
428-
massPiK = RecoDecay::m(std::array{arrayMomenta.at(0), arrayMomenta.at(1)}, std::array{massPi, massK});
407+
const auto arrayMomenta = std::array{pvec0, pvec1, pvec2};
408+
const auto massPKPi = RecoDecay::m(arrayMomenta, std::array{MassProton, MassKPlus, MassPiPlus});
409+
const auto massPiKP = RecoDecay::m(arrayMomenta, std::array{MassPiPlus, MassKPlus, MassProton});
410+
const auto massPiKPi = RecoDecay::m(arrayMomenta, std::array{MassPiPlus, MassKPlus, MassPiPlus});
411+
const auto massKKPi = RecoDecay::m(arrayMomenta, std::array{MassKPlus, MassKPlus, MassPiPlus});
412+
const auto massPiKK = RecoDecay::m(arrayMomenta, std::array{MassPiPlus, MassKPlus, MassKPlus});
413+
const auto massKPi = RecoDecay::m(std::array{arrayMomenta.at(1), arrayMomenta.at(2)}, std::array{MassKPlus, MassPiPlus});
414+
const auto massPiK = RecoDecay::m(std::array{arrayMomenta.at(0), arrayMomenta.at(1)}, std::array{MassPiPlus, MassKPlus});
429415
registry.fill(HIST("hMass3PiKPi"), massPiKPi);
430416
registry.fill(HIST("hMass3PKPi"), massPKPi);
431417
registry.fill(HIST("hMass3PiKP"), massPiKP);

0 commit comments

Comments
 (0)