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 **************************************************************************/
20 #include <TObjArray.h>
22 #include <TDirectory.h>
23 #include <Riostream.h>
25 #include "AliTPCDigitizer.h"
28 #include "AliTPCParam.h"
29 #include "AliTPCParamSR.h"
32 #include "AliRunDigitizer.h"
33 #include "AliSimDigits.h"
36 #include "AliTPCcalibDB.h"
37 #include "AliTPCCalPad.h"
38 #include "AliTPCCalROC.h"
40 ClassImp(AliTPCDigitizer)
42 //___________________________________________
43 AliTPCDigitizer::AliTPCDigitizer() :AliDigitizer(),fDebug(0)
46 // Default ctor - don't use it
51 //___________________________________________
52 AliTPCDigitizer::AliTPCDigitizer(AliRunDigitizer* manager)
53 :AliDigitizer(manager),fDebug(0)
56 // ctor which should be used
58 AliDebug(2,"(AliRunDigitizer* manager) was processed");
61 //------------------------------------------------------------------------
62 AliTPCDigitizer::~AliTPCDigitizer()
69 //------------------------------------------------------------------------
70 Bool_t AliTPCDigitizer::Init()
78 //------------------------------------------------------------------------
79 void AliTPCDigitizer::Exec(Option_t* option)
83 //------------------------------------------------------------------------
84 void AliTPCDigitizer::ExecFast(Option_t* option)
87 // merge input tree's with summable digits
88 //output stored in TreeTPCD
91 TString optionString = option;
92 if (optionString.Data() == "deb") {
93 cout<<"AliTPCDigitizer::Exec: called with option deb "<<endl;
96 //get detector and geometry
99 AliRunLoader *rl, *orl;
100 AliLoader *gime, *ogime;
104 Warning("ExecFast","gAlice is NULL. Loading from input 0");
105 rl = AliRunLoader::GetRunLoader(fManager->GetInputFolderName(0));
108 Error("ExecFast","Can not find Run Loader for input 0. Can not proceed.");
114 AliTPC *pTPC = (AliTPC *) gAlice->GetModule("TPC");
115 AliTPCParam * param = pTPC->GetParam();
117 sprintf(s,param->GetTitle());
118 sprintf(ss,"75x40_100x60");
120 printf("2 pad-length geom hits with 3 pad-lenght geom digits...\n");
122 param=new AliTPCParamSR();
125 sprintf(ss,"75x40_100x60_150x60");
126 if(strcmp(s,ss)!=0) {
127 printf("No TPC parameters found...\n");
132 pTPC->GenerNoise(500000); //create teble with noise
134 Int_t nInputs = fManager->GetNinputs();
135 Int_t * masks = new Int_t[nInputs];
136 for (Int_t i=0; i<nInputs;i++)
137 masks[i]= fManager->GetMask(i);
138 Short_t **pdig= new Short_t*[nInputs]; //pointers to the expanded digits array
139 Int_t **ptr= new Int_t*[nInputs]; //pointers to teh expanded tracks array
140 Bool_t *active= new Bool_t[nInputs]; //flag for active input segments
143 //create digits array for given sectors
146 AliSimDigits ** digarr = new AliSimDigits*[nInputs];
147 for (Int_t i1=0;i1<nInputs; i1++)
151 rl = AliRunLoader::GetRunLoader(fManager->GetInputFolderName(i1));
152 gime = rl->GetLoader("TPCLoader");
153 gime->LoadSDigits("read");
154 TTree * treear = gime->TreeS();
158 cerr<<"AliTPCDigitizer: Input tree with SDigits not found in"
159 <<" input "<< i1<<endl;
163 if (treear->GetIndex()==0)
164 treear->BuildIndex("fSegmentID","fSegmentID");
165 treear->GetBranch("Segment")->SetAddress(&digarr[i1]);
169 //create branch's in TPC treeD
170 AliSimDigits * digrow = new AliSimDigits;
172 orl = AliRunLoader::GetRunLoader(fManager->GetOutputFolderName());
173 ogime = orl->GetLoader("TPCLoader");
175 TTree * tree = ogime->TreeD();
178 ogime->MakeTree("D");
179 tree = ogime->TreeD();
181 tree->Branch("Segment","AliSimDigits",&digrow);
184 param->SetZeroSup(2);
186 Int_t zerosup = param->GetZeroSup();
187 AliTPCCalPad * gainTPC = AliTPCcalibDB::Instance()->GetPadGainFactor();
188 AliTPCCalPad * noiseTPC = AliTPCcalibDB::Instance()->GetPadNoise();
190 //Loop over segments of the TPC
192 for (Int_t segmentID=0; segmentID<param->GetNRowsTotal(); segmentID++)
195 if (!param->AdjustSectorRow(segmentID,sec,row))
197 cerr<<"AliTPC warning: invalid segment ID ! "<<segmentID<<endl;
200 AliTPCCalROC * gainROC = gainTPC->GetCalROC(sec); // pad gains per given sector
201 AliTPCCalROC * noiseROC = noiseTPC->GetCalROC(sec); // noise per given sector
202 digrow->SetID(segmentID);
207 Bool_t digitize = kFALSE;
208 for (Int_t i=0;i<nInputs; i++)
211 rl = AliRunLoader::GetRunLoader(fManager->GetInputFolderName(i));
212 gime = rl->GetLoader("TPCLoader");
214 if (gime->TreeS()->GetEntryWithIndex(segmentID,segmentID) >= 0) {
215 digarr[i]->ExpandBuffer();
216 digarr[i]->ExpandTrackBuffer();
217 nrows = digarr[i]->GetNRows();
218 ncols = digarr[i]->GetNCols();
220 if (!fRegionOfInterest || (i == 0)) digitize = kTRUE;
224 if (fRegionOfInterest && !digitize) break;
226 if (!digitize) continue;
228 digrow->Allocate(nrows,ncols);
229 digrow->AllocateTrack(3);
232 Int_t label[1000]; //stack for 300 events
235 Int_t nElems = nrows*ncols;
237 for (Int_t i=0;i<nInputs; i++)
239 pdig[i] = digarr[i]->GetDigits();
240 ptr[i] = digarr[i]->GetTracks();
243 Short_t *pdig1= digrow->GetDigits();
244 Int_t *ptr1= digrow->GetTracks() ;
248 for (Int_t elem=0;elem<nElems; elem++)
254 for (Int_t i=0;i<nInputs; i++) if (active[i])
256 // q += digarr[i]->GetDigitFast(rows,col);
259 for (Int_t tr=0;tr<3;tr++)
261 // Int_t lab = digarr[i]->GetTrackIDFast(rows,col,tr);
262 Int_t lab = ptr[i][tr*nElems];
263 if ( (lab > 1) && *(pdig[i])>zerosup)
265 label[labptr]=lab+masks[i];
272 q/=16.; //conversion factor
273 Float_t gain = gainROC->GetValue(row,elem/nrows); // get gain for given - pad-row pad
278 Float_t noisePad = noiseROC->GetValue(row,elem/nrows);
279 // Float_t noise = gRandom->Gaus(0,param->GetNoise()*param->GetNoiseNormFac());
280 Float_t noise = pTPC->GetNoise();
285 if(q >= param->GetADCSat()) q = (Short_t)(param->GetADCSat() - 1);
286 //digrow->SetDigitFast((Short_t)q,rows,col);
288 for (Int_t tr=0;tr<3;tr++)
291 ptr1[tr*nElems] = label[tr];
298 digrow->CompresBuffer(1,zerosup);
299 digrow->CompresTrackBuffer(1);
301 if (fDebug>0) cerr<<sec<<"\t"<<row<<"\n";
302 } //for (Int_t n=0; n<param->GetNRowsTotal(); n++)
305 orl = AliRunLoader::GetRunLoader(fManager->GetOutputFolderName());
306 ogime = orl->GetLoader("TPCLoader");
307 ogime->WriteDigits("OVERWRITE");
309 //fManager->GetTreeDTPC()->Write(0,TObject::kOverwrite);
312 for (Int_t i1=0;i1<nInputs; i1++) delete digarr[i1];
322 //------------------------------------------------------------------------
323 void AliTPCDigitizer::ExecSave(Option_t* option)
326 // merge input tree's with summable digits
327 //output stored in TreeTPCD
329 TString optionString = option;
330 if (optionString.Data() == "deb") {
331 cout<<"AliTPCDigitizer::Exec: called with option deb "<<endl;
334 //get detector and geometry
335 AliRunLoader *rl, *orl;
336 AliLoader *gime, *ogime;
339 orl = AliRunLoader::GetRunLoader(fManager->GetOutputFolderName());
340 ogime = orl->GetLoader("TPCLoader");
342 rl = AliRunLoader::GetRunLoader(fManager->GetInputFolderName(0));
343 gime = rl->GetLoader("TPCLoader");
346 AliRun* alirun = rl->GetAliRun();
348 AliTPC *pTPC = (AliTPC *) alirun->GetModule("TPC");
349 AliTPCParam * param = pTPC->GetParam();
350 pTPC->GenerNoise(500000); //create teble with noise
351 printf("noise %f \n", param->GetNoise()*param->GetNoiseNormFac());
353 Int_t nInputs = fManager->GetNinputs();
354 Int_t * masks = new Int_t[nInputs];
355 for (Int_t i=0; i<nInputs;i++)
356 masks[i]= fManager->GetMask(i);
358 AliSimDigits ** digarr = new AliSimDigits*[nInputs];
359 for (Int_t i1=0;i1<nInputs; i1++)
363 rl = AliRunLoader::GetRunLoader(fManager->GetInputFolderName(i1));
364 gime = rl->GetLoader("TPCLoader");
366 TTree * treear = gime->TreeS();
367 TBranch * br = treear->GetBranch("fSegmentID");
368 if (br) br->GetFile()->cd();
370 cerr<<" TPC - not existing input = \n"<<i1<<" ";
372 treear->GetBranch("Segment")->SetAddress(&digarr[i1]);
375 rl = AliRunLoader::GetRunLoader(fManager->GetInputFolderName(0));
376 gime = rl->GetLoader("TPCLoader");
377 Stat_t nentries = gime->TreeS()->GetEntries();
380 //create branch's in TPC treeD
381 AliSimDigits * digrow = new AliSimDigits;
382 TTree * tree = ogime->TreeD();
384 tree->Branch("Segment","AliSimDigits",&digrow);
385 param->SetZeroSup(2);
387 Int_t zerosup = param->GetZeroSup();
388 //Loop over segments of the TPC
390 AliTPCCalPad * gainTPC = AliTPCcalibDB::Instance()->GetPadGainFactor();
391 AliTPCCalPad * noiseTPC = AliTPCcalibDB::Instance()->GetPadNoise();
392 for (Int_t n=0; n<nentries; n++) {
393 rl = AliRunLoader::GetRunLoader(fManager->GetInputFolderName(0));
394 gime = rl->GetLoader("TPCLoader");
395 gime->TreeS()->GetEvent(n);
397 digarr[0]->ExpandBuffer();
398 digarr[0]->ExpandTrackBuffer();
401 for (Int_t i=1;i<nInputs; i++){
402 // fManager->GetInputTreeTPCS(i)->GetEntryWithIndex(digarr[0]->GetID(),digarr[0]->GetID());
403 rl = AliRunLoader::GetRunLoader(fManager->GetInputFolderName(i));
404 gime = rl->GetLoader("TPCLoader");
405 gime->TreeS()->GetEntryWithIndex(digarr[0]->GetID(),digarr[0]->GetID());
406 digarr[i]->ExpandBuffer();
407 digarr[i]->ExpandTrackBuffer();
408 if ((digarr[0]->GetID()-digarr[i]->GetID())>0)
414 if (!param->AdjustSectorRow(digarr[0]->GetID(),sec,row)) {
415 cerr<<"AliTPC warning: invalid segment ID ! "<<digarr[0]->GetID()<<endl;
419 AliTPCCalROC * gainROC = gainTPC->GetCalROC(sec); // pad gains per given sector
420 AliTPCCalROC * noiseROC = noiseTPC->GetCalROC(sec); // noise per given sector
421 digrow->SetID(digarr[0]->GetID());
423 Int_t nrows = digarr[0]->GetNRows();
424 Int_t ncols = digarr[0]->GetNCols();
425 digrow->Allocate(nrows,ncols);
426 digrow->AllocateTrack(3);
429 Int_t label[1000]; //stack for 300 events
434 for (Int_t rows=0;rows<nrows; rows++){
435 for (Int_t col=0;col<ncols; col++){
440 for (Int_t i=0;i<nInputs; i++){
441 q += digarr[i]->GetDigitFast(rows,col);
444 for (Int_t tr=0;tr<3;tr++) {
445 Int_t lab = digarr[i]->GetTrackIDFast(rows,col,tr);
446 //Int_t lab = ptr[i][tr*nElems];
448 label[labptr]=lab+masks[i];
456 q/=16.; //conversion factor
457 // Float_t noise = gRandom->Gaus(0,param->GetNoise()*param->GetNoiseNormFac());
458 Float_t gain = gainROC->GetValue(row,col);
460 Float_t noisePad = noiseROC->GetValue(row, col);
462 Float_t noise = pTPC->GetNoise();
467 if(q >= param->GetADCSat()) q = (Short_t)(param->GetADCSat() - 1);
468 digrow->SetDigitFast((Short_t)q,rows,col);
469 // *pdig1 =Short_t(q);
470 for (Int_t tr=0;tr<3;tr++){
472 ((AliSimDigits*)digrow)->SetTrackIDFast(label[tr],rows,col,tr);
473 //ptr1[tr*nElems] = label[tr];
475 // ((AliSimDigits*)digrow)->SetTrackIDFast(-1,rows,col,tr);
476 // ptr1[tr*nElems] = 1;
484 digrow->CompresBuffer(1,zerosup);
485 digrow->CompresTrackBuffer(1);
487 if (fDebug>0) cerr<<sec<<"\t"<<row<<"\n";
489 // printf("end TPC merging - end -Tree %s\t%p\n",fManager->GetInputTreeH(0)->GetName(),fManager->GetInputTreeH(0)->GetListOfBranches()->At(3));
490 //fManager->GetTreeDTPC()->Write(0,TObject::kOverwrite);
491 ogime->WriteDigits("OVERWRITE");
493 for (Int_t i=1;i<nInputs; i++)
495 rl = AliRunLoader::GetRunLoader(fManager->GetInputFolderName(i));
496 gime = rl->GetLoader("TPCLoader");
497 gime->UnloadSDigits();
499 ogime->UnloadDigits();
502 for (Int_t i1=0;i1<nInputs; i1++) delete digarr[i1];