[u/mrichter/AliRoot.git] / PWGDQ / dielectron / AliDielectronPID.h

#ifndef ALIDIELECTRONPID_H
#define ALIDIELECTRONPID_H

/* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

//#############################################################
//#                                                           # 
//#         Class AliDielectronPID                     #
//#                                                           #
//#  Authors:                                                 #
//#   Anton     Andronic, GSI / A.Andronic@gsi.de             #
//#   Ionut C.  Arsene,   GSI / I.C.Arsene@gsi.de             #
//#   Julian    Book,     Uni Ffm / Julian.Book@cern.ch       #
//#   Frederick Kramer,   Uni Ffm, / Frederick.Kramer@cern.ch #
//#   Magnus    Mager,    CERN / Magnus.Mager@cern.ch         #
//#   WooJin J. Park,     GSI / W.J.Park@gsi.de               #
//#   Jens      Wiechula, Uni HD / Jens.Wiechula@cern.ch      #
//#                                                           #
//#############################################################

#include <AliPID.h>
#include <AliAnalysisCuts.h>
#include <AliTRDPIDResponse.h>

class TF1;
class TList;
class AliVTrack;
class TGraph;
class AliPIDResponse;
class AliDielectronVarManager;
class AliDielectronVarCuts;

class AliDielectronPID : public AliAnalysisCuts {
public:
  enum DetType {kITS, kTPC, kTRD, kTRDeleEff, kTRDeleEff2D, kTOF, kEMCAL};
  enum PIDbitType {kIgnore=0, kRequire, kIfAvailable};
  
  AliDielectronPID();
  AliDielectronPID(const char*name, const char* title);

  virtual ~AliDielectronPID();

  void AddCut(DetType det, AliPID::EParticleType type, Double_t nSigmaLow, Double_t nSigmaUp=-99999.,
              Double_t min=0, Double_t max=0, Bool_t exclude=kFALSE, UInt_t pidBitType=AliDielectronPID::kRequire, 
	      Int_t var=-1);

  void AddCut(DetType det, AliPID::EParticleType type, Double_t nSigmaLow, TF1 * const funUp,
              Double_t min=0, Double_t max=0, Bool_t exclude=kFALSE, UInt_t pidBitType=AliDielectronPID::kRequire, 
	      Int_t var=-1);

  void AddCut(DetType det, AliPID::EParticleType type, TF1 * const funLow, Double_t nSigmaUp,
              Double_t min=0, Double_t max=0, Bool_t exclude=kFALSE, UInt_t pidBitType=AliDielectronPID::kRequire, 
	      Int_t var=-1);

  void AddCut(DetType det, AliPID::EParticleType type, TF1 * const funLow, TF1 * const funUp,
              Double_t min=0, Double_t max=0, Bool_t exclude=kFALSE, UInt_t pidBitType=AliDielectronPID::kRequire, 
	      Int_t var=-1);
  void AddCut(DetType det, AliPID::EParticleType type, Double_t nSigmaLow, Double_t nSigmaUp, Double_t min, Double_t max, Bool_t exclude, UInt_t pidBitType,              TF1 * const funSigma);
  void AddCut(DetType det, AliPID::EParticleType type, Double_t nSigmaLow, Double_t nSigmaUp,
  	      AliDielectronVarCuts *varcuts, Bool_t exclude=kFALSE, UInt_t pidBitType=AliDielectronPID::kRequire );
  
  void SetDefaults(Int_t def);

  //
  //Analysis cuts interface
  //const
  virtual Bool_t IsSelected(TObject* track);
  virtual Bool_t IsSelected(TList*   /* list */ ) {return kFALSE;}

  static void SetCorrGraph(TGraph * const gr) { fgFitCorr=gr; }
  static TGraph *GetCorrGraph()  { return fgFitCorr; }
  
  static void SetCorrVal(Double_t run);
  static Double_t GetCorrVal()   { return fgCorr; }
  static Double_t GetCorrValdEdx()   { return fgCorrdEdx; }
  
  static void SetCorrGraphdEdx(TGraph * const gr) { fgdEdxRunCorr=gr; }
  static TGraph *GetCorrGraphdEdx()  { return fgdEdxRunCorr; }

  static void SetEtaCorrFunction(TF1 *fun) {fgFunEtaCorr=fun;}
  static void SetCentroidCorrFunction(TF1 *fun, UInt_t varx, UInt_t vary=0, UInt_t varz=0);
  static void SetWidthCorrFunction(TF1 *fun, UInt_t varx, UInt_t vary=0, UInt_t varz=0);
  static TF1* GetEtaCorrFunction() { return fgFunEtaCorr; }
  static TF1* GetCentroidCorrFunction() { return fgFunCntrdCorr; }
  static TF1* GetWidthCorrFunction() { return fgFunWdthCorr; }

  static Double_t GetEtaCorr(const AliVTrack *track);
  static Double_t GetCntrdCorr(const AliVTrack *track) { return (fgFunCntrdCorr ? GetPIDCorr(track,fgFunCntrdCorr) : 0.0); }
  static Double_t GetWdthCorr(const AliVTrack *track)  { return (fgFunWdthCorr  ? GetPIDCorr(track,fgFunWdthCorr)  : 1.0); }

