New version of fit of alternative ExB method (Theo)
[u/mrichter/AliRoot.git] / TRD / AliTRDPreprocessorOffline.h
CommitLineData
b3fcfd96 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"
81a5aeca 14#include "AliTRDCalChamberStatus.h"
83d0cc79 15#include <THnSparse.h>
b3fcfd96 16class TObjArray;
01239968 17class AliTRDCalDet;
b3fcfd96 18class TH2I;
19class TProfile2D;
20class AliTRDCalibraVdriftLinearFit;
a0bb5615 21class AliTRDCalibraExbAltFit;
b3fcfd96 22class TH1I;
23class TH2F;
01239968 24class TString;
b3fcfd96 25
26
27class AliTRDPreprocessorOffline:public TNamed {
28public:
a0bb5615 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 };
840ec79d 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};
82b413fd 54
55
b3fcfd96 56 AliTRDPreprocessorOffline();
57 virtual ~AliTRDPreprocessorOffline();
58
83d0cc79 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;};
2a1a7b36 79 void SetRobustFitDriftVelocity(Bool_t robustFitDriftVelocity) { fRobustFitDriftVelocity = robustFitDriftVelocity;};
80 void SetRobustFitExbAlt(Bool_t robustFitExbAlt) { fRobustFitExbAlt = robustFitExbAlt;};
661c7be6 81 void SetAlternativeDriftVelocityFit(Bool_t alt) { fAlternativeVdrfitFit = alt;};
82 void SetAlternativeExbAltFit(Bool_t alt) { fAlternativeExbAltFit = alt;};
2a1a7b36 83
83d0cc79 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
82b413fd 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
840ec79d 101 { return CheckStatus(fStatusNeg, kTimeOffsetNotEnoughStatsNotFill); };
102 Bool_t IsExBErrorRange() const
103 { return CheckStatus(fStatusNeg, kExBErrorRange); };
82b413fd 104
105 Bool_t IsGainErrorOld() const
106 { return CheckStatus(fStatusPos, kGainErrorOld); };
107 Bool_t IsVdriftErrorOld() const
108 { return CheckStatus(fStatusPos, kVdriftErrorOld); };
840ec79d 109 Bool_t IsExBErrorOld() const
110 { return CheckStatus(fStatusPos, kExBErrorOld); };
82b413fd 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
82b413fd 120
83d0cc79 121 // Back corrections
01239968 122 void SetCalDetGain(AliTRDCalDet *calDetGainUsed) {fCalDetGainUsed = calDetGainUsed;};
840ec79d 123 void SetCalDetVdrift(AliTRDCalDet *calDetVdriftUsed);
124 void SetCalDetVdriftExB(AliTRDCalDet *calDetVdriftUsed,AliTRDCalDet *calDetExBUsed) {fCalDetVdriftUsed = calDetVdriftUsed; fCalDetExBUsed = calDetExBUsed;};
83d0cc79 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
01239968 128 AliTRDCalDet *GetCalDetGain() const { return fCalDetGainUsed;};
4c865c34 129 AliTRDCalDet *GetCalDetVdrift() const { return fCalDetVdriftUsed;};
83d0cc79 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
b3fcfd96 142
143 void CalibVdriftT0(const Char_t* file, Int_t startRunNumber, Int_t endRunNumber, TString ocdbStorage="");
a0bb5615 144 void CalibExbAlt(const Char_t* file, Int_t startRunNumber, Int_t endRunNumber, TString ocdbStorage="");
b3fcfd96 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="");
83d0cc79 147 void CalibChamberStatus(const Char_t* file, Int_t startRunNumber, Int_t endRunNumber, TString ocdbStorage="");
81a5aeca 148
83d0cc79 149 Bool_t ReadStatusGlobal(const Char_t* fileName="CalibObjects.root");
b3fcfd96 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");
a0bb5615 153 Bool_t ReadExbAltFitGlobal(const Char_t* fileName="CalibObjects.root");
b3fcfd96 154 Bool_t ReadPRFGlobal(const Char_t* fileName="CalibObjects.root");
155
156 Bool_t AnalyzeGain();
157 Bool_t AnalyzeVdriftT0();
158 Bool_t AnalyzeVdriftLinearFit();
a0bb5615 159 Bool_t AnalyzeExbAltFit();
a5dcf618 160 Bool_t AnalyzePRF();
161 Bool_t AnalyzeChamberStatus();
b3fcfd96 162
01239968 163 void CorrectFromDetGainUsed();
4c865c34 164 void CorrectFromDetVdriftUsed();
a5dcf618 165
b3fcfd96 166 void UpdateOCDBT0(Int_t startRunNumber, Int_t endRunNumber, const char* storagePath);
167 void UpdateOCDBVdrift(Int_t startRunNumber, Int_t endRunNumber, const char* storagePath);
eec29d56 168 void UpdateOCDBExB(Int_t startRunNumber, Int_t endRunNumber, const Char_t *storagePath);
a0bb5615 169 void UpdateOCDBExBAlt(Int_t startRunNumber, Int_t endRunNumber, const Char_t *storagePath);
b3fcfd96 170 void UpdateOCDBGain(Int_t startRunNumber, Int_t endRunNumber, const char* storagePath);
171 void UpdateOCDBPRF(Int_t startRunNumber, Int_t endRunNumber, const char* storagePath);
a5dcf618 172 void UpdateOCDBChamberStatus(Int_t startRunNumber, Int_t endRunNumber, const Char_t *storagePath);
b3fcfd96 173
54f2ff1c 174 Bool_t ValidateGain();
00d203b6 175 Bool_t ValidateVdrift();
840ec79d 176 Bool_t ValidateExB();
00d203b6 177 Bool_t ValidateT0();
178 Bool_t ValidatePRF() const;
82b413fd 179 Bool_t ValidateChamberStatus();
00d203b6 180
82b413fd 181 Int_t GetStatus() const;
182 Int_t GetStatusPos() const { return fStatusPos; }
183 Int_t GetStatusNeg() const { return fStatusNeg; }
a5dcf618 184
83d0cc79 185
4c865c34 186
187 private:
81a5aeca 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
840ec79d 192 AliTRDCalDet *fCalDetExBUsed; // CalDet used and to be corrected for
81a5aeca 193 TH2I *fCH2d; // Gain
194 TProfile2D *fPH2d; // Drift velocity first method
195 TProfile2D *fPRF2d; // PRF
83d0cc79 196 THnSparseI *fSparse; // chamberstatus
b3fcfd96 197 AliTRDCalibraVdriftLinearFit *fAliTRDCalibraVdriftLinearFit; // Drift velocity second method
a0bb5615 198 AliTRDCalibraExbAltFit* fAliTRDCalibraExbAltFit; //ExB alternative method
b3fcfd96 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
83d0cc79 203 Int_t fFirstRunGainUsed; // first run gain used
4c865c34 204 Int_t fVersionGainUsed; // VersionGainUsed
205 Int_t fSubVersionGainUsed; // SubVersionGainUsed
ca7e6e64 206 Int_t fFirstRunVdriftUsed; // FirstRunVdrift
4c865c34 207 Int_t fVersionVdriftUsed; // VersionVdriftUsed
208 Int_t fSubVersionVdriftUsed; // SubVersionVdriftUsed
83d0cc79 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
00d203b6 213 Bool_t fSwitchOnValidation; // Validation
214 Bool_t fVdriftValidated; // Vdrift validation
840ec79d 215 Bool_t fExBValidated; // ExB validation
00d203b6 216 Bool_t fT0Validated; // T0 validation
a2a4ec8e 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
83d0cc79 221 Int_t fMinStatsChamberStatus; // MinStats ChamberStatus
54f2ff1c 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
82b413fd 226 Int_t fStatusNeg; // Info but ok
227 Int_t fStatusPos; // Problems
83d0cc79 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
2a1a7b36 235 Bool_t fRobustFitDriftVelocity; // Robust fit for the drift velocity
236 Bool_t fRobustFitExbAlt; // Robust fit for the exb alt
661c7be6 237 Bool_t fAlternativeVdrfitFit; // Alternative fitting method for vdrift calibration
238 Bool_t fAlternativeExbAltFit; // Alternative fitting method for the alternative exb calibarion method
b3fcfd96 239
4c865c34 240 Int_t GetSubVersion(TString name) const;
241 Int_t GetVersion(TString name) const;
ca7e6e64 242 Int_t GetFirstRun(TString name) const;
b3fcfd96 243
244
245
246private:
247 AliTRDPreprocessorOffline& operator=(const AliTRDPreprocessorOffline&); // not implemented
248 AliTRDPreprocessorOffline(const AliTRDPreprocessorOffline&); // not implemented
2a1a7b36 249 ClassDef(AliTRDPreprocessorOffline,3)
b3fcfd96 250};
251
252#endif
01239968 253
a5dcf618 254