]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWGJE/EMCALJetTasks/UserTasks/AliAnalysisTaskChargedJetsPA.cxx
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
mult bug fix and updated selection JO
[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / UserTasks / AliAnalysisTaskChargedJetsPA.cxx
CommitLineData
12eb4ac1 1#ifndef ALIANALYSISTASKSE_H
2#include <Riostream.h>
3#include <TROOT.h>
4#include <TFile.h>
5#include <TCint.h>
6#include <TChain.h>
7#include <TTree.h>
8#include <TKey.h>
9#include <TProfile.h>
b95d0245 10#include <TProfile2D.h>
12eb4ac1 11#include <TH1F.h>
12#include <TH2F.h>
13#include <TCanvas.h>
14#include <TList.h>
15#include <TClonesArray.h>
16#include <TObject.h>
17#include <TMath.h>
18#include <TSystem.h>
19#include <TInterpreter.h>
20#include <TH1.h>
21#include "AliAnalysisTask.h"
22#include "AliCentrality.h"
23#include "AliStack.h"
24#include "AliESDEvent.h"
25#include "AliESDInputHandler.h"
26#include "AliAODEvent.h"
27#include "AliAODHandler.h"
28#include "AliAnalysisManager.h"
29#include "AliAnalysisTaskSE.h"
30#endif
31
0cb49d99 32#include <time.h>
12eb4ac1 33#include <TRandom3.h>
34#include "AliGenPythiaEventHeader.h"
b95d0245 35#include "AliAODMCHeader.h"
12eb4ac1 36#include "AliMCEvent.h"
37#include "AliLog.h"
38#include <AliEmcalJet.h>
b95d0245 39#include <AliPicoTrack.h>
12eb4ac1 40#include "AliVEventHandler.h"
41#include "AliVParticle.h"
c2b0976c 42#include "AliAODMCParticle.h"
12eb4ac1 43#include "AliAnalysisUtils.h"
06b41974 44#include "AliRhoParameter.h"
12eb4ac1 45
8628b70c 46#include "AliAnalysisTaskChargedJetsPA.h"
dc3becdd 47using std::min;
c154ba90 48
3fa9cde0 49//TODO: Not accessing the particles when using MC
8628b70c 50//TODO: FillHistogram can be done better with virtual TH1(?)
51ClassImp(AliAnalysisTaskChargedJetsPA)
52
53// ######################################################################################## DEFINE HISTOGRAMS
54void AliAnalysisTaskChargedJetsPA::Init()
55{
56 #ifdef DEBUGMODE
efb9b161 57 AliInfo("Creating histograms.");
8628b70c 58 #endif
8628b70c 59
0cb49d99 60 TH1D* tmpHisto = AddHistogram1D<TH1D>("hNumberEvents", "Number of events (0 = before cuts, 1 = after cuts)", "", 2, 0, 2, "stage","N^{Events}/cut");
61 tmpHisto->GetXaxis()->SetBinLabel(1, "Before cuts");
62 tmpHisto->GetXaxis()->SetBinLabel(2, "After cuts");
63
64 tmpHisto = AddHistogram1D<TH1D>("hEventAcceptance", "Accepted events (0 = before cuts, 1 = after pile up, 2 = after vertex)", "", 3, 0, 3, "stage","N^{Events}/cut");
65 tmpHisto->GetXaxis()->SetBinLabel(1, "Before cuts");
66 tmpHisto->GetXaxis()->SetBinLabel(2, "After pile up");
67 tmpHisto->GetXaxis()->SetBinLabel(3, "After vertex");
68
69 tmpHisto = AddHistogram1D<TH1D>("hTrackAcceptance", "Accepted tracks (0 = before cuts, 1 = after eta, 2 = after pT)", "", 3, 0, 3, "stage","N^{Tracks}/cut");
70 tmpHisto->GetXaxis()->SetBinLabel(1, "Before cuts");
71 tmpHisto->GetXaxis()->SetBinLabel(2, "After eta");
72 tmpHisto->GetXaxis()->SetBinLabel(3, "After pT");
73
74 tmpHisto = AddHistogram1D<TH1D>("hJetAcceptance", "Accepted jets (0 = before cuts, 1 = after eta, 2 = after pT, 3 = after area)", "", 4, 0, 4, "stage","N^{Jets}/cut");
75 tmpHisto->GetXaxis()->SetBinLabel(1, "Before cuts");
76 tmpHisto->GetXaxis()->SetBinLabel(2, "After eta");
77 tmpHisto->GetXaxis()->SetBinLabel(3, "After pT");
78 tmpHisto->GetXaxis()->SetBinLabel(4, "After area");
8628b70c 79
80 // NOTE: Jet histograms
81 if (fAnalyzeJets)
82 {
83 // ######## Jet spectra
c2b0976c 84 AddHistogram1D<TH1D>("hRawJetPt", "Raw jets p_{T} distribution (before cuts)", "", 500, 0., 250., "p_{T} (GeV/c)", "dN^{Jets}/dp_{T}");
b95d0245 85 AddHistogram2D<TH2D>("hJetPt", "Jets p_{T} distribution", "", 500, -50., 200., fNumberOfCentralityBins, 0, 100, "p_{T} (GeV/c)","Centrality","dN^{Jets}/dp_{T}");
86 AddHistogram2D<TH2D>("hJetPtBgrdSubtractedKTImprovedCMS", "Jets p_{T} distribution, KT background (Improved CMS) subtracted", "", 500, -50., 200., fNumberOfCentralityBins, 0, 100, "p_{T} (GeV/c)","Centrality","dN^{Jets}/dp_{T}");
088fbc16 87
c2b0976c 88 AddHistogram2D<TProfile2D>("hJetPtSubtractedRhoKTImprovedCMS", "Mean subtracted KT (CMS w/o signal) background from jets", "COLZ", 600, 0, 150, fNumberOfCentralityBins, 0, 100, "Jet p_{T}", "Centrality", "#rho mean");
89 AddHistogram2D<TH2D>("hJetPtSubtractedRhoKTImprovedCMS020", "Mean subtracted KT (CMS w/o signal) background from jets, 0-20", "COLZ", 600, 0, 150, 400,0.,40., "Jet p_{T} (GeV/c)", "#rho (GeV/c)", "dN^{Events}/dp_{T}#rho");
90
b95d0245 91 if(fAnalyzeDeprecatedBackgrounds)
92 {
93 AddHistogram2D<TH2D>("hJetPtBgrdSubtractedTR", "Jets p_{T} distribution, TR background (Cone R=0.6 around jets excluded) subtracted", "", 500, -50., 200., fNumberOfCentralityBins, 0, 100, "p_{T} (GeV/c)","Centrality","dN^{Jets}/dp_{T}");
b95d0245 94 AddHistogram2D<TH2D>("hJetPtBgrdSubtractedKTPbPb", "Jets p_{T} distribution, KT background (PbPb w/o ghosts) subtracted", "", 500, -50., 200., fNumberOfCentralityBins, 0, 100, "p_{T} (GeV/c)","Centrality","dN^{Jets}/dp_{T}");
95 AddHistogram2D<TH2D>("hJetPtBgrdSubtractedKTPbPbWithGhosts", "Jets p_{T} distribution, KT background (PbPb w/ ghosts) subtracted", "", 500, -50., 200., fNumberOfCentralityBins, 0, 100, "p_{T} (GeV/c)","Centrality","dN^{Jets}/dp_{T}");
96 AddHistogram2D<TH2D>("hJetPtBgrdSubtractedKTCMS", "Jets p_{T} distribution, KT background (CMS) subtracted", "", 500, -50., 200., fNumberOfCentralityBins, 0, 100, "p_{T} (GeV/c)","Centrality","dN^{Jets}/dp_{T}");
97 AddHistogram2D<TH2D>("hJetPtBgrdSubtractedKTMean", "Jets p_{T} distribution, KT background (Mean) subtracted", "", 500, -50., 200., fNumberOfCentralityBins, 0, 100, "p_{T} (GeV/c)","Centrality","dN^{Jets}/dp_{T}");
98 AddHistogram2D<TH2D>("hJetPtBgrdSubtractedKTTrackLike", "Jets p_{T} distribution, KT background (track-like) subtracted", "", 500, -50., 200., fNumberOfCentralityBins, 0, 100, "p_{T} (GeV/c)","Centrality","dN^{Jets}/dp_{T}");
99 }
088fbc16 100
12eb4ac1 101 // ######## Jet stuff
392aa819 102 AddHistogram2D<TH2D>("hJetConstituentPt", "Jet constituents p_{T} distribution", "", 500, -50., 200., fNumberOfCentralityBins, 0, 100, "p_{T} (GeV/c)","Centrality","dN^{Tracks}/dp_{T}");
8628b70c 103 AddHistogram1D<TH1D>("hJetCountAll", "Number of Jets", "", 200, 0., 200., "N jets","dN^{Events}/dN^{Jets}");
104 AddHistogram1D<TH1D>("hJetCountAccepted", "Number of accepted Jets", "", 200, 0., 200., "N jets","dN^{Events}/dN^{Jets}");
0cb49d99 105 AddHistogram2D<TH2D>("hJetCount", "Correlation jets/accepted jets", "", 200, 0., 200., 200, 0., 200., "N jets","N jets accepted", "d^{2}N^{Events}/dN^{Jets dN^{Jets, acc}}");
106 AddHistogram1D<TH1D>("hLeadingJetPt", "Leading jet p_{T}", "", 500, -50., 200., "p_{T} (GeV/c)","dN^{Jets}/dp_{T}");
107 AddHistogram1D<TH1D>("hSecondLeadingJetPt", "Second leading jet p_{T}", "", 500, -50., 200., "p_{T} (GeV/c)","dN^{Jets}/dp_{T}");
108 AddHistogram1D<TH1D>("hCorrectedLeadingJetPt", "Corrected leading jet p_{T}", "", 500, -50., 200., "p_{T} (GeV/c)","dN^{Jets}/dp_{T}");
109 AddHistogram1D<TH1D>("hCorrectedSecondLeadingJetPt", "Corrected second leading jet p_{T}", "", 500, -50., 200., "p_{T} (GeV/c)","dN^{Jets}/dp_{T}");
12eb4ac1 110 AddHistogram1D<TH1D>("hJetDeltaPhi", "Jets combinatorial #Delta #phi", "", 250, 0., TMath::Pi(), "#Delta #phi","dN^{Jets}/d(#Delta #phi)");
111 AddHistogram1D<TH1D>("hLeadingJetDeltaPhi", "1st and 2nd leading jet #Delta #phi", "", 250, 0., TMath::Pi(), "#Delta #phi","dN^{Jets}/d(#Delta #phi)");
088fbc16 112
113 // ########## Dijet stuff
114 AddHistogram1D<TH1D>("hDijetLeadingJetPt", "Dijet leading jet p_{T} distribution", "", 500, 0., 100., "p_{T} (GeV/c)","dN^{Jets}/dp_{T}");
115 AddHistogram1D<TH1D>("hDijetConstituentsPt", "Dijet constituents p_{T} distribution", "", 500, 0., 100., "p_{T} (GeV/c)","dN^{Jets}/dp_{T}");
116 AddHistogram2D<TH2D>("hDijetPtCorrelation", "Dijet constituents p_{T} correlation", "COLZ", 500, 5., 100., 500, 5., 100., "1st leading jet p_{T} (GeV/c)","2nd leading jet p_{T} (GeV/c)","dN^{Dijets}/d^{2}p_{T}");
8628b70c 117 }
12eb4ac1 118
8628b70c 119 // NOTE: Jet background histograms
120 if (fAnalyzeBackground)
121 {
b95d0245 122 // ########## Default background estimates
123 AddHistogram2D<TH2D>("hKTBackgroundImprovedCMS", "KT background density (Improved CMS approach)", "LEGO2", 400, 0., 40., fNumberOfCentralityBins, 0, 100, "#rho (GeV/c)","Centrality", "dN^{Events}/d#rho");
06b41974 124 AddHistogram2D<TH2D>("hKTBackgroundImprovedCMSExternal", "KT background density (Improved CMS approach from external task)", "LEGO2", 400, 0., 40., fNumberOfCentralityBins, 0, 100, "#rho (GeV/c)","Centrality", "dN^{Events}/d#rho");
b95d0245 125 AddHistogram2D<TH2D>("hDeltaPtKTImprovedCMS", "Background fluctuations #delta p_{T} (KT, Improved CMS-like)", "", 1201, -40.0, 40.0, fNumberOfCentralityBins, 0, 100, "#delta p_{T} (GeV/c)","Centrality","dN^{Jets}/d#delta p_{T}");
392aa819 126 AddHistogram2D<TH2D>("hDeltaPtKTImprovedCMSPartialExclusion", "Background fluctuations #delta p_{T} (KT, Improved CMS-like, partial jet exclusion)", "", 1201, -40.0, 40.0, fNumberOfCentralityBins, 0, 100, "#delta p_{T} (GeV/c)","Centrality","dN^{Jets}/d#delta p_{T}");
c2b0976c 127 AddHistogram2D<TH2D>("hDeltaPtKTImprovedCMSPartialExclusion_Signal", "Background fluctuations #delta p_{T} (KT, Improved CMS-like, partial jet exclusion w/ 1/N_{sig} probability)", "", 1201, -40.0, 40.0, fNumberOfCentralityBins, 0, 100, "#delta p_{T} (GeV/c)","Centrality","dN^{Jets}/d#delta p_{T}");
128 AddHistogram2D<TH2D>("hDeltaPtKTImprovedCMSFullExclusion", "Background fluctuations #delta p_{T} (KT, Improved CMS-like, full leading jet exclusion)", "", 1201, -40.0, 40.0, fNumberOfCentralityBins, 0, 100, "#delta p_{T} (GeV/c)","Centrality","dN^{Jets}/d#delta p_{T}");
0cb49d99 129 AddHistogram2D<TH2D>("hDeltaPtNoBackground", "Background fluctuations #delta p_{T} (No background)", "", 1201, -40.0, 40.0, fNumberOfCentralityBins, 0, 100, "#delta p_{T} (GeV/c)","Centrality","dN^{Jets}/d#delta p_{T}");
130 AddHistogram2D<TH2D>("hDeltaPtNoBackgroundNoEmptyCones", "Background fluctuations #delta p_{T} (No background, no empty cones)", "", 1201, -40.0, 40.0, fNumberOfCentralityBins, 0, 100, "#delta p_{T} (GeV/c)","Centrality","dN^{Jets}/d#delta p_{T}");
b95d0245 131
b95d0245 132 AddHistogram1D<TProfile>("hKTMeanBackgroundImprovedCMS", "KT background mean (Improved CMS approach)", "", 100, 0, 100, "Centrality", "#rho mean");
133
134 AddHistogram2D<TH2D>("hDijetBackground", "Background density (dijets excluded)", "", 200, 0., 20., fNumberOfCentralityBins, 0, 100, "#rho (GeV/c)","Centrality", "dN^{Events}/d#rho");
135 AddHistogram2D<TH2D>("hDijetBackgroundPerpendicular", "Background density (dijets excluded)", "", 200, 0., 20., fNumberOfCentralityBins, 0, 100, "#rho (GeV/c)","Centrality", "dN^{Events}/d#rho");
136
137 if(fAnalyzeDeprecatedBackgrounds)
138 {
139 // ########## Different background estimates
b95d0245 140 AddHistogram2D<TH2D>("hKTBackgroundPbPb", "KT background density (PbPb approach, no ghosts)", "LEGO2", 400, 0., 40., fNumberOfCentralityBins, 0, 100, "#rho (GeV/c)","Centrality", "dN^{Events}/d#rho");
141 AddHistogram2D<TH2D>("hKTBackgroundPbPbWithGhosts", "KT background density (PbPb approach w/ ghosts)", "LEGO2", 400, 0., 40., fNumberOfCentralityBins, 0, 100, "#rho (GeV/c)","Centrality", "dN^{Events}/d#rho");
142 AddHistogram2D<TH2D>("hKTBackgroundCMS", "KT background density (CMS approach)", "LEGO2", 400, 0., 40., fNumberOfCentralityBins, 0, 100, "#rho (GeV/c)","Centrality", "dN^{Events}/d#rho");
143 AddHistogram2D<TH2D>("hKTBackgroundMean", "KT background density (Mean approach)", "LEGO2", 400, 0., 40., fNumberOfCentralityBins, 0, 100, "#rho (GeV/c)","Centrality", "dN^{Events}/d#rho");
144 AddHistogram2D<TH2D>("hKTBackgroundTrackLike", "KT background density (Track-like approach)", "LEGO2", 400, 0., 40., fNumberOfCentralityBins, 0, 100, "#rho (GeV/c)","Centrality", "dN^{Events}/d#rho");
145
146 AddHistogram2D<TH2D>("hTRBackgroundNoExcl", "TR background density (No signal excluded)", "LEGO2", 400, 0., 40., fNumberOfCentralityBins, 0, 100, "#rho (GeV/c)","Centrality", "dN^{Events}/d#rho");
147 AddHistogram2D<TH2D>("hTRBackgroundCone02", "TR background density (Cones R=0.2 around signal jets excluded)", "LEGO2", 400, 0., 40., fNumberOfCentralityBins, 0, 100, "#rho (GeV/c)","Centrality", "dN^{Events}/d#rho");
148 AddHistogram2D<TH2D>("hTRBackgroundCone04", "TR background density (Cones R=0.4 around signal jets excluded)", "LEGO2", 400, 0., 40., fNumberOfCentralityBins, 0, 100, "#rho (GeV/c)","Centrality", "dN^{Events}/d#rho");
149 AddHistogram2D<TH2D>("hTRBackgroundCone06", "TR background density (Cones R=0.6 around signal jets excluded)", "LEGO2", 400, 0., 40., fNumberOfCentralityBins, 0, 100, "#rho (GeV/c)","Centrality", "dN^{Events}/d#rho");
150 AddHistogram2D<TH2D>("hTRBackgroundCone08", "TR background density (Cones R=0.8 around signal jets excluded)", "LEGO2", 400, 0., 40., fNumberOfCentralityBins, 0, 100, "#rho (GeV/c)","Centrality", "dN^{Events}/d#rho");
151 AddHistogram2D<TH2D>("hTRBackgroundExact", "TR background density (signal jets exactly excluded)", "LEGO2", 400, 0., 40., fNumberOfCentralityBins, 0, 100, "#rho (GeV/c)","Centrality", "dN^{Events}/d#rho");
152
153 // ########## Delta Pt
154 AddHistogram2D<TH2D>("hDeltaPtKTPbPb", "Background fluctuations #delta p_{T} (KT, PbPb w/o ghosts)", "", 1201, -40.0, 40.0, fNumberOfCentralityBins, 0, 100, "#delta p_{T} (GeV/c)","Centrality","dN^{Jets}/d#delta p_{T}");
155 AddHistogram2D<TH2D>("hDeltaPtKTPbPbWithGhosts", "Background fluctuations #delta p_{T} (KT, PbPb w/ ghosts)", "", 1201, -40.0, 40.0, fNumberOfCentralityBins, 0, 100, "#delta p_{T} (GeV/c)","Centrality","dN^{Jets}/d#delta p_{T}");
156 AddHistogram2D<TH2D>("hDeltaPtKTCMS", "Background fluctuations #delta p_{T} (KT, CMS-like)", "", 1201, -40.0, 40.0, fNumberOfCentralityBins, 0, 100, "#delta p_{T} (GeV/c)","Centrality","dN^{Jets}/d#delta p_{T}");
157 AddHistogram2D<TH2D>("hDeltaPtKTMean", "Background fluctuations #delta p_{T} (KT, Mean)", "", 1201, -40.0, 40.0, fNumberOfCentralityBins, 0, 100, "#delta p_{T} (GeV/c)","Centrality","dN^{Jets}/d#delta p_{T}");
158 AddHistogram2D<TH2D>("hDeltaPtKTTrackLike", "Background fluctuations #delta p_{T} (KT, track-like)", "", 1201, -40.0, 40.0, fNumberOfCentralityBins, 0, 100, "#delta p_{T} (GeV/c)","Centrality","dN^{Jets}/d#delta p_{T}");
159 AddHistogram2D<TH2D>("hDeltaPtTR", "Background fluctuations #delta p_{T} (TR, cone R=0.6)", "", 1201, -40.0, 40.0, fNumberOfCentralityBins, 0, 100, "#delta p_{T} (GeV/c)","Centrality","dN^{Jets}/d#delta p_{T}");
b95d0245 160
161 // ########## Profiles for background means vs. centrality
392aa819 162 AddHistogram1D<TProfile>("hKTMeanBackgroundPbPb", "KT background mean (PbPb approach w/o ghosts)", "", fNumberOfCentralityBins, 0, 100, "Centrality", "#rho mean");
163 AddHistogram1D<TProfile>("hKTMeanBackgroundPbPbWithGhosts", "KT background mean (PbPb approach)", "", fNumberOfCentralityBins, 0, 100, "Centrality", "#rho mean");
164 AddHistogram1D<TProfile>("hKTMeanBackgroundCMS", "KT background mean (CMS approach)", "", fNumberOfCentralityBins, 0, 100, "Centrality", "#rho mean");
165 AddHistogram1D<TProfile>("hKTMeanBackgroundMean", "KT background mean (Mean approach)", "", fNumberOfCentralityBins, 0, 100, "Centrality", "#rho mean");
166 AddHistogram1D<TProfile>("hKTMeanBackgroundTPC", "KT background mean (Track-like approach)", "", fNumberOfCentralityBins, 0, 100, "Centrality", "#rho mean");
167 AddHistogram1D<TProfile>("hTRMeanBackground", "TR background mean", "", fNumberOfCentralityBins, 0, 100, "Centrality", "#rho mean");
b95d0245 168 }
12eb4ac1 169 }
8628b70c 170
b95d0245 171
172 // NOTE: Track & Cluster & QA histograms
173 if (fAnalyzeQA)
174 {
b95d0245 175 AddHistogram1D<TH1D>("hVertexX", "X distribution of the vertex", "", 2000, -1., 1., "#Delta x(cm)","dN^{Events}/dx");
176 AddHistogram1D<TH1D>("hVertexY", "Y distribution of the vertex", "", 2000, -1., 1., "#Delta y(cm)","dN^{Events}/dy");
177 AddHistogram2D<TH2D>("hVertexXY", "XY distribution of the vertex", "COLZ", 500, -1., 1., 500, -1., 1.,"#Delta x(cm)", "#Delta y(cm)","dN^{Events}/dxdy");
0cb49d99 178 AddHistogram1D<TH1D>("hVertexZ", "Z distribution of the vertex", "", 200, -20., 20., "#Delta z(cm)","dN^{Events}/dz");
b95d0245 179 AddHistogram1D<TH1D>("hVertexR", "R distribution of the vertex", "", 100, 0., 1., "#Delta r(cm)","dN^{Events}/dr");
392aa819 180 AddHistogram1D<TH1D>("hCentralityV0M", "Centrality distribution V0M", "", fNumberOfCentralityBins, 0., 100., "Centrality","dN^{Events}");
181 AddHistogram1D<TH1D>("hCentralityV0A", "Centrality distribution V0A", "", fNumberOfCentralityBins, 0., 100., "Centrality","dN^{Events}");
182 AddHistogram1D<TH1D>("hCentralityV0C", "Centrality distribution V0C", "", fNumberOfCentralityBins, 0., 100., "Centrality","dN^{Events}");
183
184 AddHistogram2D<TH2D>("hTrackCountAcc", "Number of tracks in acceptance vs. centrality", "LEGO2", 750, 0., 750., fNumberOfCentralityBins, 0, 100, "N tracks","Centrality", "dN^{Events}/dN^{Tracks}");
185 AddHistogram2D<TH2D>("hTrackPt", "Tracks p_{T} distribution", "", 1000, 0., 250., fNumberOfCentralityBins, 0, 100, "p_{T} (GeV/c)", "Centrality", "dN^{Tracks}/dp_{T}");
186 AddHistogram2D<TH2D>("hTrackPtNegEta", "Tracks p_{T} distribution (negative #eta)", "", 1000, 0., 250., fNumberOfCentralityBins, 0, 100, "p_{T} (GeV/c)","Centrality","dN^{Tracks}/dp_{T}");
187 AddHistogram2D<TH2D>("hTrackPtPosEta", "Tracks p_{T} distribution (positive #eta)", "", 1000, 0., 250., fNumberOfCentralityBins, 0, 100, "p_{T} (GeV/c)","Centrality","dN^{Tracks}/dp_{T}");
b95d0245 188 AddHistogram1D<TH1D>("hTrackCharge", "Charge", "", 11, -5, 5, "Charge (e)","dN^{Tracks}/dq");
189 AddHistogram1D<TH1D>("hTrackPhi", "Track #phi distribution", "", 360, 0, TMath::TwoPi(), "#phi","dN^{Tracks}/d#phi");
190 AddHistogram2D<TH2D>("hTrackPhiEta", "Track angular distribution", "LEGO2", 100, 0., 2*TMath::Pi(),100, -2.5, 2.5, "#phi","#eta","dN^{Tracks}/(d#phi d#eta)");
191
392aa819 192 AddHistogram2D<TH2D>("hTrackPhiPtCut", "Track #phi distribution for different pT cuts", "LEGO2", 360, 0, TMath::TwoPi(), 20, 0, 20, "#phi", "p_{T} lower cut", "dN^{Tracks}/d#phi dp_{T}");
0cb49d99 193 AddHistogram2D<TH2D>("hTrackPhiLabel", "Track #phi distribution for different labels", "LEGO2", 360, 0, TMath::TwoPi(), 3, 0, 3, "#phi", "Label", "dN^{Tracks}/d#phi");
194 AddHistogram2D<TH2D>("hTrackPhiTrackType", "Track #phi distribution for different track types", "LEGO2", 360, 0, TMath::TwoPi(), 3, 0, 3, "#phi", "Label", "dN^{Tracks}/d#phi");
b95d0245 195 AddHistogram1D<TH1D>("hTrackEta", "Track #eta distribution", "", 180, -fTrackEtaWindow, +fTrackEtaWindow, "#eta","dN^{Tracks}/d#eta");
196 if (fAnalyzeJets)
197 {
198 // ######## Jet QA
0cb49d99 199 AddHistogram1D<TH1D>("hRawJetArea", "Jets area distribution w/o area cut", "", 200, 0., 2., "Area","dN^{Jets}/dA");
b95d0245 200 AddHistogram1D<TH1D>("hJetArea", "Jets area distribution", "", 200, 0., 2., "Area","dN^{Jets}/dA");
0cb49d99 201 AddHistogram2D<TH2D>("hRawJetPhiEta", "Raw Jets angular distribution w/o #eta cut", "LEGO2", 360, 0., 2*TMath::Pi(),100, -1.0, 1.0, "#phi","#eta","dN^{Jets}/(d#phi d#eta)");
202 AddHistogram2D<TH2D>("hJetPhiEta", "Jets angular distribution", "LEGO2", 360, 0., 2*TMath::Pi(),100, -1.0, 1.0, "#phi","#eta","dN^{Jets}/(d#phi d#eta)");
b95d0245 203 AddHistogram2D<TH2D>("hJetPtVsConstituentCount", "Jets number of constituents vs. jet p_{T}", "COLZ", 400, 0., 200., 100, 0., 100., "p_{T}","N^{Tracks}","dN^{Jets}/(dp_{T} dN^{tracks})");
204 }
205 }
206
12eb4ac1 207 // NOTE: Pythia histograms
208 if (fAnalyzePythia)
209 {
210 AddHistogram1D<TH1D>("hPythiaPtHard", "Pythia p_{T} hard distribution", "", 2000, 0, 400, "p_{T} hard","dN^{Events}/dp_{T,hard}");
8628b70c 211 }
212
8628b70c 213 // register Histograms
214 for (Int_t i = 0; i < fHistCount; i++)
215 {
216 fOutputList->Add(fHistList->At(i));
217 }
218
219 PostData(1,fOutputList); // important for merging
220
221}
222
12eb4ac1 223//________________________________________________________________________
c2b0976c 224AliAnalysisTaskChargedJetsPA::AliAnalysisTaskChargedJetsPA(const char *name, const char* trackArrayName, const char* jetArrayName, const char* backgroundJetArrayName) : AliAnalysisTaskSE(name), fOutputList(0), fAnalyzeJets(1), fAnalyzeQA(1), fAnalyzeBackground(1), fAnalyzeDeprecatedBackgrounds(1), fAnalyzePythia(0), fHasTracks(0), fHasJets(0), fHasBackgroundJets(0), fIsKinematics(0), fUseVertexCut(1), fUsePileUpCut(1), fPartialAnalysisNParts(1), fPartialAnalysisIndex(0), fJetArray(0), fTrackArray(0), fBackgroundJetArray(0), fJetArrayName(0), fTrackArrayName(0), fBackgroundJetArrayName(0), fNumPtHardBins(11), fUsePtHardBin(-1), fRhoTaskName(), fNcoll(6.88348), fRandConeRadius(0.4), fSignalJetRadius(0.4), fBackgroundJetRadius(0.4), fTRBackgroundConeRadius(0.6), fNumberRandCones(8), fNumberExcludedJets(-1), fDijetMaxAngleDeviation(10.0), fPhysicalJetRadius(0.6), fSignalJetEtaWindow(0.5), fBackgroundJetEtaWindow(0.5), fTrackEtaWindow(0.9), fMinTrackPt(0.150), fMinJetPt(1.0), fMinJetArea(0.5), fMinBackgroundJetPt(0.0), fMinDijetLeadingPt(10.0), fNumberOfCentralityBins(100), fCentralityType("V0A"), fFirstLeadingJet(0), fSecondLeadingJet(0), fNumberSignalJets(0), fCrossSection(0.0), fTrials(0.0), fRandom(0), fHelperClass(0), fInitialized(0), fTaskInstanceCounter(0), fHistList(0), fHistCount(0), fIsDEBUG(0), fEventCounter(0)
8628b70c 225{
226 #ifdef DEBUGMODE
efb9b161 227 AliInfo("Calling constructor.");
8628b70c 228 #endif
229
8628b70c 230 // Every instance of this task gets his own number
231 static Int_t instance = 0;
232 fTaskInstanceCounter = instance;
233 instance++;
234
235 fTrackArrayName = new TString(trackArrayName);
4856f9c9 236 if (fTrackArrayName->Contains("MCParticles") || fTrackArrayName->Contains("mcparticles"))
06b41974 237 fIsKinematics = kTRUE;
8628b70c 238
239 fJetArrayName = new TString(jetArrayName);
240 if (strcmp(fJetArrayName->Data(),"") == 0)
241 fAnalyzeJets = kFALSE;
242 else
243 fAnalyzeJets = kTRUE;
244
245 fBackgroundJetArrayName = new TString(backgroundJetArrayName);
246 if (strcmp(fBackgroundJetArrayName->Data(),"") == 0)
247 fAnalyzeBackground = kFALSE;
248 else
249 fAnalyzeBackground = kTRUE;
250
251 DefineOutput(1, TList::Class());
252
253 fHistList = new TList();
254
b95d0245 255 for(Int_t i=0;i<1024;i++)
256 fSignalJets[i] = NULL;
257
392aa819 258
8628b70c 259 #ifdef DEBUGMODE
efb9b161 260 AliInfo("Constructor done.");
8628b70c 261 #endif
262
263}
264
265//________________________________________________________________________
266inline Double_t AliAnalysisTaskChargedJetsPA::GetConePt(Double_t eta, Double_t phi, Double_t radius)
267{
268 Double_t tmpConePt = 0.0;
269
270 for (Int_t i = 0; i < fTrackArray->GetEntries(); i++)
271 {
272 AliVTrack* tmpTrack = static_cast<AliVTrack*>(fTrackArray->At(i));
273 if (IsTrackInAcceptance(tmpTrack))
274 if(IsTrackInCone(tmpTrack, eta, phi, radius))
275 tmpConePt = tmpConePt + tmpTrack->Pt();
276 }
277 return tmpConePt;
278}
279
12eb4ac1 280
8628b70c 281//________________________________________________________________________
282inline Double_t AliAnalysisTaskChargedJetsPA::GetPtHard()
283{
392aa819 284 #ifdef DEBUGMODE
285 AliInfo("Starting GetPtHard.");
286 #endif
287 AliGenPythiaEventHeader* pythiaHeader = dynamic_cast<AliGenPythiaEventHeader*>(MCEvent()->GenEventHeader());
288 if (MCEvent())
289 if (!pythiaHeader)
290 {
291 // Check if AOD
292 AliAODMCHeader* aodMCH = dynamic_cast<AliAODMCHeader*>(InputEvent()->FindListObject(AliAODMCHeader::StdBranchName()));
8628b70c 293
392aa819 294 if (aodMCH)
295 {
296 for(UInt_t i = 0;i<aodMCH->GetNCocktailHeaders();i++)
297 {
298 pythiaHeader = dynamic_cast<AliGenPythiaEventHeader*>(aodMCH->GetCocktailHeader(i));
299 if (pythiaHeader) break;
300 }
301 }
302 }
303
304 #ifdef DEBUGMODE
305 AliInfo("Ending GetPtHard.");
306 #endif
307 if (pythiaHeader)
308 return pythiaHeader->GetPtHard();
309
310 AliWarning(Form("In task %s: GetPtHard() failed!", GetName()));
311 return -1.0;
312}
313
314
315//________________________________________________________________________
316inline Double_t AliAnalysisTaskChargedJetsPA::GetPythiaTrials()
317{
318 #ifdef DEBUGMODE
319 AliInfo("Starting GetPythiaTrials.");
320 #endif
321 AliGenPythiaEventHeader* pythiaHeader = dynamic_cast<AliGenPythiaEventHeader*>(MCEvent()->GenEventHeader());
322 if (MCEvent())
8628b70c 323 if (!pythiaHeader)
b95d0245 324 {
325 // Check if AOD
326 AliAODMCHeader* aodMCH = dynamic_cast<AliAODMCHeader*>(InputEvent()->FindListObject(AliAODMCHeader::StdBranchName()));
327
392aa819 328 if (aodMCH)
b95d0245 329 {
392aa819 330 for(UInt_t i = 0;i<aodMCH->GetNCocktailHeaders();i++)
331 {
332 pythiaHeader = dynamic_cast<AliGenPythiaEventHeader*>(aodMCH->GetCocktailHeader(i));
333 if (pythiaHeader) break;
334 }
b95d0245 335 }
b95d0245 336 }
392aa819 337
338 #ifdef DEBUGMODE
339 AliInfo("Ending GetPythiaTrials.");
340 #endif
341 if (pythiaHeader)
342 return pythiaHeader->Trials();
343
344 AliWarning(Form("In task %s: GetPythiaTrials() failed!", GetName()));
345 return -1.0;
8628b70c 346}
347
12eb4ac1 348
392aa819 349
8628b70c 350//________________________________________________________________________
351inline Int_t AliAnalysisTaskChargedJetsPA::GetPtHardBin()
352{
392aa819 353 #ifdef DEBUGMODE
354 AliInfo("Starting GetPtHardBin.");
355 #endif
8628b70c 356 // ########## PT HARD BIN EDGES
12eb4ac1 357 const Int_t kPtHardLowerEdges[] = { 0, 5,11,21,36,57, 84,117,152,191,234};
358 const Int_t kPtHardHigherEdges[] = { 5,11,21,36,57,84,117,152,191,234,1000000};
8628b70c 359
360 Int_t tmpPtHardBin = 0;
361 Double_t tmpPtHard = GetPtHard();
362
363 for (tmpPtHardBin = 0; tmpPtHardBin <= fNumPtHardBins; tmpPtHardBin++)
12eb4ac1 364 if (tmpPtHard >= kPtHardLowerEdges[tmpPtHardBin] && tmpPtHard < kPtHardHigherEdges[tmpPtHardBin])
8628b70c 365 break;
366
392aa819 367 #ifdef DEBUGMODE
368 AliInfo("Ending GetPtHardBin.");
369 #endif
8628b70c 370 return tmpPtHardBin;
371}
372
06b41974 373//________________________________________________________________________
374Double_t AliAnalysisTaskChargedJetsPA::GetExternalRho()
375{
376 // Get rho from event.
377 AliRhoParameter *rho = 0;
378 if (!fRhoTaskName.IsNull()) {
379 rho = dynamic_cast<AliRhoParameter*>(InputEvent()->FindListObject(fRhoTaskName.Data()));
380 if (!rho) {
381 AliWarning(Form("%s: Could not retrieve rho with name %s!", GetName(), fRhoTaskName.Data()));
382 return 0;
383 }
384 }
385 else
386 return 0;
387
388 return (rho->GetVal());
389}
390
8628b70c 391
392//________________________________________________________________________
393inline Bool_t AliAnalysisTaskChargedJetsPA::IsTrackInCone(AliVTrack* track, Double_t eta, Double_t phi, Double_t radius)
394{
395 // This is to use a full cone in phi even at the edges of phi (2pi -> 0) (0 -> 2pi)
396 Double_t trackPhi = 0.0;
397 if (track->Phi() > (TMath::TwoPi() - (radius-phi)))
398 trackPhi = track->Phi() - TMath::TwoPi();
399 else if (track->Phi() < (phi+radius - TMath::TwoPi()))
400 trackPhi = track->Phi() + TMath::TwoPi();
401 else
402 trackPhi = track->Phi();
403
404 if ( TMath::Abs(trackPhi-phi)*TMath::Abs(trackPhi-phi) + TMath::Abs(track->Eta()-eta)*TMath::Abs(track->Eta()-eta) <= radius*radius)
405 return kTRUE;
406
407 return kFALSE;
408}
409
b95d0245 410//________________________________________________________________________
411inline Bool_t AliAnalysisTaskChargedJetsPA::IsTrackInJet(AliEmcalJet* jet, Int_t trackIndex)
412{
413 for (Int_t i = 0; i < jet->GetNumberOfTracks(); ++i)
414 {
415 Int_t jetTrack = jet->TrackAt(i);
416 if (jetTrack == trackIndex)
417 return kTRUE;
418 }
419 return kFALSE;
420}
421
422//________________________________________________________________________
423inline Bool_t AliAnalysisTaskChargedJetsPA::IsJetOverlapping(AliEmcalJet* jet1, AliEmcalJet* jet2)
424{
425 for (Int_t i = 0; i < jet1->GetNumberOfTracks(); ++i)
426 {
427 Int_t jet1Track = jet1->TrackAt(i);
428 for (Int_t j = 0; j < jet2->GetNumberOfTracks(); ++j)
429 {
430 Int_t jet2Track = jet2->TrackAt(j);
431 if (jet1Track == jet2Track)
432 return kTRUE;
433 }
434 }
435 return kFALSE;
436}
437
0cb49d99 438//________________________________________________________________________
439inline Bool_t AliAnalysisTaskChargedJetsPA::IsEventInAcceptance(AliVEvent* event)
440{
441 if (!event)
442 return kFALSE;
443
444 FillHistogram("hEventAcceptance", 0.5); // number of events before manual cuts
c2b0976c 445 if(fUsePileUpCut)
446 if(fHelperClass->IsPileUpEvent(event))
447 return kFALSE;
0cb49d99 448
c2b0976c 449 FillHistogram("hEventAcceptance", 1.5); // number of events after pileup cuts
0cb49d99 450
c2b0976c 451 if(fAnalyzeQA)
452 FillHistogram("hVertexZ",event->GetPrimaryVertex()->GetZ());
0cb49d99 453
c2b0976c 454 if(fUseVertexCut)
455 if(!fHelperClass->IsVertexSelected2013pA(event))
456 return kFALSE;
457 FillHistogram("hEventAcceptance", 2.5); // number of events after vertex cut
0cb49d99 458
459 return kTRUE;
460}
461
462//________________________________________________________________________
463inline Bool_t AliAnalysisTaskChargedJetsPA::IsTrackInAcceptance(AliVParticle* track)
464{
465 FillHistogram("hTrackAcceptance", 0.5);
466 if (track != 0)
c2b0976c 467 {
468 if(fIsKinematics)
469 {
470 // TODO: Only working for AOD MC
471 if((!track->Charge()) || (!(static_cast<AliAODMCParticle*>(track))->IsPhysicalPrimary()) )
472 return kFALSE;
473 }
0cb49d99 474 if (TMath::Abs(track->Eta()) <= fTrackEtaWindow)
475 {
476 FillHistogram("hTrackAcceptance", 1.5);
477 if (track->Pt() >= fMinTrackPt)
478 {
479 FillHistogram("hTrackAcceptance", 2.5);
480 return kTRUE;
481 }
482 }
c2b0976c 483 }
0cb49d99 484 return kFALSE;
485}
b95d0245 486
8628b70c 487//________________________________________________________________________
488inline Bool_t AliAnalysisTaskChargedJetsPA::IsBackgroundJetInAcceptance(AliEmcalJet *jet)
489{
490 if (jet != 0)
491 if (TMath::Abs(jet->Eta()) <= fBackgroundJetEtaWindow)
492 if (jet->Pt() >= fMinBackgroundJetPt)
493 return kTRUE;
494
495 return kFALSE;
496}
497
498//________________________________________________________________________
499inline Bool_t AliAnalysisTaskChargedJetsPA::IsSignalJetInAcceptance(AliEmcalJet *jet)
500{
0cb49d99 501 FillHistogram("hJetAcceptance", 0.5);
8628b70c 502 if (jet != 0)
503 if (TMath::Abs(jet->Eta()) <= fSignalJetEtaWindow)
0cb49d99 504 {
505 FillHistogram("hJetAcceptance", 1.5);
8628b70c 506 if (jet->Pt() >= fMinJetPt)
0cb49d99 507 {
508 FillHistogram("hJetAcceptance", 2.5);
8628b70c 509 if (jet->Area() >= fMinJetArea)
0cb49d99 510 {
511 FillHistogram("hJetAcceptance", 3.5);
8628b70c 512 return kTRUE;
0cb49d99 513 }
514 }
515 }
8628b70c 516 return kFALSE;
517}
518
519//________________________________________________________________________
520inline Bool_t AliAnalysisTaskChargedJetsPA::IsDijet(AliEmcalJet *jet1, AliEmcalJet *jet2)
521{
522 // Output from GetDeltaPhi is < pi in any case
523 if ((jet1 != 0) && (jet2 != 0))
524 if((TMath::Pi() - GetDeltaPhi(jet1->Phi(),jet2->Phi())) < fDijetMaxAngleDeviation)
525 if ((jet1->Pt() > fMinDijetLeadingPt) && (jet2->Pt() > fMinDijetLeadingPt)) //TODO: Introduce recoil jet?
526 return kTRUE;
527
528 return kFALSE;
529}
530
531//________________________________________________________________________
532void AliAnalysisTaskChargedJetsPA::ExecOnce()
533{
534 #ifdef DEBUGMODE
efb9b161 535 AliInfo("Starting ExecOnce.");
8628b70c 536 #endif
537 fInitialized = kTRUE;
538
539 // Check for track array
540 if (strcmp(fTrackArrayName->Data(), "") != 0)
541 {
542 fTrackArray = dynamic_cast<TClonesArray*>(InputEvent()->FindListObject(fTrackArrayName->Data()));
543 fHasTracks = kTRUE;
544 if (!fTrackArray)
545 {
0983dd5b 546 AliWarning(Form("%s: Could not retrieve tracks %s! This is OK, if tracks are not demanded.", GetName(), fTrackArrayName->Data()));
8628b70c 547 fHasTracks = kFALSE;
548 }
549 else
550 {
551 TClass *cl = fTrackArray->GetClass();
552 if (!cl->GetBaseClass("AliVParticle"))
553 {
554 AliError(Form("%s: Collection %s does not contain AliVParticle objects!", GetName(), fTrackArrayName->Data()));
555 fTrackArray = 0;
556 fHasTracks = kFALSE;
557 }
558 }
559 }
8628b70c 560
561 // Check for jet array
562 if (strcmp(fJetArrayName->Data(), "") != 0)
563 {
564 fJetArray = dynamic_cast<TClonesArray*>(InputEvent()->FindListObject(fJetArrayName->Data()));
565 fHasJets = kTRUE;
566
567 if (!fJetArray)
568 {
0983dd5b 569 AliWarning(Form("%s: Could not retrieve jets %s! This is OK, if jets are not demanded.", GetName(), fJetArrayName->Data()));
8628b70c 570 fHasJets = kFALSE;
571 }
572 else
573 {
574 if (!fJetArray->GetClass()->GetBaseClass("AliEmcalJet"))
575 {
576 AliError(Form("%s: Collection %s does not contain AliEmcalJet objects!", GetName(), fJetArrayName->Data()));
577 fJetArray = 0;
578 fHasJets = kFALSE;
579 }
580 }
581 }
582
583 // Check for background object
584 if (strcmp(fBackgroundJetArrayName->Data(), "") != 0)
585 {
586 fBackgroundJetArray = dynamic_cast<TClonesArray*>(InputEvent()->FindListObject(fBackgroundJetArrayName->Data()));
587 fHasBackgroundJets = kTRUE;
588 if (!fBackgroundJetArray)
589 {
590 AliInfo(Form("%s: Could not retrieve background jets %s! This is OK, if background is not demanded.", GetName(), fBackgroundJetArrayName->Data()));
591 fHasBackgroundJets = kFALSE;
592 }
593 }
594
595 // Look, if initialization is OK
12eb4ac1 596 if (!fHasTracks && fAnalyzeBackground)
8628b70c 597 {
12eb4ac1 598 AliError(Form("%s: Tracks NOT successfully casted although demanded! Deactivating background analysis",GetName()));
8628b70c 599 fAnalyzeBackground = kFALSE;
600 }
601 if ((!fHasJets && fAnalyzeJets) || (!fHasJets && fAnalyzeBackground))
602 {
603 AliError(Form("%s: Jets NOT successfully casted although demanded! Deactivating jet- and background analysis",GetName()));
604 fAnalyzeJets = kFALSE;
605 fAnalyzeBackground = kFALSE;
606 }
607 if (!fHasBackgroundJets && fAnalyzeBackground)
608 {
609 AliError(Form("%s: Background NOT successfully casted although demanded! Deactivating background analysis",GetName()));
610 fAnalyzeBackground = kFALSE;
611 }
612
06b41974 613 // Initialize helper class (for vertex selection & pile up correction)
3fa9cde0 614 fHelperClass = new AliAnalysisUtils();
06b41974 615 fHelperClass->SetCutOnZVertexSPD(kFALSE);
8628b70c 616 // Histogram init
617 Init();
618
619 #ifdef DEBUGMODE
efb9b161 620 AliInfo("ExecOnce done.");
8628b70c 621 #endif
622
623}
624
625//________________________________________________________________________
626void AliAnalysisTaskChargedJetsPA::GetSignalJets()
627{
628 // Reset vars
629 fFirstLeadingJet = NULL;
630 fSecondLeadingJet = NULL;
631 fNumberSignalJets = 0;
632
b95d0245 633 TList tmpJets;
634 for (Int_t i = 0; i < fJetArray->GetEntries(); i++)
8628b70c 635 {
636 AliEmcalJet* jet = static_cast<AliEmcalJet*>(fJetArray->At(i));
637 if (!jet)
638 {
639 AliError(Form("%s: Could not receive jet %d", GetName(), i));
640 continue;
641 }
b95d0245 642 if (!IsSignalJetInAcceptance(jet))
643 continue;
8628b70c 644
b95d0245 645 for (Int_t j = 0; j <= tmpJets.GetEntries(); j++)
8628b70c 646 {
b95d0245 647 if (j>tmpJets.GetEntries()-1) // When passed last item add the jet at the end
648 {
649 tmpJets.Add(jet);
650 break;
651 }
652
653 AliEmcalJet* listJet = static_cast<AliEmcalJet*>(tmpJets.At(j));
654
655 if(jet->Pt() < listJet->Pt()) // Insert jet before that one in list if pt smaller
656 {
657 tmpJets.AddAt(jet, j);
658 break;
659 }
8628b70c 660 }
b95d0245 661 }
662
663 for (Int_t i = 0; i < tmpJets.GetEntries(); i++)
664 {
665 AliEmcalJet* jet = static_cast<AliEmcalJet*>(tmpJets.At(i));
8628b70c 666 fSignalJets[fNumberSignalJets] = jet;
667 fNumberSignalJets++;
668 }
669
670 if (fNumberSignalJets > 0)
b95d0245 671 fFirstLeadingJet = static_cast<AliEmcalJet*>(tmpJets.At(0));
8628b70c 672 if (fNumberSignalJets > 1)
b95d0245 673 fSecondLeadingJet = static_cast<AliEmcalJet*>(tmpJets.At(1));
8628b70c 674
675}
676
677//________________________________________________________________________
678Int_t AliAnalysisTaskChargedJetsPA::GetLeadingJets(TClonesArray* jetArray, Int_t* jetIDArray, Bool_t isSignalJets)
679{
680// Writes first two leading jets into already registered array jetIDArray
681
682 if (!jetArray)
683 {
684 AliError("Could not get the jet array to get leading jets from it!");
685 return 0;
686 }
687
688 Float_t maxJetPts[] = {0, 0};
689 jetIDArray[0] = -1;
690 jetIDArray[1] = -1;
691
692 Int_t jetCount = jetArray->GetEntries();
693 Int_t jetCountAccepted = 0;
694
695 for (Int_t i = 0; i < jetCount; i++)
696 {
697 AliEmcalJet* jet = static_cast<AliEmcalJet*>(jetArray->At(i));
698 if (!jet)
699 {
700 AliError(Form("%s: Could not receive jet %d", GetName(), i));
701 continue;
702 }
703
704 if(isSignalJets)
705 {
706 if (!IsSignalJetInAcceptance(jet)) continue;
707 }
708 else
709 {
710 if (!IsBackgroundJetInAcceptance(jet)) continue;
711 }
712
713 if (jet->Pt() > maxJetPts[0])
714 {
715 maxJetPts[1] = maxJetPts[0];
716 jetIDArray[1] = jetIDArray[0];
717 maxJetPts[0] = jet->Pt();
718 jetIDArray[0] = i;
719 }
720 else if (jet->Pt() > maxJetPts[1])
721 {
722 maxJetPts[1] = jet->Pt();
723 jetIDArray[1] = i;
724 }
725 jetCountAccepted++;
726 }
727 return jetCountAccepted;
728}
12eb4ac1 729
12eb4ac1 730
731//________________________________________________________________________
e8df4140 732Double_t AliAnalysisTaskChargedJetsPA::GetCorrectedJetPt(AliEmcalJet* jet, Double_t background)
8628b70c 733{
734 #ifdef DEBUGMODE
efb9b161 735 AliInfo("Getting corrected jet spectra.");
8628b70c 736 #endif
737
738 if(!jet)
739 {
392aa819 740 AliError("Jet pointer passed to GetCorrectedJetPt() not valid!");
8628b70c 741 return -1.0;
742 }
743
744 Double_t correctedPt = -1.0;
745
0983dd5b 746 // if the passed background is not valid, do not subtract it
747 if(background < 0)
748 background = 0;
8628b70c 749
8628b70c 750 // Subtract background
e8df4140 751 correctedPt = jet->Pt() - background * jet->Area();
8628b70c 752
753 #ifdef DEBUGMODE
efb9b161 754 AliInfo("Got corrected jet spectra.");
8628b70c 755 #endif
756
757 return correctedPt;
758}
759
760
12eb4ac1 761
8628b70c 762//________________________________________________________________________
c2b0976c 763Double_t AliAnalysisTaskChargedJetsPA::GetDeltaPt(Double_t rho, Double_t leadingJetExclusionProbability)
8628b70c 764{
765 #ifdef DEBUGMODE
efb9b161 766 AliInfo("Getting Delta Pt.");
8628b70c 767 #endif
768
12eb4ac1 769 // Define an invalid delta pt
0cb49d99 770 Double_t deltaPt = -10000.0;
8628b70c 771
12eb4ac1 772 // Define eta range
8628b70c 773 Double_t etaMin, etaMax;
12eb4ac1 774 etaMin = -(fTrackEtaWindow-fRandConeRadius);
775 etaMax = +(fTrackEtaWindow-fRandConeRadius);
8628b70c 776
12eb4ac1 777 // Define random cone
778 Bool_t coneValid = kTRUE;
0983dd5b 779 Double_t tmpRandConeEta = etaMin + fRandom->Rndm()*(etaMax-etaMin);
780 Double_t tmpRandConePhi = fRandom->Rndm()*TMath::TwoPi();
8628b70c 781
12eb4ac1 782 // if there is a jet, check for overlap if demanded
c2b0976c 783 if(leadingJetExclusionProbability)
8628b70c 784 {
392aa819 785 AliEmcalJet* tmpLeading = dynamic_cast<AliEmcalJet*>(fJetArray->At(0));
786 // Get leading jet (regardless of pT)
787 for (Int_t i = 1; i<fJetArray->GetEntries(); i++)
8628b70c 788 {
392aa819 789 AliEmcalJet* tmpJet = static_cast<AliEmcalJet*>(fJetArray->At(i));
790 // if jet is in acceptance and higher, take as new leading
791 if (tmpJet)
792 if ((TMath::Abs(tmpJet->Eta()) <= fSignalJetEtaWindow) && (tmpJet->Area() >= fMinJetArea))
793 if((!tmpLeading) || (tmpJet->Pt() > tmpLeading->Pt()))
794 tmpLeading = tmpJet;
795 }
0cb49d99 796 if(tmpLeading)
392aa819 797 {
0cb49d99 798 Double_t excludedJetPhi = tmpLeading->Phi();
799 Double_t excludedJetEta = tmpLeading->Eta();
800 Double_t tmpDeltaPhi = GetDeltaPhi(tmpRandConePhi, excludedJetPhi);
e8df4140 801
0cb49d99 802 // Check, if cone has overlap with jet
803 if ( tmpDeltaPhi*tmpDeltaPhi + TMath::Abs(tmpRandConeEta-excludedJetEta)*TMath::Abs(tmpRandConeEta-excludedJetEta) <= fRandConeRadius*fRandConeRadius)
804 {
805 // Define probability to exclude the RC
c2b0976c 806 Double_t probability = leadingJetExclusionProbability;
0cb49d99 807
808 // Only exclude cone with a given probability
809 if (fRandom->Rndm()<=probability)
810 coneValid = kFALSE;
811 }
8628b70c 812 }
813 }
814
392aa819 815
8628b70c 816 // Get the cones' pt and calculate delta pt
817 if (coneValid)
818 deltaPt = GetConePt(tmpRandConeEta,tmpRandConePhi,fRandConeRadius) - (rho*fRandConeRadius*fRandConeRadius*TMath::Pi());
12eb4ac1 819
0cb49d99 820 return deltaPt;
8628b70c 821 #ifdef DEBUGMODE
efb9b161 822 AliInfo("Got Delta Pt.");
8628b70c 823 #endif
824}
825
826//________________________________________________________________________
0cb49d99 827void AliAnalysisTaskChargedJetsPA::GetKTBackgroundDensityAll(Int_t numberExcludeLeadingJets, Double_t& rhoPbPb, Double_t& rhoPbPbWithGhosts, Double_t& rhoCMS, Double_t& rhoImprovedCMS, Double_t& rhoMean, Double_t& rhoTrackLike)
8628b70c 828{
829 #ifdef DEBUGMODE
0cb49d99 830 AliInfo("Getting ALL KT background density.");
8628b70c 831 #endif
832
b95d0245 833 static Double_t tmpRhoPbPb[1024];
834 static Double_t tmpRhoPbPbWithGhosts[1024];
835 static Double_t tmpRhoMean[1024];
836 static Double_t tmpRhoCMS[1024];
837 static Double_t tmpRhoImprovedCMS[1024];
838 Double_t tmpCoveredArea = 0.0;
839 Double_t tmpSummedArea = 0.0;
840 Double_t tmpPtTrackLike = 0.0;
841 Double_t tmpAreaTrackLike = 0.0;
8628b70c 842
843 // Setting invalid values
b95d0245 844 rhoPbPb = 0.0;
845 rhoPbPbWithGhosts = 0.0;
846 rhoCMS = 0.0;
847 rhoImprovedCMS = 0.0;
848 rhoMean = 0.0;
849 rhoTrackLike = 0.0;
8628b70c 850
b95d0245 851 Int_t rhoPbPbJetCount = 0;
852 Int_t rhoPbPbWithGhostsJetCount = 0;
853 Int_t rhoCMSJetCount = 0;
854 Int_t rhoImprovedCMSJetCount = 0;
855 Int_t rhoMeanJetCount = 0;
e8df4140 856
e8df4140 857
b95d0245 858 // Find 2 leading KT jets for the original PbPb approach
859 Int_t leadingKTJets[] = {-1, -1};
860 GetLeadingJets(fBackgroundJetArray, &leadingKTJets[0], kFALSE);
861
862 // Exclude UP TO numberExcludeLeadingJets
863 if(numberExcludeLeadingJets==-1)
864 numberExcludeLeadingJets = fNumberSignalJets;
865 if (fNumberSignalJets < numberExcludeLeadingJets)
866 numberExcludeLeadingJets = fNumberSignalJets;
8628b70c 867
e8df4140 868 for (Int_t i = 0; i < fBackgroundJetArray->GetEntries(); i++)
8628b70c 869 {
e8df4140 870 AliEmcalJet* backgroundJet = static_cast<AliEmcalJet*>(fBackgroundJetArray->At(i));
b95d0245 871
e8df4140 872 if (!backgroundJet)
8628b70c 873 {
874 AliError(Form("%s: Could not receive jet %d", GetName(), i));
875 continue;
876 }
877
b95d0245 878 // Search for overlap with signal jets
879 Bool_t isOverlapping = kFALSE;
880 for(Int_t j=0;j<numberExcludeLeadingJets;j++)
e8df4140 881 {
b95d0245 882 AliEmcalJet* signalJet = fSignalJets[j];
883
884 if(IsJetOverlapping(signalJet, backgroundJet))
e8df4140 885 {
b95d0245 886 isOverlapping = kTRUE;
887 break;
888 }
889 }
e8df4140 890
b95d0245 891 tmpSummedArea += backgroundJet->Area();
892 if(backgroundJet->Pt() > 0.150)
893 tmpCoveredArea += backgroundJet->Area();
8628b70c 894
b95d0245 895 if (!IsBackgroundJetInAcceptance(backgroundJet))
896 continue;
8628b70c 897
c2b0976c 898 Double_t tmpRho = 0.0;
899 if(backgroundJet->Area())
900 tmpRho = backgroundJet->Pt() / backgroundJet->Area();
901
b95d0245 902 // PbPb approach (take ghosts into account)
903 if ((i != leadingKTJets[0]) && (i != leadingKTJets[1]))
904 {
0cb49d99 905 tmpRhoPbPbWithGhosts[rhoPbPbWithGhostsJetCount] = tmpRho;
b95d0245 906 rhoPbPbWithGhostsJetCount++;
e8df4140 907 }
b95d0245 908
909 if(backgroundJet->Pt() > 0.150)
e8df4140 910 {
b95d0245 911 // CMS approach: don't take ghosts into acount
0cb49d99 912 tmpRhoCMS[rhoCMSJetCount] = tmpRho;
b95d0245 913 rhoCMSJetCount++;
e8df4140 914
b95d0245 915 // Improved CMS approach: like CMS but excluding signal
916 if(!isOverlapping)
917 {
0cb49d99 918 tmpRhoImprovedCMS[rhoImprovedCMSJetCount] = tmpRho;
b95d0245 919 rhoImprovedCMSJetCount++;
920 }
8628b70c 921
b95d0245 922 // PbPb w/o ghosts approach (just neglect ghosts)
923 if ((i != leadingKTJets[0]) && (i != leadingKTJets[1]))
924 {
0cb49d99 925 tmpRhoPbPb[rhoPbPbJetCount] = tmpRho;
b95d0245 926 rhoPbPbJetCount++;
927 }
928 }
929
930 // (no overlap with signal jets)
931 if(!isOverlapping)
932 {
933 // Mean approach
0cb49d99 934 tmpRhoMean[rhoMeanJetCount] = tmpRho;
b95d0245 935 rhoMeanJetCount++;
936
937 // Track like approach approach
938 tmpPtTrackLike += backgroundJet->Pt();
939 tmpAreaTrackLike += backgroundJet->Area();
940 }
8628b70c 941
8628b70c 942 }
b95d0245 943
944 if (tmpAreaTrackLike > 0)
945 rhoTrackLike = tmpPtTrackLike/tmpAreaTrackLike;
946 if (rhoPbPbJetCount > 0)
947 rhoPbPb = TMath::Median(rhoPbPbJetCount, tmpRhoPbPb);
948 if (rhoPbPbWithGhostsJetCount > 0)
949 rhoPbPbWithGhosts = TMath::Median(rhoPbPbWithGhostsJetCount, tmpRhoPbPbWithGhosts);
950 if (rhoCMSJetCount > 0)
951 rhoCMS = TMath::Median(rhoCMSJetCount, tmpRhoCMS) * tmpCoveredArea/tmpSummedArea;
952 if (rhoImprovedCMSJetCount > 0)
c154ba90 953 {
b95d0245 954 rhoImprovedCMS = TMath::Median(rhoImprovedCMSJetCount, tmpRhoImprovedCMS) * tmpCoveredArea/tmpSummedArea;
c154ba90 955 }
b95d0245 956 if (rhoMeanJetCount > 0)
957 rhoMean = TMath::Mean(rhoMeanJetCount, tmpRhoMean);
958
0cb49d99 959 #ifdef DEBUGMODE
960 AliInfo("Got ALL KT background density.");
961 #endif
962}
963
964//________________________________________________________________________
965void AliAnalysisTaskChargedJetsPA::GetKTBackgroundDensity(Int_t numberExcludeLeadingJets, Double_t& rhoImprovedCMS)
966{
967 #ifdef DEBUGMODE
968 AliInfo("Getting KT background density.");
969 #endif
970
971 static Double_t tmpRhoImprovedCMS[1024];
972 Double_t tmpCoveredArea = 0.0;
973 Double_t tmpSummedArea = 0.0;
974
975 // Setting invalid values
976 rhoImprovedCMS = 0.0;
977
978 Int_t rhoImprovedCMSJetCount = 0;
979
980 // Exclude UP TO numberExcludeLeadingJets
981 if(numberExcludeLeadingJets==-1)
982 numberExcludeLeadingJets = fNumberSignalJets;
983 if (fNumberSignalJets < numberExcludeLeadingJets)
984 numberExcludeLeadingJets = fNumberSignalJets;
985
986 for (Int_t i = 0; i < fBackgroundJetArray->GetEntries(); i++)
987 {
988 AliEmcalJet* backgroundJet = static_cast<AliEmcalJet*>(fBackgroundJetArray->At(i));
989
990 if (!backgroundJet)
991 {
992 AliError(Form("%s: Could not receive jet %d", GetName(), i));
993 continue;
994 }
995
996 // Search for overlap with signal jets
997 Bool_t isOverlapping = kFALSE;
998 for(Int_t j=0;j<numberExcludeLeadingJets;j++)
999 {
1000 AliEmcalJet* signalJet = fSignalJets[j];
1001
1002 if(IsJetOverlapping(signalJet, backgroundJet))
1003 {
1004 isOverlapping = kTRUE;
1005 break;
1006 }
1007 }
1008
1009 tmpSummedArea += backgroundJet->Area();
1010 if(backgroundJet->Pt() > 0.150)
1011 tmpCoveredArea += backgroundJet->Area();
1012
1013 if (!IsBackgroundJetInAcceptance(backgroundJet))
1014 continue;
1015
1016 Double_t tmpRho = backgroundJet->Pt() / backgroundJet->Area();
1017
1018 if(backgroundJet->Pt() > 0.150)
1019 if(!isOverlapping)
1020 {
1021 tmpRhoImprovedCMS[rhoImprovedCMSJetCount] = tmpRho;
1022 rhoImprovedCMSJetCount++;
1023 }
1024 }
1025
1026 if (rhoImprovedCMSJetCount > 0)
1027 {
c154ba90 1028 rhoImprovedCMS = TMath::Median(rhoImprovedCMSJetCount, tmpRhoImprovedCMS) * tmpCoveredArea/tmpSummedArea;
0cb49d99 1029 }
8628b70c 1030 #ifdef DEBUGMODE
efb9b161 1031 AliInfo("Got KT background density.");
8628b70c 1032 #endif
1033}
1034
8628b70c 1035
8628b70c 1036//________________________________________________________________________
b95d0245 1037void AliAnalysisTaskChargedJetsPA::GetTRBackgroundDensity(Int_t numberExcludeLeadingJets, Double_t& rhoNoExclusion, Double_t& rhoConeExclusion02, Double_t& rhoConeExclusion04, Double_t& rhoConeExclusion06, Double_t& rhoConeExclusion08, Double_t& rhoExactExclusion)
8628b70c 1038{
1039 #ifdef DEBUGMODE
12eb4ac1 1040 AliInfo("Getting TR background density.");
8628b70c 1041 #endif
1042
b95d0245 1043 Double_t summedTracksPtCone04 = 0.0;
4856f9c9 1044 Double_t summedTracksPtCone02 = 0.0;
b95d0245 1045 Double_t summedTracksPtCone06 = 0.0;
1046 Double_t summedTracksPtCone08 = 0.0;
1047 Double_t summedTracksPtWithinJets = 0.0;
8628b70c 1048 Double_t summedTracksPt = 0.0;
b95d0245 1049
8628b70c 1050 // Setting invalid values
b95d0245 1051 rhoNoExclusion = 0.0;
1052 rhoConeExclusion02 = 0.0;
1053 rhoConeExclusion04 = 0.0;
1054 rhoConeExclusion06 = 0.0;
1055 rhoConeExclusion08 = 0.0;
1056 rhoExactExclusion = 0.0;
1057
8628b70c 1058 // Exclude UP TO numberExcludeLeadingJets
b95d0245 1059 if(numberExcludeLeadingJets==-1)
1060 numberExcludeLeadingJets = fNumberSignalJets;
1061 if (fNumberSignalJets < numberExcludeLeadingJets)
8628b70c 1062 numberExcludeLeadingJets = fNumberSignalJets;
1063
b95d0245 1064 for (Int_t i = 0; i < fTrackArray->GetEntries(); i++)
8628b70c 1065 {
8628b70c 1066 AliVTrack* tmpTrack = static_cast<AliVTrack*>(fTrackArray->At(i));
b95d0245 1067 Bool_t trackWithinJet = kFALSE; Bool_t trackWithin02Cone = kFALSE; Bool_t trackWithin04Cone = kFALSE; Bool_t trackWithin06Cone = kFALSE; Bool_t trackWithin08Cone = kFALSE;
1068
8628b70c 1069 if (IsTrackInAcceptance(tmpTrack))
b95d0245 1070 {
1071 // Check if tracks overlaps with jet
1072 for(Int_t j=0;j<numberExcludeLeadingJets;j++)
8628b70c 1073 {
b95d0245 1074 AliEmcalJet* signalJet = fSignalJets[j];
1075
1076 // Exact jet exclusion
1077 if (IsTrackInJet(signalJet, i))
1078 trackWithinJet = kTRUE;
1079
1080 // Cone exclusions
1081 if (IsTrackInCone(tmpTrack, signalJet->Eta(), signalJet->Phi(), 0.2))
8628b70c 1082 {
b95d0245 1083 trackWithin02Cone = kTRUE;
1084 trackWithin04Cone = kTRUE;
1085 trackWithin06Cone = kTRUE;
1086 trackWithin08Cone = kTRUE;
1087 break;
1088 }
1089 else if (IsTrackInCone(tmpTrack, signalJet->Eta(), signalJet->Phi(), 0.4))
1090 {
1091 trackWithin04Cone = kTRUE;
1092 trackWithin06Cone = kTRUE;
1093 trackWithin08Cone = kTRUE;
8628b70c 1094 }
b95d0245 1095 else if (IsTrackInCone(tmpTrack, signalJet->Eta(), signalJet->Phi(), 0.6))
8628b70c 1096 {
b95d0245 1097 trackWithin06Cone = kTRUE;
1098 trackWithin08Cone = kTRUE;
1099 }
1100 else if (IsTrackInCone(tmpTrack, signalJet->Eta(), signalJet->Phi(), 0.8))
1101 {
1102 trackWithin08Cone = kTRUE;
8628b70c 1103 }
1104 }
b95d0245 1105
1106 if(!trackWithin08Cone)
1107 {
1108 summedTracksPtCone08 += tmpTrack->Pt();
1109 }
1110 if(!trackWithin06Cone)
1111 {
1112 summedTracksPtCone06 += tmpTrack->Pt();
1113 }
1114 if(!trackWithin04Cone)
1115 {
1116 summedTracksPtCone04 += tmpTrack->Pt();
1117 }
1118 if(!trackWithin02Cone)
1119 {
1120 summedTracksPtCone02 += tmpTrack->Pt();
1121 }
1122 if(!trackWithinJet)
1123 {
1124 summedTracksPtWithinJets += tmpTrack->Pt();
1125 }
1126 summedTracksPt += tmpTrack->Pt();
1127
1128 }
8628b70c 1129 }
1130
b95d0245 1131 // Calculate the correct area where the tracks were taking from
8628b70c 1132
b95d0245 1133 Double_t tmpFullTPCArea = (2.0*fTrackEtaWindow) * TMath::TwoPi();
1134 Double_t tmpAreaCone02 = tmpFullTPCArea;
1135 Double_t tmpAreaCone04 = tmpFullTPCArea;
1136 Double_t tmpAreaCone06 = tmpFullTPCArea;
1137 Double_t tmpAreaCone08 = tmpFullTPCArea;
1138 Double_t tmpAreaWithinJets = tmpFullTPCArea;
1139 std::vector<Double_t> tmpEtas(numberExcludeLeadingJets);
1140 std::vector<Double_t> tmpPhis(numberExcludeLeadingJets);
1141
1142 for(Int_t i=0;i<numberExcludeLeadingJets;i++)
1143 {
1144 AliEmcalJet* tmpJet = fSignalJets[i];
1145 tmpEtas[i] = tmpJet->Eta();
1146 tmpPhis[i] = tmpJet->Phi();
1147 tmpAreaWithinJets -= tmpJet->Area();
8628b70c 1148 }
1149
b95d0245 1150 tmpAreaCone02 -= tmpFullTPCArea * MCGetOverlapMultipleCirclesRectancle(numberExcludeLeadingJets, tmpEtas, tmpPhis, 0.2, -fTrackEtaWindow, +fTrackEtaWindow, 0., TMath::TwoPi());
1151 tmpAreaCone04 -= tmpFullTPCArea * MCGetOverlapMultipleCirclesRectancle(numberExcludeLeadingJets, tmpEtas, tmpPhis, 0.4, -fTrackEtaWindow, +fTrackEtaWindow, 0., TMath::TwoPi());
1152 tmpAreaCone06 -= tmpFullTPCArea * MCGetOverlapMultipleCirclesRectancle(numberExcludeLeadingJets, tmpEtas, tmpPhis, 0.6, -fTrackEtaWindow, +fTrackEtaWindow, 0., TMath::TwoPi());
1153 tmpAreaCone08 -= tmpFullTPCArea * MCGetOverlapMultipleCirclesRectancle(numberExcludeLeadingJets, tmpEtas, tmpPhis, 0.8, -fTrackEtaWindow, +fTrackEtaWindow, 0., TMath::TwoPi());
1154
1155 rhoConeExclusion02 = summedTracksPtCone02/tmpAreaCone02;
1156 rhoConeExclusion04 = summedTracksPtCone04/tmpAreaCone04;
1157 rhoConeExclusion06 = summedTracksPtCone06/tmpAreaCone06;
1158 rhoConeExclusion08 = summedTracksPtCone08/tmpAreaCone08;
1159 rhoExactExclusion = summedTracksPtWithinJets/tmpAreaWithinJets;
1160 rhoNoExclusion = summedTracksPt/tmpFullTPCArea;
1161
1162
8628b70c 1163 #ifdef DEBUGMODE
12eb4ac1 1164 AliInfo("Got TR background density.");
8628b70c 1165 #endif
1166}
1167
1168//________________________________________________________________________
12eb4ac1 1169void AliAnalysisTaskChargedJetsPA::GetTRBackgroundDensity(Int_t numberExcludeLeadingJets, Double_t& rhoMean, Double_t& area, AliEmcalJet* excludeJet1, AliEmcalJet* excludeJet2, Bool_t doSearchPerpendicular)
8628b70c 1170{
1171 #ifdef DEBUGMODE
12eb4ac1 1172 AliInfo("Getting TR background density.");
8628b70c 1173 #endif
1174
1175 // Setting invalid values
1176 Double_t summedTracksPt = 0.0;
b95d0245 1177 rhoMean = 0.0;
8628b70c 1178 area = -1.0;
1179
1180 Double_t tmpRadius = 0.0;
1181 if (doSearchPerpendicular)
b95d0245 1182 tmpRadius = 0.4*TMath::Pi(); // exclude 90 degrees around jets
8628b70c 1183 else
b95d0245 1184 tmpRadius = 0.8;
8628b70c 1185
1186 numberExcludeLeadingJets = 2; // dijet is excluded here in any case
1187
1188
1189
1190 if (!fTrackArray || !fJetArray)
1191 {
1192 AliError("Could not get the track/jet array to calculate track rho!");
1193 return;
1194 }
1195
1196 Int_t trackCount = fTrackArray->GetEntries();
1197 Int_t trackCountAccepted = 0;
1198 for (Int_t i = 0; i < trackCount; i++)
1199 {
1200 AliVTrack* tmpTrack = static_cast<AliVTrack*>(fTrackArray->At(i));
1201 if (IsTrackInAcceptance(tmpTrack))
1202 {
1203 if (IsTrackInCone(tmpTrack, excludeJet1->Eta(), excludeJet1->Phi(), tmpRadius))
1204 continue;
1205
1206 if (numberExcludeLeadingJets > 1)
1207 if (IsTrackInCone(tmpTrack, excludeJet2->Eta(), excludeJet2->Phi(), tmpRadius))
1208 continue;
1209
1210 // Add track pt to array
1211 summedTracksPt = summedTracksPt + tmpTrack->Pt();
1212 trackCountAccepted++;
1213 }
1214 }
1215
1216 if (trackCountAccepted > 0)
1217 {
1218 Double_t tmpArea = 2.0*fTrackEtaWindow*TMath::TwoPi() - 2*(tmpRadius*tmpRadius * TMath::Pi()); //TPC area - excluding jet area
1219 rhoMean = summedTracksPt/tmpArea;
1220 area = tmpArea;
1221 }
1222
1223 #ifdef DEBUGMODE
12eb4ac1 1224 AliInfo("Got TR background density.");
8628b70c 1225 #endif
1226}
1227
1228//________________________________________________________________________
1229void AliAnalysisTaskChargedJetsPA::Calculate(AliVEvent* event)
1230{
1231 #ifdef DEBUGMODE
efb9b161 1232 AliInfo("Starting Calculate().");
8628b70c 1233 #endif
1234 ////////////////////// NOTE: initialization & casting
1235
c2b0976c 1236 fEventCounter++;
1237
b95d0245 1238 // Check, if analysis should be done in pt hard bins
1239 if(fUsePtHardBin != -1)
1240 if(GetPtHardBin() != fUsePtHardBin)
1241 return;
1242
c2b0976c 1243 // This is to take only every Nth event
1244 if((fEventCounter+fPartialAnalysisIndex) % fPartialAnalysisNParts != 0)
1245 return;
1246
0cb49d99 1247 FillHistogram("hNumberEvents",0.5);
b95d0245 1248
0cb49d99 1249 if(!IsEventInAcceptance(event))
1250 return;
8628b70c 1251
0cb49d99 1252 FillHistogram("hNumberEvents",1.5);
8628b70c 1253
1254 #ifdef DEBUGMODE
efb9b161 1255 AliInfo("Calculate()::Init done.");
8628b70c 1256 #endif
1257
1258 ////////////////////// NOTE: Get Centrality, (Leading)Signal jets and Background
1259
12eb4ac1 1260 // Get centrality
8628b70c 1261 AliCentrality* tmpCentrality = NULL;
1262 tmpCentrality = event->GetCentrality();
4856f9c9 1263 Double_t centralityPercentile = -1.0;
b95d0245 1264 Double_t centralityPercentileV0A = 0.0;
1265 Double_t centralityPercentileV0C = 0.0;
1266 Double_t centralityPercentileV0M = 0.0;
8628b70c 1267 if (tmpCentrality != NULL)
b95d0245 1268 {
12eb4ac1 1269 centralityPercentile = tmpCentrality->GetCentralityPercentile(fCentralityType.Data());
b95d0245 1270 centralityPercentileV0A = tmpCentrality->GetCentralityPercentile("V0A");
1271 centralityPercentileV0C = tmpCentrality->GetCentralityPercentile("V0C");
1272 centralityPercentileV0M = tmpCentrality->GetCentralityPercentile("V0M");
1273 }
8628b70c 1274
b95d0245 1275 if((centralityPercentile < 0.0) || (centralityPercentile > 100.0))
1276 {
1277 AliWarning(Form("Centrality value not valid (c=%E), setting to failsafe c=1.0.",centralityPercentile));
1278 centralityPercentile = 1.0;
1279 }
8628b70c 1280 // Get jets
1281 if (fAnalyzeBackground || fAnalyzeJets)
1282 GetSignalJets();
1283
12eb4ac1 1284 // Get background estimates
b95d0245 1285 Double_t backgroundKTImprovedCMS = -1.0;
06b41974 1286 Double_t backgroundKTImprovedCMSExternal = -1.0;
b95d0245 1287 Double_t backgroundDijet = -1.0;
1288 Double_t backgroundDijetPerpendicular = -1.0;
1289
1290 Double_t backgroundKTPbPb = -1.0;
1291 Double_t backgroundKTPbPbWithGhosts = -1.0;
1292 Double_t backgroundKTCMS = -1.0;
1293 Double_t backgroundKTMean = -1.0;
1294 Double_t backgroundKTTrackLike = -1.0;
1295 Double_t backgroundTRNoExcl = -1.0;
1296 Double_t backgroundTRCone02 = -1.0;
1297 Double_t backgroundTRCone04 = -1.0;
1298 Double_t backgroundTRCone06 = -1.0;
1299 Double_t backgroundTRCone08 = -1.0;
1300 Double_t backgroundTRExact = -1.0;
b95d0245 1301
088fbc16 1302 // Calculate background for different jet exclusions
12eb4ac1 1303
8628b70c 1304 if (fAnalyzeBackground)
1305 {
b95d0245 1306
0cb49d99 1307 if(fAnalyzeDeprecatedBackgrounds)
1308 GetKTBackgroundDensityAll (fNumberExcludedJets, backgroundKTPbPb, backgroundKTPbPbWithGhosts, backgroundKTCMS, backgroundKTImprovedCMS, backgroundKTMean, backgroundKTTrackLike);
1309 else
1310 GetKTBackgroundDensity (fNumberExcludedJets, backgroundKTImprovedCMS);
1311
1312 if(fAnalyzeDeprecatedBackgrounds)
1313 GetTRBackgroundDensity (fNumberExcludedJets, backgroundTRNoExcl, backgroundTRCone02, backgroundTRCone04, backgroundTRCone06, backgroundTRCone08, backgroundTRExact);
06b41974 1314
1315 backgroundKTImprovedCMSExternal = GetExternalRho();
8628b70c 1316 }
1317
1318 #ifdef DEBUGMODE
efb9b161 1319 AliInfo("Calculate()::Centrality&SignalJets&Background-Calculation done.");
8628b70c 1320 #endif
8628b70c 1321
b95d0245 1322 if (fAnalyzeQA)
1323 {
1324 FillHistogram("hVertexX",event->GetPrimaryVertex()->GetX());
1325 FillHistogram("hVertexY",event->GetPrimaryVertex()->GetY());
1326 FillHistogram("hVertexXY",event->GetPrimaryVertex()->GetX(), event->GetPrimaryVertex()->GetY());
b95d0245 1327 FillHistogram("hVertexR",TMath::Sqrt(event->GetPrimaryVertex()->GetX()*event->GetPrimaryVertex()->GetX() + event->GetPrimaryVertex()->GetY()*event->GetPrimaryVertex()->GetY()));
1328 FillHistogram("hCentralityV0M",centralityPercentileV0M);
1329 FillHistogram("hCentralityV0A",centralityPercentileV0A);
1330 FillHistogram("hCentralityV0C",centralityPercentileV0C);
1331
1332 Int_t trackCountAcc = 0;
1333 Int_t nTracks = fTrackArray->GetEntries();
1334 for (Int_t i = 0; i < nTracks; i++)
1335 {
1336 AliVTrack* track = static_cast<AliVTrack*>(fTrackArray->At(i));
0cb49d99 1337
1338 if (track != 0)
1339 if (track->Pt() >= fMinTrackPt)
1340 FillHistogram("hTrackPhiEta", track->Phi(),track->Eta(), 1);
1341
b95d0245 1342 if (IsTrackInAcceptance(track))
1343 {
392aa819 1344 FillHistogram("hTrackPt", track->Pt(), centralityPercentile);
b95d0245 1345 if(track->Eta() >= 0)
392aa819 1346 FillHistogram("hTrackPtPosEta", track->Pt(), centralityPercentile);
b95d0245 1347 else
392aa819 1348 FillHistogram("hTrackPtNegEta", track->Pt(), centralityPercentile);
b95d0245 1349
1350 FillHistogram("hTrackEta", track->Eta());
1351 FillHistogram("hTrackPhi", track->Phi());
c2b0976c 1352
4856f9c9 1353 if(static_cast<AliPicoTrack*>(track))
0cb49d99 1354 {
1355 FillHistogram("hTrackPhiTrackType", track->Phi(), (static_cast<AliPicoTrack*>(track))->GetTrackType());
1356 FillHistogram("hTrackPhiLabel", track->Phi(), (static_cast<AliPicoTrack*>(track))->GetLabel());
1357 }
b95d0245 1358 for(Int_t j=0;j<20;j++)
1359 if(track->Pt() > j)
1360 FillHistogram("hTrackPhiPtCut", track->Phi(), track->Pt());
1361
1362 FillHistogram("hTrackCharge", track->Charge());
1363 trackCountAcc++;
1364 }
1365 }
4856f9c9 1366 FillHistogram("hTrackCountAcc", trackCountAcc, centralityPercentile);
b95d0245 1367
1368 }
1369 #ifdef DEBUGMODE
1370 AliInfo("Calculate()::QA done.");
1371 #endif
1372
12eb4ac1 1373 ////////////////////// NOTE: Jet analysis and calculations
8628b70c 1374
12eb4ac1 1375 if (fAnalyzeJets)
8628b70c 1376 {
0cb49d99 1377 for (Int_t i = 0; i<fJetArray->GetEntries(); i++)
8628b70c 1378 {
0cb49d99 1379 AliEmcalJet* tmpJet = static_cast<AliEmcalJet*>(fJetArray->At(i));
1380 if (!tmpJet)
1381 continue;
1382
c2b0976c 1383 FillHistogram("hRawJetPt", tmpJet->Pt());
0cb49d99 1384 if (tmpJet->Pt() >= fMinJetPt)
e8df4140 1385 {
0cb49d99 1386 // ### RAW JET ANALYSIS
1387 if (tmpJet->Area() >= fMinJetArea)
1388 FillHistogram("hRawJetPhiEta", tmpJet->Phi(), tmpJet->Eta());
1389 if (TMath::Abs(tmpJet->Eta()) <= fSignalJetEtaWindow)
1390 FillHistogram("hRawJetArea", tmpJet->Area());
e8df4140 1391 }
c2b0976c 1392
0cb49d99 1393 if(IsSignalJetInAcceptance(tmpJet))
1394 {
1395 // ### SIGNAL JET ANALYSIS
1396 // Jet spectra
1397 FillHistogram("hJetPt", tmpJet->Pt(), centralityPercentile);
1398 FillHistogram("hJetPtBgrdSubtractedKTImprovedCMS", GetCorrectedJetPt(tmpJet, backgroundKTImprovedCMS), centralityPercentile);
1399 FillHistogram("hJetPtSubtractedRhoKTImprovedCMS", tmpJet->Pt(), centralityPercentile, backgroundKTImprovedCMS);
1400 if(centralityPercentile<=20.0)
1401 FillHistogram("hJetPtSubtractedRhoKTImprovedCMS020", tmpJet->Pt(), backgroundKTImprovedCMS);
1402
1403 if(fAnalyzeDeprecatedBackgrounds)
1404 {
1405 FillHistogram("hJetPtBgrdSubtractedTR", GetCorrectedJetPt(tmpJet, backgroundTRCone06), centralityPercentile);
1406 FillHistogram("hJetPtBgrdSubtractedKTPbPb", GetCorrectedJetPt(tmpJet, backgroundKTPbPb), centralityPercentile);
1407 FillHistogram("hJetPtBgrdSubtractedKTPbPbWithGhosts", GetCorrectedJetPt(tmpJet, backgroundKTPbPbWithGhosts), centralityPercentile);
1408 FillHistogram("hJetPtBgrdSubtractedKTCMS", GetCorrectedJetPt(tmpJet, backgroundKTCMS), centralityPercentile);
1409 FillHistogram("hJetPtBgrdSubtractedKTMean", GetCorrectedJetPt(tmpJet, backgroundKTMean), centralityPercentile);
1410 FillHistogram("hJetPtBgrdSubtractedKTTrackLike", GetCorrectedJetPt(tmpJet, backgroundKTTrackLike), centralityPercentile);
1411 }
e8df4140 1412
0cb49d99 1413 for(Int_t j=0; j<tmpJet->GetNumberOfTracks(); j++)
1414 FillHistogram("hJetConstituentPt", tmpJet->TrackAt(j, fTrackArray)->Pt(), centralityPercentile);
392aa819 1415
0cb49d99 1416 if(fAnalyzeQA)
1417 {
1418 FillHistogram("hJetArea", tmpJet->Area());
1419 FillHistogram("hJetPtVsConstituentCount", tmpJet->Pt(),tmpJet->GetNumberOfTracks());
1420 FillHistogram("hJetPhiEta", tmpJet->Phi(),tmpJet->Eta());
1421 }
1422 // Signal jet vs. signal jet - "Combinatorial"
1423 for (Int_t j = i+1; j<fNumberSignalJets; j++)
1424 FillHistogram("hJetDeltaPhi", GetDeltaPhi(tmpJet->Phi(), fSignalJets[j]->Phi()));
088fbc16 1425 }
12eb4ac1 1426 }
8628b70c 1427
12eb4ac1 1428 // ### DIJETS
8628b70c 1429 if(fNumberSignalJets >= 2)
1430 {
1431 FillHistogram("hLeadingJetDeltaPhi", GetDeltaPhi(fFirstLeadingJet->Phi(), fSecondLeadingJet->Phi()));
8628b70c 1432
12eb4ac1 1433 if (IsDijet(fFirstLeadingJet, fSecondLeadingJet))
8628b70c 1434 {
12eb4ac1 1435 FillHistogram("hDijetConstituentsPt", fFirstLeadingJet->Pt());
1436 FillHistogram("hDijetConstituentsPt", fSecondLeadingJet->Pt());
1437
8628b70c 1438 FillHistogram("hDijetLeadingJetPt", fFirstLeadingJet->Pt());
1439 FillHistogram("hDijetPtCorrelation", fFirstLeadingJet->Pt(), fSecondLeadingJet->Pt());
1440 Double_t dummyArea = 0;
b95d0245 1441 GetTRBackgroundDensity (2, backgroundDijet, dummyArea, fFirstLeadingJet, fSecondLeadingJet, kFALSE);
1442 GetTRBackgroundDensity (2, backgroundDijetPerpendicular, dummyArea, fFirstLeadingJet, fSecondLeadingJet, kTRUE);
8628b70c 1443 }
1444 }
1445
12eb4ac1 1446 // ### SOME JET PLOTS
1447 FillHistogram("hJetCountAll", fJetArray->GetEntries());
1448 FillHistogram("hJetCountAccepted", fNumberSignalJets);
0cb49d99 1449 FillHistogram("hJetCount", fJetArray->GetEntries(), fNumberSignalJets);
12eb4ac1 1450 if (fFirstLeadingJet)
0cb49d99 1451 {
12eb4ac1 1452 FillHistogram("hLeadingJetPt", fFirstLeadingJet->Pt());
0cb49d99 1453 FillHistogram("hCorrectedLeadingJetPt", GetCorrectedJetPt(fFirstLeadingJet,backgroundKTImprovedCMS));
1454 }
12eb4ac1 1455 if (fSecondLeadingJet)
0cb49d99 1456 {
12eb4ac1 1457 FillHistogram("hSecondLeadingJetPt", fSecondLeadingJet->Pt());
0cb49d99 1458 FillHistogram("hCorrectedSecondLeadingJetPt", GetCorrectedJetPt(fSecondLeadingJet,backgroundKTImprovedCMS));
1459 }
8628b70c 1460
8628b70c 1461 } //endif AnalyzeJets
1462
1463 #ifdef DEBUGMODE
efb9b161 1464 AliInfo("Calculate()::Jets done.");
8628b70c 1465 #endif
1466 ////////////////////// NOTE: Background analysis
1467
1468 if (fAnalyzeBackground)
1469 {
088fbc16 1470 // Calculate background in centrality classes
b95d0245 1471 FillHistogram("hKTBackgroundImprovedCMS", backgroundKTImprovedCMS, centralityPercentile);
06b41974 1472
1473 FillHistogram("hKTBackgroundImprovedCMSExternal", backgroundKTImprovedCMSExternal, centralityPercentile);
1474
b95d0245 1475 FillHistogram("hKTMeanBackgroundImprovedCMS", centralityPercentile, backgroundKTImprovedCMS);
8628b70c 1476
12eb4ac1 1477 // In case of dijets -> look at the background
b95d0245 1478 if (backgroundDijet >= 0)
1479 FillHistogram("hDijetBackground", backgroundDijet, centralityPercentile);
1480 if (backgroundDijetPerpendicular >= 0)
1481 FillHistogram("hDijetBackgroundPerpendicular", backgroundDijetPerpendicular, centralityPercentile);
1482
1483 if(fAnalyzeDeprecatedBackgrounds)
1484 {
1485 FillHistogram("hKTBackgroundPbPb", backgroundKTPbPb, centralityPercentile);
1486 FillHistogram("hKTBackgroundPbPbWithGhosts", backgroundKTPbPbWithGhosts, centralityPercentile);
1487 FillHistogram("hKTBackgroundCMS", backgroundKTCMS, centralityPercentile);
1488 FillHistogram("hKTBackgroundMean", backgroundKTMean, centralityPercentile);
1489 FillHistogram("hKTBackgroundTrackLike", backgroundKTTrackLike, centralityPercentile);
1490
1491 FillHistogram("hTRBackgroundNoExcl", backgroundTRNoExcl, centralityPercentile);
1492 FillHistogram("hTRBackgroundCone02", backgroundTRCone02, centralityPercentile);
1493 FillHistogram("hTRBackgroundCone04", backgroundTRCone04, centralityPercentile);
1494 FillHistogram("hTRBackgroundCone06", backgroundTRCone06, centralityPercentile);
1495 FillHistogram("hTRBackgroundCone08", backgroundTRCone08, centralityPercentile);
1496 FillHistogram("hTRBackgroundExact", backgroundTRExact, centralityPercentile);
1497
b95d0245 1498 // Calculate background profiles in terms of centrality
1499 FillHistogram("hKTMeanBackgroundPbPb", centralityPercentile, backgroundKTPbPb);
1500 FillHistogram("hKTMeanBackgroundPbPbWithGhosts", centralityPercentile, backgroundKTPbPbWithGhosts);
1501 FillHistogram("hKTMeanBackgroundCMS", centralityPercentile, backgroundKTCMS);
1502 FillHistogram("hKTMeanBackgroundMean", centralityPercentile, backgroundKTMean);
1503 FillHistogram("hKTMeanBackgroundTPC", centralityPercentile, backgroundKTTrackLike);
1504 FillHistogram("hTRMeanBackground", centralityPercentile, backgroundTRCone06);
1505 }
12eb4ac1 1506
1507
1508 // Calculate the delta pt
0cb49d99 1509 Double_t tmpDeltaPtNoBackground = GetDeltaPt(0.0);
1510 Double_t tmpDeltaPtKTImprovedCMS = GetDeltaPt(backgroundKTImprovedCMS);
c2b0976c 1511
1512 Double_t tmpDeltaPtKTImprovedCMSPartialExclusion = 0.0;
1513 if(fNcoll)
1514 tmpDeltaPtKTImprovedCMSPartialExclusion = GetDeltaPt(backgroundKTImprovedCMS, 1.0/fNcoll);
1515 else
1516 tmpDeltaPtKTImprovedCMSPartialExclusion = GetDeltaPt(backgroundKTImprovedCMS, 1.0);
1517
1518 Double_t tmpDeltaPtKTImprovedCMSPartialExclusion_Signal = 0.0;
1519 if(fNumberSignalJets)
1520 tmpDeltaPtKTImprovedCMSPartialExclusion_Signal = GetDeltaPt(backgroundKTImprovedCMS, 1.0/fNumberSignalJets);
1521 else
1522 tmpDeltaPtKTImprovedCMSPartialExclusion_Signal = GetDeltaPt(backgroundKTImprovedCMS, 1.0);
1523
1524 Double_t tmpDeltaPtKTImprovedCMSFullExclusion = GetDeltaPt(backgroundKTImprovedCMS, 1.0);
1525
0cb49d99 1526 Double_t tmpDeltaPtKTPbPb = 0;
1527 Double_t tmpDeltaPtKTPbPbWithGhosts = 0;
1528 Double_t tmpDeltaPtKTCMS = 0;
1529 Double_t tmpDeltaPtKTMean = 0;
1530 Double_t tmpDeltaPtKTTrackLike = 0;
1531 Double_t tmpDeltaPtTR = 0;
088fbc16 1532
0cb49d99 1533 if(fAnalyzeDeprecatedBackgrounds)
1534 {
1535 tmpDeltaPtKTPbPb = GetDeltaPt(backgroundKTPbPb);
1536 tmpDeltaPtKTPbPbWithGhosts = GetDeltaPt(backgroundKTPbPbWithGhosts);
1537 tmpDeltaPtKTCMS = GetDeltaPt(backgroundKTCMS);
1538 tmpDeltaPtKTMean = GetDeltaPt(backgroundKTMean);
1539 tmpDeltaPtKTTrackLike = GetDeltaPt(backgroundKTTrackLike);
1540 tmpDeltaPtTR = GetDeltaPt(backgroundTRCone06);
1541 }
12eb4ac1 1542
1543 // If valid, fill the delta pt histograms
b95d0245 1544
1545 if(tmpDeltaPtKTImprovedCMS > -10000.0)
1546 FillHistogram("hDeltaPtKTImprovedCMS", tmpDeltaPtKTImprovedCMS, centralityPercentile);
392aa819 1547 if(tmpDeltaPtKTImprovedCMSPartialExclusion > -10000.0)
1548 FillHistogram("hDeltaPtKTImprovedCMSPartialExclusion", tmpDeltaPtKTImprovedCMSPartialExclusion, centralityPercentile);
c2b0976c 1549 if(tmpDeltaPtKTImprovedCMSPartialExclusion_Signal > -10000.0)
1550 FillHistogram("hDeltaPtKTImprovedCMSPartialExclusion_Signal", tmpDeltaPtKTImprovedCMSPartialExclusion_Signal, centralityPercentile);
1551 if(tmpDeltaPtKTImprovedCMSFullExclusion > -10000.0)
1552 FillHistogram("hDeltaPtKTImprovedCMSFullExclusion", tmpDeltaPtKTImprovedCMSFullExclusion, centralityPercentile);
392aa819 1553
0cb49d99 1554 if(tmpDeltaPtNoBackground > 0.000001)
1555 FillHistogram("hDeltaPtNoBackgroundNoEmptyCones", tmpDeltaPtNoBackground, centralityPercentile);
1556 else if(tmpDeltaPtNoBackground > -10000.0)
1557 FillHistogram("hDeltaPtNoBackground", tmpDeltaPtNoBackground, centralityPercentile);
1558
8628b70c 1559
b95d0245 1560 if(fAnalyzeDeprecatedBackgrounds)
1561 {
1562 if(tmpDeltaPtKTPbPb > -10000.0)
1563 FillHistogram("hDeltaPtKTPbPb", tmpDeltaPtKTPbPb, centralityPercentile);
1564 if(tmpDeltaPtKTPbPbWithGhosts > -10000.0)
1565 FillHistogram("hDeltaPtKTPbPbWithGhosts", tmpDeltaPtKTPbPbWithGhosts, centralityPercentile);
1566 if(tmpDeltaPtKTCMS > -10000.0)
1567 FillHistogram("hDeltaPtKTCMS", tmpDeltaPtKTCMS, centralityPercentile);
1568 if(tmpDeltaPtKTMean > -10000.0)
1569 FillHistogram("hDeltaPtKTMean", tmpDeltaPtKTMean, centralityPercentile);
1570 if(tmpDeltaPtKTTrackLike > -10000.0)
1571 FillHistogram("hDeltaPtKTTrackLike", tmpDeltaPtKTTrackLike, centralityPercentile);
1572
b95d0245 1573 if(tmpDeltaPtTR > -10000.0)
1574 FillHistogram("hDeltaPtTR", tmpDeltaPtTR, centralityPercentile);
1575 }
8628b70c 1576 }
1577
1578 #ifdef DEBUGMODE
efb9b161 1579 AliInfo("Calculate()::Background done.");
8628b70c 1580 #endif
12eb4ac1 1581
1582 ////////////////////// NOTE: Pythia histograms
1583 if(fAnalyzePythia)
1584 {
1585 FillHistogram("hPythiaPtHard", GetPtHard());
392aa819 1586 FillHistogram("hPythiaNTrials", GetPtHardBin()+0.1, GetPythiaTrials());
4856f9c9 1587 FillHistogram("hPythiaXSection", GetPtHardBin()+0.1, fCrossSection);
12eb4ac1 1588
392aa819 1589 #ifdef DEBUGMODE
12eb4ac1 1590 AliInfo("Calculate()::Pythia done.");
1591 #endif
1592 }
0983dd5b 1593 #ifdef DEBUGMODE
1594 AliInfo("Calculate() done.");
1595 #endif
12eb4ac1 1596}
8628b70c 1597
1598//________________________________________________________________________
392aa819 1599Bool_t AliAnalysisTaskChargedJetsPA::UserNotify()
8628b70c 1600{
1601 // Implemented Notify() to read the cross sections
1602 // and number of trials from pyxsec.root
1603 //
1604 #ifdef DEBUGMODE
392aa819 1605 AliInfo("UserNotify started.");
8628b70c 1606 #endif
1607
1608 if(fAnalyzePythia)
1609 {
1610 TTree *tree = AliAnalysisManager::GetAnalysisManager()->GetTree();
1611 TFile *currFile = tree->GetCurrentFile();
1612
1613 TString file(currFile->GetName());
1614
1615 if(file.Contains("root_archive.zip#")){
1616 Ssiz_t pos1 = file.Index("root_archive",12,TString::kExact);
1617 Ssiz_t pos = file.Index("#",1,pos1,TString::kExact);
1618 file.Replace(pos+1,20,"");
1619 }
1620 else {
1621 // not an archive take the basename....
1622 file.ReplaceAll(gSystem->BaseName(file.Data()),"");
1623 }
8628b70c 1624
1625 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
1626 if(!fxsec){
1627 // next trial fetch the histgram file
1628 fxsec = TFile::Open(Form("%s%s",file.Data(),"pyxsec_hists.root"));
1629 if(!fxsec){
1630 // not a severe condition but inciate that we have no information
1631 return kFALSE;
1632 }
1633 else{
1634 // find the tlist we want to be independtent of the name so use the Tkey
1635 TKey* key = (TKey*)fxsec->GetListOfKeys()->At(0);
1636 if(!key){
1637 fxsec->Close();
1638 return kFALSE;
1639 }
1640 TList *list = dynamic_cast<TList*>(key->ReadObj());
1641 if(!list){
1642 fxsec->Close();
1643 return kFALSE;
1644 }
1645 fCrossSection = ((TProfile*)list->FindObject("h1Xsec"))->GetBinContent(1);
1646 fTrials = ((TH1F*)list->FindObject("h1Trials"))->GetBinContent(1);
1647 fxsec->Close();
1648 }
1649 } // no tree pyxsec.root
1650 else {
1651 TTree *xtree = (TTree*)fxsec->Get("Xsection");
1652 if(!xtree){
1653 fxsec->Close();
1654 return kFALSE;
1655 }
1656 UInt_t ntrials = 0;
1657 Double_t xsection = 0;
1658 xtree->SetBranchAddress("xsection",&xsection);
1659 xtree->SetBranchAddress("ntrials",&ntrials);
1660 xtree->GetEntry(0);
1661 fTrials = ntrials;
1662 fCrossSection = xsection;
1663 fxsec->Close();
1664 }
8628b70c 1665 }
0cb49d99 1666 #ifdef DEBUGMODE
1667 AliInfo("UserNotify ended.");
1668 #endif
8628b70c 1669 return kTRUE;
1670}
1671
392aa819 1672
12eb4ac1 1673//________________________________________________________________________
1674inline Double_t AliAnalysisTaskChargedJetsPA::EtaToTheta(Double_t arg)
1675 {return 2.*atan(exp(-arg));}
1676//________________________________________________________________________
1677inline Double_t AliAnalysisTaskChargedJetsPA::ThetaToEta(Double_t arg)
1678{
1679 if ((arg > TMath::Pi()) || (arg < 0.0))
1680 {
1681 AliError(Form("ThetaToEta got wrong input! (%f)", arg));
1682 return 0.0;
1683 }
1684 return -log(tan(arg/2.));
1685}
1686//________________________________________________________________________
1687inline Double_t AliAnalysisTaskChargedJetsPA::GetDeltaPhi(Double_t phi1, Double_t phi2)
1688 {return min(TMath::Abs(phi1-phi2),TMath::TwoPi() - TMath::Abs(phi1-phi2));}
1689
1690//________________________________________________________________________
1691Double_t AliAnalysisTaskChargedJetsPA::MCGetOverlapCircleRectancle(Double_t cPosX, Double_t cPosY, Double_t cRadius, Double_t rPosXmin, Double_t rPosXmax, Double_t rPosYmin, Double_t rPosYmax)
1692{
1693 const Int_t kTests = 1000;
1694 Int_t hits = 0;
1695 TRandom3 randomGen(0);
1696
1697 // Loop over kTests-many tests
1698 for (Int_t i=0; i<kTests; i++)
1699 {
1700 //Choose random position in rectangle for the tester
1701 Double_t tmpTestX = randomGen.Uniform(rPosXmin, rPosXmax);
1702 Double_t tmpTestY = randomGen.Uniform(rPosYmin, rPosYmax);
1703
1704 //Check, if tester is in circle. If yes, increment circle counter.
1705 Double_t tmpDistance = TMath::Sqrt( (tmpTestX - cPosX)*(tmpTestX - cPosX) + (tmpTestY - cPosY)*(tmpTestY - cPosY) );
1706 if(tmpDistance < cRadius)
1707 hits++;
1708 }
1709
1710 // return ratio
1711 return (static_cast<Double_t>(hits)/static_cast<Double_t>(kTests));
1712}
1713
b95d0245 1714//________________________________________________________________________
1715Double_t AliAnalysisTaskChargedJetsPA::MCGetOverlapMultipleCirclesRectancle(Int_t numCircles, std::vector<Double_t> cPosX, std::vector<Double_t> cPosY, Double_t cRadius, Double_t rPosXmin, Double_t rPosXmax, Double_t rPosYmin, Double_t rPosYmax)
1716{
1717
1718 const Int_t kTests = 1000;
1719 Int_t hits = 0;
1720 TRandom3 randomGen(0);
1721
1722 // Loop over kTests-many tests
1723 for (Int_t i=0; i<kTests; i++)
1724 {
1725 //Choose random position in rectangle for the tester
1726 Double_t tmpTestX = randomGen.Uniform(rPosXmin, rPosXmax);
1727 Double_t tmpTestY = randomGen.Uniform(rPosYmin, rPosYmax);
1728
1729 //Check, if tester is in one of the circles. If yes, increment circle counter.
1730 for(Int_t j=0; j<numCircles; j++)
1731 {
1732 Double_t tmpDistance = TMath::Sqrt( (tmpTestX - cPosX[j])*(tmpTestX - cPosX[j]) + (tmpTestY - cPosY[j])*(tmpTestY - cPosY[j]) );
1733 if(tmpDistance < cRadius)
1734 {
1735 hits++;
1736 break;
1737 }
1738 }
1739 }
1740
1741 // return ratio
1742 return (static_cast<Double_t>(hits)/static_cast<Double_t>(kTests));
1743
1744}
1745
8628b70c 1746//________________________________________________________________________
1747inline void AliAnalysisTaskChargedJetsPA::FillHistogram(const char * key, Double_t x)
1748{
1749 TH1* tmpHist = static_cast<TH1*>(fOutputList->FindObject(GetHistoName(key)));
1750 if(!tmpHist)
1751 {
392aa819 1752 AliError(Form("Cannot find histogram <%s> ",key)) ;
8628b70c 1753 return;
1754 }
1755
1756 tmpHist->Fill(x);
1757}
1758
1759//________________________________________________________________________
1760inline void AliAnalysisTaskChargedJetsPA::FillHistogram(const char * key, Double_t x, Double_t y)
1761{
1762 TH1* tmpHist = static_cast<TH1*>(fOutputList->FindObject(GetHistoName(key)));
1763 if(!tmpHist)
1764 {
392aa819 1765 AliError(Form("Cannot find histogram <%s> ",key));
8628b70c 1766 return;
1767 }
1768
1769 if (tmpHist->IsA()->GetBaseClass("TH1"))
1770 static_cast<TH1*>(tmpHist)->Fill(x,y); // Fill x with y
1771 else if (tmpHist->IsA()->GetBaseClass("TH2"))
1772 static_cast<TH2*>(tmpHist)->Fill(x,y); // Fill x,y with 1
1773}
1774
1775//________________________________________________________________________
1776inline void AliAnalysisTaskChargedJetsPA::FillHistogram(const char * key, Double_t x, Double_t y, Double_t add)
1777{
1778 TH2* tmpHist = static_cast<TH2*>(fOutputList->FindObject(GetHistoName(key)));
1779 if(!tmpHist)
1780 {
392aa819 1781 AliError(Form("Cannot find histogram <%s> ",key));
8628b70c 1782 return;
1783 }
1784
1785 tmpHist->Fill(x,y,add);
1786}
1787//________________________________________________________________________
1788template <class T> T* AliAnalysisTaskChargedJetsPA::AddHistogram1D(const char* name, const char* title, const char* options, Int_t xBins, Double_t xMin, Double_t xMax, const char* xTitle, const char* yTitle)
1789{
1790 T* tmpHist = new T(GetHistoName(name), GetHistoName(title), xBins, xMin, xMax);
1791
1792 tmpHist->GetXaxis()->SetTitle(xTitle);
1793 tmpHist->GetYaxis()->SetTitle(yTitle);
1794 tmpHist->SetOption(options);
1795 tmpHist->SetMarkerStyle(kFullCircle);
1796 tmpHist->Sumw2();
1797
1798 fHistList->Add(tmpHist);
1799 fHistCount++;
1800
1801 return tmpHist;
1802}
1803
1804//________________________________________________________________________
1805template <class T> T* AliAnalysisTaskChargedJetsPA::AddHistogram2D(const char* name, const char* title, const char* options, Int_t xBins, Double_t xMin, Double_t xMax, Int_t yBins, Double_t yMin, Double_t yMax, const char* xTitle, const char* yTitle, const char* zTitle)
1806{
1807 T* tmpHist = new T(GetHistoName(name), GetHistoName(title), xBins, xMin, xMax, yBins, yMin, yMax);
1808 tmpHist->GetXaxis()->SetTitle(xTitle);
1809 tmpHist->GetYaxis()->SetTitle(yTitle);
1810 tmpHist->GetZaxis()->SetTitle(zTitle);
1811 tmpHist->SetOption(options);
1812 tmpHist->SetMarkerStyle(kFullCircle);
1813 tmpHist->Sumw2();
1814
1815 fHistList->Add(tmpHist);
1816 fHistCount++;
1817
1818 return tmpHist;
1819}
1820
1821//________________________________________________________________________
1822void AliAnalysisTaskChargedJetsPA::Terminate(Option_t *)
1823{
1824 PostData(1, fOutputList);
1825
1826 // Mandatory
1827 fOutputList = dynamic_cast<TList*> (GetOutputData(1)); // '1' refers to the output slot
1828 if (!fOutputList) {
1829 printf("ERROR: Output list not available\n");
1830 return;
1831 }
1832}
1833
1834//________________________________________________________________________
1835AliAnalysisTaskChargedJetsPA::~AliAnalysisTaskChargedJetsPA()
1836{
1837 // Destructor. Clean-up the output list, but not the histograms that are put inside
1838 // (the list is owner and will clean-up these histograms). Protect in PROOF case.
1839 if (fOutputList && !AliAnalysisManager::GetAnalysisManager()->IsProofMode()) {
1840 delete fOutputList;
1841 }
1842}
1843
1844//________________________________________________________________________
1845void AliAnalysisTaskChargedJetsPA::UserCreateOutputObjects()
1846{
1847 // called once to create user defined output objects like histograms, plots etc.
1848 // and to put it on the output list.
1849 // Note: Saving to file with e.g. OpenFile(0) is must be before creating other objects.
1850
1851 fRandom = new TRandom3(0);
392aa819 1852
1853
8628b70c 1854 fOutputList = new TList();
1855 fOutputList->SetOwner(); // otherwise it produces leaks in merging
1856
392aa819 1857 // NOTE: Pythia histograms
1858 AddHistogram1D<TProfile>("hPythiaXSection", "Pythia cross section distribution", "", fNumPtHardBins+1, 0, fNumPtHardBins+1, "p_{T} hard bin","dN^{Events}/dp_{T,hard}");
1859 AddHistogram1D<TH1D>("hPythiaNTrials", "Pythia trials (no correction for manual cuts)", "", fNumPtHardBins+1, 0, fNumPtHardBins+1, "p_{T} hard bin", "Trials");
1860
1861
8628b70c 1862 PostData(1, fOutputList);
1863}
1864
1865//________________________________________________________________________
1866void AliAnalysisTaskChargedJetsPA::UserExec(Option_t *)
1867{
1868 #ifdef DEBUGMODE
efb9b161 1869 AliInfo("UserExec() started.");
8628b70c 1870 #endif
1871
12eb4ac1 1872 if (!InputEvent())
1873 {
1874 AliError("??? Event pointer == 0 ???");
1875 return;
1876 }
1877
1878 if (!fInitialized)
1879 ExecOnce(); // Get tracks, jets, background from arrays if not already given + Init Histos
1880
8628b70c 1881 Calculate(InputEvent());
1882
1883 PostData(1, fOutputList);
1884}