AliRDHFCutsXictoeleXifromAODtracks.h

Go to the documentation of this file.

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

/* $Id$ */

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

#include "AliRDHFCuts.h"

class AliRDHFCutsXictoeleXifromAODtracks : public AliRDHFCuts
{
 public:

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

  AliRDHFCutsXictoeleXifromAODtracks(const char* name="CutsXictoeleXi");
  virtual ~AliRDHFCutsXictoeleXifromAODtracks();
  AliRDHFCutsXictoeleXifromAODtracks(const AliRDHFCutsXictoeleXifromAODtracks& source);
  AliRDHFCutsXictoeleXifromAODtracks& operator=(const AliRDHFCutsXictoeleXifromAODtracks& 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, AliAODTrack *trkpid);
  Bool_t IsSelectedCombinedeID(AliAODTrack* trk);
  Int_t IsSelected(TLorentzVector* vtrk, TLorentzVector *vcasc, Double_t *cutvars, Int_t selectionLevel);

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

  void SetUseV0Topology(Int_t a) { fUseV0Topology=a; }
  Int_t GetUseV0Topology() { return fUseV0Topology; }

  Bool_t SingleTrkCuts(AliAODTrack *trk, AliAODTrack *trkpid, AliAODVertex *primvert);
  Bool_t SingleTrkCutsNoPID(AliAODTrack *trk, AliAODTrack *trkpid, AliAODVertex *primvert);
  Bool_t SingleCascadeCuts(AliAODcascade *casc, Double_t *vert);
  Bool_t TagConversions(AliAODTrack *etrk, Int_t *id2index, AliAODEvent *evt, Int_t ntrk, Double_t &minmass);
  Bool_t TagConversionsSameSign(AliAODTrack *etrk, Int_t *id2index, AliAODEvent *evt, Int_t ntrk, Double_t &minmass);
  Bool_t SelectWithRoughCuts(AliAODcascade *casc, AliAODTrack *trk1);

  void SetMagneticField(Double_t a){fBzkG = a;}
  void SetPrimaryVertex(Double_t *a){fPrimVert[0] = a[0];fPrimVert[1] = a[1];fPrimVert[2] = a[2];}

