[u/mrichter/AliRoot.git] / PWG3 / hfe / AliHFEcuts.h

/**************************************************************************
* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
*                                                                        *
* Author: The ALICE Off-line Project.                                    *
* Contributors are mentioned in the code where appropriate.              *
*                                                                        *
* Permission to use, copy, modify and distribute this software and its   *
* documentation strictly for non-commercial purposes is hereby granted   *
* without fee, provided that the above copyright notice appears in all   *
* copies and that both the copyright notice and this permission notice   *
* appear in the supporting documentation. The authors make no claims     *
* about the suitability of this software for any purpose. It is          *
* provided "as is" without express or implied warranty.                  *
**************************************************************************/
#ifndef __ALIHFECUTS_H__
#define __ALIHFECUTS_H__

#ifndef ROOT_TObject
#include <TObject.h>
#endif

#ifndef __ALIHFELECTRONEXTRACUTS_H__
#include "AliHFEextraCuts.h"
#endif

class AliCFManager;
class AliESDtrack;
class AliMCParticle;

class TObjArray;
class TList;

class AliHFEcuts : public TObject{
  enum{
    kDebugMode = BIT(14)
      };
  typedef enum{
    kPrimary = 0,
      kProductionVertex = 1,
      kSigmaToVertex = 2,
      kDCAToVertex = 3,
      kITSPixel = 4,
      kMaxImpactParam = 5
      } Require_t;
 public:

  typedef enum{
    kStepMCGenerated = 0,
      kStepMCInAcceptance = 1,
      kStepRecKine = 2,
      kStepRecPrim = 3,
      kStepHFEcuts = 4
      } CutStep_t;

  static const Int_t kNcutSteps;

  AliHFEcuts();
  AliHFEcuts(const AliHFEcuts &c);
  AliHFEcuts &operator=(const AliHFEcuts &c);
  ~AliHFEcuts();
    
  void Initialize(AliCFManager *cfm);
  void Initialize();

  Bool_t CheckParticleCuts(CutStep_t step, TObject *o);

  TList *GetQAhistograms() const { return fHistQA; }
    
  void SetDebugMode();
  void UnsetDebugMode() { SetBit(kDebugMode, kFALSE); }
  Bool_t IsInDebugMode() const { return TestBit(kDebugMode); };
    
  // Getters
  Bool_t IsRequireITSpixel() const { return TESTBIT(fRequirements, kITSPixel); };
  Bool_t IsRequireMaxImpactParam() const { return TESTBIT(fRequirements, kMaxImpactParam); };
  Bool_t IsRequirePrimary() const { return TESTBIT(fRequirements, kPrimary); };
  Bool_t IsRequireProdVertex() const { return TESTBIT(fRequirements, kProductionVertex); };
  Bool_t IsRequireSigmaToVertex() const { return TESTBIT(fRequirements, kSigmaToVertex); };
  Bool_t IsRequireDCAToVertex() const {return TESTBIT(fRequirements, kDCAToVertex); };
    
  // Setters
  inline void SetCutITSpixel(UChar_t cut);
  void SetMinNClustersTPC(UChar_t minClustersTPC) { fMinClustersTPC = minClustersTPC; }
  void SetMinNTrackletsTRD(UChar_t minNtrackletsTRD) { fMinTrackletsTRD = minNtrackletsTRD; }
  void SetMaxChi2perClusterTPC(Double_t chi2) { fMaxChi2clusterTPC = chi2; };
  inline void SetMaxImpactParam(Double_t radial, Double_t z);
  void SetMinRatioTPCclusters(Double_t minRatioTPC) { fMinClusterRatioTPC = minRatioTPC; };
  void SetPtRange(Double_t ptmin, Double_t ptmax){fPtRange[0] = ptmin; fPtRange[1] = ptmax;};
  inline void SetProductionVertex(Double_t xmin, Double_t xmax, Double_t ymin, Double_t ymax);
  inline void SetSigmaToVertex(Double_t sig);
    
