CORRFW/AliCFEventGenCuts.h

   1
   2 #ifndef ALICFEVENTGENCUTS_H
   3 #define ALICFEVENTGENCUTS_H
   4 /**************************************************************************
   5  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   6  *                                                                        *
   7  * Author: The ALICE Off-line Project.                                    *
   8  * Contributors are mentioned in the code where appropriate.              *
   9  *                                                                        *
  10  * Permission to use, copy, modify and distribute this software and its   *
  11  * documentation strictly for non-commercial purposes is hereby granted   *
  12  * without fee, provided that the above copyright notice appears in all   *
  13  * copies and that both the copyright notice and this permission notice   *
  14  * appear in the supporting documentation. The authors make no claims     *
  15  * about the suitability of this software for any purpose. It is          *
  16  * provided "as is" without express or implied warranty.                  *
  17  **************************************************************************/
  18 // Cut on the Event at generator level: for the moment just
  19 // the requirements on the MB process type, on the number of charged tracks
  20 // and on the vertex 3-D position are implemented
  21 // The argument of IsSelected member function (passed object) is cast into
  22 // an AliMCEvent. In the future may be modified to use AliVEvent interface
  23 // and to include more cut variables.
  24 // The class derives from AliCFCutBase
  25 // Author:S.Arcelli Silvia.Arcelli@cern.ch
  26
  27
  28 #include "AliCFCutBase.h"
  29 #include "AliGenEventHeader.h"
  30 class TBits;
  31 class AliEventGenHeader;
  32 //____________________________________________________________________________
  33 class AliCFEventGenCuts: public AliCFCutBase
  34 {
  35  public :
  36   AliCFEventGenCuts() ;
  37   AliCFEventGenCuts(Char_t* name, Char_t* title) ;
  38   AliCFEventGenCuts(const AliCFEventGenCuts& c) ;
  39   AliCFEventGenCuts& operator=(const AliCFEventGenCuts& c) ;
  40   ~AliCFEventGenCuts();
  41   void GetBitMap(TObject *obj, TBits *bitmap);
  42   Bool_t IsSelected(TObject* obj);
  43   void Init(){;};
  44   static Bool_t IsMBProcessType(Int_t isel, TObject *obj);
  45
  46   //number of embedded cuts
  47   enum{kNCuts=5};
  48
  49   //Label the MB MC processes
  50   enum PrType {
  51    kND, kSD, kDD
  52   };
  53
  54   void   SetMBProcessType(PrType process = kND) {fMBProcessType=process;} // cut values setter
  55   Int_t  GetMBProcessType()const {return fMBProcessType;} // cut values getter
  56
  57
  58
  59   void SetNTracksCut(Int_t xMin=-1, Int_t xMax=1000000) {fNTracksMin=xMin; fNTracksMax=xMax;} // cut values setter
  60
  61   void SetRequireVtxCuts(Bool_t vtx=kFALSE) {fRequireVtxCuts=vtx;} // cut values setter
  62   void SetVertexXCut(Double_t xMin=-1.e99, Double_t xMax=1.e99) { fVtxXMin=xMin; fVtxXMax=xMax;} // cut values setter
  63   void SetVertexYCut(Double_t yMin=-1.e99, Double_t yMax=1.e99) { fVtxYMin=yMin; fVtxYMax=yMax;} // cut values setter
  64   void SetVertexZCut(Double_t zMin=-1.e99, Double_t zMax=1.e99) { fVtxZMin=zMin; fVtxZMax=zMax;} // cut values setter
  65
  66   Int_t    GetNTracksMin() const {return fNTracksMin;} // cut values getter
  67   Int_t    GetNTracksMax() const {return fNTracksMax;} // cut values getter
  68   Bool_t   GetRequireVtxCuts() const {return fRequireVtxCuts;} // cut value getter
  69   Double_t GetVertexXMax() const {return fVtxXMax;} // cut values getter
  70   Double_t GetVertexYMax() const {return fVtxYMax;} // cut values getter
  71   Double_t GetVertexZMax() const {return fVtxZMax;} // cut values getter
  72   Double_t GetVertexXMin() const {return fVtxXMin;} // cut values getter
  73   Double_t GetVertexYMin() const {return fVtxYMin;} // cut values getter
  74   Double_t GetVertexZMin() const {return fVtxZMin;} // cut values getter
  75
  76
  77  private:
  78   TBits * SelectionBitMap(TObject* obj);
  79   static Int_t MBProcessType(AliGenEventHeader *genHeader);
  80
  81   Int_t fMBProcessType ; //the type of selected MB process
  82   Int_t fNTracksMin; //minimum number of particles in the event
  83   Int_t fNTracksMax; //maximum number of particles in the event
  84   Bool_t fRequireVtxCuts ; //The type of trigger to be checked
  85   Double_t fVtxXMax ; //X vertex position, maximum value
  86   Double_t fVtxYMax ; //Y vertex position, maximum value
  87   Double_t fVtxZMax ; //Z vertex position, maximum value
  88   Double_t fVtxXMin ; //X vertex position, minimum value
  89   Double_t fVtxYMin ; //Y vertex position, minimum value
  90   Double_t fVtxZMin ; //Z vertex position, minimum value
  91
  92   TBits *fBitMap ; //cut mask
  93
  94   ClassDef(AliCFEventGenCuts,1);
  95 };
  96
  97 #endif