Skip to content

Commit 9775a2d

Browse files
authored
[PWGEM/Dilepton] update treeCreatorMuonML.cxx to reduce data size (#15959)
1 parent 607b707 commit 9775a2d

2 files changed

Lines changed: 157 additions & 266 deletions

File tree

PWGEM/Dilepton/DataModel/lmeeMLTables.h

Lines changed: 14 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -153,39 +153,40 @@ DECLARE_SOA_COLUMN(YMFTatMP, yMFTatMP, float); //! y of MFT track in MFT-M
153153
DECLARE_SOA_COLUMN(XErrMFTatMP, xErrMFTatMP, float); //! x error of MFT track in MFT-MCH-MID track at matching plane
154154
DECLARE_SOA_COLUMN(YErrMFTatMP, yErrMFTatMP, float); //! y error of MFT track in MFT-MCH-MID track at matching plane
155155

156-
DECLARE_SOA_COLUMN(Chi2MFT, chi2MFT, float); //! chi2/ndf of MFT track
157-
DECLARE_SOA_COLUMN(Chi2MCHMID, chi2MCHMID, float); //! chi2/ndf of MCH-MID track
158-
DECLARE_SOA_COLUMN(Chi2MFTMCHMID, chi2MFTMCHMID, float); //! chi2/ndf of MFT-MCH-MID track
159-
DECLARE_SOA_COLUMN(NClustersMFT, nClustersMFT, uint8_t); //!
160-
DECLARE_SOA_COLUMN(IsPrimary, isPrimary, bool); //!
161-
DECLARE_SOA_COLUMN(IsCorrectMatch, isCorrectMatch, bool); //!
156+
DECLARE_SOA_COLUMN(Chi2MFT, chi2MFT, float); //! chi2/ndf of MFT track
157+
DECLARE_SOA_COLUMN(Chi2MCHMID, chi2MCHMID, float); //! chi2/ndf of MCH-MID track
158+
DECLARE_SOA_COLUMN(Chi2MFTMCHMID, chi2MFTMCHMID, float); //! chi2/ndf of MFT-MCH-MID track
159+
DECLARE_SOA_COLUMN(NClustersMFT, nClustersMFT, uint8_t); //!
160+
DECLARE_SOA_COLUMN(IsPrimaryMFT, isPrimaryMFT, bool); //!
161+
DECLARE_SOA_COLUMN(IsPrimaryMCHMID, isPrimaryMCHMID, bool); //!
162+
DECLARE_SOA_COLUMN(IsCorrectMatch, isCorrectMatch, bool); //!
163+
DECLARE_SOA_COLUMN(PdgCodeMFT, pdgCodeMFT, int); //!
164+
DECLARE_SOA_COLUMN(PdgCodeMCHMID, pdgCodeMCHMID, int); //!
162165

163166
DECLARE_SOA_COLUMN(MultMFT, multMFT, uint16_t); //! number of MFTsa tracks per collision
164167
} // namespace emmlfwdtrack
165168

166169
DECLARE_SOA_TABLE(EMFwdTracksForML, "AOD", "EMFWDTRKML", //!
167170
o2::soa::Index<>, collision::PosZ, /*collision::NumContrib,*/ mult::MultFT0C, /*evsel::NumTracksInTimeRange,*/ evsel::SumAmpFT0CInTimeRange, emmltrack::HadronicRate, emmlfwdtrack::MultMFT,
168-
// fwdtrack::TrackType,
169171

170172
emmlfwdtrack::Signed1PtMFTatMP, emmlfwdtrack::TglMFTatMP, emmlfwdtrack::PhiMFTatMP,
171173
emmlfwdtrack::XMFTatMP, emmlfwdtrack::YMFTatMP,
172174

173175
emmlfwdtrack::Signed1PtMCHMIDatMP, emmlfwdtrack::TglMCHMIDatMP, emmlfwdtrack::PhiMCHMIDatMP,
174176
emmlfwdtrack::XMCHMIDatMP, emmlfwdtrack::YMCHMIDatMP,
175177

176-
// fwdtrack::NClusters, fwdtrack::PDca, fwdtrack::RAtAbsorberEnd,
177-
// fwdtrack::Chi2MatchMCHMID,
178178
fwdtrack::Chi2MatchMCHMFT,
179-
// fwdtrack::MFTClusterSizesAndTrackFlags, emmlfwdtrack::Chi2MFTMCHMID, emmlfwdtrack::Chi2MCHMID, emmlfwdtrack::Chi2MFT, emmlfwdtrack::NClustersMFT,
180-
mcparticle::PdgCode, emmlfwdtrack::IsPrimary, emmlfwdtrack::IsCorrectMatch, mcparticle::Pt, mcparticle::Eta, mcparticle::Phi);
179+
emmlfwdtrack::PdgCodeMFT, emmlfwdtrack::IsPrimaryMFT,
180+
emmlfwdtrack::PdgCodeMCHMID, emmlfwdtrack::IsPrimaryMCHMID,
181+
emmlfwdtrack::IsCorrectMatch);
181182

182183
// iterators
183184
using EMFwdTrackForML = EMFwdTracksForML::iterator;
184185

185186
DECLARE_SOA_TABLE(EMFwdTrackErrsForML, "AOD", "EMFWDTRKERRML", //! Joinable with EMFwdTracksForML
186-
emmlfwdtrack::Signed1PtErrMFTatMP, emmlfwdtrack::TglErrMFTatMP, emmlfwdtrack::PhiErrMFTatMP,
187+
/*emmlfwdtrack::Signed1PtErrMFTatMP,*/ emmlfwdtrack::TglErrMFTatMP, emmlfwdtrack::PhiErrMFTatMP,
187188
emmlfwdtrack::XErrMFTatMP, emmlfwdtrack::YErrMFTatMP,
188-
emmlfwdtrack::Signed1PtErrMCHMIDatMP, emmlfwdtrack::TglErrMCHMIDatMP, emmlfwdtrack::PhiErrMCHMIDatMP,
189+
/*emmlfwdtrack::Signed1PtErrMCHMIDatMP,*/ emmlfwdtrack::TglErrMCHMIDatMP, emmlfwdtrack::PhiErrMCHMIDatMP,
189190
emmlfwdtrack::XErrMCHMIDatMP, emmlfwdtrack::YErrMCHMIDatMP);
190191

191192
// iterators

0 commit comments

Comments
 (0)