1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
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 **************************************************************************/
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
27 #ifndef ALIHFEEXTRACUTS_H
28 #include "AliHFEextraCuts.h"
40 class AliHFEcuts : public TNamed{
44 kStepRecKineITSTPC = 1,
50 kNcutStepsRecTrack = 7
57 kStepHFEcutsSecvtx = 0,
58 kNcutStepsSecvtxTrack = 1
62 kStepMCGeneratedZOutNoPileUpCentralityFine = 1,
63 kStepMCGeneratedEventCut = 2,
64 kStepMCInAcceptance = 3,
68 kEventStepGenerated = 0,
69 kEventStepRecNoCut = 1,
70 kEventStepRecNoPileUp = 2,
71 kEventStepRecCentralityOk = 3,
73 kEventStepReconstructed = 5,
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);
85 void Initialize(AliCFManager *cfm);
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 AliVEvent *ev);
93 TList *GetQAhistograms() const { return fHistQA; }
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); }
104 static const Char_t *MCCutName(UInt_t step){
105 if(step >= kNcutStepsMCTrack) return fgkUndefined;
106 return fgkMCCutName[step];
108 static const Char_t *RecoCutName(UInt_t step){
109 if(step >= kNcutStepsRecTrack) return fgkUndefined;
110 return fgkRecoCutName[step];
112 static const Char_t *DECutName(UInt_t step){
113 if(step >= kNcutStepsDETrack) return fgkUndefined;
114 return fgkDECutName[step];
116 static const Char_t *SecvtxCutName(UInt_t step){
117 if(step >= kNcutStepsSecvtxTrack) return fgkUndefined;
118 return fgkSecvtxCutName[step];
120 static const Char_t *EventCutName(UInt_t step){
121 if(step >= kNcutStepsEvent) return fgkUndefined;
122 return fgkEventCutName[step];
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; }
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 SetMaxChi2perTrackletTRD(Float_t maxchi2trackletTRD) { fMaxChi2TRD = maxchi2trackletTRD; }
146 void SetMaxChi2perClusterITS(Double_t chi2) { fMaxChi2clusterITS = chi2; };
147 void SetMaxChi2perClusterTPC(Double_t chi2) { fMaxChi2clusterTPC = chi2; };
148 inline void SetMaxImpactParam(Double_t radial, Double_t z);
149 inline void SetIPcutParam(Float_t p0, Float_t p1, Float_t p2, Float_t p3, Bool_t isipsigma, Bool_t isabs);
150 void SetMinRatioTPCclusters(Double_t minRatioTPC) { fMinClusterRatioTPC = minRatioTPC; };
151 void SetPtRange(Double_t ptmin, Double_t ptmax){fPtRange[0] = ptmin; fPtRange[1] = ptmax;};
152 void SetTOFsignaldxz(Double_t tofsignaldx, Double_t tofsignaldz){fTOFsignaldx = tofsignaldx; fTOFsignaldz = tofsignaldz;};
153 inline void SetProductionVertex(Double_t xmin, Double_t xmax, Double_t ymin, Double_t ymax);
154 inline void SetSigmaToVertex(Double_t sig);
155 inline void SetSigmaToVertexXY(Double_t sig);
156 inline void SetSigmaToVertexZ(Double_t sig);
157 void SetTPCmodes(UChar_t clusterDef, UChar_t ratioDef) {
158 fTPCclusterDef= clusterDef;
159 fTPCratioDef = ratioDef;
161 void SetEtaRange(Double_t etaRange){fEtaRange = etaRange;};
162 void SetVertexRange(Double_t zrange){fVertexRangeZ = zrange;};
163 void SetTOFPIDStep(Bool_t tofPidStep) {fTOFPIDStep = tofPidStep;};
164 void SetTOFMISMATCHStep(Bool_t tofMismatchStep) {fTOFMISMATCHStep = tofMismatchStep;};
165 void SetTPCPIDCleanUpStep(Bool_t tpcPIDCleanUpStep) {fTPCPIDCLEANUPStep = tpcPIDCleanUpStep;};
166 void SetITSpatternCut() { fITSpatternCut = kTRUE; }
167 void SetUseMixedVertex(Bool_t useMixedVertex) {fUseMixedVertex = useMixedVertex;};
168 void SetFractionOfSharedTPCClusters(Double_t fractionOfSharedTPCClusters) {fFractionOfSharedTPCClusters = fractionOfSharedTPCClusters;};
169 void SetMaxImpactParameterRpar(Bool_t maxImpactParameterRpar) { fMaxImpactParameterRpar = maxImpactParameterRpar; };
171 inline void CreateStandardCuts();
174 void SetAdditionalStatusRequirement(Long_t requirement) {fAdditionalStatusRequirement = requirement;}
175 void SetRequireDCAToVertex() { SETBIT(fRequirements, kDCAToVertex); CLRBIT(fRequirements, kSigmaToVertex); };
176 void SetRequireIsPrimary() { SETBIT(fRequirements, kPrimary); };
177 void SetRequireITSPixel() { SETBIT(fRequirements, kITSPixel); }
178 void SetRequireITSDrift() { SETBIT(fRequirements, kITSDrift); }
179 void UnsetRequireITSPixel() { CLRBIT(fRequirements, kITSPixel); }
180 void SetRequireProdVertex() { SETBIT(fRequirements, kProductionVertex); };
181 void SetRequireSigmaToVertex() { SETBIT(fRequirements, kSigmaToVertex); CLRBIT(fRequirements, kDCAToVertex); };
182 void UnsetVertexRequirement() { CLRBIT(fRequirements, kDCAToVertex); CLRBIT(fRequirements, kSigmaToVertex); }
183 void SetRequireKineMCCuts() { SETBIT(fRequirements, kKineMCCuts); };
185 void SetDebugLevel(Int_t level) { fDebugLevel = level; };
186 Int_t GetDebugLevel() const { return fDebugLevel; };
190 kDebugMode = BIT(14),
195 kProductionVertex = 1,
203 void SetParticleGenCutList();
204 void SetAcceptanceCutList();
205 void SetRecKineITSTPCCutList();
206 void SetRecPrimaryCutList();
207 void SetHFElectronITSCuts();
208 void SetHFElectronTOFCuts();
209 void SetHFElectronTPCCuts();
210 void SetHFElectronTRDCuts();
211 void SetHFElectronDcaCuts();
212 void SetEventCutList(Int_t istep);
214 static const Char_t* fgkMCCutName[kNcutStepsMCTrack]; // Cut step names for MC single Track cuts
215 static const Char_t* fgkRecoCutName[kNcutStepsRecTrack]; // Cut step names for Rec single Track cuts
216 static const Char_t* fgkDECutName[kNcutStepsDETrack]; // Cut step names for impact parameter cuts
217 static const Char_t* fgkSecvtxCutName[kNcutStepsSecvtxTrack]; // Cut step names for secondary vertexing cuts
218 static const Char_t* fgkEventCutName[kNcutStepsEvent]; // Cut step names for Event cuts
219 static const Char_t* fgkUndefined; // Name for undefined (overflow)
221 ULong64_t fRequirements; // Bitmap for requirements
222 UChar_t fTPCclusterDef; // TPC cluster definition
223 UChar_t fTPCratioDef; // TPC cluster ratio Definition
224 Double_t fEtaRange; // Eta range
225 Double_t fDCAtoVtx[2]; // DCA to Vertex
226 Double_t fProdVtx[4]; // Production Vertex
227 Double_t fPtRange[2]; // pt range
228 UChar_t fMinClustersTPC; // Min.Number of TPC clusters
229 UChar_t fMinClustersTPCPID; // Min.Number of TPC clusters
230 UChar_t fMinClustersITS; // Min.Number of TPC clusters
231 UChar_t fMinTrackletsTRD; // Min. Number of TRD tracklets
232 Float_t fMaxChi2TRD; // Max. Chi2 per TRD tracklet
233 UChar_t fCutITSPixel; // Cut on ITS pixel
234 Bool_t fCheckITSLayerStatus; // Check ITS layer status
235 UChar_t fCutITSDrift; // Cut on ITS drift
236 Double_t fMaxChi2clusterITS; // Max Chi2 per ITS cluster
237 Double_t fMaxChi2clusterTPC; // Max Chi2 per TPC cluster
238 Double_t fMinClusterRatioTPC; // Min. Ratio findable / found TPC clusters
239 Double_t fSigmaToVtx[3]; // Sigma To Vertex
240 Double_t fVertexRangeZ; // Vertex Range reconstructed
241 Bool_t fTRDtrackletsExact; // Require exact number of tracklets
242 Bool_t fTOFPIDStep; // TOF matching step efficiency
243 Bool_t fTOFMISMATCHStep; // TOF mismatch step
244 Bool_t fTPCPIDCLEANUPStep; // TPC PIC cleanup step
245 Bool_t fITSpatternCut; // Cut on ITS pattern
246 Bool_t fUseMixedVertex; // Use primary vertex from track only as before
247 Float_t fIPCutParams[4]; // Parameters of impact parameter cut parametrization
248 Bool_t fIsIPSigmacut; // if IP cut or IP sigma cut
249 Bool_t fIsIPAbs; // if abs IP sigma cut
250 Double_t fFractionOfSharedTPCClusters; // Fraction of shared TPC clusters
251 Bool_t fMaxImpactParameterRpar; // Max impact parameter
252 Long_t fAdditionalStatusRequirement; // Additional status bit requirement
253 Double_t fTOFsignaldx; // TOF signal Dx
254 Double_t fTOFsignaldz; // TOF signal Dz
257 TList *fHistQA; //! QA Histograms
258 TObjArray *fCutList; //! List of cut objects(Correction Framework Manager)
260 Int_t fDebugLevel; // Debug Level
262 ClassDef(AliHFEcuts, 4) // Container for HFE cuts
265 //__________________________________________________________________
266 void AliHFEcuts::SetProductionVertex(Double_t xmin, Double_t xmax, Double_t ymin, Double_t ymax){
267 // Set the production vertex constraint
268 SetRequireProdVertex();
275 //__________________________________________________________________
276 void AliHFEcuts::SetSigmaToVertex(Double_t sig){
277 SetRequireSigmaToVertex();
278 fSigmaToVtx[0] = sig;
281 //__________________________________________________________________
282 void AliHFEcuts::SetSigmaToVertexXY(Double_t sig){
283 SetRequireSigmaToVertex();
284 fSigmaToVtx[1] = sig;
287 //__________________________________________________________________
288 void AliHFEcuts::SetSigmaToVertexZ(Double_t sig){
289 SetRequireSigmaToVertex();
290 fSigmaToVtx[2] = sig;
293 //__________________________________________________________________
294 void AliHFEcuts::SetMaxImpactParam(Double_t radial, Double_t z){
295 SetRequireDCAToVertex();
296 fDCAtoVtx[0] = radial;
300 //__________________________________________________________________
301 void AliHFEcuts::SetIPcutParam(Float_t p0, Float_t p1, Float_t p2, Float_t p3, Bool_t isipsigma, Bool_t isabs){
302 // Set parameters for impact parameter cut parametrization
303 fIPCutParams[0] = p0;
304 fIPCutParams[1] = p1;
305 fIPCutParams[2] = p2;
306 fIPCutParams[3] = p3;
307 fIsIPSigmacut = isipsigma;
311 //__________________________________________________________________
312 void AliHFEcuts::SetCutITSpixel(UChar_t cut){
313 SetRequireITSPixel();
317 //__________________________________________________________________
318 void AliHFEcuts::SetCutITSdrift(UChar_t cut){
319 SetRequireITSDrift();
323 //__________________________________________________________________
324 void AliHFEcuts::CreateStandardCuts(){
326 // Standard Cuts defined by the HFE Group
328 SetRequireProdVertex();
333 //SetRequireDCAToVertex();
334 //fDCAtoVtx[0] = 0.5;
335 //fDCAtoVtx[1] = 1.5;
336 fMinClustersTPC = 80;
338 fMinTrackletsTRD = 0;
339 SetRequireITSPixel();
340 fCutITSPixel = AliHFEextraCuts::kFirst;
341 fMaxChi2clusterITS = -1.;
342 fMaxChi2clusterTPC = 4.;
343 fMinClusterRatioTPC = 0.6;
346 SetRequireKineMCCuts();