]>
Commit | Line | Data |
---|---|---|
33bbd991 | 1 | void AddTask_GammaConvV1_pPb( Int_t trainConfig = 1, //change different set of cuts |
0b67315e | 2 | Bool_t isMC = kFALSE, //run MC |
ae4f2cfb | 3 | Int_t enableQAMesonTask = 0, //enable QA in AliAnalysisTaskGammaConvV1 |
4 | Int_t enableQAPhotonTask = 0, // 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 | ) { |
0b67315e | 8 | |
33bbd991 | 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"); | |
0b67315e | 27 | |
33bbd991 | 28 | // ================== GetAnalysisManager =============================== |
29 | AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager(); | |
30 | if (!mgr) { | |
31 | Error(Form("AddTask_GammaConvV1_%i",trainConfig), "No analysis manager found."); | |
32 | return ; | |
33 | } | |
0b67315e | 34 | |
33bbd991 | 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 ================================ | |
ae4f2cfb | 45 | TString cutnumber = "8000000060084001001500000000"; |
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"); | |
0b67315e | 51 | |
33bbd991 | 52 | fV0ReaderV1->SetUseOwnXYZCalculation(kTRUE); |
53 | fV0ReaderV1->SetCreateAODs(kFALSE);// AOD Output | |
54 | fV0ReaderV1->SetUseAODConversionPhoton(kTRUE); | |
0b67315e | 55 | |
33bbd991 | 56 | if (!mgr) { |
57 | Error("AddTask_V0ReaderV1", "No analysis manager found."); | |
58 | return; | |
59 | } | |
0b67315e | 60 | |
33bbd991 | 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 | |
d8b864f8 | 74 | cout << "AOD handler: adding " << cutnumberAODBranch.Data() << " as conversion branch" << endl; |
a7f0b705 | 75 | fV0ReaderV1->SetDeltaAODBranchName(Form("GammaConv_%s_gamma",cutnumberAODBranch.Data())); |
76 | } | |
33bbd991 | 77 | fV0ReaderV1->Init(); |
0b67315e | 78 | |
33bbd991 | 79 | AliLog::SetGlobalLogLevel(AliLog::kInfo); |
0b67315e | 80 | |
33bbd991 | 81 | //connect input V0Reader |
82 | mgr->AddTask(fV0ReaderV1); | |
83 | mgr->ConnectInput(fV0ReaderV1,0,cinput); | |
0b67315e | 84 | |
33bbd991 | 85 | } |
0b67315e | 86 | |
33bbd991 | 87 | //================================================ |
88 | //========= Add task to the ANALYSIS manager ===== | |
89 | //================================================ | |
33bbd991 | 90 | // find input container |
91 | AliAnalysisTaskGammaConvV1 *task=NULL; | |
92 | task= new AliAnalysisTaskGammaConvV1(Form("GammaConvV1_%i",trainConfig)); | |
93 | task->SetIsHeavyIon(2); | |
94 | task->SetIsMC(isMC); | |
95 | // Cut Numbers to use in Analysis | |
7aed7304 | 96 | Int_t numberOfCuts = 6; |
0b67315e | 97 | |
33bbd991 | 98 | TString *cutarray = new TString[numberOfCuts]; |
99 | TString *mesonCutArray = new TString[numberOfCuts]; | |
100 | Bool_t doEtaShiftIndCuts = kFALSE; | |
101 | TString stringShift = ""; | |
0b67315e | 102 | if(trainConfig == 1){ |
33bbd991 | 103 | // Shifting in pPb direction |
104 | doEtaShiftIndCuts = kTRUE; | |
105 | stringShift = "pPb"; | |
ae4f2cfb | 106 | cutarray[ 0] = "8000001082093172003290000000"; mesonCutArray[ 0] = "01629045009000"; //standard cut Pi0 pPb |
107 | cutarray[ 1] = "8020001082093172003290000000"; mesonCutArray[ 1] = "01629045009000"; //standard cut Pi0 pPb | |
108 | cutarray[ 2] = "8240001082093172003290000000"; mesonCutArray[ 2] = "01629045009000"; //standard cut Pi0 pPb | |
109 | cutarray[ 3] = "8460001082093172003290000000"; mesonCutArray[ 3] = "01629045009000"; //standard cut Pi0 pPb | |
110 | cutarray[ 4] = "8680001082093172003290000000"; mesonCutArray[ 4] = "01629045009000"; //standard cut Pi0 pPb | |
111 | cutarray[ 5] = "8600001082093172003290000000"; mesonCutArray[ 5] = "01629045009000"; //standard cut Pi0 pPb | |
0b67315e | 112 | } else if (trainConfig == 2) { |
113 | // doEtaShiftIndCuts = kTRUE; | |
114 | // stringShift = "pPb"; | |
ae4f2cfb | 115 | cutarray[ 0] = "8000001072093172003290000000"; mesonCutArray[ 0] = "01627045009000"; |
116 | cutarray[ 1] = "8020001072093172003290000000"; mesonCutArray[ 1] = "01627045009000"; | |
117 | cutarray[ 2] = "8240001072093172003290000000"; mesonCutArray[ 2] = "01627045009000"; | |
118 | cutarray[ 3] = "8460001072093172003290000000"; mesonCutArray[ 3] = "01627045009000"; | |
119 | cutarray[ 4] = "8680001072093172003290000000"; mesonCutArray[ 4] = "01627045009000"; | |
120 | cutarray[ 5] = "8600001072093172003290000000"; mesonCutArray[ 5] = "01627045009000"; | |
0b67315e | 121 | } else if (trainConfig == 3) { |
122 | // doEtaShiftIndCuts = kTRUE; | |
123 | // stringShift = "pPb"; | |
ae4f2cfb | 124 | cutarray[ 0] = "8000001002093172003290000000"; mesonCutArray[ 0] = "01621035009000"; |
125 | cutarray[ 1] = "8020001002093172003290000000"; mesonCutArray[ 1] = "01621035009000"; | |
126 | cutarray[ 2] = "8240001002093172003290000000"; mesonCutArray[ 2] = "01621035009000"; | |
127 | cutarray[ 3] = "8460001002093172003290000000"; mesonCutArray[ 3] = "01621035009000"; | |
128 | cutarray[ 4] = "8680001002093172003290000000"; mesonCutArray[ 4] = "01621035009000"; | |
129 | cutarray[ 5] = "8600001002093172003290000000"; mesonCutArray[ 5] = "01621035009000"; | |
72395bd9 | 130 | } else if(trainConfig == 4){ |
131 | // Shifting in pPb direction | |
132 | doEtaShiftIndCuts = kTRUE; | |
133 | stringShift = "pPb"; | |
ae4f2cfb | 134 | cutarray[ 0] = "8000002082093172003290000000"; mesonCutArray[ 0] = "01629045009000"; //standard cut Pi0 pPb |
135 | cutarray[ 1] = "8020002082093172003290000000"; mesonCutArray[ 1] = "01629045009000"; //standard cut Pi0 pPb | |
136 | cutarray[ 2] = "8240002082093172003290000000"; mesonCutArray[ 2] = "01629045009000"; //standard cut Pi0 pPb | |
137 | cutarray[ 3] = "8460002082093172003290000000"; mesonCutArray[ 3] = "01629045009000"; //standard cut Pi0 pPb | |
138 | cutarray[ 4] = "8680002082093172003290000000"; mesonCutArray[ 4] = "01629045009000"; //standard cut Pi0 pPb | |
139 | cutarray[ 5] = "8600002082093172003290000000"; mesonCutArray[ 5] = "01629045009000"; //standard cut Pi0 pPb | |
72395bd9 | 140 | } else if (trainConfig == 5) { |
141 | // doEtaShiftIndCuts = kTRUE; | |
142 | // stringShift = "pPb"; | |
ae4f2cfb | 143 | cutarray[ 0] = "8000002072093172003290000000"; mesonCutArray[ 0] = "01627045009000"; //standard cut Pi0 pPb Single pT Cut changed |
144 | cutarray[ 1] = "8020002072093172003290000000"; mesonCutArray[ 1] = "01627045009000"; //standard cut Pi0 pPb Single pT Cut changed | |
145 | cutarray[ 2] = "8240002072093172003290000000"; mesonCutArray[ 2] = "01627045009000"; //standard cut Pi0 pPb Single pT Cut changed | |
146 | cutarray[ 3] = "8460002072093172003290000000"; mesonCutArray[ 3] = "01627045009000"; //standard cut Pi0 pPb Single pT Cut changed | |
147 | cutarray[ 4] = "8680002072093172003290000000"; mesonCutArray[ 4] = "01627045009000"; //standard cut Pi0 pPb Single pT Cut changed | |
148 | cutarray[ 5] = "8600002072093172003290000000"; mesonCutArray[ 5] = "01627045009000"; //standard cut Pi0 pPb Single pT Cut changed | |
72395bd9 | 149 | } else if (trainConfig == 6) { |
150 | // doEtaShiftIndCuts = kTRUE; | |
151 | // stringShift = "pPb"; | |
ae4f2cfb | 152 | cutarray[ 0] = "8000002002093172003290000000"; mesonCutArray[ 0] = "01621035009000"; //standard cut Pi0 pPb Single pT Cut changed |
153 | cutarray[ 1] = "8020002002093172003290000000"; mesonCutArray[ 1] = "01621035009000"; //standard cut Pi0 pPb Single pT Cut changed | |
154 | cutarray[ 2] = "8240002002093172003290000000"; mesonCutArray[ 2] = "01621035009000"; //standard cut Pi0 pPb Single pT Cut changed | |
155 | cutarray[ 3] = "8460002002093172003290000000"; mesonCutArray[ 3] = "01621035009000"; //standard cut Pi0 pPb Single pT Cut changed | |
156 | cutarray[ 4] = "8680002002093172003290000000"; mesonCutArray[ 4] = "01621035009000"; //standard cut Pi0 pPb Single pT Cut changed | |
157 | cutarray[ 5] = "8600002002093172003290000000"; mesonCutArray[ 5] = "01621035009000"; //standard cut Pi0 pPb Single pT Cut changed | |
bacaa826 | 158 | } else if (trainConfig == 7) { |
159 | // doEtaShiftIndCuts = kTRUE; | |
160 | // stringShift = "pPb"; | |
161 | cutarray[ 0] = "8000001002092970023220000000"; mesonCutArray[ 0] = "01621035009000"; | |
162 | cutarray[ 1] = "8000001002093663003800000000"; mesonCutArray[ 1] = "01621035009000"; | |
163 | cutarray[ 2] = "8000001002093260003800000000"; mesonCutArray[ 2] = "01621035009000"; | |
164 | cutarray[ 3] = "8000001002092370023220000000"; mesonCutArray[ 3] = "01621035009000"; | |
165 | cutarray[ 4] = "8000001002092770023220000000"; mesonCutArray[ 4] = "01621035009000"; | |
166 | cutarray[ 5] = "8000001002092551023220000000"; mesonCutArray[ 5] = "01621035009000"; | |
167 | } else if (trainConfig == 8) { | |
168 | // doEtaShiftIndCuts = kTRUE; | |
169 | // stringShift = "pPb"; | |
170 | cutarray[ 0] = "8000002002092970023220000000"; mesonCutArray[ 0] = "01621035009000"; | |
171 | cutarray[ 1] = "8000002002093663003800000000"; mesonCutArray[ 1] = "01621035009000"; | |
172 | cutarray[ 2] = "8000002002093260003800000000"; mesonCutArray[ 2] = "01621035009000"; | |
173 | cutarray[ 3] = "8000002002092370023220000000"; mesonCutArray[ 3] = "01621035009000"; | |
174 | cutarray[ 4] = "8000002002092770023220000000"; mesonCutArray[ 4] = "01621035009000"; | |
175 | cutarray[ 5] = "8000002002092551023220000000"; mesonCutArray[ 5] = "01621035009000"; | |
72395bd9 | 176 | } else { |
33bbd991 | 177 | Error(Form("GammaConvV1_%i",trainConfig), "wrong trainConfig variable no cuts have been specified for the configuration"); |
178 | return; | |
179 | } | |
0b67315e | 180 | |
33bbd991 | 181 | TList *ConvCutList = new TList(); |
182 | TList *MesonCutList = new TList(); | |
0b67315e | 183 | |
33bbd991 | 184 | TList *HeaderList = new TList(); |
185 | TObjString *Header1 = new TObjString("pi0_1"); | |
186 | HeaderList->Add(Header1); | |
187 | TObjString *Header3 = new TObjString("eta_2"); | |
188 | HeaderList->Add(Header3); | |
189 | ||
190 | ConvCutList->SetOwner(kTRUE); | |
191 | AliConversionCuts **analysisCuts = new AliConversionCuts*[numberOfCuts]; | |
192 | MesonCutList->SetOwner(kTRUE); | |
193 | AliConversionMesonCuts **analysisMesonCuts = new AliConversionMesonCuts*[numberOfCuts]; | |
0b67315e | 194 | |
195 | ||
33bbd991 | 196 | for(Int_t i = 0; i<numberOfCuts; i++){ |
0b67315e | 197 | |
33bbd991 | 198 | analysisCuts[i] = new AliConversionCuts(); |
199 | analysisCuts[i]->InitializeCutsFromCutString(cutarray[i].Data()); | |
200 | if (doEtaShiftIndCuts) { | |
201 | analysisCuts[i]->DoEtaShift(doEtaShiftIndCuts); | |
202 | analysisCuts[i]->SetEtaShift(stringShift); | |
203 | } | |
0b67315e | 204 | |
33bbd991 | 205 | ConvCutList->Add(analysisCuts[i]); |
206 | ||
207 | analysisCuts[i]->SetFillCutHistograms("",kFALSE); | |
208 | analysisMesonCuts[i] = new AliConversionMesonCuts(); | |
209 | analysisMesonCuts[i]->InitializeCutsFromCutString(mesonCutArray[i].Data()); | |
210 | MesonCutList->Add(analysisMesonCuts[i]); | |
211 | analysisMesonCuts[i]->SetFillCutHistograms(""); | |
212 | analysisCuts[i]->SetAcceptedHeader(HeaderList); | |
213 | } | |
0b67315e | 214 | |
33bbd991 | 215 | task->SetConversionCutList(numberOfCuts,ConvCutList); |
216 | task->SetMesonCutList(numberOfCuts,MesonCutList); | |
217 | task->SetMoveParticleAccordingToVertex(kTRUE); | |
218 | task->SetDoMesonAnalysis(kTRUE); | |
ae4f2cfb | 219 | task->SetDoMesonQA(enableQAMesonTask); //Attention new switch for Pi0 QA |
220 | task->SetDoPhotonQA(enableQAMesonTask); //Attention new switch small for Photon QA | |
0b67315e | 221 | |
33bbd991 | 222 | //connect containers |
223 | AliAnalysisDataContainer *coutput = | |
b0e0bc89 | 224 | mgr->CreateContainer(Form("GammaConvV1_%i",trainConfig), TList::Class(), |
33bbd991 | 225 | AliAnalysisManager::kOutputContainer,Form("GammaConvV1_%i.root",trainConfig)); |
0b67315e | 226 | |
33bbd991 | 227 | mgr->AddTask(task); |
228 | mgr->ConnectInput(task,0,cinput); | |
229 | mgr->ConnectOutput(task,1,coutput); | |
0b67315e | 230 | |
33bbd991 | 231 | return; |
0b67315e | 232 | |
33bbd991 | 233 | } |