Loaders removed from the reconstruction code (C.Cheshkov)
[u/mrichter/AliRoot.git] / PHOS / AliPHOSGetter.h
1 #ifndef ALIPHOSGETTER_H
2 #define ALIPHOSGETTER_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 "TObject.h"  
21 class TParticle ;
22 class TTree ; 
23 class TGraph ; 
24 class TF1 ; 
25
26 // --- Standard library ---
27
28 // --- AliRoot header files ---
29 #include "AliConfig.h" 
30 #include "AliPHOSLoader.h" 
31 #include "AliPHOSHit.h" 
32 #include "AliPHOSDigit.h"
33 #include "AliPHOSEmcRecPoint.h" 
34 #include "AliPHOSCpvRecPoint.h" 
35 #include "AliPHOSTrackSegment.h" 
36 #include "AliPHOSRecParticle.h" 
37 #include "AliPHOSDigitizer.h"
38 #include "AliPHOSSDigitizer.h"
39 #include "AliPHOSCalibData.h"
40
41 class AliPHOS ;  
42 class AliPHOSGeometry ;
43 class AliPHOSClusterizer ;
44 class AliPHOSTrackSegmentMaker ;  
45 class AliPHOSPID ; 
46 class AliPHOSBeamTestEvent ;
47 class AliESD ; 
48 class AliRawReader ;
49
50 class AliPHOSGetter : public TObject {
51   
52 public:  
53   // ctor: this is a singleton, the ctor should never be called but cint needs it as public
54   AliPHOSGetter() ;
55
56 public:
57   AliPHOSGetter(const AliPHOSGetter & obj) ;
58   AliPHOSGetter & operator = (const AliPHOSGetter & ) {
59     // assignement operator requested by coding convention, but not needed
60     Fatal("operator =", "not implemented") ;
61     return *this ; 
62   }
63   virtual ~AliPHOSGetter() ; 
64   
65   //=========== Instantiators ================
66   static AliPHOSGetter * Instance(const char* headerFile,
67                                   const char* version = AliConfig::GetDefaultEventFolderName(),
68                                   Option_t * openingOption = "READ" ) ; 
69   static AliPHOSGetter * Instance() ; 
70   
71   void Print(const Option_t *)const{}
72   static void Print() ; 
73   
74   //=========== General information about run ==============
75   virtual Bool_t IsLoaded(TString tree) const { return fLoadingStatus.Contains(tree) ; } 
76   virtual void   SetLoaded(TString tree) { fLoadingStatus += tree ; } 
77   
78   virtual Int_t  MaxEvent() const ; 
79   virtual Int_t  EventNumber() const ; 
80   virtual Bool_t VersionExists(TString & opt) const ; 
81   virtual UShort_t EventPattern(void) const ; 
82   virtual Float_t  BeamEnergy(void) const ;
83   
84   //========== PHOSGeometry and PHOS ============= 
85   virtual AliPHOS *         PHOS() const  ;  
86   virtual AliPHOSGeometry * PHOSGeometry() const ; 
87   
88   //========== Methods to read something from file ==========
89   virtual void   Event(Int_t event, const char * opt = "HSDRTP") ;    
90   void   Event(AliRawReader *rawReader, const char * opt = "W",Bool_t isOldRCUFormat = kFALSE) ;    
91   virtual void   Track(Int_t itrack) ;
92  
93   
94   //-----------------now getter's data--------------------------------------
95   AliPHOSCalibrationDB * CalibrationDB(){return  fcdb; }
96   void ReadCalibrationDB(const char * /*name*/, const char * /*filename*/){ ;}
97   void SetCalibrationDB(AliPHOSCalibrationDB * cdb) {fcdb = cdb ;}
98   
99   void SetCalibData(AliPHOSCalibData* calibda) { fgCalibData = calibda; }
100   AliPHOSCalibData * CalibData();
101
102   //=========== Primaries ============
103   virtual TClonesArray *    Primaries(void) ;
104   virtual TParticle * Primary(Int_t index) const ;
105   virtual Int_t       NPrimaries()const { return fNPrimaries; }
106   virtual TParticle * Secondary(const TParticle * p, Int_t index=1) const ;  
107   
108   //=========== Hits =================
109   virtual TClonesArray *  Hits(void) const ; 
110   virtual AliPHOSHit *    Hit(Int_t index) const { return dynamic_cast<AliPHOSHit*>(Hits()->At(index) );}
111   virtual TTree *         TreeH() const ; 
112   
113   //=========== SDigits ==============
114   virtual TClonesArray *      SDigits() const ;  
115   virtual AliPHOSDigit *      SDigit(Int_t index) const { return static_cast<AliPHOSDigit *>(SDigits()->At(index)) ;} 
116   virtual TTree *             TreeS() const ; 
117   virtual AliPHOSSDigitizer * SDigitizer() ;  
118   
119   virtual TString             GetSDigitsFileName() const { return PhosLoader()->GetSDigitsFileName() ; }  
120   virtual Int_t               LoadSDigits(Option_t* opt="") const { return PhosLoader()->LoadSDigits(opt) ; }
121   virtual Int_t               LoadSDigitizer(Option_t* opt="") const { return  PhosLoader()->LoadSDigitizer(opt) ; }
122   virtual Int_t               WriteSDigits(Option_t* opt="") const  { return PhosLoader()->WriteSDigits(opt) ; }
123   virtual Int_t               WriteSDigitizer(Option_t* opt="") const {
124     return  PhosLoader()->WriteSDigitizer(opt) ; }
125   
126   //========== Digits ================
127   virtual TClonesArray * Digits() const ;
128   virtual AliPHOSDigit * Digit(Int_t index) const { return static_cast<AliPHOSDigit *>(Digits()->At(index)) ;} 
129   virtual TTree *        TreeD() const ; 
130   virtual AliPHOSDigitizer * Digitizer() ;
131   virtual TString             GetDigitsFileName() const { return PhosLoader()->GetDigitsFileName() ; }  
132   virtual Int_t               LoadDigits(Option_t* opt="") const { return PhosLoader()->LoadDigits(opt) ; }
133   virtual Int_t               LoadDigitizer(Option_t* opt="") const {
134     return  PhosLoader()->LoadDigitizer(opt) ; }
135   virtual Int_t               WriteDigits(Option_t* opt="") const { return PhosLoader()->WriteDigits(opt) ; }
136   virtual Int_t               WriteDigitizer(Option_t* opt="") const {
137     return  PhosLoader()->WriteDigitizer(opt) ; }
138
139   //Methods to distinguish raw and simulated digits
140   virtual Bool_t              IsRawDigits(void) const {return fRawDigits;}
141   virtual void                SetRawDigits(Bool_t isRaw = kTRUE){fRawDigits = isRaw;}
142   
143   //========== RecPoints =============
144   virtual TObjArray *           EmcRecPoints() const;
145   virtual AliPHOSEmcRecPoint *  EmcRecPoint(Int_t index) const { return static_cast<AliPHOSEmcRecPoint *>(EmcRecPoints()->At(index)) ;} 
146   virtual TObjArray *           CpvRecPoints() const ; 
147   virtual AliPHOSCpvRecPoint *  CpvRecPoint(Int_t index) const { return static_cast<AliPHOSCpvRecPoint *>(CpvRecPoints()->At(index)) ;} 
148   virtual TTree *               TreeR() const ;
149   virtual AliPHOSClusterizer * Clusterizer() ;
150   virtual TString               GetRecPointsFileName() const { return PhosLoader()->GetRecPointsFileName() ; } 
151   virtual Int_t                 LoadRecPoints(Option_t* opt="") const { return PhosLoader()->LoadRecPoints(opt) ; }
152   virtual Int_t                 WriteRecPoints(Option_t* opt="") const { return PhosLoader()->WriteRecPoints(opt) ; }
153   
154   //========== TrackSegments   TClonesArray * TrackSegments(const char * name = 0) { 
155   virtual TClonesArray *           TrackSegments() const;
156   virtual AliPHOSTrackSegment *  TrackSegment(Int_t index) const { return static_cast<AliPHOSTrackSegment *>(TrackSegments()->At(index)) ;} 
157   virtual TTree *               TreeT() const ;
158   virtual AliPHOSTrackSegmentMaker * TrackSegmentMaker() ;
159   virtual TString               GetTracksFileName() const { return PhosLoader()->GetTracksFileName() ; } 
160   virtual Int_t                 LoadTracks(Option_t* opt="") const { return PhosLoader()->LoadTracks(opt) ; }
161   virtual Int_t                 LoadTrackSegementMaker(Option_t* opt="") const {
162     return  PhosLoader()->LoadTrackSegmentMaker(opt) ; }
163   virtual Int_t                 WriteTracks(Option_t* opt="") const { return PhosLoader()->WriteTracks(opt) ; }
164   virtual Int_t                 WriteTrackSegmentMaker(Option_t* opt="") const {
165     return  PhosLoader()->WriteTracker(opt) ; }
166   
167   //========== RecParticles ===========
168   virtual TClonesArray *         RecParticles() const;
169   virtual AliPHOSRecParticle *   RecParticle(Int_t index) const { return static_cast<AliPHOSRecParticle *>(RecParticles()->At(index)) ;} 
170   virtual TTree *               TreeP() const ;
171   virtual AliPHOSPID * PID() ;
172   virtual TString               GetRecParticlesFileName() const { return PhosLoader()->GetRecParticlesFileName() ; } 
173   virtual Int_t                 LoadRecParticles(Option_t* opt="") const { return PhosLoader()->LoadRecParticles(opt) ; }
174   virtual Int_t                 LoadPID(Option_t* opt="") const {
175     return  PhosLoader()->LoadPID(opt) ; }
176   virtual Int_t                 WriteRecParticles(Option_t* opt="") const { return PhosLoader()->WriteRecParticles(opt) ; }
177   virtual Int_t                 WritePID(Option_t* opt="") const {
178     return  PhosLoader()->WritePID(opt) ; }
179
180   //========== Raw ===========
181   virtual Int_t ReadRaw(AliRawReader *rawReader,Bool_t isOldRCUFormat) ; 
182
183   void SetDebug(Int_t level) {fgDebug = level;} // Set debug level 
184   virtual void PostSDigitizer (AliPHOSSDigitizer * sdigitizer) 
185     const {PhosLoader()->PostSDigitizer(sdigitizer);}    
186   virtual void PostDigitizer (AliPHOSDigitizer * digitizer)    
187     const {PhosLoader()->PostDigitizer(digitizer);}
188   
189   virtual TString Version() const  { return PhosLoader()->GetTitle() ; } 
190   virtual AliPHOSLoader * PhosLoader() const { return  fgPhosLoader ; }
191   virtual void Reset() ;
192   
193   virtual AliESD * ESD() const { return fESD ; }
194   
195 protected :
196   AliPHOSGetter(Int_t /*i*/) ;
197   
198 protected:
199   static AliPHOSGetter * fgObjGetter; // pointer to the unique instance of the singleton 
200   
201 private:
202   AliPHOSGetter(const char* headerFile,
203                 const char* version = AliConfig::GetDefaultEventFolderName(),
204                 Option_t * openingOption = "READ") ;
205 private:
206   
207   Int_t ReadTreeD(void) const ;
208   Int_t ReadTreeH(void) const ;
209   Int_t ReadTreeR(void) const ;
210   Int_t ReadTreeT(void) const ;
211   Int_t ReadTreeS(void) const ;
212   Int_t ReadTreeP(void) const ;
213
214   Int_t ReadTreeE(Int_t event) ;    
215   Bool_t OpenESDFile() ;
216   void ReadPrimaries(void) ;
217
218   void FitRaw(Bool_t lowGainFlag, TGraph * gLowGain, TGraph * gHighGain, TF1* signalF, Double_t & energy, Double_t & time) const; 
219
220   Int_t CalibrateRaw (Double_t energy, Int_t *relId);
221
222 private:
223   
224   AliPHOSBeamTestEvent * fBTE ;           //! Header if BeamTest Event
225   static Int_t          fgDebug ;             //! Debug level
226   
227   TString           fLoadingStatus ;     //! tells which trees are loaded
228   Int_t             fNPrimaries ;        //! # of primaries  
229   TClonesArray *    fPrimaries ;         //! list of lists of primaries
230   TFile *           fESDFile ;           //! ESD file
231   TString           fESDFileName ;       //! ESD File Name
232   AliESD *          fESD ;               //! ESD object
233   TTree *           fESDTree ;           //! ESD Tree
234   
235   Bool_t            fRawDigits ;         //! true is raw data
236
237   AliPHOSCalibrationDB * fcdb ;          //! Calibration DB for beam test run 2004
238   static AliPHOSCalibData * fgCalibData; //! instance of AliPHOSCalibData
239
240   static AliPHOSLoader * fgPhosLoader ; // the loader for the NewIO
241   
242   enum EDataTypes{kHits,kSDigits,kDigits,kRecPoints,kTracks,kNDataTypes};
243
244   
245   ClassDef(AliPHOSGetter,2)  // Algorithm class that provides methods to retrieve objects from a list knowing the index 
246     
247     };
248
249 #endif // AliPHOSGETTER_H