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

diff --git a/EMCAL/AliEMCALParton.cxx b/EMCAL/AliEMCALParton.cxx
index 8f1e2cc..010614b 100644 (file)
--- a/EMCAL/AliEMCALParton.cxx
+++ b/EMCAL/AliEMCALParton.cxx
@@ -31,6 +31,7 @@ AliEMCALParton::AliEMCALParton()
 //  fTrackPhi    = 0;
 //  fTrackPDG    = 0;
   fNTracks     = 0;
+  fPartonCode = 0;
 }
 
 AliEMCALParton::AliEMCALParton(Float_t energy, Float_t phi, Float_t eta)
@@ -44,6 +45,7 @@ AliEMCALParton::AliEMCALParton(Float_t energy, Float_t phi, Float_t eta)
 //  fTrackPhi    = 0;
 //  fTrackPDG    = 0;
   fNTracks     = 0;
+  fPartonCode = 0;
 }
 
 void AliEMCALParton::SetTrackList(Int_t NTracks, Float_t* Energy,  Float_t* Eta, Float_t* Phi, Int_t* PDG)

diff --git a/EMCAL/AliEMCALParton.h b/EMCAL/AliEMCALParton.h
index 5a91629..d90da75 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
     
 } ;