@@ -55,6 +55,10 @@ DECLARE_SOA_COLUMN(TPCNClsPID, tpcNClsPID, uint8_t); //!
5555DECLARE_SOA_COLUMN (IsForValidation, isForValidation, bool ); // !
5656DECLARE_SOA_COLUMN (Sign, sign, short ); // !
5757DECLARE_SOA_COLUMN (P, p, float ); // !
58+ DECLARE_SOA_COLUMN (PtGen, ptGen, float ); // !
59+ DECLARE_SOA_COLUMN (EtaGen, etaGen, float ); // !
60+ DECLARE_SOA_COLUMN (PhiGen, phiGen, float ); // !
61+
5862// DECLARE_SOA_DYNAMIC_COLUMN(P, p, [](float pt, float eta) -> float { return pt * std::cosh(eta); });
5963DECLARE_SOA_DYNAMIC_COLUMN (MeanClusterSizeITS, meanClusterSizeITS, [](uint32_t itsClusterSizes) -> float {
6064 int total_cluster_size = 0 , nl = 0 ;
@@ -141,7 +145,7 @@ DECLARE_SOA_TABLE_VERSIONED(EMFwdTracksForML_000, "AOD", "EMFWDTRKML", 0, //!
141145 fwdtrack::Chi2, fwdtrack::Chi2MatchMCHMID, fwdtrack::Chi2MatchMCHMFT,
142146 // fwdtrack::MCHBitMap, fwdtrack::MIDBitMap, fwdtrack::MIDBoards,
143147 fwdtrack::MFTClusterSizesAndTrackFlags, emmlfwdtrack::Chi2MFT, emmlfwdtrack::NClustersMFT, mcparticle::PdgCode, emmlfwdtrack::IsPrimary, emmlfwdtrack::IsCorrectMatchMFTMCH,
144- mcparticle::Pt, mcparticle::Eta, mcparticle::Phi );
148+ emmltrack::PtGen, emmltrack::EtaGen, emmltrack::PhiGen );
145149
146150using EMFwdTracksForML = EMFwdTracksForML_000;
147151// iterators
0 commit comments