Adding macros to create Calibration objects
[u/mrichter/AliRoot.git] / TRD / TRDbase / AliTRDrecoParam.h
1 #ifndef ALITRDRECOPARAM_H
2 #define ALITRDRECOPARAM_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice                               */
5
6 /* $Id$ */
7
8 ////////////////////////////////////////////////////////////////////////////
9 //                                                                        //
10 //  Parameter class for the TRD reconstruction                            //
11 //                                                                        //
12 ////////////////////////////////////////////////////////////////////////////
13
14 #ifndef ALIDETECTORRECOPARAM_H
15 #include "AliDetectorRecoParam.h"
16 #endif
17
18 #ifndef ALITRDCALPID_H
19 #include "AliTRDCalPID.h"
20 #endif
21
22 #ifndef ALITRDPIDRESPONSE_H
23 #include "AliTRDPIDResponse.h"
24 #endif
25
26 class TString;
27
28 class AliTRDrecoParam : public AliDetectorRecoParam
29 {
30 public:
31   enum ETRDReconstructionTask{
32     kClusterizer = 0,
33     kTracker = 1,
34     kPID = 2,
35     kTRDreconstructionTasks = 3
36   };
37   enum ETRDflags {
38      kDriftGas
39     ,kVertexConstraint
40     ,kTailCancelation
41     ,kImproveTracklet
42     ,kLUT
43     ,kGAUS
44     ,kClusterSharing
45     ,kSteerPID
46     ,kEightSlices
47     ,kCheckTimeConsistency
48     ,kLQ2D
49   };
50   AliTRDrecoParam();
51   AliTRDrecoParam(const AliTRDrecoParam &rec);
52   AliTRDrecoParam& operator=(const AliTRDrecoParam &rec);
53   ~AliTRDrecoParam() { }
54
55   Double_t GetChi2Y() const                 { return fkChi2Y;    }
56   Double_t GetChi2Z() const                 { return fkChi2Z;    }
57   Double_t GetChi2YSlope() const            { return fkChi2YSlope; }
58   Double_t GetChi2ZSlope() const            { return fkChi2ZSlope; }
59   Double_t GetChi2Cut() const               { return fChi2Cut; }
60   Double_t GetChi2YCut() const              { return fkChi2YCut; }
61   Double_t GetPhiSlope() const              { return fkPhiSlope;   }
62   Float_t  GetNClusters() const;
63   Double_t GetNMeanClusters() const         { return fkNMeanClusters; }
64   Double_t GetNSigmaClusters() const        { return fkNSigmaClusters; }
65   Double_t GetFindableClusters() const      { return fkFindable; }
66   Int_t    GetPIDLQslices() const;
67   AliTRDPIDResponse::ETRDPIDMethod GetPIDmethod() const;
68   Double_t GetMaxTheta() const              { return fkMaxTheta; }
69   Double_t GetMaxPhi() const                { return fkMaxPhi;   }
70   Double_t GetPlaneQualityThreshold() const { return fkPlaneQualityThreshold; }
71   Double_t GetPIDThreshold(Float_t /*p*/) const { return 0.;}
72   Double_t GetRoad0y() const                { return fkRoad0y;   }
73   Double_t GetRoad0z() const                { return fkRoad0z;   }
74   Double_t GetRoad1y() const                { return fkRoad1y;   }
75   Double_t GetRoad1z() const                { return fkRoad1z;   }
76   Double_t GetRoad2y() const                { return fkRoad2y;   }
77   Double_t GetRoad2z() const                { return fkRoad2z;   }
78   Double_t GetRoadzMultiplicator() const    { return fkRoadzMultiplicator; }
79   Double_t GetTrackLikelihood() const       { return fkTrackLikelihood;       }
80   void GetSysCovMatrix(Double_t *sys) const;  
81   void GetTCParams(Double_t *par) const;
82   Int_t GetStreamLevel(ETRDReconstructionTask task) const;
83   const TString *GetRawStreamVersion() const{ return &fRawStreamVersion;    };
84   Double_t GetMinMaxCutSigma() const        { return fMinMaxCutSigma;       };
85   Double_t GetMinLeftRightCutSigma() const  { return fMinLeftRightCutSigma; };
86   Double_t GetClusMaxThresh() const         { return fClusMaxThresh;        };
87   Double_t GetClusSigThresh() const         { return fClusSigThresh;        };
88   Int_t    GetTCnexp() const                { return fTCnexp;               };
89   Int_t    GetNumberOfPresamples()  const   { return fNumberOfPresamples;   };
90   Int_t    GetNumberOfPostsamples() const   { return fNumberOfPostsamples;  };
91   Int_t    GetNumberOfSeedConfigs() const   { return fNumberOfConfigs;      };
92   Int_t    GetRecEveryNTB() const           { return fRecEveryNTB;          };
93   Double_t GetClusterQmin() const           { return fClusterQmin;          };
94
95   // Tracklet parameters
96   Double_t GetCorrDZDXbiasRC(Bool_t dzdx) const { return fdzdxCorrRCbias[dzdx];}
97   Double_t GetCorrDZDX(Bool_t rc) const     { return fdzdxCorrFactor[rc];}
98   Double_t GetCorrDZDXxcross() const        { return fdzdxXcrossFactor;}
99   void     GetYcorrTailCancel(Int_t it, Double_t par[3]) const;
100   Double_t GetS2Ycorr(Bool_t rc, Bool_t chg) const     { return fS2Ycorr[2*rc+chg];}
101
102   Bool_t   IsArgon() const                  { return TESTBIT(fFlags, kDriftGas); }
103   Bool_t   IsCheckTimeConsistency() const   { return kCheckTimeConsistency;}
104   Bool_t   IsOverPtThreshold(Double_t pt) const {return Bool_t(pt>fkPtThreshold);}
105   Bool_t   IsXenon() const                  { return !TESTBIT(fFlags, kDriftGas); }
106   Bool_t   IsPIDNeuralNetwork() const       { return TESTBIT(fFlags, kSteerPID);}
107   Bool_t   IsVertexConstrained() const      { return TESTBIT(fFlags, kVertexConstraint); }
108   Bool_t   IsEightSlices() const            { return TESTBIT(fFlags, kEightSlices);}
109   Bool_t   HasImproveTracklets() const      { return TESTBIT(fFlags, kImproveTracklet);}
110   Bool_t   UseClusterSharing() const        { return TESTBIT(fFlags, kClusterSharing);}
111   Bool_t   UseLUT() const                   { return TESTBIT(fFlags, kLUT);}
112   Bool_t   UseGAUS() const                  { return TESTBIT(fFlags, kGAUS);}
113   Bool_t   UseTailCancelation() const       { return TESTBIT(fFlags, kTailCancelation); }
114         
115   static   AliTRDrecoParam *GetLowFluxParam();
116   static   AliTRDrecoParam *GetLowFluxHLTParam();
117   static   AliTRDrecoParam *GetHighFluxParam();
118   static   AliTRDrecoParam *GetHighFluxHLTParam();
119   static   AliTRDrecoParam *GetCosmicTestParam();
120
121   void     SetArgon(Bool_t b = kTRUE)                         {if(b) SETBIT(fFlags, kDriftGas); else CLRBIT(fFlags, kDriftGas);}
122   void     SetCheckTimeConsistency(Bool_t b = kTRUE)          {if(b) SETBIT(fFlags, kCheckTimeConsistency); else CLRBIT(fFlags, kCheckTimeConsistency);}
123   void     SetClusterSharing(Bool_t b = kTRUE)                {if(b) SETBIT(fFlags, kClusterSharing); else CLRBIT(fFlags, kClusterSharing);}
124   void     SetEightSlices(Bool_t b = kTRUE)                   {if(b) SETBIT(fFlags, kEightSlices); else CLRBIT(fFlags, kEightSlices);}
125   void     SetImproveTracklets(Bool_t b = kTRUE)              {if(b) SETBIT(fFlags, kImproveTracklet); else CLRBIT(fFlags, kImproveTracklet);}
126   void     SetLUT(Bool_t b=kTRUE)                             {if(b) SETBIT(fFlags, kLUT); else CLRBIT(fFlags, kLUT);}
127   void     SetGAUS(Bool_t b=kTRUE)                            {if(b) SETBIT(fFlags, kGAUS); else CLRBIT(fFlags, kGAUS);}
128   void     SetPIDNeuralNetwork(Bool_t b=kTRUE)                {if(b) SETBIT(fFlags, kSteerPID); else CLRBIT(fFlags, kSteerPID);}
129   void     SetPIDmethod(AliTRDPIDResponse::ETRDPIDMethod method);
130   void     SetPIDLQslices(Int_t s);
131   void     SetTailCancelation(Bool_t b=kTRUE)                 {if(b) SETBIT(fFlags, kTailCancelation); else CLRBIT(fFlags, kTailCancelation);}
132   void     SetXenon(Bool_t b = kTRUE)                         {if(b) CLRBIT(fFlags, kDriftGas); else SETBIT(fFlags, kDriftGas);}
133   void     SetVertexConstrained()                             {SETBIT(fFlags, kVertexConstraint);}
134   void     SetMaxTheta(Double_t maxTheta)                     {fkMaxTheta = maxTheta;}
135   void     SetMaxPhi(Double_t maxPhi)                         {fkMaxPhi = maxPhi;}
136   void     SetFindableClusters(Double_t r)                    {fkFindable = r;}
137   void     SetChi2Y(Double_t chi2)                            {fkChi2Y = chi2;}
138   void     SetChi2Z(Double_t chi2)                            {fkChi2Z = chi2;}
139   void     SetChi2YSlope(Double_t chi2YSlope)                 {fkChi2YSlope = chi2YSlope;}
140   void     SetChi2ZSlope(Double_t chi2ZSlope)                 {fkChi2ZSlope = chi2ZSlope;}
141   void     SetChi2Cut(Double_t chi2Cut)                       {fChi2Cut = chi2Cut; }
142   void     SetChi2YCut(Double_t chi2Cut)                      {fkChi2YCut = chi2Cut; }
143   void     SetPhiSlope(Double_t phiSlope)                     {fkPhiSlope = phiSlope;}
144   void     SetNMeanClusters(Double_t meanNclusters)           {fkNMeanClusters = meanNclusters;}
145   void     SetNSigmaClusters(Double_t sigmaNclusters)         {fkNSigmaClusters = sigmaNclusters;} 
146   void     SetRawStreamVersion(const Char_t *version)         {fRawStreamVersion = version; }
147   void     SetRoadzMultiplicator(Double_t mult)               {fkRoadzMultiplicator = mult; } 
148   void     SetMinMaxCutSigma(Float_t minMaxCutSigma)          { fMinMaxCutSigma   = minMaxCutSigma; }
149   void     SetMinLeftRightCutSigma(Float_t minLeftRightCutSigma) { fMinLeftRightCutSigma   = minLeftRightCutSigma; };
150   void     SetClusMaxThresh(Float_t thresh)                   { fClusMaxThresh   = thresh; };
151   void     SetClusSigThresh(Float_t thresh)                   { fClusSigThresh   = thresh; };
152   void     SetPIDThreshold(Double_t *pid);
153   void     SetPtThreshold(Double_t pt) {fkPtThreshold = pt;}
154   void     SetNexponential(Int_t nexp)                        { fTCnexp          = nexp;   };
155   void     SetTCParams(Double_t *par);
156   void     SetTrackletParams(Double_t *par=NULL);
157   void     SetStreamLevel(ETRDReconstructionTask task, Int_t level);
158   void     SetSysCovMatrix(Double_t *sys);
159   void     SetNumberOfPresamples(Int_t n)                     { fNumberOfPresamples = n;}
160   void     SetNumberOfPostsamples(Int_t n)                    { fNumberOfPostsamples = n;}
161   void     SetRecEveryTwoTB()                                 { fRecEveryNTB = 2; fkNMeanClusters = 10; }
162   void     SetClusterQmin(Double_t min)                       { fClusterQmin = min; };
163
164 private:
165   // Physics reference values for TRD
166   Double_t  fkdNchdy;                // dNch/dy
167   Double_t  fkMaxTheta;              // Maximum theta
168   Double_t  fkMaxPhi;                // Maximum phi - momentum cut
169   // Tracker params 
170   Double_t  fkRoad0y;                // Road for middle cluster
171   Double_t  fkRoad0z;                // Road for middle cluster
172
173   Double_t  fkRoad1y;                // Road in y for seeded cluster
174   Double_t  fkRoad1z;                // Road in z for seeded cluster
175
176   Double_t  fkRoad2y;                // Road in y for extrapolated cluster
177   Double_t  fkRoad2z;                // Road in z for extrapolated cluster
178   Double_t  fkPtThreshold;           // pt threshold for using TRD points for updating Kalaman track
179   Double_t  fkPlaneQualityThreshold; // Quality threshold
180   Double_t  fkRoadzMultiplicator;    // Multiplicator for the Roads in z 
181   Double_t  fkFindable;              // minimum ratio of clusters per tracklet supposed to be attached.
182   Double_t  fkChi2Z;                 // Max chi2 on the z direction for seeding clusters fit
183   Double_t  fkChi2Y;                 // Max chi2 on the y direction for seeding clusters Rieman fit
184   Double_t  fkChi2YSlope;            // Slope of the chi2-distribution in y-direction
185   Double_t  fkChi2ZSlope;            // Slope of the chi2-distribution in z-direction
186   Double_t  fChi2Cut;                // Cut on the Chi2 track/tracklet 0 used to diecide if the kalman track should be updated
187   Double_t  fkChi2YCut;                                                  // Cut on the Chi2 in y-direction in the likelihood filter
188   Double_t  fkPhiSlope;              // Slope of the distribution of the deviation between track angle and tracklet angle
189   Double_t  fkNMeanClusters;         // Mean number of clusters per tracklet
190   Double_t  fkNSigmaClusters;        // Sigma of the number of clusters per tracklet
191   Double_t  fkNClusterNoise;         // ratio of noisy clusters to the true one
192   Double_t  fkNMeanTracklets;        // Mean number of tracklets per track
193   Double_t  fkTrackLikelihood;       // Track likelihood for tracklets Rieman fit
194   
195   Double_t  fSysCovMatrix[5];        // Systematic uncertainty from calibration and alignment for each tracklet
196   Double_t  fPIDThreshold[AliTRDCalPID::kNMom];   // PID Thresholds for Electron candidate decision
197   Int_t     fNumberOfConfigs;        // Used number of seed configurations
198
199   // Reconstruction Options for TRD reconstruction
200   Int_t     fStreamLevel[kTRDreconstructionTasks]; // Stream Level
201   Long64_t  fFlags;                  // option Flags
202
203   // Raw Reader Params
204   TString   fRawStreamVersion;       // Raw Reader version
205
206   // Tracklet parameters
207   Double_t  fdzdxCorrFactor[2];      // correction of dzdx estimation due to z bias; [0] for !RC, [1] for RC
208   Double_t  fdzdxCorrRCbias[2];      // correction of dzdx estimation bias for RC; [0] for dz/dx>0, [1] for dz/dx<0
209   Double_t  fdzdxXcrossFactor;       // bias in dzdx of estimated xcross [RC]
210   Double_t  fYcorrTailCancel[4][3];  // y correction due to wrong tail cancellation. [0] bz<0 && !RC, [1] bz>0 && !RC, [2] bz<0 && RC [3] bz>0 && RC
211   Double_t  fS2Ycorr[4];             // inflation factor of error parameterization in r-phi due to wrong estimation of residuals.
212   
213   // Clusterization parameter
214   Double_t  fMinMaxCutSigma;         // Threshold sigma noise pad middle
215   Double_t  fMinLeftRightCutSigma;   // Threshold sigma noise sum pad
216   Double_t  fClusMaxThresh;          // Threshold value for cluster maximum
217   Double_t  fClusSigThresh;          // Threshold value for cluster signal
218   Int_t     fTCnexp;                 // Number of exponentials, digital filter
219   Double_t  fTCParams[8];            // Tail Cancellation parameters for drift gases 
220   Int_t     fRecEveryNTB;            // Reconstruct each nth timebin
221   Double_t  fClusterQmin;            // Threshold for the total cluster charge to be written to recPoints file
222
223   // ADC parameter
224   Int_t     fNumberOfPresamples;     // number of presamples 
225   Int_t     fNumberOfPostsamples;     // number of postsamples 
226
227   ClassDef(AliTRDrecoParam, 14)       // Reconstruction parameters for TRD detector
228
229 };
230
231 //___________________________________________________
232 inline void AliTRDrecoParam::GetSysCovMatrix(Double_t *sys) const
233 {
234   if(!sys) return;
235   memcpy(sys, fSysCovMatrix, 5*sizeof(Double_t));
236 }
237
238 //___________________________________________________
239 inline void AliTRDrecoParam::SetSysCovMatrix(Double_t *sys)
240 {
241   if(!sys) return;
242   memcpy(fSysCovMatrix, sys, 5*sizeof(Double_t));
243 }
244
245 //___________________________________________________
246 inline void AliTRDrecoParam::SetPIDThreshold(Double_t *pid)
247 {
248   if(!pid) return;
249   memcpy(fPIDThreshold, pid, AliTRDCalPID::kNMom*sizeof(Double_t));
250 }
251
252 //___________________________________________________
253 inline void AliTRDrecoParam::SetStreamLevel(ETRDReconstructionTask task, Int_t level){
254   if(task >= kTRDreconstructionTasks) return;
255   fStreamLevel[static_cast<Int_t>(task)] = level;
256 }
257
258 //___________________________________________________
259 inline Int_t AliTRDrecoParam::GetStreamLevel(ETRDReconstructionTask task) const{
260   if(task >= kTRDreconstructionTasks) return 0;
261   return fStreamLevel[static_cast<Int_t>(task)];
262 }
263
264 //___________________________________________________
265 inline void AliTRDrecoParam::GetTCParams(Double_t *par) const
266 {
267   if(!par) return;
268   if(IsArgon()) memcpy(par, &fTCParams[4], 4*sizeof(Double_t));
269   else memcpy(par, &fTCParams[0], 4*sizeof(Double_t));
270 }
271
272 //___________________________________________________
273 inline void AliTRDrecoParam::SetTCParams(Double_t *par)
274 {
275   if(!par) return;
276   memcpy(fTCParams, par, 8*sizeof(Double_t));
277 }
278
279
280 //___________________________________________________
281 inline void AliTRDrecoParam::GetYcorrTailCancel(Int_t it, Double_t par[3]) const
282 {
283   if(it<0||it>3) return;
284   memcpy(par, fYcorrTailCancel[it], 3*sizeof(Double_t));
285 }
286
287 //___________________________________________________
288 inline Int_t AliTRDrecoParam::GetPIDLQslices() const
289 {
290   if(IsPIDNeuralNetwork()) return -1;
291   return TESTBIT(fFlags, kLQ2D) ? 2 : 1;
292 }
293
294 //___________________________________________________
295 inline AliTRDPIDResponse::ETRDPIDMethod AliTRDrecoParam::GetPIDmethod() const
296 {
297   AliTRDPIDResponse::ETRDPIDMethod method = AliTRDPIDResponse::kLQ1D;
298   if(IsPIDNeuralNetwork()) method = AliTRDPIDResponse::kNN;
299   else if(TESTBIT(fFlags, kLQ2D)) method = AliTRDPIDResponse::kLQ2D;
300   return method;
301 }
302
303 //___________________________________________________
304 inline void  AliTRDrecoParam::SetPIDmethod(AliTRDPIDResponse::ETRDPIDMethod method)
305 {
306   switch(method){
307   case AliTRDPIDResponse::kLQ2D:
308     CLRBIT(fFlags, kSteerPID); 
309     SETBIT(fFlags, kLQ2D);
310     break;
311   case AliTRDPIDResponse::kNN:
312     SETBIT(fFlags, kSteerPID); 
313     break;
314   case AliTRDPIDResponse::kLQ1D:
315   default:
316     CLRBIT(fFlags, kSteerPID); 
317     CLRBIT(fFlags, kLQ2D);
318     break;
319   }
320 }
321
322 #endif