]>
Commit | Line | Data |
---|---|---|
9e64e697 | 1 | // $Id: AliAnalysisTaskEmcalJetSpectraMECpA.cxx 3010 2012-06-10 05:40:56Z loizides $ |
d7afd9ac | 2 | // |
3 | // Jet spectrum task. | |
4 | // | |
5 | // Author: R.Reed, M.Connors | |
6 | ||
7 | #include "AliAnalysisTaskEmcalJetSpectraMECpA.h" | |
8 | ||
9 | #include <TCanvas.h> | |
10 | #include <TChain.h> | |
11 | #include <TClonesArray.h> | |
12 | #include <TH1F.h> | |
13 | #include <TH2F.h> | |
14 | #include <TH3F.h> | |
15 | #include <THnSparse.h> | |
16 | #include <TList.h> | |
17 | #include <TLorentzVector.h> | |
18 | #include <TParameter.h> | |
19 | #include <TParticle.h> | |
20 | #include <TTree.h> | |
21 | #include <TVector3.h> | |
22 | ||
23 | #include "AliAODEvent.h" | |
24 | #include "AliAnalysisManager.h" | |
25 | #include "AliAnalysisTask.h" | |
26 | #include "AliCentrality.h" | |
27 | #include "AliESDEvent.h" | |
28 | #include "AliESDInputHandler.h" | |
29 | #include "AliEmcalJet.h" | |
30 | #include "AliVCluster.h" | |
31 | #include "AliRhoParameter.h" | |
32 | #include "AliEmcalParticle.h" | |
33 | ||
34 | ClassImp(AliAnalysisTaskEmcalJetSpectraMECpA) | |
35 | ||
36 | //________________________________________________________________________ | |
37 | AliAnalysisTaskEmcalJetSpectraMECpA::AliAnalysisTaskEmcalJetSpectraMECpA() : | |
38 | AliAnalysisTaskEmcalJet("spectra",kFALSE), | |
39 | fHistRhovsCent(0), | |
40 | fHistNjetvsCent(0) | |
41 | { | |
42 | // Default constructor. | |
43 | for (Int_t i = 0;i<6;++i){ | |
44 | fHistJetPtvsTrackPt[i] = 0; | |
45 | fHistRawJetPtvsTrackPt[i] = 0; | |
46 | fHistTrackPt[i] = 0; | |
47 | fHistEP0[i] = 0; | |
48 | fHistEP0A[i] = 0; | |
49 | fHistEP0C[i] = 0; | |
50 | fHistEPAvsC[i] = 0; | |
51 | fHistJetPtvsdEP[i] = 0; | |
52 | fHistJetPtvsdEPBias[i] = 0; | |
53 | fHistRhovsEP[i] = 0; | |
54 | fHistJetPtEtaPhi[i] = 0; | |
55 | ||
56 | } | |
d7afd9ac | 57 | SetMakeGeneralHistograms(kTRUE); |
58 | } | |
59 | ||
60 | //________________________________________________________________________ | |
61 | AliAnalysisTaskEmcalJetSpectraMECpA::AliAnalysisTaskEmcalJetSpectraMECpA(const char *name) : | |
62 | AliAnalysisTaskEmcalJet(name,kTRUE), | |
63 | fHistRhovsCent(0), | |
64 | fHistNjetvsCent(0) | |
65 | { | |
66 | for (Int_t i = 0;i<6;++i){ | |
67 | fHistJetPtvsTrackPt[i] = 0; | |
68 | fHistRawJetPtvsTrackPt[i] = 0; | |
69 | fHistTrackPt[i] = 0; | |
70 | fHistEP0[i] = 0; | |
71 | fHistEP0A[i] = 0; | |
72 | fHistEP0C[i] = 0; | |
73 | fHistEPAvsC[i] = 0; | |
74 | fHistJetPtvsdEP[i] = 0; | |
75 | fHistJetPtvsdEPBias[i] = 0; | |
76 | fHistRhovsEP[i] = 0; | |
77 | fHistJetPtEtaPhi[i] = 0; | |
78 | } | |
d7afd9ac | 79 | SetMakeGeneralHistograms(kTRUE); |
80 | } | |
81 | ||
82 | //________________________________________________________________________ | |
83 | void AliAnalysisTaskEmcalJetSpectraMECpA::UserCreateOutputObjects() | |
84 | { | |
85 | if (! fCreateHisto) | |
86 | return; | |
87 | AliAnalysisTaskEmcalJet::UserCreateOutputObjects(); | |
88 | ||
89 | fHistRhovsCent = new TH2F("RhovsCent", "RhovsCent", 100, 0.0, 100.0, 500, 0, 500); | |
90 | fHistNjetvsCent = new TH2F("NjetvsCent", "NjetvsCent", 100, 0.0, 100.0, 100, 0, 100); | |
91 | ||
92 | TString name; | |
93 | TString title; | |
94 | for (Int_t i = 0;i<6;++i){ | |
95 | name = TString(Form("JetPtvsTrackPt_%i",i)); | |
96 | title = TString(Form("Jet pT vs Leading Track pT cent bin %i",i)); | |
97 | fHistJetPtvsTrackPt[i] = new TH2F(name,title,1000,-500,500,100,0,100); | |
98 | fOutput->Add(fHistJetPtvsTrackPt[i]); | |
99 | name = TString(Form("RawJetPtvsTrackPt_%i",i)); | |
100 | title = TString(Form("Raw Jet pT vs Leading Track pT cent bin %i",i)); | |
101 | fHistRawJetPtvsTrackPt[i] = new TH2F(name,title,1000,-500,500,100,0,100); | |
102 | fOutput->Add(fHistRawJetPtvsTrackPt[i]); | |
103 | name = TString(Form("TrackPt_%i",i)); | |
104 | title = TString(Form("Track pT cent bin %i",i)); | |
105 | fHistTrackPt[i] = new TH1F(name,title,1000,0,200); | |
106 | fOutput->Add(fHistTrackPt[i]); | |
107 | ||
108 | name = TString(Form("EP0_%i",i)); | |
109 | title = TString(Form("EP VZero cent bin %i",i)); | |
110 | fHistEP0[i] = new TH1F(name,title,100,-TMath::Pi(),TMath::Pi()); | |
111 | fOutput->Add(fHistEP0[i]); | |
112 | name = TString(Form("EP0A_%i",i)); | |
113 | title = TString(Form("EP VZero cent bin %i",i)); | |
114 | fHistEP0A[i] = new TH1F(name,title,100,-TMath::Pi(),TMath::Pi()); | |
115 | fOutput->Add(fHistEP0A[i]); | |
116 | name = TString(Form("EP0C_%i",i)); | |
117 | title = TString(Form("EP VZero cent bin %i",i)); | |
118 | fHistEP0C[i] = new TH1F(name,title,100,-TMath::Pi(),TMath::Pi()); | |
119 | fOutput->Add(fHistEP0C[i]); | |
120 | name = TString(Form("EPAvsC_%i",i)); | |
121 | title = TString(Form("EP VZero cent bin %i",i)); | |
122 | fHistEPAvsC[i] = new TH2F(name,title,100,-TMath::Pi(),TMath::Pi(),100,-TMath::Pi(),TMath::Pi()); | |
123 | fOutput->Add(fHistEPAvsC[i]); | |
124 | name = TString(Form("JetPtvsdEP_%i",i)); | |
125 | title = TString(Form("Jet pt vs dEP cent bin %i",i)); | |
126 | fHistJetPtvsdEP[i] = new TH2F(name,title,1000,-500,500,400,-2*TMath::Pi(),2*TMath::Pi()); | |
127 | fOutput->Add(fHistJetPtvsdEP[i]); | |
128 | name = TString(Form("JetPtvsdEPBias_%i",i)); | |
129 | title = TString(Form("Bias Jet pt vs dEP cent bin %i",i)); | |
130 | fHistJetPtvsdEPBias[i] = new TH2F(name,title,1000,-500,500,400,-2*TMath::Pi(),2*TMath::Pi()); | |
131 | fOutput->Add(fHistJetPtvsdEPBias[i]); | |
132 | name = TString(Form("JetPtvsEP_%i",i)); | |
133 | title = TString(Form("Jet pt vs EP cent bin %i",i)); | |
134 | fHistJetPtvsEP[i] = new TH2F(name,title,1000,-500,500,400,-2*TMath::Pi(),2*TMath::Pi()); | |
135 | fOutput->Add(fHistJetPtvsEP[i]); | |
136 | name = TString(Form("JetPtvsEPBias_%i",i)); | |
137 | title = TString(Form("Bias Jet pt vs EP cent bin %i",i)); | |
138 | fHistJetPtvsEPBias[i] = new TH2F(name,title,1000,-500,500,400,-2*TMath::Pi(),2*TMath::Pi()); | |
139 | fOutput->Add(fHistJetPtvsEPBias[i]); | |
140 | name = TString(Form("RhovsEP_%i",i)); | |
141 | title = TString(Form("Rho vs EP cent bin %i",i)); | |
142 | fHistRhovsEP[i] = new TH2F(name,title,500,0,500,400,-2*TMath::Pi(),2*TMath::Pi()); | |
143 | fOutput->Add(fHistRhovsEP[i]); | |
144 | ||
145 | name = TString(Form("JetPtEtaPhi_%i",i)); | |
146 | title = TString(Form("JetPtEtaPhi_%i",i)); | |
147 | fHistJetPtEtaPhi[i] = new TH3F(name,title,300,-100,200,26,-0.8,0.8,400,-2*TMath::Pi(),2*TMath::Pi()); | |
148 | fOutput->Add(fHistJetPtEtaPhi[i]); | |
149 | ||
150 | ||
151 | } | |
152 | ||
153 | ||
154 | fOutput->Add(fHistRhovsCent); | |
155 | fOutput->Add(fHistNjetvsCent); | |
156 | ||
157 | PostData(1, fOutput); | |
158 | } | |
159 | ||
160 | //________________________________________________________________________ | |
161 | ||
162 | Int_t AliAnalysisTaskEmcalJetSpectraMECpA::GetCentBin(Double_t cent) const | |
163 | { | |
164 | // Get centrality bin. | |
165 | ||
166 | Int_t centbin = -1; | |
167 | if (cent>=0 && cent<10) | |
168 | centbin = 0; | |
169 | else if (cent>=10 && cent<20) | |
170 | centbin = 1; | |
171 | else if (cent>=20 && cent<30) | |
172 | centbin = 2; | |
173 | else if (cent>=30 && cent<40) | |
174 | centbin = 3; | |
175 | else if (cent>=40 && cent<50) | |
176 | centbin = 4; | |
177 | else if (cent>=50 && cent<90) | |
178 | centbin = 5; | |
179 | return centbin; | |
180 | } | |
181 | ||
182 | //________________________________________________________________________ | |
183 | ||
184 | Float_t AliAnalysisTaskEmcalJetSpectraMECpA:: RelativePhi(Double_t mphi,Double_t vphi) const | |
185 | { | |
186 | if (vphi < -1*TMath::Pi()) vphi += (2*TMath::Pi()); | |
187 | else if (vphi > TMath::Pi()) vphi -= (2*TMath::Pi()); | |
188 | if (mphi < -1*TMath::Pi()) mphi += (2*TMath::Pi()); | |
189 | else if (mphi > TMath::Pi()) mphi -= (2*TMath::Pi()); | |
190 | double dphi = mphi-vphi; | |
191 | if (dphi < -1*TMath::Pi()) dphi += (2*TMath::Pi()); | |
192 | else if (dphi > TMath::Pi()) dphi -= (2*TMath::Pi()); | |
193 | ||
194 | return dphi;//dphi in [-Pi, Pi] | |
195 | } | |
196 | ||
197 | ||
198 | //________________________________________________________________________ | |
199 | Bool_t AliAnalysisTaskEmcalJetSpectraMECpA::Run() | |
200 | { | |
9e64e697 | 201 | |
202 | ||
d7afd9ac | 203 | Int_t centbin = GetCentBin(fCent); |
204 | //for pp analyses we will just use the first centrality bin | |
205 | if (centbin == -1) | |
206 | centbin = 0; | |
207 | ||
208 | if (!fTracks){ | |
209 | cout << "Tracks not found: " << fCent << endl; | |
210 | return kTRUE; | |
211 | } | |
212 | ||
213 | const Int_t nTrack = fTracks->GetEntriesFast(); | |
214 | for (int i = 0;i<nTrack;i++){ | |
215 | AliVParticle *track = static_cast<AliVParticle*>(fTracks->At(i)); | |
216 | if (!track) | |
217 | continue; | |
218 | fHistTrackPt[centbin]->Fill(track->Pt()); | |
219 | } | |
220 | ||
221 | fHistEP0[centbin]->Fill(fEPV0); | |
222 | fHistEP0A[centbin]->Fill(fEPV0A); | |
223 | fHistEP0C[centbin]->Fill(fEPV0C); | |
224 | fHistEPAvsC[centbin]->Fill(fEPV0A,fEPV0C); | |
225 | TString fRhoScaledName = fRhoName; | |
d7afd9ac | 226 | fRho = GetRhoFromEvent(fRhoScaledName); |
227 | fRhoVal = fRho->GetVal(); | |
228 | fHistRhovsCent->Fill(fCent,fRhoVal); | |
229 | fHistRhovsEP[centbin]->Fill(fRhoVal,fEPV0); | |
230 | const Int_t Njets = fJets->GetEntriesFast(); | |
231 | ||
232 | Int_t NjetAcc = 0; | |
233 | for (Int_t iJets = 0; iJets < Njets; ++iJets) { | |
234 | AliEmcalJet *jet = static_cast<AliEmcalJet*>(fJets->At(iJets)); | |
235 | if (!jet) | |
236 | continue; | |
237 | if (jet->Area()==0) | |
238 | continue; | |
239 | if (jet->Pt()<0.1) | |
240 | continue; | |
241 | if (jet->MaxTrackPt()>100) | |
242 | continue; | |
243 | if (! AcceptJet(jet)) | |
244 | continue; | |
245 | // jets.push_back(jet); | |
246 | NjetAcc++; | |
247 | Double_t jetPt = -500; | |
248 | jetPt = jet->Pt()-jet->Area()*fRhoVal; | |
249 | fHistJetPtvsTrackPt[centbin]->Fill(jetPt,jet->MaxTrackPt()); | |
250 | fHistRawJetPtvsTrackPt[centbin]->Fill(jet->Pt(),jet->MaxTrackPt()); | |
251 | ||
252 | fHistJetPtEtaPhi[centbin]->Fill(jet->Pt(),jet->Eta(),jet->Phi()); | |
253 | ||
254 | fHistJetPtvsdEP[centbin]->Fill(jetPt,RelativePhi((fEPV0+TMath::Pi()),jet->Phi())); | |
255 | fHistJetPtvsEP[centbin]->Fill(jetPt,fEPV0); | |
256 | if (jet->MaxTrackPt()>5.0){ | |
257 | fHistJetPtvsdEPBias[centbin]->Fill(jetPt,RelativePhi((fEPV0+TMath::Pi()),jet->Phi())); | |
258 | fHistJetPtvsEPBias[centbin]->Fill(jetPt,fEPV0); | |
259 | } | |
260 | } | |
261 | ||
262 | fHistNjetvsCent->Fill(fCent,NjetAcc); | |
263 | return kTRUE; | |
264 | } | |
265 | ||
266 | ||
267 | ||
268 | ||
269 |