]>
Commit | Line | Data |
---|---|---|
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 | /* | |
17 | $Id$ | |
18 | */ | |
19 | ||
20 | ||
21 | ///////////////////////////////////////////////////////////////////// | |
22 | // Base simulation functions for ITS // | |
23 | // // | |
24 | // // | |
25 | ///////////////////////////////////////////////////////////////////// | |
26 | #include "TBranch.h" | |
27 | #include "TClonesArray.h" | |
28 | #include "TObjArray.h" | |
29 | #include "TTree.h" | |
30 | ||
31 | #include "AliRun.h" | |
32 | ||
33 | #include "AliCDBManager.h" | |
34 | #include "AliCDBId.h" | |
35 | #include "AliCDBStorage.h" | |
36 | #include "AliCDBEntry.h" | |
37 | #include "AliCDBMetaData.h" | |
38 | ||
39 | #include "AliITSdigit.h" | |
40 | #include "AliITSdigitSPD.h" | |
41 | #include "AliITSdigitSDD.h" | |
42 | #include "AliITSdigitSSD.h" | |
43 | #include "AliITSDetTypeSim.h" | |
44 | #include "AliITSpListItem.h" | |
45 | #include "AliITSresponseSDD.h" | |
46 | #include "AliITSCalibrationSDD.h" | |
47 | #include "AliITSCalibrationSSD.h" | |
48 | #include "AliITSsegmentationSPD.h" | |
49 | #include "AliITSsegmentationSDD.h" | |
50 | #include "AliITSsegmentationSSD.h" | |
51 | #include "AliITSsimulation.h" | |
52 | #include "AliITSsimulationSPD.h" | |
53 | #include "AliITSsimulationSDD.h" | |
54 | #include "AliITSsimulationSSD.h" | |
55 | ||
56 | ||
57 | const Int_t AliITSDetTypeSim::fgkNdettypes = 3; | |
58 | const Int_t AliITSDetTypeSim::fgkDefaultNModulesSPD = 240; | |
59 | const Int_t AliITSDetTypeSim::fgkDefaultNModulesSDD = 260; | |
60 | const Int_t AliITSDetTypeSim::fgkDefaultNModulesSSD = 1698; | |
61 | ||
62 | ClassImp(AliITSDetTypeSim) | |
63 | ||
64 | //---------------------------------------------------------------------- | |
65 | AliITSDetTypeSim::AliITSDetTypeSim(): | |
66 | TObject(), | |
67 | fSimulation(), // [NDet] | |
68 | fSegmentation(), // [NDet] | |
69 | fCalibration(), // [NMod] | |
70 | fPreProcess(), // [] e.g. Fill fHitModule with hits | |
71 | fPostProcess(), // [] e.g. Wright Raw data | |
72 | fNSDigits(0), //! number of SDigits | |
73 | fSDigits(), //! [NMod][NSDigits] | |
74 | fNDigits(0), //! number of Digits | |
75 | fRunNumber(0), //! Run number (to access DB) | |
76 | fDigits(), //! [NMod][NDigits] | |
77 | fHitClassName(), // String with Hit class name. | |
78 | fSDigClassName(),// String with SDigit class name. | |
79 | fDigClassName(), // String with digit class name. | |
80 | fLoader(0), // local pointer to loader | |
81 | fFirstcall(kTRUE){ // flag | |
82 | // Default Constructor | |
83 | // Inputs: | |
84 | // none. | |
85 | // Outputs: | |
86 | // none. | |
87 | // Return: | |
88 | // A properly zero-ed AliITSDetTypeSim class. | |
89 | ||
90 | fSimulation = new TObjArray(fgkNdettypes); | |
91 | fSegmentation = new TObjArray(fgkNdettypes); | |
92 | fSegmentation->SetOwner(kTRUE); | |
93 | fSDigits = new TClonesArray("AliITSpListItem",1000); | |
94 | fDigits = new TObjArray(fgkNdettypes); | |
95 | fNDigits = new Int_t[fgkNdettypes]; | |
96 | fNMod[0] = fgkDefaultNModulesSPD; | |
97 | fNMod[1] = fgkDefaultNModulesSDD; | |
98 | fNMod[2] = fgkDefaultNModulesSSD; | |
99 | SetRunNumber(); | |
100 | } | |
101 | //---------------------------------------------------------------------- | |
102 | AliITSDetTypeSim::~AliITSDetTypeSim(){ | |
103 | // Destructor | |
104 | // Inputs: | |
105 | // none. | |
106 | // Outputs: | |
107 | // none. | |
108 | // Return: | |
109 | // Nothing. | |
110 | ||
111 | if(fSimulation){ | |
112 | fSimulation->Delete(); | |
113 | delete fSimulation; | |
114 | } | |
115 | fSimulation = 0; | |
116 | if(fSegmentation){ | |
117 | fSegmentation->Delete(); | |
118 | delete fSegmentation; | |
119 | } | |
120 | fSegmentation = 0; | |
121 | if(fCalibration && fRunNumber<0){ | |
122 | AliITSresponse* rspd = ((AliITSCalibration*)fCalibration->At( | |
123 | GetITSgeom()->GetStartSPD()))->GetResponse(); | |
124 | AliITSresponse* rsdd = ((AliITSCalibration*)fCalibration->At( | |
125 | GetITSgeom()->GetStartSDD()))->GetResponse(); | |
126 | AliITSresponse* rssd = ((AliITSCalibration*)fCalibration->At( | |
127 | GetITSgeom()->GetStartSSD()))->GetResponse(); | |
128 | if(rspd) delete rspd; | |
129 | if(rsdd) delete rsdd; | |
130 | if(rssd) delete rssd; | |
131 | fCalibration->Delete(); | |
132 | delete fCalibration; | |
133 | } | |
134 | fCalibration = 0; | |
135 | if(fPreProcess){ | |
136 | fPreProcess->Delete(); | |
137 | delete fPreProcess; | |
138 | } | |
139 | fPreProcess = 0; | |
140 | if(fPostProcess){ | |
141 | fPostProcess->Delete(); | |
142 | delete fPostProcess; | |
143 | } | |
144 | fPostProcess = 0; | |
145 | if(fNDigits) delete [] fNDigits; | |
146 | fNDigits = 0; | |
147 | if (fLoader)fLoader->GetModulesFolder()->Remove(this); | |
148 | fLoader = 0; // Not deleting it. | |
149 | if (fSDigits) { | |
150 | fSDigits->Delete(); | |
151 | delete fSDigits; | |
152 | } | |
153 | fSDigits=0; | |
154 | if (fDigits) { | |
155 | fDigits->Delete(); | |
156 | delete fDigits; | |
157 | } | |
158 | fDigits=0; | |
159 | } | |
160 | //---------------------------------------------------------------------- | |
161 | AliITSDetTypeSim::AliITSDetTypeSim(const AliITSDetTypeSim &source) : TObject(source), | |
162 | fSimulation(source.fSimulation), // [NDet] | |
163 | fSegmentation(source.fSegmentation), // [NDet] | |
164 | fCalibration(source.fCalibration), // [NMod] | |
165 | fPreProcess(source.fPreProcess), // [] e.g. Fill fHitModule with hits | |
166 | fPostProcess(source.fPostProcess), // [] e.g. Wright Raw data | |
167 | fNSDigits(source.fNSDigits), //! number of SDigits | |
168 | fSDigits(source.fSDigits), //! [NMod][NSDigits] | |
169 | fNDigits(source.fNDigits), //! number of Digits | |
170 | fRunNumber(source.fRunNumber), //! Run number (to access DB) | |
171 | fDigits(source.fDigits), //! [NMod][NDigits] | |
172 | fHitClassName(source.fHitClassName), // String with Hit class name. | |
173 | fSDigClassName(source.fSDigClassName),// String with SDigit class name. | |
174 | fDigClassName(), // String with digit class name. | |
175 | fLoader(source.fLoader), // local pointer to loader | |
176 | fFirstcall(source.fFirstcall) | |
177 | { | |
178 | // Copy Constructor for object AliITSDetTypeSim not allowed | |
179 | for(Int_t i=0;i<fgkNdettypes;i++){ | |
180 | fDigClassName[i] = source.fDigClassName[i]; | |
181 | } | |
182 | } | |
183 | //---------------------------------------------------------------------- | |
184 | AliITSDetTypeSim& AliITSDetTypeSim::operator=(const AliITSDetTypeSim &source){ | |
185 | // The = operator for object AliITSDetTypeSim | |
186 | ||
187 | this->~AliITSDetTypeSim(); | |
188 | new(this) AliITSDetTypeSim(source); | |
189 | return *this; | |
190 | } | |
191 | ||
192 | //______________________________________________________________________ | |
193 | void AliITSDetTypeSim::SetITSgeom(AliITSgeom *geom){ | |
194 | // Sets/replaces the existing AliITSgeom object kept in AliITSLoader | |
195 | // | |
196 | // Inputs: | |
197 | // AliITSgoem *geom The AliITSgeom object to be used. | |
198 | // Output: | |
199 | // none. | |
200 | // Return: | |
201 | // none. | |
202 | if(!fLoader){ | |
203 | Error("SetITSgeom","No pointer to loader - nothing done"); | |
204 | return; | |
205 | } | |
206 | else { | |
207 | fLoader->SetITSgeom(geom); // protections in AliITSLoader::SetITSgeom | |
208 | } | |
209 | ||
210 | } | |
211 | //______________________________________________________________________ | |
212 | void AliITSDetTypeSim::SetLoader(AliITSLoader *loader){ | |
213 | // Sets the local copy of the AliITSLoader, and passes on the | |
214 | // AliITSgeom object as needed. | |
215 | // Inputs | |
216 | // AliITSLoader *loader pointer to AliITSLoader for local use | |
217 | // Outputs: | |
218 | // none. | |
219 | // Return: | |
220 | // none. | |
221 | ||
222 | if(fLoader==loader) return; // Same do nothing | |
223 | if(fLoader){ // alread have an existing loader | |
224 | Error("SetLoader", | |
225 | "Already have an exisiting loader ptr=%p Nothing done", | |
226 | fLoader); | |
227 | } // end if | |
228 | fLoader = loader; | |
229 | } | |
230 | //______________________________________________________________________ | |
231 | void AliITSDetTypeSim::SetSimulationModel(Int_t dettype,AliITSsimulation *sim){ | |
232 | ||
233 | //Set simulation model for detector type | |
234 | ||
235 | if(fSimulation==0) fSimulation = new TObjArray(fgkNdettypes); | |
236 | fSimulation->AddAt(sim,dettype); | |
237 | } | |
238 | //______________________________________________________________________ | |
239 | AliITSsimulation* AliITSDetTypeSim::GetSimulationModel(Int_t dettype){ | |
240 | ||
241 | //Get simulation model for detector type | |
242 | if(fSimulation==0) { | |
243 | Warning("GetSimulationModel","fSimulation is 0!"); | |
244 | return 0; | |
245 | } | |
246 | return (AliITSsimulation*)(fSimulation->At(dettype)); | |
247 | } | |
248 | //______________________________________________________________________ | |
249 | AliITSsimulation* AliITSDetTypeSim::GetSimulationModelByModule(Int_t module){ | |
250 | ||
251 | //Get simulation model by module number | |
252 | if(GetITSgeom()==0) { | |
253 | Warning("GetSimulationModelByModule","GetITSgeom() is 0!"); | |
254 | return 0; | |
255 | } | |
256 | ||
257 | return GetSimulationModel(GetITSgeom()->GetModuleType(module)); | |
258 | } | |
259 | //_______________________________________________________________________ | |
260 | void AliITSDetTypeSim::SetDefaultSegmentation(Int_t idet){ | |
261 | // Set default segmentation model objects | |
262 | AliITSsegmentation *seg; | |
263 | ||
264 | if(fSegmentation==0x0){ | |
265 | fSegmentation = new TObjArray(fgkNdettypes); | |
266 | fSegmentation->SetOwner(kTRUE); | |
267 | } | |
268 | if(GetSegmentationModel(idet)) | |
269 | delete (AliITSsegmentation*)fSegmentation->At(idet); | |
270 | if(idet==0){ | |
271 | seg = new AliITSsegmentationSPD(GetITSgeom()); | |
272 | }else if(idet==1){ | |
273 | AliITSCalibration* res=GetCalibrationModel( | |
274 | GetITSgeom()->GetStartSDD()); | |
275 | seg = new AliITSsegmentationSDD(GetITSgeom(),res); | |
276 | }else { | |
277 | seg = new AliITSsegmentationSSD(GetITSgeom()); | |
278 | } | |
279 | SetSegmentationModel(idet,seg); | |
280 | } | |
281 | //______________________________________________________________________ | |
282 | void AliITSDetTypeSim::SetSegmentationModel(Int_t dettype, | |
283 | AliITSsegmentation *seg){ | |
284 | ||
285 | //Set segmentation model for detector type | |
286 | if(fSegmentation==0x0){ | |
287 | fSegmentation = new TObjArray(fgkNdettypes); | |
288 | fSegmentation->SetOwner(kTRUE); | |
289 | } | |
290 | fSegmentation->AddAt(seg,dettype); | |
291 | } | |
292 | //______________________________________________________________________ | |
293 | AliITSsegmentation* AliITSDetTypeSim::GetSegmentationModel(Int_t dettype){ | |
294 | //Get segmentation model for detector type | |
295 | ||
296 | if(fSegmentation==0) { | |
297 | Warning("GetSegmentationModel","fSegmentation is 0!"); | |
298 | return 0; | |
299 | } | |
300 | return (AliITSsegmentation*)(fSegmentation->At(dettype)); | |
301 | } | |
302 | //_______________________________________________________________________ | |
303 | AliITSsegmentation* AliITSDetTypeSim::GetSegmentationModelByModule(Int_t module){ | |
304 | //Get segmentation model by module number | |
305 | if(GetITSgeom()==0){ | |
306 | Warning("GetSegmentationModelByModule","GetITSgeom() is 0!"); | |
307 | return 0; | |
308 | } | |
309 | return GetSegmentationModel(GetITSgeom()->GetModuleType(module)); | |
310 | } | |
311 | //_______________________________________________________________________ | |
312 | void AliITSDetTypeSim::CreateCalibrationArray() { | |
313 | //Create the container of calibration functions with correct size | |
314 | if (fCalibration) { | |
315 | Warning("CreateCalibration","pointer to calibration object exists\n"); | |
316 | fCalibration->Delete(); | |
317 | delete fCalibration; | |
318 | } | |
319 | ||
320 | Int_t nModTot = GetITSgeom()->GetIndexMax(); | |
321 | fCalibration = new TObjArray(nModTot); | |
322 | fCalibration->SetOwner(kTRUE); | |
323 | fCalibration->Clear(); | |
324 | } | |
325 | //_______________________________________________________________________ | |
326 | void AliITSDetTypeSim::SetCalibrationModel(Int_t iMod, AliITSCalibration *resp){ | |
327 | //Set response model for modules | |
328 | ||
329 | if (fCalibration==0) CreateCalibrationArray(); | |
330 | ||
331 | if (fCalibration->At(iMod)!=0) | |
332 | delete (AliITSCalibration*) fCalibration->At(iMod); | |
333 | fCalibration->AddAt(resp, iMod); | |
334 | } | |
335 | //______________________________________________________________________ | |
336 | void AliITSDetTypeSim::ResetCalibrationArray(){ | |
337 | //resets response array | |
338 | if(fCalibration && fRunNumber<0){ // if fRunNumber<0 fCalibration is owner | |
339 | AliITSresponse* rspd = ((AliITSCalibration*)fCalibration->At( | |
340 | GetITSgeom()->GetStartSPD()))->GetResponse(); | |
341 | AliITSresponse* rsdd = ((AliITSCalibration*)fCalibration->At( | |
342 | GetITSgeom()->GetStartSDD()))->GetResponse(); | |
343 | AliITSresponse* rssd = ((AliITSCalibration*)fCalibration->At( | |
344 | GetITSgeom()->GetStartSSD()))->GetResponse(); | |
345 | if(rspd) delete rspd; | |
346 | if(rsdd) delete rsdd; | |
347 | if(rssd) delete rssd; | |
348 | fCalibration->Clear(); | |
349 | }else if (fCalibration && fRunNumber>=0){ | |
350 | fCalibration->Clear(); | |
351 | } | |
352 | } | |
353 | //______________________________________________________________________ | |
354 | void AliITSDetTypeSim::ResetSegmentation(){ | |
355 | //Resets segmentation array | |
356 | if(fSegmentation) fSegmentation->Clear(); | |
357 | } | |
358 | //_______________________________________________________________________ | |
359 | AliITSCalibration* AliITSDetTypeSim::GetCalibrationModel(Int_t iMod){ | |
360 | //Get response model for module number iMod | |
361 | ||
362 | if(fCalibration==0) { | |
363 | AliError("fCalibration is 0!"); | |
364 | return 0; | |
365 | } | |
366 | return (AliITSCalibration*)(fCalibration->At(iMod)); | |
367 | } | |
368 | //_______________________________________________________________________ | |
369 | void AliITSDetTypeSim::SetDefaults(){ | |
370 | //Set defaults for segmentation and response | |
371 | ||
372 | if(GetITSgeom()==0){ | |
373 | Warning("SetDefaults","GetITSgeom() is 0!"); | |
374 | return; | |
375 | } // end if | |
376 | if (fCalibration==0) { | |
377 | CreateCalibrationArray(); | |
378 | } // end if | |
379 | ||
380 | ResetSegmentation(); | |
381 | if(!GetCalibration()){AliFatal("Exit"); exit(0);} | |
382 | ||
383 | for(Int_t idet=0;idet<fgkNdettypes;idet++){ | |
384 | //SPD | |
385 | if(idet==0){ | |
386 | if(!GetSegmentationModel(idet)) SetDefaultSegmentation(idet); | |
387 | const char *kData0=(GetCalibrationModel(GetITSgeom()->GetStartSPD()))->DataType(); | |
388 | if (strstr(kData0,"real")) { | |
389 | SetDigitClassName(idet,"AliITSdigit"); | |
390 | }else { | |
391 | SetDigitClassName(idet,"AliITSdigitSPD"); | |
392 | } // end if | |
393 | } // end if idet==0 | |
394 | //SDD | |
395 | if(idet==1){ | |
396 | if(!GetSegmentationModel(idet)) SetDefaultSegmentation(idet); | |
397 | AliITSCalibrationSDD* rsp = | |
398 | (AliITSCalibrationSDD*)GetCalibrationModel( | |
399 | GetITSgeom()->GetStartSDD()); | |
400 | const char *kopt = ((AliITSresponseSDD*)rsp->GetResponse())-> | |
401 | ZeroSuppOption(); | |
402 | if((!strstr(kopt,"2D"))&&(!strstr(kopt,"1D"))) { | |
403 | SetDigitClassName(idet,"AliITSdigit"); | |
404 | }else { | |
405 | SetDigitClassName(idet,"AliITSdigitSDD"); | |
406 | } // end if | |
407 | } // end if idet==1 | |
408 | //SSD | |
409 | if(idet==2){ | |
410 | if(!GetSegmentationModel(idet))SetDefaultSegmentation(idet); | |
411 | const char *kData2 = (GetCalibrationModel( | |
412 | GetITSgeom()->GetStartSSD())->DataType()); | |
413 | if (strstr(kData2,"real")) { | |
414 | SetDigitClassName(idet,"AliITSdigit"); | |
415 | }else { | |
416 | SetDigitClassName(idet,"AliITSdigitSSD"); | |
417 | } // end if | |
418 | } // end if idet==2 | |
419 | }// end for idet | |
420 | } | |
421 | //______________________________________________________________________ | |
422 | Bool_t AliITSDetTypeSim::GetCalibration() { | |
423 | // Get Default calibration if a storage is not defined. | |
424 | ||
425 | if(!fFirstcall){ | |
426 | AliITSCalibration* cal = GetCalibrationModel(0); | |
427 | if(cal)return kTRUE; | |
428 | } | |
429 | else { | |
430 | fFirstcall = kFALSE; | |
431 | } | |
432 | ||
433 | SetRunNumber((Int_t)AliCDBManager::Instance()->GetRun()); | |
434 | Int_t run=GetRunNumber(); | |
435 | ||
436 | Bool_t origCacheStatus = AliCDBManager::Instance()->GetCacheFlag(); | |
437 | Bool_t isCacheActive = kTRUE; | |
438 | if(GetRunNumber()<0){ | |
439 | isCacheActive=kFALSE; | |
440 | fCalibration->SetOwner(kTRUE); | |
441 | } | |
442 | else{ | |
443 | fCalibration->SetOwner(kFALSE); | |
444 | } | |
445 | ||
446 | AliCDBManager::Instance()->SetCacheFlag(isCacheActive); | |
447 | ||
448 | AliCDBEntry *entrySPD = AliCDBManager::Instance()->Get("ITS/Calib/CalibSPD", run); | |
449 | AliCDBEntry *entrySDD = AliCDBManager::Instance()->Get("ITS/Calib/CalibSDD", run); | |
450 | AliCDBEntry *entrySSD = AliCDBManager::Instance()->Get("ITS/Calib/CalibSSD", run); | |
451 | AliCDBEntry *entry2SPD = AliCDBManager::Instance()->Get("ITS/Calib/RespSPD", run); | |
452 | AliCDBEntry *entry2SDD = AliCDBManager::Instance()->Get("ITS/Calib/RespSDD", run); | |
453 | AliCDBEntry *entry2SSD = AliCDBManager::Instance()->Get("ITS/Calib/RespSSD", run); | |
454 | ||
455 | if(!entrySPD || !entrySDD || !entrySSD || !entry2SPD || !entry2SDD || !entry2SSD){ | |
456 | AliFatal("Calibration object retrieval failed!"); | |
457 | return kFALSE; | |
458 | } | |
459 | ||
460 | if(!entrySPD || !entrySDD || !entrySSD || !entry2SPD || !entry2SDD || !entry2SSD){ | |
461 | AliError("Calibration data was not found in $ALICE_ROOT!"); | |
462 | return kFALSE; | |
463 | } | |
464 | ||
465 | TObjArray *calSPD = (TObjArray *)entrySPD->GetObject(); | |
466 | if(!isCacheActive)entrySPD->SetObject(NULL); | |
467 | entrySPD->SetOwner(kTRUE); | |
468 | ||
469 | AliITSresponseSPD *pSPD = (AliITSresponseSPD*)entry2SPD->GetObject(); | |
470 | if(!isCacheActive)entry2SPD->SetObject(NULL); | |
471 | entry2SPD->SetOwner(kTRUE); | |
472 | ||
473 | TObjArray *calSDD = (TObjArray *)entrySDD->GetObject(); | |
474 | if(!isCacheActive)entrySDD->SetObject(NULL); | |
475 | entrySDD->SetOwner(kTRUE); | |
476 | ||
477 | AliITSresponseSDD *pSDD = (AliITSresponseSDD*)entry2SDD->GetObject(); | |
478 | if(!isCacheActive)entry2SDD->SetObject(NULL); | |
479 | entry2SDD->SetOwner(kTRUE); | |
480 | ||
481 | TObjArray *calSSD = (TObjArray *)entrySSD->GetObject(); | |
482 | if(!isCacheActive)entrySSD->SetObject(NULL); | |
483 | entrySSD->SetOwner(kTRUE); | |
484 | ||
485 | AliITSresponseSSD *pSSD = (AliITSresponseSSD*)entry2SSD->GetObject(); | |
486 | if(!isCacheActive)entry2SSD->SetObject(NULL); | |
487 | entry2SSD->SetOwner(kTRUE); | |
488 | ||
489 | // DB entries are deleted. In this way metadeta objects are deleted as well | |
490 | if(!isCacheActive){ | |
491 | delete entrySPD; | |
492 | delete entrySDD; | |
493 | delete entrySSD; | |
494 | delete entry2SPD; | |
495 | delete entry2SDD; | |
496 | delete entry2SSD; | |
497 | } | |
498 | ||
499 | AliCDBManager::Instance()->SetCacheFlag(origCacheStatus); | |
500 | ||
501 | if ((!pSPD)||(!pSDD)||(!pSSD)) { | |
502 | AliWarning("Can not get calibration from calibration database !"); | |
503 | return kFALSE; | |
504 | } | |
505 | if ((! calSPD)||(! calSDD)||(! calSSD)) { | |
506 | AliWarning("Can not get calibration from calibration database !"); | |
507 | return kFALSE; | |
508 | } | |
509 | fNMod[0] = calSPD->GetEntries(); | |
510 | fNMod[1] = calSDD->GetEntries(); | |
511 | fNMod[2] = calSSD->GetEntries(); | |
512 | AliInfo(Form("%i SPD, %i SDD and %i SSD in calibration database", | |
513 | fNMod[0], fNMod[1], fNMod[2])); | |
514 | AliITSCalibration* cal; | |
515 | for (Int_t i=0; i<fNMod[0]; i++) { | |
516 | cal = (AliITSCalibration*) calSPD->At(i); | |
517 | cal->SetResponse(pSPD); | |
518 | SetCalibrationModel(i, cal); | |
519 | } | |
520 | for (Int_t i=0; i<fNMod[1]; i++) { | |
521 | cal = (AliITSCalibration*) calSDD->At(i); | |
522 | cal->SetResponse(pSDD); | |
523 | Int_t iMod = i + fNMod[0]; | |
524 | SetCalibrationModel(iMod, cal); | |
525 | } | |
526 | for (Int_t i=0; i<fNMod[2]; i++) { | |
527 | cal = (AliITSCalibration*) calSSD->At(i); | |
528 | cal->SetResponse(pSSD); | |
529 | Int_t iMod = i + fNMod[0] + fNMod[1]; | |
530 | SetCalibrationModel(iMod, cal); | |
531 | } | |
532 | return kTRUE; | |
533 | } | |
534 | //_______________________________________________________________________ | |
535 | void AliITSDetTypeSim::SetDefaultSimulation(){ | |
536 | //Set default simulation for detector type | |
537 | ||
538 | if(GetITSgeom()==0){ | |
539 | Warning("SetDefaultSimulation","GetITSgeom() is 0!"); | |
540 | return; | |
541 | } | |
542 | if(fCalibration==0){ | |
543 | Warning("SetDefaultSimulation","fCalibration is 0!"); | |
544 | return; | |
545 | } | |
546 | if(fSegmentation==0){ | |
547 | Warning("SetDefaultSimulation","fSegmentation is 0!"); | |
548 | for(Int_t i=0;i<fgkNdettypes;i++) SetDefaultSegmentation(i); | |
549 | }else for(Int_t i=0;i<fgkNdettypes;i++) if(!GetSegmentationModel(i)){ | |
550 | Warning("SetDefaultSimulation", | |
551 | "Segmentation not defined for det %d - Default taken\n!",i); | |
552 | SetDefaultSegmentation(i); | |
553 | } | |
554 | AliITSsimulation* sim; | |
555 | ||
556 | for(Int_t idet=0;idet<fgkNdettypes;idet++){ | |
557 | //SPD | |
558 | if(idet==0){ | |
559 | sim = GetSimulationModel(idet); | |
560 | if(!sim){ | |
561 | sim = new AliITSsimulationSPD(this); | |
562 | SetSimulationModel(idet,sim); | |
563 | } | |
564 | } | |
565 | //SDD | |
566 | if(idet==1){ | |
567 | sim = GetSimulationModel(idet); | |
568 | if(!sim){ | |
569 | sim = new AliITSsimulationSDD(this); | |
570 | SetSimulationModel(idet,sim); | |
571 | } | |
572 | } | |
573 | //SSD | |
574 | if(idet==2){ | |
575 | sim = GetSimulationModel(idet); | |
576 | if(!sim){ | |
577 | sim = new AliITSsimulationSSD(this); | |
578 | SetSimulationModel(idet,sim); | |
579 | } | |
580 | } | |
581 | ||
582 | } | |
583 | } | |
584 | //___________________________________________________________________ | |
585 | void AliITSDetTypeSim::SetTreeAddressS(TTree* treeS, Char_t* name){ | |
586 | // Set branch address for the ITS summable digits Trees. | |
587 | char branchname[30]; | |
588 | ||
589 | if(!treeS){ | |
590 | return; | |
591 | } | |
592 | if (fSDigits == 0x0){ | |
593 | fSDigits = new TClonesArray("AliITSpListItem",1000); | |
594 | } | |
595 | TBranch *branch; | |
596 | sprintf(branchname,"%s",name); | |
597 | branch = treeS->GetBranch(branchname); | |
598 | if (branch) branch->SetAddress(&fSDigits); | |
599 | ||
600 | } | |
601 | //___________________________________________________________________ | |
602 | void AliITSDetTypeSim::SetTreeAddressD(TTree* treeD, Char_t* name){ | |
603 | // Set branch address for the digit Trees. | |
604 | ||
605 | const char *det[3] = {"SPD","SDD","SSD"}; | |
606 | TBranch *branch; | |
607 | ||
608 | char branchname[30]; | |
609 | ||
610 | if(!treeD){ | |
611 | return; | |
612 | } | |
613 | if(!fDigits){ | |
614 | fDigits = new TObjArray(fgkNdettypes); | |
615 | } | |
616 | for(Int_t i=0;i<fgkNdettypes;i++){ | |
617 | Char_t* digclass = GetDigitClassName(i); | |
618 | if(digclass==0x0){ | |
619 | if(i==0) SetDigitClassName(i,"AliITSdigitSPD"); | |
620 | if(i==1) SetDigitClassName(i,"AliITSdigitSDD"); | |
621 | if(i==2) SetDigitClassName(i,"AliITSdigitSSD"); | |
622 | digclass = GetDigitClassName(i); | |
623 | } | |
624 | TString classn = digclass; | |
625 | if(!(fDigits->At(i))){ | |
626 | fDigits->AddAt(new TClonesArray(classn.Data(),1000),i); | |
627 | }else{ | |
628 | ResetDigits(i); | |
629 | } | |
630 | ||
631 | if(fgkNdettypes==3) sprintf(branchname,"%sDigits%s",name,det[i]); | |
632 | else sprintf(branchname,"%sDigits%d",name,i+1); | |
633 | if(fDigits){ | |
634 | branch = treeD->GetBranch(branchname); | |
635 | if(branch) branch->SetAddress(&((*fDigits)[i])); | |
636 | } | |
637 | } | |
638 | ||
639 | } | |
640 | //___________________________________________________________________ | |
641 | void AliITSDetTypeSim::ResetDigits(){ | |
642 | // Reset number of digits and the digits array for the ITS detector. | |
643 | ||
644 | if(!fDigits){ | |
645 | Error("ResetDigits","fDigits is null!"); | |
646 | return; | |
647 | } | |
648 | for(Int_t i=0;i<fgkNdettypes;i++){ | |
649 | ResetDigits(i); | |
650 | } | |
651 | } | |
652 | //___________________________________________________________________ | |
653 | void AliITSDetTypeSim::ResetDigits(Int_t branch){ | |
654 | // Reset number of digits and the digits array for this branch. | |
655 | ||
656 | if(fDigits->At(branch)){ | |
657 | ((TClonesArray*)fDigits->At(branch))->Clear(); | |
658 | } | |
659 | if(fNDigits) fNDigits[branch]=0; | |
660 | ||
661 | } | |
662 | ||
663 | ||
664 | ||
665 | //_______________________________________________________________________ | |
666 | void AliITSDetTypeSim::SDigitsToDigits(Option_t* opt, Char_t* name){ | |
667 | // Standard Summable digits to Digits function. | |
668 | if(!GetITSgeom()){ | |
669 | Warning("SDigitsToDigits","GetITSgeom() is null!!"); | |
670 | return; | |
671 | } | |
672 | ||
673 | const char *all = strstr(opt,"All"); | |
674 | const char *det[3] = {strstr(opt,"SPD"),strstr(opt,"SDD"), | |
675 | strstr(opt,"SSD")}; | |
676 | if( !det[0] && !det[1] && !det[2] ) all = "All"; | |
677 | else all = 0; | |
678 | static Bool_t setDef = kTRUE; | |
679 | if(setDef) SetDefaultSimulation(); | |
680 | setDef = kFALSE; | |
681 | ||
682 | AliITSsimulation *sim =0; | |
683 | TTree* trees = fLoader->TreeS(); | |
684 | if( !(trees && GetSDigits()) ){ | |
685 | Error("SDigits2Digits","Error: No trees or SDigits. Returning."); | |
686 | return; | |
687 | } | |
688 | sprintf(name,"%s",name); | |
689 | TBranch* brchSDigits = trees->GetBranch(name); | |
690 | ||
691 | Int_t id; | |
692 | for(Int_t module=0;module<GetITSgeom()->GetIndexMax();module++){ | |
693 | id = GetITSgeom()->GetModuleType(module); | |
694 | if (!all && !det[id]) continue; | |
695 | sim = (AliITSsimulation*)GetSimulationModel(id); | |
696 | if(!sim){ | |
697 | Error("SDigit2Digits","The simulation class was not " | |
698 | "instanciated for module %d type %s!",module, | |
699 | GetITSgeom()->GetModuleTypeName(module)); | |
700 | exit(1); | |
701 | } | |
702 | sim->InitSimulationModule(module,gAlice->GetEvNumber()); | |
703 | ||
704 | fSDigits->Clear(); | |
705 | brchSDigits->GetEvent(module); | |
706 | sim->AddSDigitsToModule(fSDigits,0); | |
707 | sim->FinishSDigitiseModule(); | |
708 | fLoader->TreeD()->Fill(); | |
709 | ResetDigits(); | |
710 | } | |
711 | fLoader->TreeD()->GetEntries(); | |
712 | fLoader->TreeD()->AutoSave(); | |
713 | fLoader->TreeD()->Reset(); | |
714 | } | |
715 | //_________________________________________________________ | |
716 | void AliITSDetTypeSim::AddSumDigit(AliITSpListItem &sdig){ | |
717 | //Adds the module full of summable digits to the summable digits tree. | |
718 | ||
719 | TClonesArray &lsdig = *fSDigits; | |
720 | new(lsdig[fNSDigits++]) AliITSpListItem(sdig); | |
721 | } | |
722 | //__________________________________________________________ | |
723 | void AliITSDetTypeSim::AddRealDigit(Int_t branch, Int_t *digits){ | |
724 | // Add a real digit - as coming from data. | |
725 | ||
726 | TClonesArray &ldigits = *((TClonesArray*)fDigits->At(branch)); | |
727 | new(ldigits[fNDigits[branch]++]) AliITSdigit(digits); | |
728 | } | |
729 | //__________________________________________________________ | |
730 | void AliITSDetTypeSim::AddSimDigit(Int_t branch, AliITSdigit* d){ | |
731 | // Add a simulated digit. | |
732 | ||
733 | TClonesArray &ldigits = *((TClonesArray*)fDigits->At(branch)); | |
734 | switch(branch){ | |
735 | case 0: | |
736 | new(ldigits[fNDigits[branch]++]) AliITSdigitSPD(*((AliITSdigitSPD*)d)); | |
737 | break; | |
738 | case 1: | |
739 | new(ldigits[fNDigits[branch]++]) AliITSdigitSDD(*((AliITSdigitSDD*)d)); | |
740 | break; | |
741 | case 2: | |
742 | new(ldigits[fNDigits[branch]++]) AliITSdigitSSD(*((AliITSdigitSSD*)d)); | |
743 | break; | |
744 | } | |
745 | } | |
746 | //______________________________________________________________________ | |
747 | void AliITSDetTypeSim::AddSimDigit(Int_t branch,Float_t phys,Int_t *digits, | |
748 | Int_t *tracks,Int_t *hits,Float_t *charges){ | |
749 | // Add a simulated digit to the list. | |
750 | ||
751 | TClonesArray &ldigits = *((TClonesArray*)fDigits->At(branch)); | |
752 | AliITSCalibrationSDD *resp = 0; | |
753 | switch(branch){ | |
754 | case 0: | |
755 | new(ldigits[fNDigits[branch]++]) AliITSdigitSPD(digits,tracks,hits); | |
756 | break; | |
757 | case 1: | |
758 | resp = (AliITSCalibrationSDD*)GetCalibrationModel( | |
759 | GetITSgeom()->GetStartSDD()); | |
760 | new(ldigits[fNDigits[branch]++]) AliITSdigitSDD(phys,digits,tracks, | |
761 | hits,charges,resp); | |
762 | break; | |
763 | case 2: | |
764 | new(ldigits[fNDigits[branch]++]) AliITSdigitSSD(digits,tracks,hits); | |
765 | break; | |
766 | } | |
767 | } | |
768 | //______________________________________________________________________ | |
769 | void AliITSDetTypeSim::StoreCalibration(Int_t firstRun, Int_t lastRun, | |
770 | AliCDBMetaData &md) { | |
771 | // Store calibration in the calibration database | |
772 | // The database must be created in an external piece of code (i.e. | |
773 | // a configuration macro ) | |
774 | ||
775 | if(!AliCDBManager::Instance()->IsDefaultStorageSet()) { | |
776 | AliWarning("No storage set! Will use dummy one"); | |
777 | AliCDBManager::Instance()->SetDefaultStorage("local://$ALICE_ROOT"); | |
778 | } | |
779 | ||
780 | if (!fCalibration) { | |
781 | AliError("AliITSCalibration classes are not defined - nothing done"); | |
782 | return; | |
783 | } | |
784 | AliCDBId idRespSPD("ITS/Calib/CalibSPD",firstRun, lastRun); | |
785 | AliCDBId idRespSDD("ITS/Calib/CalibSDD",firstRun, lastRun); | |
786 | AliCDBId idRespSSD("ITS/Calib/CalibSSD",firstRun, lastRun); | |
787 | ||
788 | TObjArray respSPD(fNMod[0]); | |
789 | TObjArray respSDD(fNMod[1]-fNMod[0]); | |
790 | TObjArray respSSD(fNMod[2]-fNMod[1]); | |
791 | respSPD.SetOwner(kFALSE); | |
792 | respSSD.SetOwner(kFALSE); | |
793 | respSSD.SetOwner(kFALSE); | |
794 | ||
795 | Int_t index[fgkNdettypes]; | |
796 | for (Int_t i = 0; i<fgkNdettypes; i++ ) { | |
797 | index[i] = 0; | |
798 | for (Int_t j = 0; j<=i; j++ ) | |
799 | index[i]+=fNMod[j]; | |
800 | } | |
801 | ||
802 | for (Int_t i = 0; i<index[0]; i++ ) | |
803 | respSPD.Add(fCalibration->At(i)); | |
804 | ||
805 | for (Int_t i = index[0]; i<index[1]; i++ ) | |
806 | respSDD.Add(fCalibration->At(i)); | |
807 | ||
808 | for (Int_t i = index[1]; i<index[2]; i++ ) | |
809 | respSSD.Add(fCalibration->At(i)); | |
810 | ||
811 | AliCDBManager::Instance()->Put(&respSPD, idRespSPD, &md); | |
812 | AliCDBManager::Instance()->Put(&respSDD, idRespSDD, &md); | |
813 | AliCDBManager::Instance()->Put(&respSSD, idRespSSD, &md); | |
814 | } | |
815 | ||
816 |