Fixing wrong usage of bits
[u/mrichter/AliRoot.git] / CORRFW / AliCFEventRecCuts.h
1 #ifndef ALICFEVENTRECCUTS_H
2 #define ALICFEVENTRECCUTS_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:S.Arcelli Silvia.Arcelli@cern.ch
25
26
27 #include "AliCFCutBase.h"
28 class TH1F;
29 class TBits;
30 //_____________________________________________________________________________
31 class AliCFEventRecCuts: public AliCFCutBase 
32 {
33  public :
34   AliCFEventRecCuts() ;
35   AliCFEventRecCuts(Char_t* name, Char_t* title) ;
36   AliCFEventRecCuts(const AliCFEventRecCuts& c) ;
37   AliCFEventRecCuts& operator=(const AliCFEventRecCuts& c) ;
38   ~AliCFEventRecCuts();
39   Bool_t IsSelected(TObject* obj);
40   Bool_t IsSelected(TList* /*list*/) {return kTRUE;}
41
42   void SetNTracksCut(Int_t xMin=-1, Int_t xMax=1000000) {fNTracksMin=xMin; fNTracksMax=xMax;} // cut values setter
43
44   void SetRequireVtxCuts(Bool_t vtx=kFALSE) {fRequireVtxCuts=vtx;} // cut values setter
45   void SetVertexXCut(Double_t xMin=-1.e99, Double_t xMax=1.e99) { fVtxXMin=xMin; fVtxXMax=xMax;} // cut values setter
46   void SetVertexYCut(Double_t yMin=-1.e99, Double_t yMax=1.e99) { fVtxYMin=yMin; fVtxYMax=yMax;} // cut values setter
47   void SetVertexZCut(Double_t zMin=-1.e99, Double_t zMax=1.e99) { fVtxZMin=zMin; fVtxZMax=zMax;} // cut values setter
48
49   void SetVertexXResCut(Double_t xMax=1.e99) {fVtxXResMax=xMax;} // cut values setter
50   void SetVertexYResCut(Double_t yMax=1.e99){fVtxYResMax=yMax;} // cut values setter
51   void SetVertexZResCut(Double_t zMax=1.e99){fVtxZResMax=zMax;} // cut values setter
52
53   void SetVertexNContributors(Int_t min, Int_t max) {fVtxNCtrbMin=min; fVtxNCtrbMax=max;}
54   void SetUseTPCVertex() {fVtxTPC=kTRUE ; fVtxSPD=kFALSE;} //default is vertex from tracks
55   void SetUseSPDVertex() {fVtxTPC=kFALSE; fVtxSPD=kTRUE ;} //default is vertex from tracks
56
57   Int_t    GetNTracksMin() const {return fNTracksMin;} // cut values getter
58   Int_t    GetNTracksMax() const {return fNTracksMax;} // cut values getter
59   Bool_t   GetRequireVtxCuts() const {return fRequireVtxCuts;} // cut value getter
60   Double_t GetVertexXMax() const {return fVtxXMax;} // cut values getter
61   Double_t GetVertexYMax() const {return fVtxYMax;} // cut values getter
62   Double_t GetVertexZMax() const {return fVtxZMax;} // cut values getter
63   Double_t GetVertexXMin() const {return fVtxXMin;} // cut values getter
64   Double_t GetVertexYMin() const {return fVtxYMin;} // cut values getter
65   Double_t GetVertexZMin() const {return fVtxZMin;} // cut values getter
66   Double_t GetVertexXResMax() const {return fVtxXResMax;} // cut values getter
67   Double_t GetVertexYResMax() const {return fVtxYResMax;} // cut values getter
68   Double_t GetVertexZResMax() const {return fVtxZResMax;} // cut values getter
69
70   // QA histogram setter
71   // please use indices from the enumeration below
72   void SetHistogramBins(Int_t index, Int_t nbins, Double_t *bins);
73   void SetHistogramBins(Int_t index, Int_t nbins, Double_t xmin, Double_t xmax);
74   enum{kNTracks=0,
75        kVtxPosX,
76        kVtxPosY,
77        kVtxPosZ,
78        kVtxResX,
79        kVtxResY,
80        kVtxResZ,
81        kVtxNCtrb,
82        kNCuts,
83        kNStepQA=2
84   };
85   
86  protected:
87   void SelectionBitMap(TObject* obj);
88   void AddQAHistograms(TList *qaList) ;
89   void DefineHistograms();              // books histograms 
90   void Initialise();                    // sets everything to 0
91   void FillHistograms(TObject* obj, Bool_t b);
92
93   Int_t fNTracksMin; //minimum number of esd tracks
94   Int_t fNTracksMax; //maximum number of esd tracks
95   Bool_t fRequireVtxCuts ; //The type of trigger to be checked
96   Double_t fVtxXMax ; //X vertex position, maximum value
97   Double_t fVtxYMax ; //Y vertex position, maximum value 
98   Double_t fVtxZMax ; //Z vertex position, maximum value
99   Double_t fVtxXMin ; //X vertex position, minimum value
100   Double_t fVtxYMin ; //Y vertex position, minimum value
101   Double_t fVtxZMin ; //Z vertex position, minimum value
102   Double_t fVtxXResMax ;//Maximum value of sigma_vtx in X
103   Double_t fVtxYResMax ;//Maximum value of sigma_vtx in X
104   Double_t fVtxZResMax ;//Maximum value of sigma_vtx in X
105   Int_t    fVtxNCtrbMin; //Min number of contributors to vertex
106   Int_t    fVtxNCtrbMax; //Max number of contributors to vertex
107   Bool_t   fVtxTPC;  //Flag for use of TPC vertex
108   Bool_t   fVtxSPD;  //Flag for use of SPD vertex
109
110   TBits *fBitMap ; //cut mask
111
112   TH1F* fhQA[kNCuts][kNStepQA];         // QA Histograms
113
114   ClassDef(AliCFEventRecCuts,3);
115 };
116
117 #endif