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 TObjArrays (PHOS, EMCAL, CTS)
  12 //                 : AliCaloTrackMCReader: Fills Kinematics data in 3 TObjArrays (PHOS, EMCAL, CTS)
  13 //                 : AliCaloTrackAODReader: Fills AOD data in 3 TObjArrays (PHOS, EMCAL, CTS)
  14 //
  15 // -- Author: Gustavo Conesa (INFN-LNF)
  16
  17 // --- ROOT system ---
  18 #include "TObject.h"
  19 class TObjArray ;
  20 class TLorentzVector ;
  21 #include "TString.h"
  22 class TObjArray;
  23 class TArrayF;
  24 //#include "TTree.h"
  25 class TTree ;
  26
  27 //--- ANALYSIS system ---
  28 class AliStack ;
  29 class AliHeader ;
  30 class AliGenEventHeader ;
  31 class AliVEvent;
  32 #include "AliAODEvent.h"
  33 class AliMCEvent;
  34 class AliFidutialCut;
  35 class AliAODMCHeader;
  36
  37 class AliCaloTrackReader : public TObject {
  38
  39  public:
  40
  41   AliCaloTrackReader() ; // ctor
  42   AliCaloTrackReader(const AliCaloTrackReader & g) ; // cpy ctor
  43   AliCaloTrackReader & operator = (const AliCaloTrackReader & g) ;//cpy assignment
  44   virtual ~AliCaloTrackReader() ;//virtual dtor
  45
  46   enum inputDataType {kESD, kAOD, kMC};
  47
  48   //Select generated events, depending on comparison of pT hard and jets.
  49   virtual Bool_t ComparePtHardAndJetPt() ;
  50   virtual Bool_t IsPtHardAndJetPtComparisonSet() const {return  fComparePtHardAndJetPt ;}
  51   virtual void SetPtHardAndJetPtComparison(Bool_t compare) { fComparePtHardAndJetPt = compare ;}
  52   virtual Float_t GetPtHardAndJetFactor() const {return  fPtHardAndJetPtFactor ;}
  53   virtual void SetPtHardAndJetPtFactor(Float_t factor) { fPtHardAndJetPtFactor = factor ;}
  54
  55   virtual void InitParameters();
  56   virtual void Print(const Option_t * opt) const;
  57
  58   virtual Int_t GetDebug()         const { return fDebug ; }
  59   virtual void  SetDebug(Int_t d)        { fDebug = d ; }
  60   virtual Int_t GetDataType()      const { return fDataType ; }
  61   virtual void  SetDataType(Int_t data ) { fDataType = data ; }
  62
  63   virtual Int_t   GetEventNumber()     const {return fEventNumber ; }
  64   virtual TString GetCurrentFileName() const {return fCurrentFileName ; }
  65
  66   //Minimum pt setters and getters
  67   virtual Float_t  GetEMCALPtMin() const { return fEMCALPtMin  ; }
  68   virtual Float_t  GetPHOSPtMin()  const { return fPHOSPtMin  ; }
  69   virtual Float_t  GetCTSPtMin()   const { return fCTSPtMin  ; }
  70
  71   virtual void SetEMCALPtMin(Float_t  pt) { fEMCALPtMin = pt ; }
  72   virtual void SetPHOSPtMin(Float_t  pt)  { fPHOSPtMin = pt ; }
  73   virtual void SetCTSPtMin(Float_t  pt)   { fCTSPtMin = pt ; }
  74
  75   //Input setters and getters
  76
  77   Bool_t IsCTSSwitchedOn()  const { return fFillCTS ; }
  78   void SwitchOnCTS()    {fFillCTS = kTRUE ; }
  79   void SwitchOffCTS()   {fFillCTS = kFALSE ; }
  80
  81   Bool_t IsEMCALSwitchedOn() const { return fFillEMCAL ; }
  82   void SwitchOnEMCAL()  {fFillEMCAL = kTRUE ; }
  83   void SwitchOffEMCAL() {fFillEMCAL = kFALSE ; }
  84
  85   Bool_t IsPHOSSwitchedOn()  const { return fFillPHOS ; }
  86   void SwitchOnPHOS()   {fFillPHOS = kTRUE ; }
  87   void SwitchOffPHOS()  {fFillPHOS = kFALSE ; }
  88
  89   Bool_t IsEMCALCellsSwitchedOn() const { return fFillEMCALCells ; }
  90   void SwitchOnEMCALCells()  {fFillEMCALCells = kTRUE ; }
  91   void SwitchOffEMCALCells() {fFillEMCALCells = kFALSE ; }
  92
  93   Bool_t IsPHOSCellsSwitchedOn()  const { return fFillPHOSCells ; }
  94   void SwitchOnPHOSCells()   {fFillPHOSCells = kTRUE ; }
  95   void SwitchOffPHOSCells()  {fFillPHOSCells = kFALSE ; }
  96
  97   virtual Bool_t FillInputEvent(const Int_t iEntry, const char *currentFileName)  ;
  98   virtual void FillInputCTS()   {;}
  99   virtual void FillInputEMCAL() {;}
 100   virtual void FillInputPHOS()  {;}
 101   virtual void FillInputEMCALCells() {;}
 102   virtual void FillInputPHOSCells()  {;}
 103
 104   virtual TObjArray* GetAODCTS()   const {return fAODCTS ;}
 105   virtual TObjArray* GetAODEMCAL() const {return fAODEMCAL ;}
 106   virtual TObjArray* GetAODPHOS()  const {return fAODPHOS ;}
 107   virtual TNamed* GetEMCALCells()  const {return fEMCALCells ;}
 108   virtual TNamed* GetPHOSCells()   const {return fPHOSCells ;}
 109
 110   //Get MC  informatio
 111   //Kinematics and galice.root available
 112   virtual AliStack*    GetStack()      const ;
 113   virtual AliHeader*   GetHeader()     const ;
 114   virtual AliGenEventHeader* GetGenEventHeader() const ;
 115   //Filtered kinematics in AOD
 116   virtual TClonesArray*   GetAODMCParticles(Int_t input = 0) const ;
 117   virtual AliAODMCHeader* GetAODMCHeader(Int_t input = 0)    const ;
 118
 119   virtual AliVEvent*   GetInputEvent()  const {return fInputEvent;}
 120   virtual AliAODEvent* GetOutputEvent() const {return fOutputEvent;}
 121   virtual AliMCEvent*  GetMC()          const {return fMC;}
 122   virtual void         GetVertex(Double_t * ) const {;}
 123
 124   virtual void Init();
 125
 126   virtual void SetInputEvent(AliVEvent* input)  {fInputEvent  = input;}
 127   virtual void SetOutputEvent(AliAODEvent* aod) {fOutputEvent = aod;}
 128   virtual void SetMC(AliMCEvent* mc)            {fMC  = mc;}
 129
 130   virtual void ResetLists();
 131
 132   virtual AliFidutialCut * GetFidutialCut() const {return  fFidutialCut ;}
 133   virtual void SetFidutialCut(AliFidutialCut * fc) { fFidutialCut = fc ;}
 134
 135   virtual void SetInputOutputMCEvent(AliVEvent* /*esd*/, AliAODEvent* /*aod*/, AliMCEvent* /*mc*/) {;}
 136
 137   //Methods for mixing with external input file (AOD)
 138   virtual TTree* GetSecondInputAODTree() const {return  fSecondInputAODTree ; }
 139   //virtual void SetSecondInputAODTree(TTree * tree) {fSecondInputAODTree = tree ;
 140   //                                                                                              fSecondInputAODEvent->ReadFromTree(tree);}//Connect tree and AOD event.
 141
 142   virtual AliAODEvent* GetSecondInputAODEvent() const { return fSecondInputAODEvent ; }
 143
 144   TString GetSecondInputFileName() const    {return fSecondInputFileName ; }
 145   void SetSecondInputFileName(TString name) { fSecondInputFileName = name ; }
 146
 147   Int_t GetSecondInputFirstEvent() const    {return fSecondInputFirstEvent ; }
 148   void SetSecondInputFirstEvent(Int_t iEvent0) { fSecondInputFirstEvent = iEvent0 ; }
 149
 150   Int_t GetAODCTSNormalInputEntries()   {if(!fSecondInputAODTree) { fAODCTSNormalInputEntries   = fAODCTS->GetEntriesFast()  ;}
 151                                                                                  return fAODCTSNormalInputEntries ; }
 152   Int_t GetAODEMCALNormalInputEntries() {if(!fSecondInputAODTree) { fAODEMCALNormalInputEntries = fAODEMCAL->GetEntriesFast();}
 153                                                                                  return fAODEMCALNormalInputEntries ; }
 154   Int_t GetAODPHOSNormalInputEntries()  {if(!fSecondInputAODTree) { fAODPHOSNormalInputEntries  = fAODPHOS->GetEntriesFast() ;}
 155                                                                                  return fAODPHOSNormalInputEntries ; }
 156
 157   ULong_t GetTrackStatus() const    {return fTrackStatus ; }
 158   void SetTrackStatus(ULong_t bit) { fTrackStatus = bit ; }
 159
 160   void SwitchOnStack()              { fReadStack          = kTRUE  ; }
 161   void SwitchOffStack()             { fReadStack          = kFALSE ; }
 162   void SwitchOnAODMCParticles()     { fReadAODMCParticles = kTRUE  ; }
 163   void SwitchOffAODMCParticles()    { fReadAODMCParticles = kFALSE ; }
 164   Bool_t ReadStack()          const { return fReadStack            ; }
 165   Bool_t ReadAODMCParticles() const { return fReadAODMCParticles   ; }
 166
 167  protected:
 168   Int_t            fEventNumber; // Event number
 169   TString          fCurrentFileName; // Current file name under analysis
 170   Int_t            fDataType ;   // Select MC:Kinematics, Data:ESD/AOD, MCData:Both
 171   Int_t            fDebug;       // Debugging level
 172   AliFidutialCut * fFidutialCut; // Acceptance cuts
 173
 174   Bool_t           fComparePtHardAndJetPt;  // In MonteCarlo, jet events, reject fake events with wrong jet energy.
 175   Float_t          fPtHardAndJetPtFactor;   // Factor between ptHard and jet pT to reject/accept event.
 176
 177   Float_t        fCTSPtMin;      // pT Threshold on charged particles
 178   Float_t        fEMCALPtMin;    // pT Threshold on emcal clusters
 179   Float_t        fPHOSPtMin;     // pT Threshold on phos clusters
 180
 181   TObjArray *    fAODCTS ;        //! temporal referenced array with tracks
 182   TObjArray *    fAODEMCAL ;      //! temporal referenced array with EMCAL CaloClusters
 183   TObjArray *    fAODPHOS ;       //! temporal referenced array with PHOS CaloClusters
 184   TNamed *       fEMCALCells ;    //! temporal array with EMCAL CaloCells, ESD or AOD
 185   TNamed *       fPHOSCells ;     //! temporal array with PHOS CaloCells, ESD or AOD
 186
 187   AliVEvent   *  fInputEvent;     //! pointer to esd or aod input
 188   AliAODEvent *  fOutputEvent;    //! pointer to aod output
 189   AliMCEvent  *  fMC;             //! Monte Carlo Event Handler
 190
 191   Bool_t         fFillCTS;        // use data from CTS
 192   Bool_t         fFillEMCAL;      // use data from EMCAL
 193   Bool_t         fFillPHOS;       // use data from PHOS
 194   Bool_t         fFillEMCALCells; // use data from EMCAL
 195   Bool_t         fFillPHOSCells;  // use data from PHOS
 196
 197   TTree *        fSecondInputAODTree;    // Tree with second input AOD, for mixing analysis.
 198   AliAODEvent*   fSecondInputAODEvent;   //! pointer to second input AOD event.
 199   TString        fSecondInputFileName;   // File with AOD data to mix with normal stream of data.
 200   Int_t          fSecondInputFirstEvent; // First event to be considered in the mixing.
 201
 202   Int_t          fAODCTSNormalInputEntries;   // Number of entries in CTS   in case of standard input, larger with mixing.
 203   Int_t          fAODEMCALNormalInputEntries; // Number of entries in EMCAL in case of standard input, larger with mixing.
 204   Int_t          fAODPHOSNormalInputEntries;  // Number of entries in PHOS  in case of standard input, larger with mixing.
 205
 206   ULong_t        fTrackStatus        ; // Track selection bit, select tracks refitted in TPC, ITS ...
 207   Bool_t         fReadStack          ; // Access kine information from stack
 208   Bool_t             fReadAODMCParticles ; // Access kine information from filtered AOD MC particles
 209
 210   ClassDef(AliCaloTrackReader,6)
 211 } ;
 212
 213
 214 #endif //ALICALOTRACKREADER_H
 215
 216
 217