]>
Commit | Line | Data |
---|---|---|
8349c102 | 1 | AliAnalysisTask *AddTask_Taku_LMEEPbPb2011Eff(Bool_t runRejection=kFALSE, Bool_t setMC=kTRUE, Bool_t enableCF=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("ConfigLMEEPbPb2011MC.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("MultiDiEData"); | |
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=ConfigLMEEPbPb2011MC(4,hasMC,enableCF); | |
51 | task->AddDielectron(lowmass4); | |
52 | printf("add: %s\n",lowmass4->GetName()); | |
53 | } | |
54 | else { | |
55 | ||
56 | AliDielectron *lowmass3=ConfigLMEEPbPb2011MC(3,hasMC,enableCF); | |
57 | task->AddDielectron(lowmass3); | |
58 | printf("add: %s\n",lowmass3->GetName()) | |
59 | AliDielectron *lowmass1=ConfigLMEEPbPb2011MC(1,hasMC,enableCF); | |
60 | task->AddDielectron(lowmass1); | |
61 | printf("add: %s\n",lowmass1->GetName()) | |
62 | AliDielectron *lowmass7=ConfigLMEEPbPb2011MC(7,hasMC,enableCF); | |
63 | task->AddDielectron(lowmass7); | |
64 | printf("add: %s\n",lowmass7->GetName()) | |
65 | } | |
66 | ||
67 | mgr->AddTask(task); | |
68 | ||
69 | //create output container | |
70 | AliAnalysisDataContainer *coutput1 = | |
71 | mgr->CreateContainer("mc_tg_LMEEPbPb2011_tree", | |
72 | TTree::Class(), | |
73 | AliAnalysisManager::kExchangeContainer, | |
74 | "mc_tg_LMEEPbPb2011_default.root"); | |
75 | ||
76 | AliAnalysisDataContainer *cOutputHist1 = | |
77 | mgr->CreateContainer("mc_tg_LMEEPbPb2011_out", | |
78 | TList::Class(), | |
79 | AliAnalysisManager::kOutputContainer, | |
80 | "mc_tg_LMEEPbPb2011_out.root"); | |
81 | ||
82 | AliAnalysisDataContainer *cOutputHist2 = 0x0; | |
83 | if (enableCF) { | |
84 | cOutputHist2 = | |
85 | mgr->CreateContainer("mc_tg_LMEEPbPb2011_CF", | |
86 | TList::Class(), | |
87 | AliAnalysisManager::kOutputContainer, | |
88 | "mc_tg_LMEEPbPb2011_out.root"); | |
89 | ||
90 | } | |
91 | AliAnalysisDataContainer *cOutputHist3 = | |
92 | mgr->CreateContainer("mc_tg_EventStatPbPb2011", | |
93 | TH1D::Class(), | |
94 | AliAnalysisManager::kOutputContainer, | |
95 | "mc_tg_LMEEPbPb2011_out.root"); | |
96 | ||
97 | ||
98 | mgr->ConnectInput(task, 0, mgr->GetCommonInputContainer()); | |
99 | mgr->ConnectOutput(task, 0, coutput1 ); | |
100 | mgr->ConnectOutput(task, 1, cOutputHist1); | |
101 | if (enableCF) { | |
102 | mgr->ConnectOutput(task, 2, cOutputHist2); | |
103 | } | |
104 | mgr->ConnectOutput(task, 3, cOutputHist3); | |
105 | ||
106 | return task; | |
107 | } |