]>
Commit | Line | Data |
---|---|---|
9a476b26 | 1 | AliAnalysisTask *AddTask_cbaumann_LMEEPbPb2011SemiCent(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 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("MultiDiEDataSemiCent"); | |
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 | ||
56 | AliDielectron *lowmass3=ConfigLMEEPbPb2011(3,hasMC,enableCF); | |
50191baf | 57 | lowmass3->SetUseKF(kFALSE); |
58 | task->AddDielectron(lowmass3); | |
9a476b26 | 59 | printf("add: %s\n",lowmass3->GetName()); |
60 | } | |
61 | ||
62 | mgr->AddTask(task); | |
63 | ||
64 | //create output container | |
65 | AliAnalysisDataContainer *coutput1 = | |
66 | mgr->CreateContainer("cbaumann_LMEEPbPb2011SemiCent_tree", | |
67 | TTree::Class(), | |
68 | AliAnalysisManager::kExchangeContainer, | |
69 | "cbaumann_LMEEPbPb2011SemiCent_default.root"); | |
70 | ||
71 | AliAnalysisDataContainer *cOutputHist1 = | |
72 | mgr->CreateContainer("cbaumann_LMEEPbPb2011SemiCent_out", | |
73 | TList::Class(), | |
74 | AliAnalysisManager::kOutputContainer, | |
75 | "cbaumann_LMEEPbPb2011SemiCent_out.root"); | |
76 | ||
77 | AliAnalysisDataContainer *cOutputHist2 = 0x0; | |
78 | if (enableCF) { | |
79 | cOutputHist2 = | |
80 | mgr->CreateContainer("cbaumann_LMEEPbPb2011SemiCent_CF", | |
81 | TList::Class(), | |
82 | AliAnalysisManager::kOutputContainer, | |
83 | "cbaumann_LMEEPbPb2011SemiCent_out.root"); | |
84 | ||
85 | } | |
86 | AliAnalysisDataContainer *cOutputHist3 = | |
87 | mgr->CreateContainer("cbaumann_EventStatPbPb2011SemiCent", | |
88 | TH1D::Class(), | |
89 | AliAnalysisManager::kOutputContainer, | |
90 | "cbaumann_LMEEPbPb2011SemiCent_out.root"); | |
91 | ||
92 | ||
93 | mgr->ConnectInput(task, 0, mgr->GetCommonInputContainer()); | |
94 | mgr->ConnectOutput(task, 0, coutput1 ); | |
95 | mgr->ConnectOutput(task, 1, cOutputHist1); | |
96 | if (enableCF) { | |
97 | mgr->ConnectOutput(task, 2, cOutputHist2); | |
98 | } | |
99 | mgr->ConnectOutput(task, 3, cOutputHist3); | |
100 | ||
101 | return task; | |
102 | } |