Avoid setting the default storage in the normal reconstruction chain (Laurent, rev...
[u/mrichter/AliRoot.git] / MUON / AliMUONTrackerDataMaker.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 /// \class AliMUONTrackerDataMaker
19 /// 
20 /// Implementation of VTrackerDataMaker to read raw data and 
21 /// calibrate it (if required)
22 /// 
23 /// \author Laurent Aphecetche, Subatech
24
25 #include "AliMUONTrackerDataMaker.h"
26
27 #include "AliCDBManager.h"
28 #include "AliCDBStorage.h"
29 #include "AliCodeTimer.h"
30 #include "AliDAQ.h"
31 #include "AliLog.h"
32 #include "AliMUON2DMap.h"
33 #include "AliMUONCalibParamND.h"
34 #include "AliMUONCalibrationData.h"
35 #include "AliMUONDigitCalibrator.h"
36 #include "AliMUONRawStreamTrackerHP.h"
37 #include "AliMUONTrackerData.h"
38 #include "AliMpDDLStore.h"
39 #include "AliRawEventHeaderBase.h"
40 #include "AliRawReader.h"
41 #include "Riostream.h"
42
43 /// \cond CLASSIMP
44 ClassImp(AliMUONTrackerDataMaker)
45 /// \endcond
46
47 Int_t AliMUONTrackerDataMaker::fgkCounter(0);
48
49 //_____________________________________________________________________________
50 AliMUONTrackerDataMaker::AliMUONTrackerDataMaker(TRootIOCtor*) 
51
52 AliMUONVTrackerDataMaker(),
53 fRawReader(0x0),
54 fAccumulatedData(0x0),
55 fOneEventData(0x0),
56 fDigitCalibrator(0x0),
57 fCalibrationData(0x0), 
58 fSource(""),
59 fOCDBPath(""),
60 fNumberOfEvents(0),
61 fRunNumber(0),
62 fIsRunning(kFALSE),
63 fIsOwnerOfRawReader(kFALSE),
64 fIsEventByEvent(kFALSE)
65 {
66 /// Root IO ctor
67 }
68
69 //_____________________________________________________________________________
70 AliMUONTrackerDataMaker::AliMUONTrackerDataMaker(const AliMUONRecoParam* recoParam,
71                                                  Int_t runNumber,
72                                                  AliRawReader* rawReader,
73                                                  const char* cdbPath,
74                                                  const char* calibMode,
75                                                  Bool_t histogram,
76                                                  Double_t xmin,
77                                                  Double_t xmax)
78 :
79 AliMUONVTrackerDataMaker(),
80 fRawReader(rawReader),
81 fAccumulatedData(0x0),
82 fOneEventData(new AliMUON2DMap(true)),
83 fDigitCalibrator(0x0),
84 fCalibrationData(0x0), 
85 fSource(""),
86 fOCDBPath(cdbPath),
87 fNumberOfEvents(0),
88 fRunNumber(runNumber),
89 fIsRunning(kFALSE),
90 fIsOwnerOfRawReader(kFALSE),
91 fIsEventByEvent(kFALSE)
92 {
93   /// Ctor in which this object will NOT be the owner of the reader
94   /// and can NOT apply rewind to it, nor use Next on it. 
95   Ctor(recoParam,runNumber,calibMode,histogram,xmin,xmax);
96 }
97
98
99 //_____________________________________________________________________________
100 AliMUONTrackerDataMaker::AliMUONTrackerDataMaker(const AliMUONRecoParam* recoParam,
101                                                  AliRawReader* rawReader,
102                                                  const char* cdbPath,
103                                                  const char* calibMode,
104                                                  Bool_t histogram,
105                                                  Double_t xmin,
106                                                  Double_t xmax)
107 :
108 AliMUONVTrackerDataMaker(),
109 fRawReader(rawReader),
110 fAccumulatedData(0x0),
111 fOneEventData(new AliMUON2DMap(true)),
112 fDigitCalibrator(0x0),
113 fCalibrationData(0x0), 
114 fSource(""),
115 fOCDBPath(cdbPath),
116 fNumberOfEvents(0),
117 fRunNumber(0),
118 fIsRunning(kFALSE),
119 fIsOwnerOfRawReader(kTRUE),
120 fIsEventByEvent(kFALSE)
121 {
122   /// Ctor in which we take the ownership of the rawReader, so we can rewind
123   /// and advance it as we wish
124   
125   if (fRawReader) 
126   {
127     fRawReader->NextEvent(); // to be sure to get run number available
128     fRunNumber = fRawReader->GetRunNumber();
129     fRawReader->RewindEvents();
130   }
131   
132   Ctor(recoParam,fRunNumber,calibMode,histogram,xmin,xmax);
133 }
134
135 //_____________________________________________________________________________
136 AliMUONTrackerDataMaker::AliMUONTrackerDataMaker(AliRawReader* rawReader, Bool_t histogram)
137 :
138 AliMUONVTrackerDataMaker(),
139 fRawReader(rawReader),
140 fAccumulatedData(0x0),
141 fOneEventData(new AliMUON2DMap(true)),
142 fDigitCalibrator(0x0),
143 fCalibrationData(0x0), 
144 fSource(""),
145 fOCDBPath(""),
146 fNumberOfEvents(0),
147 fRunNumber(0),
148 fIsRunning(kFALSE),
149 fIsOwnerOfRawReader(kTRUE),
150 fIsEventByEvent(kFALSE)
151 {
152   if (fRawReader) 
153   {
154     fRawReader->NextEvent(); // to be sure to get run number available
155     fRunNumber = fRawReader->GetRunNumber();
156     fRawReader->RewindEvents();
157   }
158   
159   Ctor(0x0,fRunNumber,"",histogram);
160   
161 }
162
163 //_____________________________________________________________________________
164 void 
165 AliMUONTrackerDataMaker::Ctor(const AliMUONRecoParam* recoParam,
166                               Int_t runNumber,
167                               const char* calibMode,
168                               Bool_t histogram,
169                               Double_t xmin, Double_t xmax)
170 {
171   /// "designated constructor"
172   Bool_t calibrate = ( fOCDBPath.Length() > 0 );
173   
174   TString name;
175   TString type("RAW");
176   
177   if ( calibrate ) 
178   {
179     TString scalib(calibMode);
180     scalib.ToUpper();
181     if ( scalib == "GAIN" ) type = "CALC";
182     if ( scalib == "NOGAIN" ) type = "CALZ";
183     if ( scalib == "GAINCONSTANTCAPA") type = "CALG";
184     if ( scalib == "INJECTIONGAIN" ) type = "CALE";
185   }
186   
187   if ( !fRunNumber ) 
188   {
189     ++fgkCounter;
190     name = Form("%s%s_%d",(histogram?"H":""),type.Data(),fgkCounter);
191   }
192   else
193   {
194     name = Form("%s%s%d",(histogram?"H":""),type.Data(),fRunNumber);
195   }
196   
197   fAccumulatedData = new AliMUONTrackerData(name.Data(),"charge values",1);
198   fAccumulatedData->SetDimensionName(0,(calibrate ? "Calibrated charge" : "Raw charge"));
199   if (histogram)
200   {
201     fAccumulatedData->MakeHistogramForDimension(0,kTRUE,xmin,xmax);
202   }
203   
204   if ( calibrate ) 
205   {
206     fCalibrationData = new AliMUONCalibrationData(runNumber);
207     
208     // force the reading of calibration NOW
209     // FIXME: not really elegant and error prone (as we have the list of calib data twice, 
210     // once here and once in the digitcalibrator class, hence the change of them getting
211     // out of sync)
212     // But with the current CDBManager implementation, I don't know how to solve
213     // this better (e.g. to avoid clearing cache messages and so on).
214
215     AliCDBStorage* storage(0x0);
216     
217     if ( fOCDBPath.Length() > 0 )
218     {
219       storage = AliCDBManager::Instance()->GetDefaultStorage();
220
221       if ( storage && ( storage->GetURI() != fOCDBPath.Data() ) )
222       {
223         AliCDBManager::Instance()->SetDefaultStorage(fOCDBPath.Data());
224       }
225     }
226     
227     fCalibrationData->Pedestals();
228     fCalibrationData->Gains();
229     fCalibrationData->Neighbours();
230     fCalibrationData->HV();
231     fCalibrationData->Capacitances();
232     
233     if ( storage && ( storage->GetURI() != fOCDBPath.Data() ) )
234     {
235       AliCDBManager::Instance()->SetDefaultStorage(storage);
236     }
237     
238     fDigitCalibrator = new AliMUONDigitCalibrator(*fCalibrationData,recoParam,calibMode);
239     //FIXME: get the reco param from GUI and/or from OCDB if not used from the QA code ?
240   }
241 }
242
243 //_____________________________________________________________________________
244 AliMUONTrackerDataMaker::~AliMUONTrackerDataMaker()
245 {
246 /// dtor
247
248   delete fOneEventData;
249   delete fAccumulatedData;
250   if ( fIsOwnerOfRawReader ) delete fRawReader;
251   delete fCalibrationData;
252   delete fDigitCalibrator;
253 }
254
255 //_____________________________________________________________________________
256 Bool_t 
257 AliMUONTrackerDataMaker::Add(const AliMUONTrackerDataMaker& other)
258 {
259   /// Adds other to this
260     
261   if (!fAccumulatedData) return kFALSE;
262   
263   if ( fIsEventByEvent )
264   {
265     AliError("Cannot add event by event objects !");
266     return kFALSE;
267   }
268   
269   if ( fRunNumber != other.fRunNumber ) fRunNumber = -1;
270   
271   fSource += "\n";
272   fSource += other.fSource;
273   
274   fNumberOfEvents += other.fNumberOfEvents;
275   
276   TList list;
277   list.Add(other.fAccumulatedData);
278   
279   fAccumulatedData->Merge(&list);
280   
281   return kTRUE;
282 }
283
284 //_____________________________________________________________________________
285 Bool_t 
286 AliMUONTrackerDataMaker::NextEvent()
287 {
288   /// Read and process next event
289   
290   if ( !fIsOwnerOfRawReader ) 
291   {
292     AliError("I'm not the owner of the raw reader. Cannot use NextEvent");
293     return kFALSE;
294   }
295   
296   AliCodeTimerAuto("");
297   
298   static Int_t nphysics(0);
299   static Int_t ngood(0);
300   
301   if ( !IsRunning() ) return kTRUE;
302   
303   Bool_t ok = fRawReader->NextEvent();
304   
305   if (!ok) 
306   {
307     return kFALSE;
308   }
309   
310   Int_t eventType = fRawReader->GetType();
311   
312   ++fNumberOfEvents;
313   
314   if (eventType != AliRawEventHeaderBase::kPhysicsEvent ) 
315   {
316     return kTRUE; // for the moment
317   }
318   
319   ++nphysics;
320   
321         Bool_t pok = ProcessEvent();
322         
323         if ( pok ) ++ngood;
324         
325         AliDebug(1,Form("n %10d nphysics %10d ngood %10d",fNumberOfEvents,nphysics,ngood));
326         
327         return kTRUE;  
328 }
329
330 //_____________________________________________________________________________
331 Bool_t AliMUONTrackerDataMaker::ProcessEvent()
332 {
333   /// Process current event 
334   /// 
335   /// Note that in case of calibration, we do not simply reuse the 
336   /// AliMUONDigitCalibrator::Calibrate(AliMUONVDigitStore&) method, 
337   /// as this would require filling first a digitStore, and then calibrate it,
338   /// and then convert it into a VStore, all this taking too much time.
339   /// But we *do* reuse the AliMUONDigitCalibrator::CalibrateDigit in order not to 
340   /// duplicate this critical piece of calibration code !
341   ///
342   
343   AliCodeTimerAuto("");
344   
345   AliMUONRawStreamTrackerHP stream(fRawReader);
346   
347   stream.DisableWarnings();
348 //  stream.EnabbleErrorLogger();
349   
350   const Int_t nddls = AliDAQ::NumberOfDdls("MUONTRK");
351   TArrayI nevents(nddls);
352   
353   for ( Int_t i = 0; i < nddls; ++i ) 
354   {
355     nevents[i] = 0;
356   }
357   
358   fOneEventData->Clear();
359   
360   Int_t buspatchId;
361   UShort_t  manuId;
362   UChar_t manuChannel;
363   UShort_t adc;
364   
365   stream.First();
366   
367   while ( stream.Next(buspatchId,manuId,manuChannel,adc,kTRUE) )
368   {    
369     Int_t detElemId = AliMpDDLStore::Instance()->GetDEfromBus(buspatchId);
370     
371     Int_t ddl = AliMpDDLStore::Instance()->GetDDLfromBus(buspatchId);
372     
373     nevents[ddl] = 1;
374     
375     AliMUONVCalibParam* param = static_cast<AliMUONVCalibParam*>(fOneEventData->FindObject(detElemId,manuId));
376     if (!param)
377     {
378       param = new AliMUONCalibParamND(1,64,detElemId,manuId,
379                                       AliMUONVCalibParam::InvalidFloatValue());
380       fOneEventData->Add(param);
381     }
382     
383     Double_t charge(adc);
384     
385     if ( fDigitCalibrator ) 
386     {
387       if ( fDigitCalibrator->IsValidDigit(detElemId, manuId, manuChannel) )
388       {
389         charge = fDigitCalibrator->CalibrateDigit(detElemId, manuId, manuChannel,adc,3.0);
390       }
391       else
392       {
393         charge = 0.0;
394       }
395     }
396     
397     if (charge > 0.0 ) 
398     {
399       param->SetValueAsDouble(manuChannel,0,charge);
400     }
401   }
402   
403         Bool_t badEvent = stream.HasPaddingError() || stream.HasGlitchError();
404
405         if (!badEvent)
406   {
407     fAccumulatedData->Add(*fOneEventData,&nevents);    
408   }
409   
410   return !badEvent;
411 }
412
413
414 //_____________________________________________________________________________
415 void 
416 AliMUONTrackerDataMaker::Print(Option_t*) const
417 {
418   /// Printout
419   
420   cout << "Source=" << Source() << " Running=" << ( IsRunning() ? "YES" : "NO")
421   << endl;
422 }
423
424 //_____________________________________________________________________________
425 void AliMUONTrackerDataMaker::Rewind()
426 {
427   /// Rewind events
428   if ( fIsOwnerOfRawReader ) 
429   {
430     fRawReader->RewindEvents();
431     fNumberOfEvents=0;  
432   }
433   else
434   {
435     AliError("Wrong usage of this class : cannot rewind as I am not owner of the raw reader !");
436   }
437 }
438
439 //_____________________________________________________________________________
440 Long64_t AliMUONTrackerDataMaker::Merge(TCollection* list)
441 {
442   /// Merge objects in collection
443   
444   if (!list) return 0;
445   
446   if ( list->IsEmpty() ) return NumberOfEvents();
447   
448   TIter next(list);
449   const TObject* o(0x0);
450   
451   while ( ( o = next() ) )
452   {
453     const AliMUONTrackerDataMaker* data = dynamic_cast<const AliMUONTrackerDataMaker*>(o);
454     if (!o)
455     {
456       AliError(Form("Object named %s is not an AliMUONTrackerDataMaker ! Skipping it",
457                     o->GetName()));
458     }
459     else
460     {
461       Bool_t ok = Add(*data);
462       if (!ok)
463       {
464         AliError("Got incompatible objects");
465       }
466     }
467   }
468   
469   return NumberOfEvents();
470 }
471
472 //_____________________________________________________________________________
473 void 
474 AliMUONTrackerDataMaker::SetRawReader(AliRawReader* rawReader)
475 {
476   /// Change the rawreader (only works if isowner=true)
477   
478   if ( fIsOwnerOfRawReader ) 
479         {
480     AliFatal("Improper use of this class ! Cannot change raw reader in this case");
481         }
482         
483   fRawReader = rawReader;
484   
485 }