From fef2d0757f48691d506e4f0d1aacb715c69dd24d Mon Sep 17 00:00:00 2001
From: prino <prino@f7af4fe6-9843-0410-8265-dc069ae4e863>
Date: Thu, 8 Dec 2011 15:28:39 +0000
Subject: [PATCH] Add the possibility to select the trigger type via the event
 selection mask

---
 PWG3/vertexingHF/AliRDHFCuts.cxx |  4 ++--
 PWG3/vertexingHF/AliRDHFCuts.h   | 14 ++++++++++++--
 2 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/PWG3/vertexingHF/AliRDHFCuts.cxx b/PWG3/vertexingHF/AliRDHFCuts.cxx
index c69edbc6f4a..16ab67362b9 100644
--- a/PWG3/vertexingHF/AliRDHFCuts.cxx
+++ b/PWG3/vertexingHF/AliRDHFCuts.cxx
@@ -53,7 +53,7 @@ fMinVtxContr(1),
 fMaxVtxRedChi2(1e6),
 fMaxVtxZ(10.),
 fMinSPDMultiplicity(0),
-fTriggerMask(0),
+fTriggerMask(AliVEvent::kAnyINT),
 fTriggerClass("CINT1"),
 fTrackCuts(0),
 fnPtBins(1),
@@ -308,7 +308,7 @@ Bool_t AliRDHFCuts::IsEventSelected(AliVEvent *event) {
 
   // physics selection requirements
   if(fUsePhysicsSelection){
-    Bool_t isSelected = (((AliInputEventHandler*)(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler()))->IsEventSelected() & AliVEvent::kMB);
+    Bool_t isSelected = (((AliInputEventHandler*)(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler()))->IsEventSelected() & fTriggerMask);
     if(!isSelected) {
       if(accept) fWhyRejection=7;
       fEvRejectionBits+=1<<kPhysicsSelection;
diff --git a/PWG3/vertexingHF/AliRDHFCuts.h b/PWG3/vertexingHF/AliRDHFCuts.h
index 7b0921b2071..961e7c00a16 100644
--- a/PWG3/vertexingHF/AliRDHFCuts.h
+++ b/PWG3/vertexingHF/AliRDHFCuts.h
@@ -48,7 +48,18 @@ class AliRDHFCuts : public AliAnalysisCuts
   void SetMaxVtxRdChi2(Float_t chi2=1e6) {fMaxVtxRedChi2=chi2;}  
   void SetMaxVtxZ(Float_t z=1e6) {fMaxVtxZ=z;}  
   void SetMinSPDMultiplicity(Int_t mult=0) {fMinSPDMultiplicity=mult;}  
-  void SetTriggerMask(ULong64_t mask=0) {fTriggerMask=mask;} 
+
+  void SetTriggerMask(ULong64_t mask=0) {fTriggerMask=mask;}
+  void SetUseAnyTrigger(){fTriggerMask=AliVEvent::kAny;}
+  void SetUseMBTrigger(){fTriggerMask|=AliVEvent::kMB;}
+  void SetUseMBTriggerOnly(){fTriggerMask=AliVEvent::kMB;}
+  void SetUseCentralTrigger(){fTriggerMask|=AliVEvent::kCentral;}
+  void SetUseCentralTriggerOnly(){fTriggerMask=AliVEvent::kCentral;}
+  void SetUseSemiCentralTrigger(){fTriggerMask|=AliVEvent::kSemiCentral;}
+  void SetUseSemiCentralTriggerOnly(){fTriggerMask=AliVEvent::kSemiCentral;}
+  void SetUseEMCALTrigger(){fTriggerMask|=(AliVEvent::kEMCEJE|AliVEvent::kEMCEGA);}
+  void SetUseEMCALTriggerOnly(){fTriggerMask=(AliVEvent::kEMCEJE|AliVEvent::kEMCEGA);}
+
   void SetTriggerClass(TString trclass) {fTriggerClass=trclass;} 
   void SetVarsForOpt(Int_t nVars,Bool_t *forOpt);
   void SetGlobalIndex(){fGlobalIndex=fnVars*fnPtBins;}
@@ -237,7 +248,6 @@ class AliRDHFCuts : public AliAnalysisCuts
   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
 
-
   ClassDef(AliRDHFCuts,20);  // base class for cuts on AOD reconstructed heavy-flavour decays
 };
 
-- 
2.43.0