private:
  enum {kNmaxPID=30};

  DetType  fDetType[kNmaxPID];    //detector type of nsigma cut
  AliPID::EParticleType fPartType[kNmaxPID]; //particle type
  Float_t  fNsigmaLow[kNmaxPID];  //lower nsigma bound
  Float_t  fNsigmaUp[kNmaxPID];   //upper nsigma bound
  Double_t fmin[kNmaxPID];        //lower cut limit
  Double_t fmax[kNmaxPID];        //upper cut limit
  Bool_t   fExclude[kNmaxPID];    //use as exclusion band
  TF1     *fFunUpperCut[kNmaxPID];//use function as upper cut
  TF1     *fFunLowerCut[kNmaxPID];//use function as lower cut
  UChar_t  fNcuts;                //number of cuts
  UChar_t  fRequirePIDbit[kNmaxPID]; //How to make use of the pid bit (see)
  UShort_t fActiveCuts[kNmaxPID]; // list of activated cuts
  Double_t fSigmaFunLow[kNmaxPID]; // lower bound for fFunSigma
  Double_t fSigmaFunUp[kNmaxPID];  // upper bound for fFunSigma
  TF1      *fFunSigma[kNmaxPID];   // use function as cut range
  AliDielectronVarCuts *fVarCuts[kNmaxPID]; // varcuts

  AliPIDResponse *fPIDResponse;   //! pid response object
  
  static TGraph *fgFitCorr;       //spline fit object to correct the nsigma deviation in the TPC electron band
  static Double_t fgCorr;         //!correction value for current run. Set if fgFitCorr is set and SetCorrVal(run)
                                  // was called
  static Double_t fgCorrdEdx;     //!dEdx correction value for current run. Set if fgFitCorr is set and SetCorrVal(run)
                                  // was called
  static TF1    *fgFunEtaCorr;    //function for eta correction of electron sigma
  static TF1    *fgFunCntrdCorr;  //!function for correction of electron sigma (centroid)
  static TF1    *fgFunWdthCorr;   //!function for correction of electron sigma (width)
  static TGraph *fgdEdxRunCorr;   //run by run correction for dEdx

  static Double_t GetPIDCorr(const AliVTrack *track, TF1 *fun);
  
  Bool_t IsSelectedITS(AliVTrack * const part, Int_t icut);
  Bool_t IsSelectedTPC(AliVTrack * const part, Int_t icut);
  Bool_t IsSelectedTRD(AliVTrack * const part, Int_t icut);
  Bool_t IsSelectedTRDeleEff(AliVTrack * const part, Int_t icut, AliTRDPIDResponse::ETRDPIDMethod PIDmethod=AliTRDPIDResponse::kLQ1D);
  Bool_t IsSelectedTOF(AliVTrack * const part, Int_t icut);
  Bool_t IsSelectedEMCAL(AliVTrack * const part, Int_t icut);

  AliDielectronPID(const AliDielectronPID &c);
  AliDielectronPID &operator=(const AliDielectronPID &c);

  ClassDef(AliDielectronPID,5)         // Dielectron PID
};

