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 TRefArrays (PHOS, EMCAL, CTS)
  12 //                 : AliCaloTrackMCReader: Fills Kinematics data in 3 TRefArrays (PHOS, EMCAL, CTS)
  13 //                 : AliCaloTrackAODReader: Fills AOD data in 3 TRefArrays (PHOS, EMCAL, CTS)
  14 //
  15 // -- Author: Gustavo Conesa (INFN-LNF)
  16
  17 // --- ROOT system ---
  18 #include "TObject.h"
  19 class TRefArray ;
  20 class TLorentzVector ;
  21 class TString ;
  22 class TRefArray;
  23 class TArrayF;
  24
  25 //--- ANALYSIS system ---
  26 class AliStack ;
  27 class AliHeader ;
  28 class AliGenEventHeader ;
  29 class AliVEvent;
  30 class AliAODEvent;
  31 class AliMCEvent;
  32 class AliFidutialCut;
  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 TRefArray* GetAODCTS()   const {return fAODCTS ;}
  94   virtual TRefArray* GetAODEMCAL() const {return fAODEMCAL ;}
  95   virtual TRefArray* 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 AliVEvent*   GetInputEvent()  const {return fInputEvent;}
 103   virtual AliAODEvent* GetOutputEvent() const {return fOutputEvent;}
 104   virtual AliMCEvent*  GetMC()          const {return fMC;}
 105   virtual void         GetVertex(Double_t * ) const {;}
 106
 107   virtual void SetInputEvent(AliVEvent* input)  {fInputEvent  = input;}
 108   virtual void SetOutputEvent(AliAODEvent* aod) {fOutputEvent = aod;}
 109   virtual void SetMC(AliMCEvent* mc)            {fMC  = mc;}
 110
 111   virtual void ResetLists();
 112
 113   virtual AliFidutialCut * GetFidutialCut() const {return  fFidutialCut ;}
 114   virtual void SetFidutialCut(AliFidutialCut * fc) { fFidutialCut = fc ;}
 115
 116   virtual void SetInputOutputMCEvent(AliVEvent* /*esd*/, AliAODEvent* /*aod*/, AliMCEvent* /*mc*/) {;}
 117
 118  protected:
 119   Int_t            fEventNumber; // Event number
 120   Int_t            fDataType ;   // Select MC:Kinematics, Data:ESD/AOD, MCData:Both
 121   Int_t            fDebug;       // Debugging level
 122   AliFidutialCut * fFidutialCut; // Acceptance cuts
 123
 124   Float_t        fCTSPtMin;      // pT Threshold on charged particles
 125   Float_t        fEMCALPtMin;    // pT Threshold on emcal clusters
 126   Float_t        fPHOSPtMin;     // pT Threshold on phos clusters
 127
 128   TRefArray *    fAODCTS ;        //! temporal referenced array with tracks
 129   TRefArray *    fAODEMCAL ;      //! temporal referenced array with EMCAL CaloClusters
 130   TRefArray *    fAODPHOS ;       //! temporal referenced array with PHOS CaloClusters
 131   TNamed *       fEMCALCells ;    //! temporal array with EMCAL CaloCells, ESD or AOD
 132   TNamed *       fPHOSCells ;     //! temporal array with PHOS CaloCells, ESD or AOD
 133
 134   AliVEvent   *  fInputEvent;     //! pointer to esd or aod input
 135   AliAODEvent *  fOutputEvent;    //! pointer to aod output
 136   AliMCEvent  *  fMC;             //! Monte Carlo Event Handler
 137
 138   Bool_t         fFillCTS;        // use data from CTS
 139   Bool_t         fFillEMCAL;      // use data from EMCAL
 140   Bool_t         fFillPHOS;       // use data from PHOS
 141   Bool_t         fFillEMCALCells; // use data from EMCAL
 142   Bool_t         fFillPHOSCells;  // use data from PHOS
 143
 144   ClassDef(AliCaloTrackReader,3)
 145 } ;
 146
 147
 148 #endif //ALICALOTRACKREADER_H
 149
 150
 151