Change Mult binning scheme
[u/mrichter/AliRoot.git] / PWGDQ / dielectron / AliDielectronEventCuts.h
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 char*name, 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