PWGHF/vertexingHF/AliRDHFCutsLctopKpi.h

   1 #ifndef ALIRDHFCUTSLCTOPKPI_H
   2 #define ALIRDHFCUTSLCTOPKPI_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 AliRDHFCutsLctopKpi
  10 // class for cuts on AOD reconstructed Lc->pKpi
  11 // Author: A.Dainese, andrea.dainese@pd.infn.it
  12 //***********************************************************
  13
  14 #include "AliRDHFCuts.h"
  15 #include "AliAODPidHF.h"
  16 #include "AliAODRecoDecayHF3Prong.h"
  17
  18 class AliRDHFCutsLctopKpi : public AliRDHFCuts
  19 {
  20  public:
  21
  22  enum EPIDStrategy {
  23   kNSigma,
  24   kCombined,
  25   kCombinedSoft,
  26   kNSigmaStrong,
  27   kCombinedpPb,
  28   kCombinedpPb2,
  29   kNSigmaPbPb
  30  };
  31  enum ECutsStrategy {
  32   kStandard,
  33   kKF
  34  };
  35
  36
  37   AliRDHFCutsLctopKpi(const char* name="CutsLctopKpi");
  38
  39   virtual ~AliRDHFCutsLctopKpi();
  40
  41   AliRDHFCutsLctopKpi(const AliRDHFCutsLctopKpi& source);
  42   AliRDHFCutsLctopKpi& operator=(const AliRDHFCutsLctopKpi& source);
  43
  44   using AliRDHFCuts::GetCutVarsForOpt;
  45   virtual void GetCutVarsForOpt(AliAODRecoDecayHF *d,Float_t *vars,Int_t nvars,Int_t *pdgdaughters){
  46     return GetCutVarsForOpt(d,vars,nvars,pdgdaughters,0x0);
  47   }
  48   virtual void GetCutVarsForOpt(AliAODRecoDecayHF *d,Float_t *vars,Int_t nvars,Int_t *pdgdaughters,AliAODEvent *aod);
  49
  50   void SetPidpion(AliAODPidHF* pidPion) {
  51       if(fPidObjpion) delete fPidObjpion;
  52       fPidObjpion=new AliAODPidHF(*pidPion);
  53       }
  54   void SetPidprot(AliAODPidHF* pidProt) {
  55       if(fPidObjprot) delete fPidObjprot;
  56       fPidObjprot=new AliAODPidHF(*pidProt);
  57   }
  58
  59   virtual void SetStandardCutsPP2010();
  60   virtual void SetStandardCutsPbPb2010();
  61   virtual void SetStandardCutsPbPb2011();
  62   virtual void SetStandardCutsPPb2013();
  63
  64
  65   AliAODPidHF* GetPidpion() const {return fPidObjpion;}
  66   AliAODPidHF* GetPidprot() const {return fPidObjprot;}
  67   void SetPIDStrategy(EPIDStrategy pidStrategy) {
  68    fPIDStrategy=pidStrategy;
  69   }
  70   EPIDStrategy GetPIDStrategy() const {
  71    return fPIDStrategy;
  72   }
  73   void SetCutsStrategy(ECutsStrategy cutsStrategy) {
  74    fCutsStrategy=cutsStrategy;
  75   }
  76   ECutsStrategy GetCutsStrategy() const {
  77    return fCutsStrategy;
  78   }
  79   void SetPIDThreshold(AliPID::EParticleType species,Double_t threshold) {
  80    fPIDThreshold[static_cast<Int_t>(species)]=threshold;
  81   }
  82  Double_t GetPIDThreshold(AliPID::EParticleType species) const {
  83   return fPIDThreshold[static_cast<Int_t>(species)];
  84  }
  85  Bool_t GetUseSpecialCut(){return fUseSpecialCut;}
  86  void SetUseSpecialCut(Bool_t useSpecialCut=kTRUE){fUseSpecialCut=useSpecialCut;}
  87
  88   using AliRDHFCuts::IsSelected;
  89   virtual Int_t IsSelected(TObject* obj,Int_t selectionLevel)
  90                            {return IsSelected(obj,selectionLevel,0);}
  91   virtual Int_t IsSelected(TObject* obj,Int_t selectionLevel,AliAODEvent *aod);
  92   using AliRDHFCuts::IsSelectedPID;
  93   virtual Int_t IsSelectedPID(AliAODRecoDecayHF* obj);
  94   Int_t IsSelectedCombinedPID(AliAODRecoDecayHF* obj);
  95   Int_t IsSelectedCombinedPIDSoft(AliAODRecoDecayHF* obj);
  96   Int_t IsSelectedCombinedPIDpPb(AliAODRecoDecayHF* obj);
  97   Int_t IsSelectedCombinedPIDpPb2(AliAODRecoDecayHF* obj);
  98   Int_t IsSelectedPIDStrong(AliAODRecoDecayHF* obj);
  99   Int_t IsSelectedNSigmaPbPb(AliAODRecoDecayHF* obj);
 100   Int_t CombinePIDCuts (Int_t returnvalue, Int_t returnvaluePID) const;
 101
 102   virtual Bool_t IsInFiducialAcceptance(Double_t pt,Double_t y) const;
 103
 104   Float_t GetMassCut(Int_t iPtBin=0) const { return (GetCuts() ? fCutsRD[GetGlobalIndex(0,iPtBin)] : 1.e6);}
 105   Float_t GetDCACut(Int_t iPtBin=0) const { return (GetCuts() ? fCutsRD[GetGlobalIndex(11,iPtBin)] : 1.e6);}
 106
 107   void SetUseImpParProdCorrCut(Bool_t use){
 108     fUseImpParProdCorrCut=use;
 109   }
 110   Bool_t GetUseImpParProdCorrCut() const {
 111     return fUseImpParProdCorrCut;
 112   }
 113
 114   AliKFParticle* ReconstructKF(AliAODRecoDecayHF3Prong *d,Int_t *pdgs,Double_t field,Bool_t constraint) const;
 115  protected:
 116   AliAODPidHF *fPidObjprot;
 117   AliAODPidHF *fPidObjpion;
 118   Bool_t fUseImpParProdCorrCut; //switch for cut on d0p*d0K vs. d0K*d0pi
 119
 120 private:
 121   EPIDStrategy fPIDStrategy;                // PIS strategy (nsigma, combined)
 122   Double_t fPIDThreshold[AliPID::kSPECIES]; // PID threshold for each species
 123   ECutsStrategy fCutsStrategy;              // cut strategy (standard or KF)
 124   Bool_t fUseSpecialCut;
 125
 126   ClassDef(AliRDHFCutsLctopKpi,9);  // class for cuts on AOD reconstructed Lc->pKpi
 127 };
 128
 129 #endif
 130
 131