Skip to content

Commit 2e678d5

Browse files
zhangbiao-phyzhangbiao-phy
authored
[PWGHF] add more Pid column and Lc mass in the derived table of taskCd (#15652)
1 parent 3e7c82c commit 2e678d5

File tree

2 files changed

+17
-16
lines changed

2 files changed

+17
-16
lines changed

‎PWGHF/D2H/Tasks/taskCd.cxx‎

Lines changed: 16 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,8 @@ namespace o2::aod
6363
namespace full
6464
{
6565
// Candidate kinematics
66-
DECLARE_SOA_COLUMN(M, m, float); //! Invariant mass of candidate (GeV/c^2)
66+
DECLARE_SOA_COLUMN(MassCd, massCd, float); //! Invariant mass of cd candidate (GeV/c^2)
67+
DECLARE_SOA_COLUMN(MassLc, massLc, float); //! Invariant mass of lc candidate (GeV/c^2)
6768
DECLARE_SOA_COLUMN(Pt, pt, float); //! Transverse momentum of candidate (GeV/c)
6869
DECLARE_SOA_COLUMN(Eta, eta, float); //! eta of candidate (GeV/c)
6970
DECLARE_SOA_COLUMN(Phi, phi, float); //! phi of candidate (GeV/c)
@@ -79,6 +80,7 @@ DECLARE_SOA_COLUMN(Cpa, cpa, float); //! Cosine of po
7980
DECLARE_SOA_COLUMN(CpaXY, cpaXY, float); //! Cosine of pointing angle in XY plane
8081
DECLARE_SOA_COLUMN(Chi2PCA, chi2PCA, float); //! chi2PCA
8182
DECLARE_SOA_COLUMN(NSigmaTpcDe, nSigmaTpcDe, float); //! TPC nσ for deuteron hypothesis
83+
DECLARE_SOA_COLUMN(NSigmaTpcPr, nSigmaTpcPr, float); //! TPC nσ for proton hypothesis
8284
DECLARE_SOA_COLUMN(NSigmaTpcKa, nSigmaTpcKa, float); //! TPC nσ for kaon hypothesis
8385
DECLARE_SOA_COLUMN(NSigmaTpcPi, nSigmaTpcPi, float); //! TPC nσ for pion hypothesis
8486
DECLARE_SOA_COLUMN(NSigmaItsDe, nSigmaItsDe, float); //! ITS nσ for deuteron hypothesis
@@ -100,7 +102,8 @@ DECLARE_SOA_COLUMN(TimeStamp, timeStamp, int64_t); //! Timestamp fo
100102

101103
// Full table: include ALL columns declared above
102104
DECLARE_SOA_TABLE(HfCandCd, "AOD", "HFCANDCD",
103-
full::M,
105+
full::MassCd,
106+
full::MassLc,
104107
full::Pt,
105108
full::Eta,
106109
full::Phi,
@@ -114,15 +117,13 @@ DECLARE_SOA_TABLE(HfCandCd, "AOD", "HFCANDCD",
114117
full::Cpa,
115118
full::Chi2PCA,
116119
full::NSigmaTpcDe,
120+
full::NSigmaTpcPr,
117121
full::NSigmaItsDe,
118122
full::NSigmaTofDe,
119-
full::NItsClusters,
120123
full::NItsNClusterSize,
121-
full::NTpcClusters,
122124
full::NTpcSignalsDe,
123125
full::NTpcSignalsPi,
124126
full::NTpcSignalsKa,
125-
full::NItsSignalsDe,
126127
full::CandidateSelFlag,
127128
full::Cent);
128129
} // namespace o2::aod
@@ -219,6 +220,7 @@ struct HfTaskCd {
219220
registry.add("Data/hImpParErrProng1", "3-prong candidates;prong 1 impact parameter error (cm);entries", {HistType::kTH2F, {{100, -1., 1.}, {binsPt, "#it{p}_{T} (GeV/#it{c})"}}});
220221
registry.add("Data/hImpParErrProng2", "3-prong candidates;prong 2 impact parameter error (cm);entries", {HistType::kTH2F, {{100, -1., 1.}, {binsPt, "#it{p}_{T} (GeV/#it{c})"}}});
221222
registry.add("Data/hNsigmaTPCDeVsP", "deuteron;#it{p} (GeV/#it{c}); n#sigma^{TPC}_{d}", {HistType::kTH2F, {{200, -10.f, 10.f}, {200, -6.f, 6.f}}});
223+
registry.add("Data/hNsigmaTPCPrVsP", "proton;#it{p} (GeV/#it{c}); n#sigma^{TPC}_{p}", {HistType::kTH2F, {{200, -10.f, 10.f}, {200, -6.f, 6.f}}});
222224
registry.add("Data/hNsigmaTOFDeVsP", "deuteron;#it{p} (GeV/#it{c}); n#sigma^{TOF}_{d}", {HistType::kTH2F, {{200, -10.f, 10.f}, {200, -6.f, 6.f}}});
223225
registry.add("Data/hNsigmaITSDeVsP", "deuteron;#it{p} (GeV/#it{c}); n#sigma^{ITS}_{d}", {HistType::kTH2F, {{200, -10.f, 10.f}, {200, -6.f, 6.f}}});
224226
registry.add("Data/hTPCSignalDeVsP", "deuteron;#it{p} (GeV/#it{c}); TPC signals", {HistType::kTH2F, {{200, -10.f, 10.f}, {2000, 0, 2000}}});
@@ -290,11 +292,14 @@ struct HfTaskCd {
290292
const auto cpa = candidate.cpa();
291293
const auto cpaXY = candidate.cpaXY();
292294
float invMassCd = 0.f;
295+
float invMassLc = 0.f;
293296
if (candidate.isSelCdToDeKPi() >= selectionFlagCd) {
294297
invMassCd = HfHelper::invMassCdToDeKPi(candidate);
298+
invMassLc = HfHelper::invMassLcToPKPi(candidate);
295299
}
296300
if (candidate.isSelCdToPiKDe() >= selectionFlagCd) {
297301
invMassCd = HfHelper::invMassCdToPiKDe(candidate);
302+
invMassLc = HfHelper::invMassLcToPiKP(candidate);
298303
}
299304

300305
if (candidate.isSelCdToDeKPi() >= selectionFlagCd) {
@@ -356,13 +361,11 @@ struct HfTaskCd {
356361
if (fillTree) {
357362
int candFlag = -999;
358363

359-
float nSigmaTpcDe = 0.f, nSigmaTpcKa = 0.f, nSigmaTpcPi = 0.f;
364+
float nSigmaTpcDe = 0.f, nSigmaTpcKa = 0.f, nSigmaTpcPi = 0.f, nSigmaTpcPr = 0.f;
360365
float nSigmaItsDe = 0.f;
361366
float nSigmaTofDe = 0.f, nSigmaTofKa = 0.f, nSigmaTofPi = 0.f;
362367

363-
int itsNClusterDe = 0;
364368
int itsNClusterSizeDe = 0;
365-
int tpcNClusterDe = 0;
366369

367370
float tpcSignalsDe = 0.f;
368371
float tpcSignalsPi = 0.f;
@@ -393,13 +396,12 @@ struct HfTaskCd {
393396
pSignedDe = prong0.tpcInnerParam() * prong0.sign();
394397
pSignedPi = prong2.tpcInnerParam() * prong2.sign();
395398
nSigmaTpcDe = candidate.nSigTpcDe0();
399+
nSigmaTpcPr = candidate.nSigTpcPr0();
396400
nSigmaTofDe = candidate.nSigTofDe0();
397401
nSigmaTpcPi = candidate.nSigTpcPi2();
398402
nSigmaTofPi = candidate.nSigTofPi2();
399403
nSigmaItsDe = prong0Its.itsNSigmaDe();
400-
itsNClusterDe = prong0.itsNCls();
401404
itsNClusterSizeDe = prong0.itsClusterSizes();
402-
tpcNClusterDe = prong0.tpcNClsCrossedRows();
403405
tpcSignalsDe = prong0.tpcSignal();
404406
tpcSignalsPi = prong2.tpcSignal();
405407
itsSignalsDe = itsSignal(prong0);
@@ -408,20 +410,20 @@ struct HfTaskCd {
408410
pSignedDe = prong2.tpcInnerParam() * prong2.sign();
409411
pSignedPi = prong0.tpcInnerParam() * prong0.sign();
410412
nSigmaTpcDe = candidate.nSigTpcDe2();
413+
nSigmaTpcPr = candidate.nSigTpcPr2();
411414
nSigmaTofDe = candidate.nSigTofDe2();
412415
nSigmaTpcPi = candidate.nSigTpcPi0();
413416
nSigmaTofPi = candidate.nSigTofPi0();
414417
nSigmaItsDe = prong2Its.itsNSigmaDe();
415-
itsNClusterDe = prong2.itsNCls();
416418
itsNClusterSizeDe = prong2.itsClusterSizes();
417-
tpcNClusterDe = prong2.tpcNClsCrossedRows();
418419
tpcSignalsDe = prong2.tpcSignal();
419420
tpcSignalsPi = prong0.tpcSignal();
420421
itsSignalsDe = itsSignal(prong2);
421422
}
422423

423424
// PID QA
424425
registry.fill(HIST("Data/hNsigmaTPCDeVsP"), pSignedDe, nSigmaTpcDe);
426+
registry.fill(HIST("Data/hNsigmaTPCPrVsP"), pSignedDe, nSigmaTpcPr);
425427
registry.fill(HIST("Data/hNsigmaTOFDeVsP"), pSignedDe, nSigmaTofDe);
426428
registry.fill(HIST("Data/hNsigmaITSDeVsP"), pSignedDe, nSigmaItsDe);
427429
registry.fill(HIST("Data/hTPCSignalDeVsP"), pSignedDe, tpcSignalsDe);
@@ -435,6 +437,7 @@ struct HfTaskCd {
435437

436438
rowCandCd(
437439
invMassCd,
440+
invMassLc,
438441
pt,
439442
eta,
440443
phi,
@@ -448,15 +451,13 @@ struct HfTaskCd {
448451
cpa,
449452
chi2PCA,
450453
nSigmaTpcDe,
454+
nSigmaTpcPr,
451455
nSigmaItsDe,
452456
nSigmaTofDe,
453-
itsNClusterDe,
454457
itsNClusterSizeDe,
455-
tpcNClusterDe,
456458
tpcSignalsDe,
457459
tpcSignalsPi,
458460
tpcSignalsKa,
459-
itsSignalsDe,
460461
candFlag,
461462
cent);
462463
}

‎PWGHF/DataModel/CandidateReconstructionTables.h‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -837,7 +837,7 @@ DECLARE_SOA_TABLE(HfCand3Prong2PidAl, "AOD", "HFCAND3P2PIDAL", //!
837837
using HfCand3Prong = HfCand3ProngExt;
838838
using HfCand3ProngWPidPiKaPr = soa::Join<HfCand3Prong, HfCand3Prong0PidPi, HfCand3Prong0PidPr, HfCand3Prong0PidKa, HfCand3Prong1PidPi, HfCand3Prong1PidPr, HfCand3Prong1PidKa, HfCand3Prong2PidPi, HfCand3Prong2PidPr, HfCand3Prong2PidKa>;
839839
using HfCand3ProngWPidPiKa = soa::Join<HfCand3Prong, HfCand3Prong0PidPi, HfCand3Prong0PidKa, HfCand3Prong1PidPi, HfCand3Prong1PidKa, HfCand3Prong2PidPi, HfCand3Prong2PidKa>;
840-
using HfCand3ProngWPidPiKaDe = soa::Join<HfCand3Prong, HfCand3Prong0PidPi, HfCand3Prong0PidDe, HfCand3Prong0PidKa, HfCand3Prong1PidPi, HfCand3Prong1PidDe, HfCand3Prong1PidKa, HfCand3Prong2PidPi, HfCand3Prong2PidDe, HfCand3Prong2PidKa>;
840+
using HfCand3ProngWPidPiKaDe = soa::Join<HfCand3Prong, HfCand3Prong0PidPi, HfCand3Prong0PidDe, HfCand3Prong0PidPr, HfCand3Prong0PidKa, HfCand3Prong1PidPi, HfCand3Prong1PidDe, HfCand3Prong1PidKa, HfCand3Prong2PidPi, HfCand3Prong2PidDe, HfCand3Prong2PidPr, HfCand3Prong2PidKa>;
841841
using HfCand3ProngWPidPiKaTr = soa::Join<HfCand3Prong, HfCand3Prong0PidPi, HfCand3Prong0PidTr, HfCand3Prong0PidKa, HfCand3Prong1PidPi, HfCand3Prong1PidTr, HfCand3Prong1PidKa, HfCand3Prong2PidPi, HfCand3Prong2PidTr, HfCand3Prong2PidKa>;
842842
using HfCand3ProngWPidPiKaHe = soa::Join<HfCand3Prong, HfCand3Prong0PidPi, HfCand3Prong0PidHe, HfCand3Prong0PidKa, HfCand3Prong1PidPi, HfCand3Prong1PidHe, HfCand3Prong1PidKa, HfCand3Prong2PidPi, HfCand3Prong2PidHe, HfCand3Prong2PidKa>;
843843
using HfCand3ProngWPidPiKaAl = soa::Join<HfCand3Prong, HfCand3Prong0PidPi, HfCand3Prong0PidAl, HfCand3Prong0PidKa, HfCand3Prong1PidPi, HfCand3Prong1PidAl, HfCand3Prong1PidKa, HfCand3Prong2PidPi, HfCand3Prong2PidAl, HfCand3Prong2PidKa>;

0 commit comments

Comments
 (0)