#endif
Commit	Line	Data
8df8e382	1	#ifndef ALIDIELECTRONPID_H
	2	#define ALIDIELECTRONPID_H
	3
	4	/* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	//#############################################################
	8	//# #
	9	//# Class AliDielectronPID #
	10	//# #
	11	//# Authors: #
	12	//# Anton Andronic, GSI / A.Andronic@gsi.de #
	13	//# Ionut C. Arsene, GSI / I.C.Arsene@gsi.de #
	14	//# Julian Book, Uni Ffm / Julian.Book@cern.ch #
	15	//# Frederick Kramer, Uni Ffm, / Frederick.Kramer@cern.ch #
	16	//# Magnus Mager, CERN / Magnus.Mager@cern.ch #
	17	//# WooJin J. Park, GSI / W.J.Park@gsi.de #
	18	//# Jens Wiechula, Uni HD / Jens.Wiechula@cern.ch #
	19	//# #
	20	//#############################################################
	21
	22	#include <AliPID.h>
8df8e382	23	#include <AliAnalysisCuts.h>
6d5dea7c	24	#include <AliTRDPIDResponse.h>
8df8e382	25
8df8e382	26	class TF1;
8df8e382	27	class TList;
164bfb53	28	class AliVTrack;
48609e3d	29	class TGraph;
5720c765	30	class AliPIDResponse;
5720c765	31	class AliDielectronVarManager;
8c2440d0	32	class AliDielectronVarCuts;
8df8e382	33
	34	class AliDielectronPID : public AliAnalysisCuts {
	35	public:
6d5dea7c	36	enum DetType {kITS, kTPC, kTRD, kTRDeleEff, kTRDeleEff2D, kTOF, kEMCAL};
5720c765	37	enum PIDbitType {kIgnore=0, kRequire, kIfAvailable};
8df8e382	38
	39	AliDielectronPID();
	40	AliDielectronPID(const charname, const char title);
	41
	42	virtual ~AliDielectronPID();
	43
	44	void AddCut(DetType det, AliPID::EParticleType type, Double_t nSigmaLow, Double_t nSigmaUp=-99999.,
5720c765	45	Double_t min=0, Double_t max=0, Bool_t exclude=kFALSE, UInt_t pidBitType=AliDielectronPID::kRequire,
5720c765	46	Int_t var=-1);
8df8e382	47
8df8e382	48	void AddCut(DetType det, AliPID::EParticleType type, Double_t nSigmaLow, TF1 * const funUp,
5720c765	49	Double_t min=0, Double_t max=0, Bool_t exclude=kFALSE, UInt_t pidBitType=AliDielectronPID::kRequire,
5720c765	50	Int_t var=-1);
8df8e382	51
8df8e382	52	void AddCut(DetType det, AliPID::EParticleType type, TF1 * const funLow, Double_t nSigmaUp,
5720c765	53	Double_t min=0, Double_t max=0, Bool_t exclude=kFALSE, UInt_t pidBitType=AliDielectronPID::kRequire,
5720c765	54	Int_t var=-1);
8df8e382	55
8df8e382	56	void AddCut(DetType det, AliPID::EParticleType type, TF1 * const funLow, TF1 * const funUp,
5720c765	57	Double_t min=0, Double_t max=0, Bool_t exclude=kFALSE, UInt_t pidBitType=AliDielectronPID::kRequire,
5720c765	58	Int_t var=-1);
88204efa	59	void AddCut(DetType det, AliPID::EParticleType type, Double_t nSigmaLow, Double_t nSigmaUp, Double_t min, Double_t max, Bool_t exclude, UInt_t pidBitType, TF1 * const funSigma);
8c2440d0	60	void AddCut(DetType det, AliPID::EParticleType type, Double_t nSigmaLow, Double_t nSigmaUp,
8c2440d0	61	AliDielectronVarCuts *varcuts, Bool_t exclude=kFALSE, UInt_t pidBitType=AliDielectronPID::kRequire );
8df8e382	62
	63	void SetDefaults(Int_t def);
	64
8df8e382	65	//
	66	//Analysis cuts interface
	67	//const
	68	virtual Bool_t IsSelected(TObject* track);
	69	virtual Bool_t IsSelected(TList* /* list */ ) {return kFALSE;}
48609e3d	70
48609e3d	71	static void SetCorrGraph(TGraph * const gr) { fgFitCorr=gr; }
d327d9cd	72	static TGraph *GetCorrGraph() { return fgFitCorr; }
d327d9cd	73
48609e3d	74	static void SetCorrVal(Double_t run);
48609e3d	75	static Double_t GetCorrVal() { return fgCorr; }
d327d9cd	76	static Double_t GetCorrValdEdx() { return fgCorrdEdx; }
	77
	78	static void SetCorrGraphdEdx(TGraph * const gr) { fgdEdxRunCorr=gr; }
	79	static TGraph *GetCorrGraphdEdx() { return fgdEdxRunCorr; }
