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