[u/mrichter/AliRoot.git] / CORRFW / AliCFTrackIsPrimaryCuts.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.                  *
 **************************************************************************/

// The class AliCFTrackIsPrimaryCut is designed to select reconstructed tracks
// with a small impact parameter and tracks which are (not) daughters of kink
// decays and to provide corresponding QA histograms.
// This class inherits from the Analysis' Framework abstract base class
// AliAnalysisCuts and is a part of the Correction Framework.
// This class acts on single, reconstructed tracks, it is applicable on
// ESD and AOD data.
// It mainly consists of a IsSelected function that returns a boolean.
// This function checks whether the considered track passes a set of cuts:
// - distance to main vertex in units of sigma (resolution)
// - require that the dca calculation doesn't fail
// - accept or not accept daughter tracks of kink decays
//
// The cut values for these cuts are set with the corresponding set functions.
// All cut classes provided by the correction framework are supposed to be
// added in the Analysis Framwork's class AliAnalysisFilter and applied by
// the filter via a loop.
//
// author: I. Kraus (Ingrid.Kraus@cern.ch)
// idea taken form
// AliESDtrackCuts writte by Jan Fiete Grosse-Oetringhaus and
// AliRsnDaughterCut class written by A. Pulvirenti.

#ifndef ALICFTRACKISPRIMARYCUTS_H
#define ALICFTRACKISPRIMARYCUTS_H

#include "AliCFCutBase.h"

#include <TH2.h>
// class TH2 ;
class TBits;
class AliESDtrack ;

class AliCFTrackIsPrimaryCuts : public AliCFCutBase
{
 public :
  AliCFTrackIsPrimaryCuts() ;
  AliCFTrackIsPrimaryCuts(Char_t* name, Char_t* title) ;
  AliCFTrackIsPrimaryCuts(const AliCFTrackIsPrimaryCuts& c) ;
  AliCFTrackIsPrimaryCuts& operator=(const AliCFTrackIsPrimaryCuts& c) ;
  ~AliCFTrackIsPrimaryCuts();
  void Copy(TObject &c) const;

  Bool_t IsSelected(TObject* obj);
  Bool_t IsSelected(TList* /*list*/) {return kTRUE;}
  void GetSigmaToVertex(AliESDtrack* esdTrack);

  // cut value setter
  void SetMaxNSigmaToVertex(Double_t sigma=3)	{fNSigmaToVertexMax = sigma;}
  void SetRequireSigmaToVertex(Bool_t b=kTRUE)	{fRequireSigmaToVertex=b;}
  void SetAcceptKinkDaughters(Bool_t b=kTRUE)	{fAcceptKinkDaughters=b;}

  // QA histograms
  void DrawHistograms();
  void SaveHistograms(const Char_t* dir = 0);
  void AddQAHistograms(TList *qaList);
  // 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);

  // indeces/counters for single selections
  enum { 
    kCutNSigmaToVertex=0,	// tracks's distance to main vertex in units of sigma
    kCutRequireSigmaToVertex,	// calculation is successful
    kCutAcceptKinkDaughters,	// do (not) accept secondaries
    kDcaXY,			// controll histogram: dca in xy plane
    kDcaZ,			// controll histogram: dca along z axis
    kDcaXYnorm,			// controll histogram: normalised dca in xy plane
    kDcaZnorm,			// controll histogram: normalised dca along z axis
    kNCuts=3,			// number of single selections
    kNStepQA=2,			// number of QA steps (before/after the cuts)
    kNHist=7			// number of QA histograms
  };

 private:
  void SelectionBitMap(TObject* obj);
  void DefineHistograms(); 		// books histograms and TList
  void Initialise();			// sets everything to 0
  void FillHistograms(TObject* obj, Bool_t b);
					// Fills histograms before and after cuts
  Double_t fNSigmaToVertex;		// track distance to main vertex in units of sigma
  Double_t fNSigmaToVertexMax;		// cut value: max distance to main vertex in units of sigma
  Bool_t  fRequireSigmaToVertex;	// require calculable distance to main vertex

  TH2F* fhDcaXYvsDcaZ[2];		// Histogram: dca xy vs. z
  TH2F* fhDcaXYvsDcaZnorm[2];		// Histogram: (dca xy / sigma xy) vs. (dca z / simga z)
  Bool_t  fAcceptKinkDaughters;		// accepting kink daughters

  TH1F* fhCutStatistics;		// Histogram: statistics of what cuts the tracks did not survive
  TH2F* fhCutCorrelation;		// Histogram: 2d statistics plot

