Line data Source code
1 : #ifndef ALIPHOSLOADER_H
2 : #define ALIPHOSLOADER_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 : /* History of cvs commits:
9 : *
10 : * $Log$
11 : * Revision 1.12 2006/08/25 16:00:53 kharlov
12 : * Compliance with Effective C++AliPHOSHit.cxx
13 : *
14 : * Revision 1.11 2006/08/01 12:15:03 cvetan
15 : * Adding a constructor from TFolder. Needed by AliReconstruction plugin scheme
16 : *
17 : * Revision 1.10 2005/05/28 14:19:04 schutz
18 : * Compilation warnings fixed by T.P.
19 : *
20 : */
21 :
22 : //_________________________________________________________________________
23 : // A singleton that returns various objects
24 : // Should be used on the analysis stage to avoid confusing between different
25 : // branches of reconstruction tree: e.g. reading RecPoints and TS made from
26 : // another set of RecPoints.
27 : //
28 : // The objects are retrived from folders.
29 : //-- Author: Yves Schutz (SUBATECH) & Dmitri Peressounko (RRC KI & SUBATECH)
30 : //
31 :
32 :
33 : // --- ROOT system ---
34 : class TString ;
35 : class TParticle ;
36 : #include <TClonesArray.h>
37 : #include <TTree.h>
38 :
39 : // --- Standard library ---
40 :
41 : // --- AliRoot header files ---
42 : #include "AliRun.h"
43 : #include "AliLoader.h"
44 : #include "AliRunLoader.h"
45 : #include "AliPHOSClusterizer.h"
46 : #include "AliPHOSTrackSegmentMaker.h"
47 : #include "AliPHOSPID.h"
48 : class AliPHOS ;
49 : class AliPHOSHit ;
50 : class AliPHOSDigit ;
51 : class AliPHOSEmcRecPoint ;
52 : class AliPHOSCpvRecPoint ;
53 : class AliPHOSTrackSegment ;
54 : class AliPHOSRecParticle ;
55 : class AliPHOSSDigitizer ;
56 : class AliPHOSDigitizer ;
57 :
58 : class AliPHOSLoader : public AliLoader {
59 :
60 : public:
61 :
62 : AliPHOSLoader();
63 : AliPHOSLoader(const Char_t *detname,const Char_t *eventfoldername);
64 : AliPHOSLoader(const Char_t *detname,TFolder *topfolder);
65 :
66 : virtual ~AliPHOSLoader() ;
67 :
68 : Int_t GetEvent();//extends the method on PHOS RecPart posting
69 : Int_t SetEvent();//extends the method on PHOS RecPart posting
70 :
71 : Bool_t BranchExists(const TString& recName);
72 : Int_t LoadHits(Option_t* opt=""); //reads from disk and sends them to folder; array as well as tree
73 : Int_t LoadSDigits(Option_t* opt="");
74 : Int_t LoadDigits(Option_t* opt=""); //reads Digits from disk and sends them to folder; array as well as tree
75 : Int_t LoadRecPoints(Option_t* opt=""); //reads RecPoints from disk and sends them to folder; array as well as tree
76 : Int_t LoadTracks(Option_t* opt=""); //reads Tracks from disk and sends them to folder; array as well as tree
77 : Int_t LoadRecParticles(Option_t* opt="");
78 :
79 : Int_t PostHits()const; //Posts the
80 : Int_t PostSDigits()const;
81 : Int_t PostDigits()const;
82 : Int_t PostRecPoints()const;
83 : Int_t PostTracks()const;
84 : Int_t PostRecParticles()const;
85 :
86 : void CleanFolders();//cleans all the stuff loaded by this detector + calls AliLoader::Clean
87 :
88 : void CleanHits()const;
89 : void CleanSDigits()const;
90 : void CleanDigits()const;
91 : void CleanRecPoints()const;
92 : void CleanTracks()const;
93 : void CleanRecParticles();
94 :
95 :
96 : /*******************************************************************/
97 : /*******************************************************************/
98 : /*******************************************************************/
99 :
100 40 : TObject** HitsRef(){return GetDetectorDataRef(Hits());}
101 60 : TObject** SDigitsRef(){return GetDetectorDataRef(SDigits());}
102 18 : TObject** DigitsRef(){return GetDetectorDataRef(Digits());}
103 0 : TObject** EmcRecPointsRef(){return GetDetectorDataRef(EmcRecPoints());}
104 0 : TObject** CpvRecPointsRef(){return GetDetectorDataRef(CpvRecPoints());}
105 0 : TObject** TracksRef(){return GetDetectorDataRef(TrackSegments());}
106 0 : TObject** RecParticlesRef(){return GetDetectorDataRef(RecParticles());}
107 0 : TObject** AlarmsRef(){return GetDetectorDataRef(Alarms());}
108 : void Track(Int_t itrack) ;
109 :
110 : static AliPHOSLoader* GetPHOSLoader(const char* eventfoldername);
111 :
112 : //Method to be used when digitizing under AliDigitizationInput, who opens all files etc.
113 0 : Int_t EventNumber() { return (Int_t) GetRunLoader()->GetEventNumber();}
114 0 : Int_t MaxEvent() { return (Int_t) GetRunLoader()->TreeE()->GetEntries();}
115 :
116 : /* const AliPHOS * PHOS(); */
117 : // Alarms
118 : // TFolder * Alarms() const { return (TFolder*)(ReturnO("Alarms", 0)); }
119 0 : TObjArray * Alarms() {return 0x0;}
120 :
121 : /*********************************************/
122 : /************ TClonesArrays ***********/
123 : /*********************************************/
124 : /**** H i t s ****/
125 : TClonesArray* Hits(void);
126 : const AliPHOSHit* Hit(Int_t index);
127 : void MakeHitsArray();
128 : /**** S D i g i t s ****/
129 : TClonesArray* SDigits();
130 : const AliPHOSDigit* SDigit(Int_t index);
131 : void MakeSDigitsArray();
132 : /**** D i g i t s ****/
133 : TClonesArray* Digits();
134 : const AliPHOSDigit * Digit(Int_t index);
135 : void MakeDigitsArray();
136 : /**** R e c P o i n t s ****/
137 : TObjArray * EmcRecPoints();
138 : TObjArray * CpvRecPoints();
139 : const AliPHOSEmcRecPoint * EmcRecPoint(Int_t index) ;
140 : const AliPHOSCpvRecPoint * CpvRecPoint(Int_t index) ;
141 : void MakeRecPointsArray();
142 : /**** T r a c k S e g m e n t s ****/
143 : TClonesArray * TrackSegments();
144 : const AliPHOSTrackSegment * TrackSegment(Int_t index);
145 : void MakeTrackSegmentsArray();
146 : /**** R e c P a r t ic l e s ****/
147 : TClonesArray * RecParticles() ;
148 : const AliPHOSRecParticle * RecParticle(Int_t index);
149 : void MakeRecParticlesArray();
150 :
151 0 : void SetDebug(Int_t level) {fDebug = level;} // Set debug level
152 : void SetBranchTitle(const TString& btitle);
153 :
154 0 : Int_t GetDebug() const {return fDebug; }
155 0 : TString GetBranchTitle() const {return fBranchTitle;}
156 :
157 : protected:
158 : TString fBranchTitle; //Title of the branch
159 :
160 : private:
161 :
162 : // assignement operator requested by coding convention, but not needed
163 : AliPHOSLoader(const AliPHOSLoader &); //Not implemented
164 : AliPHOSLoader & operator = (const AliPHOSLoader & ); //Not implemented
165 :
166 : Int_t ReadHits();
167 : Int_t ReadDigits();
168 : Int_t ReadSDigits();
169 : Int_t ReadRecPoints();
170 : Int_t ReadTracks();
171 : Int_t ReadRecParticles();
172 :
173 : Int_t fDebug ; // Debug level
174 : TClonesArray *fTmpHits; //! Temporary array of hits per track
175 :
176 : static const TString fgkHitsName;//Name for TClonesArray with hits from one event
177 : static const TString fgkSDigitsName;//Name for TClonesArray
178 : static const TString fgkDigitsName;//Name for TClonesArray
179 : static const TString fgkEmcRecPointsName;//Name for TClonesArray
180 : static const TString fgkCpvRecPointsName;//Name for TClonesArray
181 : static const TString fgkTracksName;//Name for TClonesArray
182 : static const TString fgkRecParticlesName;//Name for TClonesArray
183 :
184 : static const TString fgkEmcRecPointsBranchName;//Name for branch
185 : static const TString fgkCpvRecPointsBranchName;//Name for branch
186 : static const TString fgkTrackSegmentsBranchName;//Name for branch
187 : static const TString fgkRecParticlesBranchName;//Name for branch
188 :
189 :
190 360 : ClassDef(AliPHOSLoader,4) // Algorithm class that provides methods to retrieve objects from a list knowing the index
191 :
192 : };
193 :
194 : /******************************************************************************/
195 : /**************** I N L I N E S ****************************************/
196 : /******************************************************************************/
197 :
198 : inline TClonesArray* AliPHOSLoader::Hits()
199 : {
200 330 : return (TClonesArray*)GetDetectorData(fgkHitsName);
201 : }
202 : /******************************************************************************/
203 :
204 : inline const AliPHOSHit* AliPHOSLoader::Hit(Int_t index)
205 : {
206 0 : const TClonesArray* tcarr = Hits();
207 0 : if (tcarr)
208 0 : return (const AliPHOSHit*) tcarr->At(index);
209 0 : return 0x0;
210 0 : }
211 : /******************************************************************************/
212 :
213 : inline TClonesArray* AliPHOSLoader::SDigits()
214 : {
215 682 : return dynamic_cast<TClonesArray*>(GetDetectorData(fgkSDigitsName));
216 : }
217 : /******************************************************************************/
218 :
219 : inline const AliPHOSDigit* AliPHOSLoader::SDigit(Int_t index)
220 : {
221 0 : const TClonesArray* tcarr = SDigits();
222 0 : if (tcarr)
223 0 : return (const AliPHOSDigit*) tcarr->At(index);
224 0 : return 0x0;
225 0 : }
226 : /******************************************************************************/
227 :
228 : inline TClonesArray* AliPHOSLoader::Digits()
229 : {
230 547 : return dynamic_cast<TClonesArray*>(GetDetectorData(fgkDigitsName));
231 : }
232 : /******************************************************************************/
233 :
234 : inline const AliPHOSDigit* AliPHOSLoader::Digit(Int_t index)
235 : {
236 0 : const TClonesArray* tcarr = Digits();
237 0 : if (tcarr)
238 0 : return (const AliPHOSDigit*) tcarr->At(index);
239 0 : return 0x0;
240 0 : }
241 : /******************************************************************************/
242 :
243 : inline TObjArray * AliPHOSLoader::EmcRecPoints()
244 : {
245 354 : return dynamic_cast<TObjArray*>(GetDetectorData(fgkEmcRecPointsName));
246 : }
247 : /******************************************************************************/
248 :
249 : inline const AliPHOSEmcRecPoint * AliPHOSLoader::EmcRecPoint(Int_t index)
250 : {
251 0 : TObjArray* tcarr = EmcRecPoints();
252 0 : if (tcarr)
253 0 : return (const AliPHOSEmcRecPoint*) tcarr->At(index);
254 0 : return 0x0;
255 0 : }
256 : /******************************************************************************/
257 :
258 : inline TObjArray * AliPHOSLoader::CpvRecPoints()
259 : {
260 354 : return dynamic_cast<TObjArray*>(GetDetectorData(fgkCpvRecPointsName));
261 : }
262 : /******************************************************************************/
263 :
264 : inline const AliPHOSCpvRecPoint * AliPHOSLoader::CpvRecPoint(Int_t index)
265 : {
266 0 : TObjArray* tcarr = CpvRecPoints();
267 0 : if (tcarr)
268 0 : return (const AliPHOSCpvRecPoint*) tcarr->At(index);
269 0 : return 0x0;
270 0 : }
271 : /******************************************************************************/
272 :
273 : inline TClonesArray * AliPHOSLoader::TrackSegments()
274 : {
275 330 : return dynamic_cast<TClonesArray*>(GetDetectorData(fgkTracksName));
276 : }
277 : /******************************************************************************/
278 :
279 : inline const AliPHOSTrackSegment * AliPHOSLoader::TrackSegment(Int_t index)
280 : {
281 0 : const TClonesArray* tcarr = TrackSegments();
282 0 : if (tcarr)
283 0 : return (const AliPHOSTrackSegment*) tcarr->At(index);
284 0 : return 0x0;
285 0 : }
286 : /******************************************************************************/
287 :
288 : inline TClonesArray * AliPHOSLoader::RecParticles()
289 : {
290 375 : return dynamic_cast<TClonesArray*>(GetDetectorData(fgkRecParticlesName));
291 : }
292 : /******************************************************************************/
293 :
294 : inline const AliPHOSRecParticle* AliPHOSLoader::RecParticle(Int_t index)
295 : {
296 0 : TClonesArray* tcarr = RecParticles();
297 0 : if (tcarr)
298 0 : return (const AliPHOSRecParticle*) tcarr->At(index);
299 0 : return 0x0;
300 0 : }
301 :
302 : #endif // AliPHOSLOADER_H
|