]>
Commit | Line | Data |
---|---|---|
33bbd991 | 1 | void 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 | } |