CORRFW/AliCFParticleGenCuts.h

   1 /**************************************************************************
   2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                        *
   4  * Author: The ALICE Off-line Project.                                    *
   5  * Contributors are mentioned in the code where appropriate.              *
   6  *                                                                        *
   7  * Permission to use, copy, modify and distribute this software and its   *
   8  * documentation strictly for non-commercial purposes is hereby granted   *
   9  * without fee, provided that the above copyright notice appears in all   *
  10  * copies and that both the copyright notice and this permission notice   *
  11  * appear in the supporting documentation. The authors make no claims     *
  12  * about the suitability of this software for any purpose. It is          *
  13  * provided "as is" without express or implied warranty.                  *
  14  **************************************************************************/
  15
  16 //////////////////////////////////////////////////////////////////////
  17 // AliCFParticleGenCut implementation
  18 // This class is designed to handle
  19 // particle selection at generated level.
  20 //
  21 // author : R. Vernet (renaud.vernet@cern.ch)
  22 //////////////////////////////////////////////////////////////////////
  23
  24
  25 #ifndef ALICFPARTICLEGENCUTS_H
  26 #define ALICFPARTICLEGENCUTS_H
  27
  28 #include "AliCFCutBase.h"
  29
  30 class AliMCEvent;
  31 class TObject;
  32 class AliMCParticle;
  33 class AliStack;
  34 class TList;
  35 class TH1F;
  36 class TH2F;
  37 class TBits;
  38
  39 class AliCFParticleGenCuts : public AliCFCutBase
  40 {
  41  public :
  42   AliCFParticleGenCuts() ;
  43   AliCFParticleGenCuts           (const Char_t* name, const Char_t* title) ;
  44   AliCFParticleGenCuts           (const AliCFParticleGenCuts& c) ;
  45   AliCFParticleGenCuts& operator=(const AliCFParticleGenCuts& c) ;
  46   virtual ~AliCFParticleGenCuts() { };
  47   virtual Bool_t IsSelected(TObject* obj) ;
  48   Bool_t IsSelected(TList* /*list*/) {return kTRUE;}
  49   virtual void   SetEvtInfo(TObject* mcEvent) ;
  50   //static checkers
  51   static Bool_t IsPrimaryCharged(AliMCParticle *mcPart,AliStack*stack);
  52   static Bool_t IsPrimary(AliMCParticle *mcPart,AliStack*stack);
  53   static Bool_t IsCharged(AliMCParticle *mcPart);
  54   static Bool_t IsA(AliMCParticle *mcPart, Int_t pdg, Bool_t abs=kFALSE);
  55
  56   void SetRequireIsCharged   () {fRequireIsCharged  =kTRUE; fRequireIsNeutral  =kFALSE;}
  57   void SetRequireIsNeutral   () {fRequireIsNeutral  =kTRUE; fRequireIsCharged  =kFALSE;}
  58   void SetRequireIsPrimary   () {fRequireIsPrimary  =kTRUE; fRequireIsSecondary=kFALSE;}
  59   void SetRequireIsSecondary () {fRequireIsSecondary=kTRUE; fRequireIsPrimary  =kFALSE;}
  60   void SetRequirePdgCode     (Int_t pdg)            {fRequirePdgCode=kTRUE; fPdgCode=pdg;}
  61   void SetProdVtxRangeX    (Double32_t xmin, Double32_t xmax) {fProdVtxXMin   =xmin; fProdVtxXMax   =xmax;}
  62   void SetProdVtxRangeY    (Double32_t ymin, Double32_t ymax) {fProdVtxYMin   =ymin; fProdVtxYMax   =ymax;}
  63   void SetProdVtxRangeZ    (Double32_t zmin, Double32_t zmax) {fProdVtxZMin   =zmin; fProdVtxZMax   =zmax;}
  64   void SetDecayVtxRangeX   (Double32_t xmin, Double32_t xmax) {fDecayVtxXMin  =xmin; fDecayVtxXMax  =xmax;}
  65   void SetDecayVtxRangeY   (Double32_t ymin, Double32_t ymax) {fDecayVtxYMin  =ymin; fDecayVtxYMax  =ymax;}
  66   void SetDecayVtxRangeZ   (Double32_t zmin, Double32_t zmax) {fDecayVtxZMin  =zmin; fDecayVtxZMax  =zmax;}
  67   void SetDecayLengthRange (Double32_t rmin, Double32_t rmax) {fDecayLengthMin=rmin; fDecayLengthMax=rmax;}
  68   void SetDecayRxyRange    (Double32_t rmin, Double32_t rmax) {fDecayRxyMin   =rmin; fDecayRxyMax   =rmax;}
  69
  70   enum {
  71     kCutCharge,       // ischarged cut
  72     kCutPrimSec,      // isprimary cut
  73     kCutPDGCode,      // PDG code  cut
  74     kCutProdVtxXMin,  // production vertex cut
  75     kCutProdVtxXMax,  // production vertex cut
  76     kCutProdVtxYMin,  // production vertex cut
  77     kCutProdVtxYMax,  // production vertex cut
  78     kCutProdVtxZMin,  // production vertex cut
  79     kCutProdVtxZMax,  // production vertex cut
  80     kCutDecVtxXMin,   // decay vertex cut
  81     kCutDecVtxXMax,   // decay vertex cut
  82     kCutDecVtxYMin,   // decay vertex cut
  83     kCutDecVtxYMax,   // decay vertex cut
  84     kCutDecVtxZMin,   // decay vertex cut
  85     kCutDecVtxZMax,   // decay vertex cut
  86     kCutDecLgthMin,   // decay length cut
  87     kCutDecLgthMax,   // decay length cut
  88     kCutDecRxyMin,    // transverse decay length cut
  89     kCutDecRxyMax,    // transverse decay length cut
  90     kNCuts,           // number of single selections
  91     kNStepQA=2        // number of QA steps (before/after the cuts)
  92   };
  93
  94  private:
  95   AliMCEvent* fMCInfo ;    // pointer to the MC event information
  96   Bool_t     fRequireIsCharged;   // require charged particle
  97   Bool_t     fRequireIsNeutral;   // require neutral particle
  98   Bool_t     fRequireIsPrimary;   // require primary particle
  99   Bool_t     fRequireIsSecondary; // require secondary particle
 100   Bool_t     fRequirePdgCode;     // require check of the PDG code
 101   Int_t      fPdgCode ;           // particle PDG code
 102   Double32_t fProdVtxXMin;        // min X of particle production vertex
 103   Double32_t fProdVtxYMin;        // min Y of particle production vertex
 104   Double32_t fProdVtxZMin;        // min Z of particle production vertex
 105   Double32_t fProdVtxXMax;        // max X of particle production vertex
 106   Double32_t fProdVtxYMax;        // max Y of particle production vertex
 107   Double32_t fProdVtxZMax;        // max Z of particle production vertex
 108   Double32_t fDecayVtxXMin;       // min X of particle decay vertex
 109   Double32_t fDecayVtxYMin;       // min Y of particle decay vertex
 110   Double32_t fDecayVtxZMin;       // min Z of particle decay vertex
 111   Double32_t fDecayVtxXMax;       // max X of particle decay vertex
 112   Double32_t fDecayVtxYMax;       // max Y of particle decay vertex
 113   Double32_t fDecayVtxZMax;       // max Z of particle decay vertex
 114   Double32_t fDecayLengthMin;     // min decay length (absolute)
 115   Double32_t fDecayLengthMax;     // max decay length (absolute)
 116   Double32_t fDecayRxyMin;        // min decay length in transverse plane wrt (0,0,0)
 117   Double32_t fDecayRxyMax;        // max decay length in transverse plane wrt (0,0,0)
 118
 119   //QA histos
 120   TH1F*  fhCutStatistics;               // Histogram: statistics of what cuts the tracks did not survive
 121   TH2F*  fhCutCorrelation;              // Histogram: 2d statistics plot
 122   TH1F*  fhQA[kNCuts][kNStepQA];        // QA Histograms
 123   TBits* fBitmap ;                      // stores single selection decisions
 124
 125   void SelectionBitMap(TObject* obj);
 126   void FillHistograms(TObject* obj, Bool_t afterCuts);
 127   void AddQAHistograms(TList *qaList) ;
 128   void DefineHistograms();
 129
 130   ClassDef(AliCFParticleGenCuts,1);
 131 };
 132
 133 #endif