Adding RejectList visualization in mchview
[u/mrichter/AliRoot.git] / MUON / AliMUONTrackerConditionDataMaker.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 #include "AliMUONTrackerConditionDataMaker.h"
19
20 ///\class AliMUONTrackerConditionDataMaker
21 ///
22 /// Producer of AliMUONVTrackerData from OCDB or Ascii file condition data
23 ///
24 /// \author Laurent Aphecetche, Subatech
25
26 ///\cond CLASSIMP
27 ClassImp(AliMUONTrackerConditionDataMaker)
28 ///\endcond
29
30 #include "AliCDBManager.h"
31 #include "AliCDBStorage.h"
32 #include "AliDCSValue.h"
33 #include "AliLog.h"
34 #include "AliMpManuIterator.h"
35 #include "AliMUON2DMap.h"
36 #include "AliMUONCalibParamND.h"
37 #include "AliMUONCalibParamNF.h"
38 #include "AliMUONCalibParamNI.h"
39 #include "AliMUONCalibrationData.h"
40 #include "AliMUONDigitCalibrator.h"
41 #include "AliMUONPadStatusMaker.h"
42 #include "AliMUONPadStatusMapMaker.h"
43 #include "AliMUONRejectList.h"
44 #include "AliMUONTrackerData.h"
45 #include "AliMUONTrackerIO.h"
46 #include "AliMpArrayI.h"
47 #include "AliMpConstants.h"
48 #include "AliMpDCSNamer.h"
49 #include "AliMpDDLStore.h"
50 #include "AliMpDEManager.h"
51 #include "AliMpDetElement.h"
52 #include "TClass.h"
53 #include "TMap.h"
54 #include "TObjString.h"
55 #include "Riostream.h"
56 #include "TString.h"
57 #include <sstream>
58 #include "TSystem.h"
59
60 //_____________________________________________________________________________
61 AliMUONTrackerConditionDataMaker::AliMUONTrackerConditionDataMaker():
62 AliMUONVTrackerDataMaker(),
63 fData(0x0),
64 fSource(""),
65 fIsOwnerOfData(kTRUE)
66 {
67   /// default ctor to be able to stream
68 }
69
70 //_____________________________________________________________________________
71 AliMUONTrackerConditionDataMaker::AliMUONTrackerConditionDataMaker(Int_t runNumber, const char* ocdbPath, const char* type):
72 AliMUONVTrackerDataMaker(),
73 fData(0x0),
74 fSource(Form("%s-%010d-%s",ocdbPath,runNumber,type)),
75 fIsOwnerOfData(kTRUE)
76 {
77   /// ctor from OCDB
78
79   AliCDBStorage* storage = AliCDBManager::Instance()->GetDefaultStorage();
80         
81         AliCDBManager::Instance()->SetDefaultStorage(ocdbPath);
82
83   TString stype(type);
84   stype.ToUpper();
85   
86   if ( stype == "REJECTLIST" ) 
87   {
88     
89     Int_t startOfValidity(0);
90     
91     AliMUONRejectList* rl = AliMUONCalibrationData::CreateRejectList(runNumber,&startOfValidity);    
92
93     if (rl)
94     {
95       fData = new AliMUONTrackerData(Form("RL%d",startOfValidity),"RejectList",*rl);
96     }
97     
98     delete rl;
99   }
100   else
101   {
102     Int_t startOfValidity;
103   AliMUONVStore* store = CreateStore(runNumber,ocdbPath,type,startOfValidity);
104     AliDebug(1,Form("runNumber=%d ocdbPath=%s type=%s startOfValidity=%d store=%p",
105                     runNumber,ocdbPath,type,startOfValidity,store));
106     if ( store )
107     {
108       fData = CreateData(type,*store,startOfValidity);
109     }  
110     
111     delete store;
112   }
113   
114   AliCDBManager::Instance()->SetDefaultStorage(storage);
115 }
116
117 //_____________________________________________________________________________
118 AliMUONTrackerConditionDataMaker::AliMUONTrackerConditionDataMaker(const char* filename, const char* type):
119 AliMUONVTrackerDataMaker(),
120 fData(0x0),
121 fSource(Form("%s-%s",filename,type)),
122 fIsOwnerOfData(kTRUE)
123 {
124   /// ctor from an ASCII file
125   
126   TString sFilename(gSystem->ExpandPathName(filename));
127   
128   std::ifstream in(sFilename.Data());
129   if (in.good()) 
130   {
131     std::ostringstream stream;
132     char line[1024];
133     while ( in.getline(line,1024) )
134     {
135       stream << line << "\n";    
136     }
137   
138     in.close();
139   
140     Int_t dummy;
141     
142     AliMUONVStore* store = CreateStore(-1,stream.str().c_str(),type,dummy);
143     
144     if ( store )
145     {
146       fData = CreateData(type,*store,dummy);
147     }
148     delete store;
149   }
150 }
151
152 //_____________________________________________________________________________
153 AliMUONTrackerConditionDataMaker::AliMUONTrackerConditionDataMaker(const char* data, const char* type, Bool_t) :
154 AliMUONVTrackerDataMaker(),
155 fData(0x0),
156 fSource(Form("direct-%s",type)),
157 fIsOwnerOfData(kTRUE)
158
159 {
160   /// ctor from a string containing the ASCII data
161   /// the last parameter is there just to distinguish this ctor from the previous one
162   
163   Int_t dummy;
164   
165   AliMUONVStore* store = CreateStore(-1,data,type,dummy);
166   
167   if ( store )
168   {
169     fData = CreateData(type,*store,dummy);
170   }
171   delete store;
172   
173 }
174
175 //_____________________________________________________________________________
176 AliMUONTrackerConditionDataMaker::~AliMUONTrackerConditionDataMaker()
177 {
178   /// dtor
179   if ( fIsOwnerOfData ) delete fData;
180 }
181
182
183 //_____________________________________________________________________________
184 AliMUONVTrackerData*
185 AliMUONTrackerConditionDataMaker::CreateData(const char* type, AliMUONVStore& store, Int_t startOfValidity)
186 {
187   /// Create the data source 
188   AliMUONVTrackerData* data(0x0);
189   
190   TString stype(type);
191   stype.ToUpper();
192   
193   if ( stype == "CAPACITANCES" )
194   {    
195     data = new AliMUONTrackerData(Form("CAPA%d",startOfValidity),"Capacitances",2,kTRUE);
196     data->SetDimensionName(0,"Capa");
197     data->SetDimensionName(1,"Injection gain");    
198   }
199   else if ( stype == "CONFIG" ) 
200   {
201     data = new AliMUONTrackerData(Form("CONFIG%d",startOfValidity),"Configuration",1);
202     data->SetDimensionName(0,"there");
203     data->DisableChannelLevel();
204   }
205   else if ( stype == "GAINS" ) 
206   {
207     data = new AliMUONTrackerData(Form("GAIN%d",startOfValidity),"Gains",7,kTRUE);
208     data->SetDimensionName(0,"gain");
209     data->SetDimensionName(1,"a1");
210     data->SetDimensionName(2,"a2");
211     data->SetDimensionName(3,"thres");
212     data->SetDimensionName(4,"qual1");
213     data->SetDimensionName(5,"qual2");
214     data->SetDimensionName(6,"sat");    
215   }
216   else if ( stype == "HV" ) 
217   {
218     data = new AliMUONTrackerData(Form("HV%d",startOfValidity),"High Voltages",1); //,!isSingleEvent);
219                 data->SetDimensionName(0,"HV");
220   }
221   else if ( stype == "OCCUPANCY" ) 
222   {
223     data = new AliMUONTrackerData(Form("OCC%d",startOfValidity),"OccupancyMap",store);
224     data->SetDimensionName(0,"One");
225     return data; // important to return now to avoid the data->Add(store) later on...
226   }
227   else if ( stype == "PEDESTALS" ) 
228   {
229     data  = new AliMUONTrackerData(Form("PED%d",startOfValidity),"Pedestals",2,kTRUE);
230     data->SetDimensionName(0,"Mean");
231     data->SetDimensionName(1,"Sigma");    
232   }
233   else if ( stype == "STATUS" ) 
234   {
235     data = new AliMUONTrackerData(Form("STATUS%d",startOfValidity),"Status",1,kTRUE);
236     data->SetDimensionName(0,"Bits");
237   }
238   else if ( stype == "STATUSMAP" ) 
239   {
240     data = new AliMUONTrackerData(Form("STATUSMAP%d",startOfValidity),"Status map",2,kTRUE);
241     data->SetDimensionName(0,"Bits");
242     data->SetDimensionName(1,"Dead");
243   }
244
245   if (!data)
246   {
247     AliErrorClass(Form("Could not create data for type=%s",type));
248     return 0x0;
249   }
250   
251   data->Add(store);
252   
253   return data;
254 }
255
256 //_____________________________________________________________________________
257 AliMUONVStore*
258 AliMUONTrackerConditionDataMaker::CreateHVStore(TMap& m)
259 {
260   /// Create a store from hv values
261   
262   AliMUONVStore* store = new AliMUON2DMap(kTRUE);
263   
264   TIter next(&m);
265   TObjString* s;
266   AliMpDCSNamer hvNamer("TRACKER");
267   
268   while ( ( s = static_cast<TObjString*>(next()) ) )
269   {
270     TString name(s->String());
271     
272     Int_t hvIndex = hvNamer.DCSIndexFromDCSAlias(name.Data());
273
274     Int_t detElemId = hvNamer.DetElemIdFromDCSAlias(name.Data());
275     
276     if ( hvIndex >= 0 && detElemId < 0 )
277     {
278       // skip switches
279       continue;      
280     }
281     
282     if ( !AliMpDEManager::IsValidDetElemId(detElemId) )
283     {
284       AliErrorClass(Form("Got an invalid DE = %d from alias = %s",
285                          detElemId,name.Data()));
286       continue;
287     }
288
289     Int_t nPCBs = hvNamer.NumberOfPCBs(detElemId);
290     Int_t indexMin = nPCBs ? 0 : hvIndex;
291     Int_t indexMax = nPCBs ? nPCBs : hvIndex+1;
292     
293     AliMpDetElement* de = AliMpDDLStore::Instance()->GetDetElement(detElemId);
294     
295     for ( int i = indexMin ; i < indexMax; ++i )
296     {
297       Float_t switchValue(1.0);
298       
299       if ( nPCBs ) 
300       {
301         TString switchName(hvNamer.DCSSwitchName(detElemId,i));
302
303         TPair* p = static_cast<TPair*>(m.FindObject(switchName.Data()));
304         TObjArray* a = static_cast<TObjArray*>(p->Value());
305         
306         switchValue = AliMUONPadStatusMaker::SwitchValue(*a);                                           
307       }
308       
309       const AliMpArrayI* manus = de->ManusForHV(i);
310       
311       if (!manus) continue;
312       
313       TPair* p = static_cast<TPair*>(m.FindObject(name.Data()));
314       TObjArray* a = static_cast<TObjArray*>(p->Value());
315       TIter n2(a);
316       AliDCSValue* v;
317       Float_t hvValue(0);
318       Int_t n(0);
319       Int_t noff(0);
320       
321       while ( ( v = static_cast<AliDCSValue*>(n2()) ) )
322       {
323         hvValue += v->GetFloat();
324         if ( v->GetFloat() < AliMpDCSNamer::TrackerHVOFF() ) ++noff;
325         ++n;
326       }
327       hvValue *= switchValue;  
328       
329       if ( n ) hvValue /= n;
330
331       if (noff>0 && noff<n) 
332       {
333         // that's a trip
334         hvValue = -1.0;        
335       }
336       
337       Int_t nofChannels(AliMpConstants::ManuNofChannels());
338       
339       for ( Int_t k = 0 ; k < manus->GetSize(); ++k )
340       {
341         Int_t manuId = manus->GetValue(k);
342         AliMUONVCalibParam* param = static_cast<AliMUONVCalibParam*>(store->FindObject(detElemId,manuId));
343         if ( ! param ) 
344         {
345           param = new AliMUONCalibParamND(1,nofChannels,detElemId,manuId,0);
346           store->Add(param);
347         }
348         for ( Int_t j = 0 ; j < nofChannels; ++j )
349         {
350           param->SetValueAsDouble(j,0,hvValue);
351         }
352       }
353     }
354   }
355   
356   return store;
357   
358 }
359
360 //_____________________________________________________________________________
361 AliMUONVStore*
362 AliMUONTrackerConditionDataMaker::CreateStatusStore(Int_t runNumber)
363 {
364   /// Get the status store
365   
366   AliMUONDigitCalibrator calibrator(runNumber);
367   
368   AliMUONVStore* sm = new AliMUON2DMap(kTRUE);
369   
370   AliMpManuIterator it;
371   Int_t detElemId, manuId;
372   
373   while (it.Next(detElemId,manuId))
374   {
375     AliMUONVCalibParam* np = new AliMUONCalibParamNI(1,AliMpConstants::ManuNofChannels(),detElemId,manuId);
376     for ( Int_t i = 0; i < np->Size(); ++i ) 
377     {
378       Int_t value = calibrator.PadStatus(detElemId,manuId,i);
379       np->SetValueAsInt(i,0,value); // "raw" value of the status
380     }
381     sm->Add(np);
382   }
383   
384   return sm;
385 }
386
387 //_____________________________________________________________________________
388 AliMUONVStore*
389 AliMUONTrackerConditionDataMaker::CreateStatusMapStore(Int_t runNumber)
390 {
391   /// Get the status map, and polish it a bit for representation purposes
392   
393   AliMUONDigitCalibrator calibrator(runNumber);
394   
395   AliMUONVStore* sm = new AliMUON2DMap(kTRUE);
396   
397   AliMpManuIterator it;
398   Int_t detElemId, manuId;
399   
400   while (it.Next(detElemId,manuId))
401   {
402     AliMUONVCalibParam* np = new AliMUONCalibParamNI(2,AliMpConstants::ManuNofChannels(),detElemId,manuId);
403     for ( Int_t i = 0; i < np->Size(); ++i ) 
404     {
405       Int_t value = calibrator.StatusMap(detElemId,manuId,i);
406       Int_t channelIsDead = ( value & AliMUONPadStatusMapMaker::SelfDeadMask() );
407       np->SetValueAsInt(i,0,value); // "raw" value of the status map
408       np->SetValueAsInt(i,1,channelIsDead); // simple 0 or 1 for this channel
409     }
410     sm->Add(np);
411   }
412   
413   return sm;
414 }
415
416 //_____________________________________________________________________________
417 AliMUONVStore*
418 AliMUONTrackerConditionDataMaker::CreateStore(Int_t runNumber, 
419                                               const char* source, 
420                                               const char* type, 
421                                               Int_t& startOfValidity)
422 {
423   /// Create the store by reading it from OCDB or from an ASCII file
424   
425   TString stype(type);
426   stype.ToUpper();
427   
428   AliMUONVStore* store(0x0);
429   
430   startOfValidity = 0;
431   
432   Bool_t ocdb = (runNumber>=0);
433   
434   if ( stype == "CAPACITANCES" )
435   {    
436     if ( ocdb ) 
437     {
438       store = AliMUONCalibrationData::CreateCapacitances(runNumber,&startOfValidity);    
439     }
440     else
441     {
442       store = new AliMUON2DMap(20000);
443       AliMUONTrackerIO::DecodeCapacitances(source,*store);
444     }
445   }
446   else if ( stype == "CONFIG" ) 
447   {
448     AliMUONVStore* tmp(0x0);
449     if ( ocdb ) 
450     {
451       tmp = AliMUONCalibrationData::CreateConfig(runNumber,&startOfValidity);
452     }
453     else
454     {
455       tmp = new AliMUON2DMap(kTRUE);
456       AliMUONTrackerIO::DecodeConfig(source,*tmp);
457     }
458     if ( tmp ) 
459     {
460       store = ExpandConfig(*tmp);      
461     }
462     delete tmp;
463   }
464   else if ( stype == "GAINS" ) 
465   {
466     AliMUONVStore* gains(0x0);
467     if ( ocdb ) 
468     {
469       gains = AliMUONCalibrationData::CreateGains(runNumber,&startOfValidity);
470     }
471     else
472     {
473       gains = new AliMUON2DMap(kTRUE);
474       TString comment;
475       AliMUONTrackerIO::DecodeGains(source,*gains,comment);
476     }
477     store = PatchGainStore(*gains);
478     delete gains;
479   }
480   else if ( stype == "OCCUPANCY" ) 
481   {
482     if ( ocdb ) 
483     {
484       store = AliMUONCalibrationData::CreateOccupancyMap(runNumber,&startOfValidity);
485     }
486     else
487     {
488       store = new AliMUON2DMap(kTRUE);
489       AliMUONTrackerIO::DecodeOccupancy(source,*store);
490     }
491   }
492   else if ( stype == "PEDESTALS" ) 
493   {
494     if ( ocdb ) 
495     {
496       store = AliMUONCalibrationData::CreatePedestals(runNumber,&startOfValidity);
497     }
498     else
499     {
500       store = new AliMUON2DMap(kTRUE);
501       AliMUONTrackerIO::DecodePedestals(source,*store);
502     }
503   }
504   
505   /// Below are source that can only be accessed from OCDB
506   if (!store && !ocdb) 
507   {
508     return 0x0;
509   }
510   
511   if ( stype == "HV" ) 
512   {
513     TMap* m = AliMUONCalibrationData::CreateHV(runNumber,&startOfValidity);
514                 store = CreateHVStore(*m);
515     delete m;
516   }
517   else if ( stype == "STATUS" ) 
518   {
519     store = CreateStatusStore(runNumber);
520   }
521   else if ( stype == "STATUSMAP" ) 
522   {
523     store = CreateStatusMapStore(runNumber);
524   }
525   
526   return store;
527 }
528
529 //_____________________________________________________________________________
530 AliMUONVStore*
531 AliMUONTrackerConditionDataMaker::ExpandConfig(const AliMUONVStore& manuConfig)
532 {
533   /// Convert the config from manu level to channel level (just to
534   /// be able to add it correctly to the trackerdata...)
535   
536   AliMUONVStore* store = manuConfig.Create();
537   
538   TIter next(manuConfig.CreateIterator());
539   AliMUONVCalibParam* p;
540   
541   while ( ( p = static_cast<AliMUONVCalibParam*>(next()) ) )
542   {
543     AliMUONVCalibParam* c = new AliMUONCalibParamNF(1,AliMpConstants::ManuNofChannels(),p->ID0(),p->ID1(),0.0);
544     
545     AliMpDetElement* de = AliMpDDLStore::Instance()->GetDetElement(p->ID0());
546     
547     for ( Int_t i = 0; i < c->Size(); ++i ) 
548     {
549       if ( de->IsExistingChannel(p->ID1(),i) )
550       {
551         c->SetValueAsFloat(i,0,1.0);        
552       }
553     }
554     
555     store->Add(c);
556   }
557   return store;
558 }
559
560 //_____________________________________________________________________________
561 Long64_t 
562 AliMUONTrackerConditionDataMaker::Merge(TCollection*)
563 {
564   /// Merge
565   AliError("Not implemented. Does it have sense ?");
566   return 0;
567 }
568
569 //_____________________________________________________________________________
570 AliMUONVStore*
571 AliMUONTrackerConditionDataMaker::PatchGainStore(const AliMUONVStore& gains)
572 {
573   /// Polish the gain store : 
574   /// a) adding a dimension, computed from a1, and called gain = 1/a1/0.2 
575   ///     where 0.2 is internal capa in pF, and gain is then in mV/fC
576   /// b) splitting the quality in two
577   
578   AliMUONVStore* store = gains.Create();
579   
580   TIter next(gains.CreateIterator());
581   AliMUONVCalibParam* param;
582   
583   while ( ( param = static_cast<AliMUONVCalibParam*>(next()) ) ) 
584   {
585     AliMUONVCalibParam* nd = new AliMUONCalibParamND(param->Dimension()+2,
586                                                      param->Size(),
587                                                      param->ID0(),
588                                                      param->ID1());
589     for ( Int_t i = 0; i < param->Size(); ++i ) 
590     {
591       
592       Int_t qual = param->ValueAsInt(i,3);
593                         Int_t q1 = (qual & 0xF0) >> 4;  // linear fit quality
594                         Int_t q2 = qual & 0xF;          // parabolic fit quality
595                         Double_t gain = 0.0;
596       
597       if ( param->ValueAsFloat(i,0) > 1E-9 ) gain = 1.0/param->ValueAsFloat(i,0)/0.2;
598                         
599       nd->SetValueAsDouble(i,0,gain); // gain
600       nd->SetValueAsDouble(i,1,param->ValueAsFloat(i,0)); // a1
601       nd->SetValueAsDouble(i,2,param->ValueAsFloat(i,1)); // a2
602       nd->SetValueAsInt(i,3,param->ValueAsInt(i,2)); // thres
603       nd->SetValueAsInt(i,4,q1); // qual1
604       nd->SetValueAsInt(i,5,q2); // qual2
605       nd->SetValueAsInt(i,6,param->ValueAsInt(i,4)); // sat
606     }
607     store->Add(nd);
608   }
609   
610   return store;
611 }
612