feature fixed (Christian)
[u/mrichter/AliRoot.git] / ITS / AliITSDetTypeRec.cxx
CommitLineData
3b9df642 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
7d62fb64 20////////////////////////////////////////////////////////////////////////
21// This class defines the "Standard" reconstruction for the ITS //
22// detector. //
23// //
24////////////////////////////////////////////////////////////////////////
25#include "TObjArray.h"
26#include "TTree.h"
27
fcf95fc7 28#include "AliITSresponseSDD.h"
8ba39da9 29#include "AliCDBManager.h"
30#include "AliCDBStorage.h"
31#include "AliCDBEntry.h"
32
7d62fb64 33#include "AliITSClusterFinder.h"
34#include "AliITSClusterFinderV2.h"
35#include "AliITSClusterFinderV2SPD.h"
36#include "AliITSClusterFinderV2SDD.h"
37#include "AliITSClusterFinderV2SSD.h"
38#include "AliITSClusterFinderSPD.h"
39#include "AliITSClusterFinderSDD.h"
40#include "AliITSClusterFinderSSD.h"
41#include "AliITSclusterV2.h"
42#include "AliITSgeom.h"
3b9df642 43#include "AliITSDetTypeRec.h"
7d62fb64 44#include "AliITSRawCluster.h"
45#include "AliITSRawClusterSPD.h"
46#include "AliITSRawClusterSDD.h"
47#include "AliITSRawClusterSSD.h"
48#include "AliITSRecPoint.h"
fcf95fc7 49#include "AliITSCalibrationSDD.h"
7d62fb64 50#include "AliITSsegmentationSPD.h"
51#include "AliITSsegmentationSDD.h"
52#include "AliITSsegmentationSSD.h"
53
3b9df642 54
7d62fb64 55const Int_t AliITSDetTypeRec::fgkNdettypes = 3;
8ba39da9 56const Int_t AliITSDetTypeRec::fgkDefaultNModulesSPD = 240;
57const Int_t AliITSDetTypeRec::fgkDefaultNModulesSDD = 260;
58const Int_t AliITSDetTypeRec::fgkDefaultNModulesSSD = 1698;
59
3b9df642 60ClassImp(AliITSDetTypeRec)
61
7d62fb64 62//________________________________________________________________
3b9df642 63AliITSDetTypeRec::AliITSDetTypeRec(): TObject(),
64fGeom(), //
65fReconstruction(),// [NDet]
66fSegmentation(), // [NDet]
67fCalibration(), // [NMod]
68fPreProcess(), // [] e.g. Find Calibration values
69fPostProcess(), // [] e.g. find primary vertex
3b9df642 70fDigits(), //! [NMod][NDigits]
71fClusterClassName(), // String with Cluster class name
72fDigClassName(), // String with digit class name.
73fRecPointClassName(){// String with RecPoint class name
74 // Default Constructor
75 // Inputs:
76 // none.
77 // Outputs:
78 // none.
79 // Return:
80 // A properly zero-ed AliITSDetTypeRec class.
7d62fb64 81
82 fGeom = 0;
83 fReconstruction = new TObjArray(fgkNdettypes);
84 fSegmentation = 0;
85 fCalibration = 0;
86 fPreProcess = 0;
87 fPostProcess = 0;
88 fDigits = new TObjArray(fgkNdettypes);
89 fNdtype = new Int_t[fgkNdettypes];
90 fCtype = new TObjArray(fgkNdettypes);
91 fNctype = new Int_t[fgkNdettypes];
92 fRecPoints = new TClonesArray("AliITSRecPoint",1000);
93 fNRecPoints = 0;
94 fClustersV2 = new TClonesArray("AliITSclusterV2",3000);
95 fNClustersV2 = 0;
96
97 for(Int_t i=0;i<fgkNdettypes;i++){
98 fNdtype[i]=0;
99 fNctype[i]=0;
100 }
101
102 SelectVertexer(" ");
103 fLoader = 0;
8ba39da9 104
105 fNMod[0] = fgkDefaultNModulesSPD;
106 fNMod[1] = fgkDefaultNModulesSDD;
107 fNMod[2] = fgkDefaultNModulesSSD;
108
109 SetRunNumber();
7d62fb64 110}
111//______________________________________________________________________
112AliITSDetTypeRec::AliITSDetTypeRec(const AliITSDetTypeRec &/*rec*/):TObject(/*rec*/){
113 // Copy constructor.
114
115 Error("Copy constructor","Copy constructor not allowed");
116
117}
118//______________________________________________________________________
119AliITSDetTypeRec& AliITSDetTypeRec::operator=(const AliITSDetTypeRec& /*source*/){
120 // Assignment operator. This is a function which is not allowed to be
121 // done.
122 Error("operator=","Assignment operator not allowed\n");
123 return *this;
3b9df642 124}
7d62fb64 125
126//_____________________________________________________________________
3b9df642 127AliITSDetTypeRec::~AliITSDetTypeRec(){
7d62fb64 128
129 //Destructor
fcf95fc7 130
131 if(fReconstruction){
132 fReconstruction->Delete();
133 delete fReconstruction;
134 fReconstruction = 0;
135 }
136 if(fSegmentation){
137 fSegmentation->Delete();
138 delete fSegmentation;
139 fSegmentation = 0;
140 }
141 if(fCalibration){
142 AliITSresponse* rspd = ((AliITSCalibration*)fCalibration->At(fGeom->GetStartSPD()))->GetResponse();
143 AliITSresponse* rsdd = ((AliITSCalibration*)fCalibration->At(fGeom->GetStartSDD()))->GetResponse();
144 AliITSresponse* rssd = ((AliITSCalibration*)fCalibration->At(fGeom->GetStartSSD()))->GetResponse();
145 if(rspd) delete rspd;
146 if(rsdd) delete rsdd;
147 if(rssd) delete rssd;
148 fCalibration->Delete();
149 delete fCalibration;
150 fCalibration = 0;
151 }
7d62fb64 152 if(fGeom) delete fGeom;
7d62fb64 153 if(fPreProcess) delete fPreProcess;
154 if(fPostProcess) delete fPostProcess;
155
156 if(fDigits){
157 fDigits->Delete();
158 delete fDigits;
159 fDigits=0;
160 }
161 if(fRecPoints){
162 fRecPoints->Delete();
163 delete fRecPoints;
164 fRecPoints=0;
165 }
166 if(fClustersV2){
167 fClustersV2->Delete();
168 delete fClustersV2;
169 fClustersV2=0;
170 }
171 if(fCtype) {
172 fCtype->Delete();
173 delete fCtype;
174 fCtype = 0;
175 }
176 delete [] fNctype;
8ba39da9 177 delete [] fNdtype;
7d62fb64 178 if(fLoader) delete fLoader;
179
180}
181//___________________________________________________________________
182void AliITSDetTypeRec::SetReconstructionModel(Int_t dettype,AliITSClusterFinder *clf){
183
184 //Set reconstruction model for detector type
185
186 if(fReconstruction==0) fReconstruction = new TObjArray(fgkNdettypes);
187 if(fReconstruction->At(dettype)!=0) delete fReconstruction->At(dettype);
188 fReconstruction->AddAt(clf,dettype);
189}
190//______________________________________________________________________
191AliITSClusterFinder* AliITSDetTypeRec::GetReconstructionModel(Int_t dettype){
192
193 //Get reconstruction model for detector type
194 if(fReconstruction==0) {
195 Warning("GetReconstructionModel","fReconstruction is 0!");
196 return 0;
197 }
198 return (AliITSClusterFinder*)fReconstruction->At(dettype);
199}
200
201//______________________________________________________________________
202void AliITSDetTypeRec::SetSegmentationModel(Int_t dettype,AliITSsegmentation *seg){
203
204 //Set segmentation model for detector type
205
206 if(fSegmentation==0) fSegmentation = new TObjArray(fgkNdettypes);
207 if(fSegmentation->At(dettype)!=0) delete fSegmentation->At(dettype);
208 fSegmentation->AddAt(seg,dettype);
209
210}
211//______________________________________________________________________
212AliITSsegmentation* AliITSDetTypeRec::GetSegmentationModel(Int_t dettype){
213
214 //Get segmentation model for detector type
215
216 if(fSegmentation==0) {
217 Warning("GetSegmentationModel","fSegmentation is 0!");
218 return 0;
219 }
220 return (AliITSsegmentation*)fSegmentation->At(dettype);
221
222}
223//_______________________________________________________________________
fcf95fc7 224void AliITSDetTypeRec::SetCalibrationModel(Int_t iMod, AliITSCalibration *cal){
8ba39da9 225
226 //Set calibration (response) for the module iMod of type dettype
227 if (fCalibration==0) {
228 fCalibration = new TObjArray(fGeom->GetIndexMax());
229 fCalibration->SetOwner(kTRUE);
230 fCalibration->Clear();
231 }
232
233 if (fCalibration->At(iMod) != 0)
fcf95fc7 234 delete (AliITSCalibration*) fCalibration->At(iMod);
8ba39da9 235 fCalibration->AddAt(cal,iMod);
7d62fb64 236
7d62fb64 237}
238//_______________________________________________________________________
fcf95fc7 239AliITSCalibration* AliITSDetTypeRec::GetCalibrationModel(Int_t iMod){
7d62fb64 240
8ba39da9 241 //Get calibration model for module type
7d62fb64 242
243 if(fCalibration==0) {
8ba39da9 244 Warning("GetalibrationModel","fCalibration is 0!");
7d62fb64 245 return 0;
246 }
8ba39da9 247
fcf95fc7 248 return (AliITSCalibration*)fCalibration->At(iMod);
7d62fb64 249}
250
251//______________________________________________________________________
252void AliITSDetTypeRec::SetTreeAddress(){
253 // Set branch address for the Trees.
254
255 TTree *treeR = fLoader->TreeR();
256 TTree *treeD = fLoader->TreeD();
257
258 SetTreeAddressD(treeD);
259 SetTreeAddressR(treeR);
260}
261//______________________________________________________________________
262void AliITSDetTypeRec::SetTreeAddressD(TTree *treeD){
263 // Set branch address for the tree of digits.
264
265 const char *det[4] = {"SPD","SDD","SSD","ITS"};
266 TBranch *branch;
267 Char_t* digclass;
268 Int_t i;
269 char branchname[30];
270
271 if(!treeD) return;
272 if (fDigits == 0x0) fDigits = new TObjArray(fgkNdettypes);
273 for (i=0; i<fgkNdettypes; i++) {
274 digclass = GetDigitClassName(i);
275 if(!(fDigits->At(i))) {
276 fDigits->AddAt(new TClonesArray(digclass,1000),i);
277 }else{
278 ResetDigits(i);
279 }
280 if (fgkNdettypes==3) sprintf(branchname,"%sDigits%s",det[3],det[i]);
281 else sprintf(branchname,"%sDigits%d",det[3],i+1);
282 if (fDigits) {
283 branch = treeD->GetBranch(branchname);
284 if (branch) branch->SetAddress(&((*fDigits)[i]));
285 }
286 }
287}
288
289//_______________________________________________________________________
290TBranch* AliITSDetTypeRec::MakeBranchInTree(TTree *tree, const char* name,
291 const char *classname,
292 void* address,Int_t size,
293 Int_t splitlevel, const char */*file*/)
294{
295//
296// Makes branch in given tree and diverts them to a separate file
297//
298//
299//
300
301 if (tree == 0x0) {
302 Error("MakeBranchInTree","Making Branch %s Tree is NULL",name);
303 return 0x0;
304 }
305 TBranch *branch = tree->GetBranch(name);
306 if (branch) {
307 return branch;
308 }
309 if (classname){
310 branch = tree->Branch(name,classname,address,size,splitlevel);
311 }
312 else {
313 branch = tree->Bronch(name, "TClonesArray", address, size, splitlevel);
314 }
315
316 return branch;
317}
318
319//____________________________________________________________________
320void AliITSDetTypeRec::SetDefaults(){
321
322 //Set defaults for segmentation and response
323
7d62fb64 324 if(fGeom==0){
325 Warning("SetDefaults","fGeom is 0!");
326 return;
327 }
328
329 AliITSsegmentation* seg;
8ba39da9 330 if(!GetCalibration()) {AliFatal("Exit");exit(0);}
7d62fb64 331
332 for(Int_t dettype=0;dettype<fgkNdettypes;dettype++){
333 if(dettype==0){
334 seg = new AliITSsegmentationSPD(fGeom);
335 SetSegmentationModel(dettype,seg);
3a7c3e6d 336 SetDigitClassName(dettype,"AliITSdigitSPD");
337 SetClusterClassName(dettype,"AliITSRawClusterSPD");
338
7d62fb64 339 }
340 if(dettype==1){
fcf95fc7 341 AliITSCalibrationSDD* res=(AliITSCalibrationSDD*) GetCalibrationModel(fGeom->GetStartSDD());
7d62fb64 342 seg = new AliITSsegmentationSDD(fGeom,res);
343 SetSegmentationModel(dettype,seg);
fcf95fc7 344 const char *kopt = ((AliITSresponseSDD*)res->GetResponse())->ZeroSuppOption();
3a7c3e6d 345 if((!strstr(kopt,"2D"))&&(!strstr(kopt,"1D"))) SetDigitClassName(dettype,"AliITSdigit");
346 else SetDigitClassName(dettype,"AliITSdigitSDD");
347 SetClusterClassName(dettype,"AliITSRawClusterSDD");
348
7d62fb64 349 }
350 if(dettype==2){
351 AliITSsegmentationSSD* seg2 = new AliITSsegmentationSSD(fGeom);
352 seg2->SetAngles(0.0075,0.0275); // strip angels rad P and N side.
353 seg2->SetAnglesLay5(0.0075,0.0275); // strip angels rad P and N side.
354 seg2->SetAnglesLay6(0.0275,0.0075); // strip angels rad P and N side.
7d62fb64 355 SetSegmentationModel(dettype,seg2);
7d62fb64 356 SetDigitClassName(dettype,"AliITSdigitSSD");
357 SetClusterClassName(dettype,"AliITSRawClusterSSD");
358 }
7d62fb64 359 }
7d62fb64 360
361}
8ba39da9 362//______________________________________________________________________
363Bool_t AliITSDetTypeRec::GetCalibration() {
364 // Get Default calibration if a storage is not defined.
365
06232459 366
fcf95fc7 367 AliCDBEntry *entrySPD = AliCDBManager::Instance()->Get("ITS/Calib/CalibSPD", fRunNumber);
368 AliCDBEntry *entrySDD = AliCDBManager::Instance()->Get("ITS/Calib/CalibSDD", fRunNumber);
369 AliCDBEntry *entrySSD = AliCDBManager::Instance()->Get("ITS/Calib/CalibSSD", fRunNumber);
370 AliCDBEntry *entry2SPD = AliCDBManager::Instance()->Get("ITS/Calib/RespSPD", fRunNumber);
371 AliCDBEntry *entry2SDD = AliCDBManager::Instance()->Get("ITS/Calib/RespSDD", fRunNumber);
372 AliCDBEntry *entry2SSD = AliCDBManager::Instance()->Get("ITS/Calib/RespSSD", fRunNumber);
06232459 373
fcf95fc7 374 if(!entrySPD || !entrySDD || !entrySSD || !entry2SPD || !entry2SDD || !entry2SSD){
06232459 375 AliWarning("Calibration object retrieval failed! Dummy calibration will be used.");
376 AliCDBStorage *origStorage = AliCDBManager::Instance()->GetDefaultStorage();
377 AliCDBManager::Instance()->SetDefaultStorage("local://$ALICE_ROOT");
378
fcf95fc7 379 entrySPD = AliCDBManager::Instance()->Get("ITS/Calib/CalibSPD", fRunNumber);
380 entrySDD = AliCDBManager::Instance()->Get("ITS/Calib/CalibSDD", fRunNumber);
381 entrySSD = AliCDBManager::Instance()->Get("ITS/Calib/CalibSSD", fRunNumber);
382 entry2SPD = AliCDBManager::Instance()->Get("ITS/Calib/RespSPD", fRunNumber);
383 entry2SDD = AliCDBManager::Instance()->Get("ITS/Calib/RespSDD", fRunNumber);
384 entry2SSD = AliCDBManager::Instance()->Get("ITS/Calib/RespSSD", fRunNumber);
06232459 385
386 AliCDBManager::Instance()->SetDefaultStorage(origStorage);
8ba39da9 387 }
8ba39da9 388
fcf95fc7 389
8ba39da9 390 TObjArray *respSPD = (TObjArray *)entrySPD->GetObject();
391 entrySPD->SetObject(NULL);
392 entrySPD->SetOwner(kTRUE);
fcf95fc7 393
394 AliITSresponseSPD *pSPD = (AliITSresponseSPD*)entry2SPD->GetObject();
395 entry2SPD->SetObject(NULL);
396 entry2SPD->SetOwner(kTRUE);
397
8ba39da9 398 TObjArray *respSDD = (TObjArray *)entrySDD->GetObject();
399 entrySDD->SetObject(NULL);
400 entrySDD->SetOwner(kTRUE);
fcf95fc7 401
402 AliITSresponseSDD *pSDD = (AliITSresponseSDD*)entry2SDD->GetObject();
403 entry2SDD->SetObject(NULL);
404 entry2SDD->SetOwner(kTRUE);
405
8ba39da9 406 TObjArray *respSSD = (TObjArray *)entrySSD->GetObject();
407 entrySSD->SetObject(NULL);
408 entrySSD->SetOwner(kTRUE);
fcf95fc7 409
410 AliITSresponseSSD *pSSD = (AliITSresponseSSD*)entry2SSD->GetObject();
411 entry2SSD->SetObject(NULL);
412 entry2SSD->SetOwner(kTRUE);
413
8ba39da9 414 // DB entries are dleted. In this waymetadeta objects are deleted as well
415 delete entrySPD;
416 delete entrySDD;
417 delete entrySSD;
fcf95fc7 418 delete entry2SPD;
419 delete entry2SDD;
420 delete entry2SSD;
06232459 421
fcf95fc7 422
423 if ((!pSPD)||(!pSDD)||(!pSSD) || (!respSPD) || (!respSDD) || (!respSSD)) {
8ba39da9 424 AliWarning("Can not get calibration from calibration database !");
425 return kFALSE;
426 }
fcf95fc7 427
8ba39da9 428 fNMod[0] = respSPD->GetEntries();
429 fNMod[1] = respSDD->GetEntries();
430 fNMod[2] = respSSD->GetEntries();
431 AliInfo(Form("%i SPD, %i SDD and %i SSD in calibration database",
432 fNMod[0], fNMod[1], fNMod[2]));
fcf95fc7 433 AliITSCalibration* res;
8ba39da9 434 for (Int_t i=0; i<fNMod[0]; i++) {
fcf95fc7 435 res = (AliITSCalibration*) respSPD->At(i);
436 res->SetResponse((AliITSresponse*)pSPD);
8ba39da9 437 SetCalibrationModel(i, res);
438 }
439 for (Int_t i=0; i<fNMod[1]; i++) {
fcf95fc7 440 res = (AliITSCalibration*) respSDD->At(i);
441 res->SetResponse((AliITSresponse*)pSDD);
8ba39da9 442 Int_t iMod = i + fNMod[0];
443 SetCalibrationModel(iMod, res);
444 }
445 for (Int_t i=0; i<fNMod[2]; i++) {
fcf95fc7 446 res = (AliITSCalibration*) respSSD->At(i);
447 res->SetResponse((AliITSresponse*)pSSD);
8ba39da9 448 Int_t iMod = i + fNMod[0] + fNMod[1];
449 SetCalibrationModel(iMod, res);
450 }
451
452 return kTRUE;
453}
454
7d62fb64 455
456//________________________________________________________________
457void AliITSDetTypeRec::SetDefaultClusterFinders(){
458
459 //set defaults for standard cluster finder
460
461 if(fGeom==0){
462 Warning("SetDefaults","fGeom is 0!");
463 return;
464 }
465
7d62fb64 466 AliITSClusterFinder *clf;
467
468 MakeTreeC();
469
3a7c3e6d 470 for(Int_t dettype=0;dettype<fgkNdettypes;dettype++){
7d62fb64 471 //SPD
472 if(dettype==0){
473 if(!GetReconstructionModel(dettype)){
7d62fb64 474 TClonesArray *dig0 = DigitsAddress(0);
475 TClonesArray *rec0 = ClustersAddress(0);
8ba39da9 476 clf = new AliITSClusterFinderSPD(this,dig0,rec0);
7d62fb64 477 SetReconstructionModel(dettype,clf);
8ba39da9 478
7d62fb64 479 }
480 }
8ba39da9 481
7d62fb64 482 //SDD
483 if(dettype==1){
484 if(!GetReconstructionModel(dettype)){
7d62fb64 485 TClonesArray *dig1 = DigitsAddress(1);
486 TClonesArray *rec1 = ClustersAddress(1);
8ba39da9 487 clf = new AliITSClusterFinderSDD(this,dig1,rec1);
7d62fb64 488 SetReconstructionModel(dettype,clf);
489 }
490
491 }
492 //SSD
493 if(dettype==2){
494 if(!GetReconstructionModel(dettype)){
7d62fb64 495 TClonesArray* dig2 = DigitsAddress(2);
8ba39da9 496 clf = new AliITSClusterFinderSSD(this,dig2);
7d62fb64 497 SetReconstructionModel(dettype,clf);
498 }
499 }
500
501 }
502
503
504}
505
506//________________________________________________________________
3a7c3e6d 507void AliITSDetTypeRec::SetDefaultClusterFindersV2(Bool_t rawdata){
7d62fb64 508
509 //Set defaults for cluster finder V2
510
511 if(fGeom==0){
512 Warning("SetDefaults","fGeom is 0!");
513 return;
514 }
515
7d62fb64 516 AliITSClusterFinder *clf;
517
518 MakeTreeC();
3a7c3e6d 519 for(Int_t dettype=0;dettype<fgkNdettypes;dettype++){
7d62fb64 520 //SPD
521 if(dettype==0){
522 if(!GetReconstructionModel(dettype)){
8ba39da9 523 clf = new AliITSClusterFinderV2SPD(this);
7d62fb64 524 clf->InitGeometry();
3a7c3e6d 525 if(!rawdata) clf->SetDigits(DigitsAddress(0));
7d62fb64 526 SetReconstructionModel(dettype,clf);
527
528 }
529 }
530 //SDD
531 if(dettype==1){
532 if(!GetReconstructionModel(dettype)){
8ba39da9 533 clf = new AliITSClusterFinderV2SDD(this);
534 clf->InitGeometry();
3a7c3e6d 535 if(!rawdata) clf->SetDigits(DigitsAddress(1));
7d62fb64 536 SetReconstructionModel(dettype,clf);
537 }
538
539 }
540
541 //SSD
542 if(dettype==2){
543 if(!GetReconstructionModel(dettype)){
8ba39da9 544 clf = new AliITSClusterFinderV2SSD(this);
7d62fb64 545 clf->InitGeometry();
3a7c3e6d 546 if(!rawdata) clf->SetDigits(DigitsAddress(2));
7d62fb64 547 SetReconstructionModel(dettype,clf);
548 }
549 }
550
551 }
552
553}
554//______________________________________________________________________
555void AliITSDetTypeRec::MakeBranch(Option_t* option){
556
557 //Creates branches for clusters and recpoints
558 Bool_t cR = (strstr(option,"R")!=0);
559 Bool_t cRF = (strstr(option,"RF")!=0);
560 Bool_t v2 = (strstr(option,"v2")!=0);
561
562 if(cRF)cR = kFALSE;
563
564 if(cR) MakeBranchR(0);
565 if(cRF) MakeBranchRF(0);
566 if(v2) MakeBranchR(0,"v2");
567
568}
569
570//_____________________________________________________________
571void AliITSDetTypeRec::MakeTreeC(){
572
573 //Create a separate tree to store the clusters
574 if(!fLoader){
575 Warning("MakeTreeC","ITS loader is null!");
576 return;
577 }
578 if(fLoader->TreeC()== 0x0) fLoader->MakeTree("C");
579 MakeBranchC();
580}
581
582//______________________________________________________________
583void AliITSDetTypeRec::MakeBranchC(){
584
585 //Make branches in the tree of clusters
586
587 if(!fLoader){
588 Warning("MakeBranchC","ITS loader is null!");
589 return;
590 }
591 TTree* lTC = fLoader->TreeC();
592 if(lTC==0x0){
593 Error("MakeTreeC","Can not get TreeC from loader");
594 return;
595 }
596
597 Int_t buffersize = 4000;
598 Char_t branchname[30];
599 Char_t* cluclass;
600 const char *det[4]={"SPD","SDD","SSD","ITS"};
601
602 for(Int_t i=0;i<fgkNdettypes;i++){
603 cluclass = GetClusterClassName(i);
604 if(fCtype==0x0) fCtype = new TObjArray(fgkNdettypes);
605 if(!ClustersAddress(i)){
606 fCtype->AddAt(new TClonesArray(cluclass,1000),i);
607 }
608 if(fgkNdettypes==3) sprintf(branchname,"%sClusters%s",det[3],det[i]);
609 else sprintf(branchname,"%sClusters%d",det[3],i+1);
610 if(fCtype && lTC){
611 if(lTC->GetBranch(branchname)){
612 Warning("MakeBranchC","Branch %s already exists in TreeC",branchname);
613 } else{
614 Info("MakeBranchC","Creating branch %s in TreeC",branchname);
615 lTC->Branch(branchname,&((*fCtype)[i]),buffersize);
616 }
617 }
618
619 }
620
621}
622
623//_______________________________________________________________
624void AliITSDetTypeRec::GetTreeC(Int_t event){
625
626 //Get the clusters tree for this event and
627 //sets the branch address
628
629
630 if(!fLoader){
631 Warning("GetTreeC","ITS loader is null!");
632 return;
633 }
634
635 Char_t branchname[30];
636 const char *det[4] = {"SPD","SDD","SSD","ITS"};
637 TTree* lTC = fLoader->TreeC();
638
639 ResetClusters();
640 if(lTC) fLoader->CleanRawClusters();
641
642 TBranch* branch;
643 if(lTC){
644 Char_t* cluclass;
645 for(Int_t i=0;i<fgkNdettypes;i++){
646 cluclass = GetClusterClassName(i);
647 if(fCtype==0x0) fCtype = new TObjArray(fgkNdettypes);
648 if(!fCtype->At(i)) fCtype->AddAt(new TClonesArray(cluclass,1000),i);
649 if(fgkNdettypes==3) sprintf(branchname,"%sClusters%s",det[3],det[i]);
650 else sprintf(branchname,"%sClusters%d",det[3],i+1);
651 if(fCtype){
652 branch = lTC->GetBranch(branchname);
653 if(branch) branch->SetAddress(&((*fCtype)[i]));
654 }
655 }
656 } else{
657 Error("GetTreeC","cannot find clusters Tree for vent %d",event);
658 }
659
660}
661
662//___________________________________________________________________
663void AliITSDetTypeRec::AddCluster(Int_t id, AliITSRawCluster *c){
664
665 // Adds a raw cluster to the list
666 TClonesArray &lc = *((TClonesArray*)fCtype->At(id));
667 switch(id){
668 case 0:
669 new(lc[fNctype[id]++]) AliITSRawClusterSPD(*((AliITSRawClusterSPD*)c));
670 break;
671 case 1:
672 new(lc[fNctype[id]++]) AliITSRawClusterSDD(*((AliITSRawClusterSDD*)c));
673 break;
674 case 2:
675 new(lc[fNctype[id]++]) AliITSRawClusterSSD(*((AliITSRawClusterSSD*)c));
676 break;
677 }
678}
679//___________________________________________________________________
680void AliITSDetTypeRec::ResetDigits(){
681 // Reset number of digits and the digits array for the ITS detector.
682
683 if(!fDigits) return;
684 for(Int_t i=0;i<fgkNdettypes;i++){
685 ResetDigits(i);
686 }
687}
688//___________________________________________________________________
689void AliITSDetTypeRec::ResetDigits(Int_t branch){
690 // Reset number of digits and the digits array for this branch.
691
692 if(fDigits->At(branch)) ((TClonesArray*)fDigits->At(branch))->Clear();
693 if(fNdtype) fNdtype[branch]=0;
694
695}
696
697//__________________________________________________________________
698void AliITSDetTypeRec::ResetClusters(){
699
700 //Resets number of clusters and the cluster array
701 for(Int_t i=0;i<fgkNdettypes;i++){
702 ResetClusters(i);
703 }
704}
705
706//__________________________________________________________________
707void AliITSDetTypeRec::ResetClusters(Int_t i){
708
709 //Resets number of clusters and the cluster array for this branch
710
711 if (fCtype->At(i)) ((TClonesArray*)fCtype->At(i))->Clear();
712 if (fNctype) fNctype[i]=0;
713}
714//__________________________________________________________________
715void AliITSDetTypeRec::MakeBranchR(const char *file, Option_t *opt){
716
717 //Creates tree branches for recpoints
718 // Inputs:
719 // cont char *file File name where RecPoints branch is to be written
720 // to. If blank it write the SDigits to the same
721 // file in which the Hits were found.
722
723 if(!fLoader){
724 Warning("MakeBranchR","ITS loader is null!");
725 return;
726 }
727
728 Int_t buffsz = 4000;
729 char branchname[30];
730
731 // only one branch for rec points for all detector types
732 Bool_t oFast= (strstr(opt,"Fast")!=0);
733 Bool_t v2 = (strstr(opt,"v2")!=0);
734
735 Char_t detname[10] = "ITS";
736
737
738 if(oFast){
739 sprintf(branchname,"%sRecPointsF",detname);
740 } else if(v2){
741 sprintf(branchname,"Clusters");
742 } else {
743 sprintf(branchname,"%sRecPoints",detname);
744 }
745
746 if(v2){
747
748 if(!fClustersV2)fClustersV2 = new TClonesArray("AliITSclusterV2",3000);
749 if(fLoader->TreeR()){
750 if(fClustersV2==0x0) fClustersV2 = new TClonesArray("AliITSclusterV2",3000);
751 MakeBranchInTree(fLoader->TreeR(),branchname,0,&fClustersV2,buffsz,99,file);
752
753 }
754 }else{
755 if(!fRecPoints)fRecPoints = new TClonesArray("AliITSRecPoint",1000);
756 if (fLoader->TreeR()) {
757 if(fRecPoints==0x0) fRecPoints = new TClonesArray("AliITSRecPoint",
758 1000);
759 MakeBranchInTree(fLoader->TreeR(),branchname,0,&fRecPoints,buffsz,99,file);
760 } // end if
761 }
762
763}
764//______________________________________________________________________
765void AliITSDetTypeRec::SetTreeAddressR(TTree *treeR){
766 // Set branch address for the Reconstructed points Trees.
3b9df642 767 // Inputs:
7d62fb64 768 // TTree *treeR Tree containing the RecPoints.
3b9df642 769 // Outputs:
7d62fb64 770 // none.
3b9df642 771 // Return:
7d62fb64 772
773 char branchname[30];
774 Char_t namedet[10]="ITS";
775
776 if(!treeR) return;
777 if(fRecPoints==0x0) fRecPoints = new TClonesArray("AliITSRecPoint",1000);
778 TBranch *branch1;
779 sprintf(branchname,"Clusters");
780 branch1 = treeR->GetBranch(branchname);
781 if(branch1){
782 if(fClustersV2==0x0) fClustersV2 = new TClonesArray("AliITSclusterV2",3000);
783 branch1->SetAddress(&fClustersV2);
784 }
785 else{
786 TBranch *branch;
787 sprintf(branchname,"%sRecPoints",namedet);
788 branch = treeR->GetBranch(branchname);
789 if (branch) {
790 branch->SetAddress(&fRecPoints);
791 }else {
792 sprintf(branchname,"%sRecPointsF",namedet);
793 branch = treeR->GetBranch(branchname);
794 if (branch) {
795 branch->SetAddress(&fRecPoints);
796 }
797 }
798 }
3b9df642 799}
7d62fb64 800//____________________________________________________________________
801void AliITSDetTypeRec::AddRecPoint(const AliITSRecPoint &r){
802 // Add a reconstructed space point to the list
803 // Inputs:
804 // const AliITSRecPoint &r RecPoint class to be added to the tree
805 // of reconstructed points TreeR.
806 // Outputs:
807 // none.
808 // Return:
809 // none.
810
811 TClonesArray &lrecp = *fRecPoints;
812 new(lrecp[fNRecPoints++]) AliITSRecPoint(r);
813}
814//______________________________________________________________________
815void AliITSDetTypeRec::AddClusterV2(const AliITSclusterV2 &r){
816 // Add a reconstructed space point to the list
817 // Inputs:
818 // const AliITSClusterV2 &r class to be added to the tree
819 // of reconstructed points TreeR.
820 // Outputs:
821 // none.
822 // Return:
823 // none.
824
825 TClonesArray &lrecp = *fClustersV2;
826 new(lrecp[fNClustersV2++]) AliITSclusterV2(r);
3a7c3e6d 827 }
7d62fb64 828
829//______________________________________________________________________
830void AliITSDetTypeRec::DigitsToRecPoints(Int_t evNumber,Int_t lastentry,Option_t *opt, Bool_t v2){
831 // cluster finding and reconstruction of space points
832 // the condition below will disappear when the geom class will be
833 // initialized for all versions - for the moment it is only for v5 !
834 // 7 is the SDD beam test version
835 // Inputs:
836 // Int_t evNumber Event number to be processed.
837 // Int_t lastentry Offset for module when not all of the modules
838 // are processed.
839 // Option_t *opt String indicating which ITS sub-detectors should
840 // be processed. If ="All" then all of the ITS
841 // sub detectors are processed.
842
843 if(!fGeom){
844 Warning("DigitsToRecPoints","fGeom is null!");
845 return;
846 }
847 if(!fLoader){
848 Warning("DigitsToRecPoints","ITS loader is null!");
849 return;
850 }
851
852 const char *all = strstr(opt,"All");
853 const char *det[3] = {strstr(opt,"SPD"),strstr(opt,"SDD"),
854 strstr(opt,"SSD")};
855 if(!v2) SetDefaultClusterFinders();
856 else SetDefaultClusterFindersV2();
857
858
859 TTree *treeC=fLoader->TreeC();
860 if(!treeC){
861 MakeTreeC();
862 MakeBranchC();
863 }
864 AliITSClusterFinder *rec = 0;
865 Int_t id,module,first=0;
866 for(module=0;module<fGeom->GetIndexMax();module++){
867 id = fGeom->GetModuleType(module);
868 if (!all && !det[id]) continue;
869 if(det[id]) first = fGeom->GetStartDet(id);
870 rec = (AliITSClusterFinder*)GetReconstructionModel(id);
871 TClonesArray *itsDigits = DigitsAddress(id);
872 if (!rec) {
873 Error("DigitsToRecPoints",
874 "The reconstruction class was not instanciated! event=%d",
875 evNumber);
876 exit(1);
877 }
878 ResetDigits();
879 TTree *lTD = fLoader->TreeD();
880 if (all) {
881 lTD->GetEvent(lastentry+module);
882 }else {
883 lTD->GetEvent(lastentry+(module-first));
884 }
885 Int_t ndigits = itsDigits->GetEntriesFast();
886 if(ndigits>0){
887 rec->SetDetTypeRec(this);
7d62fb64 888 rec->SetDigits(DigitsAddress(id));
889 rec->SetClusters(ClustersAddress(id));
890 rec->FindRawClusters(module);
891 } // end if
892 fLoader->TreeR()->Fill();
893 ResetRecPoints();
894 ResetClustersV2();
895 treeC->Fill();
896 ResetClusters();
897 }
898
899 fLoader->WriteRecPoints("OVERWRITE");
900 fLoader->WriteRawClusters("OVERWRITE");
901}
902//______________________________________________________________________
903void AliITSDetTypeRec::DigitsToRecPoints(AliRawReader* rawReader){
904 // cluster finding and reconstruction of space points
905 // the condition below will disappear when the geom class will be
906 // initialized for all versions - for the moment it is only for v5 !
907 // 7 is the SDD beam test version
908 // Inputs:
909 // Int_t evNumber Event number to be processed.
910 // Int_t lastentry Offset for module when not all of the modules
911 // are processed.
912 // Option_t *opt String indicating which ITS sub-detectors should
913 // be processed. If ="All" then all of the ITS
914 // sub detectors are processed.
915 // Outputs:
916 // none.
917 // Return:
918 // none.
3a7c3e6d 919 if(!fGeom){
7d62fb64 920 Warning("DigitsToRecPoints","fGeom is null!");
921 return;
922 }
923 if(!fLoader){
924 Warning("DigitsToRecPoints","ITS loader is null!");
925 return;
926 }
927
7d62fb64 928
929 AliITSClusterFinderV2 *rec = 0;
930 Int_t id=0;
931
932 if(!fLoader->TreeR()) fLoader->MakeTree("R");
933 TTree* cTree = fLoader->TreeR();
934 TClonesArray *array=new TClonesArray("AliITSclusterV2",1000);
935 cTree->Branch("Clusters",&array);
936 delete array;
937
938 TClonesArray** clusters = new TClonesArray*[fGeom->GetIndexMax()];
939 for (Int_t iModule = 0; iModule < fGeom->GetIndexMax(); iModule++) {
940 clusters[iModule] = NULL;
941 }
942 for(id=0;id<3;id++){
943 rec = (AliITSClusterFinderV2*)GetReconstructionModel(id);
8ba39da9 944 rec->SetDetTypeRec(this);
7d62fb64 945 if (!rec) {
946 Error("DigitsToRecPoints",
947 "The reconstruction class was not instanciated");
948 exit(1);
949 }
950 rec->RawdataToClusters(rawReader,clusters);
951 }
952 Int_t nClusters =0;
953 for(Int_t iModule=0;iModule<fGeom->GetIndexMax();iModule++){
954 array = clusters[iModule];
955 if(!array){
956 Error("DigitsToRecPoints","data for module %d missing!",iModule);
957 array = new TClonesArray("AliITSclusterV2");
958 }
959 cTree->SetBranchAddress("Clusters",&array);
960 cTree->Fill();
961 nClusters+=array->GetEntriesFast();
962 delete array;
963 }
964 fLoader->WriteRecPoints("OVERWRITE");
965
966 delete[] clusters;
967 Info("DigitsToRecPoints", "total number of found clustersV2 in ITS: %d\n",
968 nClusters);
969
970}
971
8ba39da9 972