Added ability to store original parton type - by default set to 0

[u/mrichter/AliRoot.git] / EMCAL / AliEMCALParton.h

diff --git a/EMCAL/AliEMCALParton.h b/EMCAL/AliEMCALParton.h
index 5a91629a5460d1c6d00856506f8731a79d39a4de..d90da75bbc0d0efbdda5a7cbcc9a875d39618c59 100644 (file)
--- a/EMCAL/AliEMCALParton.h
+++ b/EMCAL/AliEMCALParton.h
@@ -23,7 +23,8 @@ public:
   Float_t Energy()  const {return fEnergy;}
   Float_t Phi()    const {return fPhi;}
   Float_t Eta()    const {return fEta;}
-  
+  void SetPartonCode(Int_t code){fPartonCode = code;} 
+  Int_t GetPartonCode()const{return fPartonCode;}
 protected:
   Float_t   fEnergy;   // Jet Energy
   Float_t   fEta;      // Jet Phi
@@ -33,7 +34,8 @@ protected:
   Float_t  fTrackEta[1000];     // Jet Tracks Eta
   Float_t  fTrackPhi[1000];     // Jet Tracks Phi
   Int_t    fTrackPDG[1000];     // Jet Tracks PDG code
-  ClassDef(AliEMCALParton,6) // Jet for EMCAL
+  Int_t           fPartonCode;         // Store the type of parton
+  ClassDef(AliEMCALParton,7) // Jet for EMCAL
     
 } ;