[u/mrichter/AliRoot.git] / CORRFW / AliCFEventRecCuts.h

#ifndef ALICFEVENTRECCUTS_H
#define ALICFEVENTRECCUTS_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.                  *
 **************************************************************************/
// Cut on the Event at reconstructed level: for the moment 
// just the requirements on the number of charged tracks and on 
// the vertex position and resolution are implemented
// The argument of IsSelected member function (passed object) is cast into 
// an AliESDEvent. In the future may be modified to use AliVEvent interface
// and include more cut variables.
// The class derives from AliCFCutBase
// Author:S.Arcelli Silvia.Arcelli@cern.ch


#include "AliCFCutBase.h"
class TH1F;
class TBits;
//_____________________________________________________________________________
class AliCFEventRecCuts: public AliCFCutBase 
{
 public :
  AliCFEventRecCuts() ;
  AliCFEventRecCuts(Char_t* name, Char_t* title) ;
  AliCFEventRecCuts(const AliCFEventRecCuts& c) ;
  AliCFEventRecCuts& operator=(const AliCFEventRecCuts& c) ;
  ~AliCFEventRecCuts();
  void GetBitMap(TObject *obj, TBits*bitmap);
  Bool_t IsSelected(TObject* obj);
  void Init();
  void AddQAHistograms(TList *list) const;

  void SetNTracksCut(Int_t xMin=-1, Int_t xMax=1000000) {fNTracksMin=xMin; fNTracksMax=xMax;} // cut values setter

  void SetRequireVtxCuts(Bool_t vtx=kFALSE) {fRequireVtxCuts=vtx;} // cut values setter
  void SetVertexXCut(Double_t xMin=-1.e99, Double_t xMax=1.e99) { fVtxXMin=xMin; fVtxXMax=xMax;} // cut values setter
  void SetVertexYCut(Double_t yMin=-1.e99, Double_t yMax=1.e99) { fVtxYMin=yMin; fVtxYMax=yMax;} // cut values setter
  void SetVertexZCut(Double_t zMin=-1.e99, Double_t zMax=1.e99) { fVtxZMin=zMin; fVtxZMax=zMax;} // cut values setter

  void SetVertexXResCut(Double_t xMax=1.e99) {fVtxXResMax=xMax;} // cut values setter
  void SetVertexYResCut(Double_t yMax=1.e99){fVtxYResMax=yMax;} // cut values setter
  void SetVertexZResCut(Double_t zMax=1.e99){fVtxZResMax=zMax;} // cut values setter

  Int_t    GetNTracksMin() const {return fNTracksMin;} // cut values getter
  Int_t    GetNTracksMax() const {return fNTracksMax;} // cut values getter
  Bool_t   GetRequireVtxCuts() const {return fRequireVtxCuts;} // cut value getter
  Double_t GetVertexXMax() const {return fVtxXMax;} // cut values getter
  Double_t GetVertexYMax() const {return fVtxYMax;} // cut values getter
  Double_t GetVertexZMax() const {return fVtxZMax;} // cut values getter
  Double_t GetVertexXMin() const {return fVtxXMin;} // cut values getter
  Double_t GetVertexYMin() const {return fVtxYMin;} // cut values getter
  Double_t GetVertexZMin() const {return fVtxZMin;} // cut values getter
  Double_t GetVertexXResMax() const {return fVtxXResMax;} // cut values getter
  Double_t GetVertexYResMax() const {return fVtxYResMax;} // cut values getter
  Double_t GetVertexZResMax() const {return fVtxZResMax;} // cut values getter

  // QA histograms
  void FillHistogramsBeforeCuts(TObject* obj) {return FillHistograms(obj,kFALSE);}
  void FillHistogramsAfterCuts(TObject* obj)  {return FillHistograms(obj,kTRUE);}
  // QA histogram setter
  // please use indices from the enumeration below
  void SetHistogramBins(Int_t index, Int_t nbins, Double_t *bins);
  void SetHistogramBins(Int_t index, Int_t nbins, Double_t xmin, Double_t xmax);
  enum{kNTracks=0,
	 kVtxPosX,
	 kVtxPosY,
	 kVtxPosZ,
	 kVtxResX,
	 kVtxResY,
	 kVtxResZ,
	 kNCuts=7,
	 kNStepQA=2
	 };
  
 private:
  TBits *SelectionBitMap(TObject* obj);
  void DefineHistograms(); 		// books histograms 
  void Initialise();			// sets everything to 0
  void FillHistograms(TObject* obj, Bool_t b);
  Int_t fNTracksMin; //minimum number of esd tracks
  Int_t fNTracksMax; //maximum number of esd tracks
  Bool_t fRequireVtxCuts ; //The type of trigger to be checked
  Double_t fVtxXMax ; //X vertex position, maximum value
  Double_t fVtxYMax ; //Y vertex position, maximum value 
  Double_t fVtxZMax ; //Z vertex position, maximum value
  Double_t fVtxXMin ; //X vertex position, minimum value
  Double_t fVtxYMin ; //Y vertex position, minimum value
  Double_t fVtxZMin ; //Z vertex position, minimum value
  Double_t fVtxXResMax ;//Maximum value of sigma_vtx in X
  Double_t fVtxYResMax ;//Maximum value of sigma_vtx in X
  Double_t fVtxZResMax ;//Maximum value of sigma_vtx in X


