]>
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 | |
74 | fV0ReaderV1->SetDeltaAODBranchName(Form("GammaConv_%s_gamma",cutnumberAODBranch.Data())); | |
75 | } | |
33bbd991 | 76 | fV0ReaderV1->Init(); |
0b67315e | 77 | |
33bbd991 | 78 | AliLog::SetGlobalLogLevel(AliLog::kInfo); |
0b67315e | 79 | |
33bbd991 | 80 | //connect input V0Reader |
81 | mgr->AddTask(fV0ReaderV1); | |
82 | mgr->ConnectInput(fV0ReaderV1,0,cinput); | |
0b67315e | 83 | |
33bbd991 | 84 | } |
0b67315e | 85 | |
33bbd991 | 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; |
0b67315e | 96 | |
33bbd991 | 97 | TString *cutarray = new TString[numberOfCuts]; |
98 | TString *mesonCutArray = new TString[numberOfCuts]; | |
99 | Bool_t doEtaShiftIndCuts = kFALSE; | |
100 | TString stringShift = ""; | |
0b67315e | 101 | if(trainConfig == 1){ |
33bbd991 | 102 | // Shifting in pPb direction |
103 | doEtaShiftIndCuts = kTRUE; | |
104 | stringShift = "pPb"; | |
ae4f2cfb | 105 | cutarray[ 0] = "8000001082093172003290000000"; mesonCutArray[ 0] = "01629045009000"; //standard cut Pi0 pPb |
106 | cutarray[ 1] = "8020001082093172003290000000"; mesonCutArray[ 1] = "01629045009000"; //standard cut Pi0 pPb | |
107 | cutarray[ 2] = "8240001082093172003290000000"; mesonCutArray[ 2] = "01629045009000"; //standard cut Pi0 pPb | |
108 | cutarray[ 3] = "8460001082093172003290000000"; mesonCutArray[ 3] = "01629045009000"; //standard cut Pi0 pPb | |
109 | cutarray[ 4] = "8680001082093172003290000000"; mesonCutArray[ 4] = "01629045009000"; //standard cut Pi0 pPb | |
110 | cutarray[ 5] = "8600001082093172003290000000"; mesonCutArray[ 5] = "01629045009000"; //standard cut Pi0 pPb | |
0b67315e | 111 | } else if (trainConfig == 2) { |
112 | // doEtaShiftIndCuts = kTRUE; | |
113 | // stringShift = "pPb"; | |
ae4f2cfb | 114 | cutarray[ 0] = "8000001072093172003290000000"; mesonCutArray[ 0] = "01627045009000"; |
115 | cutarray[ 1] = "8020001072093172003290000000"; mesonCutArray[ 1] = "01627045009000"; | |
116 | cutarray[ 2] = "8240001072093172003290000000"; mesonCutArray[ 2] = "01627045009000"; | |
117 | cutarray[ 3] = "8460001072093172003290000000"; mesonCutArray[ 3] = "01627045009000"; | |
118 | cutarray[ 4] = "8680001072093172003290000000"; mesonCutArray[ 4] = "01627045009000"; | |
119 | cutarray[ 5] = "8600001072093172003290000000"; mesonCutArray[ 5] = "01627045009000"; | |
0b67315e | 120 | } else if (trainConfig == 3) { |
121 | // doEtaShiftIndCuts = kTRUE; | |
122 | // stringShift = "pPb"; | |
ae4f2cfb | 123 | cutarray[ 0] = "8000001002093172003290000000"; mesonCutArray[ 0] = "01621035009000"; |
124 | cutarray[ 1] = "8020001002093172003290000000"; mesonCutArray[ 1] = "01621035009000"; | |
125 | cutarray[ 2] = "8240001002093172003290000000"; mesonCutArray[ 2] = "01621035009000"; | |
126 | cutarray[ 3] = "8460001002093172003290000000"; mesonCutArray[ 3] = "01621035009000"; | |
127 | cutarray[ 4] = "8680001002093172003290000000"; mesonCutArray[ 4] = "01621035009000"; | |
128 | cutarray[ 5] = "8600001002093172003290000000"; mesonCutArray[ 5] = "01621035009000"; | |
72395bd9 | 129 | } else if(trainConfig == 4){ |
130 | // Shifting in pPb direction | |
131 | doEtaShiftIndCuts = kTRUE; | |
132 | stringShift = "pPb"; | |
ae4f2cfb | 133 | cutarray[ 0] = "8000002082093172003290000000"; mesonCutArray[ 0] = "01629045009000"; //standard cut Pi0 pPb |
134 | cutarray[ 1] = "8020002082093172003290000000"; mesonCutArray[ 1] = "01629045009000"; //standard cut Pi0 pPb | |
135 | cutarray[ 2] = "8240002082093172003290000000"; mesonCutArray[ 2] = "01629045009000"; //standard cut Pi0 pPb | |
136 | cutarray[ 3] = "8460002082093172003290000000"; mesonCutArray[ 3] = "01629045009000"; //standard cut Pi0 pPb | |
137 | cutarray[ 4] = "8680002082093172003290000000"; mesonCutArray[ 4] = "01629045009000"; //standard cut Pi0 pPb | |
138 | cutarray[ 5] = "8600002082093172003290000000"; mesonCutArray[ 5] = "01629045009000"; //standard cut Pi0 pPb | |
72395bd9 | 139 | } else if (trainConfig == 5) { |
140 | // doEtaShiftIndCuts = kTRUE; | |
141 | // stringShift = "pPb"; | |
ae4f2cfb | 142 | cutarray[ 0] = "8000002072093172003290000000"; mesonCutArray[ 0] = "01627045009000"; //standard cut Pi0 pPb Single pT Cut changed |
143 | cutarray[ 1] = "8020002072093172003290000000"; mesonCutArray[ 1] = "01627045009000"; //standard cut Pi0 pPb Single pT Cut changed | |
144 | cutarray[ 2] = "8240002072093172003290000000"; mesonCutArray[ 2] = "01627045009000"; //standard cut Pi0 pPb Single pT Cut changed | |
145 | cutarray[ 3] = "8460002072093172003290000000"; mesonCutArray[ 3] = "01627045009000"; //standard cut Pi0 pPb Single pT Cut changed | |
146 | cutarray[ 4] = "8680002072093172003290000000"; mesonCutArray[ 4] = "01627045009000"; //standard cut Pi0 pPb Single pT Cut changed | |
147 | cutarray[ 5] = "8600002072093172003290000000"; mesonCutArray[ 5] = "01627045009000"; //standard cut Pi0 pPb Single pT Cut changed | |
72395bd9 | 148 | } else if (trainConfig == 6) { |
149 | // doEtaShiftIndCuts = kTRUE; | |
150 | // stringShift = "pPb"; | |
ae4f2cfb | 151 | cutarray[ 0] = "8000002002093172003290000000"; mesonCutArray[ 0] = "01621035009000"; //standard cut Pi0 pPb Single pT Cut changed |
152 | cutarray[ 1] = "8020002002093172003290000000"; mesonCutArray[ 1] = "01621035009000"; //standard cut Pi0 pPb Single pT Cut changed | |
153 | cutarray[ 2] = "8240002002093172003290000000"; mesonCutArray[ 2] = "01621035009000"; //standard cut Pi0 pPb Single pT Cut changed | |
154 | cutarray[ 3] = "8460002002093172003290000000"; mesonCutArray[ 3] = "01621035009000"; //standard cut Pi0 pPb Single pT Cut changed | |
155 | cutarray[ 4] = "8680002002093172003290000000"; mesonCutArray[ 4] = "01621035009000"; //standard cut Pi0 pPb Single pT Cut changed | |
156 | cutarray[ 5] = "8600002002093172003290000000"; mesonCutArray[ 5] = "01621035009000"; //standard cut Pi0 pPb Single pT Cut changed | |
72395bd9 | 157 | |
158 | } else { | |
33bbd991 | 159 | Error(Form("GammaConvV1_%i",trainConfig), "wrong trainConfig variable no cuts have been specified for the configuration"); |
160 | return; | |
161 | } | |
0b67315e | 162 | |
33bbd991 | 163 | TList *ConvCutList = new TList(); |
164 | TList *MesonCutList = new TList(); | |
0b67315e | 165 | |
33bbd991 | 166 | TList *HeaderList = new TList(); |
167 | TObjString *Header1 = new TObjString("pi0_1"); | |
168 | HeaderList->Add(Header1); | |
169 | TObjString *Header3 = new TObjString("eta_2"); | |
170 | HeaderList->Add(Header3); | |
171 | ||
172 | ConvCutList->SetOwner(kTRUE); | |
173 | AliConversionCuts **analysisCuts = new AliConversionCuts*[numberOfCuts]; | |
174 | MesonCutList->SetOwner(kTRUE); | |
175 | AliConversionMesonCuts **analysisMesonCuts = new AliConversionMesonCuts*[numberOfCuts]; | |
0b67315e | 176 | |
177 | ||
33bbd991 | 178 | for(Int_t i = 0; i<numberOfCuts; i++){ |
0b67315e | 179 | |
33bbd991 | 180 | analysisCuts[i] = new AliConversionCuts(); |
181 | analysisCuts[i]->InitializeCutsFromCutString(cutarray[i].Data()); | |
182 | if (doEtaShiftIndCuts) { | |
183 | analysisCuts[i]->DoEtaShift(doEtaShiftIndCuts); | |
184 | analysisCuts[i]->SetEtaShift(stringShift); | |
185 | } | |
0b67315e | 186 | |
33bbd991 | 187 | ConvCutList->Add(analysisCuts[i]); |
188 | ||
189 | analysisCuts[i]->SetFillCutHistograms("",kFALSE); | |
190 | analysisMesonCuts[i] = new AliConversionMesonCuts(); | |
191 | analysisMesonCuts[i]->InitializeCutsFromCutString(mesonCutArray[i].Data()); | |
192 | MesonCutList->Add(analysisMesonCuts[i]); | |
193 | analysisMesonCuts[i]->SetFillCutHistograms(""); | |
194 | analysisCuts[i]->SetAcceptedHeader(HeaderList); | |
195 | } | |
0b67315e | 196 | |
33bbd991 | 197 | task->SetConversionCutList(numberOfCuts,ConvCutList); |
198 | task->SetMesonCutList(numberOfCuts,MesonCutList); | |
199 | task->SetMoveParticleAccordingToVertex(kTRUE); | |
200 | task->SetDoMesonAnalysis(kTRUE); | |
ae4f2cfb | 201 | task->SetDoMesonQA(enableQAMesonTask); //Attention new switch for Pi0 QA |
202 | task->SetDoPhotonQA(enableQAMesonTask); //Attention new switch small for Photon QA | |
0b67315e | 203 | |
33bbd991 | 204 | //connect containers |
205 | AliAnalysisDataContainer *coutput = | |
b0e0bc89 | 206 | mgr->CreateContainer(Form("GammaConvV1_%i",trainConfig), TList::Class(), |
33bbd991 | 207 | AliAnalysisManager::kOutputContainer,Form("GammaConvV1_%i.root",trainConfig)); |
0b67315e | 208 | |
33bbd991 | 209 | mgr->AddTask(task); |
210 | mgr->ConnectInput(task,0,cinput); | |
211 | mgr->ConnectOutput(task,1,coutput); | |
0b67315e | 212 | |
33bbd991 | 213 | return; |
0b67315e | 214 | |
33bbd991 | 215 | } |