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