]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWGGA/GammaConv/macros/AddTask_GammaConvV1_pPb.C
fixed bugs while streaming histos for weighting, added new trainconfig for pPb
[u/mrichter/AliRoot.git] / PWGGA / GammaConv / macros / AddTask_GammaConvV1_pPb.C
CommitLineData
33bbd991 1void AddTask_GammaConvV1_pPb( Int_t trainConfig = 1, //change different set of cuts
2 Bool_t isMC = kFALSE, //run MC
3 Bool_t enableQAMesonTask = kFALSE, //enable QA in AliAnalysisTaskGammaConvV1
4 Bool_t enableQAPhotonTask = kFALSE, // enable additional QA task
a7f0b705 5 TString fileNameInputForWeighting = "MCSpectraInput.root", // path to file for weigting input
6 TString cutnumberAODBranch = "8000000060084000001500000" // cutnumber for AOD branch
33bbd991 7 ) {
8
9 // ================= Load Librariers =================================
10 gSystem->Load("libCore.so");
11 gSystem->Load("libTree.so");
12 gSystem->Load("libGeom.so");
13 gSystem->Load("libVMC.so");
14 gSystem->Load("libPhysics.so");
15 gSystem->Load("libMinuit");
16 gSystem->Load("libSTEERBase");
17 gSystem->Load("libESD");
18 gSystem->Load("libAOD");
19 gSystem->Load("libANALYSIS");
20 gSystem->Load("libANALYSISalice");
21 gSystem->Load("libPWGGAGammaConv.so");
22 gSystem->Load("libCDB.so");
23 gSystem->Load("libSTEER.so");
24 gSystem->Load("libSTEERBase.so");
25 gSystem->Load("libTENDER.so");
26 gSystem->Load("libTENDERSupplies.so");
27
28 // ================== GetAnalysisManager ===============================
29 AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
30 if (!mgr) {
31 Error(Form("AddTask_GammaConvV1_%i",trainConfig), "No analysis manager found.");
32 return ;
33 }
34
35 // ================== GetInputEventHandler =============================
36 AliVEventHandler *inputHandler=mgr->GetInputEventHandler();
37
38 //========= Add PID Reponse to ANALYSIS manager ====
39 if(!(AliPIDResponse*)mgr->GetTask("PIDResponseTask")){
40 gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPIDResponse.C");
41 AddTaskPIDResponse(isMC);
42 }
43
44 //========= Set Cutnumber for V0Reader ================================
4803eb1f 45 TString cutnumber = "800000006008400100150000000";
33bbd991 46 Bool_t doEtaShift = kFALSE;
b0e0bc89 47 AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
33bbd991 48 //========= Add V0 Reader to ANALYSIS manager if not yet existent =====
49 if( !(AliV0ReaderV1*)mgr->GetTask("V0ReaderV1") ){
50 AliV0ReaderV1 *fV0ReaderV1 = new AliV0ReaderV1("V0ReaderV1");
51
52 fV0ReaderV1->SetUseOwnXYZCalculation(kTRUE);
53 fV0ReaderV1->SetCreateAODs(kFALSE);// AOD Output
54 fV0ReaderV1->SetUseAODConversionPhoton(kTRUE);
55
56 if (!mgr) {
57 Error("AddTask_V0ReaderV1", "No analysis manager found.");
58 return;
59 }
60
61 // Set AnalysisCut Number
62 AliConversionCuts *fCuts=NULL;
63 if(cutnumber!=""){
64 fCuts= new AliConversionCuts(cutnumber.Data(),cutnumber.Data());
65 fCuts->SetPreSelectionCutFlag(kTRUE);
66 if(fCuts->InitializeCutsFromCutString(cutnumber.Data())){
67 fCuts->DoEtaShift(doEtaShift);
68 fV0ReaderV1->SetConversionCuts(fCuts);
69 fCuts->SetFillCutHistograms("",kTRUE);
70 }
71 }
a7f0b705 72 if(inputHandler->IsA()==AliAODInputHandler::Class()){
73 // AOD mode
74 fV0ReaderV1->SetDeltaAODBranchName(Form("GammaConv_%s_gamma",cutnumberAODBranch.Data()));
75 }
33bbd991 76 fV0ReaderV1->Init();
77
78 AliLog::SetGlobalLogLevel(AliLog::kInfo);
79
33bbd991 80 //connect input V0Reader
81 mgr->AddTask(fV0ReaderV1);
82 mgr->ConnectInput(fV0ReaderV1,0,cinput);
83
84 }
85
86 //================================================
87 //========= Add task to the ANALYSIS manager =====
88 //================================================
33bbd991 89 // find input container
90 AliAnalysisTaskGammaConvV1 *task=NULL;
91 task= new AliAnalysisTaskGammaConvV1(Form("GammaConvV1_%i",trainConfig));
92 task->SetIsHeavyIon(2);
93 task->SetIsMC(isMC);
94 // Cut Numbers to use in Analysis
7aed7304 95 Int_t numberOfCuts = 6;
33bbd991 96
97 TString *cutarray = new TString[numberOfCuts];
98 TString *mesonCutArray = new TString[numberOfCuts];
99 Bool_t doEtaShiftIndCuts = kFALSE;
100 TString stringShift = "";
b0e0bc89 101 if(trainConfig == 1){
33bbd991 102 // Shifting in pPb direction
103 doEtaShiftIndCuts = kTRUE;
104 stringShift = "pPb";
4803eb1f 105 cutarray[ 0] = "800000008009317200329000000"; mesonCutArray[ 0] = "01629045000000"; //standard cut Pi0 pPb RCut Changed
106 cutarray[ 1] = "800000008109317200329000000"; mesonCutArray[ 1] = "01629045000000"; //standard cut Pi0 pPb RCut Changed
107 cutarray[ 2] = "800000008309317200329000000"; mesonCutArray[ 2] = "01629045000000"; //standard cut Pi0 pPb RCut Changed
108 cutarray[ 3] = "800000008409317200329000000"; mesonCutArray[ 3] = "01629045000000"; //standard cut Pi0 pPb RCut Changed
109 cutarray[ 4] = "800000008509317200329000000"; mesonCutArray[ 4] = "01629045000000"; //standard cut Pi0 pPb RCut Changed
110 cutarray[ 5] = "800000008609317200329000000"; mesonCutArray[ 5] = "01629045000000"; //standard cut Pi0 pPb RCut Changed
33bbd991 111 } else if (trainConfig == 2) {
e9772261 112 doEtaShiftIndCuts = kTRUE;
113 stringShift = "pPb";
4803eb1f 114 cutarray[ 0] = "800000008219317200329000000"; mesonCutArray[ 0] = "01629045000000"; //standard cut Pi0 pPb Single pT Cut changed
115 cutarray[ 1] = "800000008229317200329000000"; mesonCutArray[ 1] = "01629045000000"; //standard cut Pi0 pPb Single pT Cut changed
116 cutarray[ 2] = "800000008249317200329000000"; mesonCutArray[ 2] = "01629045000000"; //standard cut Pi0 pPb Single pT Cut changed
117 cutarray[ 3] = "800000008269317200329000000"; mesonCutArray[ 3] = "01629045000000"; //standard cut Pi0 pPb Single pT Cut changed
118 cutarray[ 4] = "800000008279317200329000000"; mesonCutArray[ 4] = "01629045000000"; //standard cut Pi0 pPb Single pT Cut changed
119 cutarray[ 5] = "800000008259317200329000000"; mesonCutArray[ 5] = "01629045000000"; //standard cut Pi0 pPb Single pT Cut changed
33bbd991 120 } else if (trainConfig == 3) {
e9772261 121 doEtaShiftIndCuts = kTRUE;
122 stringShift = "pPb";
4803eb1f 123 cutarray[ 0] = "800000008209017200329000000"; mesonCutArray[ 0] = "01629045000000"; //standard cut Pi0 pPb dedx electron Cut changed
124 cutarray[ 1] = "800000008209117200329000000"; mesonCutArray[ 1] = "01629045000000"; //standard cut Pi0 pPb dedx electron Cut changed
125 cutarray[ 2] = "800000008209217200329000000"; mesonCutArray[ 2] = "01629045000000"; //standard cut Pi0 pPb dedx electron Cut changed
126 cutarray[ 3] = "800000008209417200329000000"; mesonCutArray[ 3] = "01629045000000"; //standard cut Pi0 pPb dedx electron Cut changed
127 cutarray[ 4] = "800000008209517200329000000"; mesonCutArray[ 4] = "01629045000000"; //standard cut Pi0 pPb dedx electron Cut changed
128 cutarray[ 5] = "800000008209617200329000000"; mesonCutArray[ 5] = "01629045000000"; //standard cut Pi0 pPb dedx electron Cut changed
33bbd991 129 } else if (trainConfig == 4) {
33bbd991 130 doEtaShiftIndCuts = kTRUE;
e9772261 131 stringShift = "pPb";
4803eb1f 132 cutarray[ 0] = "800000008209316200329000000"; mesonCutArray[ 0] = "01629045000000"; //standard cut Pi0 pPb dedx Pion Line changed
133 cutarray[ 1] = "800000008209310200329000000"; mesonCutArray[ 1] = "01629045000000"; //standard cut Pi0 pPb dedx Pion Line changed
134 cutarray[ 2] = "800000008209317400329000000"; mesonCutArray[ 2] = "01629045000000"; //standard cut Pi0 pPb dedx Pion Line changed
135 cutarray[ 3] = "800000008209317300329000000"; mesonCutArray[ 3] = "01629045000000"; //standard cut Pi0 pPb dedx Pion Line changed
136 cutarray[ 4] = "800000008209357200329000000"; mesonCutArray[ 4] = "01629045000000"; //standard cut Pi0 pPb dedx Pion Line changed
137 cutarray[ 5] = "800000008209307200329000000"; mesonCutArray[ 5] = "01629045000000"; //standard cut Pi0 pPb dedx Pion Line changed
6feb1680 138 } else if (trainConfig == 5) {
139 doEtaShiftIndCuts = kFALSE;
ea924fff 140 stringShift = "pPb"; //GammaCuts
141 cutarray[ 0] = "800001100209326000380000000"; mesonCutArray[ 0] = "01631031009000";
142 cutarray[ 1] = "802001100209326000380000000"; mesonCutArray[ 1] = "01631031009000";
143 cutarray[ 2] = "804001100209326000380000000"; mesonCutArray[ 2] = "01631031009000";
144 cutarray[ 3] = "828001100209326000380000000"; mesonCutArray[ 3] = "01631031009000";
145 cutarray[ 4] = "848001100209326000380000000"; mesonCutArray[ 4] = "01631031009000";
146 cutarray[ 5] = "800001100209226000380000000"; mesonCutArray[ 5] = "01631031009000";
33bbd991 147 } else {
148 Error(Form("GammaConvV1_%i",trainConfig), "wrong trainConfig variable no cuts have been specified for the configuration");
149 return;
150 }
151
152 TList *ConvCutList = new TList();
153 TList *MesonCutList = new TList();
154
155 TList *HeaderList = new TList();
156 TObjString *Header1 = new TObjString("pi0_1");
157 HeaderList->Add(Header1);
158 TObjString *Header3 = new TObjString("eta_2");
159 HeaderList->Add(Header3);
160
161 ConvCutList->SetOwner(kTRUE);
162 AliConversionCuts **analysisCuts = new AliConversionCuts*[numberOfCuts];
163 MesonCutList->SetOwner(kTRUE);
164 AliConversionMesonCuts **analysisMesonCuts = new AliConversionMesonCuts*[numberOfCuts];
165
166
167 for(Int_t i = 0; i<numberOfCuts; i++){
168
169 analysisCuts[i] = new AliConversionCuts();
170 analysisCuts[i]->InitializeCutsFromCutString(cutarray[i].Data());
171 if (doEtaShiftIndCuts) {
172 analysisCuts[i]->DoEtaShift(doEtaShiftIndCuts);
173 analysisCuts[i]->SetEtaShift(stringShift);
174 }
175
33bbd991 176 ConvCutList->Add(analysisCuts[i]);
177
178 analysisCuts[i]->SetFillCutHistograms("",kFALSE);
179 analysisMesonCuts[i] = new AliConversionMesonCuts();
180 analysisMesonCuts[i]->InitializeCutsFromCutString(mesonCutArray[i].Data());
181 MesonCutList->Add(analysisMesonCuts[i]);
182 analysisMesonCuts[i]->SetFillCutHistograms("");
183 analysisCuts[i]->SetAcceptedHeader(HeaderList);
184 }
185
186 task->SetConversionCutList(numberOfCuts,ConvCutList);
187 task->SetMesonCutList(numberOfCuts,MesonCutList);
188 task->SetMoveParticleAccordingToVertex(kTRUE);
189 task->SetDoMesonAnalysis(kTRUE);
190 if (enableQAMesonTask) task->SetDoMesonQA(kTRUE); //Attention new switch for Pi0 QA
191 if (enableQAMesonTask) task->SetDoPhotonQA(kTRUE); //Attention new switch small for Photon QA
192
193 //connect containers
194 AliAnalysisDataContainer *coutput =
b0e0bc89 195 mgr->CreateContainer(Form("GammaConvV1_%i",trainConfig), TList::Class(),
33bbd991 196 AliAnalysisManager::kOutputContainer,Form("GammaConvV1_%i.root",trainConfig));
197
198 mgr->AddTask(task);
199 mgr->ConnectInput(task,0,cinput);
200 mgr->ConnectOutput(task,1,coutput);
201
202 return;
203
204}