Introduced better way to estimate number of clusters in TRD for offline
[u/mrichter/AliRoot.git] / TRD / 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 #ifndef ALITRDCALPID_H
18 #include "Cal/AliTRDCalPID.h"
19 #endif
20
21 class AliTRDrecoParam : public AliDetectorRecoParam
22 {
23 public:
24   AliTRDrecoParam();
25   AliTRDrecoParam(const AliTRDrecoParam &rec);
26   ~AliTRDrecoParam() { }
27
28   Double_t GetChi2Y() const                 { return fkChi2Y;    }
29   Double_t GetChi2Z() const                 { return fkChi2Z;    }
30   Double_t GetChi2YCut() const              { return fkChi2YCut; }
31   Double_t GetChi2ZCut() const              { return fkChi2ZCut; }
32   Double_t GetPhiCut() const                { return fkPhiCut;   }
33   Float_t  GetNClusters() const;
34   Double_t GetNMeanClusters() const         { return fkNMeanClusters; }
35   Double_t GetNSigmaClusters() const        { return fkNSigmaClusters; }
36   Double_t GetFindableClusters() const      { return fkFindable; }
37   Double_t GetMaxTheta() const              { return fkMaxTheta; }
38   Double_t GetMaxPhi() const                { return fkMaxPhi;   }
39   Double_t GetPlaneQualityThreshold() const { return fkPlaneQualityThreshold; }
40   Double_t GetPIDThreshold(Float_t /*p*/) const { return 0.;}
41   Double_t GetRoad0y() const                { return fkRoad0y;   }
42   Double_t GetRoad0z() const                { return fkRoad0z;   }
43   Double_t GetRoad1y() const                { return fkRoad1y;   }
44   Double_t GetRoad1z() const                { return fkRoad1z;   }
45   Double_t GetRoad2y() const                { return fkRoad2y;   }
46   Double_t GetRoad2z() const                { return fkRoad2z;   }
47   Double_t GetTrackLikelihood() const       { return fkTrackLikelihood;       }
48   inline void GetSysCovMatrix(Double_t *sys) const;  
49   Double_t GetMinMaxCutSigma() const        { return fMinMaxCutSigma;     };
50   Double_t GetMinLeftRightCutSigma() const  { return fMinLeftRightCutSigma;  };
51   Double_t GetClusMaxThresh() const         { return fClusMaxThresh;   };
52   Double_t GetClusSigThresh() const         { return fClusSigThresh;   };
53   Int_t    GetTCnexp() const                { return fTCnexp;          };
54   Int_t     GetNumberOfPresamples()  const {return fNumberOfPresamples;}
55   Int_t    GetNumberOfPostsamples() const {return fNumberOfPostsamples;}
56
57         
58   static   AliTRDrecoParam *GetLowFluxParam();
59   static   AliTRDrecoParam *GetHighFluxParam();
60   static   AliTRDrecoParam *GetCosmicTestParam();
61
62   Bool_t   IsClusterSharing() const         { return TestBit(kClusterSharing);}
63   Bool_t   IsLUT() const                    { return TestBit(kLUT);}
64   Bool_t   IsTailCancelation() const        { return TestBit(kTC);}
65   Bool_t   IsVertexConstrained() const      { return TestBit(kVertexConstrained); }
66   Bool_t   HasImproveTracklets() const       { return TestBit(kImproveTracklet); }
67
68   void     SetMaxTheta(Double_t maxTheta) {fkMaxTheta = maxTheta;}
69   void     SetMaxPhi(Double_t maxPhi) {fkMaxPhi = maxPhi;}
70   void     SetFindableClusters(Double_t r) {fkFindable = r;}
71   void     SetChi2Y(Double_t chi2) {fkChi2Y = chi2;}
72   void     SetChi2Z(Double_t chi2) {fkChi2Z = chi2;}
73   void     SetChi2YCut(Double_t chi2YCut) {fkChi2YCut = chi2YCut;}
74   void     SetChi2ZCut(Double_t chi2ZCut) {fkChi2ZCut = chi2ZCut;}
75   void     SetPhiCut(Double_t phiCut) {fkPhiCut = phiCut;}
76   void     SetNMeanClusters(Double_t meanNclusters) {fkNMeanClusters = meanNclusters;}
77   void     SetNSigmaClusters(Double_t sigmaNclusters) {fkNSigmaClusters = sigmaNclusters;} 
78   void     SetClusterSharing(Bool_t share = kTRUE)            { SetBit(kClusterSharing, share);}
79   void     SetImproveTracklets(Bool_t improve = kTRUE)         { SetBit(kImproveTracklet, improve);}
80   void     SetVertexConstrained(Bool_t vc = kTRUE)            { SetBit(kVertexConstrained, vc); }
81   void     SetLUT(Bool_t lut = kTRUE)                         { SetBit(kLUT, lut);};
82   void     SetMinMaxCutSigma(Float_t minMaxCutSigma)          { fMinMaxCutSigma   = minMaxCutSigma; }
83   void     SetMinLeftRightCutSigma(Float_t minLeftRightCutSigma) { fMinLeftRightCutSigma   = minLeftRightCutSigma; };
84   void     SetClusMaxThresh(Float_t thresh)                   { fClusMaxThresh   = thresh; };
85   void     SetClusSigThresh(Float_t thresh)                   { fClusSigThresh   = thresh; };
86   void     SetTailCancelation(Bool_t tc = kTRUE)              { SetBit(kTC, tc);  };
87   inline void SetPIDThreshold(Double_t *pid);
88   void     SetNexponential(Int_t nexp)                        { fTCnexp          = nexp;   };
89   inline void SetSysCovMatrix(Double_t *sys);
90   void     SetNumberOfPresamples(Int_t n)                     { fNumberOfPresamples = n;}
91   void     SetNumberOfPostsamples(Int_t n)                    { fNumberOfPostsamples = n;}
92
93 private:
94   enum{
95     kTC                = BIT(14) // tail cancelation
96    ,kLUT               = BIT(15) // look up table for cluster position determination 
97    ,kClusterSharing    = BIT(16) // Toggle cluster sharing
98    ,kVertexConstrained = BIT(17) // Perform vertex constrained fit
99    ,kImproveTracklet   = BIT(18) // Improve tracklet in the SA TRD track finder 
100   };
101   // Physics reference values for TRD
102   Double_t  fkdNchdy;                // dNch/dy
103   Double_t  fkMaxTheta;              // Maximum theta
104   Double_t  fkMaxPhi;                // Maximum phi - momentum cut
105
106   Double_t  fkRoad0y;                // Road for middle cluster
107   Double_t  fkRoad0z;                // Road for middle cluster
108
109   Double_t  fkRoad1y;                // Road in y for seeded cluster
110   Double_t  fkRoad1z;                // Road in z for seeded cluster
111
112   Double_t  fkRoad2y;                // Road in y for extrapolated cluster
113   Double_t  fkRoad2z;                // Road in z for extrapolated cluster
114   
115   Double_t  fkPlaneQualityThreshold; // Quality threshold
116   Double_t  fkFindable;              // minimum ratio of clusters per tracklet supposed to be attached.
117   Double_t  fkChi2Z;                 // Max chi2 on the z direction for seeding clusters fit
118   Double_t  fkChi2Y;                 // Max chi2 on the y direction for seeding clusters Rieman fit
119   Double_t  fkChi2YCut;              // Cut on the Chi2 in y-direction in the likelihood filter
120   Double_t  fkChi2ZCut;              // Cut on the Chi2 in z-direction in the likelihood filter
121   Double_t  fkPhiCut;                // Cut on the deviation of the phi angles between tracklet and track fit (lik. filter)
122   Double_t  fkNMeanClusters;         // Mean number of clusters per tracklet
123   Double_t  fkNSigmaClusters;        // Sigma of the number of clusters per tracklet
124   Double_t  fkNClusterNoise;         // ratio of noisy clusters to the true one
125   Double_t  fkNMeanTracklets;        // Mean number of tracklets per track
126   Double_t  fkTrackLikelihood;       // Track likelihood for tracklets Rieman fit
127   
128   Double_t  fSysCovMatrix[5];        // Systematic uncertainty from calibration and alignment for each tracklet
129   Double_t  fPIDThreshold[AliTRDCalPID::kNMom];
130
131   // Clusterization parameter
132   Double_t  fMinMaxCutSigma;         // Threshold sigma noise pad middle
133   Double_t  fMinLeftRightCutSigma;   // Threshold sigma noise sum pad
134   Double_t  fClusMaxThresh;          // Threshold value for cluster maximum
135   Double_t  fClusSigThresh;          // Threshold value for cluster signal
136   Int_t     fTCnexp;                 // Number of exponentials, digital filter
137   
138   // ADC parameter
139   Int_t     fNumberOfPresamples;     // number of presamples 
140   Int_t     fNumberOfPostsamples;     // number of postsamples 
141
142   ClassDef(AliTRDrecoParam, 5)       // Reconstruction parameters for TRD detector
143
144 };
145
146 //___________________________________________________
147 inline void AliTRDrecoParam::GetSysCovMatrix(Double_t *sys) const
148 {
149   if(!sys) return;
150   memcpy(sys, fSysCovMatrix, 5*sizeof(Double_t));
151 }
152
153 //___________________________________________________
154 inline void AliTRDrecoParam::SetSysCovMatrix(Double_t *sys)
155 {
156   if(!sys) return;
157   memcpy(fSysCovMatrix, sys, 5*sizeof(Double_t));
158 }
159
160 //___________________________________________________
161 inline void AliTRDrecoParam::SetPIDThreshold(Double_t *pid)
162 {
163   if(!pid) return;
164   memcpy(fPIDThreshold, pid, AliTRDCalPID::kNMom*sizeof(Double_t));
165 }
166
167
168 #endif