]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWGGA/GammaConv/macros/AddTask_GammaConvV1_pPb.C
updates to fix histogram ranges, new 2D cuts, new Analysis macro for fast QA
[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
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}