  inline void CreateStandardCuts();
    
  // Requirements
  void SetRequireDCAToVertex() { SETBIT(fRequirements, kDCAToVertex); };
  void SetRequireIsPrimary() { SETBIT(fRequirements, kPrimary); };
  void SetRequireITSPixel() { SETBIT(fRequirements, kITSPixel); }
  void SetRequireMaxImpactParam() { SETBIT(fRequirements, kMaxImpactParam); };
  void SetRequireProdVetrex() { SETBIT(fRequirements, kProductionVertex); };
  void SetRequireSigmaToVertex() { SETBIT(fRequirements, kSigmaToVertex); };
    
 private:
  void SetParticleGenCutList();
  void SetAcceptanceCutList();
  void SetRecKineCutList();
  void SetRecPrimaryCutList();
  void SetHFElectronCuts();
    
  ULong64_t fRequirements;	// Bitmap for requirements
  Double_t fDCAtoVtx[2];	// DCA to Vertex
  Double_t fProdVtx[4];	// Production Vertex
  Double_t fPtRange[2];	// pt range
  UChar_t fMinClustersTPC;	// Min.Number of TPC clusters
  UChar_t fMinTrackletsTRD;	// Min. Number of TRD tracklets
  UChar_t fCutITSPixel;	// Cut on ITS pixel
  Double_t fMaxChi2clusterTPC;	// Max Chi2 per TPC cluster
  Double_t fMinClusterRatioTPC;	// Min. Ratio findable / found TPC clusters
  Double_t fSigmaToVtx;	// Sigma To Vertex
  Double_t fMaxImpactParamR;	// Max. Impact Parameter in Radial Direction
  Double_t fMaxImpactParamZ;	// Max. Impact Parameter in Z Direction
    
  TList *fHistQA;		//! QA Histograms
  TObjArray *fCutList;	//! List of cut objects(Correction Framework Manager)
    
  ClassDef(AliHFEcuts, 1)   // Container for HFE cuts
    };

    //__________________________________________________________________
    void AliHFEcuts::SetProductionVertex(Double_t xmin, Double_t xmax, Double_t ymin, Double_t ymax){
      // Set the production vertex constraint
      SetRequireProdVetrex();
      fProdVtx[0] = xmin;
      fProdVtx[1] = xmax;
      fProdVtx[2] = ymin;
      fProdVtx[3] = ymax;
    }

//__________________________________________________________________
void AliHFEcuts::SetSigmaToVertex(Double_t sig){
  SetRequireSigmaToVertex();
  fSigmaToVtx = sig;
}

//__________________________________________________________________
void AliHFEcuts::SetMaxImpactParam(Double_t radial, Double_t z){
  SetRequireMaxImpactParam();
  fMaxImpactParamR = radial;
  fMaxImpactParamZ = z;
}

//__________________________________________________________________
void AliHFEcuts::SetCutITSpixel(UChar_t cut){
  SetRequireITSPixel();
  fCutITSPixel = cut;
}

