]>
Commit | Line | Data |
---|---|---|
9edefa04 | 1 | /************************************************************************** |
2 | * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * | |
3 | * * | |
4 | * Author: The ALICE Off-line Project. * | |
5 | * Contributors are mentioned in the code where appropriate. * | |
6 | * * | |
7 | * Permission to use, copy, modify and distribute this software and its * | |
8 | * documentation strictly for non-commercial purposes is hereby granted * | |
9 | * without fee, provided that the above copyright notice appears in all * | |
10 | * copies and that both the copyright notice and this permission notice * | |
11 | * appear in the supporting documentation. The authors make no claims * | |
12 | * about the suitability of this software for any purpose. It is * | |
13 | * provided "as is" without express or implied warranty. * | |
14 | **************************************************************************/ | |
15 | ||
16 | /* $Id$ */ | |
17 | ||
7ca4655f | 18 | #include <TClonesArray.h> |
9edefa04 | 19 | #include <TClass.h> |
6f5308d9 | 20 | #include <TGeoManager.h> |
c93255fe | 21 | #include <TTree.h> |
9edefa04 | 22 | |
aacedc3e | 23 | #include "AliITSdigit.h" |
88cb7938 | 24 | #include "AliITSLoader.h" |
7d62fb64 | 25 | #include "AliRunLoader.h" |
ad7f2bfa | 26 | #include "AliObjectLoader.h" |
6def2bd2 | 27 | #include "AliITSInitGeometry.h" |
13033bec | 28 | #include "AliLog.h" |
88cb7938 | 29 | |
aacedc3e | 30 | /////////////////////////////////////////////////////////////////////////// |
eeac5f9f | 31 | // Loader for ITS |
32 | // it manages the I/O for: | |
33 | // raw clusters, primary vertices | |
34 | // V0 and cascade | |
35 | // and tracks propagated to the origin | |
88cb7938 | 36 | ClassImp(AliITSLoader) |
37 | ||
aacedc3e | 38 | /**********************************************************************/ |
e56160b8 | 39 | AliITSLoader::AliITSLoader():AliLoader(), |
e56160b8 | 40 | fGeom(0){ |
eeac5f9f | 41 | // Default constructor |
aacedc3e | 42 | } |
43 | /*********************************************************************/ | |
44 | AliITSLoader::AliITSLoader(const Char_t *name,const Char_t *topfoldername): | |
e56160b8 | 45 | AliLoader(name,topfoldername), |
e56160b8 | 46 | fGeom(0){ |
47 | //Constructor | |
aacedc3e | 48 | AliDataLoader* rawClustersDataLoader = new AliDataLoader( |
f1b15b8d | 49 | fDetectorName + ".RawCl.root",GetDefaultRawClustersContainerName(), |
aacedc3e | 50 | "Raw Clusters"); |
51 | fDataLoaders->Add(rawClustersDataLoader); | |
52 | rawClustersDataLoader->SetEventFolder(fEventFolder); | |
53 | rawClustersDataLoader->SetFolder(GetDetectorDataFolder()); | |
54 | ||
55 | AliDataLoader* backTracksDataLoader = new AliDataLoader( | |
f1b15b8d | 56 | fDetectorName + ".BackTracks.root",GetDefaultBackTracksContainerName(), |
aacedc3e | 57 | "Back Propagated Tracks"); |
58 | fDataLoaders->Add(backTracksDataLoader); | |
59 | backTracksDataLoader->SetEventFolder(fEventFolder); | |
60 | backTracksDataLoader->SetFolder(GetDetectorDataFolder()); | |
eeac5f9f | 61 | |
aacedc3e | 62 | AliDataLoader* vertexDataLoader = new AliDataLoader( |
f1b15b8d | 63 | fDetectorName + ".Vertex.root",GetDefaultVerticesContainerName(), |
aacedc3e | 64 | "Primary Vertices","O"); |
65 | fDataLoaders->Add(vertexDataLoader); | |
66 | vertexDataLoader->SetEventFolder(fEventFolder); | |
67 | vertexDataLoader->SetFolder(GetDetectorDataFolder()); | |
68 | ||
69 | AliDataLoader* v0DataLoader = new AliDataLoader( | |
f1b15b8d | 70 | fDetectorName + ".V0s.root",GetDefaultV0ContainerName(),"V0 Vertices"); |
aacedc3e | 71 | fDataLoaders->Add(v0DataLoader); |
72 | v0DataLoader->SetEventFolder(fEventFolder); | |
73 | v0DataLoader->SetFolder(GetDetectorDataFolder()); | |
74 | ||
75 | AliDataLoader* cascadeDataLoader = new AliDataLoader( | |
f1b15b8d | 76 | fDetectorName + ".Cascades.root",GetDefaultCascadeContainerName(), |
aacedc3e | 77 | "Cascades"); |
78 | fDataLoaders->Add(cascadeDataLoader); | |
79 | cascadeDataLoader->SetEventFolder(fEventFolder); | |
80 | cascadeDataLoader->SetFolder(GetDetectorDataFolder()); | |
eeac5f9f | 81 | } |
aacedc3e | 82 | /**********************************************************************/ |
83 | AliITSLoader::AliITSLoader(const Char_t *name,TFolder *topfolder): | |
e56160b8 | 84 | AliLoader(name,topfolder), |
e56160b8 | 85 | fGeom(0){ |
aacedc3e | 86 | //ctor |
87 | AliDataLoader* rawClustersDataLoader = new AliDataLoader( | |
f1b15b8d | 88 | fDetectorName + ".RawCl.root",GetDefaultRawClustersContainerName(), |
aacedc3e | 89 | "Raw Clusters"); |
90 | fDataLoaders->Add(rawClustersDataLoader); | |
91 | rawClustersDataLoader->SetEventFolder(fEventFolder); | |
92 | rawClustersDataLoader->SetFolder(GetDetectorDataFolder()); | |
eeac5f9f | 93 | |
aacedc3e | 94 | AliDataLoader* backTracksDataLoader = new AliDataLoader( |
f1b15b8d | 95 | fDetectorName + ".BackTracks.root",GetDefaultBackTracksContainerName(), |
aacedc3e | 96 | "Back Propagated Tracks"); |
97 | fDataLoaders->Add(backTracksDataLoader); | |
98 | backTracksDataLoader->SetEventFolder(fEventFolder); | |
99 | backTracksDataLoader->SetFolder(GetDetectorDataFolder()); | |
100 | ||
101 | AliDataLoader* vertexDataLoader = new AliDataLoader( | |
f1b15b8d | 102 | fDetectorName + ".Vertex.root",GetDefaultVerticesContainerName(), |
aacedc3e | 103 | "Primary Vertices","O"); |
104 | fDataLoaders->Add(vertexDataLoader); | |
105 | vertexDataLoader->SetEventFolder(fEventFolder); | |
106 | vertexDataLoader->SetFolder(GetDetectorDataFolder()); | |
107 | ||
108 | AliDataLoader* v0DataLoader = new AliDataLoader( | |
f1b15b8d | 109 | fDetectorName + ".V0.root",GetDefaultV0ContainerName(),"V0 Vertices"); |
aacedc3e | 110 | fDataLoaders->Add(v0DataLoader); |
111 | v0DataLoader->SetEventFolder(fEventFolder); | |
112 | v0DataLoader->SetFolder(GetDetectorDataFolder()); | |
113 | ||
114 | AliDataLoader* cascadeDataLoader = new AliDataLoader( | |
f1b15b8d | 115 | fDetectorName + ".Cascade.root",GetDefaultCascadeContainerName(), |
aacedc3e | 116 | "Cascades"); |
117 | fDataLoaders->Add(cascadeDataLoader); | |
118 | cascadeDataLoader->SetEventFolder(fEventFolder); | |
119 | cascadeDataLoader->SetFolder(GetDetectorDataFolder()); | |
88cb7938 | 120 | } |
e62c1aea | 121 | |
e62c1aea | 122 | |
aacedc3e | 123 | /**********************************************************************/ |
124 | AliITSLoader::~AliITSLoader(){ | |
125 | //destructor | |
126 | AliDataLoader* dl = 0; | |
127 | UnloadRawClusters(); | |
128 | dl = GetRawClLoader(); | |
129 | fDataLoaders->Remove(dl); | |
88cb7938 | 130 | |
aacedc3e | 131 | UnloadBackTracks(); |
132 | dl = GetBackTracksDataLoader(); | |
133 | fDataLoaders->Remove(dl); | |
134 | ||
135 | UnloadVertices(); | |
136 | dl = GetVertexDataLoader(); | |
137 | fDataLoaders->Remove(dl); | |
138 | ||
139 | UnloadV0s(); | |
140 | dl = GetV0DataLoader(); | |
141 | fDataLoaders->Remove(dl); | |
142 | ||
143 | UnloadCascades(); | |
144 | dl = GetCascadeDataLoader(); | |
145 | fDataLoaders->Remove(dl); | |
e56160b8 | 146 | |
8e50d897 | 147 | if(fGeom)delete fGeom; |
023ae34b | 148 | fGeom = 0; |
88cb7938 | 149 | } |
7d62fb64 | 150 | /* |
aacedc3e | 151 | //---------------------------------------------------------------------- |
152 | AliITS* AliITSLoader::GetITS(){ | |
153 | // Returns the pointer to the ITS, kept on the file. A short cut metthod | |
154 | // Inputs: | |
155 | // none. | |
156 | // Outputs: | |
157 | // none. | |
158 | // Returns: | |
159 | // Returns a pointer to the ITS, if not found returns 0. | |
160 | AliITS *its; | |
566abf75 | 161 | |
aacedc3e | 162 | if(gAlice){ |
163 | its = dynamic_cast<AliITS*> (gAlice->GetDetector( | |
164 | GetDetectorName().Data())); | |
165 | if(its) return its; | |
166 | } // end if gAlice | |
167 | AliRunLoader *rl=0; | |
168 | rl = GetRunLoader(); | |
169 | if(!rl) return 0; | |
170 | AliRun *ar=0; | |
171 | ar = rl->GetAliRun(); | |
172 | if(!ar) return 0; | |
173 | its = dynamic_cast<AliITS*> (ar->GetDetector(GetDetectorName().Data())); | |
174 | return its; | |
88cb7938 | 175 | } |
aacedc3e | 176 | //---------------------------------------------------------------------- |
177 | void AliITSLoader::SetupDigits(AliITS *its){ | |
178 | // Sets up to store ITS Digits in side AliITS::fDtype TObjArray | |
179 | // Inputs: | |
180 | // AliITS *its Pointer to the ITS | |
181 | // Outputs: | |
182 | // none. | |
183 | // Return: | |
184 | // none. | |
88cb7938 | 185 | |
aacedc3e | 186 | its->SetTreeAddressD(TreeD()); |
187 | } | |
7d62fb64 | 188 | */ |
aacedc3e | 189 | //---------------------------------------------------------------------- |
190 | void AliITSLoader::SetupDigits(TObjArray *digPerDet,Int_t n, | |
191 | const Char_t **digclass){ | |
192 | // Sets up digPerDet to store ITS Digits. | |
193 | // Inputs: | |
194 | // TObjArray *digPerDet A pointer to a TObject Array size>=3. | |
195 | // Int_t n The size of the TObjArray and digclass array | |
196 | // Char_t **digclass Array of digit class names | |
197 | // Outputs: | |
198 | // TObjArray *digPerDet Setup and linked to the tree of digits | |
199 | // Return: | |
200 | // none. | |
201 | Int_t i,m; | |
202 | TClonesArray *cl = 0; | |
203 | TTree *td = 0; | |
204 | TBranch *br = 0; | |
ce189d7c | 205 | Char_t branch[14]; |
aacedc3e | 206 | const Char_t *det[3] = {"SPD","SDD","SSD"}; |
207 | ||
208 | if(!digPerDet){ | |
209 | Error("SetUpDigits","TObject Array digPerDet does not exist"); | |
210 | return; | |
211 | } // end if | |
212 | m = digPerDet->GetSize(); | |
213 | if(m<n){ | |
214 | Error("SetUpDigits","TObject Array digPerDet=%p must have a size" | |
215 | " at least that of n=%d",digPerDet,n); | |
216 | } // end if | |
217 | if(m<3){ | |
218 | Error("SetUpDigits","TObject Array digPerDet=%p must have a size >2", | |
219 | digPerDet); | |
220 | return; | |
221 | } // end if | |
222 | td = TreeD(); | |
223 | for(i=0;i<n;i++){ | |
224 | if(digPerDet->At(i)==0){ // set up TClones Array | |
225 | digPerDet->AddAt(new TClonesArray(digclass[i],1000),i); | |
ce189d7c | 226 | if(n==3) snprintf(branch,13,"ITSDigits%s",det[i]); |
227 | else snprintf(branch,13,"ITSDigits%d",i+1); | |
aacedc3e | 228 | br = td->GetBranch(branch); |
229 | br->SetAddress(&((*digPerDet)[i])); | |
230 | continue; // do next one. | |
231 | } // end if | |
232 | cl = dynamic_cast<TClonesArray*> (digPerDet->At(i)); | |
233 | if(!cl && digPerDet->At(i)!=0){ // not a TClonesArray | |
234 | Error("SetUpDigits","TObject Array digPerDet-At(%d)=%p must be " | |
235 | "zeroed or filled with TClonesArrays",i,digPerDet); | |
236 | return; | |
237 | } // end if | |
238 | if(!(cl->GetClass()->GetBaseClass(AliITSdigit::Class()))){ | |
239 | Error("SetUPDigits","TClones array at digPerDet[%d}=%p must be" | |
240 | "derived from AliITSdigit",i,digPerDet->At(i)); | |
241 | } // end if | |
242 | cl->Clear(); | |
c0fc8108 | 243 | if(n==3) snprintf(branch,13,"ITSDigits%s",det[i]); |
244 | else snprintf(branch,13,"ITSDigits%d",i+1); | |
aacedc3e | 245 | br = td->GetBranch(branch); |
246 | br->SetAddress(&((*digPerDet)[i])); | |
247 | continue; | |
248 | } // end for i | |
249 | } | |
250 | //--------------------------------------------------------------------- | |
251 | AliITSdigit * AliITSLoader::GetDigit(TObjArray *digPerDet,Int_t module, | |
252 | Int_t digit){ | |
253 | // Gets the digit for for a specific detector type and module. | |
254 | // To be used in conjustion with Setupdigits(AliITS *its). | |
255 | // Inputs: | |
256 | // TObjArray *digPereDet Pointer to the Array of digits | |
257 | // Int_t module Module number | |
258 | // Int_t digit Digit number | |
259 | // Outputs: | |
260 | // none. | |
261 | // Return: | |
262 | // returns the pointer to the digit. if zero then last digit for that | |
263 | // module. | |
264 | ||
265 | if(digPerDet==0){ | |
266 | Error("GetDigit","digPerDet=%p, module=%d, digit=%d", | |
267 | digPerDet,module,digit); | |
268 | return 0; | |
269 | } // end if | |
270 | return 0; | |
271 | } | |
7d62fb64 | 272 | /* |
aacedc3e | 273 | //--------------------------------------------------------------------- |
274 | AliITSdigit * AliITSLoader::GetDigit(AliITS *its,Int_t module,Int_t digit){ | |
275 | // Gets the digit for for a specific detector type and module. | |
276 | // To be used in conjustion with Setupdigits(AliITS *its). | |
277 | // Inputs: | |
278 | // AliITS *its Pointer to the ITS | |
279 | // Int_t module Module number | |
280 | // Int_t digit Digit number | |
281 | // Outputs: | |
282 | // none. | |
283 | // Return: | |
284 | // returns the pointer to the digit. if zero then last digit for that | |
285 | // module. | |
286 | //AliITSDetType *idtype; | |
287 | AliITSgeom *geom = its->GetITSgeom(); | |
288 | Int_t idet = geom->GetModuleType(module); | |
289 | TClonesArray *digits; | |
290 | ||
291 | its->ResetDigits(); | |
292 | TreeD()->GetEvent(module); | |
293 | digits = its->DigitsAddress(idet); | |
294 | if(digit>-1 && digit<digits->GetEntriesFast()){ // if in range. | |
295 | return (AliITSdigit*) digits->At(digit); | |
296 | } // end if | |
297 | return 0; | |
298 | } | |
7d62fb64 | 299 | */ |
aacedc3e | 300 | //---------------------------------------------------------------------- |
88cb7938 | 301 | void AliITSLoader::MakeTree(Option_t *opt){ |
aacedc3e | 302 | // invokes AliLoader::MakeTree + specific ITS tree(s) |
303 | // Valid options: H,S,D,R,T and C (C=raw clusters) | |
304 | AliLoader::MakeTree(opt); | |
305 | const char *oC = strstr(opt,"C"); | |
306 | if (oC) MakeRawClustersContainer(); | |
307 | ||
308 | const char *oB = strstr(opt,"B"); | |
309 | if (oB) MakeBackTracksContainer(); | |
310 | ||
311 | const char *oV0 = strstr(opt,"V0"); | |
312 | if (oV0) MakeV0Container(); | |
313 | ||
314 | const char *oX = strstr(opt,"X"); | |
315 | if (oX) MakeCascadeContainer(); | |
88cb7938 | 316 | } |
7d62fb64 | 317 | |
318 | //---------------------------------------------------------------------- | |
8e50d897 | 319 | AliITSgeom* AliITSLoader::GetITSgeom(Bool_t force) { |
7d62fb64 | 320 | // retrieves the ITS geometry from file |
8e50d897 | 321 | if(fGeom && !force)return fGeom; |
322 | if(fGeom && force){ | |
323 | delete fGeom; | |
324 | fGeom = 0; | |
325 | } | |
6f5308d9 | 326 | if(!gGeoManager){ |
327 | AliError("gGeoManager is a null pointer - ITS geometry not built"); | |
328 | return fGeom; | |
329 | } | |
108bd0fe | 330 | AliITSInitGeometry initgeom; |
6def2bd2 | 331 | fGeom = initgeom.CreateAliITSgeom(); |
332 | AliDebug(1,"AliITSgeom object has been initialized from TGeo\n"); | |
15e6a32c | 333 | AliDebug(1,Form("Geometry name: %s",(initgeom.GetGeometryName()).Data())); |
8e50d897 | 334 | return fGeom; |
7d62fb64 | 335 | } |
023ae34b | 336 | //______________________________________________________________________ |
337 | void AliITSLoader::SetITSgeom(AliITSgeom *geom){ | |
338 | // Replaces the AliITSgeom object read from file with the one | |
339 | // given. | |
340 | // Inputs: | |
341 | // AliITSgeom *geom The AliITSgeom object to replace the one | |
342 | // read from the file | |
343 | // Outputs: | |
344 | // none. | |
345 | // Return: | |
346 | // none. | |
347 | ||
348 | if(fGeom==geom) return; // Same do nothing | |
349 | if(fGeom) { | |
350 | delete fGeom; | |
351 | fGeom=0; | |
352 | }// end if | |
353 | fGeom=geom; | |
354 | } | |
f1b15b8d | 355 | |
356 | const TString& AliITSLoader::GetDefaultRawClustersContainerName() { | |
357 | //default for Raw Clusters container name | |
358 | static const TString kDefaultRawClustersContainerName = "TreeC"; | |
359 | return kDefaultRawClustersContainerName; | |
360 | } | |
361 | ||
362 | const TString& AliITSLoader::GetDefaultBackTracksContainerName() { | |
363 | //default for Back propag. tracks container name | |
364 | static const TString kDefaultBackTracksContainerName = "TreeB"; | |
365 | return kDefaultBackTracksContainerName; | |
366 | } | |
367 | ||
368 | const TString& AliITSLoader::GetDefaultVerticesContainerName() { | |
369 | //default for primary vertices container name | |
370 | static const TString kDefaultVerticesContainerName = "Vertex"; | |
371 | return kDefaultVerticesContainerName; | |
372 | } | |
373 | ||
374 | const TString& AliITSLoader::GetDefaultV0ContainerName() { | |
375 | //default for V0 container name | |
376 | static const TString kDefaultV0ContainerName = "V0"; | |
377 | return kDefaultV0ContainerName; | |
378 | } | |
379 | ||
380 | const TString& AliITSLoader::GetDefaultCascadeContainerName() { | |
381 | //default fo cascade container name | |
382 | static const TString kDefaultCascadeContainerName = "Cascade"; | |
383 | return kDefaultCascadeContainerName; | |
384 | } |