PWGHF/hfe/macros/AddTaskHFECal.C

   1 AliAnalysisTask *AddTaskHFECal()
   2 {
   3   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
   4   if (!mgr) {
   5     Error("AddTaskHFEECal", "No analysis manager found.");
   6     return NULL;
   7   }
   8
   9   if (!mgr->GetInputEventHandler()) {
  10     ::Error("AddTaskHFECal", "This task requires an input event handler");
  11     return NULL;
  12   }
  13   TString type = mgr->GetInputEventHandler()->GetDataType(); // can be "ESD" or "AOD"
  14   if (type=="AOD"){
  15     ::Error("AddTaskHFECal", "The tasks exits because AODs are in input");
  16     return NULL;
  17   }
  18   Bool_t MCthere=kFALSE;
  19   AliMCEventHandler *mcH = dynamic_cast<AliMCEventHandler*>(mgr->GetMCtruthEventHandler());
  20   if(!mcH){
  21     MCthere=kFALSE;
  22   }else{
  23     MCthere=kTRUE;
  24   }
  25
  26
  27   //analysis task
  28   gROOT->LoadMacro("$ALICE_ROOT/PWGHF/hfe/macros/configs/PbPb/ConfigHFECal.C");
  29
  30   AliAnalysisTaskHFECal *hfetaskCent = ConfigHFECal(MCthere);
  31   AliAnalysisTaskHFECal *hfetaskTrig= ConfigHFECal(MCthere);
  32
  33   mgr->AddTask(hfetaskCent);
  34   mgr->AddTask(hfetaskTrig);
  35
  36   // Semi-central trigger
  37   hfetaskCent->SelectCollisionCandidates(AliVEvent::kCentral);
  38
  39   TString containerName = mgr->GetCommonFileName();
  40   containerName += ":PWGHF_hfeCalCentral";
  41
  42   AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
  43   AliAnalysisDataContainer *coutput1 = mgr->CreateContainer("HFE_Results_EMCalCentral", TList::Class(),AliAnalysisManager::kOutputContainer, containerName.Data());
  44   mgr->ConnectInput(hfetaskCent, 0, cinput);
  45   mgr->ConnectOutput(hfetaskCent, 1, coutput1);
  46
  47   //L1 gamma trigger
  48   hfetaskTrig->SelectCollisionCandidates(AliVEvent::kEMCEGA);
  49
  50   TString containerName2 = mgr->GetCommonFileName();
  51   containerName2 += ":PWGHF_hfeCalTrigEGA";
  52
  53   AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
  54   AliAnalysisDataContainer *coutput1 = mgr->CreateContainer("HFE_Results_EMCalTrigEGA", TList::Class(),AliAnalysisManager::kOutputContainer, containerName2.Data());
  55   mgr->ConnectInput(hfetaskTrig, 0, cinput);
  56   mgr->ConnectOutput(hfetaskTrig, 1, coutput1);
  57
  58   if(MCthere)
  59     {
  60     //MB trigger
  61     AliAnalysisTaskHFECal *hfetaskMB = ConfigHFECal(MCthere);
  62     mgr->AddTask(hfetaskMB);
  63     hfetaskMB->SelectCollisionCandidates(AliVEvent::kMB);
  64
  65     TString containerName3 = mgr->GetCommonFileName();
  66     containerName3 += ":PWGHF_hfeCalkMB";
  67
  68      AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
  69      AliAnalysisDataContainer *coutput1 = mgr->CreateContainer("HFE_Results_EMCalMB", TList::Class(),AliAnalysisManager::kOutputContainer, containerName3.Data());
  70      mgr->ConnectInput(hfetaskMB, 0, cinput);
  71      mgr->ConnectOutput(hfetaskMB, 1, coutput1);
  72      }
  73
  74
  75   return NULL;
  76 }