AliRDHFCutsLctoeleLambdafromAODtracks.h

Go to the documentation of this file.

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

/* $Id$ */

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

#include "AliRDHFCuts.h"

class AliRDHFCutsLctoeleLambdafromAODtracks : public AliRDHFCuts
{
 public:

  enum EPIDStrategy{
    kNSigmaCuts,
    kNSigmaCustomizedCuts,
    kNSigmaCustomizedPtDepCuts,
    kCombinedCuts
  };

  AliRDHFCutsLctoeleLambdafromAODtracks(const char* name="CutsLctoeleLambda");
  virtual ~AliRDHFCutsLctoeleLambdafromAODtracks();
  AliRDHFCutsLctoeleLambdafromAODtracks(const AliRDHFCutsLctoeleLambdafromAODtracks& source);
  AliRDHFCutsLctoeleLambdafromAODtracks& operator=(const AliRDHFCutsLctoeleLambdafromAODtracks& 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);
  using AliRDHFCuts::IsSelectedPID;
  virtual Int_t IsSelectedPID(AliAODRecoDecayHF* obj);
  Int_t IsSelectedCombinedPID(AliAODRecoDecayHF* obj);
  Bool_t IsSelectedeID(AliAODTrack* trk);
  Bool_t IsSelectedCustomizedeID(AliAODTrack* trk);
  Bool_t IsSelectedCustomizedPtDepeID(AliAODTrack* trk);
  Bool_t IsSelectedCombinedeID(AliAODTrack* trk);

  void SetPIDStrategy(EPIDStrategy pidStrategy){fPIDStrategy=pidStrategy;}
  EPIDStrategy GetPIDStrategy() const {return fPIDStrategy;}
  void SetCombinedPIDThreshold(Double_t a){fCombinedPIDThreshold=a;}
  Double_t GetCombinedPIDThreshold(){return fCombinedPIDThreshold;}

  void SetUseOnTheFlyV0(Bool_t a) { fUseOnTheFlyV0=a; }
  Bool_t GetUseOnTheFlyV0() { return fUseOnTheFlyV0; }

  Bool_t SingleTrkCuts(AliAODTrack *trk, AliAODVertex *vert);
  Bool_t SingleTrkCutsNoPID(AliAODTrack *trk, AliAODVertex *vert);
  Bool_t SingleV0Cuts(AliAODv0 *v0, AliAODVertex *vert);
  Bool_t SelectWithRoughCuts(AliAODv0 *v0, AliAODTrack *trk1);

  void SetProdTrackTPCNclsPIDMin(Int_t a){fProdTrackTPCNclsPIDMin=a;}
  void SetProdTrackTPCNclsRatioMin(Double_t a){fProdTrackTPCNclsRatioMin=a;}
  void SetProdUseAODFilterBit(Bool_t a){fProdUseAODFilterBit=a;}
  void SetProdV0MassTolLambda(Double_t a){fProdV0MassTolLambda=a;}
  void SetProdV0MassTolLambdaRough(Double_t a){fProdV0MassTolLambdaRough=a;}
  void SetProdV0PtMin(Double_t a){fProdV0PtMin=a;}
  void SetProdV0CosPointingAngleToPrimVtxMin(Double_t a){fProdV0CosPointingAngleToPrimVtxMin=a;}
  void SetProdV0DcaDaughtersMax(Double_t a){fProdV0DcaDaughtersMax=a;}
  void SetProdV0DaughterEtaRange(Double_t a){fProdV0DaughterEtaRange=a;}
  void SetProdV0DaughterPtMin(Double_t a){fProdV0DaughterPtMin=a;}
  void SetProdV0DaughterTPCClusterMin(Double_t a){fProdV0DaughterTPCClusterMin=a;}
  void SetProdV0DaughterTPCCrossRatioMin(Double_t a){fProdV0DaughterTPCCrossRatioMin=a;}
  void SetProdRfidMinV0(Double_t a){fProdRfidMinV0=a;}
  void SetProdRfidMaxV0(Double_t a){fProdRfidMaxV0=a;}
  void SetProdDcaV0ToPrimVertexMin(Double_t a){fProdDcaV0ToPrimVertexMin=a;}
  void SetProdDcaV0PrToPrimVertexMin(Double_t a){fProdDcaV0PrToPrimVertexMin=a;}
  void SetProdDcaV0PiToPrimVertexMin(Double_t a){fProdDcaV0PiToPrimVertexMin=a;}
  void SetProdV0ProperDecayLengthMax(Double_t a){fProdV0ProperDecayLengthMax=a;}
  void SetProdMassRejK0s(Double_t a){fProdMassRejK0s=a;}
  void SetProdV0EtaRange(Double_t a, Double_t b){fProdV0EtaMin=a;fProdV0EtaMax=b;}
  void SetProdV0RapRange(Double_t a, Double_t b){fProdV0RapMin=a;fProdV0RapMax=b;}
  
