Added ability to store original parton type - by default set to 0
[u/mrichter/AliRoot.git] / EMCAL / AliEMCALParton.h
1 #ifndef ALIEMCALPARTON_H
2 #define ALIEMCALPARTON_H
3
4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5  * See cxx source for full Copyright notice     */
6
7 #include <TObject.h>
8 //*-- Author: Renan Cabrera (Creighton U.)
9
10
11 class AliEMCALParton : public TObject {
12 public:
13   AliEMCALParton();
14   AliEMCALParton(Float_t energy, Float_t phi, Float_t eta);
15   virtual ~AliEMCALParton();
16   void SetEnergy(Float_t val) {fEnergy = val;}
17   void SetPhi(Float_t val)    {fPhi    = val;}
18   void SetEta(Float_t val)    {fEta    = val;}
19   void SetTrackList(Int_t num ,Float_t* ar1,Float_t* ar2,Float_t* ar3,Int_t* ar4);
20   void GetTrackList(Float_t* ar1,Float_t* ar2,Float_t* ar3,Int_t* ar4)const;
21   
22   Int_t GetNTracks() const {return fNTracks;}
23   Float_t Energy()  const {return fEnergy;}
24   Float_t Phi()    const {return fPhi;}
25   Float_t Eta()    const {return fEta;}
26   void SetPartonCode(Int_t code){fPartonCode = code;} 
27   Int_t GetPartonCode()const{return fPartonCode;}
28 protected:
29   Float_t   fEnergy;   // Jet Energy
30   Float_t   fEta;      // Jet Phi
31   Float_t   fPhi;      // Jet Eta
32   Int_t     fNTracks;      // Number of tracks 
33   Float_t  fTrackEnergy[1000];  // Jet Tracks Energy
34   Float_t  fTrackEta[1000];     // Jet Tracks Eta
35   Float_t  fTrackPhi[1000];     // Jet Tracks Phi
36   Int_t    fTrackPDG[1000];     // Jet Tracks PDG code
37   Int_t    fPartonCode;         // Store the type of parton
38   ClassDef(AliEMCALParton,7) // Jet for EMCAL
39     
40 } ;
41
42 #endif // ALIEMCALParton_H