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 #include "TString.h"
  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   virtual TString GetCurrentFileName() const {return fCurrentFileName ; }
  55
  56   //Minimum pt setters and getters
  57   virtual Float_t  GetEMCALPtMin() const { return fEMCALPtMin  ; }
  58   virtual Float_t  GetPHOSPtMin()  const { return fPHOSPtMin  ; }
  59   virtual Float_t  GetCTSPtMin()   const { return fCTSPtMin  ; }
  60
  61   virtual void SetEMCALPtMin(Float_t  pt) { fEMCALPtMin = pt ; }
  62   virtual void SetPHOSPtMin(Float_t  pt)  { fPHOSPtMin = pt ; }
  63   virtual void SetCTSPtMin(Float_t  pt)   { fCTSPtMin = pt ; }
  64
  65   //Input setters and getters
  66
  67   Bool_t IsCTSSwitchedOn()  const { return fFillCTS ; }
  68   void SwitchOnCTS()    {fFillCTS = kTRUE ; }
  69   void SwitchOffCTS()   {fFillCTS = kFALSE ; }
  70
  71   Bool_t IsEMCALSwitchedOn() const { return fFillEMCAL ; }
  72   void SwitchOnEMCAL()  {fFillEMCAL = kTRUE ; }
  73   void SwitchOffEMCAL() {fFillEMCAL = kFALSE ; }
  74
  75   Bool_t IsPHOSSwitchedOn()  const { return fFillPHOS ; }
  76   void SwitchOnPHOS()   {fFillPHOS = kTRUE ; }
  77   void SwitchOffPHOS()  {fFillPHOS = kFALSE ; }
  78
  79   Bool_t IsEMCALCellsSwitchedOn() const { return fFillEMCALCells ; }
  80   void SwitchOnEMCALCells()  {fFillEMCALCells = kTRUE ; }
  81   void SwitchOffEMCALCells() {fFillEMCALCells = kFALSE ; }
  82
  83   Bool_t IsPHOSCellsSwitchedOn()  const { return fFillPHOSCells ; }
  84   void SwitchOnPHOSCells()   {fFillPHOSCells = kTRUE ; }
  85   void SwitchOffPHOSCells()  {fFillPHOSCells = kFALSE ; }
  86
  87   virtual void FillInputEvent(const Int_t iEntry, const char *currentFileName)  ;
  88   virtual void FillInputCTS()   {;}
  89   virtual void FillInputEMCAL() {;}
  90   virtual void FillInputPHOS()  {;}
  91   virtual void FillInputEMCALCells() {;}
  92   virtual void FillInputPHOSCells()  {;}
  93
  94   virtual TRefArray* GetAODCTS()   const {return fAODCTS ;}
  95   virtual TRefArray* GetAODEMCAL() const {return fAODEMCAL ;}
  96   virtual TRefArray* GetAODPHOS()  const {return fAODPHOS ;}
  97   virtual TNamed* GetEMCALCells()  const {return fEMCALCells ;}
  98   virtual TNamed* GetPHOSCells()   const {return fPHOSCells ;}
  99
 100   virtual AliStack*    GetStack()      const ;
 101   virtual AliHeader*   GetHeader()     const ;
 102   virtual AliGenEventHeader* GetGenEventHeader() const ;
 103   virtual AliVEvent*   GetInputEvent()  const {return fInputEvent;}
 104   virtual AliAODEvent* GetOutputEvent() const {return fOutputEvent;}
 105   virtual AliMCEvent*  GetMC()          const {return fMC;}
 106   virtual void         GetVertex(Double_t * ) const {;}
 107
 108   virtual void SetInputEvent(AliVEvent* input)  {fInputEvent  = input;}
 109   virtual void SetOutputEvent(AliAODEvent* aod) {fOutputEvent = 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 SetInputOutputMCEvent(AliVEvent* /*esd*/, AliAODEvent* /*aod*/, AliMCEvent* /*mc*/) {;}
 118
 119  protected:
 120   Int_t            fEventNumber; // Event number
 121   TString          fCurrentFileName; // Current file name under analysis
 122   Int_t            fDataType ;   // Select MC:Kinematics, Data:ESD/AOD, MCData:Both
 123   Int_t            fDebug;       // Debugging level
 124   AliFidutialCut * fFidutialCut; // Acceptance cuts
 125
 126   Float_t        fCTSPtMin;      // pT Threshold on charged particles
 127   Float_t        fEMCALPtMin;    // pT Threshold on emcal clusters
 128   Float_t        fPHOSPtMin;     // pT Threshold on phos clusters
 129
 130   TRefArray *    fAODCTS ;        //! temporal referenced array with tracks
 131   TRefArray *    fAODEMCAL ;      //! temporal referenced array with EMCAL CaloClusters
 132   TRefArray *    fAODPHOS ;       //! temporal referenced array with PHOS CaloClusters
 133   TNamed *       fEMCALCells ;    //! temporal array with EMCAL CaloCells, ESD or AOD
 134   TNamed *       fPHOSCells ;     //! temporal array with PHOS CaloCells, ESD or AOD
 135
 136   AliVEvent   *  fInputEvent;     //! pointer to esd or aod input
 137   AliAODEvent *  fOutputEvent;    //! pointer to aod output
 138   AliMCEvent  *  fMC;             //! Monte Carlo Event Handler
 139
 140   Bool_t         fFillCTS;        // use data from CTS
 141   Bool_t         fFillEMCAL;      // use data from EMCAL
 142   Bool_t         fFillPHOS;       // use data from PHOS
 143   Bool_t         fFillEMCALCells; // use data from EMCAL
 144   Bool_t         fFillPHOSCells;  // use data from PHOS
 145
 146   ClassDef(AliCaloTrackReader,4)
 147 } ;
 148
 149
 150 #endif //ALICALOTRACKREADER_H
 151
 152
 153