PWGHF/vertexingHF/AliRDHFCutsD0toKpi.h

   1 #ifndef ALIRDHFCUTSD0TOKPI_H
   2 #define ALIRDHFCUTSD0TOKPI_H
   3 /* Copyright(c) 1998-2010, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 /* $Id$ */
   7
   8 //***********************************************************
   9 // Class AliRDHFCutsD0toKpi
  10 // class for cuts on AOD reconstructed D0->Kpi
  11 // Author: A.Dainese, andrea.dainese@pd.infn.it
  12 //***********************************************************
  13
  14 #include "AliRDHFCuts.h"
  15
  16 class AliAODEvent;
  17 class AliAODRecoDecayHF;
  18 class AliAODRecoDecayHF2Prong;
  19
  20 class AliRDHFCutsD0toKpi : public AliRDHFCuts
  21 {
  22  public:
  23
  24   AliRDHFCutsD0toKpi(const char* name="CutsD0toKpi");
  25
  26   virtual ~AliRDHFCutsD0toKpi(){}
  27
  28   AliRDHFCutsD0toKpi(const AliRDHFCutsD0toKpi& source);
  29   AliRDHFCutsD0toKpi& operator=(const AliRDHFCutsD0toKpi& source);
  30
  31   using AliRDHFCuts::GetCutVarsForOpt;
  32   virtual void GetCutVarsForOpt(AliAODRecoDecayHF *d,Float_t *vars,Int_t nvars,Int_t *pdgdaughters){
  33     return GetCutVarsForOpt(d,vars,nvars,pdgdaughters,0x0);
  34   }
  35   virtual void GetCutVarsForOpt(AliAODRecoDecayHF *d,Float_t *vars,Int_t nvars,Int_t *pdgdaughters,AliAODEvent *aod);
  36
  37   using AliRDHFCuts::IsSelected;
  38   virtual Int_t IsSelected(TObject* obj,Int_t selectionLevel)
  39                          {return IsSelected(obj,selectionLevel,0);}
  40   virtual Int_t IsSelected(TObject* obj,Int_t selectionLevel,AliAODEvent* aod);
  41
  42   Float_t GetMassCut(Int_t iPtBin=0) const { return (GetCuts() ? fCutsRD[GetGlobalIndex(0,iPtBin)] : 1.e6);}
  43   Float_t GetDCACut(Int_t iPtBin=0) const { return (GetCuts() ? fCutsRD[GetGlobalIndex(1,iPtBin)] : 1.e6);}
  44   Int_t CombineSelectionLevels(Int_t selectionvalTrack,Int_t selectionvalCand,Int_t selectionvalPID)const;
  45   virtual Bool_t IsInFiducialAcceptance(Double_t pt,Double_t y) const;
  46   virtual void SetStandardCutsPP2010();
  47   virtual void SetStandardCutsPP2011_276TeV();
  48   virtual void SetStandardCutsPbPb2010();
  49   virtual void SetStandardCutsPbPb2011();
  50   void SetStandardCutsPbPb2010Peripherals();
  51   virtual Int_t IsSelectedPID(AliAODRecoDecayHF *rd);
  52   Int_t IsSelectedPIDdefault(AliAODRecoDecayHF *rd);
  53   Int_t IsSelectedSpecialCuts(AliAODRecoDecayHF *d) const;
  54   void SetUseSpecialCuts(Bool_t useSpecialCuts) {fUseSpecialCuts=useSpecialCuts;}
  55   void SetMaximumPtSpecialCuts(Double_t pt) { fPtMaxSpecialCuts=pt; }
  56   Double_t GetMaximumPtSpecialCuts() const { return fPtMaxSpecialCuts; }
  57   void SetLowPt(Bool_t lowpt,Double_t ptlow=2.) {fLowPt=lowpt;fPtLowPID=ptlow;}
  58   Bool_t GetUseSpecialCuts() const {return fUseSpecialCuts;}
  59   void SetUseDefaultPID(Bool_t defPID){fDefaultPID=defPID;}
  60   Bool_t GetIsUsedDefPID(){return fDefaultPID;}
  61   Double_t GetPtForPIDtight()const {return fPtLowPID;}
  62   void SetUseKF(Bool_t useKF);
  63   Bool_t GetIsUsedKF() const {return fUseKF;}
  64
  65
  66  protected:
  67   Int_t IsSelectedKF(AliAODRecoDecayHF2Prong* d,AliAODEvent* aod) const;
  68
  69   Bool_t fUseSpecialCuts;  // flag to switch on/off special cuts
  70   Bool_t fLowPt;           // flag to switch on/off different pid for low pt D0
  71   Bool_t fDefaultPID;      // flag to switch on/off the default pid
  72   Bool_t fUseKF;           // flag to switch on/off D0 selection via KF
  73   Double_t fPtLowPID;      // transverse momentum below which the strong PID is applied
  74   Double_t fPtMaxSpecialCuts; // transverse momentum below which the special cuts are applied
  75                               //  if set to zero, used for all pt
  76
  77   ClassDef(AliRDHFCutsD0toKpi,7);  // class for cuts on AOD reconstructed D0->Kpi
  78 };
  79
  80 #endif
  81