DA output for test shuttle.
[u/mrichter/AliRoot.git] / VZERO / AliVZERODigitizer.cxx
CommitLineData
9e04c3b6 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
090026bf 16/* $Id$ */
17
b0d2c2d3 18///_________________________________________________________________________
19///
20/// This class constructs Digits out of Hits
21///
22///
9e04c3b6 23
24// --- Standard library ---
9e04c3b6 25
26// --- ROOT system ---
090026bf 27#include <TMath.h>
9e04c3b6 28#include <TTree.h>
fe0adf2a 29#include <TMap.h>
8bd3e27a 30#include <TGeoManager.h>
31#include <TGeoPhysicalNode.h>
32#include <AliGeomManager.h>
e939a978 33#include <TRandom.h>
9672d66e 34#include <TF1.h>
4e2652e8 35#include <TH1F.h>
9e04c3b6 36
37// --- AliRoot header files ---
9e04c3b6 38#include "AliRun.h"
39#include "AliVZERO.h"
40#include "AliVZEROhit.h"
9e04c3b6 41#include "AliRunLoader.h"
42#include "AliLoader.h"
fe0adf2a 43#include "AliGRPObject.h"
9e04c3b6 44#include "AliRunDigitizer.h"
ce7090f5 45#include "AliCDBManager.h"
46#include "AliCDBStorage.h"
47#include "AliCDBEntry.h"
48#include "AliVZEROCalibData.h"
4e2652e8 49#include "AliCTPTimeParams.h"
30d5d647 50#include "AliLHCClockPhase.h"
9e04c3b6 51#include "AliVZEROdigit.h"
b0d2c2d3 52#include "AliVZERODigitizer.h"
72da5ce8 53#include "AliVZEROSDigit.h"
9e04c3b6 54
55ClassImp(AliVZERODigitizer)
56
57 AliVZERODigitizer::AliVZERODigitizer()
fe0adf2a 58 :AliDigitizer(),
59 fCalibData(GetCalibData()),
60 fPhotoCathodeEfficiency(0.18),
fe0adf2a 61 fNdigits(0),
62 fDigits(0),
4e2652e8 63 fSignalShape(NULL),
64 fPMResponse(NULL),
72da5ce8 65 fSinglePhESpectrum(NULL),
66 fEvenOrOdd(kFALSE),
67 fTask(kHits2Digits),
68 fVZERO(NULL)
9e04c3b6 69{
b0d2c2d3 70 // default constructor
72da5ce8 71 // Initialize OCDB and containers used in the digitization
b0d2c2d3 72
72da5ce8 73 Init();
74}
4e2652e8 75
72da5ce8 76//____________________________________________________________________________
77 AliVZERODigitizer::AliVZERODigitizer(AliVZERO *vzero, DigiTask_t task)
78 :AliDigitizer(),
79 fCalibData(GetCalibData()),
80 fPhotoCathodeEfficiency(0.18),
81 fNdigits(0),
82 fDigits(0),
83 fSignalShape(NULL),
84 fPMResponse(NULL),
85 fSinglePhESpectrum(NULL),
86 fEvenOrOdd(kFALSE),
87 fTask(task),
88 fVZERO(vzero)
89{
90 // constructor
91 // Initialize OCDB and containers used in the digitization
4e2652e8 92
72da5ce8 93 Init();
9e04c3b6 94}
72da5ce8 95
9e04c3b6 96//____________________________________________________________________________
97 AliVZERODigitizer::AliVZERODigitizer(AliRunDigitizer* manager)
0b2bea8b 98 :AliDigitizer(manager),
99 fCalibData(GetCalibData()),
100 fPhotoCathodeEfficiency(0.18),
0b2bea8b 101 fNdigits(0),
fe0adf2a 102 fDigits(0),
4e2652e8 103 fSignalShape(NULL),
104 fPMResponse(NULL),
72da5ce8 105 fSinglePhESpectrum(NULL),
106 fEvenOrOdd(kFALSE),
107 fTask(kHits2Digits),
108 fVZERO(NULL)
9e04c3b6 109{
110 // constructor
4e2652e8 111 // Initialize OCDB and containers used in the digitization
72da5ce8 112
113 Init();
114}
115
116//____________________________________________________________________________
117 AliVZERODigitizer::~AliVZERODigitizer()
118{
119 // destructor
9e04c3b6 120
72da5ce8 121 if (fDigits) {
122 fDigits->Delete();
123 delete fDigits;
124 fDigits=0;
125 }
126
127 if (fSignalShape) {
128 delete fSignalShape;
129 fSignalShape = NULL;
130 }
131 if (fPMResponse) {
132 delete fPMResponse;
133 fPMResponse = NULL;
134 }
135 if (fSinglePhESpectrum) {
136 delete fSinglePhESpectrum;
137 fSinglePhESpectrum = NULL;
138 }
139
140 for(Int_t i = 0 ; i < 64; ++i) {
141 if (fTime[i]) delete [] fTime[i];
142 }
143}
144
145//_____________________________________________________________________________
146Bool_t AliVZERODigitizer::Init()
147{
148 // Initialises the digitizer
149 // Initialize OCDB and containers used in the digitization
150
151 // check if the digitizer was already initialized
152 if (fSignalShape) return kTRUE;
153
9672d66e 154 fSignalShape = new TF1("VZEROSignalShape",this,&AliVZERODigitizer::SignalShape,0,200,6,"AliVZERODigitizer","SignalShape");
4e2652e8 155 // fSignalShape->SetParameters(0,1.57345e1,-4.25603e-1,2.9,6.40982,3.69339e-01);
81db2705 156 // fSignalShape->SetParameters(1.34330e+00,1.13007e+02,-4.95705e-01,
157 // 3.68911e+00,1.01040e+00, 3.94675e-01);
158 fSignalShape->SetParameters(-1.07335e+00,2.16002e+01,-1.26133e-01,
159 1.41619e+00,5.50334e-01,3.86111e-01);
4e2652e8 160 fPMResponse = new TF1("VZEROPMResponse",this,&AliVZERODigitizer::PMResponse,-kPMRespTime,2.*kPMRespTime,0,"AliVZERODigitizer","PMResponse");
161 fSinglePhESpectrum = new TF1("VZEROSinglePhESpectrum",this,&AliVZERODigitizer::SinglePhESpectrum,0,20,0,"AliVZERODigitizer","SinglePhESpectrum");
ce7090f5 162
4e2652e8 163 // Now get the CTP L0->L1 delay
164 AliCDBEntry *entry = AliCDBManager::Instance()->Get("GRP/CTP/CTPtiming");
165 if (!entry) AliFatal("CTP timing parameters are not found in OCDB !");
166 AliCTPTimeParams *ctpParams = (AliCTPTimeParams*)entry->GetObject();
167 Float_t l1Delay = (Float_t)ctpParams->GetDelayL1L0()*25.0;
168
7905581c 169 AliCDBEntry *entry1 = AliCDBManager::Instance()->Get("GRP/CTP/TimeAlign");
170 if (!entry1) AliFatal("CTP time-alignment is not found in OCDB !");
171 AliCTPTimeParams *ctpTimeAlign = (AliCTPTimeParams*)entry1->GetObject();
172 l1Delay += ((Float_t)ctpTimeAlign->GetDelayL1L0()*25.0);
173
4e2652e8 174 AliCDBEntry *entry2 = AliCDBManager::Instance()->Get("VZERO/Calib/TimeDelays");
175 if (!entry2) AliFatal("VZERO time delays are not found in OCDB !");
176 TH1F *delays = (TH1F*)entry2->GetObject();
177
30d5d647 178 AliCDBEntry *entry3 = AliCDBManager::Instance()->Get("GRP/Calib/LHCClockPhase");
179 if (!entry3) AliFatal("LHC clock-phase shift is not found in OCDB !");
180 AliLHCClockPhase *phase = (AliLHCClockPhase*)entry3->GetObject();
181
4e2652e8 182 for(Int_t i = 0 ; i < 64; ++i) {
183
184 for(Int_t j = 0; j < kNClocks; ++j) fAdc[i][j] = 0;
185 fLeadingTime[i] = fTimeWidth[i] = 0;
186
187 fPmGain[i] = fCalibData->GetGain(i);
188
189 fAdcPedestal[i][0] = fCalibData->GetPedestal(i);
190 fAdcSigma[i][0] = fCalibData->GetSigma(i);
191 fAdcPedestal[i][1] = fCalibData->GetPedestal(i+64);
192 fAdcSigma[i][1] = fCalibData->GetSigma(i+64);
193
51980118 194 Int_t board = AliVZEROCalibData::GetBoardNumber(i);
4e2652e8 195 fNBins[i] = TMath::Nint(((Float_t)(fCalibData->GetMatchWindow(board)+1)*25.0+
196 (Float_t)kMaxTDCWidth*fCalibData->GetWidthResolution(board))/
197 fCalibData->GetTimeResolution(board));
198 fNBinsLT[i] = TMath::Nint(((Float_t)(fCalibData->GetMatchWindow(board)+1)*25.0)/
199 fCalibData->GetTimeResolution(board));
200 fBinSize[i] = fCalibData->GetTimeResolution(board);
87e3cdfb 201 fHptdcOffset[i] = (((Float_t)fCalibData->GetRollOver(board)-
202 (Float_t)fCalibData->GetTriggerCountOffset(board))*25.0+
7905581c 203 fCalibData->GetTimeOffset(i)-
30d5d647 204 l1Delay-
205 phase->GetMeanPhase()+
4e2652e8 206 delays->GetBinContent(i+1)+
207 kV0Offset);
72789100 208 fClockOffset[i] = (((Float_t)fCalibData->GetRollOver(board)-
209 (Float_t)fCalibData->GetTriggerCountOffset(board))*25.0+
210 fCalibData->GetTimeOffset(i)-
211 l1Delay+
212 kV0Offset);
4e2652e8 213
214 fTime[i] = new Float_t[fNBins[i]];
215 memset(fTime[i],0,fNBins[i]*sizeof(Float_t));
216 }
72da5ce8 217
218 return kTRUE;
9e04c3b6 219}
9672d66e 220
72da5ce8 221//____________________________________________________________________________
222void AliVZERODigitizer::Exec(Option_t* /*option*/)
223{
224 // Creates digits from hits
225 fNdigits = 0;
226
227 if (fVZERO && !fManager) {
228 AliLoader *loader = fVZERO->GetLoader();
229 if (!loader) {
230 AliError("Can not get VZERO Loader via AliVZERO object!");
231 return;
232 }
233 AliRunLoader* runLoader = AliRunLoader::Instance();
234 for (Int_t iEvent = 0; iEvent < runLoader->GetNumberOfEvents(); ++iEvent) {
235 runLoader->GetEvent(iEvent);
236 if (fTask == kHits2Digits) {
237 DigitizeHits();
238 DigitizeSDigits();
239 WriteDigits(loader);
240 }
241 else {
242 DigitizeHits();
243 WriteSDigits(loader);
244 }
245 }
4e2652e8 246 }
72da5ce8 247 else if (fManager) {
248 ReadSDigits();
249 DigitizeSDigits();
250 AliRunLoader *currentLoader = AliRunLoader::GetRunLoader(fManager->GetOutputFolderName());
251 AliLoader *loader = currentLoader->GetLoader("VZEROLoader");
252 if (!loader) {
253 AliError("Cannot get VZERO Loader via RunDigitizer!");
254 return;
255 }
256 WriteDigits(loader);
4e2652e8 257 }
72da5ce8 258 else {
259 AliFatal("Invalid digitization task! Exiting!");
4e2652e8 260 }
9e04c3b6 261}
262
72da5ce8 263//____________________________________________________________________________
264void AliVZERODigitizer::AddDigit(Int_t pmnumber, Float_t time, Float_t width, Bool_t integrator, Short_t *chargeADC, Int_t *labels)
265 {
266
267// Adds Digit
268
269 TClonesArray &ldigits = *fDigits;
270
271 new(ldigits[fNdigits++]) AliVZEROdigit(pmnumber,time,width,integrator,chargeADC,labels);
272
273}
274//____________________________________________________________________________
275void AliVZERODigitizer::AddSDigit(Int_t pmnumber, Int_t nbins, Float_t *charges, Int_t *labels)
276 {
277
278// Adds SDigit
279
280 TClonesArray &ldigits = *fDigits;
281
282 new(ldigits[fNdigits++]) AliVZEROSDigit(pmnumber,nbins,charges,labels);
283
284}
285//____________________________________________________________________________
286void AliVZERODigitizer::ResetDigits()
9e04c3b6 287{
9e04c3b6 288
72da5ce8 289// Clears Digits
290
291 fNdigits = 0;
292 if (fDigits) fDigits->Clear();
9e04c3b6 293}
294
295//____________________________________________________________________________
72da5ce8 296AliVZEROCalibData* AliVZERODigitizer::GetCalibData() const
297
298{
299 AliCDBManager *man = AliCDBManager::Instance();
300
301 AliCDBEntry *entry=0;
302
303 entry = man->Get("VZERO/Calib/Data");
304
305// if(!entry){
306// AliWarning("Load of calibration data from default storage failed!");
307// AliWarning("Calibration data will be loaded from local storage ($ALICE_ROOT)");
308// Int_t runNumber = man->GetRun();
309// entry = man->GetStorage("local://$ALICE_ROOT/OCDB")
310// ->Get("VZERO/Calib/Data",runNumber);
311//
312// }
313
314 // Retrieval of data in directory VZERO/Calib/Data:
315
316
317 AliVZEROCalibData *calibdata = 0;
318
319 if (entry) calibdata = (AliVZEROCalibData*) entry->GetObject();
320 if (!calibdata) AliFatal("No calibration data from calibration database !");
321
322 return calibdata;
323
324}
325
326double AliVZERODigitizer::SignalShape(double *x, double *par)
327{
328 // this function simulates the time
329 // of arrival of the photons at the
330 // photocathode
331 Double_t xx = x[0];
332 if (xx <= par[0]) return 0;
333 Double_t a = 1./TMath::Power((xx-par[0])/par[1],1./par[2]);
334 if (xx <= par[3]) return a;
335 Double_t b = 1./TMath::Power((xx-par[3])/par[4],1./par[5]);
336 Double_t f = a*b/(a+b);
337 AliDebug(100,Form("x=%f func=%f",xx,f));
338 return f;
339}
340
341double AliVZERODigitizer::PMResponse(double *x, double * /* par */)
342{
343 // this function describes the
344 // PM time response to a single
345 // photoelectron
346 Double_t xx = x[0]+kPMRespTime;
347 return xx*xx*TMath::Exp(-xx*xx/(kPMRespTime*kPMRespTime));
348}
349
350double AliVZERODigitizer::SinglePhESpectrum(double *x, double * /* par */)
351{
352 // this function describes the
353 // PM amplitude response to a single
354 // photoelectron
355 Double_t xx = x[0];
356 if (xx < 0) return 0;
357 return (TMath::Poisson(xx,kPMNbOfSecElec)+kPMTransparency*TMath::Poisson(xx,1.0));
358}
359
360Int_t AliVZERODigitizer::Cell2Pmt(Int_t cell) const
361{
362 // The method maps the scintillator
363 // indexes to the PM ones
364 if (cell < 0 || cell >= 80) {
365 AliError(Form("Wrong VZERO cell index %d",cell));
366 return -1;
367 }
368 if (cell < 16) return cell;
369 if (cell < 48) return 8 + cell/2;
370 return cell - 16;
371}
372
373void AliVZERODigitizer::DigitizeHits()
374{
375 // Digitize the hits to the level of
376 // SDigits (fTime arrays)
4e2652e8 377
378 for(Int_t i = 0 ; i < 64; ++i) {
379 memset(fTime[i],0,fNBins[i]*sizeof(Float_t));
72da5ce8 380 fLabels[i][0] = fLabels[i][1] = fLabels[i][2] = -1;
fad64858 381 }
4e2652e8 382 Float_t integral = fPMResponse->Integral(-kPMRespTime,2.*kPMRespTime);
383 Float_t meansPhE = fSinglePhESpectrum->Mean(0,20);
384
72da5ce8 385 AliLoader* loader = fVZERO->GetLoader();
8bd3e27a 386 if (!loader) {
72da5ce8 387 AliError("Can not get VZERO Loader!");
388 return;
389 }
8bd3e27a 390 loader->LoadHits();
391 TTree* treeH = loader->TreeH();
392 if (!treeH) {
72da5ce8 393 AliError("Cannot get TreeH!");
394 return;
395 }
396 TClonesArray* hits = fVZERO->Hits();
4e2652e8 397
b0d2c2d3 398// Now makes Digits from hits
8bd3e27a 399 Int_t nTracks = (Int_t) treeH->GetEntries();
400 for (Int_t iTrack = 0; iTrack < nTracks; iTrack++) {
72da5ce8 401 fVZERO->ResetHits();
8bd3e27a 402 treeH->GetEvent(iTrack);
403 Int_t nHits = hits->GetEntriesFast();
404 for (Int_t iHit = 0; iHit < nHits; iHit++) {
4e2652e8 405 AliVZEROhit* hit = (AliVZEROhit *)hits->UncheckedAt(iHit);
406 Int_t nPhot = hit->Nphot();
407 Int_t cell = hit->Cell();
408 Int_t pmt = Cell2Pmt(cell);
127e0d90 409 if (pmt < 0) continue;
30c4ff64 410 Int_t trackLabel = hit->GetTrack();
411 for(Int_t l = 0; l < 3; ++l) {
72da5ce8 412 if (fLabels[pmt][l] < 0) {
413 fLabels[pmt][l] = trackLabel;
30c4ff64 414 break;
415 }
416 }
4e2652e8 417 Float_t dt_scintillator = gRandom->Gaus(0,kIntTimeRes);
418 Float_t t = dt_scintillator + 1e9*hit->Tof();
419 if (pmt < 32) t += kV0CDelayCables;
420 t += fHptdcOffset[pmt];
421 Int_t nPhE;
3772afd5 422 Float_t prob = fCalibData->GetLightYields(pmt)*fPhotoCathodeEfficiency; // Optical losses included!
4e2652e8 423 if (nPhot > 100)
d0ff6548 424 nPhE = (Int_t)gRandom->Gaus(prob*Float_t(nPhot)+0.5,
425 sqrt(Float_t(nPhot)*prob*(1.-prob)));
4e2652e8 426 else
427 nPhE = gRandom->Binomial(nPhot,prob);
428 Float_t charge = TMath::Qe()*fPmGain[pmt]*fBinSize[pmt]/integral;
429 for (Int_t iPhE = 0; iPhE < nPhE; ++iPhE) {
430 Float_t tPhE = t + fSignalShape->GetRandom(0,fBinSize[pmt]*Float_t(fNBins[pmt]));
431 Float_t gainVar = fSinglePhESpectrum->GetRandom(0,20)/meansPhE;
432 Int_t firstBin = TMath::Max(0,(Int_t)((tPhE-kPMRespTime)/fBinSize[pmt]));
433 Int_t lastBin = TMath::Min(fNBins[pmt]-1,(Int_t)((tPhE+2.*kPMRespTime)/fBinSize[pmt]));
434 for(Int_t iBin = firstBin; iBin <= lastBin; ++iBin) {
435 Float_t tempT = fBinSize[pmt]*(0.5+iBin)-tPhE;
436 fTime[pmt][iBin] += gainVar*charge*fPMResponse->Eval(tempT);
437 }
438 } // ph.e. loop
439 } // hit loop
440 } // track loop
8bd3e27a 441 loader->UnloadHits();
72da5ce8 442}
443
444
445void AliVZERODigitizer::DigitizeSDigits()
446{
447 // Digitize the fTime arrays (SDigits) to the level of
448 // Digits (fAdc arrays)
449 for(Int_t i = 0 ; i < 64; ++i) {
450 for(Int_t j = 0; j < kNClocks; ++j) fAdc[i][j] = 0;
451 fLeadingTime[i] = fTimeWidth[i] = 0;
452 }
4e2652e8 453
454 Float_t maximum = 0.9*fSignalShape->GetMaximum(0,200); // Not exact, one needs to do this on the convoluted
455 Float_t integral2 = fSignalShape->Integral(0,200); // function. Anyway the effect is small <10% on the 2.5 ADC thr
456 for (Int_t ipmt = 0; ipmt < 64; ++ipmt) {
ed1ea59f 457 Float_t thr = fCalibData->GetCalibDiscriThr(ipmt,kFALSE)*kChargePerADC*maximum*fBinSize[ipmt]/integral2;
4e2652e8 458 Bool_t ltFound = kFALSE, ttFound = kFALSE;
459 for (Int_t iBin = 0; iBin < fNBins[ipmt]; ++iBin) {
b6fd9c4a 460 Float_t t = fBinSize[ipmt]*Float_t(iBin);
4e2652e8 461 if (fTime[ipmt][iBin] > thr) {
462 if (!ltFound && (iBin < fNBinsLT[ipmt])) {
463 ltFound = kTRUE;
464 fLeadingTime[ipmt] = t;
465 }
466 }
467 else {
468 if (ltFound) {
469 if (!ttFound) {
470 ttFound = kTRUE;
471 fTimeWidth[ipmt] = t - fLeadingTime[ipmt];
472 }
473 }
474 }
72789100 475 Float_t tadc = t - fClockOffset[ipmt];
d0ff6548 476 Int_t clock = kNClocks/2 - Int_t(tadc/25.0);
4e2652e8 477 if (clock >= 0 && clock < kNClocks)
478 fAdc[ipmt][clock] += fTime[ipmt][iBin]/kChargePerADC;
479 }
72789100 480 AliDebug(1,Form("Channel %d Offset %f Time %f",ipmt,fClockOffset[ipmt],fLeadingTime[ipmt]));
51980118 481 Int_t board = AliVZEROCalibData::GetBoardNumber(ipmt);
4e2652e8 482 if (ltFound && ttFound) {
483 fTimeWidth[ipmt] = fCalibData->GetWidthResolution(board)*
484 Float_t(Int_t(fTimeWidth[ipmt]/fCalibData->GetWidthResolution(board)));
485 if (fTimeWidth[ipmt] < Float_t(kMinTDCWidth)*fCalibData->GetWidthResolution(board))
486 fTimeWidth[ipmt] = Float_t(kMinTDCWidth)*fCalibData->GetWidthResolution(board);
487 if (fTimeWidth[ipmt] > Float_t(kMaxTDCWidth)*fCalibData->GetWidthResolution(board))
488 fTimeWidth[ipmt] = Float_t(kMaxTDCWidth)*fCalibData->GetWidthResolution(board);
489 }
490 }
b0d2c2d3 491
72da5ce8 492 fEvenOrOdd = gRandom->Integer(2);
4e2652e8 493 for (Int_t j=0; j<64; ++j){
494 for (Int_t iClock = 0; iClock < kNClocks; ++iClock) {
72da5ce8 495 Int_t integrator = (iClock + fEvenOrOdd) % 2;
72789100 496 AliDebug(1,Form("ADC %d %d %f",j,iClock,fAdc[j][iClock]));
4e2652e8 497 fAdc[j][iClock] += gRandom->Gaus(fAdcPedestal[j][integrator], fAdcSigma[j][integrator]);
498 }
499 }
20277079 500
72da5ce8 501}
502
503void AliVZERODigitizer::WriteDigits(AliLoader *loader)
504{
505 // Take fAdc arrays filled by the previous
506 // method and produce and add digits to the digit Tree
507
508 loader->LoadDigits("UPDATE");
4e2652e8 509
72da5ce8 510 if (!loader->TreeD()) loader->MakeTree("D");
511 loader->MakeDigitsContainer();
512 TTree* treeD = loader->TreeD();
513 DigitsArray();
514 treeD->Branch("VZERODigit", &fDigits);
515
30c4ff64 516 Short_t *chargeADC = new Short_t[kNClocks];
4e2652e8 517 for (Int_t i=0; i<64; i++) {
30c4ff64 518 for (Int_t j = 0; j < kNClocks; ++j) {
4e2652e8 519 Int_t tempadc = Int_t(fAdc[i][j]);
520 if (tempadc > 1023) tempadc = 1023;
30c4ff64 521 chargeADC[j] = tempadc;
4e2652e8 522 }
72da5ce8 523 AddDigit(i, fLeadingTime[i], fTimeWidth[i], Bool_t((10+fEvenOrOdd)%2), chargeADC, fLabels[i]);
4e2652e8 524 }
30c4ff64 525 delete [] chargeADC;
4e2652e8 526
b0d2c2d3 527 treeD->Fill();
72da5ce8 528 loader->WriteDigits("OVERWRITE");
529 loader->UnloadDigits();
530 ResetDigits();
9e04c3b6 531}
532
72da5ce8 533void AliVZERODigitizer::WriteSDigits(AliLoader *loader)
9e04c3b6 534{
72da5ce8 535 // Take fTime arrays filled by the previous
536 // method and produce and add sdigits to the sdigit Tree
fe0adf2a 537
72da5ce8 538 loader->LoadSDigits("UPDATE");
fe0adf2a 539
72da5ce8 540 if (!loader->TreeS()) loader->MakeTree("S");
541 loader->MakeSDigitsContainer();
542 TTree* treeS = loader->TreeS();
543 SDigitsArray();
544 treeS->Branch("VZEROSDigit", &fDigits);
545
546 for (Int_t ipmt = 0; ipmt < 64; ++ipmt) {
547 AddSDigit(ipmt,fNBins[ipmt],fTime[ipmt],fLabels[ipmt]);
548 }
ce7090f5 549
72da5ce8 550 treeS->Fill();
551 loader->WriteSDigits("OVERWRITE");
552 loader->UnloadSDigits();
553 ResetDigits();
554}
ce7090f5 555
72da5ce8 556void AliVZERODigitizer::ReadSDigits()
ce7090f5 557{
72da5ce8 558 // Read SDigits which are then to precessed
559 // in the following method
560 for(Int_t i = 0 ; i < 64; ++i) {
561 memset(fTime[i],0,fNBins[i]*sizeof(Float_t));
562 fLabels[i][0] = fLabels[i][1] = fLabels[i][2] = -1;
563 }
ce7090f5 564
72da5ce8 565 // Loop over input files
566 Int_t nFiles= fManager->GetNinputs();
567 for (Int_t inputFile = 0; inputFile < nFiles; inputFile++) {
568 // Get the current loader
569 AliRunLoader* currentLoader =
570 AliRunLoader::GetRunLoader(fManager->GetInputFolderName(inputFile));
ce7090f5 571
72da5ce8 572 AliLoader *loader = currentLoader->GetLoader("VZEROLoader");
573 loader->LoadSDigits("READ");
574
575 // Get the tree of summable digits
576 TTree* sdigitsTree = loader->TreeS();
577 if (!sdigitsTree) {
578 AliError("No sdigit tree from manager");
579 continue;
580 }
ce7090f5 581
72da5ce8 582 // Get the branch
583 TBranch* sdigitsBranch = sdigitsTree->GetBranch("VZEROSDigit");
584 if (!sdigitsBranch) {
585 AliError("Failed to get sdigit branch");
586 return;
587 }
4e2652e8 588
72da5ce8 589 // Set the branch address
590 TClonesArray *sdigitsArray = NULL;
591 sdigitsBranch->SetAddress(&sdigitsArray);
592
593 // Sum contributions from the sdigits
594 // Get number of entries in the tree
595 Int_t nentries = Int_t(sdigitsBranch->GetEntries());
596 for (Int_t entry = 0; entry < nentries; ++entry) {
597 sdigitsBranch->GetEntry(entry);
598 // Get the number of sdigits
599 Int_t nsdigits = sdigitsArray->GetEntries();
600 for (Int_t sdigit = 0; sdigit < nsdigits; sdigit++) {
601 AliVZEROSDigit* sDigit = static_cast<AliVZEROSDigit*>(sdigitsArray->UncheckedAt(sdigit));
602 Int_t pmNumber = sDigit->PMNumber();
603 Int_t nbins = sDigit->GetNBins();
604 if (nbins != fNBins[pmNumber]) {
605 AliError(Form("Incompatible number of bins between digitizer (%d) and sdigit (%d) for PM %d! Skipping sdigit!",
606 fNBins[pmNumber],nbins,pmNumber));
607 continue;
608 }
609 // Sum the charges
610 Float_t *charges = sDigit->GetCharges();
611 for(Int_t iBin = 0; iBin < nbins; ++iBin) fTime[pmNumber][iBin] += charges[iBin];
612 // and the labels
613 Int_t *labels = sDigit->GetTracks();
614 Int_t j = 0;
615 for(Int_t i = 0; i < 3; ++i) {
616 if (fLabels[pmNumber][i] < 0) {
617 if (labels[j] < 0) break;
618 fLabels[pmNumber][i] = labels[j];
619 j++;
620 }
621 }
622 }
623 }
624 loader->UnloadSDigits();
625 }
4e2652e8 626}
627
72da5ce8 628//____________________________________________________________________
629TClonesArray*
630AliVZERODigitizer::DigitsArray()
4e2652e8 631{
72da5ce8 632 // Initialize digit array if not already and
633 // return pointer to it.
634 if (!fDigits) {
635 fDigits = new TClonesArray("AliVZEROdigit", 64);
636 fNdigits = 0;
637 }
638 return fDigits;
4e2652e8 639}
640
72da5ce8 641//____________________________________________________________________
642TClonesArray*
643AliVZERODigitizer::SDigitsArray()
4e2652e8 644{
72da5ce8 645 // Initialize sdigit array if not already and
646 // return pointer to it.
647 if (!fDigits) {
648 fDigits = new TClonesArray("AliVZEROSDigit", 64);
649 fNdigits = 0;
4e2652e8 650 }
72da5ce8 651 return fDigits;
4e2652e8 652}