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
  82
  83   Bool_t GetLinearFitForVdrift() const { return fMethodSecond;};
  84   TString GetNameList() const { return fNameList;};
  85
  86   // status
  87   Bool_t      CheckStatus(Int_t status, Int_t bitMask) const;
  88   void PrintStatus() const;
  89
  90   Bool_t      IsGainNotEnoughStatsButFill() const
  91     { return CheckStatus(fStatusNeg, kGainNotEnoughStatsButFill);  };
  92   Bool_t      IsGainNotEnoughStatsNotFill() const
  93     { return CheckStatus(fStatusNeg, kGainNotEnoughStatsNotFill);  };
  94   Bool_t      IsVdriftNotEnoughStatsButFill() const
  95     { return CheckStatus(fStatusNeg, kVdriftNotEnoughStatsButFill);  };
  96   Bool_t      IsVdriftNotEnoughStatsNotFill() const
  97     { return CheckStatus(fStatusNeg, kVdriftNotEnoughStatsNotFill);  };
  98   Bool_t      IsTimeOffsetNotEnoughStatsNotFill() const
  99     { return CheckStatus(fStatusNeg, kTimeOffsetNotEnoughStatsNotFill);  };
 100   Bool_t      IsExBErrorRange() const
 101     { return CheckStatus(fStatusNeg, kExBErrorRange);  };
 102
 103   Bool_t      IsGainErrorOld() const
 104     { return CheckStatus(fStatusPos, kGainErrorOld);  };
 105   Bool_t      IsVdriftErrorOld() const
 106     { return CheckStatus(fStatusPos, kVdriftErrorOld);  };
 107   Bool_t      IsExBErrorOld() const
 108     { return CheckStatus(fStatusPos, kExBErrorOld);  };
 109   Bool_t      IsGainErrorRange() const
 110     { return CheckStatus(fStatusPos, kGainErrorRange);  };
 111   Bool_t      IsVdriftErrorRange() const
 112     { return CheckStatus(fStatusPos, kVdriftErrorRange);  };
 113   Bool_t      IsTimeOffsetErrorRange() const
 114     { return CheckStatus(fStatusPos, kTimeOffsetErrorRange);  };
 115   Bool_t      IsChamberStatusErrorRange() const
 116     { return CheckStatus(fStatusPos, kChamberStatusErrorRange);  };
 117
 118
 119   // Back corrections
 120   void SetCalDetGain(AliTRDCalDet *calDetGainUsed) {fCalDetGainUsed = calDetGainUsed;};
 121   void SetCalDetVdrift(AliTRDCalDet *calDetVdriftUsed);
 122   void SetCalDetVdriftExB(AliTRDCalDet *calDetVdriftUsed,AliTRDCalDet *calDetExBUsed) {fCalDetVdriftUsed = calDetVdriftUsed; fCalDetExBUsed = calDetExBUsed;};
 123   Bool_t SetCalDetGain(Int_t runNumber, Int_t version, Int_t subversion);
 124   Bool_t SetCalDetVdriftExB(Int_t runNumber, Int_t versionv, Int_t subversionv, Int_t versionexb, Int_t subversionexb);
 125
 126   AliTRDCalDet *GetCalDetGain() const { return fCalDetGainUsed;};
 127   AliTRDCalDet *GetCalDetVdrift() const { return fCalDetVdriftUsed;};
 128   Int_t    GetFirstRunGainUsed() const                               { return fFirstRunGainUsed;       }
 129   Int_t    GetVersionGainUsed() const                                { return fVersionGainUsed;        }
 130   Int_t    GetSubVersionGainUsed() const                             { return fSubVersionGainUsed;     }
 131   Int_t    GetFirstRunVdriftUsed() const                             { return fFirstRunVdriftUsed;     }
 132   Int_t    GetVersionVdriftUsed() const                              { return fVersionVdriftUsed;      }
 133   Int_t    GetSubVersionVdriftUsed() const                           { return fSubVersionVdriftUsed;   }
 134   Int_t    GetFirstRunExBUsed() const                                { return fFirstRunExBUsed;        }
 135   Int_t    GetVersionExBUsed() const                                 { return fVersionExBUsed;         }
 136   Int_t    GetSubVersionExBUsed() const                              { return fSubVersionExBUsed;      }
 137
 138
 139   // Internal functions
 140
 141   void CalibVdriftT0(const Char_t* file, Int_t startRunNumber, Int_t endRunNumber, TString ocdbStorage="");
 142   void CalibExbAlt(const Char_t* file, Int_t startRunNumber, Int_t endRunNumber, TString ocdbStorage="");
 143   void CalibGain(const Char_t* file, Int_t startRunNumber, Int_t endRunNumber,  TString  ocdbStorage="");
 144   void CalibPRF(const Char_t* file, Int_t startRunNumber, Int_t endRunNumber,  TString  ocdbStorage="");
 145   void CalibChamberStatus(const Char_t* file, Int_t startRunNumber, Int_t endRunNumber, TString ocdbStorage="");
 146
 147   Bool_t ReadStatusGlobal(const Char_t* fileName="CalibObjects.root");
 148   Bool_t ReadGainGlobal(const Char_t* fileName="CalibObjects.root");
 149   Bool_t ReadVdriftT0Global(const Char_t* fileName="CalibObjects.root");
 150   Bool_t ReadVdriftLinearFitGlobal(const Char_t* fileName="CalibObjects.root");
 151   Bool_t ReadExbAltFitGlobal(const Char_t* fileName="CalibObjects.root");
 152   Bool_t ReadPRFGlobal(const Char_t* fileName="CalibObjects.root");
 153
 154   Bool_t AnalyzeGain();
 155   Bool_t AnalyzeVdriftT0();
 156   Bool_t AnalyzeVdriftLinearFit();
 157   Bool_t AnalyzeExbAltFit();
 158   Bool_t AnalyzePRF();
 159   Bool_t AnalyzeChamberStatus();
 160
 161   void CorrectFromDetGainUsed();
 162   void CorrectFromDetVdriftUsed();
 163
 164   void UpdateOCDBT0(Int_t startRunNumber, Int_t endRunNumber, const char* storagePath);
 165   void UpdateOCDBVdrift(Int_t startRunNumber, Int_t endRunNumber, const char* storagePath);
 166   void UpdateOCDBExB(Int_t startRunNumber, Int_t endRunNumber, const Char_t *storagePath);
 167   void UpdateOCDBExBAlt(Int_t startRunNumber, Int_t endRunNumber, const Char_t *storagePath);
 168   void UpdateOCDBGain(Int_t  startRunNumber, Int_t endRunNumber, const char* storagePath);
 169   void UpdateOCDBPRF(Int_t  startRunNumber, Int_t endRunNumber, const char* storagePath);
 170   void UpdateOCDBChamberStatus(Int_t startRunNumber, Int_t endRunNumber, const Char_t *storagePath);
 171
 172   Bool_t ValidateGain();
 173   Bool_t ValidateVdrift();
 174   Bool_t ValidateExB();
 175   Bool_t ValidateT0();
 176   Bool_t ValidatePRF() const;
 177   Bool_t ValidateChamberStatus();
 178
 179   Int_t    GetStatus() const;
 180   Int_t    GetStatusPos() const                                      { return fStatusPos;              }
 181   Int_t    GetStatusNeg() const                                      { return fStatusNeg;              }
 182
 183
 184
 185  private:
 186   Bool_t fMethodSecond;                      // Second Method for drift velocity
 187   TString fNameList;                         // Name of the list
 188   AliTRDCalDet *fCalDetGainUsed;             // CalDet used and to be corrected for
 189   AliTRDCalDet *fCalDetVdriftUsed;           // CalDet used and to be corrected for
 190   AliTRDCalDet *fCalDetExBUsed;              // CalDet used and to be corrected for
 191   TH2I *fCH2d;                               // Gain
 192   TProfile2D *fPH2d;                         // Drift velocity first method
 193   TProfile2D *fPRF2d;                        // PRF
 194   THnSparseI *fSparse;                       // chamberstatus
 195   AliTRDCalibraVdriftLinearFit *fAliTRDCalibraVdriftLinearFit; // Drift velocity second method
 196   AliTRDCalibraExbAltFit* fAliTRDCalibraExbAltFit; //ExB alternative method
 197   TH1I *fNEvents;                         // Number of events
 198   TH2F *fAbsoluteGain;                    // Absolute Gain calibration
 199   TObjArray * fPlots;                     // array with some plots to check
 200   TObjArray * fCalibObjects;              // array with calibration objects
 201   Int_t    fFirstRunGainUsed;             // first run gain used
 202   Int_t    fVersionGainUsed;              // VersionGainUsed
 203   Int_t    fSubVersionGainUsed;           // SubVersionGainUsed
 204   Int_t    fFirstRunVdriftUsed;           // FirstRunVdrift
 205   Int_t    fVersionVdriftUsed;            // VersionVdriftUsed
 206   Int_t    fSubVersionVdriftUsed;         // SubVersionVdriftUsed
 207   Int_t    fFirstRunExBUsed;              // FirstRunExB
 208   Int_t    fVersionExBUsed;               // VersionExBUsed
 209   Int_t    fSubVersionExBUsed;            // SubVersionExBUsed
 210   Bool_t   fNoExBUsedInReco;              // ExB not used yet in the reco
 211   Bool_t   fSwitchOnValidation;           // Validation
 212   Bool_t   fVdriftValidated;              // Vdrift validation
 213   Bool_t   fExBValidated;                 // ExB validation
 214   Bool_t   fT0Validated;                  // T0 validation
 215   Int_t    fMinStatsVdriftT0PH;           // MinStats VdriftT0
 216   Int_t    fMinStatsVdriftLinear;         // MinStats Vdrift Linear
 217   Int_t    fMinStatsGain;                 // MinStats Gain
 218   Int_t    fMinStatsPRF;                  // MinStats PRF
 219   Int_t    fMinStatsChamberStatus;        // MinStats ChamberStatus
 220   Bool_t   fBackCorrectGain;              // Back correction afterwards gain
 221   Bool_t   fBackCorrectVdrift;            // Back correction afterwards vdrift
 222   Bool_t   fNotEnoughStatisticsForTheGain;// Take the chamber per chamber distribution from the default distribution
 223   Bool_t   fNotEnoughStatisticsForTheVdriftLinear;// Take the chamber per chamber distribution from the default distribution
 224   Int_t    fStatusNeg;                    // Info but ok
 225   Int_t    fStatusPos;                    // Problems
 226   Int_t    fBadCalib[18];                 // number of bad calibrated chambers per sm
 227   Int_t    fNoData[18];                   // number of  chambers w/o data per sm
 228   Int_t    fBadCalibValidate;             // validation limit for bad calibrated chambers
 229   Int_t    fNoDataValidate;               // validation limit for chamber w/o data (sm w/o data excluded)
 230   Double_t fRMSBadCalibratedGain;         // value to decide when it is bad calibrated
 231   Double_t fRMSBadCalibratedVdrift;       // value to decide when it is bad calibrated
 232   Double_t fRMSBadCalibratedExB;          // value to decide when it is bad calibrated
 233   Bool_t   fRobustFitDriftVelocity;       // Robust fit for the drift velocity
 234   Bool_t   fRobustFitExbAlt;              // Robust fit for the exb alt
 235
 236   Int_t GetSubVersion(TString name) const;
 237   Int_t GetVersion(TString name) const;
 238   Int_t GetFirstRun(TString name) const;
 239
 240
 241
 242 private:
 243   AliTRDPreprocessorOffline& operator=(const AliTRDPreprocessorOffline&); // not implemented
 244   AliTRDPreprocessorOffline(const AliTRDPreprocessorOffline&); // not implemented
 245   ClassDef(AliTRDPreprocessorOffline,3)
 246 };
 247
 248 #endif
 249
 250