]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWGGA/GammaConv/macros/AddTask_GammaConvV1_pPb.C
Adding more bins in QA (Alis)
[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
0b67315e 2 Bool_t isMC = kFALSE, //run MC
33bbd991 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 ) {
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 ================================
0b67315e 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");
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";
72395bd9 105 cutarray[ 0] = "800000108209317200329000000"; mesonCutArray[ 0] = "01629045009000"; //standard cut Pi0 pPb
106 cutarray[ 1] = "802000108209317200329000000"; mesonCutArray[ 1] = "01629045009000"; //standard cut Pi0 pPb
107 cutarray[ 2] = "824000108209317200329000000"; mesonCutArray[ 2] = "01629045009000"; //standard cut Pi0 pPb
108 cutarray[ 3] = "846000108209317200329000000"; mesonCutArray[ 3] = "01629045009000"; //standard cut Pi0 pPb
109 cutarray[ 4] = "868000108209317200329000000"; mesonCutArray[ 4] = "01629045009000"; //standard cut Pi0 pPb
110 cutarray[ 5] = "860000108209317200329000000"; mesonCutArray[ 5] = "01629045009000"; //standard cut Pi0 pPb
0b67315e 111 } else if (trainConfig == 2) {
112 // doEtaShiftIndCuts = kTRUE;
113 // stringShift = "pPb";
72395bd9 114 cutarray[ 0] = "800000107209317200329000000"; mesonCutArray[ 0] = "01627045009000";
115 cutarray[ 1] = "802000107209317200329000000"; mesonCutArray[ 1] = "01627045009000";
116 cutarray[ 2] = "824000107209317200329000000"; mesonCutArray[ 2] = "01627045009000";
117 cutarray[ 3] = "846000107209317200329000000"; mesonCutArray[ 3] = "01627045009000";
118 cutarray[ 4] = "868000107209317200329000000"; mesonCutArray[ 4] = "01627045009000";
119 cutarray[ 5] = "860000107209317200329000000"; mesonCutArray[ 5] = "01627045009000";
0b67315e 120 } else if (trainConfig == 3) {
121 // doEtaShiftIndCuts = kTRUE;
122 // stringShift = "pPb";
621d4c94 123 cutarray[ 0] = "800000100209317200329000000"; mesonCutArray[ 0] = "01621035009000";
124 cutarray[ 1] = "802000100209317200329000000"; mesonCutArray[ 1] = "01621035009000";
125 cutarray[ 2] = "824000100209317200329000000"; mesonCutArray[ 2] = "01621035009000";
126 cutarray[ 3] = "846000100209317200329000000"; mesonCutArray[ 3] = "01621035009000";
127 cutarray[ 4] = "868000100209317200329000000"; mesonCutArray[ 4] = "01621035009000";
128 cutarray[ 5] = "860000100209317200329000000"; mesonCutArray[ 5] = "01621035009000";
72395bd9 129 } else if(trainConfig == 4){
130 // Shifting in pPb direction
131 doEtaShiftIndCuts = kTRUE;
132 stringShift = "pPb";
133 cutarray[ 0] = "800000208209317200329000000"; mesonCutArray[ 0] = "01629045009000"; //standard cut Pi0 pPb
134 cutarray[ 1] = "802000208209317200329000000"; mesonCutArray[ 1] = "01629045009000"; //standard cut Pi0 pPb
135 cutarray[ 2] = "824000208209317200329000000"; mesonCutArray[ 2] = "01629045009000"; //standard cut Pi0 pPb
136 cutarray[ 3] = "846000208209317200329000000"; mesonCutArray[ 3] = "01629045009000"; //standard cut Pi0 pPb
137 cutarray[ 4] = "868000208209317200329000000"; mesonCutArray[ 4] = "01629045009000"; //standard cut Pi0 pPb
138 cutarray[ 5] = "860000208209317200329000000"; mesonCutArray[ 5] = "01629045009000"; //standard cut Pi0 pPb
139 } else if (trainConfig == 5) {
140 // doEtaShiftIndCuts = kTRUE;
141 // stringShift = "pPb";
142 cutarray[ 0] = "800000207209317200329000000"; mesonCutArray[ 0] = "01627045009000"; //standard cut Pi0 pPb Single pT Cut changed
143 cutarray[ 1] = "802000207209317200329000000"; mesonCutArray[ 1] = "01627045009000"; //standard cut Pi0 pPb Single pT Cut changed
144 cutarray[ 2] = "824000207209317200329000000"; mesonCutArray[ 2] = "01627045009000"; //standard cut Pi0 pPb Single pT Cut changed
145 cutarray[ 3] = "846000207209317200329000000"; mesonCutArray[ 3] = "01627045009000"; //standard cut Pi0 pPb Single pT Cut changed
146 cutarray[ 4] = "868000207209317200329000000"; mesonCutArray[ 4] = "01627045009000"; //standard cut Pi0 pPb Single pT Cut changed
147 cutarray[ 5] = "860000207209317200329000000"; mesonCutArray[ 5] = "01627045009000"; //standard cut Pi0 pPb Single pT Cut changed
148 } else if (trainConfig == 6) {
149 // doEtaShiftIndCuts = kTRUE;
150 // stringShift = "pPb";
621d4c94 151 cutarray[ 0] = "800000200209317200329000000"; mesonCutArray[ 0] = "01621035009000"; //standard cut Pi0 pPb Single pT Cut changed
152 cutarray[ 1] = "802000200209317200329000000"; mesonCutArray[ 1] = "01621035009000"; //standard cut Pi0 pPb Single pT Cut changed
153 cutarray[ 2] = "824000200209317200329000000"; mesonCutArray[ 2] = "01621035009000"; //standard cut Pi0 pPb Single pT Cut changed
154 cutarray[ 3] = "846000200209317200329000000"; mesonCutArray[ 3] = "01621035009000"; //standard cut Pi0 pPb Single pT Cut changed
155 cutarray[ 4] = "868000200209317200329000000"; mesonCutArray[ 4] = "01621035009000"; //standard cut Pi0 pPb Single pT Cut changed
156 cutarray[ 5] = "860000200209317200329000000"; 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);
201 if (enableQAMesonTask) task->SetDoMesonQA(kTRUE); //Attention new switch for Pi0 QA
0b67315e 202 if (enableQAMesonTask) task->SetDoPhotonQA(kTRUE); //Attention new switch small for Photon QA
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}