]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWG4/JetTasks/AliPWG4HighPtQAMC.cxx
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
samll fixes (Marta)
[u/mrichter/AliRoot.git] / PWG4 / JetTasks / AliPWG4HighPtQAMC.cxx
CommitLineData
fdceab34 1/**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3 * *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
6 * *
7 * Permission to use, copy, modify and distribute this software and its *
8 * documentation strictly for non-commercial purposes is hereby granted *
9 * without fee, provided that the above copyright notice appears in all *
10 * copies and that both the copyright notice and this permission notice *
11 * appear in the supporting documentation. The authors make no claims *
12 * about the suitability of this software for any purpose. It is *
13 * provided "as is" without express or implied warranty. *
14 **************************************************************************/
15
16//-----------------------------------------------------------------------
17// This class compares the global reconstruction with the MC information
18// Momentum resolution is stored as function of track cuts and pt.
19// Output: Histograms for different set of cuts
20//-----------------------------------------------------------------------
21// Author : Marta Verweij - UU
22//-----------------------------------------------------------------------
23
df943115 24#ifndef ALIPWG4HighPtQAMC_CXX
25#define ALIPWG4HighPtQAMC_CXX
26
fdceab34 27#include "AliPWG4HighPtQAMC.h"
28
29#include "TH1.h"
30#include "TH2.h"
31#include "TH3.h"
b4691ee7 32#include "TProfile.h"
fdceab34 33#include "TList.h"
b4691ee7 34#include "TFile.h"
fdceab34 35#include "TChain.h"
36#include "TH3F.h"
b4691ee7 37#include "TKey.h"
38#include "TSystem.h"
39
40#include "AliAnalysisTask.h"
fdceab34 41#include "AliAnalysisManager.h"
42#include "AliESDInputHandler.h"
43#include "AliMCEvent.h"
44#include "AliMCEventHandler.h"
45#include "AliStack.h"
46#include "AliESDtrack.h"
47#include "AliESDtrackCuts.h"
48#include "AliExternalTrackParam.h"
df943115 49#include "AliLog.h"
b4691ee7 50#include "AliGenPythiaEventHeader.h"
51#include "AliGenCocktailEventHeader.h"
52//#include "AliAnalysisHelperJetTasks.h"
fdceab34 53
54using namespace std; //required for resolving the 'cout' symbol
55
56ClassImp(AliPWG4HighPtQAMC)
57
b4691ee7 58AliPWG4HighPtQAMC::AliPWG4HighPtQAMC()
59: AliAnalysisTask("AliPWG4HighPtQAMC", ""),
fdceab34 60 fESD(0),
b1cd0099 61 fMC(0),
cce400da 62 fStack(0),
36c36a0c 63 fVtx(0x0),
fdceab34 64 fTrackCuts(0),
65 fTrackCutsITS(0),
71e77a79 66 fTrackType(0),
aa3ba8d2 67 fSigmaConstrainedMax(5.),
b1cd0099 68 fPtMax(100.),
b4691ee7 69 fAvgTrials(1),
8f0faa80 70 fNEventAll(0),
71 fNEventSel(0),
cb76764e 72 fNEventReject(0),
b4691ee7 73 fh1Xsec(0),
74 fh1Trials(0),
75 fh1PtHard(0),
76 fh1PtHardTrials(0),
df943115 77 fPtAll(0),
78 fPtSel(0),
36c36a0c 79 fPtSelFakes(0),
80 fNPointTPCFakes(0),
81 fPtSelLargeLabel(0),
82 fMultRec(0),
83 fNPointTPCMultRec(0),
84 fDeltaPtMultRec(0),
57bee263 85 fPtAllvsPtMC(0),
fdceab34 86 fPtAllminPtMCvsPtAll(0),
57bee263 87 fPtAllvsPtMCvsMult(0),
88 fPtAllminPtMCvsPtAllvsMult(0),
fdceab34 89 fPtAllminPtMCvsPtAllNPointTPC(0),
90 fPtAllminPtMCvsPtAllDCAR(0),
91 fPtAllminPtMCvsPtAllDCAZ(0),
92 fPtAllminPtMCvsPtAllPhi(0),
93 fPtAllminPtMCvsPtAllNPointITS(0),
94 fPtAllminPtMCvsPtAllNSigmaToVertex(0),
95 fPtAllminPtMCvsPtAllChi2C(0),
96 fPtAllminPtMCvsPtAllRel1PtUncertainty(0),
97 fPtAllMC(0),
98 fPtSelMC(0),
99 fPtSelMCITS(0),
100 fHistList(0),
101 fPtSelITS(0),
102 fPtITSminPtMCvsPtITS(0),
103 fPtITSminPtMCvsPtITSNPointTPC(0),
104 fPtITSminPtMCvsPtITSDCAR(0),
105 fPtITSminPtMCvsPtITSDCAZ(0),
106 fPtITSminPtMCvsPtITSPhi(0),
107 fPtITSminPtMCvsPtITSNPointITS(0),
108 fPtITSminPtMCvsPtITSNSigmaToVertex(0),
109 fPtITSminPtMCvsPtITSChi2C(0),
110 fPtITSminPtMCvsPtITSRel1PtUncertainty(0),
111 fHistListITS(0)
112{
df943115 113
fdceab34 114}
115//________________________________________________________________________
116AliPWG4HighPtQAMC::AliPWG4HighPtQAMC(const char *name):
b4691ee7 117 AliAnalysisTask(name,""),
fdceab34 118 fESD(0),
b1cd0099 119 fMC(0),
cce400da 120 fStack(0),
36c36a0c 121 fVtx(0x0),
fdceab34 122 fTrackCuts(),
123 fTrackCutsITS(),
71e77a79 124 fTrackType(0),
aa3ba8d2 125 fSigmaConstrainedMax(5.),
b1cd0099 126 fPtMax(100.),
b4691ee7 127 fAvgTrials(1),
8f0faa80 128 fNEventAll(0),
129 fNEventSel(0),
cb76764e 130 fNEventReject(0),
b4691ee7 131 fh1Xsec(0),
132 fh1Trials(0),
133 fh1PtHard(0),
134 fh1PtHardTrials(0),
df943115 135 fPtAll(0),
136 fPtSel(0),
36c36a0c 137 fPtSelFakes(0),
138 fNPointTPCFakes(0),
139 fPtSelLargeLabel(0),
140 fMultRec(0),
141 fNPointTPCMultRec(0),
142 fDeltaPtMultRec(0),
57bee263 143 fPtAllvsPtMC(0),
fdceab34 144 fPtAllminPtMCvsPtAll(0),
57bee263 145 fPtAllvsPtMCvsMult(0),
146 fPtAllminPtMCvsPtAllvsMult(0),
fdceab34 147 fPtAllminPtMCvsPtAllNPointTPC(0),
148 fPtAllminPtMCvsPtAllDCAR(0),
149 fPtAllminPtMCvsPtAllDCAZ(0),
150 fPtAllminPtMCvsPtAllPhi(0),
151 fPtAllminPtMCvsPtAllNPointITS(0),
152 fPtAllminPtMCvsPtAllNSigmaToVertex(0),
153 fPtAllminPtMCvsPtAllChi2C(0),
154 fPtAllminPtMCvsPtAllRel1PtUncertainty(0),
155 fPtAllMC(0),
156 fPtSelMC(0),
157 fPtSelMCITS(0),
158 fHistList(0),
159 fPtSelITS(0),
160 fPtITSminPtMCvsPtITS(0),
161 fPtITSminPtMCvsPtITSNPointTPC(0),
162 fPtITSminPtMCvsPtITSDCAR(0),
163 fPtITSminPtMCvsPtITSDCAZ(0),
164 fPtITSminPtMCvsPtITSPhi(0),
165 fPtITSminPtMCvsPtITSNPointITS(0),
166 fPtITSminPtMCvsPtITSNSigmaToVertex(0),
167 fPtITSminPtMCvsPtITSChi2C(0),
168 fPtITSminPtMCvsPtITSRel1PtUncertainty(0),
169 fHistListITS(0)
170{
171 //
172 // Constructor. Initialization of Inputs and Outputs
173 //
f51451be 174 AliDebug(2,Form("AliPWG4HighPtQAMC Calling Constructor"));
fdceab34 175 // Input slot #0 works with a TChain ESD
176 DefineInput(0, TChain::Class());
b4691ee7 177 // Output slot #0, #1 write into a TList
fdceab34 178 DefineOutput(0, TList::Class());
fdceab34 179 DefineOutput(1, TList::Class());
df943115 180}
fdceab34 181
182//________________________________________________________________________
183void AliPWG4HighPtQAMC::ConnectInputData(Option_t *)
184{
b1cd0099 185 // Connect ESD and MC event handler here
fdceab34 186 // Called once
df943115 187 AliDebug(2,Form(">> AliPWG4HighPtSpectra::ConnectInputData \n"));
fdceab34 188 TTree* tree = dynamic_cast<TTree*> (GetInputData(0));
189 if (!tree) {
b1cd0099 190 AliDebug(2,Form( "ERROR: Could not read chain from input slot 0 \n"));
191 return;
fdceab34 192 }
b1cd0099 193
194 AliESDInputHandler *esdH = dynamic_cast<AliESDInputHandler*> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());
195
196 if (!esdH) {
197 AliDebug(2,Form("ERROR: Could not get ESDInputHandler"));
198 return;
199 } else
200 fESD = esdH->GetEvent();
201
cce400da 202 AliMCEventHandler *eventHandler = dynamic_cast<AliMCEventHandler*> (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());
203 if (!eventHandler) {
204 AliDebug(2,Form( "ERROR: Could not retrieve MC event handler \n"));
205 }
b1cd0099 206 else
207 fMC = eventHandler->MCEvent();
208
fdceab34 209}
210
b4691ee7 211
fdceab34 212//________________________________________________________________________
213void AliPWG4HighPtQAMC::CreateOutputObjects() {
214 //Create output objects
df943115 215 AliDebug(2,Form(">> AliPWG4HighPtQATPConly::CreateOutputObjects \n"));
216
217 Bool_t oldStatus = TH1::AddDirectoryStatus();
218 TH1::AddDirectory(kFALSE);
219
fdceab34 220 OpenFile(0);
221 fHistList = new TList();
b4691ee7 222 fHistList->SetOwner(kTRUE);
fdceab34 223 OpenFile(1);
224 fHistListITS = new TList();
b4691ee7 225 fHistListITS->SetOwner(kTRUE);
fdceab34 226
227 Int_t fgkNPhiBins=18;
228 Float_t kMinPhi = 0.;
229 Float_t kMaxPhi = 2.*TMath::Pi();
230
57bee263 231 Int_t fgkNPtBins=100;
232 Float_t fgkPtMin=0.;//2.;
b1cd0099 233 Float_t fgkPtMax=fPtMax;
8f0faa80 234 Int_t fgkResPtBins=80;
fdceab34 235
57bee263 236 Int_t fgkNMultBins = 50;
237 Float_t fgkMultMin = 0;
238 Float_t fgkMultMax = 500;
239
8f0faa80 240 fNEventAll = new TH1F("fNEventAll","NEventAll",1,-0.5,0.5);
241 fHistList->Add(fNEventAll);
242 fNEventSel = new TH1F("fNEventSel","NEvent Selected for analysis",1,-0.5,0.5);
243 fHistList->Add(fNEventSel);
cb76764e 244 fNEventReject = new TH1F("fNEventReject","Reason events are rejectected for analysis",20,0,20);
36c36a0c 245 //Set labels
246 fNEventReject->Fill("noESD",0);
247 fNEventReject->Fill("Trigger",0);
248 fNEventReject->Fill("noMCEvent",0);
11245558 249 fNEventReject->Fill("noStack",0);
36c36a0c 250 fNEventReject->Fill("NTracks<2",0);
251 fNEventReject->Fill("noVTX",0);
252 fNEventReject->Fill("VtxStatus",0);
253 fNEventReject->Fill("NCont<2",0);
254 fNEventReject->Fill("ZVTX>10",0);
cb76764e 255 fHistList->Add(fNEventReject);
b4691ee7 256
257 fh1Xsec = new TProfile("fh1Xsec","xsec from pyxsec.root",1,0,1);
258 fh1Xsec->GetXaxis()->SetBinLabel(1,"<#sigma>");
259 fHistList->Add(fh1Xsec);
260
261 fh1Trials = new TH1F("fh1Trials","trials root file",1,0,1);
262 fh1Trials->GetXaxis()->SetBinLabel(1,"#sum{ntrials}");
263 fHistList->Add(fh1Trials);
264
265 fh1PtHard = new TH1F("fh1PtHard","PYTHIA Pt hard;p_{T,hard}",350,-.5,349.5);
266 fHistList->Add(fh1PtHard);
267 fh1PtHardTrials = new TH1F("fh1PtHardTrials","PYTHIA Pt hard weight with trials;p_{T,hard}",350,-.5,349.5);
268 fHistList->Add(fh1PtHardTrials);
269
fdceab34 270 fPtAll = new TH1F("fPtAll","PtAll",fgkNPtBins, fgkPtMin, fgkPtMax);
271 fHistList->Add(fPtAll);
272 fPtSel = new TH1F("fPtSel","PtSel",fgkNPtBins, fgkPtMin, fgkPtMax);
273 fHistList->Add(fPtSel);
36c36a0c 274
275 fPtSelFakes = new TH1F("fPtSelFakes","PtSelFakes",fgkNPtBins, fgkPtMin, fgkPtMax);
276 fHistList->Add(fPtSelFakes);
277
278 fNPointTPCFakes = new TH1F("fNPointTPCFakes","fNPointTPCFakes",160,0.5,160.5);
279 fHistList->Add(fNPointTPCFakes);
280
281 fPtSelLargeLabel = new TH1F("fPtSelLargeLabel","PtSelLargeLabel",fgkNPtBins, fgkPtMin, fgkPtMax);
282 fHistList->Add(fPtSelLargeLabel);
283
284 fMultRec = new TH1F("fMultRec","Multiple reconstruction of tracks",20,0,20);
285 fHistList->Add(fMultRec);
286
287 fNPointTPCMultRec = new TH1F("fNPointTPCMultRec","Multiple reconstruction of tracks",160,0.5,160.5);
288 fHistList->Add(fNPointTPCMultRec);
289
290 fDeltaPtMultRec = new TH2F("fDeltaPtMultRec","Delta pT vs pT for multiple reconstructed tracks",100,0.,50.,100,-20.,20.);
291 fHistList->Add(fDeltaPtMultRec);
292
57bee263 293 fPtAllvsPtMC = new TH2F("fPtAllvsPtMC","fPtAllvsPtMC;p_{T,MC};p_{T,rec}",fgkNPtBins, fgkPtMin,fgkPtMax,fgkNPtBins, fgkPtMin,fgkPtMax);
294 fHistList->Add(fPtAllvsPtMC);
295
fdceab34 296 fPtAllminPtMCvsPtAll = new TH2F("fPtAllminPtMCvsPtAll","PtAllminPtMCvsPtAll",fgkNPtBins, fgkPtMin,fgkPtMax,fgkResPtBins,-1.,1.);
297 fPtAllminPtMCvsPtAll->SetXTitle("p_{t}^{MC}");
298 fPtAllminPtMCvsPtAll->SetYTitle("(1/p_{t}^{All}-1/p_{t}^{MC})/(1/p_{t}^{MC})");
299 fHistList->Add(fPtAllminPtMCvsPtAll);
57bee263 300
301 fPtAllvsPtMCvsMult = new TH3F("fPtAllvsPtMCvsMult","fPtAllvsPtMCvsMult;p_{T,MC};p_{T,rec};N_{tracks}",fgkNPtBins, fgkPtMin,fgkPtMax,fgkNPtBins, fgkPtMin,fgkPtMax,fgkNMultBins,fgkMultMin,fgkMultMax);
302 fHistList->Add(fPtAllvsPtMCvsMult);
303
304 fPtAllminPtMCvsPtAllvsMult = new TH3F("fPtAllminPtMCvsPtAllvsMult","fPtAllminPtMCvsPtAllvsMult",fgkNPtBins, fgkPtMin,fgkPtMax,fgkResPtBins,-1.,1.,fgkNMultBins,fgkMultMin,fgkMultMax);
305 fPtAllminPtMCvsPtAllvsMult->SetXTitle("p_{t}^{MC}");
306 fPtAllminPtMCvsPtAllvsMult->SetYTitle("(1/p_{t}^{All}-1/p_{t}^{MC})/(1/p_{t}^{MC})");
307 fPtAllminPtMCvsPtAllvsMult->SetZTitle("N_{tracks}");
308 fHistList->Add(fPtAllminPtMCvsPtAllvsMult);
fdceab34 309
310 fPtAllminPtMCvsPtAllNPointTPC = new TH3F("fPtAllminPtMCvsPtAllNPointTPC","PtAllminPtMCvsPtAllNPointTPC",fgkNPtBins, fgkPtMin,fgkPtMax,fgkResPtBins,-1.,1.,160,0.5,160.5);
311 fPtAllminPtMCvsPtAllNPointTPC->SetXTitle("p_{t}^{MC}");
312 fPtAllminPtMCvsPtAllNPointTPC->SetYTitle("(1/p_{t}^{All}-1/p_{t}^{MC})/(1/p_{t}^{MC})");
313 fPtAllminPtMCvsPtAllNPointTPC->SetZTitle("N_{point,TPC}");
314 fHistList->Add(fPtAllminPtMCvsPtAllNPointTPC);
315
8f0faa80 316 fPtAllminPtMCvsPtAllDCAR = new TH3F("fPtAllminPtMCvsPtAllDCAR","PtAllminPtMCvsPtAllDCAR",fgkNPtBins, fgkPtMin,fgkPtMax,fgkResPtBins,-1.,1.,80,-0.2,0.2);
fdceab34 317 fPtAllminPtMCvsPtAllDCAR->SetXTitle("p_{t}^{MC}");
318 fPtAllminPtMCvsPtAllDCAR->SetYTitle("(1/p_{t}^{All}-1/p_{t}^{MC})/(1/p_{t}^{MC})");
319 fPtAllminPtMCvsPtAllDCAR->SetZTitle("DCA_{R}");
320 fHistList->Add(fPtAllminPtMCvsPtAllDCAR);
321
8f0faa80 322 fPtAllminPtMCvsPtAllDCAZ = new TH3F("fPtAllminPtMCvsPtAllDCAZ","PtAllminPtMCvsPtAllDCAZ",fgkNPtBins, fgkPtMin,fgkPtMax,fgkResPtBins,-1.,1.,80,-2.,2.);
fdceab34 323 fPtAllminPtMCvsPtAllDCAZ->SetXTitle("p_{t}^{MC}");
324 fPtAllminPtMCvsPtAllDCAZ->SetYTitle("(1/p_{t}^{All}-1/p_{t}^{MC})/(1/p_{t}^{MC})");
325 fPtAllminPtMCvsPtAllDCAZ->SetZTitle("DCA_{Z}");
326 fHistList->Add(fPtAllminPtMCvsPtAllDCAZ);
327
328 fPtAllminPtMCvsPtAllPhi = new TH3F("fPtAllminPtMCvsPtAllPhi","PtAllminPtMCvsPtAllPhi",fgkNPtBins, fgkPtMin,fgkPtMax,fgkResPtBins,-1.,1.,fgkNPhiBins,kMinPhi,kMaxPhi);
329 fPtAllminPtMCvsPtAllPhi->SetXTitle("p_{t}^{MC}");
330 fPtAllminPtMCvsPtAllPhi->SetYTitle("(1/p_{t}^{All}-1/p_{t}^{MC})/(1/p_{t}^{MC})");
331 fPtAllminPtMCvsPtAllPhi->SetZTitle("#phi");
332 fHistList->Add(fPtAllminPtMCvsPtAllPhi);
333
334 fPtAllminPtMCvsPtAllNPointITS = new TH3F("fPtAllminPtMCvsPtAllNPointITS","PtAllminPtMCvsPtAllNPointITS",fgkNPtBins, fgkPtMin,fgkPtMax,fgkResPtBins,-1.,1.,9,-0.5,8.5);
335 fPtAllminPtMCvsPtAllNPointITS->SetXTitle("p_{t}^{MC}");
336 fPtAllminPtMCvsPtAllNPointITS->SetYTitle("(1/p_{t}^{All}-1/p_{t}^{MC})/(1/p_{t}^{MC})");
337 fPtAllminPtMCvsPtAllNPointITS->SetZTitle("N_{point,ITS}}");
338 fHistList->Add(fPtAllminPtMCvsPtAllNPointITS);
339
340 fPtAllminPtMCvsPtAllNSigmaToVertex = new TH3F("fPtAllminPtMCvsPtAllNSigmaToVertex","PtAllminPtMCvsPtAllNSigmaToVertex",fgkNPtBins, fgkPtMin,fgkPtMax,fgkResPtBins,-1.,1.,40,0.,8.);
341 fPtAllminPtMCvsPtAllNSigmaToVertex->SetXTitle("p_{t}^{MC}");
342 fPtAllminPtMCvsPtAllNSigmaToVertex->SetYTitle("(1/p_{t}^{All}-1/p_{t}^{MC})/(1/p_{t}^{MC})");
343 fPtAllminPtMCvsPtAllNSigmaToVertex->SetZTitle("N#sigma to vertex");
344 fHistList->Add(fPtAllminPtMCvsPtAllNSigmaToVertex);
345
346 fPtAllminPtMCvsPtAllChi2C = new TH3F("fPtAllminPtMCvsPtAllChi2C","PtAllminPtMCvsPtAllChi2C",fgkNPtBins, fgkPtMin,fgkPtMax,fgkResPtBins,-1.,1.,20,0.,10.);
347 fPtAllminPtMCvsPtAllChi2C->SetXTitle("p_{t}^{MC}");
348 fPtAllminPtMCvsPtAllChi2C->SetYTitle("(1/p_{t}^{All}-1/p_{t}^{MC})/(1/p_{t}^{MC})");
349 fPtAllminPtMCvsPtAllChi2C->SetZTitle("Constrained #chi^{2}");
350 fHistList->Add(fPtAllminPtMCvsPtAllChi2C);
351
352 fPtAllminPtMCvsPtAllRel1PtUncertainty = new TH3F("fPtAllminPtMCvsPtAllRel1PtUncertainty","PtAllminPtMCvsPtAllRel1PtUncertainty",fgkNPtBins, fgkPtMin,fgkPtMax,fgkResPtBins,-1.,1.,30,0.,0.3);
353 fPtAllminPtMCvsPtAllRel1PtUncertainty->SetXTitle("p_{t}^{MC}");
354 fPtAllminPtMCvsPtAllRel1PtUncertainty->SetYTitle("(1/p_{t}^{All}-1/p_{t}^{MC})/(1/p_{t}^{MC})");
355 fPtAllminPtMCvsPtAllRel1PtUncertainty->SetZTitle("Rel1PtUncertainty");
356 fHistList->Add(fPtAllminPtMCvsPtAllRel1PtUncertainty);
357
358 //ITSrefit
359 fPtSelITS = new TH1F("fPtSelITSrefit","PtSel",fgkNPtBins, fgkPtMin, fgkPtMax);
360 fHistListITS->Add(fPtSelITS);
361
362 fPtITSminPtMCvsPtITS = new TH2F("fPtITSminPtMCvsPtITS","PtITSminPtMCvsPtITS",fgkNPtBins, fgkPtMin,fgkPtMax,fgkResPtBins,-1.,1.);
363 fPtITSminPtMCvsPtITS->SetXTitle("p_{t}^{MC}");
364 fPtITSminPtMCvsPtITS->SetYTitle("(1/p_{t}^{ITS}-1/p_{t}^{MC})/(1/p_{t}^{MC})");
365 fHistListITS->Add(fPtITSminPtMCvsPtITS);
366
367 fPtITSminPtMCvsPtITSNPointTPC = new TH3F("fPtITSminPtMCvsPtITSNPointTPC","PtITSminPtMCvsPtITSNPointTPC",fgkNPtBins, fgkPtMin,fgkPtMax,fgkResPtBins,-1.,1.,160,0.5,160.5);
368 fPtITSminPtMCvsPtITSNPointTPC->SetXTitle("p_{t}^{MC}");
369 fPtITSminPtMCvsPtITSNPointTPC->SetYTitle("(1/p_{t}^{ITS}-1/p_{t}^{MC})/(1/p_{t}^{MC})");
370 fPtITSminPtMCvsPtITSNPointTPC->SetZTitle("N_{point,TPC}");
371 fHistListITS->Add(fPtITSminPtMCvsPtITSNPointTPC);
372
8f0faa80 373 fPtITSminPtMCvsPtITSDCAR = new TH3F("fPtITSminPtMCvsPtITSDCAR","PtITSminPtMCvsPtITSDCAR",fgkNPtBins, fgkPtMin,fgkPtMax,fgkResPtBins,-1.,1.,80,-0.2,0.2);
fdceab34 374 fPtITSminPtMCvsPtITSDCAR->SetXTitle("p_{t}^{MC}");
375 fPtITSminPtMCvsPtITSDCAR->SetYTitle("(1/p_{t}^{ITS}-1/p_{t}^{MC})/(1/p_{t}^{MC})");
376 fPtITSminPtMCvsPtITSDCAR->SetZTitle("DCA_{R}");
377 fHistListITS->Add(fPtITSminPtMCvsPtITSDCAR);
378
8f0faa80 379 fPtITSminPtMCvsPtITSDCAZ = new TH3F("fPtITSminPtMCvsPtITSDCAZ","PtITSminPtMCvsPtITSDCAZ",fgkNPtBins, fgkPtMin,fgkPtMax,fgkResPtBins,-1.,1.,80,-2.,2.);
fdceab34 380 fPtITSminPtMCvsPtITSDCAZ->SetXTitle("p_{t}^{MC}");
381 fPtITSminPtMCvsPtITSDCAZ->SetYTitle("(1/p_{t}^{ITS}-1/p_{t}^{MC})/(1/p_{t}^{MC})");
382 fPtITSminPtMCvsPtITSDCAZ->SetZTitle("DCA_{Z}");
383 fHistListITS->Add(fPtITSminPtMCvsPtITSDCAZ);
384
385 fPtITSminPtMCvsPtITSPhi = new TH3F("fPtITSminPtMCvsPtITSPhi","PtITSminPtMCvsPtITSPhi",fgkNPtBins, fgkPtMin,fgkPtMax,fgkResPtBins,-1.,1.,fgkNPhiBins,kMinPhi,kMaxPhi);
386 fPtITSminPtMCvsPtITSPhi->SetXTitle("p_{t}^{MC}");
387 fPtITSminPtMCvsPtITSPhi->SetYTitle("(1/p_{t}^{ITS}-1/p_{t}^{MC})/(1/p_{t}^{MC})");
388 fPtITSminPtMCvsPtITSPhi->SetZTitle("#phi");
389 fHistListITS->Add(fPtITSminPtMCvsPtITSPhi);
390
391 fPtITSminPtMCvsPtITSNPointITS = new TH3F("fPtITSminPtMCvsPtITSNPointITS","PtITSminPtMCvsPtITSNPointITS",fgkNPtBins, fgkPtMin,fgkPtMax,fgkResPtBins,-1.,1.,9,-0.5,8.5);
392 fPtITSminPtMCvsPtITSNPointITS->SetXTitle("p_{t}^{MC}");
393 fPtITSminPtMCvsPtITSNPointITS->SetYTitle("(1/p_{t}^{ITS}-1/p_{t}^{MC})/(1/p_{t}^{MC})}");
394 fPtITSminPtMCvsPtITSNPointITS->SetZTitle("N_{point,ITS}}");
395 fHistListITS->Add(fPtITSminPtMCvsPtITSNPointITS);
396
397 fPtITSminPtMCvsPtITSNSigmaToVertex = new TH3F("fPtITSminPtMCvsPtITSNSigmaToVertex","PtITSminPtMCvsPtITSNSigmaToVertex",fgkNPtBins, fgkPtMin,fgkPtMax,fgkResPtBins,-1.,1.,40,0.,8.);
398 fPtITSminPtMCvsPtITSNSigmaToVertex->SetXTitle("p_{t}^{MC}");
399 fPtITSminPtMCvsPtITSNSigmaToVertex->SetYTitle("(1/p_{t}^{ITS}-1/p_{t}^{MC})/(1/p_{t}^{MC})");
400 fPtITSminPtMCvsPtITSNSigmaToVertex->SetZTitle("N#sigma to vertex");
401 fHistListITS->Add(fPtITSminPtMCvsPtITSNSigmaToVertex);
402
403 fPtITSminPtMCvsPtITSChi2C = new TH3F("fPtITSminPtMCvsPtITSChi2C","PtITSminPtMCvsPtITSChi2C",fgkNPtBins, fgkPtMin,fgkPtMax,fgkResPtBins,-1.,1.,20,0.,10.);
404 fPtITSminPtMCvsPtITSChi2C->SetXTitle("p_{t}^{MC}");
405 fPtITSminPtMCvsPtITSChi2C->SetYTitle("(1/p_{t}^{ITS}-1/p_{t}^{MC})/(1/p_{t}^{MC})");
406 fPtITSminPtMCvsPtITSChi2C->SetZTitle("Constrained #chi^{2}");
407 fHistListITS->Add(fPtITSminPtMCvsPtITSChi2C);
408
409 fPtITSminPtMCvsPtITSRel1PtUncertainty = new TH3F("fPtITSminPtMCvsPtITSRel1PtUncertainty","PtITSminPtMCvsPtITSRel1PtUncertainty",fgkNPtBins, fgkPtMin,fgkPtMax,fgkResPtBins,-1.,1.,30,0.,0.3);
410 fPtITSminPtMCvsPtITSRel1PtUncertainty->SetXTitle("p_{t}^{MC}");
411 fPtITSminPtMCvsPtITSRel1PtUncertainty->SetYTitle("(1/p_{t}^{ITS}-1/p_{t}^{MC})/(1/p_{t}^{MC})");
412 fPtITSminPtMCvsPtITSRel1PtUncertainty->SetZTitle("Rel1PtUncertainty");
413 fHistListITS->Add(fPtITSminPtMCvsPtITSRel1PtUncertainty);
414
415 fPtAllMC = new TH1F("fPtAllMC","PtAll",fgkNPtBins, fgkPtMin, fgkPtMax);
416 fHistList->Add(fPtAllMC);
417 fPtSelMC = new TH1F("fPtSelMC","PtSel",fgkNPtBins, fgkPtMin, fgkPtMax);
418 fHistList->Add(fPtSelMC);
419 fPtSelMCITS = new TH1F("fPtSelMCITS","PtSel",fgkNPtBins, fgkPtMin, fgkPtMax);
420 fHistList->Add(fPtSelMCITS);
421
df943115 422 TH1::AddDirectory(oldStatus);
423
11245558 424 PostData(0, fHistList);
425 PostData(1, fHistListITS);
426
fdceab34 427}
cce400da 428
fdceab34 429//________________________________________________________________________
cce400da 430Bool_t AliPWG4HighPtQAMC::SelectEvent() {
431 //
432 // Decide if event should be selected for analysis
433 //
8f0faa80 434
cce400da 435 // Checks following requirements:
436 // - fESD available
437 // - trigger info from AliPhysicsSelection
438 // - MCevent available
439 // - number of reconstructed tracks > 1
440 // - primary vertex reconstructed
441 // - z-vertex < 10 cm
442
443 Bool_t selectEvent = kTRUE;
444
445 //fESD object available?
fdceab34 446 if (!fESD) {
b4691ee7 447 AliDebug(2,Form("ERROR: fInputEvent not available\n"));
cb76764e 448 fNEventReject->Fill("noESD",1);
cce400da 449 selectEvent = kFALSE;
450 return selectEvent;
fdceab34 451 }
452
cce400da 453 //Trigger
bfd58011 454 UInt_t isSelected = ((AliInputEventHandler*)(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler()))->IsEventSelected();
455 if(!(isSelected&AliVEvent::kMB)) { //Select collison candidates
8f0faa80 456 AliDebug(2,Form(" Trigger Selection: event REJECTED ... "));
cb76764e 457 fNEventReject->Fill("Trigger",1);
cce400da 458 selectEvent = kFALSE;
459 return selectEvent;
8f0faa80 460 }
cce400da 461
462 //MCEvent available?
463 //if yes: get stack
b1cd0099 464 if(fMC) {
465 AliDebug(2,Form("MC particles: %d", fMC->GetNumberOfTracks()));
cce400da 466 fStack = fMC->Stack(); //Particles Stack
11245558 467 if(fStack) {
468 AliDebug(2,Form("MC particles stack: %d", fStack->GetNtrack()));
469 }
470 else {
471 AliDebug(2,Form("ERROR: Could not retrieve MC eventHandler"));
472 fNEventReject->Fill("noStack",1);
473 selectEvent = kFALSE;
474 return selectEvent;
475 }
b1cd0099 476 } else {
11245558 477 AliDebug(2,Form("ERROR: Could not retrieve stack"));
cb76764e 478 fNEventReject->Fill("noMCEvent",1);
cce400da 479 selectEvent = kFALSE;
480 return selectEvent;
fdceab34 481 }
482
cce400da 483 //Check if number of reconstructed tracks is larger than 1
484 if(!fESD->GetNumberOfTracks() || fESD->GetNumberOfTracks()<2) {
485 fNEventReject->Fill("NTracks<2",1);
486 selectEvent = kFALSE;
487 return selectEvent;
cb76764e 488 }
b4691ee7 489
cce400da 490 //Check if vertex is reconstructed
36c36a0c 491 if(fTrackType==1) fVtx = fESD->GetPrimaryVertexTPC();
492 else fVtx = fESD->GetPrimaryVertexSPD();
493
494 if(!fVtx) {
495 fNEventReject->Fill("noVTX",1);
496 selectEvent = kFALSE;
497 return selectEvent;
498 }
499
500 if(!fVtx->GetStatus()) {
501 fNEventReject->Fill("VtxStatus",1);
502 selectEvent = kFALSE;
503 return selectEvent;
504 }
505
fdceab34 506 // Need vertex cut
36c36a0c 507 // TString vtxName(fVtx->GetName());
508 if(fVtx->GetNContributors()<2) {
509 fNEventReject->Fill("NCont<2",1);
510 selectEvent = kFALSE;
511 return selectEvent;
8f0faa80 512 }
36c36a0c 513
cce400da 514 //Check if z-vertex < 10 cm
75946d5d 515 double primVtx[3];
36c36a0c 516 fVtx->GetXYZ(primVtx);
8f0faa80 517 if(TMath::Sqrt(primVtx[0]*primVtx[0] + primVtx[1]*primVtx[1])>1. || TMath::Abs(primVtx[2]>10.)){
cb76764e 518 fNEventReject->Fill("ZVTX>10",1);
cce400da 519 selectEvent = kFALSE;
520 return selectEvent;
8f0faa80 521 }
36c36a0c 522
523 AliDebug(2,Form("Vertex title %s, status %d, nCont %d\n",fVtx->GetTitle(), fVtx->GetStatus(), fVtx->GetNContributors()));
fdceab34 524
cce400da 525 return selectEvent;
526
527}
528
529//________________________________________________________________________
530void AliPWG4HighPtQAMC::Exec(Option_t *) {
531 // Main loop
532 // Called for each event
533 AliDebug(2,Form(">> AliPWG4HighPtQATPConly::Exec \n"));
534 // All events without selection
535 fNEventAll->Fill(0.);
536
537 if(!SelectEvent()) {
538 // Post output data
8f0faa80 539 PostData(0, fHistList);
540 PostData(1, fHistListITS);
541 return;
542 }
fdceab34 543
cce400da 544 // ---- Get MC Header information (for MC productions in pThard bins) ----
545 Double_t ptHard = 0.;
546 Double_t nTrials = 1; // trials for MC trigger weight for real data
547
548 if(fMC){
549 AliGenPythiaEventHeader* pythiaGenHeader = GetPythiaEventHeader(fMC);
550 if(pythiaGenHeader){
551 nTrials = pythiaGenHeader->Trials();
552 ptHard = pythiaGenHeader->GetPtHard();
553
554 fh1PtHard->Fill(ptHard);
555 fh1PtHardTrials->Fill(ptHard,nTrials);
556
557 fh1Trials->Fill("#sum{ntrials}",fAvgTrials);
558 }
559 }
560
b4691ee7 561 //Need to keep track of selected events
562 fNEventSel->Fill(0.);
563
fdceab34 564 Int_t nTracks = fESD->GetNumberOfTracks();
b1cd0099 565 AliDebug(2,Form("nTracks ESD%d", nTracks));
df943115 566
cce400da 567 int nMCtracks = fStack->GetNtrack();
cba109a0 568
71e77a79 569 Float_t pt = 0.;
570 Float_t ptMC = 0.;
571 Float_t phi = 0.;
572 Float_t dca2D = 0.;
573 Float_t dcaZ = 0.;
574 Int_t nPointITS = 0;
575 Float_t chi2C = 0.;
576 Float_t nSigmaToVertex = 0.;
577 Float_t relUncertainty1Pt = 0.;
578
fdceab34 579 for (Int_t iTrack = 0; iTrack < nTracks; iTrack++) {
580
71e77a79 581 AliESDtrack *track;
582 AliESDtrack *esdtrack = fESD->GetTrack(iTrack);
583 if(!esdtrack) continue;
71e77a79 584
d756027f 585 if(fTrackType==1 || fTrackType==3)
71e77a79 586 track = AliESDtrackCuts::GetTPCOnlyTrack(fESD,esdtrack->GetID());
36c36a0c 587 else if(fTrackType==2) {
588 track = AliESDtrackCuts::GetTPCOnlyTrack(fESD,esdtrack->GetID());
589 if(!track) {
590 delete track;
591 continue;
592 }
593 AliExternalTrackParam exParam;
594 Bool_t relate = track->RelateToVertexTPC(fVtx,fESD->GetMagneticField(),kVeryBig,&exParam);
595 if( !relate ) {
596 delete track;
597 continue;
598 }
599 track->Set(exParam.GetX(),exParam.GetAlpha(),exParam.GetParameter(),exParam.GetCovariance());
600 }
71e77a79 601 else
602 track = esdtrack;
71e77a79 603
36c36a0c 604 if(!track) {
605 if(fTrackType==1 || fTrackType==2) delete track;
606 continue;
607 }
71e77a79 608
aa3ba8d2 609 if(fTrackType==2) {
610 //Cut on chi2 of constrained fit
611 if(track->GetConstrainedChi2TPC() > fSigmaConstrainedMax*fSigmaConstrainedMax) {
612 delete track;
613 continue;
614 }
615 }
616
fdceab34 617 Int_t label = TMath::Abs(track->GetLabel());
36c36a0c 618 if(label>=nMCtracks) {
619 if (fTrackCuts->AcceptTrack(track)) { fPtSelLargeLabel->Fill(pt); }
620 if(fTrackType==1 || fTrackType==2) delete track;
621 continue;
622 }
cce400da 623 TParticle *particle = fStack->Particle(label) ;
36c36a0c 624 if(!particle) {
625 if(fTrackType==1 || fTrackType==2) delete track;
626 continue;
627 }
fdceab34 628
71e77a79 629 ptMC = particle->Pt();
630
631 if(fTrackType==0) { //Global
632 pt = track->Pt();
633 phi = track->Phi();
634 track->GetImpactParameters(dca2D,dcaZ);
635 }
36c36a0c 636 else if(fTrackType==1 || fTrackType==2) { //TPConly
637 pt = track->Pt();
638 phi = track->Phi();
71e77a79 639 track->GetImpactParametersTPC(dca2D,dcaZ);
640 }
641 else {continue;}
642
643
fdceab34 644 UChar_t itsMap = track->GetITSClusterMap();
fdceab34 645 for (Int_t i=0; i < 6; i++) {
646 if (itsMap & (1 << i))
647 nPointITS ++;
648 }
71e77a79 649 nSigmaToVertex = fTrackCuts->GetSigmaToVertex(track);// Calculates the number of sigma to the vertex for a track.
650 chi2C = track->GetConstrainedChi2();
651 relUncertainty1Pt = TMath::Sqrt(track->GetSigma1Pt2())*pt;
652
fdceab34 653 fPtAll->Fill(pt);
654 fPtAllMC->Fill(ptMC);
71e77a79 655
fdceab34 656 if (fTrackCuts->AcceptTrack(track)) {
657
658 fPtSel->Fill(pt);
36c36a0c 659 if(track->GetLabel()<0) {
660 fPtSelFakes->Fill(pt);
661 fNPointTPCFakes->Fill(track->GetTPCNcls());
662 }
fdceab34 663 fPtSelMC->Fill(ptMC);
57bee263 664 fPtAllvsPtMC->Fill(ptMC,pt);
fdceab34 665 fPtAllminPtMCvsPtAll->Fill(ptMC,(1./pt-1./ptMC)/(1./ptMC) );
666 fPtAllminPtMCvsPtAllNPointTPC->Fill(ptMC,(1./pt-1./ptMC)/(1./ptMC),track->GetTPCNcls());
667 fPtAllminPtMCvsPtAllDCAR->Fill(ptMC,(1./pt-1./ptMC)/(1./ptMC),dca2D);
668 fPtAllminPtMCvsPtAllDCAZ->Fill(ptMC,(1./pt-1./ptMC)/(1./ptMC),dcaZ);
669 fPtAllminPtMCvsPtAllPhi->Fill(ptMC,(1./pt-1./ptMC)/(1./ptMC),phi);
670 fPtAllminPtMCvsPtAllNPointITS->Fill(ptMC,(1./pt-1./ptMC)/(1./ptMC),nPointITS);
671 fPtAllminPtMCvsPtAllNSigmaToVertex->Fill(ptMC,(1./pt-1./ptMC)/(1./ptMC),nSigmaToVertex);
672 fPtAllminPtMCvsPtAllChi2C->Fill(ptMC,(1./pt-1./ptMC)/(1./ptMC),chi2C);
673 fPtAllminPtMCvsPtAllRel1PtUncertainty->Fill(ptMC,(1./pt-1./ptMC)/(1./ptMC),relUncertainty1Pt);
36c36a0c 674
57bee263 675 int mult = fTrackCuts->CountAcceptedTracks(fESD);
676 fPtAllvsPtMCvsMult->Fill(ptMC,pt,mult);
677 fPtAllminPtMCvsPtAllvsMult->Fill(ptMC,(1./pt-1./ptMC)/(1./ptMC), mult);
678
36c36a0c 679 //Check if track is reconstructed multiple times
680 int multCounter = 1;
681 for (Int_t iTrack2 = iTrack+1; iTrack2 < nTracks; iTrack2++) {
682 // AliESDtrack *track2 = GetTrackForAnalysis(iTrack2);
683 AliESDtrack *track2;
684 AliESDtrack *esdtrack2 = fESD->GetTrack(iTrack2);
685 if(!esdtrack2) continue;
686
687 if(fTrackType==1)
688 track2 = AliESDtrackCuts::GetTPCOnlyTrack(fESD,esdtrack2->GetID());
689 else if(fTrackType==2) {
690 track2 = AliESDtrackCuts::GetTPCOnlyTrack(fESD,esdtrack2->GetID());
691 if(!track2) {
692 delete track2;
693 continue;
694 }
695 AliExternalTrackParam exParam2;
696 Bool_t relate = track2->RelateToVertexTPC(fVtx,fESD->GetMagneticField(),kVeryBig,&exParam2);
697 if( !relate ) {
698 delete track2;
699 continue;
700 }
701 track2->Set(exParam2.GetX(),exParam2.GetAlpha(),exParam2.GetParameter(),exParam2.GetCovariance());
702 }
703 else
704 track2 = esdtrack2;
705 if(!track2) {
706 if(fTrackType==1 || fTrackType==2) delete track2;
707 continue;
708 }
709
710 if (fTrackCuts->AcceptTrack(track2)) {
711 Int_t label2 = TMath::Abs(track2->GetLabel());
712 if(label==label2) {
713 fNPointTPCMultRec->Fill(track->GetTPCNcls());
714 fNPointTPCMultRec->Fill(track2->GetTPCNcls());
715 fDeltaPtMultRec->Fill(track->Pt(),track->Pt()-track2->Pt());
716 multCounter++;
717 }
718 }
719 if(fTrackType==1 || fTrackType==2) delete track2;
720 }//track2 loop
721 if(multCounter>1) fMultRec->Fill(multCounter);
722
723 }//fTrackCuts selection
724
fdceab34 725 //ITSrefit selection
71e77a79 726 if (fTrackCutsITS->AcceptTrack(track) && fTrackType==0) {
fdceab34 727
728 fPtSelITS->Fill(pt);
729 fPtSelMCITS->Fill(ptMC);
730 fPtITSminPtMCvsPtITS->Fill(ptMC,(1./pt-1./ptMC)/(1./ptMC) );
731 fPtITSminPtMCvsPtITSNPointTPC->Fill(ptMC,(1./pt-1./ptMC)/(1./ptMC),track->GetTPCNcls());
732 fPtITSminPtMCvsPtITSDCAR->Fill(ptMC,(1./pt-1./ptMC)/(1./ptMC),dca2D);
733 fPtITSminPtMCvsPtITSDCAZ->Fill(ptMC,(1./pt-1./ptMC)/(1./ptMC),dcaZ);
734 fPtITSminPtMCvsPtITSPhi->Fill(ptMC,(pt-ptMC)/(pt),phi);
735 fPtITSminPtMCvsPtITSNPointITS->Fill(ptMC,(pt-ptMC)/(pt),nPointITS);
736 fPtITSminPtMCvsPtITSNSigmaToVertex->Fill(ptMC,(1./pt-1./ptMC)/(1./ptMC),nSigmaToVertex);
737 fPtITSminPtMCvsPtITSChi2C->Fill(ptMC,(1./pt-1./ptMC)/(1./ptMC),chi2C);
738 fPtITSminPtMCvsPtITSRel1PtUncertainty->Fill(ptMC,(1./pt-1./ptMC)/(1./ptMC),relUncertainty1Pt);
739 }//fTrackCutsITS loop
36c36a0c 740
741 if(fTrackType==1 || fTrackType==2) delete track;
742
fdceab34 743 }//ESD track loop
744
745 // Post output data
746 PostData(0, fHistList);
747 PostData(1, fHistListITS);
748
749}
b4691ee7 750//________________________________________________________________________
751Bool_t AliPWG4HighPtQAMC::PythiaInfoFromFile(const char* currFile,Float_t &fXsec,Float_t &fTrials){
752 //
753 // get the cross section and the trails either from pyxsec.root or from pysec_hists.root
754 // This is to called in Notify and should provide the path to the AOD/ESD file
755 // Copied from AliAnalysisTaskJetSpectrum2
756 //
757
758 TString file(currFile);
759 fXsec = 0;
760 fTrials = 1;
761
762 if(file.Contains("root_archive.zip#")){
763 Ssiz_t pos1 = file.Index("root_archive",12,TString::kExact);
764 Ssiz_t pos = file.Index("#",1,pos1,TString::kExact);
765 file.Replace(pos+1,20,"");
766 }
767 else {
768 // not an archive take the basename....
769 file.ReplaceAll(gSystem->BaseName(file.Data()),"");
770 }
cb76764e 771 // Printf("%s",file.Data());
b4691ee7 772
773
774 TFile *fxsec = TFile::Open(Form("%s%s",file.Data(),"pyxsec.root")); // problem that we cannot really test the existance of a file in a archive so we have to lvie with open error message from root
775 if(!fxsec){
776 // next trial fetch the histgram file
777 fxsec = TFile::Open(Form("%s%s",file.Data(),"pyxsec_hists.root"));
778 if(!fxsec){
779 // not a severe condition but inciate that we have no information
780 return kFALSE;
781 }
782 else{
783 // find the tlist we want to be independtent of the name so use the Tkey
784 TKey* key = (TKey*)fxsec->GetListOfKeys()->At(0);
785 if(!key){
786 fxsec->Close();
787 return kFALSE;
788 }
789 TList *list = dynamic_cast<TList*>(key->ReadObj());
790 if(!list){
791 fxsec->Close();
792 return kFALSE;
793 }
794 fXsec = ((TProfile*)list->FindObject("h1Xsec"))->GetBinContent(1);
795 fTrials = ((TH1F*)list->FindObject("h1Trials"))->GetBinContent(1);
796 fxsec->Close();
797 }
798 } // no tree pyxsec.root
799 else {
800 TTree *xtree = (TTree*)fxsec->Get("Xsection");
801 if(!xtree){
802 fxsec->Close();
803 return kFALSE;
804 }
805 UInt_t ntrials = 0;
806 Double_t xsection = 0;
807 xtree->SetBranchAddress("xsection",&xsection);
808 xtree->SetBranchAddress("ntrials",&ntrials);
809 xtree->GetEntry(0);
810 fTrials = ntrials;
811 fXsec = xsection;
812 fxsec->Close();
813 }
814 return kTRUE;
815}
816//________________________________________________________________________
817Bool_t AliPWG4HighPtQAMC::Notify()
818{
819 //
820 // Implemented Notify() to read the cross sections
821 // and number of trials from pyxsec.root
822 // Copied from AliAnalysisTaskJetSpectrum2
823 //
824
825 TTree *tree = AliAnalysisManager::GetAnalysisManager()->GetTree();
826 Float_t xsection = 0;
827 Float_t ftrials = 1;
828
829 fAvgTrials = 1;
830 if(tree){
831 TFile *curfile = tree->GetCurrentFile();
832 if (!curfile) {
833 Error("Notify","No current file");
834 return kFALSE;
835 }
836 if(!fh1Xsec||!fh1Trials){
cb76764e 837 // Printf("%s%d No Histogram fh1Xsec",(char*)__FILE__,__LINE__);
b4691ee7 838 return kFALSE;
839 }
840 PythiaInfoFromFile(curfile->GetName(),xsection,ftrials);
841 fh1Xsec->Fill("<#sigma>",xsection);
842 // construct a poor man average trials
843 Float_t nEntries = (Float_t)tree->GetTree()->GetEntries();
844 if(ftrials>=nEntries && nEntries>0.)fAvgTrials = ftrials/nEntries;
845 }
846 return kTRUE;
847}
848
849//________________________________________________________________________
850AliGenPythiaEventHeader* AliPWG4HighPtQAMC::GetPythiaEventHeader(AliMCEvent *mcEvent){
851
852 if(!mcEvent)return 0;
853 AliGenEventHeader* genHeader = mcEvent->GenEventHeader();
854 AliGenPythiaEventHeader* pythiaGenHeader = dynamic_cast<AliGenPythiaEventHeader*>(genHeader);
855 if(!pythiaGenHeader){
856 // cocktail ??
857 AliGenCocktailEventHeader* genCocktailHeader = dynamic_cast<AliGenCocktailEventHeader*>(genHeader);
858
859 if (!genCocktailHeader) {
cb76764e 860 // AliWarningGeneral(Form(" %s:%d",(char*)__FILE__,__LINE__),"Unknown header type (not Pythia or Cocktail)");
b4691ee7 861 // AliWarning(Form("%s %d: Unknown header type (not Pythia or Cocktail)",(char*)__FILE__,__LINE__));
862 return 0;
863 }
864 TList* headerList = genCocktailHeader->GetHeaders();
865 for (Int_t i=0; i<headerList->GetEntries(); i++) {
866 pythiaGenHeader = dynamic_cast<AliGenPythiaEventHeader*>(headerList->At(i));
867 if (pythiaGenHeader)
868 break;
869 }
870 if(!pythiaGenHeader){
871 AliWarningGeneral(Form(" %s:%d",(char*)__FILE__,__LINE__),"Pythia event header not found");
872 return 0;
873 }
874 }
875 return pythiaGenHeader;
876
877}
878
fdceab34 879//________________________________________________________________________
880void AliPWG4HighPtQAMC::Terminate(Option_t *)
881{
df943115 882 // The Terminate() function is the last function to be called during
883 // a query. It always runs on the client, it can be used to present
884 // the results graphically or save the results to file.
fdceab34 885
fdceab34 886}
df943115 887
888#endif