]> git.uio.no Git - u/mrichter/AliRoot.git/blame_incremental - TRD/AliTRDrecoParam.h
possibility to cut on the pt of candidate (Rossella)
[u/mrichter/AliRoot.git] / TRD / AliTRDrecoParam.h
... / ...
CommitLineData
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#ifndef ALITRDCALPID_H
18#include "Cal/AliTRDCalPID.h"
19#endif
20
21class TString;
22
23class AliTRDrecoParam : public AliDetectorRecoParam
24{
25public:
26 enum ETRDReconstructionTask{
27 kClusterizer = 0,
28 kTracker = 1,
29 kPID = 2,
30 kTRDreconstructionTasks = 3
31 };
32 enum ETRDflags {
33 kDriftGas
34 ,kVertexConstraint
35 ,kTailCancelation
36 ,kImproveTracklet
37 ,kLUT
38 ,kGAUS
39 ,kClusterSharing
40 ,kSteerPID
41 ,kEightSlices
42 ,kCheckTimeConsistency
43 ,kLQ2D
44 };
45 AliTRDrecoParam();
46 AliTRDrecoParam(const AliTRDrecoParam &rec);
47 ~AliTRDrecoParam() { }
48
49 Double_t GetChi2Y() const { return fkChi2Y; }
50 Double_t GetChi2Z() const { return fkChi2Z; }
51 Double_t GetChi2YSlope() const { return fkChi2YSlope; }
52 Double_t GetChi2ZSlope() const { return fkChi2ZSlope; }
53 Double_t GetChi2YCut() const { return fkChi2YCut; }
54 Double_t GetPhiSlope() const { return fkPhiSlope; }
55 Float_t GetNClusters() const;
56 Double_t GetNMeanClusters() const { return fkNMeanClusters; }
57 Double_t GetNSigmaClusters() const { return fkNSigmaClusters; }
58 Double_t GetFindableClusters() const { return fkFindable; }
59 inline Int_t GetPIDLQslices() const;
60 Double_t GetMaxTheta() const { return fkMaxTheta; }
61 Double_t GetMaxPhi() const { return fkMaxPhi; }
62 Double_t GetPlaneQualityThreshold() const { return fkPlaneQualityThreshold; }
63 Double_t GetPIDThreshold(Float_t /*p*/) const { return 0.;}
64 Double_t GetRoad0y() const { return fkRoad0y; }
65 Double_t GetRoad0z() const { return fkRoad0z; }
66 Double_t GetRoad1y() const { return fkRoad1y; }
67 Double_t GetRoad1z() const { return fkRoad1z; }
68 Double_t GetRoad2y() const { return fkRoad2y; }
69 Double_t GetRoad2z() const { return fkRoad2z; }
70 Double_t GetRoadzMultiplicator() const { return fkRoadzMultiplicator; }
71 Double_t GetTrackLikelihood() const { return fkTrackLikelihood; }
72 inline void GetSysCovMatrix(Double_t *sys) const;
73 inline void GetTCParams(Double_t *par) const;
74 inline Int_t GetStreamLevel(ETRDReconstructionTask task) const;
75 const TString *GetRawStreamVersion() const{ return &fRawStreamVersion; };
76 Double_t GetMinMaxCutSigma() const { return fMinMaxCutSigma; };
77 Double_t GetMinLeftRightCutSigma() const { return fMinLeftRightCutSigma; };
78 Double_t GetClusMaxThresh() const { return fClusMaxThresh; };
79 Double_t GetClusSigThresh() const { return fClusSigThresh; };
80 Int_t GetTCnexp() const { return fTCnexp; };
81 Int_t GetNumberOfPresamples() const { return fNumberOfPresamples;}
82 Int_t GetNumberOfPostsamples() const { return fNumberOfPostsamples;}
83 Int_t GetNumberOfSeedConfigs() const { return fNumberOfConfigs;}
84 Int_t GetRecEveryNTB() const { return fRecEveryNTB; }
85 Bool_t IsArgon() const { return TESTBIT(fFlags, kDriftGas); }
86 Bool_t IsCheckTimeConsistency() const { return kCheckTimeConsistency;}
87 Bool_t IsOverPtThreshold(Double_t pt) const {return Bool_t(pt>fkPtThreshold);}
88 Bool_t IsXenon() const { return !TESTBIT(fFlags, kDriftGas); }
89 Bool_t IsPIDNeuralNetwork() const { return TESTBIT(fFlags, kSteerPID);}
90 Bool_t IsVertexConstrained() const { return TESTBIT(fFlags, kVertexConstraint); }
91 Bool_t IsEightSlices() const { return TESTBIT(fFlags, kEightSlices);}
92 Bool_t HasImproveTracklets() const { return TESTBIT(fFlags, kImproveTracklet);}
93 Bool_t UseClusterSharing() const { return TESTBIT(fFlags, kClusterSharing);}
94 Bool_t UseLUT() const { return TESTBIT(fFlags, kLUT);}
95 Bool_t UseGAUS() const { return TESTBIT(fFlags, kGAUS);}
96 Bool_t UseTailCancelation() const { return TESTBIT(fFlags, kTailCancelation); }
97
98 static AliTRDrecoParam *GetLowFluxParam();
99 static AliTRDrecoParam *GetLowFluxHLTParam();
100 static AliTRDrecoParam *GetHighFluxParam();
101 static AliTRDrecoParam *GetHighFluxHLTParam();
102 static AliTRDrecoParam *GetCosmicTestParam();
103
104 void SetArgon(Bool_t b = kTRUE) {if(b) SETBIT(fFlags, kDriftGas); else CLRBIT(fFlags, kDriftGas);}
105 void SetCheckTimeConsistency(Bool_t b = kTRUE) {if(b) SETBIT(fFlags, kCheckTimeConsistency); else CLRBIT(fFlags, kCheckTimeConsistency);}
106 void SetClusterSharing(Bool_t b = kTRUE) {if(b) SETBIT(fFlags, kClusterSharing); else CLRBIT(fFlags, kClusterSharing);}
107 void SetEightSlices(Bool_t b = kTRUE) {if(b) SETBIT(fFlags, kEightSlices); else CLRBIT(fFlags, kEightSlices);}
108 void SetImproveTracklets(Bool_t b = kTRUE) {if(b) SETBIT(fFlags, kImproveTracklet); else CLRBIT(fFlags, kImproveTracklet);}
109 void SetLUT(Bool_t b=kTRUE) {if(b) SETBIT(fFlags, kLUT); else CLRBIT(fFlags, kLUT);}
110 void SetGAUS(Bool_t b=kTRUE) {if(b) SETBIT(fFlags, kGAUS); else CLRBIT(fFlags, kGAUS);}
111 void SetPIDNeuralNetwork(Bool_t b=kTRUE) {if(b) SETBIT(fFlags, kSteerPID); else CLRBIT(fFlags, kSteerPID);}
112 void SetPIDLQslices(Int_t s);
113 void SetTailCancelation(Bool_t b=kTRUE) {if(b) SETBIT(fFlags, kTailCancelation); else CLRBIT(fFlags, kTailCancelation);}
114 void SetXenon(Bool_t b = kTRUE) {if(b) CLRBIT(fFlags, kDriftGas); else SETBIT(fFlags, kDriftGas);}
115 void SetVertexConstrained() {SETBIT(fFlags, kVertexConstraint);}
116 void SetMaxTheta(Double_t maxTheta) {fkMaxTheta = maxTheta;}
117 void SetMaxPhi(Double_t maxPhi) {fkMaxPhi = maxPhi;}
118 void SetFindableClusters(Double_t r) {fkFindable = r;}
119 void SetChi2Y(Double_t chi2) {fkChi2Y = chi2;}
120 void SetChi2Z(Double_t chi2) {fkChi2Z = chi2;}
121 void SetChi2YSlope(Double_t chi2YSlope) {fkChi2YSlope = chi2YSlope;}
122 void SetChi2ZSlope(Double_t chi2ZSlope) {fkChi2ZSlope = chi2ZSlope;}
123 void SetChi2YCut(Double_t chi2Cut) {fkChi2YCut = chi2Cut; }
124 void SetPhiSlope(Double_t phiSlope) {fkPhiSlope = phiSlope;}
125 void SetNMeanClusters(Double_t meanNclusters) {fkNMeanClusters = meanNclusters;}
126 void SetNSigmaClusters(Double_t sigmaNclusters) {fkNSigmaClusters = sigmaNclusters;}
127 void SetRawStreamVersion(const Char_t *version) {fRawStreamVersion = version; }
128 void SetRoadzMultiplicator(Double_t mult) {fkRoadzMultiplicator = mult; }
129 void SetMinMaxCutSigma(Float_t minMaxCutSigma) { fMinMaxCutSigma = minMaxCutSigma; }
130 void SetMinLeftRightCutSigma(Float_t minLeftRightCutSigma) { fMinLeftRightCutSigma = minLeftRightCutSigma; };
131 void SetClusMaxThresh(Float_t thresh) { fClusMaxThresh = thresh; };
132 void SetClusSigThresh(Float_t thresh) { fClusSigThresh = thresh; };
133 inline void SetPIDThreshold(Double_t *pid);
134 void SetPtThreshold(Double_t pt) {fkPtThreshold = pt;}
135 void SetNexponential(Int_t nexp) { fTCnexp = nexp; };
136 inline void SetTCParams(Double_t *par);
137 inline void SetStreamLevel(ETRDReconstructionTask task, Int_t level);
138 inline void SetSysCovMatrix(Double_t *sys);
139 void SetNumberOfPresamples(Int_t n) { fNumberOfPresamples = n;}
140 void SetNumberOfPostsamples(Int_t n) { fNumberOfPostsamples = n;}
141 void SetRecEveryTwoTB() { fRecEveryNTB = 2; fkNMeanClusters = 10; }
142
143private:
144 // Physics reference values for TRD
145 Double_t fkdNchdy; // dNch/dy
146 Double_t fkMaxTheta; // Maximum theta
147 Double_t fkMaxPhi; // Maximum phi - momentum cut
148 // Tracker params
149 Double_t fkRoad0y; // Road for middle cluster
150 Double_t fkRoad0z; // Road for middle cluster
151
152 Double_t fkRoad1y; // Road in y for seeded cluster
153 Double_t fkRoad1z; // Road in z for seeded cluster
154
155 Double_t fkRoad2y; // Road in y for extrapolated cluster
156 Double_t fkRoad2z; // Road in z for extrapolated cluster
157 Double_t fkPtThreshold; // pt threshold for using TRD points for updating Kalaman track
158 Double_t fkPlaneQualityThreshold; // Quality threshold
159 Double_t fkRoadzMultiplicator; // Multiplicator for the Roads in z
160 Double_t fkFindable; // minimum ratio of clusters per tracklet supposed to be attached.
161 Double_t fkChi2Z; // Max chi2 on the z direction for seeding clusters fit
162 Double_t fkChi2Y; // Max chi2 on the y direction for seeding clusters Rieman fit
163 Double_t fkChi2YSlope; // Slope of the chi2-distribution in y-direction
164 Double_t fkChi2ZSlope; // Slope of the chi2-distribution in z-direction
165 Double_t fkChi2YCut; // Cut on the Chi2 in y-direction in the likelihood filter
166 Double_t fkPhiSlope; // Slope of the distribution of the deviation between track angle and tracklet angle
167 Double_t fkNMeanClusters; // Mean number of clusters per tracklet
168 Double_t fkNSigmaClusters; // Sigma of the number of clusters per tracklet
169 Double_t fkNClusterNoise; // ratio of noisy clusters to the true one
170 Double_t fkNMeanTracklets; // Mean number of tracklets per track
171 Double_t fkTrackLikelihood; // Track likelihood for tracklets Rieman fit
172
173 Double_t fSysCovMatrix[5]; // Systematic uncertainty from calibration and alignment for each tracklet
174 Double_t fPIDThreshold[AliTRDCalPID::kNMom]; // PID Thresholds for Electron candidate decision
175 Int_t fNumberOfConfigs; // Used number of seed configurations
176
177 // Reconstruction Options for TRD reconstruction
178 Int_t fStreamLevel[kTRDreconstructionTasks]; // Stream Level
179 Long64_t fFlags; // option Flags
180
181 // Raw Reader Params
182 TString fRawStreamVersion; // Raw Reader version
183
184 // Clusterization parameter
185 Double_t fMinMaxCutSigma; // Threshold sigma noise pad middle
186 Double_t fMinLeftRightCutSigma; // Threshold sigma noise sum pad
187 Double_t fClusMaxThresh; // Threshold value for cluster maximum
188 Double_t fClusSigThresh; // Threshold value for cluster signal
189 Int_t fTCnexp; // Number of exponentials, digital filter
190 Double_t fTCParams[8]; // Tail Cancellation parameters for drift gases
191 Int_t fRecEveryNTB; // Reconstruct each nth timebin
192
193 // ADC parameter
194 Int_t fNumberOfPresamples; // number of presamples
195 Int_t fNumberOfPostsamples; // number of postsamples
196
197 ClassDef(AliTRDrecoParam, 12) // Reconstruction parameters for TRD detector
198
199};
200
201//___________________________________________________
202inline void AliTRDrecoParam::GetSysCovMatrix(Double_t *sys) const
203{
204 if(!sys) return;
205 memcpy(sys, fSysCovMatrix, 5*sizeof(Double_t));
206}
207
208//___________________________________________________
209inline void AliTRDrecoParam::SetSysCovMatrix(Double_t *sys)
210{
211 if(!sys) return;
212 memcpy(fSysCovMatrix, sys, 5*sizeof(Double_t));
213}
214
215//___________________________________________________
216inline void AliTRDrecoParam::SetPIDThreshold(Double_t *pid)
217{
218 if(!pid) return;
219 memcpy(fPIDThreshold, pid, AliTRDCalPID::kNMom*sizeof(Double_t));
220}
221
222//___________________________________________________
223inline void AliTRDrecoParam::SetStreamLevel(ETRDReconstructionTask task, Int_t level){
224 if(task >= kTRDreconstructionTasks) return;
225 fStreamLevel[static_cast<Int_t>(task)] = level;
226}
227
228//___________________________________________________
229inline Int_t AliTRDrecoParam::GetStreamLevel(ETRDReconstructionTask task) const{
230 if(task >= kTRDreconstructionTasks) return 0;
231 return fStreamLevel[static_cast<Int_t>(task)];
232}
233
234//___________________________________________________
235inline void AliTRDrecoParam::GetTCParams(Double_t *par) const
236{
237 if(!par) return;
238 if(IsArgon()) memcpy(par, &fTCParams[4], 4*sizeof(Double_t));
239 else memcpy(par, &fTCParams[0], 4*sizeof(Double_t));
240}
241
242//___________________________________________________
243inline void AliTRDrecoParam::SetTCParams(Double_t *par)
244{
245 if(!par) return;
246 memcpy(fTCParams, par, 8*sizeof(Double_t));
247}
248
249//___________________________________________________
250inline Int_t AliTRDrecoParam::GetPIDLQslices() const
251{
252 if(IsPIDNeuralNetwork()) return -1;
253 return TESTBIT(fFlags, kLQ2D) ? 2 : 1;
254}
255
256#endif