PWGGA/EMCALJetTasks/macros/AddTaskSAJF.C

   1 // $Id$
   2
   3 AliAnalysisTaskSAJF* AddTaskSAJF(
   4   const char *ntracks            = "Tracks",
   5   const char *nclusters          = "CaloClusters",
   6   const char *njets              = "Jets",
   7   const char *nembjets           = "EmbJets",
   8   const char *nrandtracks        = "TracksRandomized",
   9   const char *nrandclusters      = "CaloClustersRandomized",
  10   const char *nrho               = "Rho",
  11   Double_t    jetradius          = 0.4,
  12   Double_t    jetptcut           = 1,
  13   Double_t    jetareacut         = 0.4,
  14   Double_t    ptcut              = 0.15,
  15   Double_t    jetBiasTrack       = 5,
  16   Double_t    jetBiasClus        = 5,
  17   UInt_t      type               = AliAnalysisTaskEmcal::kTPC,
  18   const char *taskname           = "AliAnalysisTaskSAJF"
  19 )
  20 {
  21   // Get the pointer to the existing analysis manager via the static access method.
  22   //==============================================================================
  23   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  24   if (!mgr)
  25   {
  26     ::Error("AddTaskSAJF", "No analysis manager to connect to.");
  27     return NULL;
  28   }
  29
  30   // Check the analysis type using the event handlers connected to the analysis manager.
  31   //==============================================================================
  32   if (!mgr->GetInputEventHandler())
  33   {
  34     ::Error("AddTaskSAJF", "This task requires an input event handler");
  35     return NULL;
  36   }
  37
  38   //-------------------------------------------------------
  39   // Init the task and do settings
  40   //-------------------------------------------------------
  41
  42   TString name(taskname);
  43   name += "_";
  44   name += njets;
  45   name += "_";
  46   name += nrho;
  47   name += "_Track";
  48   name += jetBiasTrack;
  49   name += "_Clus";
  50   name += jetBiasClus;
  51   name += "_R0";
  52   name += floor(jetradius*10+0.5);
  53   name += "_PtCut";
  54   name += floor(ptcut*1000+0.5);
  55   name += "_";
  56   if (type == AliAnalysisTaskEmcal::kTPC)
  57     name += "TPC";
  58   else if (type == AliAnalysisTaskEmcal::kEMCAL)
  59     name += "EMCAL";
  60   else if (type == AliAnalysisTaskEmcal::kTPCSmall)
  61     name += "TPCSmall";
  62   AliAnalysisTaskSAJF* jetTask = new AliAnalysisTaskSAJF(name);
  63   jetTask->SetAnaType(type);
  64   jetTask->SetTracksName(ntracks);
  65   jetTask->SetClusName(nclusters);
  66   jetTask->SetJetsName(njets);
  67   jetTask->SetEmbJetsName(nembjets);
  68   jetTask->SetRandTracksName(nrandtracks);
  69   jetTask->SetRandClusName(nrandclusters);
  70   jetTask->SetRhoName(nrho);
  71   jetTask->SetPtCut(ptcut);
  72   jetTask->SetJetRadius(jetradius);
  73   jetTask->SetJetPtCut(jetptcut);
  74   jetTask->SetJetAreaCut(jetareacut);
  75   jetTask->SetPtBiasJetTrack(jetBiasTrack);
  76   jetTask->SetPtBiasJetClus(jetBiasClus);
  77
  78   //-------------------------------------------------------
  79   // Final settings, pass to manager and set the containers
  80   //-------------------------------------------------------
  81
  82   mgr->AddTask(jetTask);
  83
  84   // Create containers for input/output
  85   AliAnalysisDataContainer *cinput1  = mgr->GetCommonInputContainer()  ;
  86   TString contname(name);
  87   contname += "_histos";
  88   AliAnalysisDataContainer *coutput1 = mgr->CreateContainer(contname.Data(),
  89                                                             TList::Class(),AliAnalysisManager::kOutputContainer,
  90                                                             Form("%s", AliAnalysisManager::GetCommonFileName()));
  91   mgr->ConnectInput  (jetTask, 0,  cinput1 );
  92   mgr->ConnectOutput (jetTask, 1, coutput1 );
  93
  94   return jetTask;
  95 }