Skip to content

Commit 79ead1e

Browse files
torkjellsdatterIda Torkjellsdatter Storehaugalibuild
authored
[PWGDQ] columns added to flat B meson table (tableReader_withAssoc:AnalysisDileptonTrack) (#11248)
Co-authored-by: Ida Torkjellsdatter Storehaug <idats@eduroam-193-157-251-21.wlan.uio.no> Co-authored-by: ALICE Action Bot <alibuild@cern.ch>
1 parent 044a000 commit 79ead1e

File tree

1 file changed

+63
-6
lines changed

1 file changed

+63
-6
lines changed

‎PWGDQ/Tasks/tableReader_withAssoc.cxx‎

Lines changed: 63 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -86,13 +86,50 @@ DECLARE_SOA_COLUMN(massBcandidate, MBcandidate, float);
8686
DECLARE_SOA_COLUMN(MassDileptonCandidate, massDileptonCandidate, float);
8787
DECLARE_SOA_COLUMN(deltamassBcandidate, deltaMBcandidate, float);
8888
DECLARE_SOA_COLUMN(pTBcandidate, PtBcandidate, float);
89+
DECLARE_SOA_COLUMN(EtaBcandidate, etaBcandidate, float);
8990
DECLARE_SOA_COLUMN(LxyBcandidate, lxyBcandidate, float);
9091
DECLARE_SOA_COLUMN(LxyzBcandidate, lxyzBcandidate, float);
9192
DECLARE_SOA_COLUMN(LzBcandidate, lzBcandidate, float);
9293
DECLARE_SOA_COLUMN(TauxyBcandidate, tauxyBcandidate, float);
9394
DECLARE_SOA_COLUMN(TauzBcandidate, tauzBcandidate, float);
9495
DECLARE_SOA_COLUMN(CosPBcandidate, cosPBcandidate, float);
9596
DECLARE_SOA_COLUMN(Chi2Bcandidate, chi2Bcandidate, float);
97+
DECLARE_SOA_COLUMN(Ptassoc, ptassoc, float);
98+
DECLARE_SOA_COLUMN(Etaassoc, etaassoc, float);
99+
DECLARE_SOA_COLUMN(Ptpair, ptpair, float);
100+
DECLARE_SOA_COLUMN(Etapair, etapair, float);
101+
DECLARE_SOA_COLUMN(Ptleg1, ptleg1, float);
102+
DECLARE_SOA_COLUMN(Etaleg1, etaleg1, float);
103+
DECLARE_SOA_COLUMN(Ptleg2, ptleg2, float);
104+
DECLARE_SOA_COLUMN(Etaleg2, etaleg2, float);
105+
DECLARE_SOA_COLUMN(TPCnsigmaKaassoc, tpcnsigmaKaassoc, float);
106+
DECLARE_SOA_COLUMN(TPCnsigmaPiassoc, tpcnsigmaPiassoc, float);
107+
DECLARE_SOA_COLUMN(TPCnsigmaPrassoc, tpcnsigmaPrassoc, float);
108+
DECLARE_SOA_COLUMN(TOFnsigmaKaassoc, tofnsigmaKaassoc, float);
109+
DECLARE_SOA_COLUMN(TPCnsigmaElleg1, tpcnsigmaElleg1, float);
110+
DECLARE_SOA_COLUMN(TPCnsigmaPileg1, tpcnsigmaPileg1, float);
111+
DECLARE_SOA_COLUMN(TPCnsigmaPrleg1, tpcnsigmaPrleg1, float);
112+
DECLARE_SOA_COLUMN(TPCnsigmaElleg2, tpcnsigmaElleg2, float);
113+
DECLARE_SOA_COLUMN(TPCnsigmaPileg2, tpcnsigmaPileg2, float);
114+
DECLARE_SOA_COLUMN(TPCnsigmaPrleg2, tpcnsigmaPrleg2, float);
115+
DECLARE_SOA_COLUMN(DCAXYassoc, dcaXYassoc, float);
116+
DECLARE_SOA_COLUMN(DCAZassoc, dcaZassoc, float);
117+
DECLARE_SOA_COLUMN(DCAXYleg1, dcaXYleg1, float);
118+
DECLARE_SOA_COLUMN(DCAZleg1, dcaZleg1, float);
119+
DECLARE_SOA_COLUMN(DCAXYleg2, dcaXYleg2, float);
120+
DECLARE_SOA_COLUMN(DCAZleg2, dcaZleg2, float);
121+
DECLARE_SOA_COLUMN(ITSClusterMapassoc, itsClusterMapassoc, uint8_t);
122+
DECLARE_SOA_COLUMN(ITSClusterMapleg1, itsClusterMapleg1, uint8_t);
123+
DECLARE_SOA_COLUMN(ITSClusterMapleg2, itsClusterMapleg2, uint8_t);
124+
DECLARE_SOA_COLUMN(ITSChi2assoc, itsChi2assoc, float);
125+
DECLARE_SOA_COLUMN(ITSChi2leg1, itsChi2leg1, float);
126+
DECLARE_SOA_COLUMN(ITSChi2leg2, itsChi2leg2, float);
127+
DECLARE_SOA_COLUMN(TPCNclsassoc, tpcNclsassoc, float);
128+
DECLARE_SOA_COLUMN(TPCNclsleg1, tpcNclsleg1, float);
129+
DECLARE_SOA_COLUMN(TPCNclsleg2, tpcNclsleg2, float);
130+
DECLARE_SOA_COLUMN(TPCChi2assoc, tpcChi2assoc, float);
131+
DECLARE_SOA_COLUMN(TPCChi2leg1, tpcChi2leg1, float);
132+
DECLARE_SOA_COLUMN(TPCChi2leg2, tpcChi2leg2, float);
96133
DECLARE_SOA_BITMAP_COLUMN(IsJpsiFromBSelected, isJpsiFromBSelected, 32);
97134
// Candidate columns for prompt-non-prompt JPsi separation
98135
DECLARE_SOA_COLUMN(Massee, massJPsi2ee, float);
@@ -112,9 +149,19 @@ DECLARE_SOA_TABLE(MuonTrackCuts, "AOD", "DQANAMUONCUTSA", dqanalysisflags::IsMuo
112149
DECLARE_SOA_TABLE(MuonAmbiguities, "AOD", "DQMUONAMBA", dqanalysisflags::MuonAmbiguityInBunch, dqanalysisflags::MuonAmbiguityOutOfBunch); //! joinable to ReducedMuonTracks
113150
DECLARE_SOA_TABLE(Prefilter, "AOD", "DQPREFILTERA", dqanalysisflags::IsBarrelSelectedPrefilter); //! joinable to ReducedTracksAssoc
114151
DECLARE_SOA_TABLE(BmesonCandidates, "AOD", "DQBMESONSA",
115-
dqanalysisflags::massBcandidate, dqanalysisflags::MassDileptonCandidate, dqanalysisflags::deltamassBcandidate, dqanalysisflags::pTBcandidate,
152+
dqanalysisflags::massBcandidate, dqanalysisflags::MassDileptonCandidate, dqanalysisflags::deltamassBcandidate, dqanalysisflags::pTBcandidate, dqanalysisflags::EtaBcandidate,
116153
dqanalysisflags::LxyBcandidate, dqanalysisflags::LxyzBcandidate, dqanalysisflags::LzBcandidate,
117154
dqanalysisflags::TauxyBcandidate, dqanalysisflags::TauzBcandidate, dqanalysisflags::CosPBcandidate, dqanalysisflags::Chi2Bcandidate,
155+
dqanalysisflags::Ptassoc, dqanalysisflags::Etaassoc, dqanalysisflags::Ptpair, dqanalysisflags::Etapair,
156+
dqanalysisflags::Ptleg1, dqanalysisflags::Etaleg1, dqanalysisflags::Ptleg2, dqanalysisflags::Etaleg2,
157+
dqanalysisflags::TPCnsigmaKaassoc, dqanalysisflags::TPCnsigmaPiassoc, dqanalysisflags::TPCnsigmaPrassoc, dqanalysisflags::TOFnsigmaKaassoc,
158+
dqanalysisflags::TPCnsigmaElleg1, dqanalysisflags::TPCnsigmaPileg1, dqanalysisflags::TPCnsigmaPrleg1,
159+
dqanalysisflags::TPCnsigmaElleg2, dqanalysisflags::TPCnsigmaPileg2, dqanalysisflags::TPCnsigmaPrleg2,
160+
dqanalysisflags::DCAXYassoc, dqanalysisflags::DCAZassoc, dqanalysisflags::DCAXYleg1, dqanalysisflags::DCAZleg1, dqanalysisflags::DCAXYleg2, dqanalysisflags::DCAZleg2,
161+
dqanalysisflags::ITSClusterMapassoc, dqanalysisflags::ITSClusterMapleg1, dqanalysisflags::ITSClusterMapleg2,
162+
dqanalysisflags::ITSChi2assoc, dqanalysisflags::ITSChi2leg1, dqanalysisflags::ITSChi2leg2,
163+
dqanalysisflags::TPCNclsassoc, dqanalysisflags::TPCNclsleg1, dqanalysisflags::TPCNclsleg2,
164+
dqanalysisflags::TPCChi2assoc, dqanalysisflags::TPCChi2leg1, dqanalysisflags::TPCChi2leg2,
118165
dqanalysisflags::IsJpsiFromBSelected, dqanalysisflags::IsBarrelSelected);
119166
DECLARE_SOA_TABLE(JPsieeCandidates, "AOD", "DQPSEUDOPROPER", dqanalysisflags::Massee, dqanalysisflags::Ptee, dqanalysisflags::Lxyee, dqanalysisflags::LxyeePoleMass, dqanalysisflags::Lzee, dqanalysisflags::AmbiguousInBunchPairs, dqanalysisflags::AmbiguousOutOfBunchPairs);
120167
} // namespace o2::aod
@@ -3182,6 +3229,21 @@ struct AnalysisDileptonTrack {
31823229
// compute needed quantities
31833230
VarManager::FillDileptonHadron(dilepton, track, fValuesHadron);
31843231
VarManager::FillDileptonTrackVertexing<TCandidateType, TEventFillMap, TTrackFillMap>(event, lepton1, lepton2, track, fValuesHadron);
3232+
// table to be written out for ML analysis
3233+
BmesonsTable(fValuesHadron[VarManager::kPairMass], dilepton.mass(), fValuesHadron[VarManager::kDeltaMass], fValuesHadron[VarManager::kPairPt], fValuesHadron[VarManager::kPairEta],
3234+
fValuesHadron[VarManager::kVertexingLxy], fValuesHadron[VarManager::kVertexingLxyz], fValuesHadron[VarManager::kVertexingLz],
3235+
fValuesHadron[VarManager::kVertexingTauxy], fValuesHadron[VarManager::kVertexingTauz], fValuesHadron[VarManager::kCosPointingAngle],
3236+
fValuesHadron[VarManager::kVertexingChi2PCA],
3237+
track.pt(), track.eta(), dilepton.pt(), dilepton.eta(), lepton1.pt(), lepton1.eta(), lepton2.pt(), lepton2.eta(),
3238+
track.tpcNSigmaKa(), track.tpcNSigmaPi(), track.tpcNSigmaPr(), track.tofNSigmaKa(),
3239+
lepton1.tpcNSigmaEl(), lepton1.tpcNSigmaPi(), lepton1.tpcNSigmaPr(),
3240+
lepton2.tpcNSigmaEl(), lepton2.tpcNSigmaPi(), lepton2.tpcNSigmaPr(),
3241+
track.dcaXY(), track.dcaZ(), lepton1.dcaXY(), lepton1.dcaZ(), lepton2.dcaXY(), lepton2.dcaZ(),
3242+
track.itsClusterMap(), lepton1.itsClusterMap(), lepton2.itsClusterMap(),
3243+
track.itsChi2NCl(), lepton1.itsChi2NCl(), lepton2.itsChi2NCl(),
3244+
track.tpcNClsFound(), lepton1.tpcNClsFound(), lepton2.tpcNClsFound(),
3245+
track.tpcChi2NCl(), lepton1.tpcChi2NCl(), lepton2.tpcChi2NCl(),
3246+
dilepton.filterMap_raw(), trackSelection);
31853247
}
31863248
if constexpr (TCandidateType == VarManager::kDstarToD0KPiPi) {
31873249
trackSelection = (assoc.isBarrelSelected_raw() & fTrackCutBitMap);
@@ -3248,11 +3310,6 @@ struct AnalysisDileptonTrack {
32483310
}
32493311
} // end loop over track cuts
32503312
} // end loop over dilepton cuts
3251-
// table to be written out for ML analysis
3252-
BmesonsTable(fValuesHadron[VarManager::kPairMass], dilepton.mass(), fValuesHadron[VarManager::kDeltaMass], fValuesHadron[VarManager::kPairPt],
3253-
fValuesHadron[VarManager::kVertexingLxy], fValuesHadron[VarManager::kVertexingLxyz], fValuesHadron[VarManager::kVertexingLz],
3254-
fValuesHadron[VarManager::kVertexingTauxy], fValuesHadron[VarManager::kVertexingTauz], fValuesHadron[VarManager::kCosPointingAngle],
3255-
fValuesHadron[VarManager::kVertexingChi2PCA], dilepton.filterMap_raw(), trackSelection);
32563313
}
32573314
}
32583315
}

0 commit comments

Comments
 (0)