PWGDQ/dielectron/macrosLMEE/AddTask_cbaumann_LMEEPbPb2011Cent.C

   1 AliAnalysisTask *AddTask_cbaumann_LMEEPbPb2011Cent(Bool_t runRejection=kFALSE, Bool_t setMC=kFALSE,Bool_t enableCF=kFALSE, Bool_t switchToPhiV=kFALSE,Bool_t switchToOA=kTRUE){
   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 configLMEECutLibPath(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(configLMEECutLib.Data()))
  33         gROOT->LoadMacro(configLMEECutLibPath.Data());
  34   if (!gROOT->GetListOfGlobalFunctions()->FindObject(configFile.Data()))
  35         gROOT->LoadMacro(configFilePath.Data());
  36
  37   LMEECutLib* cutlib = new LMEECutLib();
  38   cutlib->SetMCFlag(hasMC);
  39   AliAnalysisTaskMultiDielectron *task=new AliAnalysisTaskMultiDielectron("MultiDiEDataPbPb2011Cent");
  40   if (!hasMC){ task->UsePhysicsSelection();
  41   }
  42   task->SetTriggerMask(AliVEvent::kMB+AliVEvent::kCentral+AliVEvent::kSemiCentral);
  43   task->SetEventFilter(cutlib->GetEventCuts(LMEECutLib::kPbPb2011TPCandTOF)); //
  44
  45   //load dielectron configuration file
  46
  47   //add dielectron analysis with different cuts to the task
  48   if (runRejection) {
  49
  50         AliDielectron *lowmass4=ConfigLMEEPbPb2011(4,hasMC,enableCF);
  51         task->AddDielectron(lowmass4);
  52         printf("add: %s\n",lowmass4->GetName());
  53   }
  54   else {
  55         if (switchToPhiV) {
  56
  57    AliDielectron *lowmass9=ConfigLMEEPbPb2011(9,hasMC,enableCF);
  58           lowmass9->SetUseKF(kFALSE);
  59           task->AddDielectron(lowmass9);
  60           printf("add: %s\n",lowmass9->GetName()); }
  61         else if (switchToOA) {
  62
  63    AliDielectron *lowmass12=ConfigLMEEPbPb2011(12,hasMC,enableCF);
  64           lowmass12->SetUseKF(kFALSE);
  65           task->AddDielectron(lowmass12);
  66           printf("add: %s\n",lowmass12->GetName()); }
  67
  68         else {
  69           AliDielectron *lowmass1=ConfigLMEEPbPb2011(1,hasMC,enableCF);
  70           lowmass1->SetUseKF(kFALSE);
  71           task->AddDielectron(lowmass1);
  72           printf("add: %s\n",lowmass1->GetName());
  73         }
  74         /*      AliDielectron *lowmass7=ConfigLMEEPbPb2011(7,hasMC,enableCF);
  75                         task->AddDielectron(lowmass7);
  76                         printf("add: %s\n",lowmass7->GetName());*/
  77   }
  78
  79   mgr->AddTask(task);
  80
  81   //create output container
  82   AliAnalysisDataContainer *coutput1 =
  83         mgr->CreateContainer("cbaumann_LMEEPbPb2011Cent_tree",
  84                 TTree::Class(),
  85                 AliAnalysisManager::kExchangeContainer,
  86                 "cbaumann_LMEEPbPb2011Cent_default.root");
  87
  88   AliAnalysisDataContainer *cOutputHist1 =
  89         mgr->CreateContainer("cbaumann_LMEEPbPb2011Cent_out",
  90                 TList::Class(),
  91                 AliAnalysisManager::kOutputContainer,
  92                 "cbaumann_LMEEPbPb2011Cent_out.root");
  93
  94   AliAnalysisDataContainer *cOutputHist2 = 0x0;
  95   if (enableCF) {
  96         cOutputHist2 =
  97           mgr->CreateContainer("cbaumann_LMEEPbPb2011Cent_CF",
  98                   TList::Class(),
  99                   AliAnalysisManager::kOutputContainer,
 100                   "cbaumann_LMEEPbPb2011Cent_out.root");
 101
 102   }
 103   AliAnalysisDataContainer *cOutputHist3 =
 104         mgr->CreateContainer("cbaumann_EventStatPbPb2011",
 105                 TH1D::Class(),
 106                 AliAnalysisManager::kOutputContainer,
 107                 "cbaumann_LMEEPbPb2011Cent_out.root");
 108
 109
 110   mgr->ConnectInput(task,  0, mgr->GetCommonInputContainer());
 111   mgr->ConnectOutput(task, 0, coutput1 );
 112   mgr->ConnectOutput(task, 1, cOutputHist1);
 113   if (enableCF) {
 114         mgr->ConnectOutput(task, 2, cOutputHist2);
 115   }
 116   mgr->ConnectOutput(task, 3, cOutputHist3);
 117
 118   return task;
 119 }