]> git.uio.no Git - u/mrichter/AliRoot.git/blame_incremental - JETAN/AliAnalysisTaskJetCluster.cxx
changes from saiola
[u/mrichter/AliRoot.git] / JETAN / AliAnalysisTaskJetCluster.cxx
... / ...
CommitLineData
1// **************************************
2// Task used for the correction of determiantion of reconstructed jet spectra
3// Compares input (gen) and output (rec) jets
4// *******************************************
5
6
7/**************************************************************************
8 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
9 * *
10 * Author: The ALICE Off-line Project. *
11 * Contributors are mentioned in the code where appropriate. *
12 * *
13 * Permission to use, copy, modify and distribute this software and its *
14 * documentation strictly for non-commercial purposes is hereby granted *
15 * without fee, provided that the above copyright notice appears in all *
16 * copies and that both the copyright notice and this permission notice *
17 * appear in the supporting documentation. The authors make no claims *
18 * about the suitability of this software for any purpose. It is *
19 * provided "as is" without express or implied warranty. *
20 **************************************************************************/
21
22
23#include <TROOT.h>
24#include <TRandom3.h>
25#include <TSystem.h>
26#include <TInterpreter.h>
27#include <TChain.h>
28#include <TRefArray.h>
29#include <TFile.h>
30#include <TKey.h>
31#include <TH1F.h>
32#include <TH2F.h>
33#include <TH3F.h>
34#include <TProfile.h>
35#include <TF1.h>
36#include <TList.h>
37#include <TLorentzVector.h>
38#include <TClonesArray.h>
39#include "TDatabasePDG.h"
40
41#include "AliAnalysisTaskJetCluster.h"
42#include "AliAnalysisManager.h"
43#include "AliJetFinder.h"
44#include "AliJetHeader.h"
45#include "AliJetReader.h"
46#include "AliESDEvent.h"
47#include "AliAODEvent.h"
48#include "AliAODHandler.h"
49#include "AliAODExtension.h"
50#include "AliAODTrack.h"
51#include "AliAODJet.h"
52#include "AliAODMCParticle.h"
53#include "AliMCEventHandler.h"
54#include "AliMCEvent.h"
55#include "AliStack.h"
56#include "AliGenPythiaEventHeader.h"
57#include "AliJetKineReaderHeader.h"
58#include "AliGenCocktailEventHeader.h"
59#include "AliInputEventHandler.h"
60#include "AliAODJetEventBackground.h"
61
62#include "fastjet/PseudoJet.hh"
63#include "fastjet/ClusterSequenceArea.hh"
64#include "fastjet/AreaDefinition.hh"
65#include "fastjet/JetDefinition.hh"
66// get info on how fastjet was configured
67#include "fastjet/config.h"
68
69using std::vector;
70
71ClassImp(AliAnalysisTaskJetCluster)
72
73AliAnalysisTaskJetCluster::~AliAnalysisTaskJetCluster(){
74 //
75 // Destructor
76 //
77
78 delete fRef;
79 delete fRandom;
80
81 if(fTCAJetsOut)fTCAJetsOut->Delete();
82 delete fTCAJetsOut;
83
84 if(fTCAJetsOutRan)fTCAJetsOutRan->Delete();
85 delete fTCAJetsOutRan;
86
87 if(fTCARandomConesOut)fTCARandomConesOut->Delete();
88 delete fTCARandomConesOut;
89
90 if(fTCARandomConesOutRan)fTCARandomConesOutRan->Delete();
91 delete fTCARandomConesOutRan;
92
93 if(fAODJetBackgroundOut)fAODJetBackgroundOut->Reset();
94 delete fAODJetBackgroundOut;
95}
96
97AliAnalysisTaskJetCluster::AliAnalysisTaskJetCluster():
98 AliAnalysisTaskSE(),
99 fAOD(0x0),
100 fAODExtension(0x0),
101 fRef(new TRefArray),
102 fUseAODTrackInput(kFALSE),
103 fUseAODMCInput(kFALSE),
104 fUseBackgroundCalc(kFALSE),
105 fEventSelection(kFALSE),
106 fFilterMask(0),
107 fFilterMaskBestPt(0),
108 fFilterType(0),
109 fJetTypes(kJet),
110 fTrackTypeRec(kTrackUndef),
111 fTrackTypeGen(kTrackUndef),
112 fNSkipLeadingRan(0),
113 fNSkipLeadingCone(0),
114 fNRandomCones(0),
115 fAvgTrials(1),
116 fExternalWeight(1),
117 fTrackEtaWindow(0.9),
118 fRecEtaWindow(0.5),
119 fTrackPtCut(0.),
120 fJetOutputMinPt(0.150),
121 fMaxTrackPtInJet(100.),
122 fJetTriggerPtCut(0),
123 fVtxZCut(8),
124 fVtxR2Cut(1),
125 fCentCutUp(0),
126 fCentCutLo(0),
127 fNonStdBranch(""),
128 fBackgroundBranch(""),
129 fNonStdFile(""),
130 fMomResH1(0x0),
131 fMomResH2(0x0),
132 fMomResH3(0x0),
133 fMomResH1Fit(0x0),
134 fMomResH2Fit(0x0),
135 fMomResH3Fit(0x0),
136 fhEffH1(0x0),
137 fhEffH2(0x0),
138 fhEffH3(0x0),
139 fUseTrPtResolutionSmearing(kFALSE),
140 fUseDiceEfficiency(kFALSE),
141 fDiceEfficiencyMinPt(0.),
142 fUseTrPtResolutionFromOADB(kFALSE),
143 fUseTrEfficiencyFromOADB(kFALSE),
144 fPathTrPtResolution(""),
145 fPathTrEfficiency(""),
146 fChangeEfficiencyFraction(0.),
147 fRparam(1.0),
148 fAlgorithm(fastjet::kt_algorithm),
149 fStrategy(fastjet::Best),
150 fRecombScheme(fastjet::BIpt_scheme),
151 fAreaType(fastjet::active_area),
152 fGhostArea(0.01),
153 fActiveAreaRepeats(1),
154 fGhostEtamax(1.5),
155 fTCAJetsOut(0x0),
156 fTCAJetsOutRan(0x0),
157 fTCARandomConesOut(0x0),
158 fTCARandomConesOutRan(0x0),
159 fAODJetBackgroundOut(0x0),
160 fRandom(0),
161 fh1Xsec(0x0),
162 fh1Trials(0x0),
163 fh1PtHard(0x0),
164 fh1PtHardNoW(0x0),
165 fh1PtHardTrials(0x0),
166 fh1NJetsRec(0x0),
167 fh1NConstRec(0x0),
168 fh1NConstLeadingRec(0x0),
169 fh1PtJetsRecIn(0x0),
170 fh1PtJetsLeadingRecIn(0x0),
171 fh1PtJetConstRec(0x0),
172 fh1PtJetConstLeadingRec(0x0),
173 fh1PtTracksRecIn(0x0),
174 fh1PtTracksLeadingRecIn(0x0),
175 fh1NJetsRecRan(0x0),
176 fh1NConstRecRan(0x0),
177 fh1PtJetsLeadingRecInRan(0x0),
178 fh1NConstLeadingRecRan(0x0),
179 fh1PtJetsRecInRan(0x0),
180 fh1PtTracksGenIn(0x0),
181 fh1Nch(0x0),
182 fh1CentralityPhySel(0x0),
183 fh1Centrality(0x0),
184 fh1CentralitySelect(0x0),
185 fh1ZPhySel(0x0),
186 fh1Z(0x0),
187 fh1ZSelect(0x0),
188 fh2NRecJetsPt(0x0),
189 fh2NRecTracksPt(0x0),
190 fh2NConstPt(0x0),
191 fh2NConstLeadingPt(0x0),
192 fh2JetPhiEta(0x0),
193 fh2LeadingJetPhiEta(0x0),
194 fh2JetEtaPt(0x0),
195 fh2LeadingJetEtaPt(0x0),
196 fh2TrackEtaPt(0x0),
197 fh2LeadingTrackEtaPt(0x0),
198 fh2JetsLeadingPhiEta(0x0),
199 fh2JetsLeadingPhiPt(0x0),
200 fh2TracksLeadingPhiEta(0x0),
201 fh2TracksLeadingPhiPt(0x0),
202 fh2TracksLeadingJetPhiPt(0x0),
203 fh2JetsLeadingPhiPtW(0x0),
204 fh2TracksLeadingPhiPtW(0x0),
205 fh2TracksLeadingJetPhiPtW(0x0),
206 fh2NRecJetsPtRan(0x0),
207 fh2NConstPtRan(0x0),
208 fh2NConstLeadingPtRan(0x0),
209 fh2PtNch(0x0),
210 fh2PtNchRan(0x0),
211 fh2PtNchN(0x0),
212 fh2PtNchNRan(0x0),
213 fh2TracksLeadingJetPhiPtRan(0x0),
214 fh2TracksLeadingJetPhiPtWRan(0x0),
215 fh2PtGenPtSmeared(0x0),
216 fp1Efficiency(0x0),
217 fp1PtResolution(0x0),
218 fHistList(0x0)
219{
220 //
221 // Constructor
222 //
223
224 for(int i = 0;i<3;i++){
225 fh1BiARandomCones[i] = 0;
226 fh1BiARandomConesRan[i] = 0;
227 }
228 for(int i = 0;i<kMaxCent;i++){
229 fh2JetsLeadingPhiPtC[i] = 0;
230 fh2JetsLeadingPhiPtWC[i] = 0; //! jet correlation with leading jet
231 fh2TracksLeadingJetPhiPtC[i] = 0;
232 fh2TracksLeadingJetPhiPtWC[i] = 0;
233 }
234}
235
236AliAnalysisTaskJetCluster::AliAnalysisTaskJetCluster(const char* name):
237 AliAnalysisTaskSE(name),
238 fAOD(0x0),
239 fAODExtension(0x0),
240 fRef(new TRefArray),
241 fUseAODTrackInput(kFALSE),
242 fUseAODMCInput(kFALSE),
243 fUseBackgroundCalc(kFALSE),
244 fEventSelection(kFALSE), fFilterMask(0),
245 fFilterMaskBestPt(0),
246 fFilterType(0),
247 fJetTypes(kJet),
248 fTrackTypeRec(kTrackUndef),
249 fTrackTypeGen(kTrackUndef),
250 fNSkipLeadingRan(0),
251 fNSkipLeadingCone(0),
252 fNRandomCones(0),
253 fAvgTrials(1),
254 fExternalWeight(1),
255 fTrackEtaWindow(0.9),
256 fRecEtaWindow(0.5),
257 fTrackPtCut(0.),
258 fJetOutputMinPt(0.150),
259 fMaxTrackPtInJet(100.),
260 fJetTriggerPtCut(0),
261 fVtxZCut(8),
262 fVtxR2Cut(1),
263 fCentCutUp(0),
264 fCentCutLo(0),
265 fNonStdBranch(""),
266 fBackgroundBranch(""),
267 fNonStdFile(""),
268 fMomResH1(0x0),
269 fMomResH2(0x0),
270 fMomResH3(0x0),
271 fMomResH1Fit(0x0),
272 fMomResH2Fit(0x0),
273 fMomResH3Fit(0x0),
274 fhEffH1(0x0),
275 fhEffH2(0x0),
276 fhEffH3(0x0),
277 fUseTrPtResolutionSmearing(kFALSE),
278 fUseDiceEfficiency(kFALSE),
279 fDiceEfficiencyMinPt(0.),
280 fUseTrPtResolutionFromOADB(kFALSE),
281 fUseTrEfficiencyFromOADB(kFALSE),
282 fPathTrPtResolution(""),
283 fPathTrEfficiency(""),
284 fChangeEfficiencyFraction(0.),
285 fRparam(1.0),
286 fAlgorithm(fastjet::kt_algorithm),
287 fStrategy(fastjet::Best),
288 fRecombScheme(fastjet::BIpt_scheme),
289 fAreaType(fastjet::active_area),
290 fGhostArea(0.01),
291 fActiveAreaRepeats(1),
292 fGhostEtamax(1.5),
293 fTCAJetsOut(0x0),
294 fTCAJetsOutRan(0x0),
295 fTCARandomConesOut(0x0),
296 fTCARandomConesOutRan(0x0),
297 fAODJetBackgroundOut(0x0),
298 fRandom(0),
299 fh1Xsec(0x0),
300 fh1Trials(0x0),
301 fh1PtHard(0x0),
302 fh1PtHardNoW(0x0),
303 fh1PtHardTrials(0x0),
304 fh1NJetsRec(0x0),
305 fh1NConstRec(0x0),
306 fh1NConstLeadingRec(0x0),
307 fh1PtJetsRecIn(0x0),
308 fh1PtJetsLeadingRecIn(0x0),
309 fh1PtJetConstRec(0x0),
310 fh1PtJetConstLeadingRec(0x0),
311 fh1PtTracksRecIn(0x0),
312 fh1PtTracksLeadingRecIn(0x0),
313 fh1NJetsRecRan(0x0),
314 fh1NConstRecRan(0x0),
315 fh1PtJetsLeadingRecInRan(0x0),
316 fh1NConstLeadingRecRan(0x0),
317 fh1PtJetsRecInRan(0x0),
318 fh1PtTracksGenIn(0x0),
319 fh1Nch(0x0),
320 fh1CentralityPhySel(0x0),
321 fh1Centrality(0x0),
322 fh1CentralitySelect(0x0),
323 fh1ZPhySel(0x0),
324 fh1Z(0x0),
325 fh1ZSelect(0x0),
326 fh2NRecJetsPt(0x0),
327 fh2NRecTracksPt(0x0),
328 fh2NConstPt(0x0),
329 fh2NConstLeadingPt(0x0),
330 fh2JetPhiEta(0x0),
331 fh2LeadingJetPhiEta(0x0),
332 fh2JetEtaPt(0x0),
333 fh2LeadingJetEtaPt(0x0),
334 fh2TrackEtaPt(0x0),
335 fh2LeadingTrackEtaPt(0x0),
336 fh2JetsLeadingPhiEta(0x0),
337 fh2JetsLeadingPhiPt(0x0),
338 fh2TracksLeadingPhiEta(0x0),
339 fh2TracksLeadingPhiPt(0x0),
340 fh2TracksLeadingJetPhiPt(0x0),
341 fh2JetsLeadingPhiPtW(0x0),
342 fh2TracksLeadingPhiPtW(0x0),
343 fh2TracksLeadingJetPhiPtW(0x0),
344 fh2NRecJetsPtRan(0x0),
345 fh2NConstPtRan(0x0),
346 fh2NConstLeadingPtRan(0x0),
347 fh2PtNch(0x0),
348 fh2PtNchRan(0x0),
349 fh2PtNchN(0x0),
350 fh2PtNchNRan(0x0),
351 fh2TracksLeadingJetPhiPtRan(0x0),
352 fh2TracksLeadingJetPhiPtWRan(0x0),
353 fh2PtGenPtSmeared(0x0),
354 fp1Efficiency(0x0),
355 fp1PtResolution(0x0),
356 fHistList(0x0)
357{
358 //
359 // named ctor
360 //
361
362 for(int i = 0;i<3;i++){
363 fh1BiARandomCones[i] = 0;
364 fh1BiARandomConesRan[i] = 0;
365 }
366 for(int i = 0;i<kMaxCent;i++){
367 fh2JetsLeadingPhiPtC[i] = 0;
368 fh2JetsLeadingPhiPtWC[i] = 0; //! jet correlation with leading jet
369 fh2TracksLeadingJetPhiPtC[i] = 0;
370 fh2TracksLeadingJetPhiPtWC[i] = 0;
371 }
372 DefineOutput(1, TList::Class());
373}
374
375
376
377Bool_t AliAnalysisTaskJetCluster::Notify()
378{
379 //
380 // Implemented Notify() to read the cross sections
381 // and number of trials from pyxsec.root
382 //
383 return kTRUE;
384}
385
386void AliAnalysisTaskJetCluster::UserCreateOutputObjects()
387{
388
389 //
390 // Create the output container
391 //
392
393 fRandom = new TRandom3(0);
394
395
396 // Connect the AOD
397
398
399 if (fDebug > 1) printf("AnalysisTaskJetCluster::UserCreateOutputObjects() \n");
400
401
402
403 if(fNonStdBranch.Length()!=0)
404 {
405 // only create the output branch if we have a name
406 // Create a new branch for jets...
407 // -> cleared in the UserExec....
408 // here we can also have the case that the brnaches are written to a separate file
409
410 if(fJetTypes&kJet){
411 fTCAJetsOut = new TClonesArray("AliAODJet", 0);
412 fTCAJetsOut->SetName(fNonStdBranch.Data());
413 AddAODBranch("TClonesArray",&fTCAJetsOut,fNonStdFile.Data());
414 }
415
416 if(fJetTypes&kJetRan){
417 fTCAJetsOutRan = new TClonesArray("AliAODJet", 0);
418 fTCAJetsOutRan->SetName(Form("%s_%s",fNonStdBranch.Data(),"random"));
419 if(fUseDiceEfficiency || fUseTrPtResolutionSmearing)
420 fTCAJetsOutRan->SetName(Form("%s_%sDetector%d%dFr%d",fNonStdBranch.Data(),"random",fUseTrPtResolutionSmearing,fUseDiceEfficiency,(int)(fChangeEfficiencyFraction*100.)));
421 AddAODBranch("TClonesArray",&fTCAJetsOutRan,fNonStdFile.Data());
422 }
423
424 if(fUseBackgroundCalc){
425 if(!AODEvent()->FindListObject(Form("%s_%s",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data()))){
426 fAODJetBackgroundOut = new AliAODJetEventBackground();
427 fAODJetBackgroundOut->SetName(Form("%s_%s",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data()));
428 if(fUseDiceEfficiency || fUseTrPtResolutionSmearing)
429 fAODJetBackgroundOut->SetName(Form("%s_%sDetector%d%dFr%d",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data(),fUseTrPtResolutionSmearing,fUseDiceEfficiency,(int)(fChangeEfficiencyFraction*100.)));
430
431 AddAODBranch("AliAODJetEventBackground",&fAODJetBackgroundOut,fNonStdFile.Data());
432 }
433 }
434 // create the branch for the random cones with the same R
435 TString cName = Form("%sRandomConeSkip%02d",fNonStdBranch.Data(),fNSkipLeadingCone);
436 if(fUseDiceEfficiency || fUseTrPtResolutionSmearing)
437 cName = Form("%sDetector%d%dFr%d_RandomConeSkip%02d",fNonStdBranch.Data(),fUseTrPtResolutionSmearing,fUseDiceEfficiency,(int)(fChangeEfficiencyFraction*100.),fNSkipLeadingCone);
438
439 if(fNRandomCones>0){
440 if(fJetTypes&kRC){
441 if(!AODEvent()->FindListObject(cName.Data())){
442 fTCARandomConesOut = new TClonesArray("AliAODJet", 0);
443 fTCARandomConesOut->SetName(cName.Data());
444 AddAODBranch("TClonesArray",&fTCARandomConesOut,fNonStdFile.Data());
445 }
446 }
447 // create the branch with the random for the random cones on the random event
448 if(fJetTypes&kRCRan){
449 cName = Form("%sRandomCone_random",fNonStdBranch.Data());
450 if(!AODEvent()->FindListObject(cName.Data())){
451 fTCARandomConesOutRan = new TClonesArray("AliAODJet", 0);
452 fTCARandomConesOutRan->SetName(cName.Data());
453 AddAODBranch("TClonesArray",&fTCARandomConesOutRan,fNonStdFile.Data());
454 }
455 }
456 }
457
458 if(fNonStdFile.Length()!=0){
459 //
460 // case that we have an AOD extension we need to fetch the jets from the extended output
461 // we identify the extension aod event by looking for the branchname
462 AliAODHandler *aodH = dynamic_cast<AliAODHandler*>(AliAnalysisManager::GetAnalysisManager()->GetOutputEventHandler());
463 // case that we have an AOD extension we need can fetch the background maybe from the extended output
464 fAODExtension = (aodH?aodH->GetExtension(fNonStdFile.Data()):0);
465 }
466 }
467
468
469 if(!fHistList)fHistList = new TList();
470 fHistList->SetOwner();
471 PostData(1, fHistList); // post data in any case once
472
473 Bool_t oldStatus = TH1::AddDirectoryStatus();
474 TH1::AddDirectory(kFALSE);
475
476 //
477 // Histogram
478
479 const Int_t nBinPt = 100;
480 Double_t binLimitsPt[nBinPt+1];
481 for(Int_t iPt = 0;iPt <= nBinPt;iPt++){
482 if(iPt == 0){
483 binLimitsPt[iPt] = 0.0;
484 }
485 else {// 1.0
486 binLimitsPt[iPt] = binLimitsPt[iPt-1] + 2.0;
487 }
488 }
489
490 const Int_t nBinPhi = 90;
491 Double_t binLimitsPhi[nBinPhi+1];
492 for(Int_t iPhi = 0;iPhi<=nBinPhi;iPhi++){
493 if(iPhi==0){
494 binLimitsPhi[iPhi] = -1.*TMath::Pi();
495 }
496 else{
497 binLimitsPhi[iPhi] = binLimitsPhi[iPhi-1] + 1/(Float_t)nBinPhi * TMath::Pi()*2;
498 }
499 }
500
501
502
503 const Int_t nBinEta = 40;
504 Double_t binLimitsEta[nBinEta+1];
505 for(Int_t iEta = 0;iEta<=nBinEta;iEta++){
506 if(iEta==0){
507 binLimitsEta[iEta] = -2.0;
508 }
509 else{
510 binLimitsEta[iEta] = binLimitsEta[iEta-1] + 0.1;
511 }
512 }
513
514 const int nChMax = 5000;
515
516 fh1Xsec = new TProfile("fh1Xsec","xsec from pyxsec.root",1,0,1);
517 fh1Xsec->GetXaxis()->SetBinLabel(1,"<#sigma>");
518
519 fh1Trials = new TH1F("fh1Trials","trials root file",1,0,1);
520 fh1Trials->GetXaxis()->SetBinLabel(1,"#sum{ntrials}");
521
522
523 fh1NJetsRec = new TH1F("fh1NJetsRec","N reconstructed jets",120,-0.5,119.5);
524 fh1NJetsRecRan = new TH1F("fh1NJetsRecRan","N reconstructed jets",120,-0.5,119.5);
525
526 fh1NConstRec = new TH1F("fh1NConstRec","# jet constituents",120,-0.5,119.5);
527 fh1NConstRecRan = new TH1F("fh1NConstRecRan","# jet constituents",120,-0.5,119.5);
528 fh1NConstLeadingRec = new TH1F("fh1NConstLeadingRec","jet constituents",120,-0.5,119.5);
529 fh1NConstLeadingRecRan = new TH1F("fh1NConstLeadingRecRan","jet constituents",120,-0.5,119.5);
530
531
532 fh1PtHard = new TH1F("fh1PtHard","PYTHIA Pt hard;p_{T,hard}",nBinPt,binLimitsPt);
533 fh1PtHardNoW = new TH1F("fh1PtHardNoW","PYTHIA Pt hard no weight;p_{T,hard}",nBinPt,binLimitsPt);
534 fh1PtHardTrials = new TH1F("fh1PtHardTrials","PYTHIA Pt hard weight with trials;p_{T,hard}",nBinPt,binLimitsPt);
535
536 fh1PtJetsRecIn = new TH1F("fh1PtJetsRecIn","Rec jets P_T;p_{T} (GeV/c)",nBinPt,binLimitsPt);
537 fh1PtJetsRecInRan = new TH1F("fh1PtJetsRecInRan","Rec jets P_T;p_{T} (GeV/c)",nBinPt,binLimitsPt);
538 fh1PtJetsLeadingRecIn = new TH1F("fh1PtJetsLeadingRecIn","Rec jets P_T;p_{T} (GeV/c)",nBinPt,binLimitsPt);
539 fh1PtJetsLeadingRecInRan = new TH1F("fh1PtJetsLeadingRecInRan","Rec jets P_T;p_{T} (GeV/c)",nBinPt,binLimitsPt);
540 fh1PtJetConstRec = new TH1F("fh1PtJetsConstRec","Rec jets constituents P_T;p_{T} (GeV/c)",nBinPt,binLimitsPt);
541 fh1PtJetConstLeadingRec = new TH1F("fh1PtJetsConstLeadingRec","Rec jets constituents P_T;p_{T} (GeV/c)",nBinPt,binLimitsPt);
542 fh1PtTracksRecIn = new TH1F("fh1PtTracksRecIn",Form("Rec tracks P_T #eta < %1.2f;p_{T} (GeV/c)",fTrackEtaWindow),nBinPt,binLimitsPt);
543 fh1PtTracksLeadingRecIn = new TH1F("fh1PtTracksLeadingRecIn",Form("Rec tracks P_T #eta < %1.2f ;p_{T} (GeV/c)",fTrackEtaWindow),nBinPt,binLimitsPt);
544 fh1PtTracksGenIn = new TH1F("fh1PtTracksGenIn",Form("gen tracks P_T #eta < %1.2f ;p_{T} (GeV/c)",fTrackEtaWindow),nBinPt,binLimitsPt);
545 fh1Nch = new TH1F("fh1Nch","charged multiplicity; N_{ch}",nChMax,-0.5,nChMax-0.5);
546
547 fh1Centrality = new TH1F("fh1Centrality",";cent (%)",111,-0.5,110.5);
548 fh1CentralitySelect = new TH1F("fh1CentralitySelect",";cent (%)",111,-0.5,110.5);
549 fh1CentralityPhySel = new TH1F("fh1CentralityPhySel",";cent (%)",111,-0.5,110.5);
550
551 fh1Z = new TH1F("fh1Z",";zvtx",100,-25,25);
552 fh1ZSelect = new TH1F("fh1ZSelect",";zvtx",100,-25,25);
553 fh1ZPhySel = new TH1F("fh1ZPhySel",";zvtx",100,-25,25);
554
555 fh2NRecJetsPt = new TH2F("fh2NRecJetsPt","Number of jets above threshhold;p_{T,cut} (GeV/c);N_{jets}",nBinPt,binLimitsPt,50,-0.5,49.5);
556 fh2NRecJetsPtRan = new TH2F("fh2NRecJetsPtRan","Number of jets above threshhold;p_{T,cut} (GeV/c);N_{jets}",nBinPt,binLimitsPt,50,-0.5,49.5);
557 fh2NRecTracksPt = new TH2F("fh2NRecTracksPt","Number of tracks above threshhold;p_{T,cut} (GeV/c);N_{tracks}",nBinPt,binLimitsPt,50,-0.5,49.5);
558 //
559
560
561 fh2NConstPt = new TH2F("fh2NConstPt","Number of constituents ;p_{T} (GeV/c);N",nBinPt,binLimitsPt,50,-0.5,49.5);
562 fh2NConstLeadingPt = new TH2F("fh2NConstLeadingPt","Number of constituents ;p_{T} (GeV/c);N",nBinPt,binLimitsPt,50,-0.5,49.5);
563 fh2NConstPtRan = new TH2F("fh2NConstPtRan","Number of constituents ;p_{T} (GeV/c);N",nBinPt,binLimitsPt,50,-0.5,49.5);
564 fh2NConstLeadingPtRan = new TH2F("fh2NConstLeadingPtRan","Number of constituents ;p_{T} (GeV/c);N",nBinPt,binLimitsPt,50,-0.5,49.5);
565
566 fh2PtNch = new TH2F("fh2PtNch","p_T of cluster vs. multiplicity; N_{ch};p_{T} (GeV/c);",nChMax,-0.5,nChMax-0.5,nBinPt,binLimitsPt);
567 fh2PtNchRan = new TH2F("fh2PtNchRan","p_T of cluster vs. multiplicity ran; N_{ch};p_{T} (GeV/c);",nChMax,-0.5,nChMax-0.5,nBinPt,binLimitsPt);
568 fh2PtNchN = new TH2F("fh2PtNchN","p_T of cluster vs. multiplicity N weighted; N_{ch};p_{T} (GeV/c);",nChMax,-0.5,nChMax-0.5,nBinPt,binLimitsPt);
569 fh2PtNchNRan = new TH2F("fh2PtNchNRan","p_T of cluster vs. multiplicity N weighted ran; N_{ch};p_{T} (GeV/c);",nChMax,-0.5,nChMax-0.5,nBinPt,binLimitsPt);
570
571
572
573 fh2JetPhiEta = new TH2F("fh2JetPhiEta","eta vs phi all jets;#phi;#eta",
574 nBinPhi,0.,2.*TMath::Pi(),nBinEta,binLimitsEta);
575 fh2LeadingJetPhiEta = new TH2F("fh2LeadingJetPhiEta","eta vs phi leading jets;#phi;#eta",
576 nBinPhi,0.,2.*TMath::Pi(),nBinEta,binLimitsEta);
577
578 fh2JetEtaPt = new TH2F("fh2JetEtaPt","pt vs eta all jets;#eta;p_{T}",
579 nBinEta,binLimitsEta,nBinPt,binLimitsPt);
580 fh2LeadingJetEtaPt = new TH2F("fh2LeadingJetEtaPt","pT vs eta leading jets;#eta;p_{T}",
581 nBinEta,binLimitsEta,nBinPt,binLimitsPt);
582
583 fh2TrackEtaPt = new TH2F("fh2TrackEtaPt","pt vs eta all jets;#eta;p_{T}",
584 nBinEta,binLimitsEta,nBinPt,binLimitsPt);
585 fh2LeadingTrackEtaPt = new TH2F("fh2LeadingTrackEtaPt","pT vs eta leading jets;#eta;p_{T}",
586 nBinEta,binLimitsEta,nBinPt,binLimitsPt);
587
588
589
590 fh2JetsLeadingPhiEta = new TH2F("fh2JetsLeadingPhiEta","delta eta vs delta phi to leading jet;#Delta#phi;#Delta#eta",
591 nBinPhi,binLimitsPhi,nBinEta,binLimitsEta);
592 fh2JetsLeadingPhiPt = new TH2F("fh2JetsLeadingPhiPt","leading p_T vs delta phi to leading jet;#Delta#phi;p_{T} (GeV/c)",
593 nBinPhi,binLimitsPhi,nBinPt,binLimitsPt);
594 fh2TracksLeadingPhiEta = new TH2F("fh2TracksLeadingPhiEta","delta eta vs delta phi to leading track;#Delta#phi;#Delta#eta",
595 nBinPhi,binLimitsPhi,nBinEta,binLimitsEta);
596 fh2TracksLeadingPhiPt = new TH2F("fh2TracksLeadingPhiPt","leading p_T vs delta phi to leading jet;#Delta#phi;p_{T} (GeV/c)",
597 nBinPhi,binLimitsPhi,nBinPt,binLimitsPt);
598 fh2TracksLeadingJetPhiPt = new TH2F("fh2TracksLeadingJetPhiPt","leading p_T vs delta phi to leading jet;#Delta#phi;p_{T} (GeV/c)",
599 nBinPhi,binLimitsPhi,nBinPt,binLimitsPt);
600 fh2TracksLeadingJetPhiPtRan = new TH2F("fh2TracksLeadingJetPhiPtRan","leading p_T vs delta phi to leading jet;#Delta#phi;p_{T} (GeV/c)",
601 nBinPhi,binLimitsPhi,nBinPt,binLimitsPt);
602
603 fh2JetsLeadingPhiPtW = new TH2F("fh2JetsLeadingPhiPtW","leading p_T vs delta phi p_T weigted to leading jet;#Delta#phi;p_{T} (GeV/c)",
604 nBinPhi,binLimitsPhi,nBinPt,binLimitsPt);
605 fh2TracksLeadingPhiPtW = new TH2F("fh2TracksLeadingPhiPtW","leading p_T vs delta phi to leading jet (p_T weighted);#Delta#phi;p_{T} (GeV/c)",
606 nBinPhi,binLimitsPhi,nBinPt,binLimitsPt);
607
608 fh2TracksLeadingJetPhiPtW = new TH2F("fh2TracksLeadingJetPhiPtW","leading p_T vs delta phi to leading jet;#Delta#phi;p_{T} (GeV/c)",
609 nBinPhi,binLimitsPhi,nBinPt,binLimitsPt);
610 fh2TracksLeadingJetPhiPtWRan = new TH2F("fh2TracksLeadingJetPhiPtWRan","leading p_T vs delta phi to leading jet;#Delta#phi;p_{T} (GeV/c)",
611 nBinPhi,binLimitsPhi,nBinPt,binLimitsPt);
612
613 //Detector level effects histos
614 fh2PtGenPtSmeared = new TH2F("fh2PtGenPtSmeared","fh2PtGenPtSmeared",nBinPt,binLimitsPt,nBinPt,binLimitsPt);
615
616 fp1Efficiency = new TProfile("fp1Efficiency","fp1Efficiency",nBinPt,binLimitsPt);
617 fp1PtResolution = new TProfile("fp1PtResolution","fp1PtResolution",nBinPt,binLimitsPt);
618
619 fHistList->Add(fh2PtGenPtSmeared);
620 fHistList->Add(fp1Efficiency);
621 fHistList->Add(fp1PtResolution);
622
623 if(fNRandomCones>0&&fUseBackgroundCalc){
624 for(int i = 0;i<3;i++){
625 fh1BiARandomCones[i] = new TH1F(Form("fh1BiARandomCones%d",i),";B_{i}^{A} (GeV/c)",200,-100,100);
626 fh1BiARandomConesRan[i] = new TH1F(Form("fh1BiARandomConesRan%d",i),";B_{i}^{A} (GeV/c)",200,-100,100);
627 }
628 }
629
630 for(int i = 0;i < kMaxCent;i++){
631 fh2JetsLeadingPhiPtC[i] = (TH2F*)fh2JetsLeadingPhiPt->Clone(Form("%s_C%02d",fh2JetsLeadingPhiPt->GetName(),i+1));
632 fh2JetsLeadingPhiPtWC[i]= (TH2F*)fh2JetsLeadingPhiPtW->Clone(Form("%s_C%02d",fh2JetsLeadingPhiPtW->GetName(),i+1));
633 fh2TracksLeadingJetPhiPtC[i] = (TH2F*)fh2TracksLeadingJetPhiPt->Clone(Form("%s_C%02d",fh2TracksLeadingJetPhiPt->GetName(),i+1));
634 fh2TracksLeadingJetPhiPtWC[i] = (TH2F*)fh2TracksLeadingJetPhiPtW->Clone(Form("%s_C%02d",fh2TracksLeadingJetPhiPtW->GetName(),i+1));
635 }
636
637 const Int_t saveLevel = 3; // large save level more histos
638 if(saveLevel>0){
639 fHistList->Add(fh1Xsec);
640 fHistList->Add(fh1Trials);
641
642 fHistList->Add(fh1NJetsRec);
643 fHistList->Add(fh1NConstRec);
644 fHistList->Add(fh1NConstLeadingRec);
645 fHistList->Add(fh1PtJetsRecIn);
646 fHistList->Add(fh1PtJetsLeadingRecIn);
647 fHistList->Add(fh1PtTracksRecIn);
648 fHistList->Add(fh1PtTracksLeadingRecIn);
649 fHistList->Add(fh1PtJetConstRec);
650 fHistList->Add(fh1PtJetConstLeadingRec);
651 fHistList->Add(fh1NJetsRecRan);
652 fHistList->Add(fh1NConstRecRan);
653 fHistList->Add(fh1PtJetsLeadingRecInRan);
654 fHistList->Add(fh1NConstLeadingRecRan);
655 fHistList->Add(fh1PtJetsRecInRan);
656 fHistList->Add(fh1Nch);
657 fHistList->Add(fh1Centrality);
658 fHistList->Add(fh1CentralitySelect);
659 fHistList->Add(fh1CentralityPhySel);
660 fHistList->Add(fh1Z);
661 fHistList->Add(fh1ZSelect);
662 fHistList->Add(fh1ZPhySel);
663 if(fNRandomCones>0&&fUseBackgroundCalc){
664 for(int i = 0;i<3;i++){
665 fHistList->Add(fh1BiARandomCones[i]);
666 fHistList->Add(fh1BiARandomConesRan[i]);
667 }
668 }
669 for(int i = 0;i < kMaxCent;i++){
670 fHistList->Add(fh2JetsLeadingPhiPtC[i]);
671 fHistList->Add(fh2JetsLeadingPhiPtWC[i]);
672 fHistList->Add(fh2TracksLeadingJetPhiPtC[i]);
673 fHistList->Add(fh2TracksLeadingJetPhiPtWC[i]);
674 }
675
676 fHistList->Add(fh2NRecJetsPt);
677 fHistList->Add(fh2NRecTracksPt);
678 fHistList->Add(fh2NConstPt);
679 fHistList->Add(fh2NConstLeadingPt);
680 fHistList->Add(fh2PtNch);
681 fHistList->Add(fh2PtNchRan);
682 fHistList->Add(fh2PtNchN);
683 fHistList->Add(fh2PtNchNRan);
684 fHistList->Add(fh2JetPhiEta);
685 fHistList->Add(fh2LeadingJetPhiEta);
686 fHistList->Add(fh2JetEtaPt);
687 fHistList->Add(fh2LeadingJetEtaPt);
688 fHistList->Add(fh2TrackEtaPt);
689 fHistList->Add(fh2LeadingTrackEtaPt);
690 fHistList->Add(fh2JetsLeadingPhiEta );
691 fHistList->Add(fh2JetsLeadingPhiPt);
692 fHistList->Add(fh2TracksLeadingPhiEta);
693 fHistList->Add(fh2TracksLeadingPhiPt);
694 fHistList->Add(fh2TracksLeadingJetPhiPt);
695 fHistList->Add(fh2JetsLeadingPhiPtW);
696 fHistList->Add(fh2TracksLeadingPhiPtW);
697 fHistList->Add(fh2TracksLeadingJetPhiPtW);
698 fHistList->Add(fh2NRecJetsPtRan);
699 fHistList->Add(fh2NConstPtRan);
700 fHistList->Add(fh2NConstLeadingPtRan);
701 fHistList->Add(fh2TracksLeadingJetPhiPtRan);
702 fHistList->Add(fh2TracksLeadingJetPhiPtWRan);
703 }
704
705 // =========== Switch on Sumw2 for all histos ===========
706 for (Int_t i=0; i<fHistList->GetEntries(); ++i) {
707 TH1 *h1 = dynamic_cast<TH1*>(fHistList->At(i));
708 if (h1){
709 h1->Sumw2();
710 continue;
711 }
712 THnSparse *hn = dynamic_cast<THnSparse*>(fHistList->At(i));
713 if(hn)hn->Sumw2();
714 }
715 TH1::AddDirectory(oldStatus);
716}
717
718void AliAnalysisTaskJetCluster::LocalInit()
719{
720 //
721 // Initialization
722 //
723
724 if (fDebug > 1) printf("AnalysisTaskJetCluster::Init() \n");
725
726 if(fUseTrPtResolutionFromOADB) LoadTrPtResolutionRootFileFromOADB();
727 if(fUseTrEfficiencyFromOADB) LoadTrEfficiencyRootFileFromOADB();
728
729
730 FitMomentumResolution();
731
732}
733
734void AliAnalysisTaskJetCluster::UserExec(Option_t */*option*/)
735{
736
737 // handle and reset the output jet branch
738
739 if(fTCAJetsOut)fTCAJetsOut->Delete();
740 if(fTCAJetsOutRan)fTCAJetsOutRan->Delete();
741 if(fTCARandomConesOut)fTCARandomConesOut->Delete();
742 if(fTCARandomConesOutRan)fTCARandomConesOutRan->Delete();
743 if(fAODJetBackgroundOut)fAODJetBackgroundOut->Reset();
744
745 AliAODJetEventBackground* externalBackground = 0;
746 if(!externalBackground&&fBackgroundBranch.Length()){
747 externalBackground = (AliAODJetEventBackground*)(AODEvent()->FindListObject(fBackgroundBranch.Data()));
748 if((!externalBackground)&&fAODExtension)externalBackground = (AliAODJetEventBackground*)(fAODExtension->GetAOD()->FindListObject(fBackgroundBranch.Data()));
749 if(!externalBackground)Printf("%s:%d Background branch not found %s",(char*)__FILE__,__LINE__,fBackgroundBranch.Data());;
750 }
751 //
752 // Execute analysis for current event
753 //
754 AliESDEvent *fESD = 0;
755 if(fUseAODTrackInput){
756 fAOD = dynamic_cast<AliAODEvent*>(InputEvent());
757 if(!fAOD){
758 Printf("%s:%d AODEvent not found in Input Manager %d",(char*)__FILE__,__LINE__,fUseAODTrackInput);
759 return;
760 }
761 // fetch the header
762 }
763 else{
764 // assume that the AOD is in the general output...
765 fAOD = AODEvent();
766 if(!fAOD){
767 Printf("%s:%d AODEvent not found in the Output",(char*)__FILE__,__LINE__);
768 return;
769 }
770 if(fDebug>0){
771 fESD = dynamic_cast<AliESDEvent*> (InputEvent());
772 }
773 }
774
775 //Check if information is provided detector level effects
776 if(!fMomResH1 || !fMomResH2 || !fMomResH3) fUseTrPtResolutionSmearing = kFALSE;
777 if(!fhEffH1 || !fhEffH2 || !fhEffH3) fUseDiceEfficiency = kFALSE;
778
779 Bool_t selectEvent = false;
780 Bool_t physicsSelection = true;// handled by the framework(fInputHandler->IsEventSelected()&AliVEvent::kMB)==AliVEvent::kMB;
781
782 Float_t cent = 0;
783 Float_t zVtx = 0;
784 Int_t cenClass = -1;
785 if(fAOD){
786 const AliAODVertex *vtxAOD = fAOD->GetPrimaryVertex();
787 TString vtxTitle(vtxAOD->GetTitle());
788 zVtx = vtxAOD->GetZ();
789
790 cent = fAOD->GetHeader()->GetCentrality();
791 if(cent<10)cenClass = 0;
792 else if(cent<30)cenClass = 1;
793 else if(cent<50)cenClass = 2;
794 else if(cent<80)cenClass = 3;
795 if(physicsSelection){
796 fh1CentralityPhySel->Fill(cent);
797 fh1ZPhySel->Fill(zVtx);
798 }
799
800 if(fEventSelection){
801 if(vtxAOD->GetNContributors()>2&&!vtxTitle.Contains("TPCVertex")){
802 Float_t yvtx = vtxAOD->GetY();
803 Float_t xvtx = vtxAOD->GetX();
804 Float_t r2 = yvtx*yvtx+xvtx*xvtx;
805 if(TMath::Abs(zVtx)<fVtxZCut&&r2<fVtxR2Cut){ // apply vertex cut later on
806 if(physicsSelection){
807 selectEvent = true;
808 }
809 }
810 }
811 if(fCentCutUp>0){
812 if(cent<fCentCutLo||cent>fCentCutUp){
813 selectEvent = false;
814 }
815 }
816 }else{
817 selectEvent = true;
818 }
819 }
820
821
822 if(!selectEvent){
823 PostData(1, fHistList);
824 return;
825 }
826 fh1Centrality->Fill(cent);
827 fh1Z->Fill(zVtx);
828 fh1Trials->Fill("#sum{ntrials}",1);
829
830
831 if (fDebug > 10)Printf("%s:%d",(char*)__FILE__,__LINE__);
832
833 // ==== General variables needed
834
835
836
837 // we simply fetch the tracks/mc particles as a list of AliVParticles
838
839 TList recParticles;
840 TList genParticles;
841
842 Int_t nT = GetListOfTracks(&recParticles,fTrackTypeRec);
843 Float_t nCh = recParticles.GetEntries();
844 fh1Nch->Fill(nCh);
845 if(fDebug>2)Printf("%s:%d Selected Rec tracks: %d %d",(char*)__FILE__,__LINE__,nT,recParticles.GetEntries());
846 nT = GetListOfTracks(&genParticles,fTrackTypeGen);
847 if(fDebug>2)Printf("%s:%d Selected Gen tracks: %d %d",(char*)__FILE__,__LINE__,nT,genParticles.GetEntries());
848
849 // find the jets....
850
851 vector<fastjet::PseudoJet> inputParticlesRec;
852 vector<fastjet::PseudoJet> inputParticlesRecRan;
853
854 // Generate the random cones
855
856 AliAODJet vTmpRan(1,0,0,1);
857 for(int i = 0; i < recParticles.GetEntries(); i++){
858 AliVParticle *vp = (AliVParticle*)recParticles.At(i);
859
860 // Carefull energy is not well determined in real data, should not matter for p_T scheme?
861 // we take total momentum here
862
863 if((!fUseTrPtResolutionSmearing) && (!fUseDiceEfficiency)) {
864 //Add particles to fastjet in case we are not running toy model
865 fastjet::PseudoJet jInp(vp->Px(),vp->Py(),vp->Pz(),vp->P());
866 jInp.set_user_index(i);
867 inputParticlesRec.push_back(jInp);
868 }
869 else if(fUseDiceEfficiency) {
870
871 // Dice to decide if particle is kept or not - toy model for efficiency
872 //
873 Double_t rnd = fRandom->Uniform(1.);
874 Double_t pT = vp->Pt();
875 Double_t eff[3] = {0.};
876 Double_t pTtmp = pT;
877 if(pT>10.) pTtmp = 10.;
878 Double_t eff1 = fhEffH1->GetBinContent(fhEffH1->FindBin(pTtmp));
879 Double_t eff2 = fhEffH2->GetBinContent(fhEffH2->FindBin(pTtmp));
880 Double_t eff3 = fhEffH3->GetBinContent(fhEffH3->FindBin(pTtmp));
881 Int_t cat[3] = {0};
882 //Sort efficiencies from large to small
883 if(eff1>eff2 && eff1>eff3) {
884 eff[0] = eff1;
885 cat[0] = 1;
886 if(eff2>eff3) {
887 eff[1] = eff2;
888 eff[2] = eff3;
889 cat[1] = 2;
890 cat[2] = 3;
891 } else {
892 eff[1] = eff3;
893 eff[2] = eff2;
894 cat[1] = 3;
895 cat[2] = 2;
896 }
897 }
898 else if(eff2>eff1 && eff2>eff3) {
899 eff[0] = eff2;
900 cat[0] = 2;
901 if(eff1>eff3) {
902 eff[1] = eff1;
903 eff[2] = eff3;
904 cat[1] = 1;
905 cat[2] = 3;
906 } else {
907 eff[1] = eff3;
908 eff[2] = eff1;
909 cat[1] = 3;
910 cat[2] = 1;
911 }
912 }
913 else if(eff3>eff1 && eff3>eff2) {
914 eff[0] = eff3;
915 cat[0] = 3;
916 if(eff1>eff2) {
917 eff[1] = eff1;
918 eff[2] = eff2;
919 cat[1] = 1;
920 cat[2] = 2;
921 } else {
922 eff[1] = eff2;
923 eff[2] = eff1;
924 cat[1] = 2;
925 cat[2] = 1;
926 }
927 }
928
929 Double_t sumEff = eff[0]+eff[1]+eff[2];
930 fp1Efficiency->Fill(vp->Pt(),sumEff);
931 if(rnd>sumEff && pT > fDiceEfficiencyMinPt) continue;
932
933 if(fUseTrPtResolutionSmearing) {
934 //Smear momentum of generated particle
935 Double_t smear = 1.;
936 //Select hybrid track category
937 if(rnd<=eff[2])
938 smear = GetMomentumSmearing(cat[2],pT);
939 else if(rnd<=(eff[2]+eff[1]))
940 smear = GetMomentumSmearing(cat[1],pT);
941 else
942 smear = GetMomentumSmearing(cat[0],pT);
943
944 fp1PtResolution->Fill(vp->Pt(),smear);
945
946 Double_t sigma = vp->Pt()*smear;
947 Double_t pTrec = fRandom->Gaus(vp->Pt(),sigma);
948
949 Double_t phi = vp->Phi();
950 Double_t theta = 2.*TMath::ATan(TMath::Exp(-1.*(vp->Eta())));
951 Double_t pX = pTrec * TMath::Cos(phi);
952 Double_t pY = pTrec * TMath::Sin(phi);
953 Double_t pZ = pTrec/TMath::Tan(theta);
954 Double_t p=TMath::Sqrt(pTrec*pTrec+pZ*pZ);
955
956 fh2PtGenPtSmeared->Fill(vp->Pt(),pTrec);
957
958 fastjet::PseudoJet jInp(pX,pY,pZ,p);
959 jInp.set_user_index(i);
960 inputParticlesRec.push_back(jInp);
961
962 }
963 else {
964 fastjet::PseudoJet jInp(vp->Px(),vp->Py(),vp->Pz(),vp->P());
965 jInp.set_user_index(i);
966 inputParticlesRec.push_back(jInp);
967
968 }
969
970 }
971
972 // the randomized input changes eta and phi, but keeps the p_T
973 if(i>=fNSkipLeadingRan){// eventually skip the leading particles
974 Double_t pT = vp->Pt();
975 Double_t eta = 2.*fTrackEtaWindow * fRandom->Rndm() - fTrackEtaWindow;
976 Double_t phi = 2.* TMath::Pi() * fRandom->Rndm();
977
978 Double_t theta = 2.*TMath::ATan(TMath::Exp(-eta));
979 Double_t pZ = pT/TMath::Tan(theta);
980
981 Double_t pX = pT * TMath::Cos(phi);
982 Double_t pY = pT * TMath::Sin(phi);
983 Double_t p = TMath::Sqrt(pT*pT+pZ*pZ);
984 fastjet::PseudoJet jInpRan(pX,pY,pZ,p);
985
986 jInpRan.set_user_index(i);
987 inputParticlesRecRan.push_back(jInpRan);
988 vTmpRan.SetPxPyPzE(pX,pY,pZ,p);
989 }
990
991 // fill the tref array, only needed when we write out jets
992 if(fTCAJetsOut){
993 if(i == 0){
994 fRef->Delete(); // make sure to delete before placement new...
995 if((!fUseTrPtResolutionSmearing) && (!fUseDiceEfficiency)) {
996 new(fRef) TRefArray(TProcessID::GetProcessWithUID(vp)); //TRefArray does not work with toy model ...
997 }
998 }
999 if((!fUseTrPtResolutionSmearing) && (!fUseDiceEfficiency)) fRef->Add(vp); //TRefArray does not work with toy model ...
1000 }
1001 }// recparticles
1002
1003 if(inputParticlesRec.size()==0){
1004 if(fDebug)Printf("%s:%d No input particles found, skipping event",(char*)__FILE__,__LINE__);
1005 PostData(1, fHistList);
1006 return;
1007 }
1008
1009 // run fast jet
1010 // employ setters for these...
1011
1012
1013 // now create the object that holds info about ghosts
1014 /*
1015 if(!fUseBackgroundCalc&& fNonStdBranch.Length()==0){
1016 // reduce CPU time...
1017 fGhostArea = 0.5;
1018 fActiveAreaRepeats = 0;
1019 }
1020 */
1021
1022 fastjet::GhostedAreaSpec ghostSpec(fGhostEtamax, fActiveAreaRepeats, fGhostArea);
1023 fastjet::AreaType areaType = fastjet::active_area;
1024 fastjet::AreaDefinition areaDef = fastjet::AreaDefinition(areaType,ghostSpec);
1025 fastjet::JetDefinition jetDef(fAlgorithm, fRparam, fRecombScheme, fStrategy);
1026 fastjet::ClusterSequenceArea clustSeq(inputParticlesRec, jetDef,areaDef);
1027
1028 //range where to compute background
1029 Double_t phiMin = 0, phiMax = 0, rapMin = 0, rapMax = 0;
1030 phiMin = 0;
1031 phiMax = 2*TMath::Pi();
1032 rapMax = fGhostEtamax - fRparam;
1033 rapMin = - fGhostEtamax + fRparam;
1034 fastjet::RangeDefinition range(rapMin,rapMax, phiMin, phiMax);
1035
1036
1037 const vector <fastjet::PseudoJet> &inclusiveJets = clustSeq.inclusive_jets();
1038 const vector <fastjet::PseudoJet> &sortedJets = sorted_by_pt(inclusiveJets);
1039
1040
1041 fh1NJetsRec->Fill(sortedJets.size());
1042
1043 // loop over all jets an fill information, first on is the leading jet
1044
1045 Int_t nRecOver = inclusiveJets.size();
1046 Int_t nRec = inclusiveJets.size();
1047 if(inclusiveJets.size()>0){
1048 AliAODJet leadingJet (sortedJets[0].px(), sortedJets[0].py(), sortedJets[0].pz(), sortedJets[0].E());
1049 Double_t area = clustSeq.area(sortedJets[0]);
1050 leadingJet.SetEffArea(area,0);
1051 Float_t pt = leadingJet.Pt();
1052 Int_t nAodOutJets = 0;
1053 Int_t nAodOutTracks = 0;
1054 AliAODJet *aodOutJet = 0;
1055
1056 Int_t iCount = 0;
1057 for(int i = 1;i <= fh2NRecJetsPt->GetNbinsX();i++){
1058 Float_t ptCut = fh2NRecJetsPt->GetXaxis()->GetBinCenter(i);
1059 while(pt<ptCut&&iCount<nRec){
1060 nRecOver--;
1061 iCount++;
1062 if(iCount<nRec){
1063 pt = sortedJets[iCount].perp();
1064 }
1065 }
1066 if(nRecOver<=0)break;
1067 fh2NRecJetsPt->Fill(ptCut,nRecOver);
1068 }
1069 Float_t phi = leadingJet.Phi();
1070 if(phi<0)phi+=TMath::Pi()*2.;
1071 Float_t eta = leadingJet.Eta();
1072 Float_t pTback = 0;
1073 if(externalBackground){
1074 // carefull has to be filled in a task before
1075 // todo, ReArrange to the botom
1076 pTback = externalBackground->GetBackground(2)*leadingJet.EffectiveAreaCharged();
1077 }
1078 pt = leadingJet.Pt() - pTback;
1079 // correlation of leading jet with tracks
1080 TIterator *recIter = recParticles.MakeIterator();
1081 recIter->Reset();
1082 AliVParticle *tmpRecTrack = 0;
1083 while((tmpRecTrack = (AliVParticle*)(recIter->Next()))){
1084 Float_t tmpPt = tmpRecTrack->Pt();
1085 // correlation
1086 Float_t tmpPhi = tmpRecTrack->Phi();
1087 if(tmpPhi<0)tmpPhi+=TMath::Pi()*2.;
1088 Float_t dPhi = phi - tmpPhi;
1089 if(dPhi>TMath::Pi())dPhi = dPhi - 2.*TMath::Pi();
1090 if(dPhi<(-1.*TMath::Pi()))dPhi = dPhi + 2.*TMath::Pi();
1091 fh2TracksLeadingJetPhiPt->Fill(dPhi,pt);
1092 fh2TracksLeadingJetPhiPtW->Fill(dPhi,pt,tmpPt);
1093 if(cenClass>=0){
1094 fh2TracksLeadingJetPhiPtC[cenClass]->Fill(dPhi,pt);
1095 fh2TracksLeadingJetPhiPtWC[cenClass]->Fill(dPhi,pt,tmpPt);
1096 }
1097
1098 }
1099
1100
1101 TLorentzVector vecareab;
1102 for(int j = 0; j < nRec;j++){
1103 AliAODJet tmpRec (sortedJets[j].px(), sortedJets[j].py(), sortedJets[j].pz(), sortedJets[j].E());
1104 aodOutJet = 0;
1105 nAodOutTracks = 0;
1106 Float_t tmpPt = tmpRec.Pt();
1107
1108 if(tmpPt>fJetOutputMinPt&&fTCAJetsOut){// cut on the non-background subtracted...
1109 aodOutJet = new ((*fTCAJetsOut)[nAodOutJets++]) AliAODJet(tmpRec);
1110 aodOutJet->GetRefTracks()->Clear();
1111 Double_t area1 = clustSeq.area(sortedJets[j]);
1112 aodOutJet->SetEffArea(area1,0);
1113 fastjet::PseudoJet vecarea=clustSeq.area_4vector(sortedJets[j]);
1114 vecareab.SetPxPyPzE(vecarea.px(),vecarea.py(),vecarea.pz(),vecarea.e());
1115 aodOutJet->SetVectorAreaCharged(&vecareab);
1116 }
1117
1118
1119 Float_t tmpPtBack = 0;
1120 if(externalBackground){
1121 // carefull has to be filled in a task before
1122 // todo, ReArrange to the botom
1123 tmpPtBack = externalBackground->GetBackground(2)*tmpRec.EffectiveAreaCharged();
1124 }
1125 tmpPt = tmpPt - tmpPtBack;
1126 if(tmpPt<0)tmpPt = 0; // avoid negative weights...
1127
1128 fh1PtJetsRecIn->Fill(tmpPt);
1129 // Fill Spectra with constituentsemacs
1130 const vector<fastjet::PseudoJet> &constituents = clustSeq.constituents(sortedJets[j]);
1131
1132 fh1NConstRec->Fill(constituents.size());
1133 fh2PtNch->Fill(nCh,tmpPt);
1134 fh2PtNchN->Fill(nCh,tmpPt,constituents.size());
1135 fh2NConstPt->Fill(tmpPt,constituents.size());
1136 // loop over constiutents and fill spectrum
1137
1138 AliVParticle *partLead = 0;
1139 Float_t ptLead = -1;
1140
1141 for(unsigned int ic = 0; ic < constituents.size();ic++){
1142 AliVParticle *part = (AliVParticle*)recParticles.At(constituents[ic].user_index());
1143 if(!part) continue;
1144 fh1PtJetConstRec->Fill(part->Pt());
1145 if(aodOutJet){
1146 if((!fUseTrPtResolutionSmearing) && (!fUseDiceEfficiency)) aodOutJet->AddTrack(fRef->At(constituents[ic].user_index()));
1147 if(part->Pt()>fMaxTrackPtInJet){
1148 aodOutJet->SetTrigger(AliAODJet::kHighTrackPtTriggered);
1149 }
1150 }
1151 if(part->Pt()>ptLead){
1152 ptLead = part->Pt();
1153 partLead = part;
1154 }
1155 if(j==0)fh1PtJetConstLeadingRec->Fill(part->Pt());
1156 }
1157
1158 AliAODTrack *aodT = 0;
1159 if(partLead&&aodOutJet){
1160 //set pT of leading constituent of jet
1161 aodOutJet->SetPtLeading(partLead->Pt());
1162 aodT = dynamic_cast<AliAODTrack*>(partLead);
1163 if(aodT){
1164 if(aodT->TestFilterBit(fFilterMaskBestPt)){
1165 aodOutJet->SetTrigger(AliAODJet::kHighTrackPtBest);
1166 }
1167 }
1168 }
1169
1170 // correlation
1171 Float_t tmpPhi = tmpRec.Phi();
1172 Float_t tmpEta = tmpRec.Eta();
1173 if(tmpPhi<0)tmpPhi+=TMath::Pi()*2.;
1174 if(j==0){
1175 fh1PtJetsLeadingRecIn->Fill(tmpPt);
1176 fh2LeadingJetPhiEta->Fill(tmpPhi,tmpEta);
1177 fh2LeadingJetEtaPt->Fill(tmpEta,tmpPt);
1178 fh1NConstLeadingRec->Fill(constituents.size());
1179 fh2NConstLeadingPt->Fill(tmpPt,constituents.size());
1180 continue;
1181 }
1182 fh2JetPhiEta->Fill(tmpRec.Phi(),tmpEta);
1183 fh2JetEtaPt->Fill(tmpEta,tmpPt);
1184 Float_t dPhi = phi - tmpPhi;
1185 if(dPhi>TMath::Pi())dPhi = dPhi - 2.*TMath::Pi();
1186 if(dPhi<(-1.*TMath::Pi()))dPhi = dPhi + 2.*TMath::Pi();
1187 Float_t dEta = eta - tmpRec.Eta();
1188 fh2JetsLeadingPhiEta->Fill(dPhi,dEta);
1189 fh2JetsLeadingPhiPt->Fill(dPhi,pt);
1190 if(cenClass>=0){
1191 fh2JetsLeadingPhiPtC[cenClass]->Fill(dPhi,pt);
1192 fh2JetsLeadingPhiPtWC[cenClass]->Fill(dPhi,pt,tmpPt);
1193 }
1194 fh2JetsLeadingPhiPtW->Fill(dPhi,pt,tmpPt);
1195 }// loop over reconstructed jets
1196 delete recIter;
1197
1198
1199
1200 // Add the random cones...
1201 if(fNRandomCones>0&&fTCARandomConesOut){
1202 // create a random jet within the acceptance
1203 Double_t etaMax = fTrackEtaWindow - fRparam;
1204 Int_t nCone = 0;
1205 Int_t nConeRan = 0;
1206 Double_t pTC = 1; // small number
1207 for(int ir = 0;ir < fNRandomCones;ir++){
1208 Double_t etaC = etaMax*2.*(fRandom->Rndm()-0.5); // +- etamax
1209 Double_t phiC = fRandom->Rndm()*2.*TMath::Pi(); // 0 - 2pi
1210 // massless jet
1211 Double_t thetaC = 2.*TMath::ATan(TMath::Exp(-etaC));
1212 Double_t pZC = pTC/TMath::Tan(thetaC);
1213 Double_t pXC = pTC * TMath::Cos(phiC);
1214 Double_t pYC = pTC * TMath::Sin(phiC);
1215 Double_t pC = TMath::Sqrt(pTC*pTC+pZC*pZC);
1216 AliAODJet tmpRecC (pXC,pYC,pZC, pC);
1217 bool skip = false;
1218 for(int jj = 0; jj < TMath::Min(nRec,fNSkipLeadingCone);jj++){// test for overlap with leading jets
1219 AliAODJet jet (sortedJets[jj].px(), sortedJets[jj].py(), sortedJets[jj].pz(), sortedJets[jj].E());
1220 if(jet.DeltaR(& tmpRecC)<2.*fRparam+0.2){
1221 skip = true;
1222 break;
1223 }
1224 }
1225 // test for overlap with previous cones to avoid double counting
1226 for(int iic = 0;iic<ir;iic++){
1227 AliAODJet *iicone = (AliAODJet*)fTCARandomConesOut->At(iic);
1228 if(iicone){
1229 if(iicone->DeltaR(&tmpRecC)<2.*fRparam){
1230 skip = true;
1231 break;
1232 }
1233 }
1234 }
1235 if(skip)continue;
1236 tmpRecC.SetBgEnergy(0,0); // this is use as temporary storage of the summed p_T below
1237 tmpRecC.SetPtLeading(-1.);
1238 if(fTCARandomConesOut)new ((*fTCARandomConesOut)[nCone++]) AliAODJet(tmpRecC);
1239 if(fTCARandomConesOutRan)new ((*fTCARandomConesOutRan)[nConeRan++]) AliAODJet(tmpRecC);
1240 }// loop over random cones creation
1241
1242
1243 // loop over the reconstructed particles and add up the pT in the random cones
1244 // maybe better to loop over randomized particles not in the real jets...
1245 // but this by definition brings dow average energy in the whole event
1246 AliAODJet vTmpRanR(1,0,0,1);
1247 for(int i = 0; i < recParticles.GetEntries(); i++){
1248 AliVParticle *vp = (AliVParticle*)recParticles.At(i);
1249 if(fTCARandomConesOut){
1250 for(int ir = 0;ir < fNRandomCones;ir++){
1251 AliAODJet *jC = (AliAODJet*)fTCARandomConesOut->At(ir);
1252 if(jC&&jC->DeltaR(vp)<fRparam){
1253 if(vp->Pt()>fMaxTrackPtInJet)jC->SetTrigger(AliAODJet::kHighTrackPtTriggered);
1254 jC->SetBgEnergy(jC->ChargedBgEnergy()+vp->Pt(),0);
1255 if(vp->Pt() > jC->GetPtLeading()) jC->SetPtLeading(vp->Pt());
1256 }
1257 }
1258 }// add up energy in cone
1259
1260 // the randomized input changes eta and phi, but keeps the p_T
1261 if(i>=fNSkipLeadingRan){// eventually skip the leading particles
1262 Double_t pTR = vp->Pt();
1263 Double_t etaR = 2.*fTrackEtaWindow* fRandom->Rndm() - fTrackEtaWindow;
1264 Double_t phiR = 2.* TMath::Pi() * fRandom->Rndm();
1265
1266 Double_t thetaR = 2.*TMath::ATan(TMath::Exp(-etaR));
1267 Double_t pZR = pTR/TMath::Tan(thetaR);
1268
1269 Double_t pXR = pTR * TMath::Cos(phiR);
1270 Double_t pYR = pTR * TMath::Sin(phiR);
1271 Double_t pR = TMath::Sqrt(pTR*pTR+pZR*pZR);
1272 vTmpRanR.SetPxPyPzE(pXR,pYR,pZR,pR);
1273 if(fTCARandomConesOutRan){
1274 for(int ir = 0;ir < fTCARandomConesOutRan->GetEntriesFast();ir++){
1275 AliAODJet *jC = (AliAODJet*)fTCARandomConesOutRan->At(ir);
1276 if(jC&&jC->DeltaR(&vTmpRanR)<fRparam){
1277 if(vTmpRanR.Pt()>fMaxTrackPtInJet)jC->SetTrigger(AliAODJet::kHighTrackPtTriggered);
1278 jC->SetBgEnergy(jC->ChargedBgEnergy()+vTmpRanR.Pt(),0);
1279 if(vTmpRanR.Pt() > jC->GetPtLeading()) jC->SetPtLeading(vTmpRanR.Pt());
1280 }
1281 }
1282 }
1283 }
1284 }// loop over recparticles
1285
1286 Float_t jetArea = fRparam*fRparam*TMath::Pi();
1287 if(fTCARandomConesOut){
1288 for(int ir = 0;ir < fTCARandomConesOut->GetEntriesFast();ir++){
1289 // rescale the momentum vectors for the random cones
1290
1291 AliAODJet *rC = (AliAODJet*)fTCARandomConesOut->At(ir);
1292 if(rC){
1293 Double_t etaC = rC->Eta();
1294 Double_t phiC = rC->Phi();
1295 // massless jet, unit vector
1296 pTC = rC->ChargedBgEnergy();
1297 if(pTC<=0)pTC = 0.001; // for almost empty events
1298 Double_t thetaC = 2.*TMath::ATan(TMath::Exp(-etaC));
1299 Double_t pZC = pTC/TMath::Tan(thetaC);
1300 Double_t pXC = pTC * TMath::Cos(phiC);
1301 Double_t pYC = pTC * TMath::Sin(phiC);
1302 Double_t pC = TMath::Sqrt(pTC*pTC+pZC*pZC);
1303 rC->SetPxPyPzE(pXC,pYC,pZC, pC);
1304 rC->SetBgEnergy(0,0);
1305 rC->SetEffArea(jetArea,0);
1306 }
1307 }
1308 }
1309 if(fTCARandomConesOutRan){
1310 for(int ir = 0;ir < fTCARandomConesOutRan->GetEntriesFast();ir++){
1311 AliAODJet* rC = (AliAODJet*)fTCARandomConesOutRan->At(ir);
1312 // same wit random
1313 if(rC){
1314 Double_t etaC = rC->Eta();
1315 Double_t phiC = rC->Phi();
1316 // massless jet, unit vector
1317 pTC = rC->ChargedBgEnergy();
1318 if(pTC<=0)pTC = 0.001;// for almost empty events
1319 Double_t thetaC = 2.*TMath::ATan(TMath::Exp(-etaC));
1320 Double_t pZC = pTC/TMath::Tan(thetaC);
1321 Double_t pXC = pTC * TMath::Cos(phiC);
1322 Double_t pYC = pTC * TMath::Sin(phiC);
1323 Double_t pC = TMath::Sqrt(pTC*pTC+pZC*pZC);
1324 rC->SetPxPyPzE(pXC,pYC,pZC, pC);
1325 rC->SetBgEnergy(0,0);
1326 rC->SetEffArea(jetArea,0);
1327 }
1328 }
1329 }
1330 }// if(fNRandomCones
1331
1332 //background estimates:all bckg jets(0) & wo the 2 hardest(1)
1333
1334
1335
1336
1337
1338 if(fAODJetBackgroundOut){
1339 vector<fastjet::PseudoJet> jets2=sortedJets;
1340 if(jets2.size()>2) jets2.erase(jets2.begin(),jets2.begin()+2);
1341 Double_t bkg1=0;
1342 Double_t sigma1=0.;
1343 Double_t meanarea1=0.;
1344 Double_t bkg2=0;
1345 Double_t sigma2=0.;
1346 Double_t meanarea2=0.;
1347
1348 clustSeq.get_median_rho_and_sigma(jets2, range, true, bkg1, sigma1, meanarea1, true);
1349 fAODJetBackgroundOut->SetBackground(0,bkg1,sigma1,meanarea1);
1350
1351 // fh1BiARandomCones[0]->Fill(omCone-(bkg1*areaRandomCone));
1352 // fh1BiARandomConesRan[0]->Fill(ptRandomConeRan-(bkg1*areaRandomCone));
1353
1354 clustSeq.get_median_rho_and_sigma(jets2, range, false, bkg2, sigma2, meanarea2, true);
1355 fAODJetBackgroundOut->SetBackground(1,bkg2,sigma2,meanarea2);
1356 // fh1BiARandomCones[1]->Fill(ptRandomCone-(bkg2*areaRandomCone));
1357 // fh1BiARandomConesRan[1]->Fill(ptRandomConeRan-(bkg2*areaRandomCone));
1358
1359 }
1360 }
1361
1362
1363
1364
1365
1366 // fill track information
1367 Int_t nTrackOver = recParticles.GetSize();
1368 // do the same for tracks and jets
1369
1370 if(nTrackOver>0){
1371 TIterator *recIter = recParticles.MakeIterator();
1372 AliVParticle *tmpRec = (AliVParticle*)(recIter->Next());
1373 Float_t pt = tmpRec->Pt();
1374
1375 // Printf("Leading track p_t %3.3E",pt);
1376 for(int i = 1;i <= fh2NRecTracksPt->GetNbinsX();i++){
1377 Float_t ptCut = fh2NRecTracksPt->GetXaxis()->GetBinCenter(i);
1378 while(pt<ptCut&&tmpRec){
1379 nTrackOver--;
1380 tmpRec = (AliVParticle*)(recIter->Next());
1381 if(tmpRec){
1382 pt = tmpRec->Pt();
1383 }
1384 }
1385 if(nTrackOver<=0)break;
1386 fh2NRecTracksPt->Fill(ptCut,nTrackOver);
1387 }
1388
1389 recIter->Reset();
1390 AliVParticle *leading = (AliVParticle*)recParticles.At(0);
1391 Float_t phi = leading->Phi();
1392 if(phi<0)phi+=TMath::Pi()*2.;
1393 Float_t eta = leading->Eta();
1394 pt = leading->Pt();
1395 while((tmpRec = (AliVParticle*)(recIter->Next()))){
1396 Float_t tmpPt = tmpRec->Pt();
1397 Float_t tmpEta = tmpRec->Eta();
1398 fh1PtTracksRecIn->Fill(tmpPt);
1399 fh2TrackEtaPt->Fill(tmpEta,tmpPt);
1400 if(tmpRec==leading){
1401 fh1PtTracksLeadingRecIn->Fill(tmpPt);
1402 fh2LeadingTrackEtaPt->Fill(tmpEta,tmpPt);
1403 continue;
1404 }
1405 // correlation
1406 Float_t tmpPhi = tmpRec->Phi();
1407
1408 if(tmpPhi<0)tmpPhi+=TMath::Pi()*2.;
1409 Float_t dPhi = phi - tmpPhi;
1410 if(dPhi>TMath::Pi())dPhi = dPhi - 2.*TMath::Pi();
1411 if(dPhi<(-1.*TMath::Pi()))dPhi = dPhi + 2.*TMath::Pi();
1412 Float_t dEta = eta - tmpRec->Eta();
1413 fh2TracksLeadingPhiEta->Fill(dPhi,dEta);
1414 fh2TracksLeadingPhiPt->Fill(dPhi,pt);
1415 fh2TracksLeadingPhiPtW->Fill(dPhi,pt,tmpPt);
1416 }
1417 delete recIter;
1418 }
1419
1420 // find the random jets
1421
1422 fastjet::ClusterSequenceArea clustSeqRan(inputParticlesRecRan, jetDef, areaDef);
1423
1424 // fill the jet information from random track
1425 const vector <fastjet::PseudoJet> &inclusiveJetsRan = clustSeqRan.inclusive_jets();
1426 const vector <fastjet::PseudoJet> &sortedJetsRan = sorted_by_pt(inclusiveJetsRan);
1427
1428 fh1NJetsRecRan->Fill(sortedJetsRan.size());
1429
1430 // loop over all jets an fill information, first on is the leading jet
1431
1432 Int_t nRecOverRan = inclusiveJetsRan.size();
1433 Int_t nRecRan = inclusiveJetsRan.size();
1434
1435 if(inclusiveJetsRan.size()>0){
1436 AliAODJet leadingJet (sortedJetsRan[0].px(), sortedJetsRan[0].py(), sortedJetsRan[0].pz(), sortedJetsRan[0].E());
1437 Float_t pt = leadingJet.Pt();
1438
1439 Int_t iCount = 0;
1440 TLorentzVector vecarearanb;
1441
1442 for(int i = 1;i <= fh2NRecJetsPtRan->GetNbinsX();i++){
1443 Float_t ptCut = fh2NRecJetsPtRan->GetXaxis()->GetBinCenter(i);
1444 while(pt<ptCut&&iCount<nRecRan){
1445 nRecOverRan--;
1446 iCount++;
1447 if(iCount<nRecRan){
1448 pt = sortedJetsRan[iCount].perp();
1449 }
1450 }
1451 if(nRecOverRan<=0)break;
1452 fh2NRecJetsPtRan->Fill(ptCut,nRecOverRan);
1453 }
1454 Float_t phi = leadingJet.Phi();
1455 if(phi<0)phi+=TMath::Pi()*2.;
1456 pt = leadingJet.Pt();
1457
1458 // correlation of leading jet with random tracks
1459
1460 for(unsigned int ip = 0; ip < inputParticlesRecRan.size();ip++)
1461 {
1462 Float_t tmpPt = inputParticlesRecRan[ip].perp();
1463 // correlation
1464 Float_t tmpPhi = inputParticlesRecRan[ip].phi();
1465 if(tmpPhi<0)tmpPhi+=TMath::Pi()*2.;
1466 Float_t dPhi = phi - tmpPhi;
1467 if(dPhi>TMath::Pi())dPhi = dPhi - 2.*TMath::Pi();
1468 if(dPhi<(-1.*TMath::Pi()))dPhi = dPhi + 2.*TMath::Pi();
1469 fh2TracksLeadingJetPhiPtRan->Fill(dPhi,pt);
1470 fh2TracksLeadingJetPhiPtWRan->Fill(dPhi,pt,tmpPt);
1471 }
1472
1473 Int_t nAodOutJetsRan = 0;
1474 AliAODJet *aodOutJetRan = 0;
1475 for(int j = 0; j < nRecRan;j++){
1476 AliAODJet tmpRec (sortedJetsRan[j].px(), sortedJetsRan[j].py(), sortedJetsRan[j].pz(), sortedJetsRan[j].E());
1477 Float_t tmpPt = tmpRec.Pt();
1478 fh1PtJetsRecInRan->Fill(tmpPt);
1479 // Fill Spectra with constituents
1480 const vector<fastjet::PseudoJet> &constituents = clustSeqRan.constituents(sortedJetsRan[j]);
1481 fh1NConstRecRan->Fill(constituents.size());
1482 fh2NConstPtRan->Fill(tmpPt,constituents.size());
1483 fh2PtNchRan->Fill(nCh,tmpPt);
1484 fh2PtNchNRan->Fill(nCh,tmpPt,constituents.size());
1485
1486
1487 if(tmpPt>fJetOutputMinPt&&fTCAJetsOutRan){
1488 aodOutJetRan = new ((*fTCAJetsOutRan)[nAodOutJetsRan++]) AliAODJet(tmpRec);
1489 Double_t arearan=clustSeqRan.area(sortedJetsRan[j]);
1490 aodOutJetRan->GetRefTracks()->Clear();
1491 aodOutJetRan->SetEffArea(arearan,0);
1492 fastjet::PseudoJet vecarearan=clustSeqRan.area_4vector(sortedJetsRan[j]);
1493 vecarearanb.SetPxPyPzE(vecarearan.px(),vecarearan.py(),vecarearan.pz(),vecarearan.e());
1494 aodOutJetRan->SetVectorAreaCharged(&vecarearanb);
1495
1496 }
1497
1498 // correlation
1499 Float_t tmpPhi = tmpRec.Phi();
1500 if(tmpPhi<0)tmpPhi+=TMath::Pi()*2.;
1501
1502 if(j==0){
1503 fh1PtJetsLeadingRecInRan->Fill(tmpPt);
1504 fh1NConstLeadingRecRan->Fill(constituents.size());
1505 fh2NConstLeadingPtRan->Fill(tmpPt,constituents.size());
1506 continue;
1507 }
1508 }
1509
1510
1511 if(fAODJetBackgroundOut){
1512 Double_t bkg3=0.;
1513 Double_t sigma3=0.;
1514 Double_t meanarea3=0.;
1515 clustSeqRan.get_median_rho_and_sigma(sortedJetsRan ,range, false, bkg3, sigma3, meanarea3, true);
1516 fAODJetBackgroundOut->SetBackground(2,bkg3,sigma3,meanarea3);
1517 // float areaRandomCone = rRandomCone2 *TMath::Pi();
1518 /*
1519 fh1BiARandomCones[2]->Fill(ptRandomCone-(bkg3*areaRandomCone));
1520 fh1BiARandomConesRan[2]->Fill(ptRandomConeRan-(bkg3*areaRandomCone));
1521 */
1522 }
1523
1524
1525
1526 }
1527
1528
1529 // do the event selection if activated
1530 if(fJetTriggerPtCut>0){
1531 bool select = false;
1532 Float_t minPt = fJetTriggerPtCut;
1533 /*
1534 // hard coded for now ...
1535 // 54.50 44.5 29.5 18.5 for anti-kt rejection 1E-3
1536 if(cent<10)minPt = 50;
1537 else if(cent<30)minPt = 42;
1538 else if(cent<50)minPt = 28;
1539 else if(cent<80)minPt = 18;
1540 */
1541 float rho = 0;
1542 if(externalBackground)rho = externalBackground->GetBackground(2);
1543 if(fTCAJetsOut){
1544 for(int i = 0;i < fTCAJetsOut->GetEntriesFast();i++){
1545 AliAODJet *jet = (AliAODJet*)fTCAJetsOut->At(i);
1546 Float_t ptSub = jet->Pt() - rho *jet->EffectiveAreaCharged();
1547 if(ptSub>=minPt){
1548 select = true;
1549 break;
1550 }
1551 }
1552 }
1553
1554 if(select){
1555 static AliAODHandler *aodH = dynamic_cast<AliAODHandler*>(AliAnalysisManager::GetAnalysisManager()->GetOutputEventHandler());
1556 fh1CentralitySelect->Fill(cent);
1557 fh1ZSelect->Fill(zVtx);
1558 aodH->SetFillAOD(kTRUE);
1559 }
1560 }
1561 if (fDebug > 2){
1562 if(fTCAJetsOut)Printf("%s:%d Rec Jets %d",(char*)__FILE__,__LINE__,fTCAJetsOut->GetEntriesFast());
1563 if(fTCAJetsOutRan)Printf("%s:%d Rec Jets Ran %d",(char*)__FILE__,__LINE__,fTCAJetsOutRan->GetEntriesFast());
1564 if(fTCARandomConesOut)Printf("%s:%d RC %d",(char*)__FILE__,__LINE__,fTCARandomConesOut->GetEntriesFast());
1565 if(fTCARandomConesOutRan)Printf("%s:%d RC Ran %d",(char*)__FILE__,__LINE__,fTCARandomConesOutRan->GetEntriesFast());
1566 }
1567 PostData(1, fHistList);
1568}
1569
1570void AliAnalysisTaskJetCluster::Terminate(Option_t */*option*/)
1571{
1572 //
1573 // Terminate analysis
1574 //
1575 if (fDebug > 1) printf("AnalysisJetCluster: Terminate() \n");
1576
1577 if(fMomResH1Fit) delete fMomResH1Fit;
1578 if(fMomResH2Fit) delete fMomResH2Fit;
1579 if(fMomResH3Fit) delete fMomResH3Fit;
1580
1581}
1582
1583
1584Int_t AliAnalysisTaskJetCluster::GetListOfTracks(TList *list,Int_t type){
1585
1586 //
1587 // get list of tracks/particles for different types
1588 //
1589
1590 if(fDebug>2)Printf("%s:%d Selecting tracks with %d",(char*)__FILE__,__LINE__,type);
1591
1592 Int_t iCount = 0;
1593 if(type==kTrackAOD || type==kTrackAODextra || type==kTrackAODextraonly){
1594 if(type!=kTrackAODextraonly) {
1595 AliAODEvent *aod = 0;
1596 if(fUseAODTrackInput)aod = dynamic_cast<AliAODEvent*>(InputEvent());
1597 else aod = AODEvent();
1598 if(!aod){
1599 if(fDebug>2)Printf("%s:%d No AOD",(char*)__FILE__,__LINE__);
1600 return iCount;
1601 }
1602
1603 for(int it = 0;it < aod->GetNumberOfTracks();++it){
1604 AliAODTrack *tr = aod->GetTrack(it);
1605 Bool_t bGood = false;
1606 if(fFilterType == 0)bGood = true;
1607 else if(fFilterType == 1)bGood = tr->IsHybridTPCConstrainedGlobal();
1608 else if(fFilterType == 2)bGood = tr->IsHybridGlobalConstrainedGlobal();
1609 if((fFilterMask>0)&&((!tr->TestFilterBit(fFilterMask)||(!bGood)))){
1610 if(fDebug>10)Printf("%s:%d Not matching filter %d/%d %d/%d",(char*)__FILE__,__LINE__,it,aod->GetNumberOfTracks(),fFilterMask,tr->GetFilterMap());
1611 continue;
1612 }
1613 if(TMath::Abs(tr->Eta())>fTrackEtaWindow){
1614 if(fDebug>10)Printf("%s:%d Not matching eta %d/%d",(char*)__FILE__,__LINE__,it,aod->GetNumberOfTracks());
1615 continue;
1616 }
1617 if(tr->Pt()<fTrackPtCut){
1618 if(fDebug>10)Printf("%s:%d Not matching pt %d/%d",(char*)__FILE__,__LINE__,it,aod->GetNumberOfTracks());
1619 continue;
1620 }
1621 if(fDebug>10)Printf("%s:%d MATCHED %d/%d",(char*)__FILE__,__LINE__,it,aod->GetNumberOfTracks());
1622 list->Add(tr);
1623 iCount++;
1624 }
1625 }
1626 if(type==kTrackAODextra || type==kTrackAODextraonly) {
1627 AliAODEvent *aod = 0;
1628 if(fUseAODTrackInput)aod = dynamic_cast<AliAODEvent*>(InputEvent());
1629 else aod = AODEvent();
1630
1631 if(!aod){
1632 return iCount;
1633 }
1634 TClonesArray *aodExtraTracks = dynamic_cast<TClonesArray*>(aod->FindListObject("aodExtraTracks"));
1635 if(!aodExtraTracks)return iCount;
1636 for(int it =0; it<aodExtraTracks->GetEntries(); it++) {
1637 AliVParticle *track = dynamic_cast<AliVParticle*> ((*aodExtraTracks)[it]);
1638 if (!track) continue;
1639
1640 AliAODTrack *trackAOD = dynamic_cast<AliAODTrack*> (track);
1641 if(!trackAOD)continue;
1642 Bool_t bGood = false;
1643 if(fFilterType == 0)bGood = true;
1644 else if(fFilterType == 1)bGood = trackAOD->IsHybridTPCConstrainedGlobal();
1645 else if(fFilterType == 2)bGood = trackAOD->IsHybridGlobalConstrainedGlobal();
1646 if((fFilterMask>0)&&((!trackAOD->TestFilterBit(fFilterMask)||(!bGood))))continue;
1647 if(TMath::Abs(trackAOD->Eta())>fTrackEtaWindow) continue;
1648 if(trackAOD->Pt()<fTrackPtCut) continue;
1649 list->Add(trackAOD);
1650 iCount++;
1651 }
1652 }
1653 }
1654 else if (type == kTrackKineAll||type == kTrackKineCharged){
1655 AliMCEvent* mcEvent = MCEvent();
1656 if(!mcEvent)return iCount;
1657 // we want to have alivpartilces so use get track
1658 for(int it = 0;it < mcEvent->GetNumberOfTracks();++it){
1659 if(!mcEvent->IsPhysicalPrimary(it))continue;
1660 AliMCParticle* part = (AliMCParticle*)mcEvent->GetTrack(it);
1661 if(type == kTrackKineAll){
1662 if(part->Pt()<fTrackPtCut)continue;
1663 list->Add(part);
1664 iCount++;
1665 }
1666 else if(type == kTrackKineCharged){
1667 if(part->Particle()->GetPDG()->Charge()==0)continue;
1668 if(part->Pt()<fTrackPtCut)continue;
1669 list->Add(part);
1670 iCount++;
1671 }
1672 }
1673 }
1674 else if (type == kTrackAODMCCharged || type == kTrackAODMCAll || type == kTrackAODMCChargedAcceptance) {
1675 AliAODEvent *aod = 0;
1676 if(fUseAODMCInput)aod = dynamic_cast<AliAODEvent*>(InputEvent());
1677 else aod = AODEvent();
1678 if(!aod)return iCount;
1679 TClonesArray *tca = dynamic_cast<TClonesArray*>(aod->FindListObject(AliAODMCParticle::StdBranchName()));
1680 if(!tca)return iCount;
1681 for(int it = 0;it < tca->GetEntriesFast();++it){
1682 AliAODMCParticle *part = (AliAODMCParticle*)(tca->At(it));
1683 if(!part->IsPhysicalPrimary())continue;
1684 if(type == kTrackAODMCAll){
1685 if(part->Pt()<fTrackPtCut)continue;
1686 list->Add(part);
1687 iCount++;
1688 }
1689 else if (type == kTrackAODMCCharged || type == kTrackAODMCChargedAcceptance ){
1690 if(part->Charge()==0)continue;
1691 if(part->Pt()<fTrackPtCut)continue;
1692 if(kTrackAODMCCharged){
1693 list->Add(part);
1694 }
1695 else {
1696 if(TMath::Abs(part->Eta())>fTrackEtaWindow)continue;
1697 list->Add(part);
1698 }
1699 iCount++;
1700 }
1701 }
1702 }// AODMCparticle
1703 list->Sort();
1704 return iCount;
1705}
1706
1707void AliAnalysisTaskJetCluster::LoadTrPtResolutionRootFileFromOADB() {
1708
1709 TFile *f = TFile::Open(fPathTrPtResolution.Data());
1710
1711 if(!f)return;
1712
1713 TProfile *fProfPtPtSigma1PtGlobSt = (TProfile*)f->Get("fProfPtPtSigma1PtGlobSt");
1714 TProfile *fProfPtPtSigma1PtGlobCnoITS = (TProfile*)f->Get("fProfPtPtSigma1PtGlobCnoITS");
1715 TProfile *fProfPtPtSigma1PtGlobCnoSPD = (TProfile*)f->Get("fProfPtPtSigma1PtGlobCnoSPD");
1716
1717 SetSmearResolution(kTRUE);
1718 SetMomentumResolutionHybrid(fProfPtPtSigma1PtGlobSt,fProfPtPtSigma1PtGlobCnoITS,fProfPtPtSigma1PtGlobCnoSPD);
1719
1720
1721}
1722
1723void AliAnalysisTaskJetCluster::LoadTrEfficiencyRootFileFromOADB() {
1724
1725 TFile *f = TFile::Open(fPathTrEfficiency.Data());
1726 if(!f)return;
1727
1728 TH1D *hEffPosGlobSt = (TH1D*)f->Get("hEffPosGlobSt");
1729 TH1D *hEffPosGlobCnoITS = (TH1D*)f->Get("hEffPosGlobCnoITS");
1730 TH1D *hEffPosGlobCnoSPD = (TH1D*)f->Get("hEffPosGlobCnoSPD");
1731
1732 SetDiceEfficiency(kTRUE);
1733
1734 if(fChangeEfficiencyFraction>0.) {
1735
1736 TH1D *hEffPosGlobStMin = (TH1D*)hEffPosGlobSt->Clone("hEffPosGlobStMin");
1737
1738 for(int bin=1; bin<=hEffPosGlobSt->GetNbinsX(); bin++) {
1739 Double_t content = hEffPosGlobSt->GetBinContent(bin);
1740 hEffPosGlobStMin->SetBinContent(bin,content-fChangeEfficiencyFraction);
1741 }
1742
1743 SetEfficiencyHybrid(hEffPosGlobStMin,hEffPosGlobCnoITS,hEffPosGlobCnoSPD);
1744
1745 }
1746 else
1747 SetEfficiencyHybrid(hEffPosGlobSt,hEffPosGlobCnoITS,hEffPosGlobCnoSPD);
1748
1749}
1750
1751void AliAnalysisTaskJetCluster::SetMomentumResolutionHybrid(TProfile *p1, TProfile *p2, TProfile *p3) {
1752
1753 //
1754 // set mom res profiles
1755 //
1756
1757 if(fMomResH1) delete fMomResH1;
1758 if(fMomResH2) delete fMomResH2;
1759 if(fMomResH3) delete fMomResH3;
1760
1761 fMomResH1 = new TProfile(*p1);//(TProfile*)p1->Clone("fMomResH1");
1762 fMomResH2 = new TProfile(*p2);//(TProfile*)p2->Clone("fMomResH2");
1763 fMomResH3 = new TProfile(*p3);//(TProfile*)p3->Clone("fMomResH3");
1764
1765}
1766
1767void AliAnalysisTaskJetCluster:: SetEfficiencyHybrid(TH1 *h1, TH1 *h2, TH1 *h3) {
1768 //
1769 // set tracking efficiency histos
1770 //
1771
1772 fhEffH1 = (TH1*)h1->Clone("fhEffH1");
1773 fhEffH2 = (TH1*)h2->Clone("fhEffH2");
1774 fhEffH3 = (TH1*)h3->Clone("fhEffH3");
1775}
1776
1777Double_t AliAnalysisTaskJetCluster::GetMomentumSmearing(Int_t cat, Double_t pt) {
1778
1779 //
1780 // Get smearing on generated momentum
1781 //
1782
1783 //printf("GetMomentumSmearing for cat %d and pt = %f \n",cat,pt);
1784
1785 TProfile *fMomRes = 0x0;
1786 if(cat==1) fMomRes = (TProfile*)fMomResH1->Clone("fMomRes");
1787 if(cat==2) fMomRes = (TProfile*)fMomResH2->Clone("fMomRes");
1788 if(cat==3) fMomRes = (TProfile*)fMomResH3->Clone("fMomRes");
1789
1790 if(!fMomRes) {
1791 return 0.;
1792 }
1793
1794
1795 Double_t smear = 0.;
1796
1797 if(pt>20.) {
1798 if(cat==1 && fMomResH1Fit) smear = fMomResH1Fit->Eval(pt);
1799 if(cat==2 && fMomResH2Fit) smear = fMomResH2Fit->Eval(pt);
1800 if(cat==3 && fMomResH3Fit) smear = fMomResH3Fit->Eval(pt);
1801 }
1802 else {
1803
1804 Int_t bin = fMomRes->FindBin(pt);
1805
1806 smear = fRandom->Gaus(fMomRes->GetBinContent(bin),fMomRes->GetBinError(bin));
1807
1808 }
1809
1810 if(fMomRes) delete fMomRes;
1811
1812 return smear;
1813}
1814
1815void AliAnalysisTaskJetCluster::FitMomentumResolution() {
1816 //
1817 // Fit linear function on momentum resolution at high pT
1818 //
1819
1820 if(!fMomResH1Fit && fMomResH1) {
1821 fMomResH1Fit = new TF1("fMomResH1Fit","[0]+[1]*x",0.,200.);
1822 fMomResH1->Fit(fMomResH1Fit,"LL V0","",5.,30.);
1823 fMomResH1Fit ->SetRange(5.,100.);
1824 }
1825
1826 if(!fMomResH2Fit && fMomResH2) {
1827 fMomResH2Fit = new TF1("fMomResH2Fit","[0]+[1]*x",0.,200.);
1828 fMomResH2->Fit(fMomResH2Fit,"LL V0","",5.,30.);
1829 fMomResH2Fit ->SetRange(5.,100.);
1830 }
1831
1832 if(!fMomResH3Fit && fMomResH3) {
1833 fMomResH3Fit = new TF1("fMomResH3Fit","[0]+[1]*x",0.,200.);
1834 fMomResH3->Fit(fMomResH3Fit,"LL V0","",5.,30.);
1835 fMomResH3Fit ->SetRange(5.,100.);
1836 }
1837
1838}
1839
1840/*
1841Int_t AliAnalysisTaskJetCluster::AddParticlesFastJet(TList &particles,vector<fastjet::PseudoJet> &inputParticles){
1842 for(int i = 0; i < particles.GetEntries(); i++){
1843 AliVParticle *vp = (AliVParticle*)particles.At(i);
1844 // Carefull energy is not well determined in real data, should not matter for p_T scheme?
1845 fastjet::PseudoJet jInp(vp->Px(),vp->Py(),vp->Pz(),vp->E());
1846 jInp.set_user_index(i);
1847 inputParticles.push_back(jInp);
1848 }
1849
1850 return 0;
1851
1852}
1853*/