Place the config and root file at the right place
[u/mrichter/AliRoot.git] / PWGHF / hfe / AliHFEcuts.h
CommitLineData
809a4336 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**************************************************************************/
50685501 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//
c2690925 20#ifndef ALIHFECUTS_H
21#define ALIHFECUTS_H
22
e3fc062d 23#ifndef ROOT_TNamed
24#include <TNamed.h>
809a4336 25#endif
26
75d81601 27#ifndef ALIHFEEXTRACUTS_H
809a4336 28#include "AliHFEextraCuts.h"
29#endif
30
31class AliCFManager;
32class AliESDtrack;
e17c1f86 33class AliMCEvent;
809a4336 34class AliMCParticle;
e17c1f86 35class AliVEvent;
809a4336 36
37class TObjArray;
38class TList;
39
e3fc062d 40class AliHFEcuts : public TNamed{
75d81601 41 public:
42 typedef enum{
3a72645a 43 kStepRecNoCut = 0,
44 kStepRecKineITSTPC = 1,
45 kStepRecPrim = 2,
46 kStepHFEcutsITS = 3,
9250ffbf 47 kStepHFEcutsTOF = 4,
8c1c76e9 48 kStepHFEcutsTPC = 5,
49 kStepHFEcutsTRD = 6,
50 kNcutStepsRecTrack = 7
3a72645a 51 } RecoCutStep_t;
52 typedef enum{
53 kStepHFEcutsDca = 0,
54 kNcutStepsDETrack = 1
55 } DECutStep_t;
56 typedef enum{
c2690925 57 kStepHFEcutsSecvtx = 0,
58 kNcutStepsSecvtxTrack = 1
59 } SecvtxCutStep_t;
60 typedef enum{
75d81601 61 kStepMCGenerated = 0,
c2690925 62 kStepMCGeneratedZOutNoPileUpCentralityFine = 1,
3a72645a 63 kStepMCGeneratedEventCut = 2,
64 kStepMCInAcceptance = 3,
65 kNcutStepsMCTrack = 4
66 } MCCutStep_t;
0792aa82 67 typedef enum{
68 kEventStepGenerated = 0,
e3fc062d 69 kEventStepRecNoCut = 1,
3a72645a 70 kEventStepRecNoPileUp = 2,
c2690925 71 kEventStepRecCentralityOk = 3,
72 kEventStepZRange = 4,
73 kEventStepReconstructed = 5,
74 kNcutStepsEvent = 6
0792aa82 75 } EventCutStep_t;
809a4336 76
75d81601 77 AliHFEcuts();
e3fc062d 78 AliHFEcuts(const Char_t *name, const Char_t *title);
75d81601 79 AliHFEcuts(const AliHFEcuts &c);
80 AliHFEcuts &operator=(const AliHFEcuts &c);
e3fc062d 81 void Copy(TObject &o) const;
8c1c76e9 82 Long64_t Merge(const TCollection *list);
75d81601 83 ~AliHFEcuts();
809a4336 84
75d81601 85 void Initialize(AliCFManager *cfm);
86 void Initialize();
809a4336 87
3a72645a 88 Bool_t CheckParticleCuts(UInt_t step, TObject *o);
e156c3bb 89 Bool_t CheckEventCuts(const char*namestep, TObject *o);
e17c1f86 90 void SetRecEvent(const AliVEvent *ev);
76d0b522 91 void SetMCEvent(const AliVEvent *ev);
75d81601 92
93 TList *GetQAhistograms() const { return fHistQA; }
809a4336 94
e3fc062d 95 void SetQAOn() {SetBit(kDebugMode, kTRUE); };
96 void UnsetQA() {SetBit(kDebugMode, kFALSE); };
97 Bool_t IsQAOn() const { return TestBit(kDebugMode); };
9bcfd1ab 98 void SetAOD() { SetBit(kAOD, kTRUE); }
99 void SetESD() { SetBit(kAOD, kFALSE); }
9bcfd1ab 100 Bool_t IsAOD() const { return TestBit(kAOD); }
101 Bool_t IsESD() const { return !TestBit(kAOD); }
3a72645a 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 }
c2690925 116 static const Char_t *SecvtxCutName(UInt_t step){
117 if(step >= kNcutStepsSecvtxTrack) return fgkUndefined;
118 return fgkSecvtxCutName[step];
119 }
3a72645a 120 static const Char_t *EventCutName(UInt_t step){
121 if(step >= kNcutStepsEvent) return fgkUndefined;
122 return fgkEventCutName[step];
123 }
124
75d81601 125 // Getters
126 Bool_t IsRequireITSpixel() const { return TESTBIT(fRequirements, kITSPixel); };
5cd679b7 127 Bool_t IsRequireITSdrift() const { return TESTBIT(fRequirements, kITSDrift); };
75d81601 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); };
faee3b18 133 Bool_t IsRequireKineMCCuts() const {return TESTBIT(fRequirements, kKineMCCuts); };
3a72645a 134 Double_t GetVertexRange() const {return fVertexRangeZ; };
f9f097c0 135 Int_t GetMinTrackletsTRD() const { return fMinTrackletsTRD; }
63bdf450 136 Bool_t GetUseMixedVertex() const { return fUseMixedVertex;};
809a4336 137
75d81601 138 // Setters
139 inline void SetCutITSpixel(UChar_t cut);
5cd679b7 140 inline void SetCutITSdrift(UChar_t cut);
0792aa82 141 void SetCheckITSLayerStatus(Bool_t checkITSLayerStatus) { fCheckITSLayerStatus = checkITSLayerStatus; }
75d81601 142 void SetMinNClustersTPC(UChar_t minClustersTPC) { fMinClustersTPC = minClustersTPC; }
e17c1f86 143 void SetMinNClustersTPCPID(UChar_t minClustersTPC) { fMinClustersTPCPID = minClustersTPC; }
3a72645a 144 void SetMinNClustersITS(UChar_t minClustersITS) { fMinClustersITS = minClustersITS; }
e17c1f86 145 void SetMinNTrackletsTRD(UChar_t minNtrackletsTRD, Bool_t exact = kFALSE) { fMinTrackletsTRD = minNtrackletsTRD; fTRDtrackletsExact = exact; }
cedf0381 146 void SetMaxChi2perTrackletTRD(Float_t maxchi2trackletTRD) { fMaxChi2TRD = maxchi2trackletTRD; }
8c1c76e9 147 void SetMaxChi2perClusterITS(Double_t chi2) { fMaxChi2clusterITS = chi2; };
75d81601 148 void SetMaxChi2perClusterTPC(Double_t chi2) { fMaxChi2clusterTPC = chi2; };
149 inline void SetMaxImpactParam(Double_t radial, Double_t z);
63bdf450 150 inline void SetIPcutParam(Float_t p0, Float_t p1, Float_t p2, Float_t p3, Bool_t isIPcharge, Bool_t isipsigma, Bool_t isopp);
75d81601 151 void SetMinRatioTPCclusters(Double_t minRatioTPC) { fMinClusterRatioTPC = minRatioTPC; };
152 void SetPtRange(Double_t ptmin, Double_t ptmax){fPtRange[0] = ptmin; fPtRange[1] = ptmax;};
1c051dd4 153 void SetTOFsignaldxz(Double_t tofsignaldx, Double_t tofsignaldz){fTOFsignaldx = tofsignaldx; fTOFsignaldz = tofsignaldz;};
96167a04 154 void SetAODFilterBit(Int_t bit) { fAODFilterBit = bit; };
75d81601 155 inline void SetProductionVertex(Double_t xmin, Double_t xmax, Double_t ymin, Double_t ymax);
a5fb8511 156 void SetProductionVertexZ(Double_t zmin, Double_t zmax) { fProdVtxZ[0] = zmin; fProdVtxZ[1] = zmax; }
75d81601 157 inline void SetSigmaToVertex(Double_t sig);
8c1c76e9 158 inline void SetSigmaToVertexXY(Double_t sig);
159 inline void SetSigmaToVertexZ(Double_t sig);
76d0b522 160 void SetTPCmodes(UChar_t clusterDef, UChar_t ratioDef) {
e3ae862b 161 fTPCclusterDef= clusterDef;
162 fTPCratioDef = ratioDef;
163 }
cddc311d 164 void SetEtaRange(Double_t etaRange){fEtaRange[0] = -etaRange; fEtaRange[1] = etaRange;};
165 void SetEtaRange(Double_t etamin, Double_t etamax){fEtaRange[0] = etamin; fEtaRange[1] = etamax;};
a5fb8511 166 void SetPhiRange(Double_t phimin, Double_t phimax){fPhiRange[0] = phimin; fPhiRange[1] = phimax;};
e3ae862b 167 void SetVertexRange(Double_t zrange){fVertexRangeZ = zrange;};
c2690925 168 void SetTOFPIDStep(Bool_t tofPidStep) {fTOFPIDStep = tofPidStep;};
169 void SetTOFMISMATCHStep(Bool_t tofMismatchStep) {fTOFMISMATCHStep = tofMismatchStep;};
8c1c76e9 170 void SetTPCPIDCleanUpStep(Bool_t tpcPIDCleanUpStep) {fTPCPIDCLEANUPStep = tpcPIDCleanUpStep;};
cedf0381 171 void SetITSpatternCut() { fITSpatternCut = kTRUE; }
63bdf450 172 inline void SetUseMixedVertex(Bool_t useMixedVertex);
173 inline void SetUseSPDVertex(Bool_t useSPDVertex);
174 void SetUseCorrelationVertex() { fUseCorrelationVertex = kTRUE;};
d87a9d29 175 void SetSPDVtxResolutionCut() {fSPDVtxResolution = kTRUE;}
4437a0d2 176 void SetpApileupCut() { fPApileupCut = kTRUE; }
e17c1f86 177 void SetFractionOfSharedTPCClusters(Double_t fractionOfSharedTPCClusters) {fFractionOfSharedTPCClusters = fractionOfSharedTPCClusters;};
9250ffbf 178 void SetMaxImpactParameterRpar(Bool_t maxImpactParameterRpar) { fMaxImpactParameterRpar = maxImpactParameterRpar; };
809a4336 179
75d81601 180 inline void CreateStandardCuts();
809a4336 181
75d81601 182 // Requirements
11ff28c5 183 void SetAdditionalStatusRequirement(Long_t requirement) {fAdditionalStatusRequirement = requirement;}
67fe7bd0 184 void SetRequireDCAToVertex() { SETBIT(fRequirements, kDCAToVertex); CLRBIT(fRequirements, kSigmaToVertex); };
75d81601 185 void SetRequireIsPrimary() { SETBIT(fRequirements, kPrimary); };
186 void SetRequireITSPixel() { SETBIT(fRequirements, kITSPixel); }
5cd679b7 187 void SetRequireITSDrift() { SETBIT(fRequirements, kITSDrift); }
11ff28c5 188 void UnsetRequireITSPixel() { CLRBIT(fRequirements, kITSPixel); }
75d81601 189 void SetRequireProdVertex() { SETBIT(fRequirements, kProductionVertex); };
67fe7bd0 190 void SetRequireSigmaToVertex() { SETBIT(fRequirements, kSigmaToVertex); CLRBIT(fRequirements, kDCAToVertex); };
3a72645a 191 void UnsetVertexRequirement() { CLRBIT(fRequirements, kDCAToVertex); CLRBIT(fRequirements, kSigmaToVertex); }
faee3b18 192 void SetRequireKineMCCuts() { SETBIT(fRequirements, kKineMCCuts); };
bbeea05b 193 void SetRejectKinkDaughters() { fRejectKinkDaughters = kTRUE; }
194 void SetAcceptKinkDaughters() { fRejectKinkDaughters = kFALSE; }
195 void SetRejectKinkMothers() { fRejectKinkMothers = kTRUE; }
196 void SetAcceptKinkMothers() { fRejectKinkMothers = kFALSE; }
dbe3abbe 197
75d81601 198 void SetDebugLevel(Int_t level) { fDebugLevel = level; };
199 Int_t GetDebugLevel() const { return fDebugLevel; };
200
201 private:
50685501 202 enum{
9bcfd1ab 203 kDebugMode = BIT(14),
204 kAOD = BIT(15)
50685501 205 };
206 typedef enum{
207 kPrimary = 0,
208 kProductionVertex = 1,
209 kSigmaToVertex = 2,
210 kDCAToVertex = 3,
211 kITSPixel = 4,
faee3b18 212 kMaxImpactParam = 5,
5cd679b7 213 kKineMCCuts = 6,
cedf0381 214 kITSDrift = 7
50685501 215 } Require_t;
75d81601 216 void SetParticleGenCutList();
217 void SetAcceptanceCutList();
722347d8 218 void SetRecKineITSTPCCutList();
75d81601 219 void SetRecPrimaryCutList();
220 void SetHFElectronITSCuts();
9250ffbf 221 void SetHFElectronTOFCuts();
8c1c76e9 222 void SetHFElectronTPCCuts();
75d81601 223 void SetHFElectronTRDCuts();
3a72645a 224 void SetHFElectronDcaCuts();
0792aa82 225 void SetEventCutList(Int_t istep);
3a72645a 226
227 static const Char_t* fgkMCCutName[kNcutStepsMCTrack]; // Cut step names for MC single Track cuts
228 static const Char_t* fgkRecoCutName[kNcutStepsRecTrack]; // Cut step names for Rec single Track cuts
229 static const Char_t* fgkDECutName[kNcutStepsDETrack]; // Cut step names for impact parameter cuts
c2690925 230 static const Char_t* fgkSecvtxCutName[kNcutStepsSecvtxTrack]; // Cut step names for secondary vertexing cuts
3a72645a 231 static const Char_t* fgkEventCutName[kNcutStepsEvent]; // Cut step names for Event cuts
232 static const Char_t* fgkUndefined; // Name for undefined (overflow)
dbe3abbe 233
a5fb8511 234 ULong64_t fRequirements; // Bitmap for requirements
235 UChar_t fTPCclusterDef; // TPC cluster definition
236 UChar_t fTPCratioDef; // TPC cluster ratio Definition
237 Double_t fEtaRange[2]; // Eta range
238 Double_t fPhiRange[2]; // Phi range
239 Double_t fDCAtoVtx[2]; // DCA to Vertex
240 Double_t fProdVtx[4]; // Production Vertex
241 Double_t fProdVtxZ[2]; // Production Vertex in Z direction
242 Double_t fPtRange[2]; // pt range
243 UChar_t fMinClustersTPC; // Min.Number of TPC clusters
244 UChar_t fMinClustersTPCPID; // Min.Number of TPC clusters
245 UChar_t fMinClustersITS; // Min.Number of TPC clusters
246 UChar_t fMinTrackletsTRD; // Min. Number of TRD tracklets
247 Float_t fMaxChi2TRD; // Max. Chi2 per TRD tracklet
248 UChar_t fCutITSPixel; // Cut on ITS pixel
249 Bool_t fCheckITSLayerStatus; // Check ITS layer status
250 UChar_t fCutITSDrift; // Cut on ITS drift
251 Double_t fMaxChi2clusterITS; // Max Chi2 per ITS cluster
252 Double_t fMaxChi2clusterTPC; // Max Chi2 per TPC cluster
253 Double_t fMinClusterRatioTPC; // Min. Ratio findable / found TPC clusters
254 Double_t fSigmaToVtx[3]; // Sigma To Vertex
255 Double_t fVertexRangeZ; // Vertex Range reconstructed
256 Bool_t fTRDtrackletsExact; // Require exact number of tracklets
257 Bool_t fTOFPIDStep; // TOF matching step efficiency
258 Bool_t fTOFMISMATCHStep; // TOF mismatch step
259 Bool_t fTPCPIDCLEANUPStep; // TPC PIC cleanup step
260 Bool_t fITSpatternCut; // Cut on ITS pattern
261 Bool_t fUseMixedVertex; // Use primary vertex from track if there otherwise SPD vertex
262 Bool_t fUseSPDVertex; // Use primary SPD vertex
263 Bool_t fUseCorrelationVertex; // Use the correlation of the vertex in z
264 Bool_t fSPDVtxResolution; // Check resolution of the SPD vertex
265 Bool_t fPApileupCut; // Apply pA pileup cut
266 Float_t fIPCutParams[4]; // Parameters of impact parameter cut parametrization
267 Bool_t fIsIPSigmacut; // if IP cut or IP sigma cut
268 Bool_t fIsIPcharge; // if cut on IP * charge (cut using only positive side of distribution, to eliminate conversions)
269 Bool_t fIsIPOpp; // if IP*charge cut on side of the photon peak
270 Double_t fFractionOfSharedTPCClusters; // Fraction of shared TPC clusters
271 Bool_t fMaxImpactParameterRpar; // Max impact parameter
272 Long_t fAdditionalStatusRequirement; // Additional status bit requirement
273 Double_t fTOFsignaldx; // TOF signal Dx
274 Double_t fTOFsignaldz; // TOF signal Dz
275 Int_t fAODFilterBit; // AOD Filter Bit Number
276 Bool_t fRejectKinkDaughters; // Reject Kink Daughters
277 Bool_t fRejectKinkMothers; // Reject Kink Daughters
809a4336 278
a5fb8511 279 TList *fHistQA; //! QA Histograms
280 TObjArray *fCutList; //! List of cut objects(Correction Framework Manager)
dbe3abbe 281
a5fb8511 282 Int_t fDebugLevel; // Debug Level
809a4336 283
a5fb8511 284 ClassDef(AliHFEcuts, 6) // Container for HFE cuts
75d81601 285};
809a4336 286
75d81601 287//__________________________________________________________________
288void AliHFEcuts::SetProductionVertex(Double_t xmin, Double_t xmax, Double_t ymin, Double_t ymax){
289 // Set the production vertex constraint
290 SetRequireProdVertex();
291 fProdVtx[0] = xmin;
292 fProdVtx[1] = xmax;
293 fProdVtx[2] = ymin;
294 fProdVtx[3] = ymax;
295}
809a4336 296
297//__________________________________________________________________
298void AliHFEcuts::SetSigmaToVertex(Double_t sig){
299 SetRequireSigmaToVertex();
8c1c76e9 300 fSigmaToVtx[0] = sig;
301}
302
303//__________________________________________________________________
304void AliHFEcuts::SetSigmaToVertexXY(Double_t sig){
305 SetRequireSigmaToVertex();
306 fSigmaToVtx[1] = sig;
307}
308
309//__________________________________________________________________
310void AliHFEcuts::SetSigmaToVertexZ(Double_t sig){
311 SetRequireSigmaToVertex();
312 fSigmaToVtx[2] = sig;
809a4336 313}
314
315//__________________________________________________________________
316void AliHFEcuts::SetMaxImpactParam(Double_t radial, Double_t z){
ad75027f 317 SetRequireDCAToVertex();
318 fDCAtoVtx[0] = radial;
319 fDCAtoVtx[1] = z;
809a4336 320}
321
322//__________________________________________________________________
63bdf450 323void AliHFEcuts::SetIPcutParam(Float_t p0, Float_t p1, Float_t p2, Float_t p3, Bool_t isipsigma, Bool_t isIPcharge, Bool_t isopp){
c2690925 324 // Set parameters for impact parameter cut parametrization
325 fIPCutParams[0] = p0;
326 fIPCutParams[1] = p1;
327 fIPCutParams[2] = p2;
328 fIPCutParams[3] = p3;
329 fIsIPSigmacut = isipsigma;
63bdf450 330 fIsIPcharge = isIPcharge;
331 fIsIPOpp = isopp;
c2690925 332}
333
334//__________________________________________________________________
809a4336 335void AliHFEcuts::SetCutITSpixel(UChar_t cut){
336 SetRequireITSPixel();
337 fCutITSPixel = cut;
338}
339
340//__________________________________________________________________
5cd679b7 341void AliHFEcuts::SetCutITSdrift(UChar_t cut){
342 SetRequireITSDrift();
343 fCutITSDrift = cut;
344}
63bdf450 345//__________________________________________________________________
346void AliHFEcuts::SetUseMixedVertex(Bool_t useMixedVertex){
347 //
348 // Choice of a vertex
349 //
350 fUseMixedVertex = useMixedVertex;
351 if(useMixedVertex) fUseSPDVertex = kFALSE;
352}
353//__________________________________________________________________
354void AliHFEcuts::SetUseSPDVertex(Bool_t useSPDVertex){
355 //
356 // Choice of a vertex
357 //
358 fUseSPDVertex = useSPDVertex;
359 if(useSPDVertex) fUseMixedVertex = kFALSE;
360}
5cd679b7 361
362//__________________________________________________________________
809a4336 363void AliHFEcuts::CreateStandardCuts(){
364 //
365 // Standard Cuts defined by the HFE Group
366 //
ad75027f 367 SetRequireProdVertex();
faee3b18 368 fProdVtx[0] = 0;
722347d8 369 fProdVtx[1] = 3;
faee3b18 370 fProdVtx[2] = 0;
722347d8 371 fProdVtx[3] = 3;
a5fb8511 372 fProdVtxZ[0] = -15;
373 fProdVtxZ[1] = 15;
faee3b18 374 //SetRequireDCAToVertex();
375 //fDCAtoVtx[0] = 0.5;
376 //fDCAtoVtx[1] = 1.5;
377 fMinClustersTPC = 80;
3a72645a 378 fMinClustersITS = 4;
722347d8 379 fMinTrackletsTRD = 0;
dbe3abbe 380 SetRequireITSPixel();
722347d8 381 fCutITSPixel = AliHFEextraCuts::kFirst;
8c1c76e9 382 fMaxChi2clusterITS = -1.;
3a72645a 383 fMaxChi2clusterTPC = 4.;
809a4336 384 fMinClusterRatioTPC = 0.6;
385 fPtRange[0] = 0.1;
420bd2ea 386 fPtRange[1] = 100.;
bbeea05b 387 fRejectKinkDaughters = kTRUE;
388 fRejectKinkMothers = kTRUE;
faee3b18 389 SetRequireKineMCCuts();
809a4336 390}
391#endif