1 /**************************************************************************
2 * Copyright(c) 1998-2000, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
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 **************************************************************************/
18 Revision 1.8 2002/10/23 07:17:33 alibrary
19 Introducing Riostream.h
21 Revision 1.7 2002/10/14 14:57:42 hristov
22 Merging the VirtualMC branch to the main development branch (HEAD)
24 Revision 1.3.6.1 2002/06/10 15:26:11 hristov
27 Revision 1.6 2002/04/06 14:41:04 kowal2
28 Added #include<stdlib.h> and log
37 #include <TObjArray.h>
39 #include <TDirectory.h>
40 #include <Riostream.h>
42 #include "AliTPCDigitizer.h"
45 #include "AliTPCParam.h"
46 #include "AliTPCParamSR.h"
49 #include "AliRunDigitizer.h"
50 #include "AliSimDigits.h"
53 ClassImp(AliTPCDigitizer)
55 //___________________________________________
56 AliTPCDigitizer::AliTPCDigitizer() :AliDigitizer()
58 // Default ctor - don't use it
62 //___________________________________________
63 AliTPCDigitizer::AliTPCDigitizer(AliRunDigitizer* manager)
64 :AliDigitizer(manager)
66 // ctor which should be used
69 cerr<<"AliTPCDigitizer::AliTPCDigitizer"
70 <<"(AliRunDigitizer* manager) was processed"<<endl;
73 //------------------------------------------------------------------------
74 AliTPCDigitizer::~AliTPCDigitizer()
81 //------------------------------------------------------------------------
82 Bool_t AliTPCDigitizer::Init()
90 //------------------------------------------------------------------------
91 void AliTPCDigitizer::Exec(Option_t* option)
95 //------------------------------------------------------------------------
96 void AliTPCDigitizer::ExecFast(Option_t* option)
99 // merge input tree's with summable digits
100 //output stored in TreeTPCD
103 TString optionString = option;
104 if (optionString.Data() == "deb") {
105 cout<<"AliTPCDigitizer::Exec: called with option deb "<<endl;
108 //get detector and geometry
109 AliTPC *pTPC = (AliTPC *) gAlice->GetModule("TPC");
110 AliTPCParam * param = pTPC->GetParam();
112 sprintf(s,param->GetTitle());
113 sprintf(ss,"75x40_100x60");
115 printf("2 pad-length geom hits with 3 pad-lenght geom digits...\n");
117 param=new AliTPCParamSR();
120 sprintf(ss,"75x40_100x60_150x60");
121 if(strcmp(s,ss)!=0) {
122 printf("No TPC parameters found...\n");
127 pTPC->GenerNoise(500000); //create teble with noise
129 Int_t nInputs = fManager->GetNinputs();
130 Int_t * masks = new Int_t[nInputs];
131 for (Int_t i=0; i<nInputs;i++)
132 masks[i]= fManager->GetMask(i);
133 Short_t **pdig= new Short_t*[nInputs]; //pointers to the expanded digits array
134 Int_t **ptr= new Int_t*[nInputs]; //pointers to teh expanded tracks array
136 //create digits array for given sectors
138 AliSimDigits ** digarr = new AliSimDigits*[nInputs];
139 for (Int_t i1=0;i1<nInputs; i1++){
142 TTree * treear = fManager->GetInputTreeTPCS(i1);
144 cerr<<"AliTPCDigitizer: Input tree with SDigits not found in"
145 <<" input "<< i1<<endl;
148 treear->GetBranch("Segment")->SetAddress(&digarr[i1]);
149 if (treear->GetIndex()==0)
150 treear->BuildIndex("fSegmentID","fSegmentID");
152 Stat_t nentries = fManager->GetInputTreeTPCS(0)->GetEntries();
155 //create branch's in TPC treeD
156 AliSimDigits * digrow = new AliSimDigits;
157 TTree * tree = fManager->GetTreeDTPC();
158 tree->Branch("Segment","AliSimDigits",&digrow);
161 param->SetZeroSup(2);
163 Int_t zerosup = param->GetZeroSup();
165 //Loop over segments of the TPC
167 for (Int_t n=0; n<nentries; n++) {
168 // for (Int_t n=0; n<300; n++) {
169 fManager->GetInputTreeTPCS(0)->GetEvent(n);
170 digarr[0]->ExpandBuffer();
171 digarr[0]->ExpandTrackBuffer();
173 for (Int_t i=1;i<nInputs; i++){
174 fManager->GetInputTreeTPCS(i)->GetEntryWithIndex(digarr[0]->GetID(),digarr[0]->GetID());
175 if ((digarr[0]->GetID()-digarr[i]->GetID())>0)
176 printf("problem - not corresponding segment in background event\n");
178 digarr[i]->ExpandBuffer();
179 digarr[i]->ExpandTrackBuffer();
182 if (!param->AdjustSectorRow(digarr[0]->GetID(),sec,row)) {
183 cerr<<"AliTPC warning: invalid segment ID ! "<<digarr[0]->GetID()<<endl;
187 digrow->SetID(digarr[0]->GetID());
189 Int_t nrows = digarr[0]->GetNRows();
190 Int_t ncols = digarr[0]->GetNCols();
191 digrow->Allocate(nrows,ncols);
192 digrow->AllocateTrack(3);
195 Int_t label[1000]; //stack for 300 events
198 Int_t nElems = nrows*ncols;
200 for (Int_t i=0;i<nInputs; i++){
201 pdig[i] = digarr[i]->GetDigits();
202 ptr[i] = digarr[i]->GetTracks();
204 Short_t *pdig1= digrow->GetDigits();
205 Int_t *ptr1= digrow->GetTracks() ;
209 // for (Int_t rows=0;rows<nrows; rows++){
210 // for (Int_t col=0;col<ncols; col++){
211 for (Int_t elem=0;elem<nElems; elem++){
212 //for (Int_t elem=nElems;elem<nElems; elem++){
217 for (Int_t i=0;i<nInputs; i++){
218 // q += digarr[i]->GetDigitFast(rows,col);
221 for (Int_t tr=0;tr<3;tr++) {
222 // Int_t lab = digarr[i]->GetTrackIDFast(rows,col,tr);
223 Int_t lab = ptr[i][tr*nElems];
224 if ( (lab > 1) && *(pdig[i])>zerosup) {
225 label[labptr]=lab+masks[i];
233 q/=16.; //conversion factor
234 // Float_t noise = gRandom->Gaus(0,param->GetNoise()*param->GetNoiseNormFac());
235 Float_t noise = pTPC->GetNoise();
240 if(q > param->GetADCSat()) q = (Short_t)(param->GetADCSat());
241 //digrow->SetDigitFast((Short_t)q,rows,col);
243 for (Int_t tr=0;tr<3;tr++){
245 // ((AliSimDigits*)digrow)->SetTrackIDFast(label[tr],rows,col,tr);
246 ptr1[tr*nElems] = label[tr];
248 // ((AliSimDigits*)digrow)->SetTrackIDFast(-1,rows,col,tr);
249 // ptr1[tr*nElems] = 1;
256 digrow->CompresBuffer(1,zerosup);
257 digrow->CompresTrackBuffer(1);
259 if (fDebug>0) cerr<<sec<<"\t"<<row<<"\n";
261 fManager->GetTreeDTPC()->Write(0,TObject::kOverwrite);
263 for (Int_t i1=0;i1<nInputs; i1++) delete digarr[i1];
270 //------------------------------------------------------------------------
271 void AliTPCDigitizer::ExecSave(Option_t* option)
274 // merge input tree's with summable digits
275 //output stored in TreeTPCD
277 TString optionString = option;
278 if (optionString.Data() == "deb") {
279 cout<<"AliTPCDigitizer::Exec: called with option deb "<<endl;
282 //get detector and geometry
283 printf("TPC merging -1 -Tree %s\t%p\n",fManager->GetInputTreeH(0)->GetName(),fManager->GetInputTreeH(0)->GetListOfBranches()->At(3));
284 AliTPC *pTPC = (AliTPC *) gAlice->GetModule("TPC");
285 AliTPCParam * param = pTPC->GetParam();
286 pTPC->GenerNoise(500000); //create teble with noise
287 printf("noise %f \n", param->GetNoise()*param->GetNoiseNormFac());
289 Int_t nInputs = fManager->GetNinputs();
290 Int_t * masks = new Int_t[nInputs];
291 for (Int_t i=0; i<nInputs;i++)
292 masks[i]= fManager->GetMask(i);
293 // Short_t **pdig= new Short_t*[nInputs]; //pointers to the expanded digits array
294 //Int_t **ptr= new Int_t*[nInputs]; //pointers to teh expanded tracks array
296 //create digits array for given sectors
298 printf("TPC merging -2 -Tree %s\t%p\n",fManager->GetInputTreeH(0)->GetName(),fManager->GetInputTreeH(0)->GetListOfBranches()->At(3));
299 AliSimDigits ** digarr = new AliSimDigits*[nInputs];
300 for (Int_t i1=0;i1<nInputs; i1++){
303 TTree * treear = fManager->GetInputTreeTPCS(i1);
304 printf("TPC merging -2.7 -Tree %s\t%p\n",fManager->GetInputTreeH(0)->GetName(),fManager->GetInputTreeH(0)->GetListOfBranches()->At(3));
305 TBranch * br = treear->GetBranch("fSegmentID");
306 if (br) br->GetFile()->cd();
307 printf("TPC merging -2.75 -Tree %s\t%p\n",fManager->GetInputTreeH(0)->GetName(),fManager->GetInputTreeH(0)->GetListOfBranches()->At(3));
308 //treear->BuildIndex("fSegmentID","fSegmentID");
310 cerr<<" TPC - not existing input = \n"<<i1<<" ";
312 printf("TPC merging -2.8 -Tree %s\t%p\n",fManager->GetInputTreeH(0)->GetName(),fManager->GetInputTreeH(0)->GetListOfBranches()->At(3));
313 treear->GetBranch("Segment")->SetAddress(&digarr[i1]);
314 printf("TPC merging -2.9 -Tree %s\t%p\n",fManager->GetInputTreeH(0)->GetName(),fManager->GetInputTreeH(0)->GetListOfBranches()->At(3));
316 Stat_t nentries = fManager->GetInputTreeTPCS(0)->GetEntries();
319 //create branch's in TPC treeD
320 AliSimDigits * digrow = new AliSimDigits;
321 TTree * tree = fManager->GetTreeDTPC();
322 //if (tree->GetBranch("Segment") ) tree->GetBranch("Segment")->SetAddress(&digrow);
324 tree->Branch("Segment","AliSimDigits",&digrow);
326 printf("TPC merging -3 -Tree %s\t%p\n",fManager->GetInputTreeH(0)->GetName(),fManager->GetInputTreeH(0)->GetListOfBranches()->At(3));
327 param->SetZeroSup(2);
329 Int_t zerosup = param->GetZeroSup();
330 //Loop over segments of the TPC
332 for (Int_t n=0; n<nentries; n++) {
333 // for (Int_t n=0; n<300; n++) {
334 fManager->GetInputTreeTPCS(0)->GetEvent(n);
335 digarr[0]->ExpandBuffer();
336 digarr[0]->ExpandTrackBuffer();
338 for (Int_t i=1;i<nInputs; i++){
339 fManager->GetInputTreeTPCS(i)->GetEntryWithIndex(digarr[0]->GetID(),digarr[0]->GetID());
340 //fManager->GetInputTreeTPCS(i)->GetEntryWithIndex(digarr[0]->GetID(),1);
341 digarr[i]->ExpandBuffer();
342 digarr[i]->ExpandTrackBuffer();
343 if ((digarr[0]->GetID()-digarr[i]->GetID())>0)
349 if (!param->AdjustSectorRow(digarr[0]->GetID(),sec,row)) {
350 cerr<<"AliTPC warning: invalid segment ID ! "<<digarr[0]->GetID()<<endl;
354 digrow->SetID(digarr[0]->GetID());
356 Int_t nrows = digarr[0]->GetNRows();
357 Int_t ncols = digarr[0]->GetNCols();
358 digrow->Allocate(nrows,ncols);
359 digrow->AllocateTrack(3);
362 Int_t label[1000]; //stack for 300 events
367 for (Int_t rows=0;rows<nrows; rows++){
368 for (Int_t col=0;col<ncols; col++){
373 for (Int_t i=0;i<nInputs; i++){
374 q += digarr[i]->GetDigitFast(rows,col);
377 for (Int_t tr=0;tr<3;tr++) {
378 Int_t lab = digarr[i]->GetTrackIDFast(rows,col,tr);
379 //Int_t lab = ptr[i][tr*nElems];
381 label[labptr]=lab+masks[i];
389 q/=16.; //conversion factor
390 // Float_t noise = gRandom->Gaus(0,param->GetNoise()*param->GetNoiseNormFac());
391 Float_t noise = pTPC->GetNoise();
396 if(q > param->GetADCSat()) q = (Short_t)(param->GetADCSat());
397 digrow->SetDigitFast((Short_t)q,rows,col);
398 // *pdig1 =Short_t(q);
399 for (Int_t tr=0;tr<3;tr++){
401 ((AliSimDigits*)digrow)->SetTrackIDFast(label[tr],rows,col,tr);
402 //ptr1[tr*nElems] = label[tr];
404 // ((AliSimDigits*)digrow)->SetTrackIDFast(-1,rows,col,tr);
405 // ptr1[tr*nElems] = 1;
413 digrow->CompresBuffer(1,zerosup);
414 digrow->CompresTrackBuffer(1);
416 if (fDebug>0) cerr<<sec<<"\t"<<row<<"\n";
418 printf("end TPC merging - end -Tree %s\t%p\n",fManager->GetInputTreeH(0)->GetName(),fManager->GetInputTreeH(0)->GetListOfBranches()->At(3));
419 fManager->GetTreeDTPC()->Write(0,TObject::kOverwrite);
421 for (Int_t i1=0;i1<nInputs; i1++) delete digarr[i1];