PWGPP/macros/AddTaskMTRchamberEfficiency.C

   1 #if !defined(__CINT__) || defined(__MAKECINT__)
   2 #include "TString.h"
   3 #include "TObjArray.h"
   4
   5 #include "AliLog.h"
   6 #include "AliVEventHandler.h"
   7
   8 #include "AliAnalysisManager.h"
   9 #include "AliAnalysisDataContainer.h"
  10
  11 #include "AliMuonTrackCuts.h"
  12 #include "AliAnalysisTaskTrigChEff.h"
  13 #endif
  14
  15 AliAnalysisTaskTrigChEff* AddTaskMTRchamberEfficiency(Bool_t useGhosts = kFALSE, Bool_t isMC = kFALSE)
  16 {
  17   //
  18   // Task for the determination of the MUON trigger chamber efficiency
  19   //
  20   // stocco@subatech.in2p3.fr
  21   //
  22
  23   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  24   if (!mgr) {
  25     ::Error("AddTaskMTRchamberEfficiency", "No analysis manager to connect to.");
  26     return NULL;
  27   }
  28
  29   TString type = mgr->GetInputEventHandler()->GetDataType();
  30   if (!type.Contains("ESD") && !type.Contains("AOD")) {
  31     ::Error("AddTaskMTRchamberEfficiency", "AliAnalysisTaskTrigChEff task needs the manager to have an ESD or AOD input handler.");
  32     return NULL;
  33   }
  34
  35   // Create cuts
  36   AliMuonTrackCuts* muonTrackCuts = new AliMuonTrackCuts("StdMuonTrackCuts", "StdMuonTrackCuts");
  37   muonTrackCuts->SetIsMC(isMC);
  38   muonTrackCuts->ApplySharpPtCutInMatching(kTRUE);
  39
  40   // Create task
  41   AliAnalysisTaskTrigChEff* taskTrigChEff = new AliAnalysisTaskTrigChEff("TriggerChamberEfficiency", *muonTrackCuts);
  42   taskTrigChEff->SetUseGhostTracks(useGhosts);
  43   if ( isMC ) taskTrigChEff->SetTrigClassPatterns("ANY");
  44   else {
  45     TString trigClassPatterns = taskTrigChEff->GetDefaultTrigClassPatterns();
  46     trigClassPatterns.Prepend("ANY,!CMUP,");
  47     taskTrigChEff->SetTrigClassPatterns(trigClassPatterns);
  48   }
  49   taskTrigChEff->GetMuonEventCuts()->SetFilterMask(AliMuonEventCuts::kSelectedTrig);
  50   taskTrigChEff->GetMuonTrackCuts()->SetAllowDefaultParams();
  51   mgr->AddTask(taskTrigChEff);
  52
  53   // Create container
  54   TString currName = "";
  55   TString outputfile = mgr->GetCommonFileName();
  56   if ( ! outputfile.IsNull() ) outputfile += ":MTR_ChamberEffMap";
  57   else outputfile = "TestTrigChEffAnalysis.root";
  58
  59   AliAnalysisDataContainer *coutput1 = mgr->CreateContainer("testMTRChamberEff",TObjArray::Class(),AliAnalysisManager::kOutputContainer,outputfile.Data());
  60   AliAnalysisDataContainer *coutput2 = mgr->CreateContainer("triggerChamberEff", TList::Class(), AliAnalysisManager::kOutputContainer,outputfile.Data());
  61
  62    // Connect containers
  63    mgr->ConnectInput  (taskTrigChEff,  0, mgr->GetCommonInputContainer());
  64    mgr->ConnectOutput (taskTrigChEff,  1, coutput1);
  65    mgr->ConnectOutput (taskTrigChEff,  2, coutput2);
  66
  67    return taskTrigChEff;
  68 }