AliRDHFCutsDstoKKpi.h

Go to the documentation of this file.

#ifndef ALIRDHFCUTSDSTOKKPI_H
#define ALIRDHFCUTSDSTOKKPI_H
/* Copyright(c) 1998-2010, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */ 

//***********************************************************
//***********************************************************

#include "AliRDHFCuts.h"

class AliRDHFCutsDstoKKpi : public AliRDHFCuts 
{
 public:

  AliRDHFCutsDstoKKpi(const char* name="CutsDstoKKpi");
  
  virtual ~AliRDHFCutsDstoKKpi();

  AliRDHFCutsDstoKKpi(const AliRDHFCutsDstoKKpi& source);
  AliRDHFCutsDstoKKpi& operator=(const AliRDHFCutsDstoKKpi& source); 
 
  using AliRDHFCuts::GetCutVarsForOpt;
  virtual void GetCutVarsForOpt(AliAODRecoDecayHF *d,Float_t *vars,Int_t nvars,Int_t *pdgdaughters){
    return GetCutVarsForOpt(d,vars,nvars,pdgdaughters,0x0);
  }
  virtual void GetCutVarsForOpt(AliAODRecoDecayHF *d,Float_t *vars,Int_t nvars,Int_t *pdgdaughters,AliAODEvent *aod);

  using AliRDHFCuts::IsSelected;
  virtual Int_t IsSelected(TObject* obj,Int_t selectionLevel){
    return IsSelected(obj,selectionLevel,0x0);
  }
  virtual Int_t IsSelected(TObject* obj,Int_t selectionLevel,AliAODEvent* aod);

  virtual Int_t IsSelectedPID(AliAODRecoDecayHF *rd);
  Int_t IsSelectedPID(Double_t Pt, TObjArray aodtracks);
  virtual Int_t IsSelectedPIDBayes(AliAODRecoDecayHF *rd);
  Int_t IsSelectedPIDBayes(Double_t Pt, TObjArray aodtracks);
  virtual void SetStandardCutsPP2010();
   
  virtual Bool_t IsInFiducialAcceptance(Double_t pt,Double_t y) const;
  Float_t GetMassCut(Int_t iPtBin=0) const { return (GetCuts() ? fCutsRD[GetGlobalIndex(0,iPtBin)] : 1.e6);}
  Float_t GetDCACut(Int_t iPtBin=0) const { return (GetCuts() ? fCutsRD[GetGlobalIndex(11,iPtBin)] : 1.e6);}
  Float_t GetPhiMassCut(Int_t iPtBin=0) const { return (GetCuts() ? fCutsRD[GetGlobalIndex(12,iPtBin)] : 1.e6);}
  Float_t GetK0starMassCut(Int_t iPtBin=0) const { return (GetCuts() ? fCutsRD[GetGlobalIndex(13,iPtBin)] : 1.e6);}

  UInt_t GetPIDTrackTPCTOFBitMap(AliAODTrack *track) const;

  void Setd0MeasMinusExpCut(Int_t nPtBins, Float_t *cutval);
  void Setd0Cut(Int_t nPtBins, Float_t *cutval);
  void EnableK0starChannel(){fCheckK0star=kTRUE;}
  void DisableK0starChannel(){fCheckK0star=kFALSE;}
  Bool_t IsK0starChannelEnabled() const{ return fCheckK0star;}

  const Float_t *Getd0MeasMinusExpCut() const {return fMaxd0MeasMinusExp;} 
  const Float_t *Getd0Cut() const {return fMaxd0;} 

  virtual Int_t PreSelect(TObjArray aodTracks);

  enum TrackPIDBit{kTPCPionLess1,kTPCPionMore1Less2,kTPCPionMore2Less3,kTPCPionMore3,
                   kTPCKaonLess1,kTPCKaonMore1Less2,kTPCKaonMore2Less3,kTPCKaonMore3,
                   kTPCProtonLess1,kTPCProtonMore1Less2,kTPCProtonMore2Less3,kTPCProtonMore3,
                   kTOFPionLess1,kTOFPionMore1Less2,kTOFPionMore2Less3,kTOFPionMore3,
                   kTOFKaonLess1,kTOFKaonMore1Less2,kTOFKaonMore2Less3,kTOFKaonMore3,
                   kTOFProtonLess1,kTOFProtonMore1Less2,kTOFProtonMore2Less3,kTOFProtonMore3};
                
                 
  enum EDsPid {kConservative, kStrong, kStrongPDep, kBayesianMaxProb, kBayesianThreshold, kBayesianWeights};
  void SetPidOption(Int_t opt){
    fPidOption=opt;
  }
  void SetUseBayesianPIDWithMaxProb(Double_t dist=0.01){
    fPidOption=kBayesianMaxProb;
    fDistToMaxProb=dist;
    fPidHF->SetUseCombined(kTRUE);
  }

  void SetUseBayesianPIDWithThresholds(Double_t thr=0.05){
    fPidOption=kBayesianThreshold;
    fBayesThreshold=thr;
    fPidHF->SetUseCombined(kTRUE);
  }
  void SetUseBayesianPIDWithWeights(){
    fPidOption=kBayesianWeights;
    fPidHF->SetUseCombined(kTRUE);
  }
  void SetMaxPtStrongPid(Float_t spid){fMaxPtStrongPid=spid;}
  void SetMaxPStrongPidK(Float_t spid){fMaxPStrongPidK=spid;}
  void SetMaxPStrongPidpi(Float_t spid){fMaxPStrongPidpi=spid;}
  
  void SetUseReferencePhiMass(Double_t value){fUseRefPhiMass = kTRUE; fPhiMassRef = value;}
    
  Int_t GetPidOption() const {return fPidOption;}
  Double_t GetWeightForKKpi() const {return fWeightKKpi;}
  Double_t GetWeightForpiKK() const {return fWeightpiKK;}

  Bool_t IsCutOnResonancesApplied() const {return fCutOnResonances;}
  void  ApplyCutOnResonances(Bool_t opt=kTRUE){
    fCutOnResonances=opt;
  }

  Double_t ComputeInvMass2(AliAODTrack* track1, AliAODTrack* track2, Int_t pdg1, Int_t pdg2);

  virtual void PrintAll() const;

 protected:
 
  Bool_t fCutOnResonances;  
  Int_t fPidOption;         
  Float_t fMaxPtStrongPid; 
  Float_t fMaxPStrongPidK; 
  Float_t fMaxPStrongPidpi; 
  Double_t fDistToMaxProb; 
  Double_t fBayesThreshold;
  Double_t fWeightKKpi; 
  Double_t fWeightpiKK; 
  Double_t fPhiMassRef; 
  Bool_t fUseRefPhiMass; 
  Bool_t fUsed0MeasMinusExpCut; 
  Float_t* fMaxd0MeasMinusExp;  //[fnPtBins] cut values on d0meas-d0exp
  Bool_t fUsed0Cut; 
  Float_t* fMaxd0;  //[fnPtBins] cut values on d0
  Bool_t fCheckK0star;  

  ClassDef(AliRDHFCutsDstoKKpi,7);  
};

#endif