Place the config and root file at the right place
[u/mrichter/AliRoot.git] / PWGHF / hfe / AliHFEextraEventCuts.h
1 #ifndef ALIHFEEXTRAEVENTCUTS_H
2 #define ALIHFEEXTRAEVENTCUTS_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:R.Bailhache
25
26
27 #include "AliCFCutBase.h"
28 class TH1F;
29 class TBits;
30 class AliVVertex;
31 class AliVEvent;
32 class AliAnalysisUtils;
33 //_____________________________________________________________________________
34 class AliHFEextraEventCuts: public AliCFCutBase 
35 {
36  public :
37   AliHFEextraEventCuts() ;
38   AliHFEextraEventCuts(Char_t* name, Char_t* title) ;
39   AliHFEextraEventCuts(const AliHFEextraEventCuts& c) ;
40   AliHFEextraEventCuts& operator=(const AliHFEextraEventCuts& c) ;
41   ~AliHFEextraEventCuts();
42   Bool_t IsSelected(TObject* obj);
43   Bool_t IsSelected(TList* /*list*/) {return kTRUE;}
44
45   void SetRequireVtxCuts(Bool_t vtx=kFALSE) {fRequireVtxCuts=vtx;} // cut values setter
46   void SetVertexZCut(Double_t zMin=-1.e99, Double_t zMax=1.e99) { fVtxZMin=zMin; fVtxZMax=zMax;} // cut values setter
47
48   void SetVertexNContributors(Int_t min) {fVtxNCtrbMin=min;}
49   void SetUseMixedVertex() {fVtxMixed=kTRUE; fVtxSPD=kFALSE;} //default is vertex from tracks
50   void SetUseSPDVertex() {fVtxSPD=kTRUE; fVtxMixed=kFALSE;} //default is vertex from tracks
51   void SetCheckCorrelationSPDVtx() {fCheckCorrelationSPDVtx=kTRUE;} // check the correlation between z of different vertices
52   void SetCheckSPDResolution() {fVtxResolution = kTRUE;} // check resolution on the SPD vertex
53   void SetpAPileupCut() { fPApileupCut = kTRUE; } 
54  
55   Bool_t   GetRequireVtxCuts() const {return fRequireVtxCuts;} // cut value getter
56   Double_t GetVertexZMax() const {return fVtxZMax;} // cut values getter
57   Double_t GetVertexZMin() const {return fVtxZMin;} // cut values getter
58   
59   // QA histogram setter
60   // please use indices from the enumeration below
61   void SetHistogramBins(Int_t index, Int_t nbins, Double_t *bins);
62   void SetHistogramBins(Int_t index, Int_t nbins, Double_t xmin, Double_t xmax);
63   enum{kVtxPosZ,
64        kVtxNCtrb,
65        kCorrelation,
66        kResolution,
67        kpApileup,
68        kNCuts,
69        kNStepQA=2
70   };
71   
72  protected:
73   void SelectionBitMap(TObject* obj);
74   void AddQAHistograms(TList *qaList) ;
75   void DefineHistograms();              // books histograms 
76   void Initialise();                    // sets everything to 0
77   void FillHistograms(TObject* obj, Bool_t b);
78   const AliVVertex *GetPrimaryVertexSPD(const AliVEvent * const inputEvent);
79   const AliVVertex *GetPrimaryVertexTracks(const AliVEvent *const inputEvent);
80
81   AliAnalysisUtils *fAnalysisUtils;     // Analysis Utils for pA pileup cut
82   Bool_t fRequireVtxCuts ;              // The type of trigger to be checked
83   Double_t fVtxZMax ;                   // Z vertex position, maximum value
84   Double_t fVtxZMin ;                   // Z vertex position, minimum value
85   Int_t    fVtxNCtrbMin;                // Min number of contributors to vertex
86   Bool_t   fVtxMixed;                   // Flag for use of mixed vertex (primary vertex with track, if not SPD vertex)
87   Bool_t   fVtxSPD;                     // Flag for use of SPD vertex 
88   Bool_t   fCheckCorrelationSPDVtx;     // Check the correlation SPD, track vertex
89   Bool_t   fVtxResolution;              // Check vertex resolution cut
90   Bool_t   fPApileupCut;                // Apply pA-pileup cut
91  
92   TBits *fBitMap ; //cut mask
93
94   TH1F* fhQA[kNCuts][kNStepQA];         // QA Histograms
95
96   ClassDef(AliHFEextraEventCuts,3);
97 };
98
99 #endif