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

#ifndef ALIDIELECTRONEVENTCUTS_H
#define ALIDIELECTRONEVENTCUTS_H

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

//#############################################################
//#                                                           # 
//#         Class AliDielectronEventCuts                     #
//#                                                           #
//#  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 <TF1.h>

#include <AliAnalysisUtils.h>
#include <AliAnalysisCuts.h>

class AliTriggerAnalysis;
class AliESDVertex;
class AliAODVertex;


class AliDielectronEventCuts : public AliAnalysisCuts {
public:
  enum EVtxType { kVtxTracks=0, kVtxSPD, kVtxTPC, kVtxAny, kVtxTracksOrSPD };
  
  AliDielectronEventCuts();
  AliDielectronEventCuts(const char*name, const char* title);

  virtual ~AliDielectronEventCuts();


  void SetVertexType(EVtxType type)             { fVtxType=type;                }
  void SetVertexZ(Double_t zmin, Double_t zmax) { fVtxZmin=zmin; fVtxZmax=zmax; }
  void SetRequireVertex(Bool_t req=kTRUE)       { fRequireVtx=req;              }
  void SetRequireV0and(UChar_t type=1)          { fRequireV0and=type;           }
  void SetMinVtxContributors(Int_t min=1)       { fMinVtxContributors=min;      }
  void SetCutOnMultipicityITSTPC(Bool_t mult=kTRUE) { fMultITSTPC=mult;         }
  void SetCentralityRange(Double_t min, Double_t max) { fCentMin=min; fCentMax=max; }
  void SetCutOnV0MultipicityNTrks(TF1* parMean, TF1* parSigma, Double_t cutSigma=3.) { fparMean=parMean; fparSigma=parSigma; fcutSigma=cutSigma; }
  void SetCutOnNVtxContributorsGloablTPC(TF1* parMin, TF1* parMax) { fparMinVtxContributors=parMin; fparMaxVtxContributors=parMax; }
  void SetRequire2013vertexandevent(Bool_t req13 = kTRUE) {fRequire13sel = req13; }
  //
  //Analysis cuts interface
  //
  virtual Bool_t IsSelected(TObject* event);
  Bool_t IsSelectedESD(TObject* event);
  Bool_t IsSelectedAOD(TObject* event);
  virtual Bool_t IsSelected(TList*   /* list */ ) {return kFALSE;}

  void Print(const Option_t* option = "") const;

private:
  static const char* fgkVtxNames[AliDielectronEventCuts::kVtxTracksOrSPD+1];  //vertex names
  Double_t fVtxZmin;                // minimum z vertex position
  Double_t fVtxZmax;                // maximum z vertex position
  Bool_t   fRequireVtx;             // require a vertex
  Int_t    fMinVtxContributors;     // min number of vertex contributors
  Bool_t   fMultITSTPC;             // if to cut on the ITS TPC multiplicity correlation (Pb-Pb)
  Double_t fCentMin;                // minimum multiplity percentile
  Double_t fCentMax;                // maximum multiplity percentile
  EVtxType fVtxType;                // vertex type
  Bool_t fRequire13sel;             //bit to select event and vertex selection proposed for 2013 in 
                                    //https://twiki.cern.ch/twiki/bin/viewauth/ALICE/PAVertexSelectionStudies
  AliAnalysisUtils fUtils;          //data member to use utility class for event and vertex selection in 2013
  

  UChar_t fRequireV0and;             // use V0and triggered events only

  AliTriggerAnalysis *fTriggerAnalysis; //! trigger analysis class
  const AliESDVertex *fkVertex;         //! current vertex
  const AliAODVertex *fkVertexAOD;      //! current vertex AOD

  TF1*     fparMean;                // parametrization of the mean values
  TF1*     fparSigma;               // parametrization of the sigmas
  Double_t fcutSigma;               // number of absolut sigmas inclusion
  TF1*     fparMinVtxContributors;  // parametrization of #vtx contributors global vs TPC (lower limit) 
  TF1*     fparMaxVtxContributors;  // parametrization of #vtx contributors global vs TPC (upper limit) 
  AliDielectronEventCuts(const AliDielectronEventCuts &c);
  AliDielectronEventCuts &operator=(const AliDielectronEventCuts &c);

  
  ClassDef(AliDielectronEventCuts,2)         // Dielectron EventCuts
};