  TBits *fBitMap ; //cut mask


  TH1F* fhQA[kNCuts][kNStepQA];		// QA Histograms
  //QA Histogram parameters
  Int_t fhNBinsNTracks;//size of array of bin limits, N Tracks ESD
  Double_t *fhBinLimNTracks;//[fhNBinsNTracks] bin limits, N Tracks ESD
  Int_t fhNBinsVtxPosX;//size of array of bin limits, Vtx Pos X
  Double_t *fhBinLimVtxPosX;//[fhNBinsVtxPosX] bin limits, Vtx Pos X
  Int_t fhNBinsVtxPosY;//size of array of bin limits, Vtx Pos Y
  Double_t *fhBinLimVtxPosY;//[fhNBinsVtxPosY] bin limits, Vtx Pos Y
  Int_t fhNBinsVtxPosZ;//size of array of bin limits, Vtx Pos Z
  Double_t *fhBinLimVtxPosZ;//[fhNBinsVtxPosZ] bin limits, Vtx Pos Z
  Int_t fhNBinsVtxResX;//size of array of bin limits, Vtx Res X
  Double_t *fhBinLimVtxResX;//[fhNBinsVtxResX] bin limits, Vtx Res X
  Int_t fhNBinsVtxResY;//size of array of bin limits, Vtx Res Y
  Double_t *fhBinLimVtxResY;//[fhNBinsVtxResY] bin limits, Vtx Res Y
  Int_t fhNBinsVtxResZ;//size of array of bin limits, Vtx Res Z
  Double_t *fhBinLimVtxResZ;//[fhNBinsVtxResZ] bin limits, Vtx Res Z

  ClassDef(AliCFEventRecCuts,2);
};

#endif
Commit	Line	Data
563113d0	1	#ifndef ALICFEVENTRECCUTS_H
	2	#define ALICFEVENTRECCUTS_H
	3	/**************************************************************************
	4	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* *
	6	* Author: The ALICE Off-line Project. *
	7	* Contributors are mentioned in the code where appropriate. *
	8	* *
	9	* Permission to use, copy, modify and distribute this software and its *
	10	* documentation strictly for non-commercial purposes is hereby granted *
	11	* without fee, provided that the above copyright notice appears in all *
	12	* copies and that both the copyright notice and this permission notice *
	13	* appear in the supporting documentation. The authors make no claims *
	14	* about the suitability of this software for any purpose. It is *
	15	* provided "as is" without express or implied warranty. *
	16	**************************************************************************/
	17	// Cut on the Event at reconstructed level: for the moment
	18	// just the requirements on the number of charged tracks and on
	19	// the vertex position and resolution are implemented
	20	// The argument of IsSelected member function (passed object) is cast into
	21	// an AliESDEvent. In the future may be modified to use AliVEvent interface
	22	// and include more cut variables.
	23	// The class derives from AliCFCutBase
	24	// Author:S.Arcelli Silvia.Arcelli@cern.ch
	25
	26
	27	#include "AliCFCutBase.h"
