Possibility to specify event selection criteria within the raw-data input URI. The...
[u/mrichter/AliRoot.git] / ITS / AliITSBeamTestDigitizer.cxx
1 ////////////////////////////////////////////////////
2 //  Class to manage the                           //
3 //  ITS beam test conversion from rawdata         //
4 //  to digits. It executes the digitization for   //
5 //  SPD, SDD and SSD.                             //
6 //                                                //
7 //                                                //
8 //  Origin:  E. Crescio crescio@to.infn.it        //
9 //           J. Conrad  Jan.Conrad@cern.ch        //
10 ////////////////////////////////////////////////////
11 #include "AliHeader.h"
12 #include "AliRun.h"
13 #include "AliRunLoader.h"
14 #include "AliITSEventHeader.h"
15 #include "AliITSLoader.h"
16 #include "AliITSBeamTestDigSDD.h"
17 #include "AliITSBeamTestDigSPD.h"
18 #include "AliITSBeamTestDigSSD.h"
19 #include "AliITSBeamTestDigitizer.h"
20 #include "AliRawReaderDate.h"
21 #include "AliRawReaderRoot.h"
22
23 const TString AliITSBeamTestDigitizer::fgkDefaultDigitsFileName="ITS.Digits.root";  
24
25 ClassImp(AliITSBeamTestDigitizer)
26
27
28 //_____________________________________________________________
29 AliITSBeamTestDigitizer::AliITSBeamTestDigitizer():TTask(), 
30 fEvIn(0),
31 fEvFin(0),
32 fRunNumber(-1),
33 fDATEEvType(7),
34 fFlagHeader(kTRUE),
35 fFlagInit(kFALSE),
36 fOptDate(kFALSE),
37 fDigitsFileName(0),
38 fRawdataFileName(0),
39 fPeriod(kNov04),
40 fRunLoader(0),
41 fLoader(0),
42 fHeader(0){  
43   //
44   // Default constructor
45   //
46   SetFlagInit();
47   SetOptDate();
48
49
50 //_____________________________________________________________
51 AliITSBeamTestDigitizer::AliITSBeamTestDigitizer(const Text_t* name, const Text_t* title, Char_t* opt,const char* filename):TTask(name,title),
52 fEvIn(0),
53 fEvFin(0),
54 fRunNumber(-1),
55 fDATEEvType(7),
56 fFlagHeader(kTRUE),
57 fFlagInit(kFALSE),
58 fOptDate(kFALSE),
59 fDigitsFileName(0),
60 fRawdataFileName(0),
61 fPeriod(kNov04),
62 fRunLoader(0),
63 fLoader(0),
64 fHeader(0) 
65 {  
66   //
67   // Standard constructor 
68   //
69   SetOptDate();
70
71   TString choice(opt);
72   Bool_t aug04 = choice.Contains("Aug04");
73   Bool_t nov04 = choice.Contains("Nov04");
74   if(aug04) fPeriod=kAug04;
75   if(nov04) fPeriod=kNov04;
76   Init(filename);
77  } 
78
79 //_____________________________________________________________
80 AliITSBeamTestDigitizer::AliITSBeamTestDigitizer(const Text_t* name, const Text_t* title, Int_t run, Char_t* opt,const char* filename):TTask(name,title), 
81 fEvIn(0),
82 fEvFin(0),
83 fRunNumber(run),
84 fDATEEvType(7),
85 fFlagHeader(kTRUE),
86 fFlagInit(kFALSE),
87 fOptDate(kFALSE),
88 fDigitsFileName(0),
89 fRawdataFileName(0),
90 fPeriod(kNov04),
91 fRunLoader(0),
92 fLoader(0),
93 fHeader(0){  
94   //
95   // Constructor 
96   //
97   SetOptDate();
98   TString choice(opt);
99   Bool_t aug04 = choice.Contains("Aug04");
100   Bool_t nov04 = choice.Contains("Nov04");
101   if(aug04) fPeriod=kAug04;
102   if(nov04) fPeriod=kNov04;
103
104   Init(filename);
105  } 
106
107 //___________________________________________________________
108 void AliITSBeamTestDigitizer::Init(const char* filename){
109
110   //
111   //Initialization of run loader and its loader 
112   //creation of galice.root
113   //
114
115
116   fRunLoader = AliRunLoader::Open(filename,AliConfig::GetDefaultEventFolderName(),"update");
117   if (fRunLoader == 0x0)
118     {
119       Error("AliITSBeamTestDigitizer","Can not load the session",filename);
120       return;
121     }
122   fRunLoader->LoadgAlice();
123   gAlice = fRunLoader->GetAliRun();
124   
125   if(!gAlice) {
126     Error("AliITSBeamTestDigitizer","gAlice not found on file. Aborting.");
127     return;
128   } 
129   fRunLoader->MakeTree("E");  
130   fLoader = (AliITSLoader*)fRunLoader->GetLoader("ITSLoader");
131
132   fDigitsFileName=fgkDefaultDigitsFileName;
133   this->Add(new AliITSBeamTestDigSPD("DigSPD","SPD Digitization")); 
134   this->Add(new AliITSBeamTestDigSDD("DigSDD","SDD Digitization")); 
135   this->Add(new AliITSBeamTestDigSSD("DigSSD","SSD Digitization"));
136
137   SetFlagInit(kTRUE);
138 }
139
140
141 //______________________________________________________________________
142 AliITSBeamTestDigitizer::AliITSBeamTestDigitizer(const AliITSBeamTestDigitizer &bt):TTask(bt),
143 fEvIn(bt.fEvIn),
144 fEvFin(bt.fEvFin),
145 fRunNumber(bt.fRunNumber),
146 fDATEEvType(bt.fDATEEvType),
147 fFlagHeader(bt.fFlagHeader),
148 fFlagInit(bt.fFlagInit),
149 fOptDate(bt.fOptDate),
150 fDigitsFileName(bt.fDigitsFileName),
151 fRawdataFileName(bt.fRawdataFileName),
152 fPeriod(bt.fPeriod),
153 fRunLoader(bt.fRunLoader),
154 fLoader(bt.fLoader),
155 fHeader(bt.fHeader){
156   // Copy constructor. 
157 }
158 //______________________________________________________________________
159 AliITSBeamTestDigitizer& AliITSBeamTestDigitizer::operator=(const AliITSBeamTestDigitizer &source){
160     // Assignment operator. 
161   this->~AliITSBeamTestDigitizer();
162   new(this) AliITSBeamTestDigitizer(source);
163   return *this;
164 }
165
166
167 //______________________________________________________________
168 AliITSBeamTestDigitizer::~AliITSBeamTestDigitizer(){
169
170   //Destructor
171   //  if(fBt) delete fBt;
172   if(fLoader) delete fLoader;
173   if(fHeader) delete fHeader;
174
175
176
177 //_____________________________________________________________
178 void AliITSBeamTestDigitizer::SetNumberOfEventsPerFile(Int_t nev)
179 {
180   //Sets number of events per file
181
182   if(fRunLoader) fRunLoader->SetNumberOfEventsPerFile(nev);
183   else Warning("SetNumberOfEventsPerFile","fRunLoader is 0");
184 }
185
186
187 //____________________________________________________
188 void AliITSBeamTestDigitizer::ExecDigitization(){
189
190   // Execution of digitisation for SPD,SDD and SSD
191
192   if(!GetFlagInit()){
193     Warning("ExecDigitization()","Run Init() please..");
194     return;
195   }
196   fLoader->SetDigitsFileName(fDigitsFileName);
197   fLoader->LoadDigits("recreate");
198  
199   AliRawReader* rd;
200
201
202   TString strFile = fRawdataFileName;
203   strFile += Form("?EventType=%d",fDATEEvType);
204   if(GetOptDate()) rd = new AliRawReaderDate(strFile.Data(),fEvIn);
205   else rd = new AliRawReaderRoot(strFile.Data(),fEvIn);
206
207   AliHeader* header = fRunLoader->GetHeader();
208   Int_t iev=fEvIn-1;
209
210   
211   AliITSBeamTestDigSDD* digSDD = (AliITSBeamTestDigSDD*)fTasks->FindObject("DigSDD");
212   AliITSBeamTestDigSPD* digSPD = (AliITSBeamTestDigSPD*)fTasks->FindObject("DigSPD");
213   AliITSBeamTestDigSSD* digSSD = (AliITSBeamTestDigSSD*)fTasks->FindObject("DigSSD");
214
215
216   do{
217     iev++;
218     if(fEvFin!=0){
219       if(iev>fEvFin) break;
220     } 
221     AliITSEventHeader* itsh = new AliITSEventHeader("ITSHeader");
222     fRunLoader->SetEventNumber(iev);
223    
224     rd->RequireHeader(fFlagHeader);
225  
226     digSDD->SetRawReader(rd);
227     digSPD->SetRawReader(rd);
228     digSSD->SetRawReader(rd);
229     
230     if(fLoader->TreeD() == 0x0) fLoader->MakeTree("D");
231
232     TTree* treeD = (TTree*)fLoader->TreeD();
233    
234     // Make branches outside the dig-classes
235
236     TClonesArray* digitsSPD = new TClonesArray("AliITSdigitSPD",1000);
237     treeD->Branch("ITSDigitsSPD",&digitsSPD);
238  
239     TClonesArray* digitsSDD = new TClonesArray("AliITSdigitSDD",1000);
240     treeD->Branch("ITSDigitsSDD",&digitsSDD);
241    
242     TClonesArray* digitsSSD = new TClonesArray("AliITSdigitSSD",1000);
243     treeD->Branch("ITSDigitsSSD",&digitsSSD);
244
245
246     digSSD->SetTree(treeD);
247     digSDD->SetTree(treeD);
248     digSPD->SetTree(treeD);
249     
250     AliITSgeom* geom = fLoader->GetITSgeom();
251
252     digSSD->SetITSgeom(geom);
253     digSDD->SetITSgeom(geom);
254     digSPD->SetITSgeom(geom);
255     
256     digSSD->SetITSEventHeader(itsh);
257     digSDD->SetITSEventHeader(itsh);
258     digSPD->SetITSEventHeader(itsh);
259
260     digSDD->SetBtPeriod(GetBeamTestPeriod());
261     if(GetBeamTestPeriod()==1)digSDD->SetThreshold(16);
262     else digSDD->SetThreshold(0);
263     ExecuteTask(0);  
264
265     header->SetEventNrInRun(iev);
266     header->SetEvent(iev);
267     header->SetRun(fRunNumber);
268     fRunLoader->GetHeader()->AddDetectorEventHeader(itsh);
269     fRunLoader->TreeE()->Fill();
270     header->Reset(fRunNumber,iev);
271     
272     delete digitsSPD;
273     delete digitsSDD;
274     delete digitsSSD;
275
276    }while(rd->NextEvent());
277
278   
279   fRunLoader->WriteHeader("OVERWRITE");
280   fRunLoader->WriteRunLoader("OVERWRITE");
281
282   delete rd;
283   fLoader->UnloadDigits();
284   fLoader->UnloadRawClusters();
285   fRunLoader->UnloadHeader();
286
287   
288 }
289
290
291
292 //_______________________________________________
293 void AliITSBeamTestDigitizer:: SetActive(const TString& subdet,Bool_t value){
294
295   //Sets active sub-tasks (detectors)
296   
297   Bool_t sdd = subdet.Contains("SDD");
298   Bool_t spd = subdet.Contains("SPD");
299   Bool_t ssd = subdet.Contains("SSD");
300
301   if(sdd){
302   AliITSBeamTestDigSDD* digSDD = (AliITSBeamTestDigSDD*)fTasks->FindObject("DigSDD");
303   digSDD->SetActive(value);
304   }
305   
306  if(spd){
307   AliITSBeamTestDigSPD* digSPD = (AliITSBeamTestDigSPD*)fTasks->FindObject("DigSPD");
308   digSPD->SetActive(value);
309   
310   }
311  
312   if(ssd){
313   AliITSBeamTestDigSSD* digSSD = (AliITSBeamTestDigSSD*)fTasks->FindObject("DigSSD");
314   digSSD->SetActive(value);
315   
316   }
317
318
319
320 }
321