Simple example macro to make some analyzis for the ESD tracks - see comments (Not...
[u/mrichter/AliRoot.git] / ITS / AliITS.cxx
CommitLineData
4c039060 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
88cb7938 16/* $Id$ */
2ab0c725 17
4c039060 18
fe4da5cc 19///////////////////////////////////////////////////////////////////////////////
04366a57 20// //
21// An overview of the basic philosophy of the ITS code development //
22// and analysis is show in the figure below. //
23//Begin_Html //
24/*
a92b2b7d 25<img src="picts/ITS/ITS_Analysis_schema.gif">
fe4da5cc 26</pre>
27<br clear=left>
28<font size=+2 color=red>
58005f18 29<p>Roberto Barbera is in charge of the ITS Offline code (1999).
fe4da5cc 30<a href="mailto:roberto.barbera@ct.infn.it">Roberto Barbera</a>.
31</font>
32<pre>
33*/
34//End_Html
58005f18 35//
36// AliITS. Inner Traking System base class.
37// This class contains the base procedures for the Inner Tracking System
38//
39//Begin_Html
40/*
a92b2b7d 41<img src="picts/ITS/AliITS_Class_Diagram.gif">
58005f18 42</pre>
43<br clear=left>
44<font size=+2 color=red>
45<p>This show the class diagram of the different elements that are part of
46the AliITS class.
47</font>
48<pre>
49*/
50//End_Html
51//
52// Version: 0
53// Written by Rene Brun, Federico Carminati, and Roberto Barbera
54//
55// Version: 1
56// Modified and documented by Bjorn S. Nilsen
57// July 11 1999
58//
3bd79107 59// Version: 2
60// Modified and documented by A. Bologna
61// October 18 1999
62//
58005f18 63// AliITS is the general base class for the ITS. Also see AliDetector for
64// futher information.
65//
fe4da5cc 66///////////////////////////////////////////////////////////////////////////////
88cb7938 67
143d1056 68#include <stdlib.h>
caf37aec 69#include <TClonesArray.h>
9c74c52b 70#include <TFile.h>
7d62fb64 71#include <TParticle.h>
d5da1ecf 72#include <TString.h>
88cb7938 73#include <TTree.h>
88cb7938 74#include <TVirtualMC.h>
7d62fb64 75#include "AliDetector.h"
3bd79107 76#include "AliITS.h"
7d62fb64 77#include "AliITSDetTypeSim.h"
f77f13c8 78#include "AliITSDDLRawData.h"
88cb7938 79#include "AliITSLoader.h"
88cb7938 80#include "AliITShit.h"
81#include "AliITSmodule.h"
7d62fb64 82#include "AliITSpListItem.h"
83#include "AliITSsimulation.h"
84#include "AliITSsimulationFastPoints.h"
5d12ce38 85#include "AliMC.h"
7d62fb64 86#include "AliITSDigitizer.h"
87#include "AliITSRecPoint.h"
f77f13c8 88#include "AliRun.h"
7d62fb64 89#include "AliLog.h"
828e06c7 90
3bd79107 91ClassImp(AliITS)
3bd79107 92
2aea926d 93//______________________________________________________________________
7d62fb64 94AliITS::AliITS() : AliDetector(){
95 // Default initializer for ITS
96 // The default constructor of the AliITS class. In addition to
97 // creating the AliITS class it zeros the variables fIshunt (a member
98 // of AliDetector class), fEuclidOut, and fIdN, and zeros the pointers
99 // fIdSens, and fIdName. The AliDetector default constructor
100 // is also called.
101
102 fEuclidOut=0;
103 fOpt="All";
104 fIdSens=0;
105 fIdName=0;
106 fDetTypeSim=0;
107 fIshunt = 0; // not zeroed in AliDetector.
108 fHits =0;
109 fNhits=0;
110 fITSmodules=0;
111
aa044888 112// SetDetectors(); // default to fOpt="All". This variable not written out.
2aea926d 113 SetMarkerColor(kRed);
fe4da5cc 114}
2aea926d 115//______________________________________________________________________
7d62fb64 116AliITS::AliITS(const char *name, const char *title):AliDetector(name,title){
117 // The standard Constructor for the ITS class.
118 // It also zeros the variables
119 // fIshunt (a member of AliDetector class), fEuclidOut, and zeros
120 // the pointers fIdSens and fIdName. To help in displaying hits via the
121 // ROOT macro display.C AliITS also sets the marker color to red. The
122 // variables passes with this constructor, const char *name and *title,
123 // are used by the constructor of AliDetector class. See AliDetector
124 // class for a description of these parameters and its constructor
125 // functions.
126
127 fEuclidOut=0;
128 fOpt="All";
129 fIdSens=0;
130 fIdName=0;
131 fDetTypeSim=0;
132 fHits = new TClonesArray("AliITShit",1560);
133 if(gAlice->GetMCApp()) gAlice->GetMCApp()->AddHitList(fHits);
134 fNhits=0;
135 fITSmodules = 0;
136
137 fIshunt = 0; // not zeroed in AliDetector
138 // Not done in AliDetector.
139
140 fEuclidOut = 0;
141 SetDetectors(); // default to fOpt="All". This variable not written out.
142
143 fIdName = 0;
144 fIdSens = 0;
145
146 fDetTypeSim = new AliITSDetTypeSim();
147
148 SetMarkerColor(kRed);
fe4da5cc 149
3bd79107 150}
2aea926d 151//______________________________________________________________________
58005f18 152AliITS::~AliITS(){
4a5bebc2 153 // Default destructor for ITS.
2aea926d 154 // The default destructor of the AliITS class. In addition to deleting
7d62fb64 155 // the AliITS class it deletes the memory pointed to by
156 // fIdSens, fIdName, fDetTypeSim and it's contents.
4a5bebc2 157 // Inputs:
158 // none.
159 // Outputs:
160 // none.
161 // Return:
162 // none.
88cb7938 163 if (fHits) {
164 fHits->Delete();
165 delete fHits;
166 fHits=0;
167 }
88cb7938 168 if(fITSmodules) {
169 this->ClearModules();
170 delete fITSmodules;
279a35d7 171 fITSmodules = 0;
88cb7938 172 }// end if fITSmodules!=0
173
7d62fb64 174 delete[] fIdName; // Array of TStrings
175 delete[] fIdSens;
176
177 if (fDetTypeSim){
178 delete fDetTypeSim;
179 fDetTypeSim = 0;
180 }
3bd79107 181}
2aea926d 182//______________________________________________________________________
ac74f489 183AliITS::AliITS(const AliITS &source) : AliDetector(source){
4a5bebc2 184 // Copy constructor. This is a function which is not allowed to be
185 // done to the ITS. It exits with an error.
186 // Inputs:
187 // AliITS &source An AliITS class.
188 // Outputs:
189 // none.
190 // Return:
191 // none.
3bd79107 192
2aea926d 193 if(this==&source) return;
4a5bebc2 194 Error("Copy constructor",
88cb7938 195 "You are not allowed to make a copy of the AliITS");
2aea926d 196 exit(1);
3bd79107 197}
2aea926d 198//______________________________________________________________________
7d62fb64 199AliITS& AliITS::operator=(const AliITS &source){
4a5bebc2 200 // Assignment operator. This is a function which is not allowed to be
201 // done to the ITS. It exits with an error.
202 // Inputs:
203 // AliITS &source An AliITS class.
204 // Outputs:
205 // none.
206 // Return:
207 // none.
3bd79107 208
2aea926d 209 if(this==&source) return *this;
4a5bebc2 210 Error("operator=","You are not allowed to make a copy of the AliITS");
2aea926d 211 exit(1);
212 return *this; //fake return
3bd79107 213}
2aea926d 214//______________________________________________________________________
aacedc3e 215AliDigitizer* AliITS::CreateDigitizer(AliRunDigitizer* manager)const{
216 // Creates the AliITSDigitizer in a standard way for use via AliModule.
217 // This function can not be included in the .h file because of problems
218 // with the order of inclusion (recursive).
4a5bebc2 219 // Inputs:
aacedc3e 220 // AliRunDigitizer *manager The Manger class for Digitization
221 // Output:
222 // none.
4a5bebc2 223 // Return:
aacedc3e 224 // A new AliITSRunDigitizer (cast as a AliDigitizer).
3bd79107 225
aacedc3e 226 return new AliITSDigitizer(manager);
3bd79107 227}
2aea926d 228//______________________________________________________________________
229void AliITS::Init(){
4a5bebc2 230 // Initializer ITS after it has been built
2aea926d 231 // This routine initializes the AliITS class. It is intended to be
232 // called from the Init function in AliITSv?. Besides displaying a banner
233 // indicating that it has been called it initializes the array fIdSens
234 // and sets the default segmentation, response, digit and raw cluster
235 // classes therefore it should be called after a call to CreateGeometry.
4a5bebc2 236 // Inputs:
237 // none.
238 // Outputs:
239 // none.
240 // Return:
241 // none.
3bd79107 242 Int_t i;
3bd79107 243
2aea926d 244 SetDefaults();
245 // Array of TStrings
0e73d04c 246 if(gMC) for(i=0;i<fIdN;i++) fIdSens[i] = gMC->VolId(fIdName[i]);
7d62fb64 247
248 WriteGeometry();
3bd79107 249}
7d62fb64 250
251//______________________________________________________________________
252void AliITS::WriteGeometry(){
253
254 //Writes ITS geomtry on gAlice
255
256 if(!fLoader) MakeLoader(AliConfig::GetDefaultEventFolderName());
257 AliRunLoader* rl = fLoader->GetRunLoader();
258 rl->CdGAFile();
259 AliITSgeom* geom = GetITSgeom();
260 geom->Write();
261
262}
263
2aea926d 264//______________________________________________________________________
265void AliITS::SetDefaults(){
4a5bebc2 266 // sets the default segmentation, response, digit and raw cluster classes.
267 // Inputs:
268 // none.
269 // Outputs:
270 // none.
271 // Return:
272 // none.
7d62fb64 273 AliInfoClass("Setting Defaults");
274
275 if(!fDetTypeSim) {
276 Error("SetDefaults()","fDetTypeSim is 0!");
277 return;
278 }
279
280 fDetTypeSim->SetDefaults();
281
282
3bd79107 283}
2aea926d 284//______________________________________________________________________
285void AliITS::SetDefaultSimulation(){
4a5bebc2 286 // sets the default simulation.
287 // Inputs:
288 // none.
289 // Outputs:
290 // none.
291 // Return:
292 // none.
7d62fb64 293 if(!fDetTypeSim) {
294 Error("SetDefaultSimulation()","fDetTypeSim is 0!");
295 return;
296 }
297
298 fDetTypeSim->SetDefaultSimulation();
04366a57 299
04366a57 300}
301
302
2aea926d 303//______________________________________________________________________
88cb7938 304void AliITS::MakeBranch(Option_t* option){
2aea926d 305 // Creates Tree branches for the ITS.
4a5bebc2 306 // Inputs:
307 // Option_t *option String of Tree types S,D, and/or R.
308 // const char *file String of the file name where these branches
309 // are to be stored. If blank then these branches
310 // are written to the same tree as the Hits were
311 // read from.
312 // Outputs:
313 // none.
314 // Return:
315 // none.
7d62fb64 316 if(!fDetTypeSim) {
317 Error("MakeBranch","fDetTypeSim is 0!");
318 return;
319 }
04366a57 320
7d62fb64 321 Bool_t cH = (strstr(option,"H")!=0);
322 Bool_t cS = (strstr(option,"S")!=0);
323 Bool_t cD = (strstr(option,"D")!=0);
324
325 if(cH && (fHits == 0x0)) fHits = new TClonesArray("AliITShit", 1560);
326 AliDetector::MakeBranch(option);
327
328 if(cS) MakeBranchS(0);
329 if(cD) MakeBranchD(0);
330
331
332}
333//___________________________________________________________________
334void AliITS::MakeBranchS(const char* fl){
335
336 // Creates Tree Branch for the ITS summable digits.
337 // Inputs:
338 // cont char *fl File name where SDigits branch is to be written
339 // to. If blank it write the SDigits to the same
340 // file in which the Hits were found.
341
342
343 if(!fDetTypeSim){
344 Error("MakeBranchS","fDetTypeSim is 0!");
345 }
346 Int_t buffersize = 4000;
347 char branchname[30];
348
349 // only one branch for SDigits.
350 sprintf(branchname,"%s",GetName());
351
352 if(fLoader->TreeS()){
353 if(fDetTypeSim->GetSDigits()==0x0) fDetTypeSim->SetSDigits(new TClonesArray("AliITSpListItem",1000));
354 TClonesArray* sdig = (TClonesArray*)fDetTypeSim->GetSDigits();
355 MakeBranchInTree(fLoader->TreeS(),branchname,&sdig,buffersize,fl);
356 }
357}
358//______________________________________________________________________
359void AliITS::MakeBranchD(const char* file){
360
361 //Make branch for digits
362 if(!fDetTypeSim) {
363 Warning("MakeBranchD","fDetTypeSim is 0!");
364 return;
365 }
366 fDetTypeSim->SetLoader(fLoader);
367 MakeBranchInTreeD(fLoader->TreeD(),file);
368}
369
370//___________________________________________________________________
371void AliITS:: MakeBranchInTreeD(TTree* treeD, const char* file){
372 // Creates Tree branches for the ITS.
373
374 if(!fDetTypeSim){
375 Error("MakeBranchS","fDetTypeSim is 0!");
376 }
377 fDetTypeSim->SetLoader(fLoader);
378
379 const Char_t *det[3] = {"SPD","SDD","SSD"};
380 Char_t* digclass;
381 Int_t buffersize = 4000;
382 Char_t branchname[30];
383
384 if(!fDetTypeSim->GetDigits()){
385 fDetTypeSim->SetDigits(new TObjArray(fgkNTYPES));
386 }
387 for(Int_t i=0;i<fgkNTYPES;i++){
388 digclass = fDetTypeSim->GetDigitClassName(i);
389 TString classn = digclass;
390 if(!((fDetTypeSim->GetDigits())->At(i))){
391 (fDetTypeSim->GetDigits())->AddAt(new TClonesArray(classn.Data(),1000),i);
392 }
393 else ResetDigits(i);
394 if(fgkNTYPES==3) sprintf(branchname,"%sDigits%s",GetName(),det[i]);
395 else sprintf(branchname,"%sDigits%d",GetName(),i+1);
396 TObjArray* dig = DigitsAddress(i);
397 if(GetDigits() && treeD) AliDetector::MakeBranchInTree(treeD,branchname, &dig,buffersize,file);
398 }
2aea926d 399
fe4da5cc 400}
2aea926d 401//______________________________________________________________________
402void AliITS::SetTreeAddress(){
403 // Set branch address for the Trees.
4a5bebc2 404 // Inputs:
405 // none.
406 // Outputs:
407 // none.
408 // Return:
409 // none.
aacedc3e 410
7d62fb64 411 if(!fDetTypeSim) {
412 Error("SetTreeAddress","fDetTypeSim is 0!");
413 return;
414 }
415
416 fDetTypeSim->SetLoader(fLoader);
417
418 TTree *treeS = fLoader->TreeS();
419 TTree *treeD = fLoader->TreeD();
420 if (fLoader->TreeH() && (fHits == 0x0)) {
421 fHits = new TClonesArray("AliITShit", 1560);
422 }
423 AliDetector::SetTreeAddress();
424
425 fDetTypeSim->SetTreeAddressS(treeS, (Char_t*)GetName());
426 fDetTypeSim->SetTreeAddressD(treeD, (Char_t*)GetName());
427
2aea926d 428
fe4da5cc 429}
2aea926d 430//______________________________________________________________________
2aea926d 431void AliITS::AddHit(Int_t track, Int_t *vol, Float_t *hits){
432 // Add an ITS hit
433 // The function to add information to the AliITShit class. See the
434 // AliITShit class for a full description. This function allocates the
435 // necessary new space for the hit information and passes the variable
436 // track, and the pointers *vol and *hits to the AliITShit constructor
437 // function.
4a5bebc2 438 // Inputs:
439 // Int_t track Track number which produced this hit.
440 // Int_t *vol Array of Integer Hit information. See AliITShit.h
441 // Float_t *hits Array of Floating Hit information. see AliITShit.h
442 // Outputs:
443 // none.
444 // Return:
445 // none.
7d62fb64 446 TClonesArray &lhits = *fHits;
447 new(lhits[fNhits++]) AliITShit(fIshunt,track,vol,hits);
3bd79107 448}
3bd79107 449
2aea926d 450//______________________________________________________________________
451void AliITS::FillModules(Int_t evnt,Int_t bgrev,Int_t nmodules,
8e8eae84 452 Option_t *option, const char *filename){
7d62fb64 453 // fill the modules with the sorted by module hits; add hits from
454 // background if option=Add.
455
456 static TTree *trH1; //Tree with background hits
457 static Bool_t first=kTRUE;
458 static TFile *file;
459 const char *addBgr = strstr(option,"Add");
460
461 evnt = nmodules; // Dummy use of variables to remove warnings
462 if (addBgr ) {
463 if(first) {
464 file=new TFile(filename);
465 } // end if first
466 first=kFALSE;
467 file->cd();
468 file->ls();
469 // Get Hits Tree header from file
470 if(trH1) delete trH1;
471 trH1=0;
472
473 char treeName[20];
474 sprintf(treeName,"TreeH%d",bgrev);
475 trH1 = (TTree*)gDirectory->Get(treeName);
476 if (!trH1) {
477 Error("FillModules","cannot find Hits Tree for event:%d",bgrev);
478 } // end if !trH1
479 // Set branch addresses
480 } // end if addBgr
481
482 FillModules(fLoader->TreeH(),0); // fill from this file's tree.
3bd79107 483
7d62fb64 484 if (addBgr ) {
485 FillModules(trH1,10000000); // Default mask 10M.
486 TTree *fAli=fLoader->GetRunLoader()->TreeK();
487 TFile *fileAli=0;
488 if (fAli) fileAli =fAli->GetCurrentFile();
489 fileAli->cd();
490 } // end if add
491
492
3bd79107 493}
7d62fb64 494
495
2aea926d 496//______________________________________________________________________
b4012daf 497void AliITS::FillModules(TTree *treeH, Int_t mask) {
498 // fill the modules with the sorted by module hits;
499 // can be called many times to do a merging
500 // Inputs:
501 // TTree *treeH The tree containing the hits to be copied into
502 // the modules.
503 // Int_t mask The track number mask to indecate which file
504 // this hits came from.
505 // Outputs:
506 // none.
507 // Return:
508 // none.
509
88cb7938 510 if (treeH == 0x0)
511 {
512 Error("FillModules","Tree is NULL");
513 }
b4012daf 514 Int_t lay,lad,det,index;
515 AliITShit *itsHit=0;
516 AliITSmodule *mod=0;
517 char branchname[20];
518 sprintf(branchname,"%s",GetName());
519 TBranch *branch = treeH->GetBranch(branchname);
520 if (!branch) {
88cb7938 521 Error("FillModules","%s branch in TreeH not found",branchname);
522 return;
b4012daf 523 } // end if !branch
524 branch->SetAddress(&fHits);
525 Int_t nTracks =(Int_t) treeH->GetEntries();
526 Int_t iPrimTrack,h;
527 for(iPrimTrack=0; iPrimTrack<nTracks; iPrimTrack++){
88cb7938 528 ResetHits();
529 Int_t nBytes = treeH->GetEvent(iPrimTrack);
530 if (nBytes <= 0) continue;
531 Int_t nHits = fHits->GetEntriesFast();
532 for(h=0; h<nHits; h++){
533 itsHit = (AliITShit *)fHits->UncheckedAt(h);
534 itsHit->GetDetectorID(lay,lad,det);
7d62fb64 535 if (GetITSgeom()) {
536 index = GetITSgeom()->GetModuleIndex(lay,lad,det);
88cb7938 537 } else {
538 index=det-1; // This should not be used.
539 } // end if [You must have fITSgeom for this to work!]
540 mod = GetModule(index);
541 itsHit->SetTrack(itsHit->GetTrack()+mask); // Set track mask.
542 mod->AddHit(itsHit,iPrimTrack,h);
543 } // end loop over hits
b4012daf 544 } // end loop over tracks
545}
7d62fb64 546
b4012daf 547//______________________________________________________________________
7d62fb64 548void AliITS::InitModules(Int_t size,Int_t &nmodules){
549 // Initialize the modules array.
4a5bebc2 550 // Inputs:
7d62fb64 551 // Int_t size Size of array of the number of modules to be
552 // created. If size <=0 then the number of modules
553 // is gotten from AliITSgeom class kept in fITSgeom.
4a5bebc2 554 // Outputs:
7d62fb64 555 // Int_t &nmodules The number of modules existing.
4a5bebc2 556 // Return:
557 // none.
2ab0c725 558
7d62fb64 559 if(fITSmodules){
560 fITSmodules->Delete();
561 delete fITSmodules;
562 } // end fir fITSmoudles
88cb7938 563
7d62fb64 564 if(!fDetTypeSim) {
565 Error("InitModules","fDetTypeSim is null!");
566 return;
567 }
828e06c7 568
7d62fb64 569 Int_t nl,indexMAX,index;
570
571 if(size<=0){ // default to using data stored in AliITSgeom
572 if(fDetTypeSim->GetITSgeom()==0) {
573 Error("InitModules","fITSgeom not defined");
574 return;
575 } // end if fITSgeom==0
576 nl = fDetTypeSim->GetITSgeom()->GetNlayers();
577 indexMAX = fDetTypeSim->GetITSgeom()->GetIndexMax();
578 nmodules = indexMAX;
579 fITSmodules = new TObjArray(indexMAX);
580 for(index=0;index<indexMAX;index++){
581 fITSmodules->AddAt( new AliITSmodule(index),index);
582 } // end for index
583 }else{
584 fITSmodules = new TObjArray(size);
585 for(index=0;index<size;index++) {
586 fITSmodules->AddAt( new AliITSmodule(index),index);
587 } // end for index
588
589 nmodules = size;
590 } // end i size<=0
2aea926d 591}
592//______________________________________________________________________
593void AliITS::Hits2SDigits(){
594 // Standard Hits to summable Digits function.
4a5bebc2 595 // Inputs:
596 // none.
597 // Outputs:
598 // none.
7d62fb64 599
2aea926d 600
7d62fb64 601 if(!fDetTypeSim) {
602 Error("Hits2SDigits","fDetTypeSim is null!");
603 return;
604
605 }
606
607 SetDefaults();
608 fLoader->LoadHits("read");
609 fLoader->LoadSDigits("recreate");
610 AliRunLoader* rl = fLoader->GetRunLoader();
611 fDetTypeSim->SetLoader(fLoader);
612 for (Int_t iEvent = 0; iEvent < rl->GetNumberOfEvents(); iEvent++) {
aacedc3e 613 // Do the Hits to Digits operation. Use Standard input values.
614 // Event number from file, no background hit merging , use size from
615 // AliITSgeom class, option="All", input from this file only.
7d62fb64 616 rl->GetEvent(iEvent);
617 if (!fLoader->TreeS()) fLoader->MakeTree("S");
618 MakeBranch("S");
619 SetTreeAddress();
620 HitsToPreDigits(iEvent,0,-1," ",fOpt," ");
621 } // end for iEvent
88cb7938 622
7d62fb64 623 fLoader->UnloadHits();
624 fLoader->UnloadSDigits();
625
85a5290f 626}
627//______________________________________________________________________
7d62fb64 628void AliITS::Hits2Digits(){
629
630 //Conversion from hits to digits
631 if(!fDetTypeSim) {
632 Error("Hits2SDigits","fDetTypeSim is 0!");
633 return;
634 }
635
636 fDetTypeSim->SetLoader(fLoader);
637 SetDefaults();
638
639 fLoader->LoadHits("read");
640 fLoader->LoadDigits("recreate");
641 AliRunLoader* rl = fLoader->GetRunLoader();
642 for (Int_t iEvent = 0; iEvent < rl->GetNumberOfEvents(); iEvent++) {
643 rl->GetEvent(iEvent);
644 if (!fLoader->TreeD()) fLoader->MakeTree("D");
645 MakeBranch("D");
646 SetTreeAddress();
647 HitsToDigits(iEvent,0,-1," ",fOpt," ");
648 }
649
650 fLoader->UnloadHits();
651 fLoader->UnloadDigits();
652
2aea926d 653}
7d62fb64 654
2aea926d 655//______________________________________________________________________
7d62fb64 656void AliITS::HitsToDigits(Int_t evNumber,Int_t bgrev,Int_t size,
657 Option_t *option,Option_t *opt,
658 const char *filename){
659 // Keep galice.root for signal and name differently the file for
660 // background when add! otherwise the track info for signal will be lost !
661 // the condition below will disappear when the geom class will be
662 // initialized for all versions - for the moment it is only for v5 !
663 // 7 is the SDD beam test version.
4a5bebc2 664 // Inputs:
7d62fb64 665 // Int_t evnt Event to be processed.
666 // Int_t bgrev Background Hit tree number.
667 // Int_t nmodules Not used.
668 // Option_t *option String indicating if merging hits or not. To
669 // merge hits set equal to "Add". Otherwise no
670 // background hits are considered.
671 // Test_t *filename File name containing the background hits..
4a5bebc2 672 // Outputs:
673 // none.
7d62fb64 674 // Return:
675 // none.
2aea926d 676
7d62fb64 677 if(!fDetTypeSim) {
678 Error("HitsToDigits","fDetTypeSim is null!");
679 return;
680 }
681 fDetTypeSim->SetLoader(fLoader);
682 if(!GetITSgeom()) return; // need transformations to do digitization.
683 AliITSgeom *geom = GetITSgeom();
85a5290f 684
7d62fb64 685 const char *all = strstr(opt,"All");
686 const char *det[3] = {strstr(opt,"SPD"),strstr(opt,"SDD"),
687 strstr(opt,"SSD")};
688 static Bool_t setDef=kTRUE;
689 if (setDef) SetDefaultSimulation();
690 setDef=kFALSE;
691
692 Int_t nmodules;
693 InitModules(size,nmodules);
694 FillModules(evNumber,bgrev,nmodules,option,filename);
695
696 AliITSsimulation *sim = 0;
697 AliITSmodule *mod = 0;
698 Int_t id;
699 for(Int_t module=0;module<geom->GetIndexMax();module++){
700 id = geom->GetModuleType(module);
701 if (!all && !det[id]) continue;
702 sim = (AliITSsimulation*)fDetTypeSim->GetSimulationModel(id);
703 if (!sim) {
704 Error("HitsToDigits","The simulation class was not "
705 "instanciated for module %d type %s!",module,
706 geom->GetModuleTypeName(module));
707 exit(1);
708 } // end if !sim
709 mod = (AliITSmodule *)fITSmodules->At(module);
710 sim->DigitiseModule(mod,module,evNumber);
711 // fills all branches - wasted disk space
712 fLoader->TreeD()->Fill();
713 ResetDigits();
714 } // end for module
715
716 ClearModules();
717
718 fLoader->TreeD()->GetEntries();
719 fLoader->TreeD()->AutoSave();
720 // reset tree
721 fLoader->TreeD()->Reset();
2aea926d 722}
7d62fb64 723//_____________________________________________________________________
724void AliITS::Hits2PreDigits(){
725 // Turn hits into SDigits
726
727 if(!fDetTypeSim) {
728 Error("Hits2SDigits","fDetTypeSim is 0!");
729 return;
730 }
731
732 fDetTypeSim->SetLoader(fLoader);
733 SetDefaults();
734
735 HitsToPreDigits(fLoader->GetRunLoader()->GetEventNumber(),
736 0,-1," ",fOpt," ");
737}
738
2aea926d 739//______________________________________________________________________
2aea926d 740void AliITS::HitsToPreDigits(Int_t evNumber,Int_t bgrev,Int_t size,
aacedc3e 741 Option_t *option,Option_t *opt,
742 const char *filename){
4a5bebc2 743 // Keep galice.root for signal and name differently the file for
2aea926d 744 // background when add! otherwise the track info for signal will be lost !
745 // the condition below will disappear when the geom class will be
4a5bebc2 746 // initialized for all versions - for the moment it is only for v5 !
747 // 7 is the SDD beam test version.
748 // Inputs:
749 // Int_t evnt Event to be processed.
750 // Int_t bgrev Background Hit tree number.
751 // Int_t nmodules Not used.
752 // Option_t *option String indicating if merging hits or not. To
753 // merge hits set equal to "Add". Otherwise no
754 // background hits are considered.
755 // Test_t *filename File name containing the background hits..
756 // Outputs:
757 // none.
758 // Return:
759 // none.
2aea926d 760
7d62fb64 761
762 if(!fDetTypeSim) {
763 Error("HitsToPreDigits","fDetTypeSim is null!");
764 return;
765 }
766 fDetTypeSim->SetLoader(fLoader);
767
768 if(!GetITSgeom()){
769 Error("HitsToPreDigits","fGeom is null!");
770 return; // need transformations to do digitization.
771 }
772 AliITSgeom *geom = GetITSgeom();
773
774 const char *all = strstr(opt,"All");
775 const char *det[3] = {strstr(opt,"SPD"),strstr(opt,"SDD"),
776 strstr(opt,"SSD")};
777 static Bool_t setDef=kTRUE;
778 if (setDef) SetDefaultSimulation();
779 setDef=kFALSE;
780
781 Int_t nmodules;
782 InitModules(size,nmodules);
783 FillModules(evNumber,bgrev,nmodules,option,filename);
784
785
786 AliITSsimulation *sim = 0;
787 AliITSmodule *mod = 0;
788 Int_t id,module;
789 for(module=0;module<geom->GetIndexMax();module++){
790 id = geom->GetModuleType(module);
791 if (!all && !det[id]) continue;
792 sim = (AliITSsimulation*)GetSimulationModel(id);
793 if (!sim) {
794 Error("HitsToPreDigits","The simulation class was not "
795 "instanciated for module %d type %s!",module,
796 geom->GetModuleTypeName(module));
797 exit(1);
798 } // end if !sim
799 mod = (AliITSmodule *)fITSmodules->At(module);
800 sim->SDigitiseModule(mod,module,evNumber);
801 // fills all branches - wasted disk space
802 fLoader->TreeS()->Fill();
803 fDetTypeSim->ResetSDigits();
804 } // end for module
805
806 ClearModules();
807
808
809 fLoader->TreeS()->GetEntries();
810 fLoader->TreeS()->AutoSave();
811 fLoader->WriteSDigits("OVERWRITE");
812 // reset tree
813 fLoader->TreeS()->Reset();
2aea926d 814}
7d62fb64 815
816//_____________________________________________________________________
817void AliITS::HitsToFastRecPoints(Int_t evNumber,Int_t bgrev,Int_t size,
818 Option_t *opt0,Option_t *opt1,
819 const char *flnm){
820 // keep galice.root for signal and name differently the file for
2aea926d 821 // background when add! otherwise the track info for signal will be lost !
822 // the condition below will disappear when the geom class will be
4a5bebc2 823 // initialized for all versions - for the moment it is only for v5 !
4a5bebc2 824 // Inputs:
825 // Int_t evnt Event to be processed.
826 // Int_t bgrev Background Hit tree number.
7d62fb64 827 // Int_t size Size used by InitModules. See InitModules.
828 // Option_t *opt0 Option passed to FillModules. See FillModules.
829 // Option_t *opt1 String indicating if merging hits or not. To
4a5bebc2 830 // merge hits set equal to "Add". Otherwise no
831 // background hits are considered.
7d62fb64 832 // Test_t *flnm File name containing the background hits..
4a5bebc2 833 // Outputs:
834 // none.
835 // Return:
836 // none.
837
04366a57 838
00a7cc50 839
7d62fb64 840 if(!GetITSgeom()){
841 Error("HitsToPreDigits","fGeom is null!");
842 return; // need transformations to do digitization.
843 }
844 AliITSgeom *geom = GetITSgeom();
845
846 AliITSLoader *pITSloader = (AliITSLoader*)fLoader;
847
848 const char *all = strstr(opt1,"All");
849 const char *det[3] ={strstr(opt1,"SPD"),strstr(opt1,"SDD"),
850 strstr(opt1,"SSD")};
851 Int_t nmodules;
852 InitModules(size,nmodules);
853 FillModules(evNumber,bgrev,nmodules,opt0,flnm);
854
855 AliITSsimulation *sim = 0;
856 AliITSmodule *mod = 0;
857 Int_t id,module;
858
859 TTree *lTR = pITSloader->TreeR();
00a7cc50 860 if(!lTR) {
861 pITSloader->MakeTree("R");
862 lTR = pITSloader->TreeR();
863 }
864
7d62fb64 865 TClonesArray* ptarray = new TClonesArray("AliITSRecPoint",1000);
00a7cc50 866 TBranch* branch = (TBranch*)lTR->Branch("ITSRecPointsF",&ptarray);
867 branch->SetAddress(&ptarray);
7d62fb64 868 //m.b. : this change is nothing but a nice way to make sure
869 //the CPU goes up !
870 AliDebug(1,Form("N mod = %d",geom->GetIndexMax()));
871 for(module=0;module<geom->GetIndexMax();module++){
872 id = geom->GetModuleType(module);
873 if (!all && !det[id]) continue;
874 sim = (AliITSsimulation*)GetSimulationModel(id);
875 if (!sim) {
876 Error("HitsToFastPoints","The simulation class was not "
877 "instanciated for module %d type %x!",module,
878 geom->GetModuleTypeName(module));
879 exit(1);
880 } // end if !sim
881 mod = (AliITSmodule *)fITSmodules->At(module);
882 sim->CreateFastRecPoints(mod,module,gRandom,ptarray);
883 lTR->Fill();
884 } // end for module
885
886 ClearModules();
887 fLoader->WriteRecPoints("OVERWRITE");
888 delete ptarray;
2aea926d 889}
7d62fb64 890//_____________________________________________________________________
891Int_t AliITS::Hits2Clusters(TTree *hTree, TTree *cTree) {
892 //------------------------------------------------------------
893 // This function creates ITS clusters
894 //------------------------------------------------------------
895 if(!GetITSgeom()){
896 Error("HitsToPreDigits","fGeom is null!");
897 return 1; // need transformations to do digitization.
898 }
899 AliITSgeom *geom=GetITSgeom();
900 Int_t mmax=geom->GetIndexMax();
901
902 InitModules(-1,mmax);
903 FillModules(hTree,0);
904
7d62fb64 905 TClonesArray *points = new TClonesArray("AliITSRecPoint",1000);
00a7cc50 906 TBranch *branch=cTree->GetBranch("ITSRecPoints");
907 if (!branch) cTree->Branch("ITSRecPoints",&points);
908 else branch->SetAddress(&points);
7d62fb64 909
910 AliITSsimulationFastPoints sim;
911 Int_t ncl=0;
912 for (Int_t m=0; m<mmax; m++) {
913 AliITSmodule *mod=GetModule(m);
914 sim.CreateFastRecPoints(mod,m,gRandom,points);
00a7cc50 915 ncl+=points->GetEntriesFast();
7d62fb64 916 cTree->Fill();
7d62fb64 917 points->Clear();
918 }
919
920 Info("Hits2Clusters","Number of found fast clusters : %d",ncl);
921
922 //cTree->Write();
923
7d62fb64 924 delete points;
925 return 0;
926}
7d62fb64 927
7d62fb64 928//_____________________________________________________________________
929void AliITS::CheckLabels(Int_t lab[3]) const {
930 //------------------------------------------------------------
931 // Tries to find mother's labels
932 //------------------------------------------------------------
933
934 if(lab[0]<0 && lab[1]<0 && lab[2]<0) return; // In case of no labels just exit
935
936 Int_t ntracks = gAlice->GetMCApp()->GetNtrack();
937 for (Int_t i=0;i<3;i++){
938 Int_t label = lab[i];
939 if (label>=0 && label<ntracks) {
940 TParticle *part=(TParticle*)gAlice->GetMCApp()->Particle(label);
941 if (part->P() < 0.005) {
942 Int_t m=part->GetFirstMother();
943 if (m<0) {
944 continue;
945 }
946 if (part->GetStatusCode()>0) {
947 continue;
948 }
949 lab[i]=m;
950 }
951 }
952 }
953
954}
955
956//______________________________________________________________________
957void AliITS::SDigitsToDigits(Option_t *opt){
958 // Standard Summable digits to Digits function.
959 // Inputs:
960 // none.
961 // Outputs:
962 // none.
963 if(!fDetTypeSim) {
964 Error("SDigitsToSDigits","fDetTypeSim is 0!");
965 return;
966 }
967
968 fDetTypeSim->SetLoader(fLoader);
969 SetDefaults();
970 fDetTypeSim->SDigitsToDigits(opt,(Char_t*)GetName());
971
972}
973
2aea926d 974//______________________________________________________________________
2aea926d 975void AliITS::ResetDigits(){
4a5bebc2 976 // Reset number of digits and the digits array for the ITS detector.
977 // Inputs:
978 // none.
979 // Outputs:
980 // none.
7d62fb64 981 if(!fDetTypeSim) {
982 Error("ResetDigits","fDetTypeSim is 0!");
983 return;
984 }
985
986 fDetTypeSim->ResetDigits();
3bd79107 987
988
2aea926d 989}
990//______________________________________________________________________
7d62fb64 991void AliITS::ResetDigits(Int_t branch){
4a5bebc2 992 // Reset number of digits and the digits array for this branch.
993 // Inputs:
994 // none.
995 // Outputs:
996 // none.
828e06c7 997
7d62fb64 998 if(!fDetTypeSim) {
999 Error("ResetDigits","fDetTypeSim is 0!");
1000 return;
1001 }
1002
1003 fDetTypeSim->ResetDigits(branch);
1004
2aea926d 1005}
1006//______________________________________________________________________
1007void AliITS::AddSumDigit(AliITSpListItem &sdig){
4a5bebc2 1008 // Adds the a module full of summable digits to the summable digits tree.
1009 // Inputs:
1010 // AliITSpListItem &sdig SDigit to be added to SDigits tree.
1011 // Outputs:
1012 // none.
1013 // Return:
1014 // none.
828e06c7 1015
7d62fb64 1016 if(!fDetTypeSim) {
1017 Error("AddSumDigit","fDetTypeSim is 0!");
1018 return;
1019 }
1020 fDetTypeSim->AddSumDigit(sdig);
1021
828e06c7 1022}
2aea926d 1023//______________________________________________________________________
7d62fb64 1024void AliITS::AddRealDigit(Int_t branch, Int_t *digits){
4a5bebc2 1025 // Add a real digit - as coming from data.
1026 // Inputs:
1027 // Int_t id Detector type number.
1028 // Int_t *digits Integer array containing the digits info. See
1029 // AliITSdigit.h
1030 // Outputs:
1031 // none.
1032 // Return:
1033 // none.
9c74c52b 1034
7d62fb64 1035 if(!fDetTypeSim) {
1036 Error("AddRealDigit","fDetTypeSim is 0!");
1037 return;
1038 }
1039 fDetTypeSim->AddRealDigit(branch,digits);
1040
2aea926d 1041}
1042//______________________________________________________________________
7d62fb64 1043void AliITS::AddSimDigit(Int_t branch, AliITSdigit *d){
4a5bebc2 1044 // Add a simulated digit.
1045 // Inputs:
1046 // Int_t id Detector type number.
1047 // AliITSdigit *d Digit to be added to the Digits Tree. See
1048 // AliITSdigit.h
1049 // Outputs:
1050 // none.
1051 // Return:
1052 // none.
2aea926d 1053
7d62fb64 1054 if(!fDetTypeSim) {
1055 Error("AddSimDigit","fDetTypeSim is 0!");
1056 return;
1057 }
1058 fDetTypeSim->AddSimDigit(branch,d);
1059
2aea926d 1060}
1061//______________________________________________________________________
7d62fb64 1062void AliITS::AddSimDigit(Int_t branch,Float_t phys,Int_t *digits,Int_t *tracks,
88cb7938 1063 Int_t *hits,Float_t *charges){
7f1a504b 1064 // Add a simulated digit to the list.
1065 // Inputs:
1066 // Int_t id Detector type number.
1067 // Float_t phys Physics indicator. See AliITSdigits.h
1068 // Int_t *digits Integer array containing the digits info. See
1069 // AliITSdigit.h
1070 // Int_t *tracks Integer array [AliITSdigitS?D::GetNTracks()]
1071 // containing the track numbers that contributed to
1072 // this digit.
1073 // Int_t *hits Integer array [AliITSdigitS?D::GetNTracks()]
1074 // containing the hit numbers, from AliITSmodule, that
1075 // contributed to this digit.
1076 // Float_t *charge Floating point array of the signals contributed
1077 // to this digit by each track.
1078 // Outputs:
1079 // none.
1080 // Return:
1081 // none.
2aea926d 1082
7d62fb64 1083 if(!fDetTypeSim) {
1084 Error("AddSimDigit","fDetTypeSim is 0!");
1085 return;
1086 }
1087 fDetTypeSim->AddSimDigit(branch,phys,digits,tracks,hits,charges);
1088
2aea926d 1089}
0421c3d1 1090//______________________________________________________________________
aacedc3e 1091void AliITS::Digits2Raw(){
1092 // convert digits of the current event to raw data
0421c3d1 1093
7d62fb64 1094 if(!fDetTypeSim) {
1095 Error("Digits2Raw","fDetTypeSim is 0!");
1096 return;
1097 }
1098 fDetTypeSim->SetLoader(fLoader);
1099 SetDefaults();
1100 fDetTypeSim->GetLoader()->LoadDigits();
1101 TTree* digits = fDetTypeSim->GetLoader()->TreeD();
0421c3d1 1102 if (!digits) {
aacedc3e 1103 Error("Digits2Raw", "no digits tree");
1104 return;
0421c3d1 1105 }
7d62fb64 1106 fDetTypeSim->SetTreeAddressD(digits,(Char_t*)GetName());
0421c3d1 1107
1108 AliITSDDLRawData rawWriter;
1109 //Verbose level
1110 // 0: Silent
1111 // 1: cout messages
1112 // 2: txt files with digits
1113 //BE CAREFUL, verbose level 2 MUST be used only for debugging and
1114 //it is highly suggested to use this mode only for debugging digits files
1115 //reasonably small, because otherwise the size of the txt files can reach
1116 //quickly several MB wasting time and disk space.
1117 rawWriter.SetVerbose(0);
1118
1119 //SILICON PIXEL DETECTOR
1120 Info("Digits2Raw", "Formatting raw data for SPD");
1121 rawWriter.RawDataSPD(digits->GetBranch("ITSDigitsSPD"));
1122
1123 //SILICON DRIFT DETECTOR
1124 Info("Digits2Raw", "Formatting raw data for SDD");
1125 rawWriter.RawDataSDD(digits->GetBranch("ITSDigitsSDD"));
1126
1127 //SILICON STRIP DETECTOR
1128 Info("Digits2Raw", "Formatting raw data for SSD");
1129 rawWriter.RawDataSSD(digits->GetBranch("ITSDigitsSSD"));
1130
1131 fLoader->UnloadDigits();
7d62fb64 1132
2aea926d 1133
e3b819ce 1134
7d62fb64 1135
2aea926d 1136}
88cb7938 1137
88cb7938 1138
828e06c7 1139
04366a57 1140
1141//______________________________________________________________________
aacedc3e 1142AliLoader* AliITS::MakeLoader(const char* topfoldername){
1143 //builds ITSgetter (AliLoader type)
1144 //if detector wants to use castomized getter, it must overload this method
1145
1146 Info("MakeLoader","Creating AliITSLoader. Top folder is %s.",
1147 topfoldername);
1148 fLoader = new AliITSLoader(GetName(),topfoldername);
1149 return fLoader;
2aea926d 1150}