//__________________________________________________________________
void AliHFEcuts::CreateStandardCuts(){
  //
  // Standard Cuts defined by the HFE Group
  //
  SetRequireProdVetrex();
  fProdVtx[0] = -1;
  fProdVtx[1] = 1;
  fProdVtx[2] = -1;
  fProdVtx[3] = 1;
  SetRequireDCAToVertex();
  fDCAtoVtx[0] = 4.;
  fDCAtoVtx[1] = 10.;
  fMinClustersTPC = 50;
  fMinTrackletsTRD = 6;
  fCutITSPixel = AliHFEextraCuts::kAny;
  fMaxChi2clusterTPC = 3.5;
  fMinClusterRatioTPC = 0.6;
  fPtRange[0] = 0.1;
  fPtRange[1] = 20.;
  fSigmaToVtx = 4.;
  SetRequireMaxImpactParam();
  fMaxImpactParamR = 3.;
  fMaxImpactParamZ = 12.;
}
#endif
Commit	Line	Data
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	**************************************************************************/
	15	#ifndef __ALIHFECUTS_H__
	16	#define __ALIHFECUTS_H__
	17
	18	#ifndef ROOT_TObject
	19	#include <TObject.h>
	20	#endif
	21
	22	#ifndef __ALIHFELECTRONEXTRACUTS_H__
	23	#include "AliHFEextraCuts.h"
	24	#endif
	25
	26	class AliCFManager;
	27	class AliESDtrack;
	28	class AliMCParticle;
	29
	30	class TObjArray;
	31	class TList;
	32
	33	class AliHFEcuts : public TObject{
	34	enum{
	35	kDebugMode = BIT(14)
	36	};
	37	typedef enum{
	38	kPrimary = 0,
	39	kProductionVertex = 1,
	40	kSigmaToVertex = 2,
	41	kDCAToVertex = 3,
	42	kITSPixel = 4,
	43	kMaxImpactParam = 5
	44	} Require_t;
	45	public:
	46
	47	typedef enum{
	48	kStepMCGenerated = 0,
	49	kStepMCInAcceptance = 1,
	50	kStepRecKine = 2,
	51	kStepRecPrim = 3,
	52	kStepHFEcuts = 4
	53	} CutStep_t;
	54
	55	static const Int_t kNcutSteps;
	56
	57	AliHFEcuts();
	58	AliHFEcuts(const AliHFEcuts &c);
	59	AliHFEcuts &operator=(const AliHFEcuts &c);
	60	~AliHFEcuts();
	61
	62	void Initialize(AliCFManager *cfm);
	63	void Initialize();
	64
65	Bool_t CheckParticleCuts(CutStep_t step, TObject *o);
66
67	TList *GetQAhistograms() const { return fHistQA; }
68
69	void SetDebugMode();
70	void UnsetDebugMode() { SetBit(kDebugMode, kFALSE); }
71	Bool_t IsInDebugMode() const { return TestBit(kDebugMode); };
72
73	// Getters
74	Bool_t IsRequireITSpixel() const { return TESTBIT(fRequirements, kITSPixel); };
75	Bool_t IsRequireMaxImpactParam() const { return TESTBIT(fRequirements, kMaxImpactParam); };
76	Bool_t IsRequirePrimary() const { return TESTBIT(fRequirements, kPrimary); };
77	Bool_t IsRequireProdVertex() const { return TESTBIT(fRequirements, kProductionVertex); };
78	Bool_t IsRequireSigmaToVertex() const { return TESTBIT(fRequirements, kSigmaToVertex); };
79	Bool_t IsRequireDCAToVertex() const {return TESTBIT(fRequirements, kDCAToVertex); };
80
81	// Setters
82	inline void SetCutITSpixel(UChar_t cut);
83	void SetMinNClustersTPC(UChar_t minClustersTPC) { fMinClustersTPC = minClustersTPC; }
84	void SetMinNTrackletsTRD(UChar_t minNtrackletsTRD) { fMinTrackletsTRD = minNtrackletsTRD; }
85	void SetMaxChi2perClusterTPC(Double_t chi2) { fMaxChi2clusterTPC = chi2; };
86	inline void SetMaxImpactParam(Double_t radial, Double_t z);
87	void SetMinRatioTPCclusters(Double_t minRatioTPC) { fMinClusterRatioTPC = minRatioTPC; };
88	void SetPtRange(Double_t ptmin, Double_t ptmax){fPtRange[0] = ptmin; fPtRange[1] = ptmax;};
89	inline void SetProductionVertex(Double_t xmin, Double_t xmax, Double_t ymin, Double_t ymax);
90	inline void SetSigmaToVertex(Double_t sig);
91
92	inline void CreateStandardCuts();
93
94	// Requirements
95	void SetRequireDCAToVertex() { SETBIT(fRequirements, kDCAToVertex); };
96	void SetRequireIsPrimary() { SETBIT(fRequirements, kPrimary); };
97	void SetRequireITSPixel() { SETBIT(fRequirements, kITSPixel); }
98	void SetRequireMaxImpactParam() { SETBIT(fRequirements, kMaxImpactParam); };
99	void SetRequireProdVetrex() { SETBIT(fRequirements, kProductionVertex); };
100	void SetRequireSigmaToVertex() { SETBIT(fRequirements, kSigmaToVertex); };
101
102	private:
103	void SetParticleGenCutList();
104	void SetAcceptanceCutList();
105	void SetRecKineCutList();
106	void SetRecPrimaryCutList();
107	void SetHFElectronCuts();
108
109	ULong64_t fRequirements; // Bitmap for requirements
110	Double_t fDCAtoVtx[2]; // DCA to Vertex
111	Double_t fProdVtx[4]; // Production Vertex
112	Double_t fPtRange[2]; // pt range
113	UChar_t fMinClustersTPC; // Min.Number of TPC clusters
114	UChar_t fMinTrackletsTRD; // Min. Number of TRD tracklets
115	UChar_t fCutITSPixel; // Cut on ITS pixel
116	Double_t fMaxChi2clusterTPC; // Max Chi2 per TPC cluster
117	Double_t fMinClusterRatioTPC; // Min. Ratio findable / found TPC clusters
118	Double_t fSigmaToVtx; // Sigma To Vertex
119	Double_t fMaxImpactParamR; // Max. Impact Parameter in Radial Direction
120	Double_t fMaxImpactParamZ; // Max. Impact Parameter in Z Direction
121
122	TList *fHistQA; //! QA Histograms
123	TObjArray *fCutList; //! List of cut objects(Correction Framework Manager)
124
125	ClassDef(AliHFEcuts, 1) // Container for HFE cuts
126	};
127
128	//__________________________________________________________________
129	void AliHFEcuts::SetProductionVertex(Double_t xmin, Double_t xmax, Double_t ymin, Double_t ymax){
130	// Set the production vertex constraint
131	SetRequireProdVetrex();
132	fProdVtx[0] = xmin;
133	fProdVtx[1] = xmax;
134	fProdVtx[2] = ymin;
135	fProdVtx[3] = ymax;
136	}
137
138	//__________________________________________________________________
139	void AliHFEcuts::SetSigmaToVertex(Double_t sig){
140	SetRequireSigmaToVertex();
141	fSigmaToVtx = sig;
142	}
143
144	//__________________________________________________________________
145	void AliHFEcuts::SetMaxImpactParam(Double_t radial, Double_t z){
146	SetRequireMaxImpactParam();
147	fMaxImpactParamR = radial;
148	fMaxImpactParamZ = z;
149	}
150
151	//__________________________________________________________________
152	void AliHFEcuts::SetCutITSpixel(UChar_t cut){
153	SetRequireITSPixel();
154	fCutITSPixel = cut;
155	}
156
157	//__________________________________________________________________
158	void AliHFEcuts::CreateStandardCuts(){
159	//
160	// Standard Cuts defined by the HFE Group
161	//
162	SetRequireProdVetrex();
163	fProdVtx[0] = -1;
164	fProdVtx[1] = 1;
165	fProdVtx[2] = -1;
166	fProdVtx[3] = 1;
167	SetRequireDCAToVertex();
168	fDCAtoVtx[0] = 4.;
169	fDCAtoVtx[1] = 10.;
170	fMinClustersTPC = 50;
171	fMinTrackletsTRD = 6;
172	fCutITSPixel = AliHFEextraCuts::kAny;
173	fMaxChi2clusterTPC = 3.5;
174	fMinClusterRatioTPC = 0.6;
175	fPtRange[0] = 0.1;
176	fPtRange[1] = 20.;
177	fSigmaToVtx = 4.;
178	SetRequireMaxImpactParam();
179	fMaxImpactParamR = 3.;
180	fMaxImpactParamZ = 12.;
181	}
182	#endif