Skip to content

Commit a211a2f

Browse files
author
Ida Torkjellsdatter Storehaug
committed
tableReader_withAssoc:AnalysisDileptonTrack: columns added to flat B meson table
1 parent 7607086 commit a211a2f

File tree

1 file changed

+64
-6
lines changed

1 file changed

+64
-6
lines changed

‎PWGDQ/Tasks/tableReader_withAssoc.cxx‎

Lines changed: 64 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -85,13 +85,50 @@ DECLARE_SOA_COLUMN(massBcandidate, MBcandidate, float);
8585
DECLARE_SOA_COLUMN(MassDileptonCandidate, massDileptonCandidate, float);
8686
DECLARE_SOA_COLUMN(deltamassBcandidate, deltaMBcandidate, float);
8787
DECLARE_SOA_COLUMN(pTBcandidate, PtBcandidate, float);
88+
DECLARE_SOA_COLUMN(EtaBcandidate, etaBcandidate, float);
8889
DECLARE_SOA_COLUMN(LxyBcandidate, lxyBcandidate, float);
8990
DECLARE_SOA_COLUMN(LxyzBcandidate, lxyzBcandidate, float);
9091
DECLARE_SOA_COLUMN(LzBcandidate, lzBcandidate, float);
9192
DECLARE_SOA_COLUMN(TauxyBcandidate, tauxyBcandidate, float);
9293
DECLARE_SOA_COLUMN(TauzBcandidate, tauzBcandidate, float);
9394
DECLARE_SOA_COLUMN(CosPBcandidate, cosPBcandidate, float);
9495
DECLARE_SOA_COLUMN(Chi2Bcandidate, chi2Bcandidate, float);
96+
DECLARE_SOA_COLUMN(Ptassoc, ptassoc, float);
97+
DECLARE_SOA_COLUMN(Etaassoc, etaassoc, float);
98+
DECLARE_SOA_COLUMN(Ptpair, ptpair, float);
99+
DECLARE_SOA_COLUMN(Etapair, etapair, float);
100+
DECLARE_SOA_COLUMN(Ptleg1, ptleg1, float);
101+
DECLARE_SOA_COLUMN(Etaleg1, etaleg1, float);
102+
DECLARE_SOA_COLUMN(Ptleg2, ptleg2, float);
103+
DECLARE_SOA_COLUMN(Etaleg2, etaleg2, float);
104+
DECLARE_SOA_COLUMN(TPCnsigmaKaassoc, tpcnsigmaKaassoc, float);
105+
DECLARE_SOA_COLUMN(TPCnsigmaPiassoc, tpcnsigmaPiassoc, float);
106+
DECLARE_SOA_COLUMN(TPCnsigmaPrassoc, tpcnsigmaPrassoc, float);
107+
DECLARE_SOA_COLUMN(TOFnsigmaKaassoc, tofnsigmaKaassoc, float);
108+
DECLARE_SOA_COLUMN(TPCnsigmaElleg1, tpcnsigmaElleg1, float);
109+
DECLARE_SOA_COLUMN(TPCnsigmaPileg1, tpcnsigmaPileg1, float);
110+
DECLARE_SOA_COLUMN(TPCnsigmaPrleg1, tpcnsigmaPrleg1, float);
111+
DECLARE_SOA_COLUMN(TPCnsigmaElleg2, tpcnsigmaElleg2, float);
112+
DECLARE_SOA_COLUMN(TPCnsigmaPileg2, tpcnsigmaPileg2, float);
113+
DECLARE_SOA_COLUMN(TPCnsigmaPrleg2, tpcnsigmaPrleg2, float);
114+
DECLARE_SOA_COLUMN(DCAXYassoc, dcaXYassoc, float);
115+
DECLARE_SOA_COLUMN(DCAZassoc, dcaZassoc, float);
116+
DECLARE_SOA_COLUMN(DCAXYleg1, dcaXYleg1, float);
117+
DECLARE_SOA_COLUMN(DCAZleg1, dcaZleg1, float);
118+
DECLARE_SOA_COLUMN(DCAXYleg2, dcaXYleg2, float);
119+
DECLARE_SOA_COLUMN(DCAZleg2, dcaZleg2, float);
120+
DECLARE_SOA_COLUMN(ITSClusterMapassoc, itsClusterMapassoc, uint8_t);
121+
DECLARE_SOA_COLUMN(ITSClusterMapleg1, itsClusterMapleg1, uint8_t);
122+
DECLARE_SOA_COLUMN(ITSClusterMapleg2, itsClusterMapleg2, uint8_t);
123+
DECLARE_SOA_COLUMN(ITSChi2assoc, itsChi2assoc, float);
124+
DECLARE_SOA_COLUMN(ITSChi2leg1, itsChi2leg1, float);
125+
DECLARE_SOA_COLUMN(ITSChi2leg2, itsChi2leg2, float);
126+
DECLARE_SOA_COLUMN(TPCNclsassoc, tpcNclsassoc, float);
127+
DECLARE_SOA_COLUMN(TPCNclsleg1, tpcNclsleg1, float);
128+
DECLARE_SOA_COLUMN(TPCNclsleg2, tpcNclsleg2, float);
129+
DECLARE_SOA_COLUMN(TPCChi2assoc, tpcChi2assoc, float);
130+
DECLARE_SOA_COLUMN(TPCChi2leg1, tpcChi2leg1, float);
131+
DECLARE_SOA_COLUMN(TPCChi2leg2, tpcChi2leg2, float);
95132
DECLARE_SOA_BITMAP_COLUMN(IsJpsiFromBSelected, isJpsiFromBSelected, 32);
96133
// Candidate columns for prompt-non-prompt JPsi separation
97134
DECLARE_SOA_COLUMN(Massee, massJPsi2ee, float);
@@ -111,9 +148,19 @@ DECLARE_SOA_TABLE(MuonTrackCuts, "AOD", "DQANAMUONCUTSA", dqanalysisflags::IsMuo
111148
DECLARE_SOA_TABLE(MuonAmbiguities, "AOD", "DQMUONAMBA", dqanalysisflags::MuonAmbiguityInBunch, dqanalysisflags::MuonAmbiguityOutOfBunch); //! joinable to ReducedMuonTracks
112149
DECLARE_SOA_TABLE(Prefilter, "AOD", "DQPREFILTERA", dqanalysisflags::IsBarrelSelectedPrefilter); //! joinable to ReducedTracksAssoc
113150
DECLARE_SOA_TABLE(BmesonCandidates, "AOD", "DQBMESONSA",
114-
dqanalysisflags::massBcandidate, dqanalysisflags::MassDileptonCandidate, dqanalysisflags::deltamassBcandidate, dqanalysisflags::pTBcandidate,
151+
dqanalysisflags::massBcandidate, dqanalysisflags::MassDileptonCandidate, dqanalysisflags::deltamassBcandidate, dqanalysisflags::pTBcandidate, dqanalysisflags::EtaBcandidate,
115152
dqanalysisflags::LxyBcandidate, dqanalysisflags::LxyzBcandidate, dqanalysisflags::LzBcandidate,
116153
dqanalysisflags::TauxyBcandidate, dqanalysisflags::TauzBcandidate, dqanalysisflags::CosPBcandidate, dqanalysisflags::Chi2Bcandidate,
154+
dqanalysisflags::Ptassoc, dqanalysisflags::Etaassoc, dqanalysisflags::Ptpair, dqanalysisflags::Etapair,
155+
dqanalysisflags::Ptleg1, dqanalysisflags::Etaleg1,dqanalysisflags::Ptleg2, dqanalysisflags::Etaleg2,
156+
dqanalysisflags::TPCnsigmaKaassoc, dqanalysisflags::TPCnsigmaPiassoc, dqanalysisflags::TPCnsigmaPrassoc, dqanalysisflags::TOFnsigmaKaassoc,
157+
dqanalysisflags::TPCnsigmaElleg1, dqanalysisflags::TPCnsigmaPileg1, dqanalysisflags::TPCnsigmaPrleg1,
158+
dqanalysisflags::TPCnsigmaElleg2, dqanalysisflags::TPCnsigmaPileg2, dqanalysisflags::TPCnsigmaPrleg2,
159+
dqanalysisflags::DCAXYassoc, dqanalysisflags::DCAZassoc,dqanalysisflags::DCAXYleg1, dqanalysisflags::DCAZleg1, dqanalysisflags::DCAXYleg2, dqanalysisflags::DCAZleg2,
160+
dqanalysisflags::ITSClusterMapassoc, dqanalysisflags::ITSClusterMapleg1, dqanalysisflags::ITSClusterMapleg2,
161+
dqanalysisflags::ITSChi2assoc, dqanalysisflags::ITSChi2leg1, dqanalysisflags::ITSChi2leg2,
162+
dqanalysisflags::TPCNclsassoc, dqanalysisflags::TPCNclsleg1, dqanalysisflags::TPCNclsleg2,
163+
dqanalysisflags::TPCChi2assoc, dqanalysisflags::TPCChi2leg1, dqanalysisflags::TPCChi2leg2,
117164
dqanalysisflags::IsJpsiFromBSelected, dqanalysisflags::IsBarrelSelected);
118165
DECLARE_SOA_TABLE(JPsieeCandidates, "AOD", "DQPSEUDOPROPER", dqanalysisflags::Massee, dqanalysisflags::Ptee, dqanalysisflags::Lxyee, dqanalysisflags::LxyeePoleMass, dqanalysisflags::Lzee, dqanalysisflags::AmbiguousInBunchPairs, dqanalysisflags::AmbiguousOutOfBunchPairs);
119166
} // namespace o2::aod
@@ -3133,6 +3180,22 @@ struct AnalysisDileptonTrack {
31333180
// compute needed quantities
31343181
VarManager::FillDileptonHadron(dilepton, track, fValuesHadron);
31353182
VarManager::FillDileptonTrackVertexing<TCandidateType, TEventFillMap, TTrackFillMap>(event, lepton1, lepton2, track, fValuesHadron);
3183+
// table to be written out for ML analysis
3184+
BmesonsTable(fValuesHadron[VarManager::kPairMass], dilepton.mass(), fValuesHadron[VarManager::kDeltaMass], fValuesHadron[VarManager::kPairPt], fValuesHadron[VarManager::kPairEta],
3185+
fValuesHadron[VarManager::kVertexingLxy], fValuesHadron[VarManager::kVertexingLxyz], fValuesHadron[VarManager::kVertexingLz],
3186+
fValuesHadron[VarManager::kVertexingTauxy], fValuesHadron[VarManager::kVertexingTauz], fValuesHadron[VarManager::kCosPointingAngle],
3187+
fValuesHadron[VarManager::kVertexingChi2PCA],
3188+
track.pt(), track.eta(), dilepton.pt(), dilepton.eta(), lepton1.pt(), lepton1.eta(), lepton2.pt(), lepton2.eta(),
3189+
track.tpcNSigmaKa(), track.tpcNSigmaPi(), track.tpcNSigmaPr(), track.tofNSigmaKa(),
3190+
lepton1.tpcNSigmaEl(), lepton1.tpcNSigmaPi(), lepton1.tpcNSigmaPr(),
3191+
lepton2.tpcNSigmaEl(), lepton2.tpcNSigmaPi(), lepton2.tpcNSigmaPr(),
3192+
track.dcaXY(), track.dcaZ(), lepton1.dcaXY(), lepton1.dcaZ(), lepton2.dcaXY(), lepton2.dcaZ(),
3193+
track.itsClusterMap(), lepton1.itsClusterMap(), lepton2.itsClusterMap(),
3194+
track.itsChi2NCl(), lepton1.itsChi2NCl(), lepton2.itsChi2NCl(),
3195+
track.tpcNClsFound(), lepton1.tpcNClsFound(), lepton2.tpcNClsFound(),
3196+
track.tpcChi2NCl(), lepton1.tpcChi2NCl(), lepton2.tpcChi2NCl(),
3197+
dilepton.filterMap_raw(), trackSelection);
3198+
31363199
}
31373200
if constexpr (TCandidateType == VarManager::kDstarToD0KPiPi) {
31383201
trackSelection = (assoc.isBarrelSelected_raw() & fTrackCutBitMap);
@@ -3199,11 +3262,6 @@ struct AnalysisDileptonTrack {
31993262
}
32003263
} // end loop over track cuts
32013264
} // end loop over dilepton cuts
3202-
// table to be written out for ML analysis
3203-
BmesonsTable(fValuesHadron[VarManager::kPairMass], dilepton.mass(), fValuesHadron[VarManager::kDeltaMass], fValuesHadron[VarManager::kPairPt],
3204-
fValuesHadron[VarManager::kVertexingLxy], fValuesHadron[VarManager::kVertexingLxyz], fValuesHadron[VarManager::kVertexingLz],
3205-
fValuesHadron[VarManager::kVertexingTauxy], fValuesHadron[VarManager::kVertexingTauz], fValuesHadron[VarManager::kCosPointingAngle],
3206-
fValuesHadron[VarManager::kVertexingChi2PCA], dilepton.filterMap_raw(), trackSelection);
32073265
}
32083266
}
32093267
}

0 commit comments

Comments
 (0)