  void SetProdTrackTPCNclsPIDMin(Int_t a){fProdTrackTPCNclsPIDMin=a;}
  void SetProdTrackTPCNclsRatioMin(Double_t a){fProdTrackTPCNclsRatioMin=a;}
  void SetProdUseAODFilterBit(Bool_t a){fProdUseAODFilterBit=a;}
  void SetProdAODFilterBit(Int_t a){fProdAODFilterBit=a;}
  void SetProdRejectTrackWithShared(Bool_t a){fProdRejectTrackWithShared=a;}
  void SetProdMassTolLambda(Double_t a){fProdMassTolLambda=a;}
  void SetProdMassTolXiRough(Double_t a){fProdMassTolXiRough=a;}
  void SetProdMassTolXi(Double_t a){fProdMassTolXi=a;}
  void SetProdMassRejOmega(Double_t a){fProdMassRejOmega=a;}
  void SetProdRfidMinV0(Double_t a){fProdRfidMinV0=a;}
  void SetProdRfidMaxV0(Double_t a){fProdRfidMaxV0=a;}
  void SetProdRfidMinXi(Double_t a){fProdRfidMinXi=a;}
  void SetProdRfidMaxXi(Double_t a){fProdRfidMaxXi=a;}
  void SetProdCascProperDecayLengthMax(Double_t a){fProdCascProperDecayLengthMax=a;}
  void SetProdDcaXiDaughtersMax(Double_t a){fProdDcaXiDaughtersMax=a;}
  void SetProdDcaV0DaughtersMax(Double_t a){fProdDcaV0DaughtersMax=a;}
  void SetProdDcaBachToPrimVertexMin(Double_t a){fProdDcaBachToPrimVertexMin=a;}
  void SetProdDcaV0ToPrimVertexMin(Double_t a){fProdDcaV0ToPrimVertexMin=a;}
  void SetProdDcaV0PrToPrimVertexMin(Double_t a){fProdDcaV0PrToPrimVertexMin=a;}
  void SetProdDcaV0PiToPrimVertexMin(Double_t a){fProdDcaV0PiToPrimVertexMin=a;}
  void SetProdXiCosineOfPoiningAngleMin(Double_t a){fProdXiCosineOfPoiningAngleMin=a;}
  void SetProdV0CosineOfPoiningAngleXiMin(Double_t a){fProdV0CosineOfPoiningAngleXiMin=a;}
  void SetProdCascNTPCClustersMin(Double_t a){fProdCascNTPCClustersMin=a;}
  void SetProdCascEtaRange(Double_t a, Double_t b){fProdCascEtaMin=a;fProdCascEtaMax=b;}
  void SetProdCascRapRange(Double_t a, Double_t b){fProdCascRapMin=a;fProdCascRapMax=b;}
  void SetProdCascPtRange(Double_t a, Double_t b){fProdCascPtMin=a;fProdCascPtMax=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;}
  Int_t   GetProdAODFilterBit(){return fProdAODFilterBit;}
  Bool_t   GetProdRejectTrackWithShared(){return fProdRejectTrackWithShared;}
  Double_t GetProdMassTolLambda(){return fProdMassTolLambda;}
  Double_t GetProdMassTolXiRough(){return fProdMassTolXiRough;}
  Double_t GetProdMassTolXi(){return fProdMassTolXi;}
  Double_t GetProdMassRejOmega(){return fProdMassRejOmega;}
  Double_t GetProdRfidMinV0(){return fProdRfidMinV0;}
  Double_t GetProdRfidMaxV0(){return fProdRfidMaxV0;}
  Double_t GetProdRfidMinXi(){return fProdRfidMinXi;}
  Double_t GetProdRfidMaxXi(){return fProdRfidMaxXi;}
  Double_t GetProdCascProperDecayLengthMax(){return fProdCascProperDecayLengthMax;}
  Double_t GetProdDcaXiDaughtersMax(){return fProdDcaXiDaughtersMax;}
  Double_t GetProdDcaV0DaughtersMax(){return fProdDcaV0DaughtersMax;}
  Double_t GetProdDcaBachToPrimVertexMin(){return fProdDcaBachToPrimVertexMin;}
  Double_t GetProdDcaV0ToPrimVertexMin(){return fProdDcaV0ToPrimVertexMin;}
  Double_t GetProdDcaV0PrToPrimVertexMin(){return fProdDcaV0PrToPrimVertexMin;}
  Double_t GetProdDcaV0PiToPrimVertexMin(){return fProdDcaV0PiToPrimVertexMin;}
  Double_t GetProdXiCosineOfPoiningAngleMin(){return fProdXiCosineOfPoiningAngleMin;}
  Double_t GetProdV0CosineOfPoiningAngleXiMin(){return fProdV0CosineOfPoiningAngleXiMin;}
  Double_t GetProdRoughMassTol(){return fProdRoughMassTol;}
  Double_t GetProdRoughPtMin(){return fProdRoughPtMin;}
  Double_t GetProdCascNTPCClustersMin(){return fProdCascNTPCClustersMin;}
  void GetProdCascEtaRange(Double_t &a, Double_t &b){a=fProdCascEtaMin;b=fProdCascEtaMax;}
  void GetProdCascRapRange(Double_t &a, Double_t &b){a=fProdCascRapMin;b=fProdCascRapMax;}
  void GetProdCascPtRange(Double_t &a, Double_t &b){a=fProdCascPtMin;b=fProdCascPtMax;}

