e7668a725dce7e1438c2dcf5d323de6d027b6922
[u/mrichter/AliRoot.git] / PWGHF / hfe / AliHFEcuts.h
1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3 *                                                                        *
4 * Author: The ALICE Off-line Project.                                    *
5 * Contributors are mentioned in the code where appropriate.              *
6 *                                                                        *
7 * Permission to use, copy, modify and distribute this software and its   *
8 * documentation strictly for non-commercial purposes is hereby granted   *
9 * without fee, provided that the above copyright notice appears in all   *
10 * copies and that both the copyright notice and this permission notice   *
11 * appear in the supporting documentation. The authors make no claims     *
12 * about the suitability of this software for any purpose. It is          *
13 * provided "as is" without express or implied warranty.                  *
14 **************************************************************************/
15 //
16 // Cut container class for the ALICE HFE group
17 // Serves also as interface to the correction Framework
18 // Provides a set of standard cuts
19 //
20 #ifndef ALIHFECUTS_H
21 #define ALIHFECUTS_H
22
23 #ifndef ROOT_TNamed
24 #include <TNamed.h>
25 #endif
26
27 #ifndef ALIHFEEXTRACUTS_H
28 #include "AliHFEextraCuts.h"
29 #endif
30
31 class AliCFManager;
32 class AliESDtrack;
33 class AliMCEvent;
34 class AliMCParticle;
35 class AliVEvent;
36
37 class TObjArray;
38 class TList;
39
40 class AliHFEcuts : public TNamed{
41   public:
42     typedef enum{
43       kStepRecNoCut = 0,
44       kStepRecKineITSTPC = 1,
45       kStepRecPrim = 2,
46       kStepHFEcutsITS = 3,
47       kStepHFEcutsTOF = 4,
48       kStepHFEcutsTPC = 5,
49       kStepHFEcutsTRD = 6,
50       kNcutStepsRecTrack = 7
51     } RecoCutStep_t;
52     typedef enum{
53       kStepHFEcutsDca = 0, 
54       kNcutStepsDETrack = 1
55     } DECutStep_t;
56     typedef enum{
57       kStepHFEcutsSecvtx = 0, 
58       kNcutStepsSecvtxTrack = 1
59     } SecvtxCutStep_t;
60     typedef enum{
61       kStepMCGenerated = 0,
62       kStepMCGeneratedZOutNoPileUpCentralityFine = 1,
63       kStepMCGeneratedEventCut = 2,
64       kStepMCInAcceptance = 3,
65       kNcutStepsMCTrack =  4
66     } MCCutStep_t;
67     typedef enum{
68       kEventStepGenerated = 0,
69       kEventStepRecNoCut = 1,
70       kEventStepRecNoPileUp = 2,
71       kEventStepRecCentralityOk = 3,
72       kEventStepZRange = 4,
73       kEventStepReconstructed = 5,
74       kNcutStepsEvent = 6
75     } EventCutStep_t;
76
77     AliHFEcuts();
78     AliHFEcuts(const Char_t *name, const Char_t *title);
79     AliHFEcuts(const AliHFEcuts &c);
80     AliHFEcuts &operator=(const AliHFEcuts &c);
81     void Copy(TObject &o) const;
82     Long64_t Merge(const TCollection *list);
83     ~AliHFEcuts();
84     
85     void Initialize(AliCFManager *cfm);
86     void Initialize();
87
88     Bool_t CheckParticleCuts(UInt_t step, TObject *o);
89     Bool_t CheckEventCuts(const char*namestep, TObject *o);
90     void SetRecEvent(const AliVEvent *ev);
91     void SetMCEvent(const AliMCEvent *ev);
92   
93     TList *GetQAhistograms() const { return fHistQA; }
94     
95     void SetQAOn() {SetBit(kDebugMode, kTRUE); };
96     void UnsetQA() {SetBit(kDebugMode, kFALSE); };
97     Bool_t IsQAOn() const { return TestBit(kDebugMode); };
98     void SetAOD() { SetBit(kAOD, kTRUE); }
99     void SetESD() { SetBit(kAOD, kFALSE); }
100     Bool_t IsAOD() const { return TestBit(kAOD); }
101     Bool_t IsESD() const { return !TestBit(kAOD); }
102
103     // Cut Names
104     static const Char_t *MCCutName(UInt_t step){
105       if(step >= kNcutStepsMCTrack) return fgkUndefined;
106       return fgkMCCutName[step];
107     };
108     static const Char_t *RecoCutName(UInt_t step){
109       if(step >= kNcutStepsRecTrack) return fgkUndefined;
110       return fgkRecoCutName[step];
111     }
112     static const Char_t *DECutName(UInt_t step){
113       if(step >= kNcutStepsDETrack) return fgkUndefined;
114       return fgkDECutName[step];
115     }
116     static const Char_t *SecvtxCutName(UInt_t step){
117       if(step >= kNcutStepsSecvtxTrack) return fgkUndefined;
118       return fgkSecvtxCutName[step];
119     }
120     static const Char_t *EventCutName(UInt_t step){
121       if(step >= kNcutStepsEvent) return fgkUndefined;
122       return fgkEventCutName[step];
123     }
124    
125     // Getters
126     Bool_t IsRequireITSpixel() const { return TESTBIT(fRequirements, kITSPixel); };
127     Bool_t IsRequireITSdrift() const { return TESTBIT(fRequirements, kITSDrift); };
128     Bool_t IsRequireMaxImpactParam() const { return TESTBIT(fRequirements, kMaxImpactParam); };
129     Bool_t IsRequirePrimary() const { return TESTBIT(fRequirements, kPrimary); };
130     Bool_t IsRequireProdVertex() const { return TESTBIT(fRequirements, kProductionVertex); };
131     Bool_t IsRequireSigmaToVertex() const { return TESTBIT(fRequirements, kSigmaToVertex); };
132     Bool_t IsRequireDCAToVertex() const {return TESTBIT(fRequirements, kDCAToVertex); };
133     Bool_t IsRequireKineMCCuts() const {return TESTBIT(fRequirements, kKineMCCuts); };
134     Double_t GetVertexRange() const {return fVertexRangeZ; };
135     Int_t GetMinTrackletsTRD() const { return fMinTrackletsTRD; }
136     
137     // Setters
138     inline void SetCutITSpixel(UChar_t cut);
139     inline void SetCutITSdrift(UChar_t cut);
140     void SetCheckITSLayerStatus(Bool_t checkITSLayerStatus) { fCheckITSLayerStatus = checkITSLayerStatus; }
141     void SetMinNClustersTPC(UChar_t minClustersTPC) { fMinClustersTPC = minClustersTPC; }
142     void SetMinNClustersTPCPID(UChar_t minClustersTPC) { fMinClustersTPCPID = minClustersTPC; }
143     void SetMinNClustersITS(UChar_t minClustersITS) { fMinClustersITS = minClustersITS; }
144     void SetMinNTrackletsTRD(UChar_t minNtrackletsTRD, Bool_t exact = kFALSE) { fMinTrackletsTRD = minNtrackletsTRD; fTRDtrackletsExact = exact; }
145     void SetMaxChi2perClusterITS(Double_t chi2) { fMaxChi2clusterITS = chi2; };
146     void SetMaxChi2perClusterTPC(Double_t chi2) { fMaxChi2clusterTPC = chi2; };
147     inline void SetMaxImpactParam(Double_t radial, Double_t z);
148     inline void SetIPcutParam(Float_t p0, Float_t p1, Float_t p2, Float_t p3, Bool_t isipsigma, Bool_t isabs);
149     void SetMinRatioTPCclusters(Double_t minRatioTPC) { fMinClusterRatioTPC = minRatioTPC; };
150     void SetPtRange(Double_t ptmin, Double_t ptmax){fPtRange[0] = ptmin; fPtRange[1] = ptmax;};
151     void SetTOFsignaldxz(Double_t tofsignaldx, Double_t tofsignaldz){fTOFsignaldx = tofsignaldx; fTOFsignaldz = tofsignaldz;};
152     inline void SetProductionVertex(Double_t xmin, Double_t xmax, Double_t ymin, Double_t ymax);
153     inline void SetSigmaToVertex(Double_t sig);
154     inline void SetSigmaToVertexXY(Double_t sig);
155     inline void SetSigmaToVertexZ(Double_t sig);
156     void SetTPCmodes(AliHFEextraCuts::ETPCclusterDef_t clusterDef, AliHFEextraCuts::ETPCclrDef_t ratioDef) {
157       fTPCclusterDef= clusterDef;
158       fTPCratioDef = ratioDef;
159     }
160     void SetEtaRange(Double_t etaRange){fEtaRange = etaRange;};
161     void SetVertexRange(Double_t zrange){fVertexRangeZ = zrange;};
162     void SetTOFPIDStep(Bool_t tofPidStep) {fTOFPIDStep = tofPidStep;};
163     void SetTOFMISMATCHStep(Bool_t tofMismatchStep) {fTOFMISMATCHStep = tofMismatchStep;};
164     void SetTPCPIDCleanUpStep(Bool_t tpcPIDCleanUpStep) {fTPCPIDCLEANUPStep = tpcPIDCleanUpStep;};
165     void SetUseMixedVertex(Bool_t useMixedVertex) {fUseMixedVertex = useMixedVertex;};    
166     void SetFractionOfSharedTPCClusters(Double_t fractionOfSharedTPCClusters) {fFractionOfSharedTPCClusters = fractionOfSharedTPCClusters;};
167     void SetMaxImpactParameterRpar(Bool_t maxImpactParameterRpar) { fMaxImpactParameterRpar = maxImpactParameterRpar; };
168     
169     inline void CreateStandardCuts();
170     
171     // Requirements
172     void SetAdditionalStatusRequirement(Long_t requirement) {fAdditionalStatusRequirement = requirement;}
173     void SetRequireDCAToVertex() { SETBIT(fRequirements, kDCAToVertex); CLRBIT(fRequirements, kSigmaToVertex); };
174     void SetRequireIsPrimary() { SETBIT(fRequirements, kPrimary); };
175     void SetRequireITSPixel() { SETBIT(fRequirements, kITSPixel); }
176     void SetRequireITSDrift() { SETBIT(fRequirements, kITSDrift); }
177     void UnsetRequireITSPixel() { CLRBIT(fRequirements, kITSPixel); }
178     void SetRequireProdVertex() { SETBIT(fRequirements, kProductionVertex); };
179     void SetRequireSigmaToVertex() { SETBIT(fRequirements, kSigmaToVertex); CLRBIT(fRequirements, kDCAToVertex); };
180     void UnsetVertexRequirement() { CLRBIT(fRequirements, kDCAToVertex); CLRBIT(fRequirements, kSigmaToVertex); }
181     void SetRequireKineMCCuts() { SETBIT(fRequirements, kKineMCCuts); };
182
183     void SetDebugLevel(Int_t level) { fDebugLevel = level; };
184     Int_t GetDebugLevel() const { return fDebugLevel; };
185
186   private:
187     enum{
188       kDebugMode = BIT(14),
189       kAOD = BIT(15)
190     };
191     typedef enum{
192       kPrimary = 0,
193       kProductionVertex = 1,
194       kSigmaToVertex = 2,
195       kDCAToVertex = 3,
196       kITSPixel = 4,
197       kMaxImpactParam = 5,
198       kKineMCCuts = 6,
199       kITSDrift = 7,
200     } Require_t;
201     void SetParticleGenCutList();
202     void SetAcceptanceCutList();
203     void SetRecKineITSTPCCutList();
204     void SetRecPrimaryCutList();
205     void SetHFElectronITSCuts();
206     void SetHFElectronTOFCuts();
207     void SetHFElectronTPCCuts();
208     void SetHFElectronTRDCuts();
209     void SetHFElectronDcaCuts();
210     void SetEventCutList(Int_t istep);
211
212     static const Char_t* fgkMCCutName[kNcutStepsMCTrack];     // Cut step names for MC single Track cuts
213     static const Char_t* fgkRecoCutName[kNcutStepsRecTrack];  // Cut step names for Rec single Track cuts
214     static const Char_t* fgkDECutName[kNcutStepsDETrack];     // Cut step names for impact parameter cuts
215     static const Char_t* fgkSecvtxCutName[kNcutStepsSecvtxTrack];     // Cut step names for secondary vertexing cuts
216     static const Char_t* fgkEventCutName[kNcutStepsEvent];    // Cut step names for Event cuts
217     static const Char_t* fgkUndefined;                        // Name for undefined (overflow)
218   
219     ULong64_t fRequirements;      // Bitmap for requirements
220     AliHFEextraCuts::ETPCclusterDef_t fTPCclusterDef;       // TPC cluster definition
221     AliHFEextraCuts::ETPCclrDef_t fTPCratioDef;             // TPC cluster ratio Definition
222     Double_t fEtaRange;               // Eta range
223     Double_t fDCAtoVtx[2];            // DCA to Vertex
224     Double_t fProdVtx[4];               // Production Vertex
225     Double_t fPtRange[2];               // pt range
226     UChar_t fMinClustersTPC;        // Min.Number of TPC clusters
227     UChar_t fMinClustersTPCPID;   // Min.Number of TPC clusters
228     UChar_t fMinClustersITS;        // Min.Number of TPC clusters
229     UChar_t fMinTrackletsTRD;       // Min. Number of TRD tracklets
230     UChar_t fCutITSPixel;               // Cut on ITS pixel
231     Bool_t  fCheckITSLayerStatus;       // Check ITS layer status
232     UChar_t fCutITSDrift;               // Cut on ITS drift
233     Double_t fMaxChi2clusterITS;        // Max Chi2 per ITS cluster
234     Double_t fMaxChi2clusterTPC;        // Max Chi2 per TPC cluster
235     Double_t fMinClusterRatioTPC;       // Min. Ratio findable / found TPC clusters
236     Double_t fSigmaToVtx[3];        // Sigma To Vertex
237     Double_t fVertexRangeZ;             // Vertex Range reconstructed
238     Bool_t   fTRDtrackletsExact;        // Require exact number of tracklets
239     Bool_t   fTOFPIDStep;               // TOF matching step efficiency
240     Bool_t   fTOFMISMATCHStep;        // TOF mismatch step
241     Bool_t   fTPCPIDCLEANUPStep;      // TPC PIC cleanup step
242     Bool_t   fUseMixedVertex;         // Use primary vertex from track only as before
243     Float_t  fIPCutParams[4];         // Parameters of impact parameter cut parametrization
244     Bool_t   fIsIPSigmacut;           // if IP cut or IP sigma cut 
245     Bool_t   fIsIPAbs;                // if abs IP sigma cut
246     Double_t fFractionOfSharedTPCClusters; // Fraction of shared TPC clusters
247     Bool_t   fMaxImpactParameterRpar;      // Max impact parameter
248     Long_t   fAdditionalStatusRequirement; // Additional status bit requirement 
249     Double_t fTOFsignaldx;                 // TOF signal Dx
250     Double_t fTOFsignaldz;                 // TOF signal Dz
251
252     
253     TList *fHistQA;                         //! QA Histograms
254     TObjArray *fCutList;                //! List of cut objects(Correction Framework Manager)
255
256     Int_t fDebugLevel;            // Debug Level
257     
258   ClassDef(AliHFEcuts, 4)         // Container for HFE cuts
259 };
260
261 //__________________________________________________________________
262 void AliHFEcuts::SetProductionVertex(Double_t xmin, Double_t xmax, Double_t ymin, Double_t ymax){
263   // Set the production vertex constraint
264   SetRequireProdVertex();
265   fProdVtx[0] = xmin;
266   fProdVtx[1] = xmax;
267   fProdVtx[2] = ymin;
268   fProdVtx[3] = ymax;
269 }
270
271 //__________________________________________________________________
272 void AliHFEcuts::SetSigmaToVertex(Double_t sig){
273   SetRequireSigmaToVertex();
274   fSigmaToVtx[0] = sig;
275 }
276
277 //__________________________________________________________________
278 void AliHFEcuts::SetSigmaToVertexXY(Double_t sig){
279   SetRequireSigmaToVertex();
280   fSigmaToVtx[1] = sig;
281 }
282
283 //__________________________________________________________________
284 void AliHFEcuts::SetSigmaToVertexZ(Double_t sig){
285   SetRequireSigmaToVertex();
286   fSigmaToVtx[2] = sig;
287 }
288
289 //__________________________________________________________________
290 void AliHFEcuts::SetMaxImpactParam(Double_t radial, Double_t z){
291   SetRequireDCAToVertex();
292   fDCAtoVtx[0] = radial;
293   fDCAtoVtx[1] = z;
294 }
295
296 //__________________________________________________________________
297 void AliHFEcuts::SetIPcutParam(Float_t p0, Float_t p1, Float_t p2, Float_t p3, Bool_t isipsigma, Bool_t isabs){
298   // Set parameters for impact parameter cut parametrization
299   fIPCutParams[0] = p0;
300   fIPCutParams[1] = p1;
301   fIPCutParams[2] = p2;
302   fIPCutParams[3] = p3;
303   fIsIPSigmacut = isipsigma;
304   fIsIPAbs = isabs;
305 }
306
307 //__________________________________________________________________
308 void AliHFEcuts::SetCutITSpixel(UChar_t cut){
309   SetRequireITSPixel();
310   fCutITSPixel = cut;
311 }
312
313 //__________________________________________________________________
314 void AliHFEcuts::SetCutITSdrift(UChar_t cut){
315   SetRequireITSDrift();
316   fCutITSDrift = cut;
317 }
318
319 //__________________________________________________________________
320 void AliHFEcuts::CreateStandardCuts(){
321   //
322   // Standard Cuts defined by the HFE Group
323   //
324   SetRequireProdVertex();
325   fProdVtx[0] = 0;
326   fProdVtx[1] = 3;
327   fProdVtx[2] = 0;
328   fProdVtx[3] = 3;
329   //SetRequireDCAToVertex();
330   //fDCAtoVtx[0] = 0.5;
331   //fDCAtoVtx[1] = 1.5;
332   fMinClustersTPC = 80;
333   fMinClustersITS = 4;
334   fMinTrackletsTRD = 0;
335   SetRequireITSPixel();
336   fCutITSPixel = AliHFEextraCuts::kFirst;
337   fMaxChi2clusterITS = -1.;
338   fMaxChi2clusterTPC = 4.;
339   fMinClusterRatioTPC = 0.6;
340   fPtRange[0] = 0.1;
341   fPtRange[1] = 20.;
342   SetRequireKineMCCuts();
343 }
344 #endif