PWG3/hfe/AddTaskHFE.C

   1 AliAnalysisTask *AddTaskHFE(){
   2   //get the current analysis manager
   3   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
   4   if (!mgr) {
   5     Error("AddTaskHFE", "No analysis manager found.");
   6     return NULL;
   7   }
   8   if (!mgr->GetInputEventHandler()) {
   9     ::Error("AddTaskHFE", "This task requires an input event handler");
  10     return NULL;
  11   }
  12   TString type = mgr->GetInputEventHandler()->GetDataType(); // can be "ESD" or "AOD"
  13   if (type=="AOD"){
  14     ::Error("AddTaskHFE", "The tasks exits because AODs are in input");
  15     return NULL;
  16   }
  17
  18   //============= Set Task Name ===================
  19   //TString taskName=("AliAnalysisTaskHFE.cxx+");
  20   //===============================================
  21
  22   AliHFEcuts *hfecuts = new AliHFEcuts;
  23   hfecuts->CreateStandardCuts();
  24   hfecuts->SetCutITSpixel(AliHFEextraCuts::kFirst);
  25   hfecuts->SetMinNTrackletsTRD(1);
  26   //hfecuts->SetCheckITSLayerStatus(kFALSE);
  27
  28   AliAnalysisTaskHFE *task = new AliAnalysisTaskHFE("Heavy Flavour Electron Analysis");
  29   task->SetPIDStrategy(4);
  30   task->SetHFECuts(hfecuts);
  31   task->SetQAOn(AliAnalysisTaskHFE::kMCqa);
  32   task->SetSecVtxOn();
  33   mgr->AddTask(task);
  34
  35   //----------------------
  36   //create data containers
  37   //----------------------
  38
  39   //find input container
  40   AliAnalysisDataContainer *cinput  = mgr->GetCommonInputContainer();
  41   TString containerName = mgr->GetCommonFileName();
  42   containerName += ":PWG3_hfe";
  43
  44   task->ConnectOutput(0, mgr->CreateContainer("nEvents", TH1I::Class(),
  45                                               AliAnalysisManager::kOutputContainer, containerName.Data()));
  46   task->ConnectOutput(1, mgr->CreateContainer("Results", TList::Class(),
  47                                               AliAnalysisManager::kOutputContainer, containerName.Data()));
  48   task->ConnectOutput(2, mgr->CreateContainer("QA", TList::Class(),
  49                                               AliAnalysisManager::kOutputContainer, containerName.Data()));
  50   mgr->ConnectInput  (task,  0, cinput );
  51
  52   return task;
  53 }