AliRDHFCutsLctoV0.h

Go to the documentation of this file.

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

/* $Id$ */

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

#include "AliRDHFCuts.h"

class AliRDHFCutsLctoV0 : public AliRDHFCuts
{
 public:

  enum ELctoV0channel {
    kLcToK0Spr=0x0001,
    kLcToLBarpi=0x0002,
    kLcToLpi=0x0004
  };

 enum ELctoV0pidStrategy {
  kTOFandTPC=0,
  kTOForTPCveto=1,
  kTOFandTPCasym1=2,
  kTOFandTPCasym2=3,
  kTPClowTOFhigh=4,
  kTPClowTOFintermediateTOForTPChigh=5,
  kTPClowTOFintermediateTOForTPChighA=6,
  kTPClowTOForTPCintermediate_high=7,
  kTPClowTOForTPCintermediate_highA=8,
  kCombinedTPCTOF=9
 };

  AliRDHFCutsLctoV0(const char* name="CutsLctoV0", Short_t v0channel=0);

  virtual ~AliRDHFCutsLctoV0();

  AliRDHFCutsLctoV0(const AliRDHFCutsLctoV0& source);
  AliRDHFCutsLctoV0& operator=(const AliRDHFCutsLctoV0& source);

  using AliRDHFCuts::GetCutVarsForOpt;
  virtual void GetCutVarsForOpt(AliAODRecoDecayHF *d,Float_t *vars,Int_t nvars,Int_t *pdgdaughters);

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

  using AliRDHFCuts::IsSelectedPID;
  virtual Int_t IsSelectedPID(AliAODRecoDecayHF* obj);


  Bool_t PreSelect(TObject* obj, AliAODv0 *v0, AliVTrack *bachelorTrack);

  Int_t IsSelectedSingleCut(TObject* obj, Int_t selectionLevel, Int_t cutIndex, AliAODEvent* aod=0x0);

  Int_t CombineCuts (Int_t returnvalueTrack, Int_t returnvalue, Int_t returnvaluePID) 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(7,iPtBin)] : 1.e6);}

  void SetPidSelectionFlag(Int_t a) {fPidSelectionFlag=a;}
  Int_t GetPidSelectionFlag() {return fPidSelectionFlag;}

  Bool_t AreLctoV0DaughtersSelected(AliAODRecoDecayHF *rd, AliAODEvent* aod=0x0) const;
  Bool_t ApplySingleProtonCuts(AliAODTrack *trk, AliAODEvent* aod);
  Bool_t ApplySingleK0Cuts(AliAODv0 *v0, AliAODEvent* aod);
  Bool_t ApplyCandidateCuts(AliAODRecoDecayHF *rd, AliAODEvent* aod, Bool_t spdfirst);

  Int_t GetV0Type();

  void SetHighPtCut(Float_t highPtCut) {fHighPtCut=highPtCut;};
  Float_t GetHighPtCut() const {return fHighPtCut;};

  void SetLowPtCut(Float_t lowPtCut) {fLowPtCut=lowPtCut;};
  Float_t GetLowPtCut() const {return fLowPtCut;};

  void SetMinCombinedProbability(Int_t nPBins, Float_t *minProb);
  const Float_t *GetMinCombinedProbability() {return fMinCombProb;}

  void SetExcludedCut(Int_t excludedCut) {fExcludedCut=excludedCut;}
  Int_t GetExcludedCut(){return fExcludedCut;}

  virtual void SetStandardCutsPP2010();
  virtual void SetStandardCutsPbPb2010();
  virtual void SetStandardCutsPbPb2011();

  virtual Bool_t IsInFiducialAcceptance(Double_t pt,Double_t y) const;

  void AddTrackCutsV0daughters(AliESDtrackCuts* v0daug)
  { delete fV0daughtersCuts; fV0daughtersCuts = new AliESDtrackCuts(*v0daug); }
  virtual AliESDtrackCuts *GetTrackCutsV0daughters() const {return fV0daughtersCuts;}

  Double_t GetProtonEmissionAngleCMS(AliAODRecoDecayHF *d);
  Double_t GetReSignedd0(AliAODRecoDecayHF *d);
  void SetMagneticField(Double_t a){fBzkG = a;}

  void SetBachelorPLimitsForPID(Int_t nPBins, Float_t *pMin);
  const Float_t *GetBachelorPLimitsForPID() {return fBachelorPLimitsForPID;}

  const Float_t GetNTPCSigmaCutForPreselection() {return fNTPCSigmaCutForPreselection;}
  void SetGetNTPCSigmaCutForPreselection(Float_t a) {fNTPCSigmaCutForPreselection = a;}

  virtual void PrintAll() const;

  const Int_t GetNBachelorPBins() {return fNBachelorPBins;}
  void SetNBachelorPBins(Int_t nPbins) {fNBachelorPBins=nPbins;}

 protected:

  void CheckPID(AliAODTrack *bachelor, AliAODTrack * /*v0Neg*/, AliAODTrack * /*v0Pos*/,
    Bool_t &isBachelorID1, Bool_t &isBachelorID2, Bool_t &isBachelorID4);

  Int_t GetBachelorPBin(Double_t bachelorP) const;

 private:

  Int_t fPidSelectionFlag;
  AliESDtrackCuts *fV0daughtersCuts; 
  Float_t     fV0Type; 
  Float_t fHighPtCut;  
  Float_t fLowPtCut;   
  Int_t   fExcludedCut; 
  Float_t *fMinCombProbVsLcPt; //[fnPtBins] min value for combined PID probabilities (vs Lc pT)
  Double_t fBzkG; 
  Int_t fNBachelorPBins; // number of bachelor p bins
  Float_t *fMinCombProb; //[fNBachelorPBins] min value for combined PID probabilities (vs bachelor p)
  Float_t *fBachelorPLimitsForPID;  //[fNBachelorPBins] limits for bachelor p bins
  Float_t fNTPCSigmaCutForPreselection; 

  //UShort_t fV0channel;

  ClassDef(AliRDHFCutsLctoV0,9);  
};

#endif