  TH1F* fhQA[kNHist][kNStepQA];		// QA Histograms
  TBits *fBitmap ; 			// stores single selection decisions

  // QA histogram setters
  Int_t fhNBinsNSigma;			// number of bins+1: dca in units of sigma
  Int_t fhNBinsRequireSigma;		// number of bins+1: require successful calcuation
  Int_t fhNBinsAcceptKink;		// number of bins+1: acceptkink daughters
  Int_t fhNBinsDcaXY;			// number of bins+1: dca in transverse plane
  Int_t fhNBinsDcaZ;			// number of bins+1: dca along beam axis
  Int_t fhNBinsDcaXYnorm;		// number of bins+1: normalised dca in transverse plane
  Int_t fhNBinsDcaZnorm;		// number of bins+1: normalised dca along beam axis

  Double_t *fhBinLimNSigma; //[fhNBinsNSigma] bin limits: dca in units of sigma
  Double_t *fhBinLimRequireSigma;//[fhNBinsRequireSigma] bin limits: require successful calcuation
  Double_t *fhBinLimAcceptKink;//[fhNBinsAcceptKink] bin limits: acceptkink daughters
  Double_t *fhBinLimDcaXY;//[fhNBinsDcaXY] bin limits: dca in transverse plane
  Double_t *fhBinLimDcaZ; //[fhNBinsDcaZ] bin limits: dca along beam axis
  Double_t *fhBinLimDcaXYnorm; //[fhNBinsDcaXYnorm] bin limits: normalised dca in transverse plane
  Double_t *fhBinLimDcaZnorm;//[fhNBinsDcaZnorm] bin limits: normalised dca along beam axis

  ClassDef(AliCFTrackIsPrimaryCuts,2);
};

#endif
Commit	Line	Data
563113d0	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	// The class AliCFTrackIsPrimaryCut is designed to select reconstructed tracks
	17	// with a small impact parameter and tracks which are (not) daughters of kink
	18	// decays and to provide corresponding QA histograms.
	19	// This class inherits from the Analysis' Framework abstract base class
	20	// AliAnalysisCuts and is a part of the Correction Framework.
	21	// This class acts on single, reconstructed tracks, it is applicable on
	22	// ESD and AOD data.
	23	// It mainly consists of a IsSelected function that returns a boolean.
	24	// This function checks whether the considered track passes a set of cuts:
	25	// - distance to main vertex in units of sigma (resolution)
	26	// - require that the dca calculation doesn't fail
	27	// - accept or not accept daughter tracks of kink decays
	28	//
	29	// The cut values for these cuts are set with the corresponding set functions.
	30	// All cut classes provided by the correction framework are supposed to be
	31	// added in the Analysis Framwork's class AliAnalysisFilter and applied by
	32	// the filter via a loop.
	33	//
	34	// author: I. Kraus (Ingrid.Kraus@cern.ch)
	35	// idea taken form
	36	// AliESDtrackCuts writte by Jan Fiete Grosse-Oetringhaus and
	37	// AliRsnDaughterCut class written by A. Pulvirenti.
	38
	39	#ifndef ALICFTRACKISPRIMARYCUTS_H
	40	#define ALICFTRACKISPRIMARYCUTS_H
	41
	42	#include "AliCFCutBase.h"
	43