#endif
Commit	Line	Data
	1	#ifndef ALIDIELECTRONEVENTCUTS_H
	2	#define ALIDIELECTRONEVENTCUTS_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 AliDielectronEventCuts #
	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 <TF1.h>
	23
	24	#include <AliAnalysisUtils.h>
	25	#include <AliAnalysisCuts.h>
	26
	27	class AliTriggerAnalysis;
	28	class AliESDVertex;
	29	class AliAODVertex;
	30
	31
	32	class AliDielectronEventCuts : public AliAnalysisCuts {
	33	public:
	34	enum EVtxType { kVtxTracks=0, kVtxSPD, kVtxTPC, kVtxAny, kVtxTracksOrSPD };
	35
	36	AliDielectronEventCuts();
	37	AliDielectronEventCuts(const charname, const char title);
	38
	39	virtual ~AliDielectronEventCuts();
	40
	41
	42	void SetVertexType(EVtxType type) { fVtxType=type; }
	43	void SetVertexZ(Double_t zmin, Double_t zmax) { fVtxZmin=zmin; fVtxZmax=zmax; }
	44	void SetRequireVertex(Bool_t req=kTRUE) { fRequireVtx=req; }
	45	void SetRequireV0and(UChar_t type=1) { fRequireV0and=type; }
	46	void SetMinVtxContributors(Int_t min=1) { fMinVtxContributors=min; }
	47	void SetCutOnMultipicityITSTPC(Bool_t mult=kTRUE) { fMultITSTPC=mult; }
	48	void SetCentralityRange(Double_t min, Double_t max) { fCentMin=min; fCentMax=max; }
	49	void SetCutOnV0MultipicityNTrks(TF1* parMean, TF1* parSigma, Double_t cutSigma=3.) { fparMean=parMean; fparSigma=parSigma; fcutSigma=cutSigma; }
	50	void SetCutOnNVtxContributorsGloablTPC(TF1* parMin, TF1* parMax) { fparMinVtxContributors=parMin; fparMaxVtxContributors=parMax; }
	51	void SetRequire2013vertexandevent(Bool_t req13 = kTRUE) {fRequire13sel = req13; }
	52	//
	53	//Analysis cuts interface
	54	//
	55	virtual Bool_t IsSelected(TObject* event);
	56	Bool_t IsSelectedESD(TObject* event);
	57	Bool_t IsSelectedAOD(TObject* event);
	58	virtual Bool_t IsSelected(TList* /* list */ ) {return kFALSE;}
	59
	60	void Print(const Option_t* option = "") const;
	61
	62	private:
	63	static const char* fgkVtxNames[AliDielectronEventCuts::kVtxTracksOrSPD+1]; //vertex names
	64	Double_t fVtxZmin; // minimum z vertex position
	65	Double_t fVtxZmax; // maximum z vertex position
	66	Bool_t fRequireVtx; // require a vertex
	67	Int_t fMinVtxContributors; // min number of vertex contributors
	68	Bool_t fMultITSTPC; // if to cut on the ITS TPC multiplicity correlation (Pb-Pb)
	69	Double_t fCentMin; // minimum multiplity percentile
	70	Double_t fCentMax; // maximum multiplity percentile
	71	EVtxType fVtxType; // vertex type
	72	Bool_t fRequire13sel; //bit to select event and vertex selection proposed for 2013 in
	73	//https://twiki.cern.ch/twiki/bin/viewauth/ALICE/PAVertexSelectionStudies
	74	AliAnalysisUtils fUtils; //data member to use utility class for event and vertex selection in 2013
	75
	76
	77	UChar_t fRequireV0and; // use V0and triggered events only
	78
	79	AliTriggerAnalysis *fTriggerAnalysis; //! trigger analysis class
	80	const AliESDVertex *fkVertex; //! current vertex
	81	const AliAODVertex *fkVertexAOD; //! current vertex AOD
	82
	83	TF1* fparMean; // parametrization of the mean values
	84	TF1* fparSigma; // parametrization of the sigmas
	85	Double_t fcutSigma; // number of absolut sigmas inclusion
	86	TF1* fparMinVtxContributors; // parametrization of #vtx contributors global vs TPC (lower limit)
	87	TF1* fparMaxVtxContributors; // parametrization of #vtx contributors global vs TPC (upper limit)
	88	AliDielectronEventCuts(const AliDielectronEventCuts &c);
	89	AliDielectronEventCuts &operator=(const AliDielectronEventCuts &c);
	90
	91
	92	ClassDef(AliDielectronEventCuts,2) // Dielectron EventCuts
	93	};
	94
	95
	96	#endif