e27311df	28	class TH1F;
563113d0	29	class TBits;
	30	//_____________________________________________________________________________
	31	class AliCFEventRecCuts: public AliCFCutBase
	32	{
	33	public :
	34	AliCFEventRecCuts() ;
	35	AliCFEventRecCuts(Char_t* name, Char_t* title) ;
	36	AliCFEventRecCuts(const AliCFEventRecCuts& c) ;
	37	AliCFEventRecCuts& operator=(const AliCFEventRecCuts& c) ;
	38	~AliCFEventRecCuts();
	39	void GetBitMap(TObject obj, TBitsbitmap);
	40	Bool_t IsSelected(TObject* obj);
e27311df	41	void Init();
e27311df	42	void AddQAHistograms(TList *list) const;
563113d0	43
	44	void SetNTracksCut(Int_t xMin=-1, Int_t xMax=1000000) {fNTracksMin=xMin; fNTracksMax=xMax;} // cut values setter
	45
	46	void SetRequireVtxCuts(Bool_t vtx=kFALSE) {fRequireVtxCuts=vtx;} // cut values setter
	47	void SetVertexXCut(Double_t xMin=-1.e99, Double_t xMax=1.e99) { fVtxXMin=xMin; fVtxXMax=xMax;} // cut values setter
	48	void SetVertexYCut(Double_t yMin=-1.e99, Double_t yMax=1.e99) { fVtxYMin=yMin; fVtxYMax=yMax;} // cut values setter
	49	void SetVertexZCut(Double_t zMin=-1.e99, Double_t zMax=1.e99) { fVtxZMin=zMin; fVtxZMax=zMax;} // cut values setter
	50
	51	void SetVertexXResCut(Double_t xMax=1.e99) {fVtxXResMax=xMax;} // cut values setter
	52	void SetVertexYResCut(Double_t yMax=1.e99){fVtxYResMax=yMax;} // cut values setter
	53	void SetVertexZResCut(Double_t zMax=1.e99){fVtxZResMax=zMax;} // cut values setter
	54
	55	Int_t GetNTracksMin() const {return fNTracksMin;} // cut values getter
	56	Int_t GetNTracksMax() const {return fNTracksMax;} // cut values getter
	57	Bool_t GetRequireVtxCuts() const {return fRequireVtxCuts;} // cut value getter
	58	Double_t GetVertexXMax() const {return fVtxXMax;} // cut values getter
	59	Double_t GetVertexYMax() const {return fVtxYMax;} // cut values getter
	60	Double_t GetVertexZMax() const {return fVtxZMax;} // cut values getter
	61	Double_t GetVertexXMin() const {return fVtxXMin;} // cut values getter
	62	Double_t GetVertexYMin() const {return fVtxYMin;} // cut values getter
	63	Double_t GetVertexZMin() const {return fVtxZMin;} // cut values getter
	64	Double_t GetVertexXResMax() const {return fVtxXResMax;} // cut values getter
	65	Double_t GetVertexYResMax() const {return fVtxYResMax;} // cut values getter
	66	Double_t GetVertexZResMax() const {return fVtxZResMax;} // cut values getter
e27311df	67
	68	// QA histograms
	69	void FillHistogramsBeforeCuts(TObject* obj) {return FillHistograms(obj,kFALSE);}
	70	void FillHistogramsAfterCuts(TObject* obj) {return FillHistograms(obj,kTRUE);}
	71	// QA histogram setter
	72	// please use indices from the enumeration below
	73	void SetHistogramBins(Int_t index, Int_t nbins, Double_t *bins);
	74	void SetHistogramBins(Int_t index, Int_t nbins, Double_t xmin, Double_t xmax);
	75	enum{kNTracks=0,
	76	kVtxPosX,
	77	kVtxPosY,
	78	kVtxPosZ,
	79	kVtxResX,
	80	kVtxResY,
	81	kVtxResZ,
	82	kNCuts=7,
	83	kNStepQA=2
	84	};
563113d0	85
	86	private:
	87	TBits SelectionBitMap(TObject obj);
e27311df	88	void DefineHistograms(); // books histograms
	89	void Initialise(); // sets everything to 0
	90	void FillHistograms(TObject* obj, Bool_t b);
563113d0	91	Int_t fNTracksMin; //minimum number of esd tracks
	92	Int_t fNTracksMax; //maximum number of esd tracks
	93	Bool_t fRequireVtxCuts ; //The type of trigger to be checked
	94	Double_t fVtxXMax ; //X vertex position, maximum value
	95	Double_t fVtxYMax ; //Y vertex position, maximum value
	96	Double_t fVtxZMax ; //Z vertex position, maximum value
	97	Double_t fVtxXMin ; //X vertex position, minimum value
	98	Double_t fVtxYMin ; //Y vertex position, minimum value
	99	Double_t fVtxZMin ; //Z vertex position, minimum value
	100	Double_t fVtxXResMax ;//Maximum value of sigma_vtx in X
	101	Double_t fVtxYResMax ;//Maximum value of sigma_vtx in X
	102	Double_t fVtxZResMax ;//Maximum value of sigma_vtx in X
	103
	104
	105	TBits *fBitMap ; //cut mask
	106
e27311df	107
	108	TH1F* fhQA[kNCuts][kNStepQA]; // QA Histograms
	109	//QA Histogram parameters
	110	Int_t fhNBinsNTracks;//size of array of bin limits, N Tracks ESD
	111	Double_t *fhBinLimNTracks;//[fhNBinsNTracks] bin limits, N Tracks ESD
	112	Int_t fhNBinsVtxPosX;//size of array of bin limits, Vtx Pos X
	113	Double_t *fhBinLimVtxPosX;//[fhNBinsVtxPosX] bin limits, Vtx Pos X
	114	Int_t fhNBinsVtxPosY;//size of array of bin limits, Vtx Pos Y
	115	Double_t *fhBinLimVtxPosY;//[fhNBinsVtxPosY] bin limits, Vtx Pos Y
	116	Int_t fhNBinsVtxPosZ;//size of array of bin limits, Vtx Pos Z
	117	Double_t *fhBinLimVtxPosZ;//[fhNBinsVtxPosZ] bin limits, Vtx Pos Z
	118	Int_t fhNBinsVtxResX;//size of array of bin limits, Vtx Res X
	119	Double_t *fhBinLimVtxResX;//[fhNBinsVtxResX] bin limits, Vtx Res X
	120	Int_t fhNBinsVtxResY;//size of array of bin limits, Vtx Res Y
	121	Double_t *fhBinLimVtxResY;//[fhNBinsVtxResY] bin limits, Vtx Res Y
	122	Int_t fhNBinsVtxResZ;//size of array of bin limits, Vtx Res Z
	123	Double_t *fhBinLimVtxResZ;//[fhNBinsVtxResZ] bin limits, Vtx Res Z
	124
	125	ClassDef(AliCFEventRecCuts,2);
563113d0	126	};
	127
	128	#endif