fixes from Laurent for the MC branch in the AOD filters

[u/mrichter/AliRoot.git] / PWG3 / muon / AliAnalysisTaskESDMuonFilter.h

diff --git a/PWG3/muon/AliAnalysisTaskESDMuonFilter.h b/PWG3/muon/AliAnalysisTaskESDMuonFilter.h
index a6af6aa8f63a6f78db72a8a40d8d9de7ac1e5048..96e1ded70d1fb6f80699c02300598e3607bf2531 100644 (file)
--- a/PWG3/muon/AliAnalysisTaskESDMuonFilter.h
+++ b/PWG3/muon/AliAnalysisTaskESDMuonFilter.h
@@ -23,8 +23,8 @@ class AliStack;
 class AliAnalysisTaskESDMuonFilter : public AliAnalysisTaskSE\r
 {\r
  public:\r
-    AliAnalysisTaskESDMuonFilter(Bool_t onlyMuon=kTRUE, Bool_t keepAllEvents=kTRUE);\r
-    AliAnalysisTaskESDMuonFilter(const char* name, Bool_t onlyMuon=kTRUE, Bool_t keepAllEvents=kTRUE);\r
+    AliAnalysisTaskESDMuonFilter(Bool_t onlyMuon=kTRUE, Bool_t keepAllEvents=kTRUE, Int_t mcMode=0);\r
+    AliAnalysisTaskESDMuonFilter(const char* name, Bool_t onlyMuon=kTRUE, Bool_t keepAllEvents=kTRUE, Int_t mcMode=0);\r
     virtual ~AliAnalysisTaskESDMuonFilter() {;}\r
     // Implementation of interface methods\r
     virtual void UserCreateOutputObjects();\r
@@ -40,6 +40,8 @@ class AliAnalysisTaskESDMuonFilter : public AliAnalysisTaskSE
     void SetWriteMuonAOD(Bool_t enableMuonAOD){fEnableMuonAOD = enableMuonAOD;}\r
     void SetWriteDimuonAOD(Bool_t enableDimuonAOD){fEnableDimuonAOD = enableDimuonAOD;}\r
 \r
+  void PrintTask(Option_t *option="", Int_t indent=0) const;\r
+\r
  private:\r
     AliAnalysisTaskESDMuonFilter(const AliAnalysisTaskESDMuonFilter&);\r
     AliAnalysisTaskESDMuonFilter& operator=(const AliAnalysisTaskESDMuonFilter&);\r
@@ -51,8 +53,9 @@ class AliAnalysisTaskESDMuonFilter : public AliAnalysisTaskSE
   Bool_t fEnableDimuonAOD; // flag for enabling Dimuon AOD production\r
   Bool_t fOnlyMuon; // flag for disabling branches irrelevant for (most) muon analyses\r
   Bool_t fKeepAllEvents; // keep even events where there's no muons (to get e.g. unbiased vertex distribution)\r
+  Int_t  fMCMode; // whether and how we're filtering MC data\r
   \r
-  ClassDef(AliAnalysisTaskESDMuonFilter, 3); // Analysis task for standard ESD filtering\r
+  ClassDef(AliAnalysisTaskESDMuonFilter, 5); // Analysis task for standard ESD filtering\r
 };\r
  \r
 class AliAnalysisNonMuonTrackCuts : public AliAnalysisCuts\r