Interface to define the run pass in AliMuonTrackCut (ZHANG Xiaoming)

diff --git a/PWG/FLOW/Tasks/AliFlowTrackCuts.cxx b/PWG/FLOW/Tasks/AliFlowTrackCuts.cxx
index dec4156b207243f07673efb406ebda3da26e3754..565168fdb61dba811af2c7cf8a1e5c302f707ff3 100644 (file)
--- a/PWG/FLOW/Tasks/AliFlowTrackCuts.cxx
+++ b/PWG/FLOW/Tasks/AliFlowTrackCuts.cxx
@@ -1203,13 +1203,14 @@ AliFlowTrackCuts* AliFlowTrackCuts::GetStandardITSTPCTrackCuts2009(Bool_t selPri
 }
 
 //-----------------------------------------------------------------------------
-AliFlowTrackCuts* AliFlowTrackCuts::GetStandardMuonTrackCuts(Bool_t isMC)
+AliFlowTrackCuts* AliFlowTrackCuts::GetStandardMuonTrackCuts(Bool_t isMC, Int_t passN)
 {
 // XZhang 20120604
   AliFlowTrackCuts* cuts = new AliFlowTrackCuts("standard muon track cuts");
   cuts->SetParamType(kMUON);
   cuts->SetStandardMuonTrackCuts();
   cuts->SetIsMuonMC(isMC);
+  cuts->SetMuonPassNumber(passN);
   return cuts;
 }
 

diff --git a/PWG/FLOW/Tasks/AliFlowTrackCuts.h b/PWG/FLOW/Tasks/AliFlowTrackCuts.h
index 5273c5955d10cd9c9034480cdfcf4e3ead5c4150..7362fb39809a49e2319e3f0e082dcaa5e439d31f 100644 (file)
--- a/PWG/FLOW/Tasks/AliFlowTrackCuts.h
+++ b/PWG/FLOW/Tasks/AliFlowTrackCuts.h
@@ -48,7 +48,7 @@ class AliFlowTrackCuts : public AliFlowTrackSimpleCuts {
   static AliFlowTrackCuts* GetStandardGlobalTrackCuts2010();
   static AliFlowTrackCuts* GetStandardITSTPCTrackCuts2009(Bool_t selPrimaries=kTRUE);
   static AliFlowTrackCuts* GetStandardVZEROOnlyTrackCuts();
-  static AliFlowTrackCuts* GetStandardMuonTrackCuts(Bool_t isMC=kFALSE);  // XZhang 20120604
+  static AliFlowTrackCuts* GetStandardMuonTrackCuts(Bool_t isMC=kFALSE, Int_t passN=2);  // XZhang 20120604
 
   Int_t Count(AliVEvent* event=NULL);
 
@@ -114,9 +114,11 @@ class AliFlowTrackCuts : public AliFlowTrackSimpleCuts {
   void SetPriors(Float_t centr = 0); // set my favourite priors for Bayesian PID (requested if Bayesian PID is used)
   void SetFlowTagType(AliFlowTrackSimple::tagType t) {fFlowTagType=t;}
 
-  void SetStandardMuonTrackCuts() { InitMuonCuts();     fMuonTrackCuts->SetDefaultFilterMask(); return; }  // XZhang 20120604
-  void SetIsMuonMC(Bool_t isMC)   { InitMuonCuts();     fMuonTrackCuts->SetIsMC(isMC);          return; }  // XZhang 20120604
-  void SetRunsMuon( const AliInputEventHandler* eventHandler )   { if (fMuonTrackCuts) fMuonTrackCuts->SetRun(eventHandler); }  // XZhang 20120604
+  AliMuonTrackCuts *GetMuonTrackCuts() { InitMuonCuts(); return fMuonTrackCuts; }                           // XZhang 20121014
+  void SetStandardMuonTrackCuts()      { InitMuonCuts(); fMuonTrackCuts->SetDefaultFilterMask(); return; }  // XZhang 20120604
+  void SetIsMuonMC(Bool_t isMC)        { InitMuonCuts(); fMuonTrackCuts->SetIsMC(isMC);          return; }  // XZhang 20120604
+  void SetMuonPassNumber(Int_t passN)  { InitMuonCuts(); fMuonTrackCuts->SetPassNumber(passN);   return; }  // XZhang 20121013
+  void SetRunsMuon(const AliInputEventHandler* eventHandler) { if (fMuonTrackCuts) fMuonTrackCuts->SetRun(eventHandler); }  // XZhang 20120604
 
   Int_t GetMinNClustersTPC() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetMinNClusterTPC();}
   Int_t GetMinNClustersITS() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetMinNClustersITS();}

diff --git a/PWG/muon/AddTaskMuonsHF.C b/PWG/muon/AddTaskMuonsHF.C
index 15c19e35b69ae816c43cfa6d4082202f2a93bf82..62c851068ee2d87b275cad5dca154df675e0bef8 100644 (file)
--- a/PWG/muon/AddTaskMuonsHF.C
+++ b/PWG/muon/AddTaskMuonsHF.C
@@ -1,4 +1,4 @@
-AliAnalysisTaskSEMuonsHF* AddTaskMuonsHF(Int_t mode=0, Bool_t isMC=kFALSE, Bool_t isTree=kFALSE, Bool_t isSel)
+AliAnalysisTaskSEMuonsHF* AddTaskMuonsHF(Int_t mode=0, Int_t passN=2, Bool_t isMC=kFALSE, Bool_t isTree=kFALSE, Bool_t isSel=kTRUE)
 {
   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
   if (!mgr) {
@@ -31,8 +31,8 @@ AliAnalysisTaskSEMuonsHF* AddTaskMuonsHF(Int_t mode=0, Bool_t isMC=kFALSE, Bool_
                           999999.0,   // up limit of vt
                          -999999.0,   // centrality min
                           999999.0};  // centrality max
-  AliMuonTrackCuts *cutsMuon = new AliMuonTrackCuts("cutsMuon", "cutsMuon"); cutsMuon->SetIsMC(isMC);
-  AliMuonPairCuts  *cutsDimu = new  AliMuonPairCuts("cutsDimu", "cutsDimu"); cutsDimu->SetIsMC(isMC);
+  AliMuonTrackCuts *cutsMuon = new AliMuonTrackCuts("cutsMuon", "cutsMuon"); cutsMuon->SetIsMC(isMC); cutsMuon->SetPassNumber(passN);
+  AliMuonPairCuts  *cutsDimu = new  AliMuonPairCuts("cutsDimu", "cutsDimu"); cutsDimu->SetIsMC(isMC); cutsDimu->GetMuonTrackCuts()->SetPassNumber(passN);
   AliAnalysisTaskSEMuonsHF *taskMuonsHF = new AliAnalysisTaskSEMuonsHF("MuonsHF Analysis Task",*cutsMuon,*cutsDimu);
   taskMuonsHF->SetAnaMode(mode);
   taskMuonsHF->SetUseMC(isMC);