updated for photonic electron efficiency

[u/mrichter/AliRoot.git] / PWGHF / vertexingHF / AliRDHFCuts.h

diff --git a/PWGHF/vertexingHF/AliRDHFCuts.h b/PWGHF/vertexingHF/AliRDHFCuts.h
index 443efc95d043bb2ab9cf64104643117ee8024735..a0bbdca9357128a08bab2b2ddf58cb854ed8cce8 100644 (file)
--- a/PWGHF/vertexingHF/AliRDHFCuts.h
+++ b/PWGHF/vertexingHF/AliRDHFCuts.h
@@ -100,8 +100,9 @@ class AliRDHFCuts : public AliAnalysisCuts
     fTriggerMask=(AliVEvent::kEMCEJE|AliVEvent::kEMCEGA);
     fUseOnlyOneTrigger=kTRUE;
   }
-
-  void SetTriggerClass(TString trclass) {fTriggerClass=trclass;} 
+  void SetRemoveTrackletOutliers(Bool_t opt) {fRemoveTrackletOutliers=opt;}
+  void SetTriggerClass(TString trclass0, TString trclass1="") {fTriggerClass[0]=trclass0; fTriggerClass[1]=trclass1;} 
+  void ApplySPDDeadPbPb2011(){fApplySPDDeadPbPb2011=kTRUE;}
   void SetVarsForOpt(Int_t nVars,Bool_t *forOpt);
   void SetGlobalIndex(){fGlobalIndex=fnVars*fnPtBins;}
   void SetGlobalIndex(Int_t nVars,Int_t nptBins){fnVars=nVars; fnPtBins=nptBins; SetGlobalIndex();}
@@ -113,12 +114,14 @@ class AliRDHFCuts : public AliAnalysisCuts
           {delete fTrackCuts; fTrackCuts=new AliESDtrackCuts(*cuts); return;}
   void SetUsePID(Bool_t flag=kTRUE) {fUsePID=flag; return;}
   void SetUseAOD049(Bool_t flag=kTRUE) {fUseAOD049=flag; return;}
+  void SetKinkRejection(Bool_t flag=kTRUE) {fKinkReject=flag; return;}
   void SetUseCentrality(Int_t flag=1);    // see enum below
   void SetPidHF(AliAODPidHF* pidObj) {
     if(fPidHF) delete fPidHF;
     fPidHF=new AliAODPidHF(*pidObj);
   }
   void SetRemoveDaughtersFromPrim(Bool_t removeDaughtersPrim) {fRemoveDaughtersFromPrimary=removeDaughtersPrim;}
+  void SetRecomputePrimaryVertex(Bool_t opt) {fRecomputePrimVertex=opt;}
   void SetMinPtCandidate(Double_t ptCand=-1.) {fMinPtCand=ptCand; return;}
   void SetMaxPtCandidate(Double_t ptCand=1000.) {fMaxPtCand=ptCand; return;}
   void SetOptPileup(Int_t opt=0){
@@ -154,6 +157,7 @@ class AliRDHFCuts : public AliAnalysisCuts
   void    GetVarPtIndex(Int_t iGlob, Int_t& iVar, Int_t& iPtBin) const;
   Bool_t  GetIsUsePID() const {return fUsePID;}
   Bool_t  GetUseAOD049() const {return fUseAOD049;}
+  Bool_t  GetUseKinkRejection() const {return fKinkReject;}
   Bool_t  GetIsPrimaryWithoutDaughters() const {return fRemoveDaughtersFromPrimary;}
   Bool_t GetOptPileUp() const {return fOptPileup;}
   Int_t GetUseCentrality() const {return fUseCentrality;}
@@ -241,6 +245,7 @@ class AliRDHFCuts : public AliAnalysisCuts
   void SetNVars(Int_t nVars){fnVars=nVars;}
 
   Bool_t IsSignalMC(AliAODRecoDecay *d,AliAODEvent *aod,Int_t pdg) const;
+  Bool_t RecomputePrimaryVertex(AliAODEvent* event) const;
 
   // cuts on the event
   Int_t fMinVtxType; // 0: not cut; 1: SPDZ; 2: SPD3D; 3: Tracks
@@ -250,7 +255,7 @@ class AliRDHFCuts : public AliAnalysisCuts
   Int_t fMinSPDMultiplicity; // SPD multiplicity
   ULong64_t fTriggerMask; // trigger mask
   Bool_t fUseOnlyOneTrigger; // flag to select one trigger only
-  TString  fTriggerClass; // trigger class
+  TString  fTriggerClass[2]; // trigger class
   // quality cuts on the daughter tracks
   AliESDtrackCuts *fTrackCuts; // tracks for daughter tracks (AOD converted to ESD on the flight!)
   // cuts on the candidate
@@ -270,6 +275,7 @@ class AliRDHFCuts : public AliAnalysisCuts
   Int_t fWhyRejection; // used to code the step at which candidate was rejected
   UInt_t fEvRejectionBits; //bit map storing the full info about event rejection
   Bool_t fRemoveDaughtersFromPrimary; // flag to switch on the removal of duaghters from the primary vertex computation
+  Bool_t fRecomputePrimVertex; // flag to recompute primary vertex
   Bool_t fUseMCVertex; // use MC primary vertex 
   Bool_t fUsePhysicsSelection; // use Physics selection criteria
   Int_t  fOptPileup;      // option for pielup selection
@@ -290,8 +296,11 @@ class AliRDHFCuts : public AliAnalysisCuts
   Bool_t  fKeepSignalMC; // IsSelected returns always kTRUE for MC signal
   Bool_t fIsCandTrackSPDFirst; // flag to select the track kFirst criteria for pt < ptlimit
   Double_t fMaxPtCandTrackSPDFirst; // maximum pt of the candidate for which to check if the daughters fulfill kFirst criteria
+  Bool_t fApplySPDDeadPbPb2011;  // flag to apply SPD dead module map of PbPb2011
+  Bool_t fRemoveTrackletOutliers; // flag to apply cut on tracklets vs. centrality for 2011 data
+  Bool_t fKinkReject; // flag to reject kink daughters
 
-  ClassDef(AliRDHFCuts,21);  // base class for cuts on AOD reconstructed heavy-flavour decays
+  ClassDef(AliRDHFCuts,24);  // base class for cuts on AOD reconstructed heavy-flavour decays
 };
 
 #endif