[PWGJE] Enhance GammaJetAnalysisTree with additional MC information (#16981)

Author: fjonasALICE

PWGJE/DataModel/GammaJetAnalysisTree.h

@@ -9,9 +9,8 @@
 // granted to it by virtue of its status as an Intergovernmental Organization
 // or submit itself to any jurisdiction.
 
-///
+/// \file GammaJetAnalysisTree.h
 /// \brief Table definitions for gamma-jet analyses
-///
 /// \author Florian Jonas <florian.jonas@cern.ch>
 
 #ifndef PWGJE_DATAMODEL_GAMMAJETANALYSISTREE_H_
@@ -106,8 +105,16 @@ namespace gjgammamcinfo
 {
 DECLARE_SOA_COLUMN(Origin, origin, uint16_t);
 DECLARE_SOA_COLUMN(LeadingEnergyFraction, leadingEnergyFraction, float); // fraction of energy from the leading MC particle
+// further information about the particle that produced this cluster, one could also linnk to mcgen particle table
+// but for now this is easier as the MCgen table is filtered and I do not want to deal with ensuring the filered table exists and
+// does not filter out a mother
+DECLARE_SOA_COLUMN(Eta, eta, float);
+DECLARE_SOA_COLUMN(Phi, phi, float);
+DECLARE_SOA_COLUMN(Energy, energy, float);
+DECLARE_SOA_COLUMN(Pt, pt, float);
+DECLARE_SOA_COLUMN(MCIso, mcIso, float);
 } // namespace gjgammamcinfo
-DECLARE_SOA_TABLE(GjGammaMCInfos, "AOD", "GJGAMMAMCINFO", gjgamma::GjEventId, gjgammamcinfo::Origin, gjgammamcinfo::LeadingEnergyFraction)
+DECLARE_SOA_TABLE(GjGammaMCInfos, "AOD", "GJGAMMAMCINFO", gjgamma::GjEventId, gjgammamcinfo::Origin, gjgammamcinfo::LeadingEnergyFraction, gjgammamcinfo::Eta, gjgammamcinfo::Phi, gjgammamcinfo::Energy, gjgammamcinfo::Pt, gjgammamcinfo::MCIso)
 
 // Generator level particle information from the MC collision that was matched to the reconstructed collision
 namespace gjmcparticle
@@ -178,6 +185,16 @@ DECLARE_SOA_COLUMN(PerpConeRho, perpconerho, float);
 } // namespace gjmcjet
 DECLARE_SOA_TABLE(GjMCJets, "AOD", "GJMCJET", gjgamma::GjEventId, gjmcjet::Pt, gjmcjet::Eta, gjmcjet::Phi, gjmcjet::Radius, gjmcjet::Energy, gjmcjet::Mass, gjmcjet::Area, gjmcjet::PerpConeRho)
 
+// MC gen level substructure
+namespace gjmcjetsubstructure
+{
+DECLARE_SOA_COLUMN(EnergyMother, energyMother, std::vector<float>); //! energy of mother subjet at each splitting
+DECLARE_SOA_COLUMN(PtLeading, ptLeading, std::vector<float>);       //! pt of leading subjet at each splitting
+DECLARE_SOA_COLUMN(PtSubLeading, ptSubLeading, std::vector<float>); //! pt of subleading subjet at each splitting
+DECLARE_SOA_COLUMN(Theta, theta, std::vector<float>);               //! opening angle theta at each splitting
+} // namespace gjmcjetsubstructure
+DECLARE_SOA_TABLE(GjMCJetSubstructures, "AOD", "GJMCJETSUBSTR", gjgamma::GjEventId, gjmcjetsubstructure::EnergyMother, gjmcjetsubstructure::PtLeading, gjmcjetsubstructure::PtSubLeading, gjmcjetsubstructure::Theta)
+
 } // namespace o2::aod
 
 #endif // PWGJE_DATAMODEL_GAMMAJETANALYSISTREE_H_