TRD/AliTRDPreprocessorOffline.h

   1 #ifndef ALITRDPREPROCESSOROFFLINE_H
   2 #define ALITRDPREPROCESSOROFFLINE_H
   3
   4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   5  * See cxx source for full Copyright notice                               */
   6
   7 //
   8 //
   9 //    Class to create OCDB entries - processing the results of the OFFLINE calibration
  10 //
  11
  12
  13 #include "TNamed.h"
  14 #include "AliTRDCalChamberStatus.h"
  15 #include <THnSparse.h>
  16 class TObjArray;
  17 class AliTRDCalDet;
  18 class TH2I;
  19 class TProfile2D;
  20 class AliTRDCalibraVdriftLinearFit;
  21 class AliTRDCalibraExbAltFit;
  22 class TH1I;
  23 class TH2F;
  24 class TString;
  25
  26
  27 class AliTRDPreprocessorOffline:public TNamed {
  28 public:
  29   enum{ kGain = 0,
  30         kVdriftPHDet = 1,
  31         kVdriftPHPad = 2,
  32         kT0PHDet = 3,
  33         kT0PHPad = 4,
  34         kVdriftLinear = 5,
  35         kLorentzLinear = 6,
  36         kChamberStatus = 7,
  37         kPRF = 8,
  38         kExbAlt = 9,
  39         kNumCalibObjs = 10
  40   };
  41   enum { kGainNotEnoughStatsButFill = 2,
  42          kVdriftNotEnoughStatsButFill = 4,
  43          kGainNotEnoughStatsNotFill = 8,
  44          kVdriftNotEnoughStatsNotFill = 16,
  45          kTimeOffsetNotEnoughStatsNotFill = 32,
  46          kExBErrorRange = 64};
  47   enum { kGainErrorOld = 2,
  48          kVdriftErrorOld = 4,
  49          kExBErrorOld = 8,
  50          kGainErrorRange = 16,
  51          kVdriftErrorRange = 32,
  52          kTimeOffsetErrorRange = 64,
  53          kChamberStatusErrorRange = 128};
  54
  55
  56   AliTRDPreprocessorOffline();
  57   virtual ~AliTRDPreprocessorOffline();
  58
  59   Bool_t Init(const Char_t* fileName);
  60   void Process(const Char_t* file, Int_t startRunNumber, Int_t endRunNumber, TString ocdbStorage);
  61
  62   // settings
  63   void SetLinearFitForVdrift(Bool_t methodsecond) { fMethodSecond = methodsecond;};
  64   void SetNameList(TString nameList) { fNameList = nameList;};
  65   void     SetMinStatsVdriftT0PH(Int_t minStatsVdriftT0PH)           { fMinStatsVdriftT0PH = minStatsVdriftT0PH; }
  66   void     SetMinStatsVdriftLinear(Int_t minStatsVdriftLinear)       { fMinStatsVdriftLinear = minStatsVdriftLinear; }
  67   void     SetMinStatsGain(Int_t minStatsGain)                       { fMinStatsGain = minStatsGain; }
  68   void     SetMinStatsPRF(Int_t minStatsPRF)                         { fMinStatsPRF = minStatsPRF; }
  69   void     SetMinStatsChamberStatus(Int_t minStatsChamberStatus)     { fMinStatsChamberStatus = minStatsChamberStatus; }
  70   void     SetLimitValidateNoData(Int_t nodatavalidate)              { fNoDataValidate = nodatavalidate; };
  71   void     SetLimitValidateBadCalib(Int_t badcalibvalidate)          { fBadCalibValidate = badcalibvalidate; };
  72   void     SetBackCorrectGain(Bool_t backCorrectGain)                { fBackCorrectGain = backCorrectGain; }
  73   void     SetBackCorrectVdrift(Bool_t backCorrectVdrift)            { fBackCorrectVdrift = backCorrectVdrift; }
  74   void     SetNoExBUsedInReco(Bool_t noExBUsedInReco)                { fNoExBUsedInReco    = noExBUsedInReco;   };
  75   void     SetSwitchOnValidation(Bool_t switchOnValidation)          { fSwitchOnValidation = switchOnValidation;};
  76   void     SetRMSBadCalibratedGain(Double_t rms)                     { fRMSBadCalibratedGain = rms;};
  77   void     SetRMSBadCalibratedVdrift(Double_t rms)                   { fRMSBadCalibratedVdrift = rms;};
  78   void     SetRMSBadCalibratedExB(Double_t rms)                      { fRMSBadCalibratedExB = rms;};
  79   void     SetRobustFitDriftVelocity(Bool_t robustFitDriftVelocity)  { fRobustFitDriftVelocity = robustFitDriftVelocity;};
  80   void     SetRobustFitExbAlt(Bool_t robustFitExbAlt)                { fRobustFitExbAlt = robustFitExbAlt;};
  81   void     SetAlternativeDriftVelocityFit(Bool_t alt)                { fAlternativeVdrfitFit = alt;};
  82   void     SetAlternativeExbAltFit(Bool_t alt)                       { fAlternativeExbAltFit = alt;};
  83
  84
  85   Bool_t GetLinearFitForVdrift() const { return fMethodSecond;};
  86   TString GetNameList() const { return fNameList;};
  87
  88   // status
  89   Bool_t      CheckStatus(Int_t status, Int_t bitMask) const;
  90   void PrintStatus() const;
  91
  92   Bool_t      IsGainNotEnoughStatsButFill() const
  93     { return CheckStatus(fStatusNeg, kGainNotEnoughStatsButFill);  };
  94   Bool_t      IsGainNotEnoughStatsNotFill() const
  95     { return CheckStatus(fStatusNeg, kGainNotEnoughStatsNotFill);  };
  96   Bool_t      IsVdriftNotEnoughStatsButFill() const
  97     { return CheckStatus(fStatusNeg, kVdriftNotEnoughStatsButFill);  };
  98   Bool_t      IsVdriftNotEnoughStatsNotFill() const
  99     { return CheckStatus(fStatusNeg, kVdriftNotEnoughStatsNotFill);  };
 100   Bool_t      IsTimeOffsetNotEnoughStatsNotFill() const
 101     { return CheckStatus(fStatusNeg, kTimeOffsetNotEnoughStatsNotFill);  };
 102   Bool_t      IsExBErrorRange() const
 103     { return CheckStatus(fStatusNeg, kExBErrorRange);  };
 104
 105   Bool_t      IsGainErrorOld() const
 106     { return CheckStatus(fStatusPos, kGainErrorOld);  };
 107   Bool_t      IsVdriftErrorOld() const
 108     { return CheckStatus(fStatusPos, kVdriftErrorOld);  };
 109   Bool_t      IsExBErrorOld() const
 110     { return CheckStatus(fStatusPos, kExBErrorOld);  };
 111   Bool_t      IsGainErrorRange() const
 112     { return CheckStatus(fStatusPos, kGainErrorRange);  };
 113   Bool_t      IsVdriftErrorRange() const
 114     { return CheckStatus(fStatusPos, kVdriftErrorRange);  };
 115   Bool_t      IsTimeOffsetErrorRange() const
 116     { return CheckStatus(fStatusPos, kTimeOffsetErrorRange);  };
 117   Bool_t      IsChamberStatusErrorRange() const
 118     { return CheckStatus(fStatusPos, kChamberStatusErrorRange);  };
 119
 120
 121   // Back corrections
 122   void SetCalDetGain(AliTRDCalDet *calDetGainUsed) {fCalDetGainUsed = calDetGainUsed;};
 123   void SetCalDetVdrift(AliTRDCalDet *calDetVdriftUsed);
 124   void SetCalDetVdriftExB(AliTRDCalDet *calDetVdriftUsed,AliTRDCalDet *calDetExBUsed) {fCalDetVdriftUsed = calDetVdriftUsed; fCalDetExBUsed = calDetExBUsed;};
 125   Bool_t SetCalDetGain(Int_t runNumber, Int_t version, Int_t subversion);
 126   Bool_t SetCalDetVdriftExB(Int_t runNumber, Int_t versionv, Int_t subversionv, Int_t versionexb, Int_t subversionexb);
 127
 128   AliTRDCalDet *GetCalDetGain() const { return fCalDetGainUsed;};
 129   AliTRDCalDet *GetCalDetVdrift() const { return fCalDetVdriftUsed;};
 130   Int_t    GetFirstRunGainUsed() const                               { return fFirstRunGainUsed;       }
 131   Int_t    GetVersionGainUsed() const                                { return fVersionGainUsed;        }
 132   Int_t    GetSubVersionGainUsed() const                             { return fSubVersionGainUsed;     }
 133   Int_t    GetFirstRunVdriftUsed() const                             { return fFirstRunVdriftUsed;     }
 134   Int_t    GetVersionVdriftUsed() const                              { return fVersionVdriftUsed;      }
 135   Int_t    GetSubVersionVdriftUsed() const                           { return fSubVersionVdriftUsed;   }
 136   Int_t    GetFirstRunExBUsed() const                                { return fFirstRunExBUsed;        }
 137   Int_t    GetVersionExBUsed() const                                 { return fVersionExBUsed;         }
 138   Int_t    GetSubVersionExBUsed() const                              { return fSubVersionExBUsed;      }
 139
 140
 141   // Internal functions
 142
 143   void CalibVdriftT0(const Char_t* file, Int_t startRunNumber, Int_t endRunNumber, TString ocdbStorage="");
 144   void CalibExbAlt(const Char_t* file, Int_t startRunNumber, Int_t endRunNumber, TString ocdbStorage="");
 145   void CalibGain(const Char_t* file, Int_t startRunNumber, Int_t endRunNumber,  TString  ocdbStorage="");
 146   void CalibPRF(const Char_t* file, Int_t startRunNumber, Int_t endRunNumber,  TString  ocdbStorage="");
 147   void CalibChamberStatus(const Char_t* file, Int_t startRunNumber, Int_t endRunNumber, TString ocdbStorage="");
 148
 149   Bool_t ReadStatusGlobal(const Char_t* fileName="CalibObjects.root");
 150   Bool_t ReadGainGlobal(const Char_t* fileName="CalibObjects.root");
 151   Bool_t ReadVdriftT0Global(const Char_t* fileName="CalibObjects.root");
 152   Bool_t ReadVdriftLinearFitGlobal(const Char_t* fileName="CalibObjects.root");
 153   Bool_t ReadExbAltFitGlobal(const Char_t* fileName="CalibObjects.root");
 154   Bool_t ReadPRFGlobal(const Char_t* fileName="CalibObjects.root");
 155
 156   Bool_t AnalyzeGain();
 157   Bool_t AnalyzeVdriftT0();
 158   Bool_t AnalyzeVdriftLinearFit();
 159   Bool_t AnalyzeExbAltFit();
 160   Bool_t AnalyzePRF();
 161   Bool_t AnalyzeChamberStatus();
 162
 163   void CorrectFromDetGainUsed();
 164   void CorrectFromDetVdriftUsed();
 165
 166   void UpdateOCDBT0(Int_t startRunNumber, Int_t endRunNumber, const char* storagePath);
 167   void UpdateOCDBVdrift(Int_t startRunNumber, Int_t endRunNumber, const char* storagePath);
 168   void UpdateOCDBExB(Int_t startRunNumber, Int_t endRunNumber, const Char_t *storagePath);
 169   void UpdateOCDBExBAlt(Int_t startRunNumber, Int_t endRunNumber, const Char_t *storagePath);
 170   void UpdateOCDBGain(Int_t  startRunNumber, Int_t endRunNumber, const char* storagePath);
 171   void UpdateOCDBPRF(Int_t  startRunNumber, Int_t endRunNumber, const char* storagePath);
 172   void UpdateOCDBChamberStatus(Int_t startRunNumber, Int_t endRunNumber, const Char_t *storagePath);
 173
 174   Bool_t ValidateGain();
 175   Bool_t ValidateVdrift();
 176   Bool_t ValidateExB();
 177   Bool_t ValidateT0();
 178   Bool_t ValidatePRF() const;
 179   Bool_t ValidateChamberStatus();
 180
 181   Int_t    GetStatus() const;
 182   Int_t    GetStatusPos() const                                      { return fStatusPos;              }
 183   Int_t    GetStatusNeg() const                                      { return fStatusNeg;              }
 184
 185
 186
 187  private:
 188   Bool_t fMethodSecond;                      // Second Method for drift velocity
 189   TString fNameList;                         // Name of the list
 190   AliTRDCalDet *fCalDetGainUsed;             // CalDet used and to be corrected for
 191   AliTRDCalDet *fCalDetVdriftUsed;           // CalDet used and to be corrected for
 192   AliTRDCalDet *fCalDetExBUsed;              // CalDet used and to be corrected for
 193   TH2I *fCH2d;                               // Gain
 194   TProfile2D *fPH2d;                         // Drift velocity first method
 195   TProfile2D *fPRF2d;                        // PRF
 196   THnSparseI *fSparse;                       // chamberstatus
 197   AliTRDCalibraVdriftLinearFit *fAliTRDCalibraVdriftLinearFit; // Drift velocity second method
 198   AliTRDCalibraExbAltFit* fAliTRDCalibraExbAltFit; //ExB alternative method
 199   TH1I *fNEvents;                         // Number of events
 200   TH2F *fAbsoluteGain;                    // Absolute Gain calibration
 201   TObjArray * fPlots;                     // array with some plots to check
 202   TObjArray * fCalibObjects;              // array with calibration objects
 203   Int_t    fFirstRunGainUsed;             // first run gain used
 204   Int_t    fVersionGainUsed;              // VersionGainUsed
 205   Int_t    fSubVersionGainUsed;           // SubVersionGainUsed
 206   Int_t    fFirstRunVdriftUsed;           // FirstRunVdrift
 207   Int_t    fVersionVdriftUsed;            // VersionVdriftUsed
 208   Int_t    fSubVersionVdriftUsed;         // SubVersionVdriftUsed
 209   Int_t    fFirstRunExBUsed;              // FirstRunExB
 210   Int_t    fVersionExBUsed;               // VersionExBUsed
 211   Int_t    fSubVersionExBUsed;            // SubVersionExBUsed
 212   Bool_t   fNoExBUsedInReco;              // ExB not used yet in the reco
 213   Bool_t   fSwitchOnValidation;           // Validation
 214   Bool_t   fVdriftValidated;              // Vdrift validation
 215   Bool_t   fExBValidated;                 // ExB validation
 216   Bool_t   fT0Validated;                  // T0 validation
 217   Int_t    fMinStatsVdriftT0PH;           // MinStats VdriftT0
 218   Int_t    fMinStatsVdriftLinear;         // MinStats Vdrift Linear
 219   Int_t    fMinStatsGain;                 // MinStats Gain
 220   Int_t    fMinStatsPRF;                  // MinStats PRF
 221   Int_t    fMinStatsChamberStatus;        // MinStats ChamberStatus
 222   Bool_t   fBackCorrectGain;              // Back correction afterwards gain
 223   Bool_t   fBackCorrectVdrift;            // Back correction afterwards vdrift
 224   Bool_t   fNotEnoughStatisticsForTheGain;// Take the chamber per chamber distribution from the default distribution
 225   Bool_t   fNotEnoughStatisticsForTheVdriftLinear;// Take the chamber per chamber distribution from the default distribution
 226   Int_t    fStatusNeg;                    // Info but ok
 227   Int_t    fStatusPos;                    // Problems
 228   Int_t    fBadCalib[18];                 // number of bad calibrated chambers per sm
 229   Int_t    fNoData[18];                   // number of  chambers w/o data per sm
 230   Int_t    fBadCalibValidate;             // validation limit for bad calibrated chambers
 231   Int_t    fNoDataValidate;               // validation limit for chamber w/o data (sm w/o data excluded)
 232   Double_t fRMSBadCalibratedGain;         // value to decide when it is bad calibrated
 233   Double_t fRMSBadCalibratedVdrift;       // value to decide when it is bad calibrated
 234   Double_t fRMSBadCalibratedExB;          // value to decide when it is bad calibrated
 235   Bool_t   fRobustFitDriftVelocity;       // Robust fit for the drift velocity
 236   Bool_t   fRobustFitExbAlt;              // Robust fit for the exb alt
 237   Bool_t   fAlternativeVdrfitFit;         // Alternative fitting method for vdrift calibration
 238   Bool_t   fAlternativeExbAltFit;         // Alternative fitting method for the alternative exb calibarion method
 239
 240   Int_t GetSubVersion(TString name) const;
 241   Int_t GetVersion(TString name) const;
 242   Int_t GetFirstRun(TString name) const;
 243
 244
 245
 246 private:
 247   AliTRDPreprocessorOffline& operator=(const AliTRDPreprocessorOffline&); // not implemented
 248   AliTRDPreprocessorOffline(const AliTRDPreprocessorOffline&); // not implemented
 249   ClassDef(AliTRDPreprocessorOffline,3)
 250 };
 251
 252 #endif
 253
 254