5720c765	80
5720c765	81	static void SetEtaCorrFunction(TF1 *fun) {fgFunEtaCorr=fun;}
61749cc7	82	static void SetCentroidCorrFunction(TF1 *fun, UInt_t varx, UInt_t vary=0, UInt_t varz=0);
61749cc7	83	static void SetWidthCorrFunction(TF1 *fun, UInt_t varx, UInt_t vary=0, UInt_t varz=0);
5720c765	84	static TF1* GetEtaCorrFunction() { return fgFunEtaCorr; }
61749cc7	85	static TF1* GetCentroidCorrFunction() { return fgFunCntrdCorr; }
61749cc7	86	static TF1* GetWidthCorrFunction() { return fgFunWdthCorr; }
5720c765	87
0c09cae4	88	static Double_t GetEtaCorr(const AliVTrack *track);
61749cc7	89	static Double_t GetCntrdCorr(const AliVTrack *track) { return (fgFunCntrdCorr ? GetPIDCorr(track,fgFunCntrdCorr) : 0.0); }
61749cc7	90	static Double_t GetWdthCorr(const AliVTrack *track) { return (fgFunWdthCorr ? GetPIDCorr(track,fgFunWdthCorr) : 1.0); }
5720c765	91
8df8e382	92	private:
a823f01b	93	enum {kNmaxPID=30};
5720c765	94
8df8e382	95	DetType fDetType[kNmaxPID]; //detector type of nsigma cut
	96	AliPID::EParticleType fPartType[kNmaxPID]; //particle type
	97	Float_t fNsigmaLow[kNmaxPID]; //lower nsigma bound
	98	Float_t fNsigmaUp[kNmaxPID]; //upper nsigma bound
5720c765	99	Double_t fmin[kNmaxPID]; //lower cut limit
5720c765	100	Double_t fmax[kNmaxPID]; //upper cut limit
8df8e382	101	Bool_t fExclude[kNmaxPID]; //use as exclusion band
	102	TF1 *fFunUpperCut[kNmaxPID];//use function as upper cut
	103	TF1 *fFunLowerCut[kNmaxPID];//use function as lower cut
	104	UChar_t fNcuts; //number of cuts
61d106d3	105	UChar_t fRequirePIDbit[kNmaxPID]; //How to make use of the pid bit (see)
5720c765	106	UShort_t fActiveCuts[kNmaxPID]; // list of activated cuts
88204efa	107	Double_t fSigmaFunLow[kNmaxPID]; // lower bound for fFunSigma
	108	Double_t fSigmaFunUp[kNmaxPID]; // upper bound for fFunSigma
	109	TF1 *fFunSigma[kNmaxPID]; // use function as cut range
8c2440d0	110	AliDielectronVarCuts *fVarCuts[kNmaxPID]; // varcuts
8df8e382	111
5720c765	112	AliPIDResponse *fPIDResponse; //! pid response object
2a14a7b1	113
48609e3d	114	static TGraph *fgFitCorr; //spline fit object to correct the nsigma deviation in the TPC electron band
	115	static Double_t fgCorr; //!correction value for current run. Set if fgFitCorr is set and SetCorrVal(run)
	116	// was called
d327d9cd	117	static Double_t fgCorrdEdx; //!dEdx correction value for current run. Set if fgFitCorr is set and SetCorrVal(run)
d327d9cd	118	// was called
5720c765	119	static TF1 *fgFunEtaCorr; //function for eta correction of electron sigma
04c83ead	120	static TF1 *fgFunCntrdCorr; //!function for correction of electron sigma (centroid)
04c83ead	121	static TF1 *fgFunWdthCorr; //!function for correction of electron sigma (width)
d327d9cd	122	static TGraph *fgdEdxRunCorr; //run by run correction for dEdx
8c2440d0	123
61749cc7	124	static Double_t GetPIDCorr(const AliVTrack track, TF1 fun);
8d42b5b9	125
ffbede40	126	Bool_t IsSelectedITS(AliVTrack * const part, Int_t icut);
	127	Bool_t IsSelectedTPC(AliVTrack * const part, Int_t icut);
	128	Bool_t IsSelectedTRD(AliVTrack * const part, Int_t icut);
6d5dea7c	129	Bool_t IsSelectedTRDeleEff(AliVTrack * const part, Int_t icut, AliTRDPIDResponse::ETRDPIDMethod PIDmethod=AliTRDPIDResponse::kLQ1D);
ffbede40	130	Bool_t IsSelectedTOF(AliVTrack * const part, Int_t icut);
5720c765	131	Bool_t IsSelectedEMCAL(AliVTrack * const part, Int_t icut);
8df8e382	132
8df8e382	133	AliDielectronPID(const AliDielectronPID &c);
	134	AliDielectronPID &operator=(const AliDielectronPID &c);
	135
61749cc7	136	ClassDef(AliDielectronPID,5) // Dielectron PID
8df8e382	137	};
8df8e382	138
8df8e382	139	#endif