8a39c96a25f9501c4be16a871ed8774e459f41e4
[u/mrichter/AliRoot.git] / PWG / CaloTrackCorrBase / AliCaloTrackMCReader.h
1 #ifndef ALICALOTRACKMCREADER_H
2 #define ALICALOTRACKMCREADER_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice     */
5 /* $Id:  $ */
6
7 //_________________________________________________________________________
8 // Class for reading data (Kinematics) in order to do prompt gamma 
9 // or other particle  correlations
10 // Separates generated particles into charged (CTS) 
11 // and neutral (PHOS or EMCAL acceptance)
12 //
13 // Now, it only works with data stored in Kinematics.root and 
14 // not in filtered Kinematics branch in AODs
15 //
16 //*-- Author: Gustavo Conesa (LNF-INFN) 
17 //////////////////////////////////////////////////////////////////////////////
18
19 //-- Author: Gustavo Conesa (INFN-LNF)
20
21 // --- ROOT system ---
22 class TArrayI   ;
23 class TParticle ;
24
25 // --- AliRoot system ---
26 #include "AliCaloTrackReader.h" 
27 class AliVCluster ;
28 class AliAODTrack ;
29 class AliAODEvent ;
30 class AliMCEvent  ;
31 class AliVEvent   ;
32
33 class AliCaloTrackMCReader : public AliCaloTrackReader {
34   
35  public: 
36   AliCaloTrackMCReader() ; // ctor
37   virtual ~AliCaloTrackMCReader() ;//virtual dtor
38
39   void   InitParameters();
40   
41   void   Print(const Option_t * opt) const; 
42   
43   void   SwitchOnPi0Decay()           { fDecayPi0 = kTRUE  ; } 
44   void   SwitchOffPi0Decay()          { fDecayPi0 = kFALSE ; } 
45   Int_t  IsPi0DecaySwitchedOn() const { return fDecayPi0   ; } 
46   
47   void      AddNeutralParticlesArray(TArrayI & array)  
48       { fNeutralParticlesArray   = new TArrayI(array) ; }
49   TArrayI * GetNeutralParticlesArray()      const {return  fNeutralParticlesArray;}
50   Bool_t    SkipNeutralParticles(Int_t pdg) const ;
51   
52   void      AddChargedParticlesArray(TArrayI & array)  
53       { fChargedParticlesArray   = new TArrayI(array) ; }
54   TArrayI * GetChargedParticlesArray()   const {return  fChargedParticlesArray;}
55   Bool_t    KeepChargedParticles(Int_t pdg) const ;
56
57   void AddStatusArray(TArrayI & array)  
58       { fStatusArray   = new TArrayI(array) ; }
59   TArrayI * GetStatusArray()             const {return  fStatusArray;}
60   
61   void   SwitchOnStatusSelection()  { fKeepAllStatus = kFALSE ; }
62   void   SwitchOffStatusSelection() { fKeepAllStatus = kTRUE  ; }
63   Bool_t KeepParticleWithStatus(Int_t status) const ;
64   
65   void   SwitchOnOnlyGeneratorParticles()  {fOnlyGeneratorParticles = kTRUE  ; }
66   void   SwitchOffOnlyGeneratorParticles() {fOnlyGeneratorParticles = kFALSE ; }
67   
68   void      GetVertex(Double_t v[3]) const ;
69   Double_t* GetVertex(const Int_t evtIndex) const {return fVertex[evtIndex];}
70   void      GetVertex(Double_t vertex[3], const Int_t evtIndex) const 
71     {vertex[0]=fVertex[evtIndex][0];  vertex[1]=fVertex[evtIndex][1];  vertex[2]=fVertex[evtIndex][2];}   
72   
73   Bool_t FillInputEvent(const Int_t iEntry, const char * currentFileName) ;
74   AliVEvent*  GetInputEvent() const {return (AliVEvent *) GetMC();}
75   void   SetInputOutputMCEvent(AliVEvent* esd, AliAODEvent* aod, AliMCEvent* mc) ;
76   
77   void   SetCaloClusterPID(const Int_t pdgCode, AliVCluster *calo)     const ;
78   void   SetTrackChargeAndPID(const Int_t pdgCode, AliAODTrack *track) const ;
79   
80   void   SwitchOnOverlapCheck()  {fCheckOverlap = kTRUE  ; }
81   void   SwitchOffOverlapCheck() {fCheckOverlap = kFALSE ; }
82
83   Float_t GetEMCALOverlapAngle() const {return fEMCALOverlapAngle ; }
84   Float_t GetPHOSOverlapAngle()  const {return fPHOSOverlapAngle  ; }
85   void    SetEMCALOverlapAngle(Float_t angle)  {fEMCALOverlapAngle = angle ; }
86   void    SetPHOSOverlapAngle(Float_t angle)   {fPHOSOverlapAngle  = angle ; }
87     
88  private:
89   
90   void   CheckOverlap(const Float_t anglethres, const Int_t imom, Int_t & iPrimary, Int_t & index, TLorentzVector & mom, Int_t & pdg);
91   void   MakePi0Decay(const TLorentzVector p0, TLorentzVector &p1, TLorentzVector &p2) const ;//, Double_t &angle); 
92   void   FillCalorimeters(Int_t & iParticle, TParticle* particle, TLorentzVector & momentum) ;
93   
94   private:
95   Bool_t    fDecayPi0 ;              // If not decayed, decay pi0 by hand
96   TArrayI * fNeutralParticlesArray ; // Do not keep neutral particles of this list in calorimeter.
97   TArrayI * fChargedParticlesArray ; // Keep charged particles of this list in calorimeter.
98   TArrayI * fStatusArray ;           // Keep particles with status of the list.
99   Bool_t    fKeepAllStatus ;         // Do or do not select particles depending on their status code.
100   Bool_t    fCheckOverlap;           // Check of overlapped photons from pi0 enter the calorimeter
101   Float_t   fEMCALOverlapAngle;      // Aperture angle of photons from decay that is not resolved by EMCAL, in radians
102   Float_t   fPHOSOverlapAngle;       // Aperture angle of photons from decay that is not resolved by PHOS, in radians
103   Int_t     fIndex2ndPhoton;         // Check overlap of first decay photon already done, internal use.
104   Bool_t    fOnlyGeneratorParticles; // Use particles only generated by PYTHIA/HERWIG/... and not by the MC tranport G3/G4/FLUKA ...
105   
106   AliCaloTrackMCReader(const AliCaloTrackMCReader & g) ; // cpy ctor    
107   AliCaloTrackMCReader & operator = (const AliCaloTrackMCReader & g) ;//cpy assignment
108   
109   ClassDef(AliCaloTrackMCReader,4)
110 } ;
111
112
113 #endif //ALICALOTRACKMCREADER_H
114
115
116