PWG4/PartCorrBase/AliCaloTrackReader.h

   1 #ifndef ALICALOTRACKREADER_H
   2 #define ALICALOTRACKREADER_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 // Base class for reading data: MonteCarlo, ESD or AOD, of PHOS EMCAL and
   9 // Central Barrel Tracking detectors.
  10 // Not all MC particles/tracks/clusters are kept, some kinematical restrictions are done.
  11 // Mother class of : AliCaloTrackESDReader: Fills ESD data in 3 TClonesArrays (PHOS, EMCAL, CTS)
  12 //                 : AliCaloTrackMCReader: Fills Kinematics data in 3 TClonesArrays (PHOS, EMCAL, CTS)
  13 //                 : AliCaloTrackAODReader: Fills AOD data in 3 TClonesArrays (PHOS, EMCAL, CTS)
  14 //
  15 // -- Author: Gustavo Conesa (INFN-LNF)
  16
  17 // --- ROOT system ---
  18 #include "TObject.h"
  19 class TClonesArray ;
  20 class TLorentzVector ;
  21 class TString ;
  22 #include "TArrayF.h"
  23
  24 //--- ANALYSIS system ---
  25 class AliStack ;
  26 class AliHeader ;
  27 class AliGenEventHeader ;
  28 #include "AliESDEvent.h"
  29 #include "AliAODEvent.h"
  30 #include "AliMCEvent.h"
  31 class AliLog ;
  32 #include "AliFidutialCut.h"
  33
  34 class AliCaloTrackReader : public TObject {
  35
  36 public:
  37
  38   AliCaloTrackReader() ; // ctor
  39   AliCaloTrackReader(const AliCaloTrackReader & g) ; // cpy ctor
  40   AliCaloTrackReader & operator = (const AliCaloTrackReader & g) ;//cpy assignment
  41   virtual ~AliCaloTrackReader() ;//virtual dtor
  42
  43   enum inputDataType {kESD, kAOD, kMC};
  44
  45   virtual void InitParameters();
  46   virtual void Print(const Option_t * opt) const;
  47
  48   virtual Int_t GetDebug()         const { return fDebug ; }
  49   virtual void  SetDebug(Int_t d)        { fDebug = d ; }
  50   virtual Int_t GetDataType()      const { return fDataType ; }
  51   virtual void  SetDataType(Int_t data ) { fDataType = data ; }
  52
  53   virtual Int_t GetEventNumber() const {return fEventNumber ; }
  54
  55   //Minimum pt setters and getters
  56   virtual Float_t  GetEMCALPtMin() const { return fEMCALPtMin  ; }
  57   virtual Float_t  GetPHOSPtMin()  const { return fPHOSPtMin  ; }
  58   virtual Float_t  GetCTSPtMin()   const { return fCTSPtMin  ; }
  59
  60   virtual void SetEMCALPtMin(Float_t  pt) { fEMCALPtMin = pt ; }
  61   virtual void SetPHOSPtMin(Float_t  pt)  { fPHOSPtMin = pt ; }
  62   virtual void SetCTSPtMin(Float_t  pt)   { fCTSPtMin = pt ; }
  63
  64   //Input setters and getters
  65
  66   Bool_t IsCTSSwitchedOn()  const { return fFillCTS ; }
  67   void SwitchOnCTS()    {fFillCTS = kTRUE ; }
  68   void SwitchOffCTS()   {fFillCTS = kFALSE ; }
  69
  70   Bool_t IsEMCALSwitchedOn() const { return fFillEMCAL ; }
  71   void SwitchOnEMCAL()  {fFillEMCAL = kTRUE ; }
  72   void SwitchOffEMCAL() {fFillEMCAL = kFALSE ; }
  73
  74   Bool_t IsPHOSSwitchedOn()  const { return fFillPHOS ; }
  75   void SwitchOnPHOS()   {fFillPHOS = kTRUE ; }
  76   void SwitchOffPHOS()  {fFillPHOS = kFALSE ; }
  77
  78   Bool_t IsEMCALCellsSwitchedOn() const { return fFillEMCALCells ; }
  79   void SwitchOnEMCALCells()  {fFillEMCALCells = kTRUE ; }
  80   void SwitchOffEMCALCells() {fFillEMCALCells = kFALSE ; }
  81
  82   Bool_t IsPHOSCellsSwitchedOn()  const { return fFillPHOSCells ; }
  83   void SwitchOnPHOSCells()   {fFillPHOSCells = kTRUE ; }
  84   void SwitchOffPHOSCells()  {fFillPHOSCells = kFALSE ; }
  85
  86   virtual void FillInputEvent(Int_t iEntry)  ;
  87   virtual void FillInputCTS()   {;}
  88   virtual void FillInputEMCAL() {;}
  89   virtual void FillInputPHOS()  {;}
  90   virtual void FillInputEMCALCells() {;}
  91   virtual void FillInputPHOSCells()  {;}
  92
  93   virtual TClonesArray* GetAODCTS()   const {return fAODCTS ;}
  94   virtual TClonesArray* GetAODEMCAL() const {return fAODEMCAL ;}
  95   virtual TClonesArray* GetAODPHOS()  const {return fAODPHOS ;}
  96   virtual TNamed* GetEMCALCells()     const {return fEMCALCells ;}
  97   virtual TNamed* GetPHOSCells()      const {return fPHOSCells ;}
  98
  99   virtual AliStack*          GetStack()  const ;
 100   virtual AliHeader*         GetHeader() const ;
 101   virtual AliGenEventHeader* GetGenEventHeader() const ;
 102   virtual AliESDEvent* GetESD() const {return fESD;}
 103   virtual AliAODEvent* GetAOD() const {return fAOD;}
 104   virtual AliMCEvent*  GetMC()  const {return fMC;}
 105   virtual AliVEvent*   GetInputEvent()        const {return (new AliESDEvent());}
 106   virtual void         GetVertex(Double_t * ) const {;}
 107
 108   virtual void SetESD( AliESDEvent* esd) {fESD = esd;}
 109   virtual void SetAOD(AliAODEvent* aod)  {fAOD = aod;}
 110   virtual void SetMC(AliMCEvent* mc)     {fMC  = mc;}
 111
 112   virtual void ResetLists();
 113
 114   virtual AliFidutialCut * GetFidutialCut() const {return  fFidutialCut ;}
 115   virtual void SetFidutialCut(AliFidutialCut * fc) { fFidutialCut = fc ;}
 116
 117   virtual void SetInputEvent(TObject* /*esd*/, TObject* /*aod*/, TObject* /*mc*/) {;}
 118
 119  protected:
 120   Int_t                    fEventNumber; // Event number
 121   Int_t            fDataType ;   // Select MC:Kinematics, Data:ESD/AOD, MCData:Both
 122   Int_t            fDebug;       // Debugging level
 123   AliFidutialCut * fFidutialCut; // Acceptance cuts
 124
 125   Float_t        fCTSPtMin;      // pT  Threshold on charged particles
 126   Float_t        fEMCALPtMin;    // pT Threshold on emcal clusters
 127   Float_t        fPHOSPtMin;     // pT  Threshold on phos clusters
 128
 129   TClonesArray * fAODCTS ;        //! temporal array with tracks
 130   TClonesArray * fAODEMCAL ;      //! temporal array with EMCAL CaloClusters
 131   TClonesArray * fAODPHOS ;       //! temporal array with PHOS CaloClusters
 132   TNamed *       fEMCALCells ;    //! temporal array with EMCAL CaloCells, ESD or AOD
 133   TNamed *       fPHOSCells ;     //! temporal array with PHOS CaloCells, ESD or AOD
 134
 135   AliESDEvent *  fESD;            //! pointer to esd
 136   AliAODEvent *  fAOD;            //! pointer to aod
 137   AliMCEvent  *  fMC;             //! Monte Carlo Event Handler
 138
 139   Bool_t         fFillCTS;        // use data from CTS
 140   Bool_t         fFillEMCAL;      // use data from EMCAL
 141   Bool_t         fFillPHOS;       // use data from PHOS
 142   Bool_t         fFillEMCALCells; // use data from EMCAL
 143   Bool_t         fFillPHOSCells;  // use data from PHOS
 144
 145   ClassDef(AliCaloTrackReader,1)
 146 } ;
 147
 148
 149 #endif //ALICALOTRACKREADER_H
 150
 151
 152