  void SetProdRoughMassTol(Double_t a){fProdRoughMassTol=a;}
  void SetProdRoughPtMin(Double_t a){fProdRoughPtMin=a;}
  
  Int_t GetProdTrackTPCNclsPIDMin(){return fProdTrackTPCNclsPIDMin;}
  Double_t GetProdTrackTPCNclsRatioMin(){return fProdTrackTPCNclsRatioMin;}
  Bool_t   GetProdUseAODFilterBit(){return fProdUseAODFilterBit;}
  Double_t GetProdV0MassTolLambda(){return fProdV0MassTolLambda;}
  Double_t GetProdV0MassTolLambdaRough(){return fProdV0MassTolLambdaRough;}
  Double_t GetProdV0PtMin(){return fProdV0PtMin;}
  Double_t GetProdV0CosPointingAngleToPrimVtxMin(){return fProdV0CosPointingAngleToPrimVtxMin;}
  Double_t GetProdV0DcaDaughtersMax(){return fProdV0DcaDaughtersMax;}
  Double_t GetProdV0DaughterEtaRange(){return fProdV0DaughterEtaRange;}
  Double_t GetProdV0DaughterPtMin(){return fProdV0DaughterPtMin;}
  Double_t GetProdV0DaughterTPCClusterMin(){return fProdV0DaughterTPCClusterMin;}
  Double_t GetProdV0DaughterTPCCrossRatioMin(){return fProdV0DaughterTPCCrossRatioMin;}
  Double_t GetProdRfidMinV0(){return fProdRfidMinV0;}
  Double_t GetProdRfidMaxV0(){return fProdRfidMaxV0;}
  Double_t GetProdDcaV0ToPrimVertexMin(){return fProdDcaV0ToPrimVertexMin;}
  Double_t GetProdDcaV0PrToPrimVertexMin(){return fProdDcaV0PrToPrimVertexMin;}
  Double_t GetProdDcaV0PiToPrimVertexMin(){return fProdDcaV0PiToPrimVertexMin;}
  Double_t GetProdV0ProperDecayLengthMax(){return fProdV0ProperDecayLengthMax;}
  Double_t GetProdMassRejK0s(){return fProdMassRejK0s;}
  void GetProdV0EtaRange(Double_t &a, Double_t &b){a=fProdV0EtaMin;b=fProdV0EtaMax;}
  void GetProdV0RapRange(Double_t &a, Double_t &b){a=fProdV0RapMin;b=fProdV0RapMax;}

  Double_t GetProdRoughMassTol(){return fProdRoughMassTol;}
  Double_t GetProdRoughPtMin(){return fProdRoughPtMin;}

  void SetUseLambdaPID(Bool_t a){fUseLambdaPID=a;}
  Bool_t GetUseLambdaPID(){return fUseLambdaPID;}
  void SetPidProton(AliAODPidHF* pidProton) { 
      if(fPidObjProton) delete fPidObjProton;
      fPidObjProton=new AliAODPidHF(*pidProton);
      }
  AliAODPidHF* GetPidProton() const {return fPidObjProton;}
  void SetPidPion(AliAODPidHF* pidPion) { 
      if(fPidObjPion) delete fPidObjPion;
      fPidObjPion=new AliAODPidHF(*pidPion);
      }
  AliAODPidHF* GetPidPion() const {return fPidObjPion;}
  void GetSigmaElectronTPCRange(Double_t &a,Double_t &b){a=fSigmaElectronTPCMin;b=fSigmaElectronTPCMax;}
  void GetSigmaElectronTOFRange(Double_t &a,Double_t &b){a=fSigmaElectronTOFMin;b=fSigmaElectronTOFMax;}
  void GetSigmaElectronTPCPtDepPars(Double_t &a,Double_t &b){a=fSigmaElectronTPCPtDepPar0;b=fSigmaElectronTPCPtDepPar1;}
  void GetSigmaElectronTPCPtDepPars(Double_t &a,Double_t &b,Double_t &c){a=fSigmaElectronTPCPtDepPar0;b=fSigmaElectronTPCPtDepPar1;c=fSigmaElectronTPCPtDepPar2;}

