PWG/EMCAL/macros/AddTaskEmcalPicoTrackMaker.C

   1 // $Id$
   2
   3 AliEmcalPicoTrackMaker* AddTaskEmcalPicoTrackMaker(
   4   const char *name         = "PicoTracks",
   5   const char *inname       = "tracks",
   6   const char *runperiod    = "",
   7   Bool_t      includeNoITS = kTRUE,
   8   Double_t ptmin           = 0,
   9   Double_t ptmax           = 1000,
  10   Double_t etamin          = -10,
  11   Double_t etamax          = +10,
  12   Double_t phimin          = -10,
  13   Double_t phimax          = +10,
  14   AliESDtrackCuts *cuts    = 0
  15 )
  16 {
  17   // Get the pointer to the existing analysis manager via the static access method.
  18   //==============================================================================
  19   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  20   if (!mgr)
  21   {
  22     ::Error("AddTaskEmcalPicoTrackMaker", "No analysis manager to connect to.");
  23     return NULL;
  24   }
  25
  26   // Check the analysis type using the event handlers connected to the analysis manager.
  27   //==============================================================================
  28   if (!mgr->GetInputEventHandler())
  29   {
  30     ::Error("AddTaskEmcalPicoTrackMaker", "This task requires an input event handler");
  31     return NULL;
  32   }
  33
  34   //-------------------------------------------------------
  35   // Init the task and do settings
  36   //-------------------------------------------------------
  37
  38   AliEmcalPicoTrackMaker *eTask = new AliEmcalPicoTrackMaker();
  39   eTask->SetTracksOutName(name);
  40   eTask->SetTracksInName(inname);
  41   eTask->SetIncludeNoITS(includeNoITS);
  42   eTask->SetTrackPtLimits(ptmin, ptmax);
  43   eTask->SetTrackEtaLimits(etamin, etamax);
  44   eTask->SetTrackPhiLimits(phimin, phimax);
  45
  46   TString runPeriod(runperiod);
  47   runPeriod.ToLower();
  48   if (runPeriod == "lhc11h") {
  49     eTask->SetAODfilterBits(256,512); // hybrid tracks for LHC11h
  50     eTask->SetMC(kFALSE);
  51   } else if (runPeriod == "lhc11a") {
  52     eTask->SetAODfilterBits(256,16); // hybrid tracks for LHC11a
  53     eTask->SetMC(kFALSE);
  54   } else if (runPeriod == "lhc12a15a" || runPeriod == "lhc12a15e") {
  55     eTask->SetAODfilterBits(256,16); // hybrid tracks for LHC12a15a and LHC12a15e
  56     eTask->SetMC(kTRUE);
  57   } else if (runPeriod.Contains(":")) {
  58     TObjArray *arr = runPeriod.Tokenize(":");
  59     TString arg1(arr->At(0)->GetName());
  60     TString arg2("-1");
  61     if (arr->GetEntries()>1)
  62       arg2 = arr->At(1)->GetName();
  63     eTask->SetAODfilterBits(arg1.Atoi(),arg2.Atoi());
  64     delete arr;
  65   } else {
  66     if (runPeriod.IsNull())
  67       ::Warning("Run period %s not known. It will use IsHybridGlobalConstrainedGlobal.");
  68   }
  69   eTask->SetESDtrackCuts(cuts);
  70
  71   //-------------------------------------------------------
  72   // Final settings, pass to manager and set the containers
  73   //-------------------------------------------------------
  74   mgr->AddTask(eTask);
  75
  76   // Create containers for input/output
  77   AliAnalysisDataContainer *cinput1  = mgr->GetCommonInputContainer();
  78   mgr->ConnectInput  (eTask, 0,  cinput1 );
  79
  80   return eTask;
  81 }