PWGDQ/dielectron/macrosLMEE/AddTask_cbaumann_LMEEPbPb2011.C

   1 AliAnalysisTask *AddTask_cbaumann_LMEEPbPb2011(Bool_t runRejection=kFALSE, Bool_t setMC=kFALSE,Bool_t enableCF=kFALSE){
   2   //get the current analysis manager
   3   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
   4   if (!mgr) {
   5     Error("AddTask_cbaumann_LMEEPbPb2011", "No analysis manager found.");
   6     return 0;
   7   }
   8
   9   //create config File names: TRAIN_ROOT is for running on GSI train,
  10   // ALICE_ROOT for CERN Lego trains
  11   TString configBasePath("$TRAIN_ROOT/cbaumann_dielectron/");
  12   TString trainRoot=gSystem->Getenv("TRAIN_ROOT");
  13   if (trainRoot.IsNull()) configBasePath= "$ALICE_ROOT/PWGDQ/dielectron/macrosLMEE/";
  14   TString configFile("ConfigLMEEPbPb2011.C");
  15   TString configLMEECutLib("LMEECutLib.C");
  16
  17   TString configFilePath(configBasePath+configFile);
  18   TString configLMEECutLib(configBasePath+configLMEECutLib);
  19
  20   //AOD Usage currently tested with separate task, to be merged
  21   if (mgr->GetInputEventHandler()->IsA()==AliAODInputHandler::Class()){
  22     ::Info("AddTaskLMEEPbPb2011", "no dedicated AOD configuration");
  23   }
  24
  25   //Do we have an MC handler?
  26   Bool_t hasMC=setMC;
  27   if (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler()!=0x0)
  28     hasMC=kTRUE;
  29
  30
  31
  32   if (!gROOT->GetListOfGlobalFunctions()->FindObject(configFile.Data()))
  33     gROOT->LoadMacro(configFilePath.Data());
  34   if (!gROOT->GetListOfGlobalFunctions()->FindObject(configLMEECutLib.Data()))
  35     gROOT->LoadMacro(configLMEECutLibPath.Data());
  36
  37   LMEECutLib* cutlib = new LMEECutLib();
  38   AliAnalysisTaskMultiDielectron *task=new AliAnalysisTaskMultiDielectron("MultiDiEData");
  39   if (!hasMC){ task->UsePhysicsSelection();
  40   }
  41   task->SetTriggerMask(AliVEvent::kMB+AliVEvent::kCentral+AliVEvent::kSemiCentral);
  42   task->SetEventFilter(cutlib->GetEventCuts(LMEECutLib::kPbPb2011TPCandTOF)); //
  43
  44   //load dielectron configuration file
  45
  46   //add dielectron analysis with different cuts to the task
  47   if (runRejection) {
  48
  49     AliDielectron *lowmass4=ConfigLMEEPbPb2011(4,hasMC,enableCF);
  50     task->AddDielectron(lowmass4);
  51     printf("add: %s\n",lowmass4->GetName());
  52   }
  53   else {
  54
  55     AliDielectron *lowmass3=ConfigLMEEPbPb2011(3,hasMC,enableCF);
  56     task->AddDielectron(lowmass3);
  57     printf("add: %s\n",lowmass3->GetName())
  58       AliDielectron *lowmass1=ConfigLMEEPbPb2011(1,hasMC,enableCF);
  59     task->AddDielectron(lowmass1);
  60     printf("add: %s\n",lowmass1->GetName())
  61       AliDielectron *lowmass7=ConfigLMEEPbPb2011(7,hasMC,enableCF);
  62     task->AddDielectron(lowmass7);
  63     printf("add: %s\n",lowmass7->GetName())
  64   }
  65
  66   mgr->AddTask(task);
  67
  68   //create output container
  69   AliAnalysisDataContainer *coutput1 =
  70     mgr->CreateContainer("cbaumann_LMEEPbPb2011_tree",
  71         TTree::Class(),
  72         AliAnalysisManager::kExchangeContainer,
  73         "cbaumann_LMEEPbPb2011_default.root");
  74
  75   AliAnalysisDataContainer *cOutputHist1 =
  76     mgr->CreateContainer("cbaumann_LMEEPbPb2011_out",
  77         TList::Class(),
  78         AliAnalysisManager::kOutputContainer,
  79         "cbaumann_LMEEPbPb2011_out.root");
  80   if (enableCF) {
  81     AliAnalysisDataContainer *cOutputHist2 =
  82       mgr->CreateContainer("cbaumann_LMEEPbPb2011_CF",
  83           TList::Class(),
  84           AliAnalysisManager::kOutputContainer,
  85           "cbaumann_LMEEPbPb2011_out.root");
  86
  87   }
  88   AliAnalysisDataContainer *cOutputHist3 =
  89     mgr->CreateContainer("cbaumann_EventStatPbPb2011",
  90         TH1D::Class(),
  91         AliAnalysisManager::kOutputContainer,
  92         "cbaumann_LMEEPbPb2011_out.root");
  93
  94
  95   mgr->ConnectInput(task,  0, mgr->GetCommonInputContainer());
  96   mgr->ConnectOutput(task, 0, coutput1 );
  97   mgr->ConnectOutput(task, 1, cOutputHist1);
  98   if (enableCF) {
  99     mgr->ConnectOutput(task, 2, cOutputHist2);
 100   }
 101   mgr->ConnectOutput(task, 3, cOutputHist3);
 102
 103   return task;
 104 }