EMCAL/AliEMCALGetter.h

   1 #ifndef ALIEMCALGETTER_H
   2 #define ALIEMCALGETTER_H
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 /* $Id$ */
   7
   8 //_________________________________________________________________________
   9 //  A singleton that returns various objects
  10 //  Should be used on the analysis stage to avoid confusing between different
  11 //  branches of reconstruction tree: e.g. reading RecPoints and TS made from
  12 //  another set of RecPoints.
  13 //
  14 //  The objects are retrived from folders.
  15 //*-- Author: Yves Schutz (SUBATECH) & Dmitri Peressounko (RRC KI & SUBATECH)
  16 //
  17
  18
  19 // --- ROOT system ---
  20 #include "TClonesArray.h"
  21 #include "TFolder.h"
  22 #include "TTree.h"
  23 #include "TFile.h"
  24 class TString ;
  25 class TParticle ;
  26 class TTask ;
  27
  28 // --- Standard library ---
  29 #include <stdlib.h>
  30 #include <iostream.h>
  31
  32 // --- AliRoot header files ---
  33
  34 #include "AliRun.h"
  35 #include "AliEMCALv1.h"
  36 #include "AliEMCALHit.h"
  37 #include "AliEMCALDigit.h"
  38 #include "AliEMCALDigitizer.h"
  39 #include "AliEMCALSDigitizer.h"
  40 #include "AliEMCALTowerRecPoint.h"
  41 class AliEMCALGeometry ;
  42 class AliEMCALClusterizerv1 ;
  43 //class AliEMCALTrackSegment ;
  44 //class AliEMCALTrackSegmentMaker ;
  45 //class AliEMCALRecParticle ;
  46 //class AliEMCALPID ;
  47
  48 class AliEMCALGetter : public TObject {
  49
  50  public:
  51
  52   AliEMCALGetter(){
  53     // ctor: this is a singleton, the ctor should never be called but cint needs it as public
  54     cerr << "ERROR: AliPHOGetter is a singleton default ctor not callable" << endl ;
  55     abort() ;
  56   }
  57   AliEMCALGetter(const AliEMCALGetter & obj) {
  58     // cpy ctor requested by Coding Convention
  59     // but not yet needed
  60     abort() ;
  61   }
  62
  63   virtual ~AliEMCALGetter() ;
  64
  65   void CloseFile() ;
  66   const TFolder * Folder(const TString what) const ;
  67   const Bool_t HasFailed(void) const {return fFailed ;}
  68   Bool_t PostPrimaries(void ) const ;
  69   Bool_t PostHits(void ) const ;
  70   Bool_t PostSDigits(      const char * name,  const char * file = 0) const ;
  71   Bool_t PostDigits(       const char * name ) const ;
  72   Bool_t PostRecPoints(    const char * name ) const ;
  73   //Bool_t PostTrackSegments(const char * name) const ;
  74   //Bool_t PostRecParticles( const char * name) const ;
  75
  76   Bool_t PostClusterizer( const char * name) const ;
  77   Bool_t PostClusterizer(AliEMCALClusterizerv1 * clu) const ;
  78   Bool_t PostSDigitizer (AliEMCALSDigitizer * sdigitizer) const ;
  79   Bool_t PostSDigitizer ( const char * name, const char * file ) const ;
  80   Bool_t PostDigitizer (AliEMCALDigitizer * digitizer) const ;
  81   Bool_t PostDigitizer  ( const char * name) const ;
  82   //Bool_t PostTrackSegmentMaker(AliEMCALTrackSegmentMaker * tsm) const ;
  83   //Bool_t PostTrackSegmentMaker(const char * name ) const ;
  84   //Bool_t PostPID  (AliEMCALPID * pid) const ;
  85   //Bool_t PostPID  (const char * name ) const ;
  86   //Bool_t PostQA   (void) const ;
  87
  88
  89   void   Event(const Int_t event, const char * opt = "HSDR") ;
  90   void   Track(Int_t itrack) ;
  91
  92   //Method to be used when digitizing under AliRunDigitizer, who opens all files etc.
  93   void   ReadTreeS(TTree * treeS,Int_t input) ;
  94
  95   Int_t  EventNumber()       { return (Int_t) gAlice->GetEvNumber() ; }
  96   Int_t  MaxEvent()          { return (Int_t) gAlice->TreeE()->GetEntries() ; }
  97   static AliEMCALGetter * GetInstance(const char* headerFile,
  98                                      const char* branchTitle = "Default", const Option_t * rw="" ) ;
  99   static AliEMCALGetter *   GetInstance() ;
 100
 101   const AliEMCALv1 *         EMCAL()  ;
 102   AliEMCALGeometry * EMCALGeometry() ;
 103    // Alarms
 104   //TFolder * Alarms() const { return (TFolder*)(ReturnO("Alarms", 0)) ; }
 105   //TObjArray *  Alarms(const char * name ) const { return (TObjArray*)(ReturnO("Alarms", name)) ; }
 106
 107   // QA Tasks
 108   //TTask * QATasks(const char * name = 0) const { return (TTask*)(ReturnT("QATasks", name)) ; }
 109
 110   // Primaries
 111   TClonesArray *  Primaries(void) const { return (TClonesArray*)(ReturnO("Primaries")) ; }
 112
 113
 114   // Hits
 115   TTree *               TreeH(TString filename="") ;
 116   const TClonesArray *  Hits(void) { return static_cast<const TClonesArray*>(ReturnO("Hits")) ; }
 117   const AliEMCALHit *   Hit(Int_t index)  { return static_cast<const AliEMCALHit*>(Hits()->At(index) );}
 118
 119   // SDigits
 120   TTree *         TreeS(TString filename="") ;
 121   TClonesArray *  SDigits(const char * name = 0, const char * file=0) {
 122     return static_cast<TClonesArray*>(ReturnO("SDigits", name, file)) ;
 123   }
 124   const AliEMCALDigit *  SDigit(Int_t index) { return static_cast<const AliEMCALDigit*>(SDigits()->At(index)) ;}
 125
 126   AliEMCALSDigitizer *  SDigitizer(const char * name =0) const {
 127     return ((AliEMCALSDigitizer*)(ReturnT("SDigitizer", name))) ;
 128   }
 129
 130   // Digits
 131   TTree *         TreeD(TString filename="") ;
 132   TClonesArray *  Digits(const char * name = 0)const  {
 133     return static_cast<TClonesArray*>(ReturnO("Digits", name)) ;
 134   }
 135   const AliEMCALDigit *  Digit(Int_t index) { return static_cast<const AliEMCALDigit *>(Digits()->At(index)) ;}
 136   AliEMCALDigitizer *  Digitizer(const char * name =0) const {
 137     return (AliEMCALDigitizer*)(ReturnT("Digitizer", name)) ;
 138   }
 139
 140   //  RecPoints
 141   TObjArray * TowerRecPoints(const char * name = 0) const {
 142     return (TObjArray*)(ReturnO("TowerRecPoints", name)) ; }
 143   TObjArray * PreShowerRecPoints(const char * name = 0) const {
 144     return (TObjArray*)(ReturnO("PreShoRecPoints", name)) ; }
 145   const AliEMCALTowerRecPoint *  TowerRecPoint(Int_t index) {
 146     return static_cast<const AliEMCALTowerRecPoint *>(TowerRecPoints()->At(index)) ;}
 147   const AliEMCALTowerRecPoint *  PreShowerRecPoint(Int_t index) {
 148     return static_cast<const AliEMCALTowerRecPoint *>(PreShowerRecPoints()->At(index)) ;}
 149
 150     AliEMCALClusterizerv1 * Clusterizer (const char * name =0) const {
 151     return (AliEMCALClusterizerv1*)(ReturnT("Clusterizer", name)) ; }
 152
 153   // TrackSegments
 154   //TClonesArray * TrackSegments(const char * name = 0) const
 155    //                { return (TClonesArray*)(ReturnO("TrackSegments", name)) ; }
 156   //AliEMCALTrackSegmentMaker * TrackSegmentMaker (const char * name =0) const
 157     //               { return (AliEMCALTrackSegmentMaker*)(ReturnT("TrackSegmentMaker", name)) ; }
 158
 159   // RecParticles
 160   //TClonesArray * RecParticles(const char * name = 0) const
 161     //               { return (TClonesArray*)(ReturnO("RecParticles", name)) ; }
 162     //AliEMCALPID * PID(const char * name =0) const
 163       //             { return (AliEMCALPID*)(ReturnT("PID", name)) ; }
 164
 165   // Primaries
 166   TTree *                     TreeK(TString filename="") ;
 167   const TParticle *           Primary(Int_t index) const ;
 168   const Int_t                 NPrimaries()const { return fNPrimaries; }
 169
 170   void RemoveTask(TString opt, TString name) const ;
 171   void RemoveObjects(TString opt, TString name) const ;
 172   void RemoveSDigits() const ;
 173   void SetDebug(Int_t level) {fDebug = level;} // Set debug level
 174
 175   AliEMCALGetter & operator = (const AliEMCALGetter & ) {
 176     // assignement operator requested by coding convention, but not needed
 177     abort() ;
 178     return *this ;
 179   }
 180
 181   TFolder * SDigitsFolder() { return dynamic_cast<TFolder*>(fSDigitsFolder->FindObject("EMCAL")) ; }
 182
 183  private:
 184
 185   AliEMCALGetter(const char* headerFile, const char* branchTitle ="Default", const Option_t * rw ="") ;
 186   void CreateWhiteBoard() const ;
 187   TObject * ReturnO(TString what, TString name=0, TString file=0) const ;
 188   const TTask * ReturnT(TString what,TString name=0) const ;
 189   void DefineBranchTitles(char* branch, char* branchTitle) ;
 190   void ReadTreeD() ;
 191   void ReadTreeH() ;
 192   void ReadTreeR() ;
 193   void ReadTreeS(Int_t event) ;
 194   //void ReadTreeQA() ;
 195   void ReadPrimaries() ;
 196
 197   TObject ** PrimariesRef(void) const ;
 198   TObject ** HitsRef(void) const ;
 199   TObject ** SDigitsRef(const char * name, const char * file = 0 ) const;
 200   TObject ** DigitsRef (const char * name)   const ;
 201   TObject ** TowerRecPointsRef (const char * name) const ;
 202   TObject ** PreShoRecPointsRef (const char * name) const ;
 203   //TObject ** TrackSegmentsRef(const char * name)   const ;
 204   //TObject ** RecParticlesRef (const char * name)   const ;
 205   //TObject ** AlarmsRef (void)   const ;
 206
 207   TObject ** SDigitizerRef (const char * name) const ;
 208   TObject ** DigitizerRef  (const char * name) const ;
 209   TObject ** ClusterizerRef(const char * name) const ;
 210   //TObject ** TSMakerRef    (const char * name) const ;
 211   //TObject ** PIDRef        (const char * name) const ;
 212
 213  private:
 214
 215   static TFile *        fFile ;              //!
 216   TString        fHeaderFile ;        //! File in which gAlice lives
 217   TString        fBranchTitle ;       //!
 218   //TString        fTrackSegmentsTitle ;//!
 219   TString        fRecPointsTitle ;    //!
 220   //TString        fRecParticlesTitle ; //!
 221   TString        fDigitsTitle ;       //!
 222   TString        fSDigitsTitle ;      //!
 223
 224   Bool_t         fFailed ;            //! true if file is not opened and/or galice not found
 225   Int_t          fDebug ;             // Debug level
 226
 227   AliRun *       fAlice ;             //! needed to read TreeK if in an other file than fHeaderFile
 228   Int_t          fNPrimaries ;        //! # of primaries
 229
 230   TObjArray *    fPrimaries ;         //! list of lists of primaries-for the case of mixing
 231
 232   TFolder *      fModuleFolder ;      //!Folder that contains the modules
 233   TFolder *      fPrimariesFolder ;   //!Folder that contains the Primary Particles
 234   TFolder *      fHitsFolder ;        //!Folder that contains the Hits
 235   TFolder *      fSDigitsFolder ;     //!Folder that contains the SDigits
 236   TFolder *      fDigitsFolder ;      //!Folder that contains the Digits
 237   TFolder *      fRecoFolder ;        //!Folder that contains the reconstructed objects (RecPoints, TrackSegments, RecParticles)
 238   //TFolder *      fQAFolder ;          //!Folder that contains the QA objects
 239   TFolder *      fTasksFolder ;       //!Folder that contains the Tasks (sdigitizer, digitizer, reconstructioner)
 240
 241   static AliEMCALGetter * fgObjGetter; // pointer to the unique instance of the singleton
 242
 243   ClassDef(AliEMCALGetter,1)  // Algorithm class that provides methods to retrieve objects from a list knowing the index
 244
 245 };
 246
 247 #endif // AliEMCALGETTER_H