Skip to content

Commit 2cfd001

Browse files
author
Chiara De Martin
committed
use PDG_t for PDG codes instead of numbers
1 parent ab9da47 commit 2cfd001

File tree

1 file changed

+64
-54
lines changed

1 file changed

+64
-54
lines changed

‎PWGLF/Tasks/QC/v0cascadesqa.cxx‎

Lines changed: 64 additions & 54 deletions
Original file line numberDiff line numberDiff line change
@@ -157,6 +157,13 @@ struct v0cascadesQA {
157157
histos_eve.add("GeneratedParticles", "GeneratedParticles", {HistType::kTH3F, {{14, 0.0f, 14.0f}, {100, 0, 10}, {100, 0.f, 50.f}}});
158158
histos_eve.add("hEventCounter", "hEventCounter", {HistType::kTH1F, {{2, 0.0f, 2.0f}}});
159159
histos_eve.add("hEventCounterMC", "hEventCounterMC", {HistType::kTH1F, {{2, 0.0f, 2.0f}}});
160+
histos_eve.add("GenK0sPtVsEta", "GenK0sPtVsEta", {HistType::kTH2F, {axisPt, axisEta}});
161+
histos_eve.add("GenLambdaPtVsEta", "GenLambdaPtVsEta", {HistType::kTH2F, {axisPt, axisEta}});
162+
histos_eve.add("GenAntiLambdaPtVsEta", "GenAntiLambdaPtVsEta", {HistType::kTH2F, {axisPt, axisEta}});
163+
histos_eve.add("GenXiMinusPtVsEta", "GenXiMinusPtVsEta", {HistType::kTH2F, {axisPt, axisEta}});
164+
histos_eve.add("GenXiPlusPtVsEta", "GenXiPlusPtVsEta", {HistType::kTH2F, {axisPt, axisEta}});
165+
histos_eve.add("GenOmegaMinusPtVsEta", "GenOmegaMinusPtVsEta", {HistType::kTH2F, {axisPt, axisEta}});
166+
histos_eve.add("GenOmegaPlusPtVsEta", "GenOmegaPlusPtVsEta", {HistType::kTH2F, {axisPt, axisEta}});
160167

161168
histos_V0.add("CosPA", "CosPA", HistType::kTH1F, {axisV0CosPA});
162169
histos_V0.add("V0Radius", "V0Radius", HistType::kTH1F, {axisV0Radius});
@@ -323,23 +330,38 @@ struct v0cascadesQA {
323330
}
324331
double R_Decay = TMath::Sqrt(vx * vx + vy * vy);
325332

333+
if (mcparticle.pdgCode() == PDG_t::kK0Short)
334+
histos_eve.fill(HIST("GenK0sPtVsEta"), mcparticle.pt(), mcparticle.eta());
335+
if (mcparticle.pdgCode() == PDG_t::kLambda0)
336+
histos_eve.fill(HIST("GenLambdaPtVsEta"), mcparticle.pt(), mcparticle.eta());
337+
if (mcparticle.pdgCode() == PDG_t::kLambda0Bar)
338+
histos_eve.fill(HIST("GenAntiLambdaPtVsEta"), mcparticle.pt(), mcparticle.eta());
339+
if (mcparticle.pdgCode() == PDG_t::kXiMinus)
340+
histos_eve.fill(HIST("GenXiMinusPtVsEta"), mcparticle.pt(), mcparticle.eta());
341+
if (mcparticle.pdgCode() == PDG_t::kXiPlusBar)
342+
histos_eve.fill(HIST("GenXiPlusPtVsEta"), mcparticle.pt(), mcparticle.eta());
343+
if (mcparticle.pdgCode() == PDG_t::kOmegaMinus)
344+
histos_eve.fill(HIST("GenOmegaMinusPtVsEta"), mcparticle.pt(), mcparticle.eta());
345+
if (mcparticle.pdgCode() == PDG_t::kOmegaPlusBar)
346+
histos_eve.fill(HIST("GenOmegaPlusPtVsEta"), mcparticle.pt(), mcparticle.eta());
347+
326348
if (mcparticle.isPhysicalPrimary() && TMath::Abs(mcparticle.y()) < V0_rapidity) {
327-
if (mcparticle.pdgCode() == 310)
328-
histos_eve.fill(HIST("GeneratedParticles"), 0.5, mcparticle.pt(), R_Decay); // K0s
329-
if (mcparticle.pdgCode() == 3122)
330-
histos_eve.fill(HIST("GeneratedParticles"), 2.5, mcparticle.pt(), R_Decay); // Lambda
331-
if (mcparticle.pdgCode() == -3122)
332-
histos_eve.fill(HIST("GeneratedParticles"), 4.5, mcparticle.pt(), R_Decay); // AntiLambda
349+
if (mcparticle.pdgCode() == PDG_t::kK0Short)
350+
histos_eve.fill(HIST("GeneratedParticles"), 0.5, mcparticle.pt(), R_Decay);
351+
if (mcparticle.pdgCode() == PDG_t::kLambda0)
352+
histos_eve.fill(HIST("GeneratedParticles"), 2.5, mcparticle.pt(), R_Decay);
353+
if (mcparticle.pdgCode() == PDG_t::kLambda0Bar)
354+
histos_eve.fill(HIST("GeneratedParticles"), 4.5, mcparticle.pt(), R_Decay);
333355
}
334356
if (mcparticle.isPhysicalPrimary() && TMath::Abs(mcparticle.y()) < Casc_rapidity) {
335-
if (mcparticle.pdgCode() == 3312)
336-
histos_eve.fill(HIST("GeneratedParticles"), 6.5, mcparticle.pt(), R_Decay); // Xi-
337-
if (mcparticle.pdgCode() == -3312)
338-
histos_eve.fill(HIST("GeneratedParticles"), 8.5, mcparticle.pt(), R_Decay); // Xi+
339-
if (mcparticle.pdgCode() == 3334)
340-
histos_eve.fill(HIST("GeneratedParticles"), 10.5, mcparticle.pt(), R_Decay); // Omega-
341-
if (mcparticle.pdgCode() == -3334)
342-
histos_eve.fill(HIST("GeneratedParticles"), 12.5, mcparticle.pt(), R_Decay); // Omega+
357+
if (mcparticle.pdgCode() == PDG_t::kXiMinus)
358+
histos_eve.fill(HIST("GeneratedParticles"), 6.5, mcparticle.pt(), R_Decay);
359+
if (mcparticle.pdgCode() == PDG_t::kXiPlusBar)
360+
histos_eve.fill(HIST("GeneratedParticles"), 8.5, mcparticle.pt(), R_Decay);
361+
if (mcparticle.pdgCode() == PDG_t::kOmegaMinus)
362+
histos_eve.fill(HIST("GeneratedParticles"), 10.5, mcparticle.pt(), R_Decay);
363+
if (mcparticle.pdgCode() == PDG_t::kOmegaPlusBar)
364+
histos_eve.fill(HIST("GeneratedParticles"), 12.5, mcparticle.pt(), R_Decay);
343365
}
344366
}
345367
}
@@ -486,7 +508,7 @@ struct v0cascadesQA {
486508
}
487509
auto v0mcparticle = v0.mcParticle();
488510
Int_t lPDG = 0;
489-
if (TMath::Abs(v0mcparticle.pdgCode()) == 310 || TMath::Abs(v0mcparticle.pdgCode()) == 3122) {
511+
if (TMath::Abs(v0mcparticle.pdgCode()) == 310 || TMath::Abs(v0mcparticle.pdgCode()) == PDG_t::kLambda0) {
490512
lPDG = v0mcparticle.pdgCode();
491513
}
492514

@@ -505,12 +527,12 @@ struct v0cascadesQA {
505527
histos_V0.fill(HIST("InvMassK0STrue"), v0.pt(), v0.v0radius(), v0.mK0Short());
506528
}
507529
}
508-
if (lPDG == 3122) {
530+
if (lPDG == PDG_t::kLambda0) {
509531
if (TMath::Abs(v0.yLambda()) < V0_rapidity && CtauLambda < lifetimecut->get("lifetimecutLambda")) {
510532
histos_V0.fill(HIST("InvMassLambdaTrue"), v0.pt(), v0.v0radius(), v0.mLambda());
511533
}
512534
}
513-
if (lPDG == -3122) {
535+
if (lPDG == PDG_t::kLambda0Bar) {
514536
if (TMath::Abs(v0.yLambda()) < V0_rapidity && CtauLambda < lifetimecut->get("lifetimecutLambda")) {
515537
histos_V0.fill(HIST("InvMassAntiLambdaTrue"), v0.pt(), v0.v0radius(), v0.mAntiLambda());
516538
}
@@ -530,31 +552,19 @@ struct v0cascadesQA {
530552
return;
531553
}
532554

533-
int dauEtaFlag = 0;
534555
for (auto& casc : Cascades) {
535556
auto bachelor = casc.bachelor_as<DaughterTracks>();
536557
auto posdau = casc.posTrack_as<DaughterTracks>();
537558
auto negdau = casc.negTrack_as<DaughterTracks>();
538559

539-
if (posdau.eta() < 0. && negdau.eta() < 0. && bachelor.eta() < 0.) {
540-
dauEtaFlag = -1;
541-
} else if (posdau.eta() >= 0. && negdau.eta() >= 0. && bachelor.eta() >= 0.) {
542-
dauEtaFlag = 1;
543-
} else {
544-
dauEtaFlag = 0;
545-
}
546-
547560
// check TPC
548561
if (checkDauTPC && (!posdau.hasTPC() || !negdau.hasTPC() || !bachelor.hasTPC())) {
549562
continue;
550563
}
551564

552-
// histos_Casc.fill(HIST("XiProgSelections"), );
553-
// histos_Casc.fill(HIST("OmegaProgSelections"), );
554565
histos_Casc.fill(HIST("CascCosPA"), casc.casccosPA(collision.posX(), collision.posY(), collision.posZ()), casc.sign());
555566
histos_Casc.fill(HIST("V0CosPA"), casc.v0cosPA(collision.posX(), collision.posY(), collision.posZ()), casc.sign());
556567

557-
// double v0cospatoxi = RecoDecay::CPA(array{casc.x(), casc.y(), casc.z()}, array{casc.xlambda(), casc.ylambda(), casc.zlambda()}, array{v0.px(), v0.py(), v0.pz()});
558568
double v0cospatoxi = RecoDecay::cpa(array{casc.x(), casc.y(), casc.z()}, array{casc.xlambda(), casc.ylambda(), casc.zlambda()}, array{casc.pxpos() + casc.pxneg(), casc.pypos() + casc.pyneg(), casc.pzpos() + casc.pzneg()});
559569

560570
histos_Casc.fill(HIST("V0CosPAToXi"), v0cospatoxi, casc.sign());
@@ -673,38 +683,38 @@ struct v0cascadesQA {
673683
bool isXiMinusCascade = MomOfBachIsPrimary && !(MomOfNegIsPrimary) && !(MomOfPosIsPrimary) &&
674684
particleMotherOfNeg == particleMotherOfPos &&
675685
particleMotherOfV0 == particleMotherOfBach &&
676-
particleMotherOfBach.pdgCode() == 3312 &&
677-
bachelor.pdgCode() == -211 &&
678-
particleMotherOfNeg.pdgCode() == 3122 &&
679-
mcnegtrack.pdgCode() == -211 &&
680-
mcpostrack.pdgCode() == 2212;
686+
particleMotherOfBach.pdgCode() == PDG_t::kXiMinus &&
687+
bachelor.pdgCode() == PDG_t::kPiMinus &&
688+
particleMotherOfNeg.pdgCode() == PDG_t::kLambda0 &&
689+
mcnegtrack.pdgCode() == PDG_t::kPiMinus &&
690+
mcpostrack.pdgCode() == PDG_t::kProton;
681691

682692
bool isOmegaMinusCascade = MomOfBachIsPrimary && !(MomOfNegIsPrimary) && !(MomOfPosIsPrimary) &&
683693
particleMotherOfNeg == particleMotherOfPos &&
684694
particleMotherOfV0 == particleMotherOfBach &&
685-
particleMotherOfBach.pdgCode() == 3334 &&
686-
bachelor.pdgCode() == -321 &&
687-
particleMotherOfNeg.pdgCode() == 3122 &&
688-
mcnegtrack.pdgCode() == -211 &&
689-
mcpostrack.pdgCode() == 2212;
695+
particleMotherOfBach.pdgCode() == PDG_t::kOmegaMinus &&
696+
bachelor.pdgCode() == PDG_t::kKMinus &&
697+
particleMotherOfNeg.pdgCode() == PDG_t::kLambda0 &&
698+
mcnegtrack.pdgCode() == PDG_t::kPiMinus &&
699+
mcpostrack.pdgCode() == PDG_t::kProton;
690700

691701
bool isXiPlusCascade = MomOfBachIsPrimary && !(MomOfNegIsPrimary) && !(MomOfPosIsPrimary) &&
692702
particleMotherOfNeg == particleMotherOfPos &&
693703
particleMotherOfV0 == particleMotherOfBach &&
694-
particleMotherOfBach.pdgCode() == -3312 &&
695-
bachelor.pdgCode() == 211 &&
696-
particleMotherOfNeg.pdgCode() == -3122 &&
697-
mcnegtrack.pdgCode() == -2212 &&
698-
mcpostrack.pdgCode() == 211;
704+
particleMotherOfBach.pdgCode() == PDG_t::kXiPlusBar &&
705+
bachelor.pdgCode() == PDG_t::kPiPlus &&
706+
particleMotherOfNeg.pdgCode() == PDG_t::kLambda0Bar &&
707+
mcnegtrack.pdgCode() == PDG_t::kProtonBar &&
708+
mcpostrack.pdgCode() == PDG_t::kPiPlus;
699709

700710
bool isOmegaPlusCascade = MomOfBachIsPrimary && !(MomOfNegIsPrimary) && !(MomOfPosIsPrimary) &&
701711
particleMotherOfNeg == particleMotherOfPos &&
702712
particleMotherOfV0 == particleMotherOfBach &&
703-
particleMotherOfBach.pdgCode() == -3334 &&
704-
bachelor.pdgCode() == 321 &&
705-
particleMotherOfNeg.pdgCode() == -3122 &&
706-
mcnegtrack.pdgCode() == -2212 &&
707-
mcpostrack.pdgCode() == 211;
713+
particleMotherOfBach.pdgCode() == PDG_t::kOmegaPlusBar &&
714+
bachelor.pdgCode() == PDG_t::kKPlus &&
715+
particleMotherOfNeg.pdgCode() == PDG_t::kLambda0Bar &&
716+
mcnegtrack.pdgCode() == PDG_t::kProtonBar &&
717+
mcpostrack.pdgCode() == PDG_t::kPiPlus;
708718

709719
if (isXiMinusCascade) {
710720
histos_Casc.fill(HIST("QA_XiMinusCandidates"), 8.5);
@@ -741,11 +751,11 @@ struct v0cascadesQA {
741751
if ((particleMotherOfV0 == particleMotherOfBach)) {
742752
histos_Casc.fill(HIST("QA_XiMinusCandidates"), 6.5);
743753
}
744-
if (particleMotherOfBach.pdgCode() == 3312 &&
745-
bachelor.pdgCode() == -211 &&
746-
particleMotherOfNeg.pdgCode() == 3122 &&
747-
mcnegtrack.pdgCode() == -211 &&
748-
mcpostrack.pdgCode() == 2212) {
754+
if (particleMotherOfBach.pdgCode() == PDG_t::kXiMinus &&
755+
bachelor.pdgCode() == PDG_t::kPiMinus &&
756+
particleMotherOfNeg.pdgCode() == PDG_t::kLambda0 &&
757+
mcnegtrack.pdgCode() == PDG_t::kPiMinus &&
758+
mcpostrack.pdgCode() == PDG_t::kProton) {
749759
histos_Casc.fill(HIST("QA_XiMinusCandidates"), 7.5);
750760
}
751761
}

0 commit comments

Comments
 (0)