  void SetExcludePionTPC(Bool_t a){fExcludePionTPC=a;}
  void SetExcludeProtonTPC(Bool_t a){fExcludeProtonTPC=a;}
  void SetExcludeKaonTPC(Bool_t a){fExcludeKaonTPC=a;}
  void SetExcludenSigmaPionTPC(Double_t a){fExcludenSigmaPionTPC=a;}
  void SetExcludenSigmaProtonTPC(Double_t a){fExcludenSigmaProtonTPC=a;}
  void SetExcludenSigmaKaonTPC(Double_t a){fExcludenSigmaKaonTPC=a;}
  void SetSigmaElectronTPCRange(Double_t a,Double_t b){fSigmaElectronTPCMin=a;fSigmaElectronTPCMax=b;}
  void SetSigmaElectronTOFRange(Double_t a,Double_t b){fSigmaElectronTOFMin=a;fSigmaElectronTOFMax=b;}
  void SetSigmaElectronTPCPtDepPars(Double_t a,Double_t b){fSigmaElectronTPCPtDepPar0=a;fSigmaElectronTPCPtDepPar1=b;}
  void SetSigmaElectronTPCPtDepPars(Double_t a,Double_t b,Double_t c){fSigmaElectronTPCPtDepPar0=a;fSigmaElectronTPCPtDepPar1=b;fSigmaElectronTPCPtDepPar2=c;}
  Bool_t IsPeakRegion(AliAODv0 *c);
  Bool_t IsPeakRegion(TLorentzVector *c);
  Bool_t IsSideBand(AliAODv0 *c);
  Bool_t IsSideBand(TLorentzVector *c);

 protected:
  
 private:

  EPIDStrategy fPIDStrategy;        
  Double_t fCombinedPIDThreshold;   
  Bool_t fUseLambdaPID;            
  AliAODPidHF *fPidObjProton;         
  AliAODPidHF *fPidObjPion;         
  Bool_t   fUseOnTheFlyV0;          
  
  Int_t fProdTrackTPCNclsPIDMin;      
  Double_t fProdTrackTPCNclsRatioMin;      
  Bool_t   fProdUseAODFilterBit;    
  Double_t fProdV0MassTolLambda;       
  Double_t fProdV0MassTolLambdaRough;       
  Double_t fProdV0PtMin;            
  Double_t fProdV0CosPointingAngleToPrimVtxMin;
  Double_t fProdV0DcaDaughtersMax;  
  Double_t fProdV0DaughterEtaRange; 
  Double_t fProdV0DaughterPtMin;    
  Double_t fProdV0DaughterTPCClusterMin;
  Double_t fProdV0DaughterTPCCrossRatioMin;
  Double_t fProdRfidMinV0;          
  Double_t fProdRfidMaxV0;          
  Double_t fProdDcaV0ToPrimVertexMin;  
  Double_t fProdDcaV0PrToPrimVertexMin;  
  Double_t fProdDcaV0PiToPrimVertexMin;  
  Double_t fProdV0ProperDecayLengthMax;        
  Double_t fProdMassRejK0s;          
  Double_t fProdV0EtaMin; 
  Double_t fProdV0EtaMax; 
  Double_t fProdV0RapMin; 
  Double_t fProdV0RapMax; 
  Double_t fProdRoughMassTol;       
  Double_t fProdRoughPtMin;         

  Bool_t fExcludePionTPC;    
  Bool_t fExcludeProtonTPC;  
  Bool_t fExcludeKaonTPC;    
  Double_t fExcludenSigmaPionTPC; 
  Double_t fExcludenSigmaProtonTPC; 
  Double_t fExcludenSigmaKaonTPC; 
  Double_t fSigmaElectronTPCMin;  
  Double_t fSigmaElectronTPCPtDepPar0; 
  Double_t fSigmaElectronTPCPtDepPar1; 
  Double_t fSigmaElectronTPCPtDepPar2; 
  Double_t fSigmaElectronTPCMax; 
  Double_t fSigmaElectronTOFMin; 
  Double_t fSigmaElectronTOFMax; 

  ClassDef(AliRDHFCutsLctoeleLambdafromAODtracks,4);
};

#endif