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