045e6d86	44	#include <TH2.h>
045e6d86	45	// class TH2 ;
563113d0	46	class TBits;
563113d0	47	class AliESDtrack ;
	48
	49	class AliCFTrackIsPrimaryCuts : public AliCFCutBase
	50	{
	51	public :
	52	AliCFTrackIsPrimaryCuts() ;
	53	AliCFTrackIsPrimaryCuts(Char_t* name, Char_t* title) ;
	54	AliCFTrackIsPrimaryCuts(const AliCFTrackIsPrimaryCuts& c) ;
	55	AliCFTrackIsPrimaryCuts& operator=(const AliCFTrackIsPrimaryCuts& c) ;
	56	~AliCFTrackIsPrimaryCuts();
	57	void Copy(TObject &c) const;
	58
563113d0	59	Bool_t IsSelected(TObject* obj);
264ebaac	60	Bool_t IsSelected(TList* /list/) {return kTRUE;}
107a3100	61	void GetSigmaToVertex(AliESDtrack* esdTrack);
563113d0	62
	63	// cut value setter
	64	void SetMaxNSigmaToVertex(Double_t sigma=3) {fNSigmaToVertexMax = sigma;}
	65	void SetRequireSigmaToVertex(Bool_t b=kTRUE) {fRequireSigmaToVertex=b;}
	66	void SetAcceptKinkDaughters(Bool_t b=kTRUE) {fAcceptKinkDaughters=b;}
	67
	68	// QA histograms
563113d0	69	void DrawHistograms();
563113d0	70	void SaveHistograms(const Char_t* dir = 0);
107a3100	71	void AddQAHistograms(TList *qaList);
563113d0	72	// QA histogram setter
	73	// please use indices from the enumeration below
	74	void SetHistogramBins(Int_t index, Int_t nbins, Double_t *bins);
	75	void SetHistogramBins(Int_t index, Int_t nbins, Double_t xmin, Double_t xmax);
	76
	77	// indeces/counters for single selections
	78	enum {
	79	kCutNSigmaToVertex=0, // tracks's distance to main vertex in units of sigma
	80	kCutRequireSigmaToVertex, // calculation is successful
	81	kCutAcceptKinkDaughters, // do (not) accept secondaries
	82	kDcaXY, // controll histogram: dca in xy plane
	83	kDcaZ, // controll histogram: dca along z axis
	84	kDcaXYnorm, // controll histogram: normalised dca in xy plane
	85	kDcaZnorm, // controll histogram: normalised dca along z axis
	86	kNCuts=3, // number of single selections
	87	kNStepQA=2, // number of QA steps (before/after the cuts)
	88	kNHist=7 // number of QA histograms
	89	};
	90
	91	private:
107a3100	92	void SelectionBitMap(TObject* obj);
563113d0	93	void DefineHistograms(); // books histograms and TList
	94	void Initialise(); // sets everything to 0
	95	void FillHistograms(TObject* obj, Bool_t b);
	96	// Fills histograms before and after cuts
107a3100	97	Double_t fNSigmaToVertex; // track distance to main vertex in units of sigma
107a3100	98	Double_t fNSigmaToVertexMax; // cut value: max distance to main vertex in units of sigma
563113d0	99	Bool_t fRequireSigmaToVertex; // require calculable distance to main vertex
	100
	101	TH2F* fhDcaXYvsDcaZ[2]; // Histogram: dca xy vs. z
	102	TH2F* fhDcaXYvsDcaZnorm[2]; // Histogram: (dca xy / sigma xy) vs. (dca z / simga z)
	103	Bool_t fAcceptKinkDaughters; // accepting kink daughters
	104
045e6d86	105	TH1F* fhCutStatistics; // Histogram: statistics of what cuts the tracks did not survive
045e6d86	106	TH2F* fhCutCorrelation; // Histogram: 2d statistics plot
563113d0	107
563113d0	108	TH1F* fhQA[kNHist][kNStepQA]; // QA Histograms
045e6d86	109	TBits *fBitmap ; // stores single selection decisions
563113d0	110
563113d0	111	// QA histogram setters
db6722a5	112	Int_t fhNBinsNSigma; // number of bins+1: dca in units of sigma
	113	Int_t fhNBinsRequireSigma; // number of bins+1: require successful calcuation
	114	Int_t fhNBinsAcceptKink; // number of bins+1: acceptkink daughters
	115	Int_t fhNBinsDcaXY; // number of bins+1: dca in transverse plane
	116	Int_t fhNBinsDcaZ; // number of bins+1: dca along beam axis
	117	Int_t fhNBinsDcaXYnorm; // number of bins+1: normalised dca in transverse plane
	118	Int_t fhNBinsDcaZnorm; // number of bins+1: normalised dca along beam axis
045e6d86	119
	120	Double_t *fhBinLimNSigma; //[fhNBinsNSigma] bin limits: dca in units of sigma
	121	Double_t *fhBinLimRequireSigma;//[fhNBinsRequireSigma] bin limits: require successful calcuation
	122	Double_t *fhBinLimAcceptKink;//[fhNBinsAcceptKink] bin limits: acceptkink daughters
	123	Double_t *fhBinLimDcaXY;//[fhNBinsDcaXY] bin limits: dca in transverse plane
	124	Double_t *fhBinLimDcaZ; //[fhNBinsDcaZ] bin limits: dca along beam axis
	125	Double_t *fhBinLimDcaXYnorm; //[fhNBinsDcaXYnorm] bin limits: normalised dca in transverse plane
	126	Double_t *fhBinLimDcaZnorm;//[fhNBinsDcaZnorm] bin limits: normalised dca along beam axis
	127
	128	ClassDef(AliCFTrackIsPrimaryCuts,2);
563113d0	129	};
	130
	131	#endif