PWGDQ/dielectron/macrosLMEE/ESDonlyTasks/AddTask_cbaumann_LMEEPbPb2011Cent.C

   1 AliAnalysisTask *AddTask_cbaumann_LMEEPbPb2011Cent(Bool_t runRejection=kFALSE, Bool_t setMC=kFALSE,Bool_t enableCF=kFALSE, Bool_t switchToPhiV=kTRUE,Bool_t switchToOA=kFALSE, Bool_t getFromAlien=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   // 3 options:
  10   // TRAIN_ROOT is for running on GSI train,
  11   // ALICE_ROOT for CERN Lego trains
  12   // getFromAlien: load files from user ALIEN dir
  13   TString configBasePath("$TRAIN_ROOT/cbaumann_dielectron/");
  14   TString trainRoot=gSystem->Getenv("TRAIN_ROOT");
  15   if (trainRoot.IsNull()) configBasePath= "$ALICE_ROOT/PWGDQ/dielectron/macrosLMEE/";
  16
  17   if (getFromAlien &&
  18       (!gSystem->Exec("alien_cp alien:///alice/cern.ch/user/c/cbaumann/PWGDQ/dielectron/macrosLMEE/ConfigLMEEPbPb2011.C .")) &&
  19       (!gSystem->Exec("alien_cp alien:///alice/cern.ch/user/c/cbaumann/PWGDQ/dielectron/macrosLMEE/LMEECutLib.C ."))
  20      ) {
  21         configBasePath=Form("%s/",gSystem->pwd());
  22   }
  23   TString configFile("ConfigLMEEPbPb2011.C");
  24   TString configLMEECutLib("LMEECutLib.C");
  25
  26   TString configFilePath(configBasePath+configFile);
  27   TString configLMEECutLibPath(configBasePath+configLMEECutLib);
  28
  29   //AOD Usage currently tested with separate task, to be merged
  30   if (mgr->GetInputEventHandler()->IsA()==AliAODInputHandler::Class()){
  31         ::Info("AddTaskLMEEPbPb2011", "no dedicated AOD configuration");
  32   }
  33
  34   //Do we have an MC handler?
  35   Bool_t hasMC=setMC;
  36   if (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler()!=0x0)
  37         hasMC=kTRUE;
  38
  39
  40
  41   if (!gROOT->GetListOfGlobalFunctions()->FindObject(configLMEECutLib.Data()))
  42         gROOT->LoadMacro(configLMEECutLibPath.Data());
  43   if (!gROOT->GetListOfGlobalFunctions()->FindObject(configFile.Data()))
  44         gROOT->LoadMacro(configFilePath.Data());
  45
  46   LMEECutLib* cutlib = new LMEECutLib();
  47   cutlib->SetMCFlag(hasMC);
  48   AliAnalysisTaskMultiDielectron *task=new AliAnalysisTaskMultiDielectron("MultiDiEDataPbPb2011Cent");
  49   if (!hasMC){ task->UsePhysicsSelection();
  50   }
  51   task->SetTriggerMask(AliVEvent::kMB+AliVEvent::kCentral+AliVEvent::kSemiCentral);
  52   task->SetEventFilter(cutlib->GetEventCuts(LMEECutLib::kPbPb2011TPCandTOF)); //
  53
  54   //load dielectron configuration file
  55
  56   //add dielectron analysis with different cuts to the task
  57   if (runRejection) {
  58
  59         AliDielectron *lowmass4=ConfigLMEEPbPb2011(4,hasMC,enableCF);
  60         task->AddDielectron(lowmass4);
  61         printf("add: %s\n",lowmass4->GetName());
  62   }
  63   else {
  64         if (switchToPhiV) {
  65
  66    AliDielectron *lowmass9=ConfigLMEEPbPb2011(9,hasMC,enableCF);
  67           lowmass9->SetUseKF(kFALSE);
  68           task->AddDielectron(lowmass9);
  69           printf("add: %s\n",lowmass9->GetName()); }
  70
  71         else if (switchToOA) {
  72
  73    AliDielectron *lowmass12=ConfigLMEEPbPb2011(12,hasMC,enableCF);
  74           lowmass12->SetUseKF(kFALSE);
  75           task->AddDielectron(lowmass12);
  76           printf("add: %s\n",lowmass12->GetName()); }
  77
  78         else {
  79           AliDielectron *lowmass1=ConfigLMEEPbPb2011(1,hasMC,enableCF);
  80           lowmass1->SetUseKF(kFALSE);
  81           task->AddDielectron(lowmass1);
  82           printf("add: %s\n",lowmass1->GetName());
  83         }
  84         /*      AliDielectron *lowmass7=ConfigLMEEPbPb2011(7,hasMC,enableCF);
  85                         task->AddDielectron(lowmass7);
  86                         printf("add: %s\n",lowmass7->GetName());*/
  87   }
  88
  89   mgr->AddTask(task);
  90
  91   //create output container
  92   AliAnalysisDataContainer *coutput1 =
  93         mgr->CreateContainer("cbaumann_LMEEPbPb2011Cent_tree",
  94                 TTree::Class(),
  95                 AliAnalysisManager::kExchangeContainer,
  96                 "cbaumann_LMEEPbPb2011Cent_default.root");
  97
  98   AliAnalysisDataContainer *cOutputHist1 =
  99         mgr->CreateContainer("cbaumann_LMEEPbPb2011Cent_out",
 100                 TList::Class(),
 101                 AliAnalysisManager::kOutputContainer,
 102                 "cbaumann_LMEEPbPb2011Cent_out.root");
 103
 104   AliAnalysisDataContainer *cOutputHist2 = 0x0;
 105   if (enableCF) {
 106         cOutputHist2 =
 107           mgr->CreateContainer("cbaumann_LMEEPbPb2011Cent_CF",
 108                   TList::Class(),
 109                   AliAnalysisManager::kOutputContainer,
 110                   "cbaumann_LMEEPbPb2011Cent_out.root");
 111
 112   }
 113   AliAnalysisDataContainer *cOutputHist3 =
 114         mgr->CreateContainer("cbaumann_EventStatPbPb2011",
 115                 TH1D::Class(),
 116                 AliAnalysisManager::kOutputContainer,
 117                 "cbaumann_LMEEPbPb2011Cent_out.root");
 118
 119
 120   mgr->ConnectInput(task,  0, mgr->GetCommonInputContainer());
 121   mgr->ConnectOutput(task, 0, coutput1 );
 122   mgr->ConnectOutput(task, 1, cOutputHist1);
 123   if (enableCF) {
 124         mgr->ConnectOutput(task, 2, cOutputHist2);
 125   }
 126   mgr->ConnectOutput(task, 3, cOutputHist3);
 127
 128   return task;
 129 }