  Bool_t GetUseCascadePID(){return fUseCascadePID;}
  void SetUseCascadePID(Bool_t a){fUseCascadePID=a;}
  void SetPidCascPi(AliAODPidHF* pidPion) { 
      if(fPidObjCascPi) delete fPidObjCascPi;
      fPidObjCascPi=new AliAODPidHF(*pidPion);
      }
  AliAODPidHF* GetPidCascPi() const {return fPidObjCascPi;}
  void SetPidCascPr(AliAODPidHF* pidProton) { 
      if(fPidObjCascPr) delete fPidObjCascPr;
      fPidObjCascPr=new AliAODPidHF(*pidProton);
      }
  AliAODPidHF* GetPidCascPr() const {return fPidObjCascPr;}
  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;}
  Double_t GetConversionMassMax(){return fConversionMassMax;}
  Double_t GetEleXiMassMax(){return fEleXiMassMax;}

  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;}
  void SetConversionMassMax(Double_t a){fConversionMassMax=a;}
  void SetEleXiMassMax(Double_t a){fEleXiMassMax=a;}

  Bool_t IsPeakRegion(AliAODcascade *c);
  Bool_t IsPeakRegion(TLorentzVector *c);
  Bool_t IsSideBand(AliAODcascade *c);
  Bool_t IsSideBand(TLorentzVector *c);

  void SetSftPosR125(AliAODTrack *track,Double_t bfield,Double_t priVtx[3], Double_t *XSftR125);
  Double_t dEtaSR125(Double_t *postrack1,Double_t *postrack2);
  Double_t dPhiSR125(Double_t *postrack1,Double_t *postrack2);
  Double_t GetdPhiSdEtaSR125(AliAODTrack *tracke, AliAODTrack *trackp,AliAODTrack *trackn, AliAODTrack *trackb, Double_t bfield,Double_t priVtx[3], Double_t &dPhiS_ep, Double_t &dEtaS_ep,Double_t &dPhiS_en, Double_t &dEtaS_en, Double_t &dPhiS_eb, Double_t &dEtaS_eb);
  Double_t DeltaPhi(AliAODcascade *casc, AliAODTrack *trk);
  Double_t CosOpeningAngle(AliAODcascade *casc, AliAODTrack *trk);
  Double_t DeltaEta(AliAODcascade *casc, AliAODTrack *trk);

 protected:
  
 private:

  EPIDStrategy fPIDStrategy;        
  Double_t fCombinedPIDThreshold;   
  Bool_t fUseCascadePID;            
  AliAODPidHF *fPidObjCascPi;         
  AliAODPidHF *fPidObjCascPr;         
//  Bool_t   fUseOnTheFlyV0;          /// Flag to check if we use on-the-fly v0
  Int_t   fUseV0Topology;          
  Double_t fBzkG; 
  Double_t fPrimVert[3];
  
  Int_t fProdTrackTPCNclsPIDMin;      
  Double_t fProdTrackTPCNclsRatioMin;      
  Bool_t   fProdUseAODFilterBit;    
  Int_t    fProdAODFilterBit;    
  Bool_t   fProdRejectTrackWithShared;    
  Double_t fProdMassTolLambda;      
  Double_t fProdMassTolXiRough;          
  Double_t fProdMassTolXi;          
  Double_t fProdMassRejOmega;          
  Double_t fProdRfidMinV0;          
  Double_t fProdRfidMaxV0;          
  Double_t fProdRfidMinXi;          
  Double_t fProdRfidMaxXi;          
  Double_t fProdCascProperDecayLengthMax;        
  Double_t fProdDcaXiDaughtersMax;  
  Double_t fProdDcaV0DaughtersMax;  
  Double_t fProdDcaBachToPrimVertexMin;  
  Double_t fProdDcaV0ToPrimVertexMin;  
  Double_t fProdDcaV0PrToPrimVertexMin;  
  Double_t fProdDcaV0PiToPrimVertexMin;  
  Double_t fProdXiCosineOfPoiningAngleMin;  
  Double_t fProdV0CosineOfPoiningAngleXiMin;  
  Double_t fProdCascNTPCClustersMin;         
  Double_t fProdCascEtaMin; 
  Double_t fProdCascEtaMax; 
  Double_t fProdCascRapMin; 
  Double_t fProdCascRapMax; 
  Double_t fProdCascPtMin; 
  Double_t fProdCascPtMax; 
  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; 

  Double_t fConversionMassMax; 
  Double_t fEleXiMassMax; 
  
  ClassDef(AliRDHFCutsXictoeleXifromAODtracks,9);
};

#endif