Accounted different btw. online and offline module numbering
[u/mrichter/AliRoot.git] / VZERO / AliVZEROReconstructor.cxx
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
16 /* $Id$ */
17
18 ///////////////////////////////////////////////////////////////////////////////
19 ///                                                                          //
20 /// class for VZERO reconstruction                                           //
21 ///                                                                          //
22 ///////////////////////////////////////////////////////////////////////////////
23
24 #include "AliRunLoader.h"
25 #include "AliRawReader.h"
26 #include "AliGRPObject.h"
27 #include "AliCDBManager.h"
28 #include "AliCDBStorage.h"
29 #include "AliCDBEntry.h"
30 #include "AliVZEROReconstructor.h"
31 #include "AliVZERORawStream.h"
32 #include "AliESDEvent.h"
33 #include "AliVZEROTriggerMask.h"
34 #include "AliESDfriend.h"
35 #include "AliESDVZEROfriend.h"
36 #include "AliVZEROdigit.h"
37 #include "AliVZEROCalibData.h"
38
39 ClassImp(AliVZEROReconstructor)
40
41 //_____________________________________________________________________________
42 AliVZEROReconstructor:: AliVZEROReconstructor(): AliReconstructor(),
43                         fESDVZERO(0x0),
44                         fESD(0x0),
45                         fESDVZEROfriend(0x0),
46                         fCalibData(GetCalibData()),
47                         fCollisionMode(0),
48                         fBeamEnergy(0.)
49 {
50   // Default constructor  
51   // Get calibration data
52   
53   // fCalibData = GetCalibData(); 
54
55 }
56
57
58 //_____________________________________________________________________________
59 AliVZEROReconstructor& AliVZEROReconstructor::operator = 
60   (const AliVZEROReconstructor& /*reconstructor*/)
61 {
62 // assignment operator
63
64   Fatal("operator =", "assignment operator not implemented");
65   return *this;
66 }
67
68 //_____________________________________________________________________________
69 AliVZEROReconstructor::~AliVZEROReconstructor()
70 {
71 // destructor
72
73    delete fESDVZERO;
74    delete fESDVZEROfriend;
75 }
76
77 //_____________________________________________________________________________
78 void AliVZEROReconstructor::Init()
79 {
80 // initializer
81
82   fESDVZERO  = new AliESDVZERO;
83   fESDVZEROfriend = new AliESDVZEROfriend;
84   
85   GetCollisionMode();  // fCollisionMode =1 for Pb-Pb simulated data
86 }
87
88 //______________________________________________________________________
89 void AliVZEROReconstructor::ConvertDigits(AliRawReader* rawReader, TTree* digitsTree) const
90 {
91 // converts RAW to digits - pedestal is subtracted 
92
93   if (!digitsTree) {
94     AliError("No digits tree!");
95     return;
96   }
97
98   TClonesArray* digitsArray = new TClonesArray("AliVZEROdigit");
99   digitsTree->Branch("VZERODigit", &digitsArray);
100
101   fESDVZEROfriend->Reset();
102
103   rawReader->Reset();
104   AliVZERORawStream rawStream(rawReader);
105   if (rawStream.Next()) {  
106      Float_t ADC_max[64], adc[64]; 
107      Bool_t BBFlag[64], BGFlag[64], integrator[64]; 
108      Float_t time[64], width[64];  
109      for(Int_t i=0; i<64; i++) {
110          // Search for the maximum charge in the train of 21 LHC clocks 
111          // regardless of the integrator which has been operated:
112          ADC_max[i] = 0.0;
113          Int_t imax = 0;
114          for(Int_t iClock=0; iClock<21; iClock++){
115              if(rawStream.GetPedestal(i,iClock) > ADC_max[i])  
116                 {ADC_max[i]= rawStream.GetPedestal(i,iClock);
117                  imax      = iClock;}
118          }
119          // Convert i (FEE channel numbering) to j (aliroot channel numbering)
120          Int_t j       =  rawStream.GetOfflineChannel(i);
121          Int_t board   = j / 8;
122          adc[j]        =  ADC_max[i];
123          time[j]       =  rawStream.GetTime(i)/ (25./256.) * fCalibData->GetTimeResolution(board);
124          width[j]      =  rawStream.GetWidth(i) / 0.4 * fCalibData->GetWidthResolution(board);
125          BBFlag[j]     =  rawStream.GetBBFlag(i,imax);
126          BGFlag[j]     =  rawStream.GetBGFlag(i,imax); 
127          integrator[j] =  rawStream.GetIntegratorFlag(i,imax); 
128
129          // Filling the esd friend object
130          fESDVZEROfriend->SetBBScalers(j,rawStream.GetBBScalers(i));
131          fESDVZEROfriend->SetBGScalers(j,rawStream.GetBGScalers(i));
132          for (Int_t iBunch = 0; iBunch < AliESDVZEROfriend::kNBunches; iBunch++) {
133              fESDVZEROfriend->SetChargeMB(j,iBunch,rawStream.GetChargeMB(i,iBunch));
134              fESDVZEROfriend->SetIntMBFlag(j,iBunch,rawStream.GetIntMBFlag(i,iBunch));
135              fESDVZEROfriend->SetBBMBFlag(j,iBunch,rawStream.GetBBMBFlag(i,iBunch));
136              fESDVZEROfriend->SetBGMBFlag(j,iBunch,rawStream.GetBGMBFlag(i,iBunch));
137          }
138          for (Int_t iEv = 0; iEv < AliESDVZEROfriend::kNEvOfInt; iEv++) {
139              fESDVZEROfriend->SetPedestal(j,iEv,rawStream.GetPedestal(i,iEv));
140              fESDVZEROfriend->SetIntegratorFlag(j,iEv,rawStream.GetIntegratorFlag(i,iEv));
141              fESDVZEROfriend->SetBBFlag(j,iEv,rawStream.GetBBFlag(i,iEv));
142              fESDVZEROfriend->SetBGFlag(j,iEv,rawStream.GetBGFlag(i,iEv));
143          }
144          fESDVZEROfriend->SetTime(j,time[j]);
145          fESDVZEROfriend->SetWidth(j,width[j]);
146      }  
147
148      // Filling the esd friend object
149      fESDVZEROfriend->SetTriggerInputs(rawStream.GetTriggerInputs());
150      fESDVZEROfriend->SetTriggerInputsMask(rawStream.GetTriggerInputsMask());
151
152      for(Int_t iScaler = 0; iScaler < AliESDVZEROfriend::kNScalers; iScaler++)
153          fESDVZEROfriend->SetTriggerScalers(iScaler,rawStream.GetTriggerScalers(iScaler));
154
155      for(Int_t iBunch = 0; iBunch < AliESDVZEROfriend::kNBunches; iBunch++)
156          fESDVZEROfriend->SetBunchNumbersMB(iBunch,rawStream.GetBunchNumbersMB(iBunch));
157      
158
159      // Channels(aliroot numbering) will be ordered in the tree
160      for(Int_t iChannel = 0; iChannel < 64; iChannel++) {
161          if(fCalibData->IsChannelDead(iChannel)){
162             adc[iChannel]  = (Float_t) kInvalidADC; 
163             time[iChannel] = (Float_t) kInvalidTime;     
164          }
165          new ((*digitsArray)[digitsArray->GetEntriesFast()])
166              AliVZEROdigit(iChannel, adc[iChannel], time[iChannel],
167                            width[iChannel], BBFlag[iChannel], BGFlag[iChannel],integrator[iChannel]);
168         
169      }          
170      digitsTree->Fill();
171   }
172 }
173
174 //______________________________________________________________________
175 void AliVZEROReconstructor::FillESD(TTree* digitsTree, TTree* /*clustersTree*/,
176                                     AliESDEvent* esd) const
177 {
178 // fills multiplicities to the ESD
179     
180   if (!digitsTree) {
181       AliError("No digits tree!");
182       return;
183   }
184
185   TClonesArray* digitsArray = NULL;
186   TBranch* digitBranch = digitsTree->GetBranch("VZERODigit");
187   digitBranch->SetAddress(&digitsArray);
188
189   Float_t   mult[64];  
190   Float_t    adc[64]; 
191   Float_t   time[64]; 
192   Float_t  width[64];
193   Bool_t  BBFlag[64];
194   Bool_t  BGFlag[64];
195    
196   for (Int_t i=0; i<64; i++){
197        adc[i]    = 0.0;
198        mult[i]   = 0.0;
199        time[i]   = 0.0;
200        width[i]  = 0.0;
201        BBFlag[i] = kFALSE;
202        BGFlag[i] = kFALSE;
203   }
204      
205   // loop over VZERO entries to get multiplicity
206   Int_t nEntries = (Int_t)digitsTree->GetEntries();
207   for (Int_t e=0; e<nEntries; e++) {
208     digitsTree->GetEvent(e);
209
210     Int_t nDigits = digitsArray->GetEntriesFast();
211     
212     for (Int_t d=0; d<nDigits; d++) {    
213         AliVZEROdigit* digit = (AliVZEROdigit*)digitsArray->At(d);      
214         Int_t  pmNumber      = digit->PMNumber(); 
215         // Pedestal retrieval and suppression: 
216         Float_t  pedestal      = fCalibData->GetPedestal(d);
217         adc[pmNumber]   =  digit->ADC() - pedestal; 
218         time[pmNumber]  =  digit->Time();
219         width[pmNumber] =  digit->Width();
220         BBFlag[pmNumber]=  digit->BBFlag();
221         BGFlag[pmNumber]=  digit->BGFlag();
222         // printf("PM = %d,  MIP per ADC channel = %f \n",pmNumber, fCalibData->GetMIPperADC(pmNumber));
223                 //AliInfo(Form("PM = %d,  ADC = %f TDC %f",pmNumber, digit->ADC(),digit->Time()));
224         // cut of ADC at 1MIP/2 
225         if(fCollisionMode >0) { 
226            Float_t MIP = 2.0;  
227            if (adc[pmNumber] > (MIP/2.) ) mult[pmNumber] += adc[pmNumber]*(1.0/MIP) ;
228            }
229            else{    
230            if (adc[pmNumber] > (1.0/fCalibData->GetMIPperADC(pmNumber) /2.) ) 
231                mult[pmNumber] += adc[pmNumber]*fCalibData->GetMIPperADC(pmNumber);
232         }           
233     } // end of loop over digits
234   } // end of loop over events in digits tree
235          
236   fESDVZERO->SetMultiplicity(mult);
237   fESDVZERO->SetADC(adc);
238   fESDVZERO->SetTime(time);
239   fESDVZERO->SetWidth(width);
240   fESDVZERO->SetBBFlag(BBFlag);
241   fESDVZERO->SetBGFlag(BGFlag);
242
243   // now get the trigger mask
244
245   AliVZEROTriggerMask *TriggerMask = new AliVZEROTriggerMask();
246   TriggerMask->SetAdcThreshold(10.0/2.0);
247   TriggerMask->SetTimeWindowWidthBBA(50);
248   TriggerMask->SetTimeWindowWidthBGA(20);
249   TriggerMask->SetTimeWindowWidthBBC(50);
250   TriggerMask->SetTimeWindowWidthBGC(20);
251   TriggerMask->FillMasks(digitsTree,digitsArray);
252
253   fESDVZERO->SetBBtriggerV0A(TriggerMask->GetBBtriggerV0A());
254   fESDVZERO->SetBGtriggerV0A(TriggerMask->GetBGtriggerV0A());
255   fESDVZERO->SetBBtriggerV0C(TriggerMask->GetBBtriggerV0C());
256   fESDVZERO->SetBGtriggerV0C(TriggerMask->GetBGtriggerV0C());
257   
258   if (esd) { 
259      AliDebug(1, Form("Writing VZERO data to ESD tree"));
260      esd->SetVZEROData(fESDVZERO);
261   }
262
263   if (esd) {
264      AliESDfriend *fr = (AliESDfriend*)esd->FindListObject("AliESDfriend");
265      if (fr) {
266         AliDebug(1, Form("Writing VZERO friend data to ESD tree"));
267         fr->SetVZEROfriend(fESDVZEROfriend);
268     }
269   }
270 }
271
272 //_____________________________________________________________________________
273 AliCDBStorage* AliVZEROReconstructor::SetStorage(const char *uri) 
274 {
275 // Sets the storage  
276
277   Bool_t deleteManager = kFALSE;
278   
279   AliCDBManager *manager = AliCDBManager::Instance();
280   AliCDBStorage *defstorage = manager->GetDefaultStorage();
281   
282   if(!defstorage || !(defstorage->Contains("VZERO"))){ 
283      AliWarning("No default storage set or default storage doesn't contain VZERO!");
284      manager->SetDefaultStorage(uri);
285      deleteManager = kTRUE;
286   }
287  
288   AliCDBStorage *storage = manager->GetDefaultStorage();
289
290   if(deleteManager){
291      AliCDBManager::Instance()->UnsetDefaultStorage();
292      defstorage = 0;   // the storage is killed by AliCDBManager::Instance()->Destroy()
293   }
294
295   return storage; 
296 }
297
298 //____________________________________________________________________________
299 void AliVZEROReconstructor::GetCollisionMode()
300 {
301 // Retrieves the collision mode from GRP data
302
303 // Initialization of the GRP entry 
304 // To be replace by a call to AliRunInfo object
305   
306   AliCDBEntry*  entry = AliCDBManager::Instance()->Get("GRP/GRP/Data");
307   AliGRPObject* grpData = 0x0;
308    
309   if(entry){
310     TMap* m = dynamic_cast<TMap*>(entry->GetObject());  // old GRP entry
311     if(m){
312        grpData = new AliGRPObject();
313        grpData->ReadValuesFromMap(m);
314     }
315     else{
316        grpData = dynamic_cast<AliGRPObject*>(entry->GetObject());  // new GRP entry
317     }
318   }
319
320   if(!grpData) { AliError("No GRP entry found in OCDB!");
321                   return; }
322
323 // Retrieval of simulated collision mode 
324
325   TString beamType = grpData->GetBeamType();
326   if(beamType==AliGRPObject::GetInvalidString()){
327      AliError("GRP/GRP/Data entry:  missing value for the beam type !");
328      AliError("\t VZERO cannot retrieve beam type\n");
329      return;
330   }
331
332    if( (beamType.CompareTo("P-P") ==0)  || (beamType.CompareTo("p-p") ==0) ){
333        fCollisionMode=0;
334   }
335    else if( (beamType.CompareTo("Pb-Pb") ==0)  || (beamType.CompareTo("A-A") ==0) ){
336        fCollisionMode=1;
337    }
338     
339   fBeamEnergy = grpData->GetBeamEnergy();
340   if(fBeamEnergy==AliGRPObject::GetInvalidFloat()) {
341      AliError("GRP/GRP/Data entry:  missing value for the beam energy ! Using 0");
342      fBeamEnergy = 0.;
343   }
344   
345 //    printf("\n ++++++ Beam type and collision mode retrieved as %s %d @ %1.3f GeV ++++++\n\n",beamType.Data(), fCollisionMode, fBeamEnergy);
346
347 }
348
349 //_____________________________________________________________________________
350 AliVZEROCalibData* AliVZEROReconstructor::GetCalibData() const
351 {
352   // Gets calibration object for VZERO set
353
354   AliCDBManager *man = AliCDBManager::Instance();
355
356   AliCDBEntry *entry=0;
357
358   entry = man->Get("VZERO/Calib/Data");
359
360 //   if(!entry){
361 //     AliWarning("Load of calibration data from default storage failed!");
362 //     AliWarning("Calibration data will be loaded from local storage ($ALICE_ROOT)");
363 //     Int_t runNumber = man->GetRun();
364 //     entry = man->GetStorage("local://$ALICE_ROOT/OCDB")
365 //       ->Get("VZERO/Calib/Data",runNumber);
366 //      
367 //   }
368
369   // Retrieval of data in directory VZERO/Calib/Data:
370
371   AliVZEROCalibData *calibdata = 0;
372
373   if (entry) calibdata = (AliVZEROCalibData*) entry->GetObject();
374   if (!calibdata)  AliFatal("No calibration data from calibration database !");
375
376   return calibdata;
377 }
378