Skip to content

Commit a815f52

Browse files
echizzalechizzal
authored
Remove function to calculate inv mass (#6986)
1 parent 9fdee58 commit a815f52

1 file changed

Lines changed: 25 additions & 46 deletions

File tree

‎EventFiltering/PWGCF/CFFilterPPPhi.cxx‎

Lines changed: 25 additions & 46 deletions
Original file line numberDiff line numberDiff line change
@@ -202,6 +202,7 @@ struct CFFillterPPPhi {
202202
// phi cuts
203203
registry.add("TrackCuts/Phi/fPtPhiBefore", "Transverse momentum V0s;p_{T} (GeV/c);Entries", HistType::kTH1F, {{1000, 0, 10}});
204204
registry.add("TrackCuts/Phi/fInvMassPhiBefore", "Invariant mass V0s;M_{KK};Entries", HistType::kTH1F, {{7000, 0.8, 1.5}});
205+
205206
registry.add("TrackCuts/Phi/fEtaPhiBefore", "Pseudorapidity of V0;#eta;Entries", HistType::kTH1F, {{1000, -2, 2}});
206207
registry.add("TrackCuts/Phi/fPhiPhiBefore", "Azimuthal angle of V0;#phi;Entries", HistType::kTH1F, {{720, 0, TMath::TwoPi()}});
207208

@@ -211,12 +212,13 @@ struct CFFillterPPPhi {
211212
registry.add("TrackCuts/Phi/fPhiPhi", "Azimuthal angle of V0;#phi;Entries", HistType::kTH1F, {{720, 0, TMath::TwoPi()}});
212213

213214
// phi daughter
215+
registry.add("TrackCuts/Phi/PosDaughter/Pt", "Transverse momentum Pos Daugh tracks;p_{T} (GeV/c);Entries", HistType::kTH1F, {{1000, 0, 10}});
214216
registry.add("TrackCuts/Phi/PosDaughter/Eta", "Phi Pos Daugh Eta;#eta;Entries", HistType::kTH1F, {{1000, -2, 2}});
215-
registry.add("TrackCuts/Phi/PosDaughter/DCAXY", "Phi Pos Daugh DCAXY;DCA_{XY};Entries", HistType::kTH1F, {{1000, -2.5f, 2.5f}});
216-
registry.add("TrackCuts/Phi/PosDaughter/fTPCncls", "Phi Pos Daugh TPCncls;TPC Clusters;Entries", HistType::kTH1F, {{163, -1.0f, 162.0f}});
217+
registry.add("TrackCuts/Phi/PosDaughter/Phi", "Azimuthal angle of Pos Daugh tracks;#phi;Entries", HistType::kTH1F, {{720, 0, TMath::TwoPi()}});
218+
219+
registry.add("TrackCuts/Phi/NegDaughter/Pt", "Transverse momentum Neg Daugh tracks;p_{T} (GeV/c);Entries", HistType::kTH1F, {{1000, 0, 10}});
217220
registry.add("TrackCuts/Phi/NegDaughter/Eta", "Phi Neg Daugh Eta;#eta;Entries", HistType::kTH1F, {{1000, -2, 2}});
218-
registry.add("TrackCuts/Phi/NegDaughter/DCAXY", "Phi Neg Daugh DCAXY;DCA_{XY};Entries", HistType::kTH1F, {{1000, -2.5f, 2.5f}});
219-
registry.add("TrackCuts/Phi/NegDaughter/fTPCncls", "Phi Neg Daugh TPCncls;TPC Clusters;Entries", HistType::kTH1F, {{163, -1.0f, 162.0f}});
221+
registry.add("TrackCuts/Phi/NegDaughter/Phi", "Azimuthal angle of Neg Daugh tracks;#phi;Entries", HistType::kTH1F, {{720, 0, TMath::TwoPi()}});
220222

221223
// triggers
222224
registry.add("ppphi/fMultiplicity", "Multiplicity of all triggered events;Mult;Entries", HistType::kTH1F, {{1000, 0, 1000}});
@@ -245,7 +247,7 @@ struct CFFillterPPPhi {
245247
bool isSelectedTrackProton(T const& track)
246248
{
247249
bool isSelected = false;
248-
if (track.pt() < ConfTrkPtPrUp.value && track.pt() > ConfTrkPtPrDown.value && std::abs(track.eta()) <= ConfTrkEtaPr.value && std::abs(track.dcaXY()) <= ConfTrkDCAxyPr.value && std::abs(track.dcaZ()) <= ConfTrkDCAzPr.value && track.tpcNClsCrossedRows() >= ConfNCrossedPr.value && track.tpcNClsFound() >= ConfNClusPr.value && track.tpcCrossedRowsOverFindableCls() >= ConfTrkTPCfClsPr.value) {
250+
if (track.pt() <= ConfTrkPtPrUp.value && track.pt() >= ConfTrkPtPrDown.value && std::abs(track.eta()) <= ConfTrkEtaPr.value && std::abs(track.dcaXY()) <= ConfTrkDCAxyPr.value && std::abs(track.dcaZ()) <= ConfTrkDCAzPr.value && track.tpcNClsCrossedRows() >= ConfNCrossedPr.value && track.tpcNClsFound() >= ConfNClusPr.value && track.tpcCrossedRowsOverFindableCls() >= ConfTrkTPCfClsPr.value) {
249251
if (track.tpcInnerParam() < ConfTrkPTPCPrThr.value && std::abs(track.tpcNSigmaPr()) <= ConfTrkPrSigmaPID.value) {
250252
isSelected = true;
251253
}
@@ -260,7 +262,7 @@ struct CFFillterPPPhi {
260262
bool isSelectedTrackKaon(T const& track)
261263
{
262264
bool isSelected = false;
263-
if (track.pt() < ConfTrkPtKaUp.value && track.pt() > ConfTrkPtKaDown.value && std::abs(track.eta()) <= ConfTrkEtaKa.value && std::abs(track.dcaXY()) <= ConfTrkDCAxyKa.value && std::abs(track.dcaZ()) <= ConfTrkDCAzKa.value && track.tpcNClsCrossedRows() >= ConfNCrossedKa.value && track.tpcNClsFound() >= ConfNClusKa.value && track.tpcCrossedRowsOverFindableCls() >= ConfTrkTPCfClsKa.value) {
265+
if (track.pt() <= ConfTrkPtKaUp.value && track.pt() >= ConfTrkPtKaDown.value && std::abs(track.eta()) <= ConfTrkEtaKa.value && std::abs(track.dcaXY()) <= ConfTrkDCAxyKa.value && std::abs(track.dcaZ()) <= ConfTrkDCAzKa.value && track.tpcNClsCrossedRows() >= ConfNCrossedKa.value && track.tpcNClsFound() >= ConfNClusKa.value && track.tpcCrossedRowsOverFindableCls() >= ConfTrkTPCfClsKa.value) {
264266
if (track.tpcInnerParam() < ConfTrkPTPCKaThr.value && std::abs(track.tpcNSigmaKa()) <= ConfTrkKaSigmaPID.value) {
265267
isSelected = true;
266268
}
@@ -272,38 +274,10 @@ struct CFFillterPPPhi {
272274
}
273275

274276
float mMassProton = o2::constants::physics::MassProton;
275-
float mMassKaon = o2::constants::physics::MassKPlus;
276-
float mMassPhi = o2::constants::physics::MassPhi;
277-
278-
// float getInvMass(const ROOT::Math::PtEtaPhiMVector part1,
279-
// const ROOT::Math::PtEtaPhiMVector part2)
280-
// {
281-
// const ROOT::Math::PtEtaPhiMVector trackSum = part1 + part2;
282-
// return trackSum.M();
283-
// }
284-
285-
float CalculateInvMass(const ROOT::Math::PtEtaPhiMVector part1,
286-
const ROOT::Math::PtEtaPhiMVector part2)
287-
{
288-
Double_t invMass = 0;
289-
290-
float massDP = mMassKaon;
291-
float massDN = mMassKaon;
292-
293-
float EDaugP = TMath::Sqrt(
294-
massDP * massDP + part1.Px() * part1.Px() + part1.Py() * part1.Py() + part1.Pz() * part1.Pz());
295-
float EDaugN = TMath::Sqrt(
296-
massDN * massDN + part2.Px() * part2.Px() + part2.Py() * part2.Py() + part2.Pz() * part2.Pz());
277+
float mMassKaonPlus = o2::constants::physics::MassKPlus;
278+
float mMassKaonMinus = o2::constants::physics::MassKMinus;
297279

298-
float energysum = EDaugP + EDaugN;
299-
float pSum2 = (part2.Px() + part1.Px()) * (part2.Px() + part1.Px()) +
300-
301-
(part2.Py() + part1.Py()) * (part2.Py() + part1.Py()) +
302-
303-
(part2.Pz() + part1.Pz()) * (part2.Pz() + part1.Pz());
304-
invMass = TMath::Sqrt(energysum * energysum - pSum2);
305-
return invMass;
306-
}
280+
float mMassPhi = o2::constants::physics::MassPhi;
307281

308282
float getkstar(const ROOT::Math::PtEtaPhiMVector part1,
309283
const ROOT::Math::PtEtaPhiMVector part2)
@@ -421,8 +395,9 @@ struct CFFillterPPPhi {
421395
}
422396

423397
if (isSelectedTrackKaon(track)) {
424-
ROOT::Math::PtEtaPhiMVector temp(track.pt(), track.eta(), track.phi(), mMassKaon);
398+
ROOT::Math::PtEtaPhiMVector temp(track.pt(), track.eta(), track.phi(), mMassKaonPlus);
425399
if (track.sign() > 0) {
400+
temp.SetM(mMassKaonPlus);
426401
kaons.push_back(temp);
427402
registry.fill(HIST("TrackCuts/Kaon/fPKaon"), track.p());
428403
registry.fill(HIST("TrackCuts/Kaon/fPTPCKaon"), track.tpcInnerParam());
@@ -445,6 +420,8 @@ struct CFFillterPPPhi {
445420
// KaonIndex.push_back(track.globalIndex());
446421
}
447422
if (track.sign() < 0) {
423+
temp.SetM(mMassKaonMinus);
424+
448425
antikaons.push_back(temp);
449426
registry.fill(HIST("TrackCuts/AntiKaon/fPAntiKaon"), track.p());
450427
registry.fill(HIST("TrackCuts/AntiKaon/fPTPCAntiKaon"), track.tpcInnerParam());
@@ -472,29 +449,31 @@ struct CFFillterPPPhi {
472449

473450
for (const auto& postrack : kaons) {
474451
for (const auto& negtrack : antikaons) {
452+
475453
ROOT::Math::PtEtaPhiMVector temp = postrack + negtrack;
476-
temp.SetM(mMassPhi);
454+
// temp.SetM(mMassPhi);
455+
registry.fill(HIST("TrackCuts/Phi/fInvMassPhiBefore"), temp.M());
477456

478-
registry.fill(HIST("TrackCuts/Phi/fInvMassPhiBefore"), CalculateInvMass(postrack, negtrack));
479457
registry.fill(HIST("TrackCuts/Phi/fPtPhiBefore"), temp.pt());
480458
registry.fill(HIST("TrackCuts/Phi/fEtaPhiBefore"), temp.eta());
481459
registry.fill(HIST("TrackCuts/Phi/fPhiPhiBefore"), temp.phi());
482460

483-
if ((CalculateInvMass(postrack, negtrack) > ConfResoInvMassLowLimit.value) && (CalculateInvMass(postrack, negtrack) < ConfResoInvMassUpLimit.value)) {
461+
if ((temp.M() >= ConfResoInvMassLowLimit.value) && (temp.M() <= ConfResoInvMassUpLimit.value)) {
484462
// ROOT::Math::PtEtaPhiMVector temp = postrack + negtrack;
485463
phi.push_back(temp);
486464

487465
registry.fill(HIST("TrackCuts/Phi/fPtPhi"), temp.pt());
488466
registry.fill(HIST("TrackCuts/Phi/fEtaPhi"), temp.eta());
489467
registry.fill(HIST("TrackCuts/Phi/fPhiPhi"), temp.phi());
490-
registry.fill(HIST("TrackCuts/Phi/fInvMassPhi"), CalculateInvMass(postrack, negtrack));
468+
registry.fill(HIST("TrackCuts/Phi/fInvMassPhi"), temp.M());
491469

470+
registry.fill(HIST("TrackCuts/Phi/PosDaughter/Pt"), postrack.pt());
492471
registry.fill(HIST("TrackCuts/Phi/PosDaughter/Eta"), postrack.eta());
493-
// registry.fill(HIST("TrackCuts/Phi/PosDaughter/DCAXY"), postrack.dcaXY());
494-
// registry.fill(HIST("TrackCuts/Phi/PosDaughter/fTPCncls"), postrack.tpcNClsFound());
472+
registry.fill(HIST("TrackCuts/Phi/PosDaughter/Phi"), postrack.phi());
473+
474+
registry.fill(HIST("TrackCuts/Phi/NegDaughter/Pt"), negtrack.pt());
495475
registry.fill(HIST("TrackCuts/Phi/NegDaughter/Eta"), negtrack.eta());
496-
// registry.fill(HIST("TrackCuts/Phi/NegDaughter/DCAXY"), negtrack.dcaXY());
497-
// registry.fill(HIST("TrackCuts/Phi/NegDaughter/fTPCncls"), negtrack.tpcNClsFound());
476+
registry.fill(HIST("TrackCuts/Phi/NegDaughter/Phi"), negtrack.phi());
498477
}
499478
}
500479
}

0 commit comments

Comments
 (0)