19 |
|
|
20 |
|
#include "DataFormats/HeavyIonEvent/interface/CentralityBins.h" |
21 |
|
#include "DataFormats/HeavyIonEvent/interface/Centrality.h" |
22 |
+ |
#include "Geometry/Records/interface/CaloGeometryRecord.h" |
23 |
|
|
24 |
|
#include "DataFormats/CaloTowers/interface/CaloTowerCollection.h" |
24 |
– |
#include "DataFormats/PatCandidates/interface/Jet.h" |
25 |
|
#include "DataFormats/JetReco/interface/CaloJetCollection.h" |
26 |
|
#include "SimDataFormats/HiGenData/interface/GenHIEvent.h" |
27 |
|
#include "SimDataFormats/GeneratorProducts/interface/GenEventInfoProduct.h" |
28 |
+ |
#include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h" |
29 |
|
#include "DataFormats/JetReco/interface/GenJetCollection.h" |
30 |
|
#include "DataFormats/VertexReco/interface/Vertex.h" |
31 |
|
#include "DataFormats/Math/interface/deltaPhi.h" |
39 |
|
#include "DataFormats/L1GlobalTrigger/interface/L1GlobalTriggerObjectMap.h" |
40 |
|
#include "L1Trigger/GlobalTrigger/interface/L1GlobalTrigger.h" |
41 |
|
|
42 |
+ |
#include "DataFormats/EcalRecHit/interface/EcalRecHitCollections.h" |
43 |
+ |
#include "DataFormats/HcalRecHit/interface/HcalRecHitCollections.h" |
44 |
+ |
|
45 |
|
using namespace std; |
46 |
|
using namespace edm; |
47 |
|
using namespace reco; |
48 |
|
|
49 |
< |
HiInclusiveJetAnalyzer::HiInclusiveJetAnalyzer(const edm::ParameterSet& iConfig) { |
50 |
< |
|
49 |
> |
HiInclusiveJetAnalyzer::HiInclusiveJetAnalyzer(const edm::ParameterSet& iConfig) : |
50 |
> |
geo(0) |
51 |
> |
{ |
52 |
|
|
53 |
|
jetTag_ = iConfig.getParameter<InputTag>("jetTag"); |
54 |
< |
vtxTag_ = iConfig.getUntrackedParameter<edm::InputTag>("vtxTag",edm::InputTag("hiSelectedVertex")); |
54 |
> |
matchTag_ = iConfig.getUntrackedParameter<InputTag>("matchTag",jetTag_); |
55 |
> |
|
56 |
> |
vtxTag_ = iConfig.getUntrackedParameter<edm::InputTag>("vtxTag",edm::InputTag("hiSelectedVertex")); |
57 |
> |
trackTag_ = iConfig.getParameter<InputTag>("trackTag"); |
58 |
> |
useQuality_ = iConfig.getUntrackedParameter<bool>("useQuality",1); |
59 |
> |
trackQuality_ = iConfig.getUntrackedParameter<string>("trackQuality","highPurity"); |
60 |
|
|
61 |
|
isMC_ = iConfig.getUntrackedParameter<bool>("isMC",false); |
62 |
+ |
fillGenJets_ = iConfig.getUntrackedParameter<bool>("fillGenJets",false); |
63 |
+ |
|
64 |
+ |
doTrigger_ = iConfig.getUntrackedParameter<bool>("doTrigger",false); |
65 |
+ |
|
66 |
+ |
rParam = iConfig.getParameter<double>("rParam"); |
67 |
+ |
hardPtMin_ = iConfig.getUntrackedParameter<double>("hardPtMin",4); |
68 |
+ |
jetPtMin_ = iConfig.getUntrackedParameter<double>("jetPtMin",0); |
69 |
|
|
70 |
|
if(isMC_){ |
71 |
|
genjetTag_ = iConfig.getParameter<InputTag>("genjetTag"); |
74 |
|
verbose_ = iConfig.getUntrackedParameter<bool>("verbose",false); |
75 |
|
|
76 |
|
useCentrality_ = iConfig.getUntrackedParameter<bool>("useCentrality",false); |
77 |
< |
useVtx_ = iConfig.getUntrackedParameter<bool>("useVtx",true); |
77 |
> |
useVtx_ = iConfig.getUntrackedParameter<bool>("useVtx",false); |
78 |
|
useJEC_ = iConfig.getUntrackedParameter<bool>("useJEC",true); |
79 |
+ |
usePat_ = iConfig.getUntrackedParameter<bool>("usePAT",true); |
80 |
|
|
81 |
< |
if(!isMC_){ |
81 |
> |
doLifeTimeTagging_ = iConfig.getUntrackedParameter<bool>("doLifeTimeTagging",false); |
82 |
> |
doLifeTimeTaggingExtras_ = iConfig.getUntrackedParameter<bool>("doLifeTimeTaggingExtras",true); |
83 |
> |
saveBfragments_ = iConfig.getUntrackedParameter<bool>("saveBfragments",false); |
84 |
> |
skipCorrections_ = iConfig.getUntrackedParameter<bool>("skipCorrections",false); |
85 |
> |
|
86 |
> |
pfCandidateLabel_ = iConfig.getUntrackedParameter<edm::InputTag>("pfCandidateLabel",edm::InputTag("particleFlowTmp")); |
87 |
> |
|
88 |
> |
EBSrc_ = iConfig.getUntrackedParameter<edm::InputTag>("EBRecHitSrc",edm::InputTag("ecalRecHit","EcalRecHitsEB")); |
89 |
> |
EESrc_ = iConfig.getUntrackedParameter<edm::InputTag>("EERecHitSrc",edm::InputTag("ecalRecHit","EcalRecHitsEE")); |
90 |
> |
HcalRecHitHFSrc_ = iConfig.getUntrackedParameter<edm::InputTag>("hcalHFRecHitSrc",edm::InputTag("hfreco")); |
91 |
> |
HcalRecHitHBHESrc_ = iConfig.getUntrackedParameter<edm::InputTag>("hcalHBHERecHitSrc",edm::InputTag("hbhereco")); |
92 |
> |
|
93 |
> |
genParticleSrc_ = iConfig.getUntrackedParameter<edm::InputTag>("genParticles",edm::InputTag("hiGenParticles")); |
94 |
> |
|
95 |
> |
if(doTrigger_){ |
96 |
|
L1gtReadout_ = iConfig.getParameter<edm::InputTag>("L1gtReadout"); |
97 |
|
hltResName_ = iConfig.getUntrackedParameter<string>("hltTrgResults","TriggerResults::HLT"); |
98 |
|
|
108 |
|
} |
109 |
|
} |
110 |
|
|
111 |
< |
cout<<" jet collection : "<<jetTag_<<endl; |
112 |
< |
if(isMC_)cout<<" genjet collection : "<<genjetTag_<<endl; |
111 |
> |
// cout<<" jet collection : "<<jetTag_<<endl; |
112 |
> |
doSubEvent_ = 0; |
113 |
|
|
114 |
+ |
if(isMC_){ |
115 |
+ |
// cout<<" genjet collection : "<<genjetTag_<<endl; |
116 |
+ |
genPtMin_ = iConfig.getUntrackedParameter<double>("genPtMin",0); |
117 |
+ |
doSubEvent_ = iConfig.getUntrackedParameter<bool>("doSubEvent",1); |
118 |
+ |
} |
119 |
|
|
120 |
|
|
121 |
|
} |
139 |
|
string jetTagTitle = jetTag_.label()+" Jet Analysis Tree"; |
140 |
|
t = fs1->make<TTree>("t",jetTagTitle.c_str()); |
141 |
|
|
105 |
– |
|
142 |
|
// TTree* t= new TTree("t","Jet Response Analyzer"); |
143 |
< |
t->Branch("run",&jets_.run,"run/I"); |
143 |
> |
//t->Branch("run",&jets_.run,"run/I"); |
144 |
|
t->Branch("evt",&jets_.evt,"evt/I"); |
145 |
< |
t->Branch("lumi",&jets_.lumi,"lumi/I"); |
145 |
> |
//t->Branch("lumi",&jets_.lumi,"lumi/I"); |
146 |
|
t->Branch("b",&jets_.b,"b/F"); |
147 |
< |
t->Branch("vx",&jets_.vx,"vx/F"); |
148 |
< |
t->Branch("vy",&jets_.vy,"vy/F"); |
149 |
< |
t->Branch("vz",&jets_.vz,"vz/F"); |
150 |
< |
t->Branch("hf",&jets_.hf,"hf/F"); |
147 |
> |
if (useVtx_) { |
148 |
> |
t->Branch("vx",&jets_.vx,"vx/F"); |
149 |
> |
t->Branch("vy",&jets_.vy,"vy/F"); |
150 |
> |
t->Branch("vz",&jets_.vz,"vz/F"); |
151 |
> |
} |
152 |
> |
if (useCentrality_) { |
153 |
> |
t->Branch("hf",&jets_.hf,"hf/F"); |
154 |
> |
t->Branch("bin",&jets_.bin,"bin/I"); |
155 |
> |
} |
156 |
> |
|
157 |
|
t->Branch("nref",&jets_.nref,"nref/I"); |
116 |
– |
t->Branch("bin",&jets_.bin,"bin/I"); |
158 |
|
t->Branch("rawpt",jets_.rawpt,"rawpt[nref]/F"); |
159 |
< |
t->Branch("jtpt",jets_.jtpt,"jtpt[nref]/F"); |
159 |
> |
if(!skipCorrections_) t->Branch("jtpt",jets_.jtpt,"jtpt[nref]/F"); |
160 |
|
t->Branch("jteta",jets_.jteta,"jteta[nref]/F"); |
161 |
|
t->Branch("jty",jets_.jty,"jty[nref]/F"); |
162 |
|
t->Branch("jtphi",jets_.jtphi,"jtphi[nref]/F"); |
163 |
+ |
t->Branch("jtpu",jets_.jtpu,"jtpu[nref]/F"); |
164 |
+ |
t->Branch("jtm",jets_.jtm,"jtm[nref]/F"); |
165 |
+ |
|
166 |
+ |
// jet ID information, jet composition |
167 |
+ |
t->Branch("discr_fr01", jets_.discr_fr01,"discr_fr01[nref]/F"); |
168 |
|
|
169 |
+ |
t->Branch("trackMax", jets_.trackMax,"trackMax[nref]/F"); |
170 |
+ |
t->Branch("trackSum", jets_.trackSum,"trackSum[nref]/F"); |
171 |
+ |
t->Branch("trackN", jets_.trackN,"trackN[nref]/I"); |
172 |
+ |
t->Branch("trackHardSum", jets_.trackHardSum,"trackHardSum[nref]/F"); |
173 |
+ |
t->Branch("trackHardN", jets_.trackHardN,"trackHardN[nref]/I"); |
174 |
+ |
|
175 |
+ |
t->Branch("chargedMax", jets_.chargedMax,"chargedMax[nref]/F"); |
176 |
+ |
t->Branch("chargedSum", jets_.chargedSum,"chargedSum[nref]/F"); |
177 |
+ |
t->Branch("chargedN", jets_.chargedN,"chargedN[nref]/I"); |
178 |
+ |
t->Branch("chargedHardSum", jets_.chargedHardSum,"chargedHardSum[nref]/F"); |
179 |
+ |
t->Branch("chargedHardN", jets_.chargedHardN,"chargedHardN[nref]/I"); |
180 |
+ |
|
181 |
+ |
t->Branch("photonMax", jets_.photonMax,"photonMax[nref]/F"); |
182 |
+ |
t->Branch("photonSum", jets_.photonSum,"photonSum[nref]/F"); |
183 |
+ |
t->Branch("photonN", jets_.photonN,"photonN[nref]/I"); |
184 |
+ |
t->Branch("photonHardSum", jets_.photonHardSum,"photonHardSum[nref]/F"); |
185 |
+ |
t->Branch("photonHardN", jets_.photonHardN,"photonHardN[nref]/I"); |
186 |
+ |
|
187 |
+ |
t->Branch("neutralMax", jets_.neutralMax,"neutralMax[nref]/F"); |
188 |
+ |
t->Branch("neutralSum", jets_.neutralSum,"neutralSum[nref]/F"); |
189 |
+ |
t->Branch("neutralN", jets_.neutralN,"neutralN[nref]/I"); |
190 |
+ |
|
191 |
+ |
t->Branch("hcalSum", jets_.hcalSum,"hcalSum[nref]/F"); |
192 |
+ |
t->Branch("ecalSum", jets_.ecalSum,"ecalSum[nref]/F"); |
193 |
+ |
|
194 |
+ |
t->Branch("eMax", jets_.eMax,"eMax[nref]/F"); |
195 |
+ |
t->Branch("eSum", jets_.eSum,"eSum[nref]/F"); |
196 |
+ |
t->Branch("eN", jets_.eN,"eN[nref]/I"); |
197 |
+ |
|
198 |
+ |
t->Branch("muMax", jets_.muMax,"muMax[nref]/F"); |
199 |
+ |
t->Branch("muSum", jets_.muSum,"muSum[nref]/F"); |
200 |
+ |
t->Branch("muN", jets_.muN,"muN[nref]/I"); |
201 |
+ |
|
202 |
+ |
if(!skipCorrections_) t->Branch("matchedPt", jets_.matchedPt,"matchedPt[nref]/F"); |
203 |
+ |
t->Branch("matchedRawPt", jets_.matchedRawPt,"matchedRawPt[nref]/F"); |
204 |
+ |
t->Branch("matchedR", jets_.matchedR,"matchedR[nref]/F"); |
205 |
+ |
|
206 |
+ |
// b-jet discriminators |
207 |
+ |
if (doLifeTimeTagging_) { |
208 |
+ |
|
209 |
+ |
t->Branch("discr_csvMva",jets_.discr_csvMva,"discr_csvMva[nref]/F"); |
210 |
+ |
t->Branch("discr_csvSimple",jets_.discr_csvSimple,"discr_csvSimple[nref]/F"); |
211 |
+ |
t->Branch("discr_muByIp3",jets_.discr_muByIp3,"discr_muByIp3[nref]/F"); |
212 |
+ |
t->Branch("discr_muByPt",jets_.discr_muByPt,"discr_muByPt[nref]/F"); |
213 |
+ |
t->Branch("discr_prob",jets_.discr_prob,"discr_prob[nref]/F"); |
214 |
+ |
t->Branch("discr_probb",jets_.discr_probb,"discr_probb[nref]/F"); |
215 |
+ |
t->Branch("discr_tcHighEff",jets_.discr_tcHighEff,"discr_tcHighEff[nref]/F"); |
216 |
+ |
t->Branch("discr_tcHighPur",jets_.discr_tcHighPur,"discr_tcHighPur[nref]/F"); |
217 |
+ |
|
218 |
+ |
t->Branch("nsvtx", jets_.nsvtx, "nsvtx[nref]/I"); |
219 |
+ |
t->Branch("svtxntrk", jets_.svtxntrk, "svtxntrk[nref]/I"); |
220 |
+ |
t->Branch("svtxdl", jets_.svtxdl, "svtxdl[nref]/F"); |
221 |
+ |
t->Branch("svtxdls", jets_.svtxdls, "svtxdls[nref]/F"); |
222 |
+ |
t->Branch("svtxm", jets_.svtxm, "svtxm[nref]/F"); |
223 |
+ |
t->Branch("svtxpt", jets_.svtxpt, "svtxpt[nref]/F"); |
224 |
+ |
|
225 |
+ |
t->Branch("nIPtrk",jets_.nIPtrk,"nIPtrk[nref]/I"); |
226 |
+ |
t->Branch("nselIPtrk",jets_.nselIPtrk,"nselIPtrk[nref]/I"); |
227 |
+ |
|
228 |
+ |
if (doLifeTimeTaggingExtras_) { |
229 |
+ |
t->Branch("nIP",&jets_.nIP,"nIP/I"); |
230 |
+ |
t->Branch("ipJetIndex",jets_.ipJetIndex,"ipJetIndex[nIP]/I"); |
231 |
+ |
t->Branch("ipPt",jets_.ipPt,"ipPt[nIP]/F"); |
232 |
+ |
t->Branch("ipProb0",jets_.ipProb0,"ipProb0[nIP]/F"); |
233 |
+ |
t->Branch("ipProb1",jets_.ipProb1,"ipProb1[nIP]/F"); |
234 |
+ |
t->Branch("ip2d",jets_.ip2d,"ip2d[nIP]/F"); |
235 |
+ |
t->Branch("ip2dSig",jets_.ip2dSig,"ip2dSig[nIP]/F"); |
236 |
+ |
t->Branch("ip3d",jets_.ip3d,"ip3d[nIP]/F"); |
237 |
+ |
t->Branch("ip3dSig",jets_.ip3dSig,"ip3dSig[nIP]/F"); |
238 |
+ |
t->Branch("ipDist2Jet",jets_.ipDist2Jet,"ipDist2Jet[nIP]/F"); |
239 |
+ |
t->Branch("ipDist2JetSig",jets_.ipDist2JetSig,"ipDist2JetSig[nIP]/F"); |
240 |
+ |
t->Branch("ipClosest2Jet",jets_.ipClosest2Jet,"ipClosest2Jet[nIP]/F"); |
241 |
+ |
|
242 |
+ |
} |
243 |
+ |
|
244 |
+ |
t->Branch("mue", jets_.mue, "mue[nref]/F"); |
245 |
+ |
t->Branch("mupt", jets_.mupt, "mupt[nref]/F"); |
246 |
+ |
t->Branch("mueta", jets_.mueta, "mueta[nref]/F"); |
247 |
+ |
t->Branch("muphi", jets_.muphi, "muphi[nref]/F"); |
248 |
+ |
t->Branch("mudr", jets_.mudr, "mudr[nref]/F"); |
249 |
+ |
t->Branch("muptrel", jets_.muptrel, "muptrel[nref]/F"); |
250 |
+ |
t->Branch("muchg", jets_.muchg, "muchg[nref]/I"); |
251 |
+ |
} |
252 |
+ |
|
253 |
+ |
|
254 |
|
if(isMC_){ |
255 |
+ |
t->Branch("beamId1",&jets_.beamId1,"beamId1/I"); |
256 |
+ |
t->Branch("beamId2",&jets_.beamId2,"beamId2/I"); |
257 |
+ |
|
258 |
|
t->Branch("pthat",&jets_.pthat,"pthat/F"); |
259 |
|
|
260 |
|
// Only matched gen jets |
266 |
|
t->Branch("refdrjt",jets_.refdrjt,"refdrjt[nref]/F"); |
267 |
|
// matched parton |
268 |
|
t->Branch("refparton_pt",jets_.refparton_pt,"refparton_pt[nref]/F"); |
269 |
< |
t->Branch("refparton_flavor",jets_.refparton_flavor,"refparton_flavor[nref]/F"); |
269 |
> |
t->Branch("refparton_flavor",jets_.refparton_flavor,"refparton_flavor[nref]/I"); |
270 |
> |
t->Branch("refparton_flavorForB",jets_.refparton_flavorForB,"refparton_flavorForB[nref]/I"); |
271 |
> |
|
272 |
> |
t->Branch("genChargedSum", jets_.genChargedSum,"genChargedSum[nref]/F"); |
273 |
> |
t->Branch("genHardSum", jets_.genHardSum,"genHardSum[nref]/F"); |
274 |
> |
t->Branch("signalChargedSum", jets_.signalChargedSum,"signalChargedSum[nref]/F"); |
275 |
> |
t->Branch("signalHardSum", jets_.signalHardSum,"signalHardSum[nref]/F"); |
276 |
> |
|
277 |
> |
if(doSubEvent_){ |
278 |
> |
t->Branch("subid",jets_.subid,"subid[nref]/I"); |
279 |
> |
} |
280 |
> |
|
281 |
> |
if(fillGenJets_){ |
282 |
> |
// For all gen jets, matched or unmatched |
283 |
> |
t->Branch("ngen",&jets_.ngen,"ngen/I"); |
284 |
> |
t->Branch("genmatchindex",jets_.genmatchindex,"genmatchindex[ngen]/I"); |
285 |
> |
t->Branch("genpt",jets_.genpt,"genpt[ngen]/F"); |
286 |
> |
t->Branch("geneta",jets_.geneta,"geneta[ngen]/F"); |
287 |
> |
t->Branch("geny",jets_.geny,"geny[ngen]/F"); |
288 |
> |
t->Branch("genphi",jets_.genphi,"genphi[ngen]/F"); |
289 |
> |
t->Branch("gendphijt",jets_.gendphijt,"gendphijt[ngen]/F"); |
290 |
> |
t->Branch("gendrjt",jets_.gendrjt,"gendrjt[ngen]/F"); |
291 |
> |
|
292 |
> |
if(doSubEvent_){ |
293 |
> |
t->Branch("gensubid",jets_.gensubid,"gensubid[ngen]/I"); |
294 |
> |
} |
295 |
> |
} |
296 |
> |
|
297 |
> |
if(saveBfragments_ ) { |
298 |
> |
t->Branch("bMult",&jets_.bMult,"bMult/I"); |
299 |
> |
t->Branch("bJetIndex",jets_.bJetIndex,"bJetIndex[bMult]/I"); |
300 |
> |
t->Branch("bStatus",jets_.bStatus,"bStatus[bMult]/I"); |
301 |
> |
t->Branch("bVx",jets_.bVx,"bVx[bMult]/F"); |
302 |
> |
t->Branch("bVy",jets_.bVy,"bVy[bMult]/F"); |
303 |
> |
t->Branch("bVz",jets_.bVz,"bVz[bMult]/F"); |
304 |
> |
t->Branch("bPt",jets_.bPt,"bPt[bMult]/F"); |
305 |
> |
t->Branch("bEta",jets_.bEta,"bEta[bMult]/F"); |
306 |
> |
t->Branch("bPhi",jets_.bPhi,"bPhi[bMult]/F"); |
307 |
> |
t->Branch("bPdg",jets_.bPdg,"bPdg[bMult]/I"); |
308 |
> |
t->Branch("bChg",jets_.bChg,"bChg[bMult]/I"); |
309 |
> |
} |
310 |
|
|
137 |
– |
// For all gen jets, matched or unmatched |
138 |
– |
t->Branch("ngen",&jets_.ngen,"ngen/I"); |
139 |
– |
t->Branch("genmatchindex",jets_.genmatchindex,"genmatchindex[ngen]/I"); |
140 |
– |
t->Branch("genpt",jets_.genpt,"genpt[ngen]/F"); |
141 |
– |
t->Branch("geneta",jets_.geneta,"geneta[ngen]/F"); |
142 |
– |
t->Branch("geny",jets_.geny,"geny[ngen]/F"); |
143 |
– |
t->Branch("genphi",jets_.genphi,"genphi[ngen]/F"); |
144 |
– |
t->Branch("gendphijt",jets_.gendphijt,"gendphijt[ngen]/F"); |
145 |
– |
t->Branch("gendrjt",jets_.gendrjt,"gendrjt[ngen]/F"); |
311 |
|
} |
312 |
< |
|
312 |
> |
/* |
313 |
|
if(!isMC_){ |
314 |
|
t->Branch("nL1TBit",&jets_.nL1TBit,"nL1TBit/I"); |
315 |
|
t->Branch("l1TBit",jets_.l1TBit,"l1TBit[nL1TBit]/O"); |
321 |
|
t->Branch("hltBit",jets_.hltBit,"hltBit[nHLTBit]/O"); |
322 |
|
|
323 |
|
} |
324 |
< |
|
324 |
> |
*/ |
325 |
|
TH1D::SetDefaultSumw2(); |
326 |
|
|
327 |
|
|
342 |
|
|
343 |
|
LogDebug("HiInclusiveJetAnalyzer")<<"START event: "<<event<<" in run "<<run<<endl; |
344 |
|
|
180 |
– |
|
345 |
|
int bin = -1; |
346 |
|
double hf = 0.; |
347 |
|
double b = 999.; |
348 |
|
|
349 |
+ |
if(!geo){ |
350 |
+ |
edm::ESHandle<CaloGeometry> pGeo; |
351 |
+ |
iSetup.get<CaloGeometryRecord>().get(pGeo); |
352 |
+ |
geo = pGeo.product(); |
353 |
+ |
} |
354 |
|
if(useCentrality_){ |
186 |
– |
//if(!isMC_){ |
355 |
|
if(!centrality_) centrality_ = new CentralityProvider(iSetup); |
356 |
|
centrality_->newEvent(iEvent,iSetup); // make sure you do this first in every event |
357 |
|
//double c = centrality_->centralityValue(); |
361 |
|
|
362 |
|
bin = centrality_->getBin(); |
363 |
|
b = centrality_->bMean(); |
196 |
– |
//} |
197 |
– |
/* |
198 |
– |
else{ |
199 |
– |
|
200 |
– |
TFile * centFile = new TFile("/net/hidsk0001/d00/scratch/mnguyen/CMSSW_3_9_1_patch1/src/macros/Hydjet_CentTable.root"); |
201 |
– |
|
202 |
– |
edm::Handle<reco::Centrality> cent; |
203 |
– |
iEvent.getByLabel(edm::InputTag("hiCentrality"),cent); |
204 |
– |
//cout<<" grabbed centrality "<<endl; |
205 |
– |
CentralityBins::RunMap cmap = getCentralityFromFile(centFile, "makeCentralityTableTFile", "HFhitsHydjet_2760GeV", 1, 1); |
206 |
– |
|
207 |
– |
// Still getting cent from hits. When tower tables become available, we need to switch |
208 |
– |
hf = cent->EtHFhitSum(); |
209 |
– |
//cout<<" hf "<<hf<<endl; |
210 |
– |
bin = cmap[run]->getBin(hf); |
211 |
– |
b = cmap[run]->bMeanOfBin(bin); |
212 |
– |
} |
213 |
– |
*/ |
364 |
|
} |
365 |
|
|
216 |
– |
|
217 |
– |
|
218 |
– |
|
219 |
– |
//double jetPtMin = 35.; |
220 |
– |
|
221 |
– |
|
366 |
|
// loop the events |
367 |
|
|
368 |
|
jets_.bin = bin; |
369 |
|
jets_.hf = hf; |
370 |
|
|
371 |
+ |
reco::Vertex::Point vtx(0,0,0); |
372 |
+ |
if (useVtx_) { |
373 |
+ |
edm::Handle<vector<reco::Vertex> >vertex; |
374 |
+ |
iEvent.getByLabel(vtxTag_, vertex); |
375 |
+ |
|
376 |
+ |
if(vertex->size()>0) { |
377 |
+ |
jets_.vx=vertex->begin()->x(); |
378 |
+ |
jets_.vy=vertex->begin()->y(); |
379 |
+ |
jets_.vz=vertex->begin()->z(); |
380 |
+ |
vtx = vertex->begin()->position(); |
381 |
+ |
} |
382 |
+ |
} |
383 |
+ |
|
384 |
+ |
edm::Handle<pat::JetCollection> patjets; |
385 |
+ |
if(usePat_)iEvent.getByLabel(jetTag_, patjets); |
386 |
|
|
387 |
< |
if (useVtx_) { |
388 |
< |
edm::Handle<vector<reco::Vertex> >vertex; |
230 |
< |
iEvent.getByLabel(vtxTag_, vertex); |
231 |
< |
|
232 |
< |
if(vertex->size()>0) { |
233 |
< |
jets_.vx=vertex->begin()->x(); |
234 |
< |
jets_.vy=vertex->begin()->y(); |
235 |
< |
jets_.vz=vertex->begin()->z(); |
236 |
< |
} |
237 |
< |
} |
387 |
> |
edm::Handle<pat::JetCollection> matchedjets; |
388 |
> |
iEvent.getByLabel(matchTag_, matchedjets); |
389 |
|
|
390 |
+ |
edm::Handle<reco::JetView> jets; |
391 |
+ |
iEvent.getByLabel(jetTag_, jets); |
392 |
|
|
393 |
+ |
edm::Handle<reco::PFCandidateCollection> pfCandidates; |
394 |
+ |
iEvent.getByLabel(pfCandidateLabel_,pfCandidates); |
395 |
+ |
|
396 |
+ |
edm::Handle<reco::TrackCollection> tracks; |
397 |
+ |
iEvent.getByLabel(trackTag_,tracks); |
398 |
|
|
399 |
< |
edm::Handle<pat::JetCollection> jets; |
400 |
< |
iEvent.getByLabel(jetTag_, jets); |
399 |
> |
edm::Handle<edm::SortedCollection<EcalRecHit,edm::StrictWeakOrdering<EcalRecHit> > > ebHits; |
400 |
> |
edm::Handle<edm::SortedCollection<EcalRecHit,edm::StrictWeakOrdering<EcalRecHit> > > eeHits; |
401 |
|
|
402 |
< |
|
403 |
< |
// FILL JRA TREE |
402 |
> |
edm::Handle<HFRecHitCollection> hfHits; |
403 |
> |
edm::Handle<HBHERecHitCollection> hbheHits; |
404 |
|
|
405 |
+ |
iEvent.getByLabel(HcalRecHitHBHESrc_,hbheHits); |
406 |
+ |
iEvent.getByLabel(HcalRecHitHFSrc_,hfHits); |
407 |
+ |
iEvent.getByLabel(EBSrc_,ebHits); |
408 |
+ |
iEvent.getByLabel(EESrc_,eeHits); |
409 |
+ |
|
410 |
+ |
edm::Handle<reco::GenParticleCollection> genparts; |
411 |
+ |
iEvent.getByLabel(genParticleSrc_,genparts); |
412 |
+ |
|
413 |
+ |
|
414 |
+ |
// FILL JRA TREE |
415 |
|
jets_.b = b; |
416 |
|
jets_.nref = 0; |
417 |
+ |
|
418 |
|
|
419 |
< |
if(!isMC_){ |
419 |
> |
if(doTrigger_){ |
420 |
|
fillL1Bits(iEvent); |
421 |
|
fillHLTBits(iEvent); |
422 |
|
} |
423 |
|
|
424 |
< |
for(unsigned int j = 0 ; j < jets->size(); ++j){ |
425 |
< |
const pat::Jet& jet = (*jets)[j]; |
424 |
> |
for(unsigned int j = 0; j < jets->size(); ++j){ |
425 |
> |
const reco::Jet& jet = (*jets)[j]; |
426 |
> |
if (useJEC_ && usePat_){ |
427 |
> |
jets_.rawpt[jets_.nref]=(*patjets)[j].correctedJet("Uncorrected").pt(); |
428 |
> |
} |
429 |
|
|
430 |
< |
//cout<<" jet pt "<<jet.pt()<<endl; |
431 |
< |
//if(jet.pt() < jetPtMin) continue; |
432 |
< |
if (useJEC_) jets_.rawpt[jets_.nref]=jet.correctedJet("Uncorrected").pt(); |
430 |
> |
if(doLifeTimeTagging_){ |
431 |
> |
|
432 |
> |
if(jetTag_.label()=="icPu5patJets"){ |
433 |
> |
jets_.discr_csvMva[jets_.nref] = (*patjets)[j].bDiscriminator("icPu5CombinedSecondaryVertexMVABJetTags"); |
434 |
> |
jets_.discr_csvSimple[jets_.nref] = (*patjets)[j].bDiscriminator("icPu5CombinedSecondaryVertexBJetTags"); |
435 |
> |
jets_.discr_muByIp3[jets_.nref] = (*patjets)[j].bDiscriminator("icPu5SoftMuonByIP3dBJetTags"); |
436 |
> |
jets_.discr_muByPt[jets_.nref] = (*patjets)[j].bDiscriminator("icPu5SoftMuonByPtBJetTags"); |
437 |
> |
jets_.discr_prob[jets_.nref] = (*patjets)[j].bDiscriminator("icPu5JetBProbabilityBJetTags"); |
438 |
> |
jets_.discr_probb[jets_.nref] = (*patjets)[j].bDiscriminator("icPu5JetProbabilityBJetTags"); |
439 |
> |
jets_.discr_tcHighEff[jets_.nref] = (*patjets)[j].bDiscriminator("icPu5TrackCountingHighEffBJetTags"); |
440 |
> |
jets_.discr_tcHighPur[jets_.nref] = (*patjets)[j].bDiscriminator("icPu5TrackCountingHighPurBJetTags"); |
441 |
> |
} |
442 |
> |
else if(jetTag_.label()=="akPu3PFpatJets"){ |
443 |
> |
jets_.discr_csvMva[jets_.nref] = (*patjets)[j].bDiscriminator("akPu3PFCombinedSecondaryVertexMVABJetTags"); |
444 |
> |
jets_.discr_csvSimple[jets_.nref] = (*patjets)[j].bDiscriminator("akPu3PFCombinedSecondaryVertexBJetTags"); |
445 |
> |
jets_.discr_muByIp3[jets_.nref] = (*patjets)[j].bDiscriminator("akPu3PFSoftMuonByIP3dBJetTags"); |
446 |
> |
jets_.discr_muByPt[jets_.nref] = (*patjets)[j].bDiscriminator("akPu3PFSoftMuonByPtBJetTags"); |
447 |
> |
jets_.discr_prob[jets_.nref] = (*patjets)[j].bDiscriminator("akPu3PFJetBProbabilityBJetTags"); |
448 |
> |
jets_.discr_probb[jets_.nref] = (*patjets)[j].bDiscriminator("akPu3PFJetProbabilityBJetTags"); |
449 |
> |
jets_.discr_tcHighEff[jets_.nref] = (*patjets)[j].bDiscriminator("akPu3PFTrackCountingHighEffBJetTags"); |
450 |
> |
jets_.discr_tcHighPur[jets_.nref] = (*patjets)[j].bDiscriminator("akPu3PFTrackCountingHighPurBJetTags"); |
451 |
> |
} |
452 |
> |
else{ |
453 |
> |
// cout<<" b-tagging variables not filled for this collection, turn of doLifeTimeTagging "<<endl; |
454 |
> |
} |
455 |
> |
|
456 |
> |
const reco::SecondaryVertexTagInfo& tagInfoSV=*(*patjets)[j].tagInfoSecondaryVertex(); |
457 |
> |
|
458 |
> |
jets_.nsvtx[jets_.nref] = tagInfoSV.nVertices(); |
459 |
> |
|
460 |
> |
if (tagInfoSV.nVertices()>0) { |
461 |
> |
jets_.svtxntrk[jets_.nref] = tagInfoSV.nVertexTracks(0); |
462 |
> |
// this is the 3d flight distance, for 2-D use (0,true) |
463 |
> |
Measurement1D m1D = tagInfoSV.flightDistance(0); |
464 |
> |
jets_.svtxdl[jets_.nref] = m1D.value(); |
465 |
> |
jets_.svtxdls[jets_.nref] = m1D.significance(); |
466 |
> |
|
467 |
> |
const reco::Vertex& svtx = tagInfoSV.secondaryVertex(0); |
468 |
> |
//cout<<" SV: vx: "<<svtx.x()<<" vy "<<svtx.y()<<" vz "<<svtx.z()<<endl; |
469 |
> |
//cout<<" PV: vx: "<<jet.vx()<<" vy "<<jet.vy()<<" vz "<<jet.vz()<<endl; |
470 |
> |
jets_.svtxm[jets_.nref] = svtx.p4().mass(); |
471 |
> |
jets_.svtxpt[jets_.nref] = svtx.p4().pt(); |
472 |
> |
//cout<<" chi2 "<<svtx.chi2()<<" ndof "<<svtx.ndof()<<endl; |
473 |
> |
} |
474 |
> |
|
475 |
> |
const reco::TrackIPTagInfo& tagInfoIP=*(*patjets)[j].tagInfoTrackIP(); |
476 |
> |
|
477 |
> |
jets_.nIPtrk[jets_.nref] = tagInfoIP.tracks().size(); |
478 |
> |
jets_.nselIPtrk[jets_.nref] = tagInfoIP.selectedTracks().size(); |
479 |
> |
|
480 |
> |
if (doLifeTimeTaggingExtras_) { |
481 |
> |
|
482 |
> |
TrackRefVector selTracks=tagInfoIP.selectedTracks(); |
483 |
> |
|
484 |
> |
GlobalPoint pv(tagInfoIP.primaryVertex()->position().x(),tagInfoIP.primaryVertex()->position().y(),tagInfoIP.primaryVertex()->position().z()); |
485 |
> |
|
486 |
> |
for(int it=0;it<jets_.nselIPtrk[jets_.nref] ;it++) |
487 |
> |
{ |
488 |
> |
jets_.ipJetIndex[jets_.nIP + it]= jets_.nref; |
489 |
> |
TrackIPTagInfo::TrackIPData data = tagInfoIP.impactParameterData()[it]; |
490 |
> |
jets_.ipPt[jets_.nIP + it] = selTracks[it]->pt(); |
491 |
> |
jets_.ipProb0[jets_.nIP + it] = tagInfoIP.probabilities(0)[it]; |
492 |
> |
jets_.ipProb1[jets_.nIP + it] = tagInfoIP.probabilities(1)[it]; |
493 |
> |
jets_.ip2d[jets_.nIP + it] = data.ip2d.value(); |
494 |
> |
jets_.ip2dSig[jets_.nIP + it] = data.ip2d.significance(); |
495 |
> |
jets_.ip3d[jets_.nIP + it] = data.ip3d.value(); |
496 |
> |
jets_.ip3dSig[jets_.nIP + it] = data.ip3d.significance(); |
497 |
> |
jets_.ipDist2Jet[jets_.nIP + it] = data.distanceToJetAxis.value(); |
498 |
> |
jets_.ipDist2JetSig[jets_.nIP + it] = data.distanceToJetAxis.significance(); |
499 |
> |
jets_.ipClosest2Jet[jets_.nIP + it] = (data.closestToJetAxis - pv).mag(); //decay length |
500 |
> |
} |
501 |
> |
|
502 |
> |
jets_.nIP += jets_.nselIPtrk[jets_.nref]; |
503 |
> |
|
504 |
> |
} |
505 |
> |
|
506 |
> |
const reco::PFCandidateCollection *pfCandidateColl = &(*pfCandidates); |
507 |
> |
int pfMuonIndex = getPFJetMuon(jet, pfCandidateColl); |
508 |
> |
if(pfMuonIndex >=0){ |
509 |
> |
const reco::PFCandidate muon = pfCandidateColl->at(pfMuonIndex); |
510 |
> |
jets_.mupt[jets_.nref] = muon.pt(); |
511 |
> |
jets_.mueta[jets_.nref] = muon.eta(); |
512 |
> |
jets_.muphi[jets_.nref] = muon.phi(); |
513 |
> |
jets_.mue[jets_.nref] = muon.energy(); |
514 |
> |
jets_.mudr[jets_.nref] = reco::deltaR(jet,muon); |
515 |
> |
jets_.muptrel[jets_.nref] = getPtRel(muon, jet); |
516 |
> |
jets_.muchg[jets_.nref] = muon.charge(); |
517 |
> |
}else{ |
518 |
> |
jets_.mupt[jets_.nref] = 0.0; |
519 |
> |
jets_.mueta[jets_.nref] = 0.0; |
520 |
> |
jets_.muphi[jets_.nref] = 0.0; |
521 |
> |
jets_.mue[jets_.nref] = 0.0; |
522 |
> |
jets_.mudr[jets_.nref] = 9.9; |
523 |
> |
jets_.muptrel[jets_.nref] = 0.0; |
524 |
> |
jets_.muchg[jets_.nref] = 0; |
525 |
> |
} |
526 |
> |
|
527 |
> |
} |
528 |
> |
|
529 |
> |
// Jet ID variables |
530 |
> |
|
531 |
> |
jets_.muMax[jets_.nref] = 0; |
532 |
> |
jets_.muSum[jets_.nref] = 0; |
533 |
> |
jets_.muN[jets_.nref] = 0; |
534 |
> |
|
535 |
> |
jets_.eMax[jets_.nref] = 0; |
536 |
> |
jets_.eSum[jets_.nref] = 0; |
537 |
> |
jets_.eN[jets_.nref] = 0; |
538 |
> |
|
539 |
> |
jets_.neutralMax[jets_.nref] = 0; |
540 |
> |
jets_.neutralSum[jets_.nref] = 0; |
541 |
> |
jets_.neutralN[jets_.nref] = 0; |
542 |
> |
|
543 |
> |
jets_.photonMax[jets_.nref] = 0; |
544 |
> |
jets_.photonSum[jets_.nref] = 0; |
545 |
> |
jets_.photonN[jets_.nref] = 0; |
546 |
> |
jets_.photonHardSum[jets_.nref] = 0; |
547 |
> |
jets_.photonHardN[jets_.nref] = 0; |
548 |
> |
|
549 |
> |
jets_.chargedMax[jets_.nref] = 0; |
550 |
> |
jets_.chargedSum[jets_.nref] = 0; |
551 |
> |
jets_.chargedN[jets_.nref] = 0; |
552 |
> |
jets_.chargedHardSum[jets_.nref] = 0; |
553 |
> |
jets_.chargedHardN[jets_.nref] = 0; |
554 |
> |
|
555 |
> |
jets_.trackMax[jets_.nref] = 0; |
556 |
> |
jets_.trackSum[jets_.nref] = 0; |
557 |
> |
jets_.trackN[jets_.nref] = 0; |
558 |
> |
jets_.trackHardSum[jets_.nref] = 0; |
559 |
> |
jets_.trackHardN[jets_.nref] = 0; |
560 |
> |
|
561 |
> |
jets_.hcalSum[jets_.nref] = 0; |
562 |
> |
jets_.ecalSum[jets_.nref] = 0; |
563 |
> |
|
564 |
> |
jets_.genChargedSum[jets_.nref] = 0; |
565 |
> |
jets_.genHardSum[jets_.nref] = 0; |
566 |
> |
|
567 |
> |
jets_.signalChargedSum[jets_.nref] = 0; |
568 |
> |
jets_.signalHardSum[jets_.nref] = 0; |
569 |
> |
|
570 |
> |
jets_.subid[jets_.nref] = -1; |
571 |
> |
|
572 |
> |
for(unsigned int icand = 0; icand < tracks->size(); ++icand){ |
573 |
> |
const reco::Track& track = (*tracks)[icand]; |
574 |
> |
if(useQuality_ ){ |
575 |
> |
bool goodtrack = track.quality(reco::TrackBase::qualityByName(trackQuality_)); |
576 |
> |
if(!goodtrack) continue; |
577 |
> |
} |
578 |
> |
|
579 |
> |
double dr = deltaR(jet,track); |
580 |
> |
if(dr < rParam){ |
581 |
> |
double ptcand = track.pt(); |
582 |
> |
jets_.trackSum[jets_.nref] += ptcand; |
583 |
> |
jets_.trackN[jets_.nref] += 1; |
584 |
> |
|
585 |
> |
if(ptcand > hardPtMin_){ |
586 |
> |
jets_.trackHardSum[jets_.nref] += ptcand; |
587 |
> |
jets_.trackHardN[jets_.nref] += 1; |
588 |
> |
|
589 |
> |
} |
590 |
> |
if(ptcand > jets_.trackMax[jets_.nref]) jets_.trackMax[jets_.nref] = ptcand; |
591 |
> |
|
592 |
> |
} |
593 |
> |
} |
594 |
> |
|
595 |
> |
for(unsigned int icand = 0; icand < pfCandidates->size(); ++icand){ |
596 |
> |
const reco::PFCandidate& track = (*pfCandidates)[icand]; |
597 |
> |
double dr = deltaR(jet,track); |
598 |
> |
if(dr < rParam){ |
599 |
> |
double ptcand = track.pt(); |
600 |
> |
int pfid = track.particleId(); |
601 |
> |
|
602 |
> |
switch(pfid){ |
603 |
> |
|
604 |
> |
case 1: |
605 |
> |
jets_.chargedSum[jets_.nref] += ptcand; |
606 |
> |
jets_.chargedN[jets_.nref] += 1; |
607 |
> |
if(ptcand > hardPtMin_){ |
608 |
> |
jets_.chargedHardSum[jets_.nref] += ptcand; |
609 |
> |
jets_.chargedHardN[jets_.nref] += 1; |
610 |
> |
} |
611 |
> |
if(ptcand > jets_.chargedMax[jets_.nref]) jets_.chargedMax[jets_.nref] = ptcand; |
612 |
> |
break; |
613 |
> |
|
614 |
> |
case 2: |
615 |
> |
jets_.eSum[jets_.nref] += ptcand; |
616 |
> |
jets_.eN[jets_.nref] += 1; |
617 |
> |
if(ptcand > jets_.eMax[jets_.nref]) jets_.eMax[jets_.nref] = ptcand; |
618 |
> |
break; |
619 |
> |
|
620 |
> |
case 3: |
621 |
> |
jets_.muSum[jets_.nref] += ptcand; |
622 |
> |
jets_.muN[jets_.nref] += 1; |
623 |
> |
if(ptcand > jets_.muMax[jets_.nref]) jets_.muMax[jets_.nref] = ptcand; |
624 |
> |
break; |
625 |
> |
|
626 |
> |
case 4: |
627 |
> |
jets_.photonSum[jets_.nref] += ptcand; |
628 |
> |
jets_.photonN[jets_.nref] += 1; |
629 |
> |
if(ptcand > hardPtMin_){ |
630 |
> |
jets_.photonHardSum[jets_.nref] += ptcand; |
631 |
> |
jets_.photonHardN[jets_.nref] += 1; |
632 |
> |
} |
633 |
> |
if(ptcand > jets_.photonMax[jets_.nref]) jets_.photonMax[jets_.nref] = ptcand; |
634 |
> |
break; |
635 |
> |
|
636 |
> |
case 5: |
637 |
> |
jets_.neutralSum[jets_.nref] += ptcand; |
638 |
> |
jets_.neutralN[jets_.nref] += 1; |
639 |
> |
if(ptcand > jets_.neutralMax[jets_.nref]) jets_.neutralMax[jets_.nref] = ptcand; |
640 |
> |
break; |
641 |
> |
|
642 |
> |
default: |
643 |
> |
break; |
644 |
> |
|
645 |
> |
} |
646 |
> |
} |
647 |
> |
} |
648 |
> |
|
649 |
> |
// Calorimeter fractions |
650 |
> |
|
651 |
> |
for(unsigned int i = 0; i < hbheHits->size(); ++i){ |
652 |
> |
const HBHERecHit & hit= (*hbheHits)[i]; |
653 |
> |
math::XYZPoint pos = getPosition(hit.id(),vtx); |
654 |
> |
double dr = deltaR(jet.eta(),jet.phi(),pos.eta(),pos.phi()); |
655 |
> |
if(dr < rParam){ |
656 |
> |
jets_.hcalSum[jets_.nref] += getEt(pos,hit.energy()); |
657 |
> |
} |
658 |
> |
} |
659 |
> |
|
660 |
> |
for(unsigned int i = 0; i < hfHits->size(); ++i){ |
661 |
> |
const HFRecHit & hit= (*hfHits)[i]; |
662 |
> |
math::XYZPoint pos = getPosition(hit.id(),vtx); |
663 |
> |
double dr = deltaR(jet.eta(),jet.phi(),pos.eta(),pos.phi()); |
664 |
> |
if(dr < rParam){ |
665 |
> |
jets_.hcalSum[jets_.nref] += getEt(pos,hit.energy()); |
666 |
> |
} |
667 |
> |
} |
668 |
> |
|
669 |
> |
|
670 |
> |
for(unsigned int i = 0; i < ebHits->size(); ++i){ |
671 |
> |
const EcalRecHit & hit= (*ebHits)[i]; |
672 |
> |
math::XYZPoint pos = getPosition(hit.id(),vtx); |
673 |
> |
double dr = deltaR(jet.eta(),jet.phi(),pos.eta(),pos.phi()); |
674 |
> |
if(dr < rParam){ |
675 |
> |
jets_.ecalSum[jets_.nref] += getEt(pos,hit.energy()); |
676 |
> |
} |
677 |
> |
} |
678 |
> |
|
679 |
> |
for(unsigned int i = 0; i < eeHits->size(); ++i){ |
680 |
> |
const EcalRecHit & hit= (*eeHits)[i]; |
681 |
> |
math::XYZPoint pos = getPosition(hit.id(),vtx); |
682 |
> |
double dr = deltaR(jet.eta(),jet.phi(),pos.eta(),pos.phi()); |
683 |
> |
if(dr < rParam){ |
684 |
> |
jets_.ecalSum[jets_.nref] += getEt(pos,hit.energy()); |
685 |
> |
} |
686 |
> |
} |
687 |
> |
|
688 |
> |
// Alternative reconstruction matching (PF for calo, calo for PF) |
689 |
> |
|
690 |
> |
double drMin = 100; |
691 |
> |
for(unsigned int imatch = 0 ; imatch < matchedjets->size(); ++imatch){ |
692 |
> |
const pat::Jet& mjet = (*matchedjets)[imatch]; |
693 |
> |
|
694 |
> |
double dr = deltaR(jet,mjet); |
695 |
> |
if(dr < drMin){ |
696 |
> |
jets_.matchedPt[jets_.nref] = mjet.pt(); |
697 |
> |
jets_.matchedRawPt[jets_.nref] = mjet.correctedJet("Uncorrected").pt(); |
698 |
> |
jets_.matchedR[jets_.nref] = dr; |
699 |
> |
drMin = dr; |
700 |
> |
} |
701 |
> |
} |
702 |
> |
|
703 |
> |
// if(etrk.quality(reco::TrackBase::qualityByName(qualityString_))) pev_.trkQual[pev_.nTrk]=1; |
704 |
> |
|
705 |
> |
///////////////////////////////////////////////////////////////// |
706 |
> |
// Jet core pt^2 discriminant for fake jets |
707 |
> |
// Edited by Yue Shi Lai <ylai@mit.edu> |
708 |
> |
|
709 |
> |
// Initial value is 0 |
710 |
> |
jets_.discr_fr01[jets_.nref] = 0; |
711 |
> |
// Start with no directional adaption, i.e. the fake rejection |
712 |
> |
// axis is the jet axis |
713 |
> |
float pseudorapidity_adapt = jets_.jteta[jets_.nref]; |
714 |
> |
float azimuth_adapt = jets_.jtphi[jets_.nref]; |
715 |
> |
|
716 |
> |
// Unadapted discriminant with adaption search |
717 |
> |
for (size_t iteration = 0; iteration < 2; iteration++) { |
718 |
> |
float pseudorapidity_adapt_new = pseudorapidity_adapt; |
719 |
> |
float azimuth_adapt_new = azimuth_adapt; |
720 |
> |
float max_weighted_perp = 0; |
721 |
> |
float perp_square_sum = 0; |
722 |
> |
|
723 |
> |
for (size_t index_pf_candidate = 0; |
724 |
> |
index_pf_candidate < pfCandidates->size(); |
725 |
> |
index_pf_candidate++) { |
726 |
> |
const reco::PFCandidate &p = |
727 |
> |
(*pfCandidates)[index_pf_candidate]; |
728 |
> |
|
729 |
> |
switch (p.particleId()) { |
730 |
> |
//case 1: // Charged hadron |
731 |
> |
//case 3: // Muon |
732 |
> |
case 4: // Photon |
733 |
> |
{ |
734 |
> |
const float dpseudorapidity = |
735 |
> |
p.eta() - pseudorapidity_adapt; |
736 |
> |
const float dazimuth = |
737 |
> |
reco::deltaPhi(p.phi(), azimuth_adapt); |
738 |
> |
// The Gaussian scale factor is 0.5 / (0.1 * 0.1) |
739 |
> |
// = 50 |
740 |
> |
const float angular_weight = |
741 |
> |
exp(-50.0F * (dpseudorapidity * dpseudorapidity + |
742 |
> |
dazimuth * dazimuth)); |
743 |
> |
const float weighted_perp = |
744 |
> |
angular_weight * p.pt() * p.pt(); |
745 |
> |
const float weighted_perp_square = |
746 |
> |
weighted_perp * p.pt(); |
747 |
> |
|
748 |
> |
perp_square_sum += weighted_perp_square; |
749 |
> |
if (weighted_perp >= max_weighted_perp) { |
750 |
> |
pseudorapidity_adapt_new = p.eta(); |
751 |
> |
azimuth_adapt_new = p.phi(); |
752 |
> |
max_weighted_perp = weighted_perp; |
753 |
> |
} |
754 |
> |
} |
755 |
> |
default: |
756 |
> |
break; |
757 |
> |
} |
758 |
> |
} |
759 |
> |
// Update the fake rejection value |
760 |
> |
jets_.discr_fr01[jets_.nref] = std::max( |
761 |
> |
jets_.discr_fr01[jets_.nref], perp_square_sum); |
762 |
> |
// Update the directional adaption |
763 |
> |
pseudorapidity_adapt = pseudorapidity_adapt_new; |
764 |
> |
azimuth_adapt = azimuth_adapt_new; |
765 |
> |
} |
766 |
> |
|
767 |
|
jets_.jtpt[jets_.nref] = jet.pt(); |
768 |
|
jets_.jteta[jets_.nref] = jet.eta(); |
769 |
|
jets_.jtphi[jets_.nref] = jet.phi(); |
770 |
|
jets_.jty[jets_.nref] = jet.eta(); |
771 |
< |
|
771 |
> |
jets_.jtpu[jets_.nref] = jet.pileup(); |
772 |
> |
jets_.jtm[jets_.nref] = jet.mass(); |
773 |
|
|
774 |
< |
if(jet.genJet()){ |
775 |
< |
jets_.refpt[jets_.nref] = jet.genJet()->pt(); |
776 |
< |
jets_.refeta[jets_.nref] = jet.genJet()->eta(); |
777 |
< |
jets_.refphi[jets_.nref] = jet.genJet()->phi(); |
778 |
< |
jets_.refy[jets_.nref] = jet.genJet()->eta(); |
779 |
< |
jets_.refdphijt[jets_.nref] = reco::deltaPhi(jet.phi(),jet.genJet()->phi()); |
780 |
< |
jets_.refdrjt[jets_.nref] = reco::deltaR(jet.eta(),jet.phi(),jet.genJet()->eta(),jet.genJet()->phi()); |
781 |
< |
} |
782 |
< |
else{ |
783 |
< |
jets_.refpt[jets_.nref] = -999.; |
784 |
< |
jets_.refeta[jets_.nref] = -999.; |
785 |
< |
jets_.refphi[jets_.nref] = -999.; |
786 |
< |
jets_.refy[jets_.nref] = -999.; |
787 |
< |
jets_.refdphijt[jets_.nref] = -999.; |
788 |
< |
jets_.refdrjt[jets_.nref] = -999.; |
789 |
< |
} |
790 |
< |
|
791 |
< |
// matched partons |
792 |
< |
if (jet.genParton()) { |
793 |
< |
jets_.refparton_pt[jets_.nref] = jet.genParton()->pt(); |
794 |
< |
jets_.refparton_flavor[jets_.nref] = jet.genParton()->pdgId(); |
795 |
< |
} else { |
796 |
< |
jets_.refparton_pt[jets_.nref] = -999; |
797 |
< |
jets_.refparton_flavor[jets_.nref] = -999; |
798 |
< |
} |
774 |
> |
if(isMC_ && usePat_){ |
775 |
> |
|
776 |
> |
for(UInt_t i = 0; i < genparts->size(); ++i){ |
777 |
> |
const reco::GenParticle& p = (*genparts)[i]; |
778 |
> |
if (p.status()!=1) continue; |
779 |
> |
if (p.charge()==0) continue; |
780 |
> |
double dr = deltaR(jet,p); |
781 |
> |
if(dr < rParam){ |
782 |
> |
double ppt = p.pt(); |
783 |
> |
jets_.genChargedSum[jets_.nref] += ppt; |
784 |
> |
if(ppt > hardPtMin_) jets_.genHardSum[jets_.nref] += ppt; |
785 |
> |
if(p.collisionId() == 0){ |
786 |
> |
jets_.signalChargedSum[jets_.nref] += ppt; |
787 |
> |
if(ppt > hardPtMin_) jets_.signalHardSum[jets_.nref] += ppt; |
788 |
> |
} |
789 |
> |
|
790 |
> |
} |
791 |
> |
} |
792 |
> |
|
793 |
> |
const reco::GenJet * genjet = (*patjets)[j].genJet(); |
794 |
> |
|
795 |
> |
if(genjet){ |
796 |
> |
jets_.refpt[jets_.nref] = genjet->pt(); |
797 |
> |
jets_.refeta[jets_.nref] = genjet->eta(); |
798 |
> |
jets_.refphi[jets_.nref] = genjet->phi(); |
799 |
> |
jets_.refy[jets_.nref] = genjet->eta(); |
800 |
> |
jets_.refdphijt[jets_.nref] = reco::deltaPhi(jet.phi(), genjet->phi()); |
801 |
> |
jets_.refdrjt[jets_.nref] = reco::deltaR(jet.eta(),jet.phi(),genjet->eta(),genjet->phi()); |
802 |
> |
|
803 |
> |
if(doSubEvent_){ |
804 |
> |
const GenParticle* gencon = genjet->getGenConstituent(0); |
805 |
> |
jets_.subid[jets_.nref] = gencon->collisionId(); |
806 |
> |
} |
807 |
> |
|
808 |
> |
}else{ |
809 |
> |
jets_.refpt[jets_.nref] = -999.; |
810 |
> |
jets_.refeta[jets_.nref] = -999.; |
811 |
> |
jets_.refphi[jets_.nref] = -999.; |
812 |
> |
jets_.refy[jets_.nref] = -999.; |
813 |
> |
jets_.refdphijt[jets_.nref] = -999.; |
814 |
> |
jets_.refdrjt[jets_.nref] = -999.; |
815 |
> |
} |
816 |
> |
|
817 |
> |
jets_.refparton_flavorForB[jets_.nref] = (*patjets)[j].partonFlavour(); |
818 |
> |
|
819 |
> |
// matched partons |
820 |
> |
const reco::GenParticle & parton = *(*patjets)[j].genParton(); |
821 |
> |
|
822 |
> |
if((*patjets)[j].genParton()){ |
823 |
> |
jets_.refparton_pt[jets_.nref] = parton.pt(); |
824 |
> |
jets_.refparton_flavor[jets_.nref] = parton.pdgId(); |
825 |
> |
|
826 |
> |
if(saveBfragments_ && abs(jets_.refparton_flavorForB[jets_.nref])==5){ |
827 |
> |
|
828 |
> |
usedStringPts.clear(); |
829 |
> |
|
830 |
> |
// uncomment this if you want to know the ugly truth about parton matching -matt |
831 |
> |
//if(jet.pt() > 50 &&abs(parton.pdgId())!=5 && parton.pdgId()!=21) |
832 |
> |
// cout<<" Identified as a b, but doesn't match b or gluon, id = "<<parton.pdgId()<<endl; |
833 |
> |
|
834 |
> |
jets_.bJetIndex[jets_.bMult] = jets_.nref; |
835 |
> |
jets_.bStatus[jets_.bMult] = parton.status(); |
836 |
> |
jets_.bVx[jets_.bMult] = parton.vx(); |
837 |
> |
jets_.bVy[jets_.bMult] = parton.vy(); |
838 |
> |
jets_.bVz[jets_.bMult] = parton.vz(); |
839 |
> |
jets_.bPt[jets_.bMult] = parton.pt(); |
840 |
> |
jets_.bEta[jets_.bMult] = parton.eta(); |
841 |
> |
jets_.bPhi[jets_.bMult] = parton.phi(); |
842 |
> |
jets_.bPdg[jets_.bMult] = parton.pdgId(); |
843 |
> |
jets_.bChg[jets_.bMult] = parton.charge(); |
844 |
> |
jets_.bMult++; |
845 |
> |
saveDaughters(parton); |
846 |
> |
} |
847 |
> |
|
848 |
> |
|
849 |
> |
} else { |
850 |
> |
jets_.refparton_pt[jets_.nref] = -999; |
851 |
> |
jets_.refparton_flavor[jets_.nref] = -999; |
852 |
> |
} |
853 |
> |
|
854 |
|
|
855 |
+ |
} |
856 |
|
|
857 |
|
jets_.nref++; |
858 |
|
|
862 |
|
|
863 |
|
if(isMC_){ |
864 |
|
|
865 |
+ |
edm::Handle<HepMCProduct> hepMCProduct; |
866 |
+ |
iEvent.getByLabel(eventInfoTag_,hepMCProduct); |
867 |
+ |
const HepMC::GenEvent* MCEvt = hepMCProduct->GetEvent(); |
868 |
+ |
|
869 |
+ |
std::pair<HepMC::GenParticle*,HepMC::GenParticle*> beamParticles = MCEvt->beam_particles(); |
870 |
+ |
if(beamParticles.first != 0)jets_.beamId1 = beamParticles.first->pdg_id(); |
871 |
+ |
if(beamParticles.second != 0)jets_.beamId2 = beamParticles.second->pdg_id(); |
872 |
+ |
|
873 |
|
edm::Handle<GenEventInfoProduct> hEventInfo; |
874 |
|
iEvent.getByLabel(eventInfoTag_,hEventInfo); |
875 |
|
//jets_.pthat = hEventInfo->binningValues()[0]; |
887 |
|
float genjet_pt = genjet.pt(); |
888 |
|
|
889 |
|
// threshold to reduce size of output in minbias PbPb |
890 |
< |
if(genjet_pt>20.){ |
890 |
> |
if(genjet_pt>genPtMin_){ |
891 |
|
|
892 |
|
jets_.genpt[jets_.ngen] = genjet_pt; |
893 |
|
jets_.geneta[jets_.ngen] = genjet.eta(); |
894 |
|
jets_.genphi[jets_.ngen] = genjet.phi(); |
895 |
|
jets_.geny[jets_.ngen] = genjet.eta(); |
896 |
|
|
897 |
+ |
if(doSubEvent_){ |
898 |
+ |
const GenParticle* gencon = genjet.getGenConstituent(0); |
899 |
+ |
jets_.gensubid[jets_.ngen] = gencon->collisionId(); |
900 |
+ |
} |
901 |
|
|
902 |
|
// find matching patJet if there is one |
903 |
|
|
904 |
|
jets_.gendrjt[jets_.ngen] = -1.0; |
905 |
|
jets_.genmatchindex[jets_.ngen] = -1; |
906 |
|
|
332 |
– |
|
907 |
|
for(unsigned int ijet = 0 ; ijet < jets->size(); ++ijet){ |
908 |
|
const pat::Jet& jet = (*jets)[ijet]; |
909 |
< |
|
910 |
< |
if(jet.genJet()){ |
911 |
< |
if(fabs(genjet.pt()-jet.genJet()->pt())<0.0001 && |
912 |
< |
fabs(genjet.eta()-jet.genJet()->eta())<0.0001 && |
913 |
< |
(fabs(genjet.phi()-jet.genJet()->phi())<0.0001 || fabs(genjet.phi()-jet.genJet()->phi() - 2.0*TMath::Pi()) < 0.0001 )){ |
909 |
> |
const reco::GenJet* matchedGenJet = (*patjets)[ijet].genJet(); |
910 |
> |
if(matchedGenJet){ |
911 |
> |
// poor man's matching, someone fix please |
912 |
> |
if(fabs(genjet.pt()-matchedGenJet->pt())<0.0001 && |
913 |
> |
fabs(genjet.eta()-matchedGenJet->eta())<0.0001 && |
914 |
> |
(fabs(genjet.phi()-matchedGenJet->phi())<0.0001 || fabs(genjet.phi()-matchedGenJet->phi() - 2.0*TMath::Pi()) < 0.0001 )){ |
915 |
|
|
916 |
|
jets_.genmatchindex[jets_.ngen] = (int)ijet; |
917 |
|
jets_.gendphijt[jets_.ngen] = reco::deltaPhi(jet.phi(),genjet.phi()); |
927 |
|
} |
928 |
|
|
929 |
|
} |
930 |
+ |
|
931 |
|
t->Fill(); |
932 |
+ |
memset(&jets_,0,sizeof jets_); |
933 |
+ |
|
934 |
|
} |
935 |
|
|
936 |
|
|
1030 |
|
} |
1031 |
|
|
1032 |
|
|
1033 |
+ |
int |
1034 |
+ |
HiInclusiveJetAnalyzer::getPFJetMuon(const pat::Jet& pfJet, const reco::PFCandidateCollection *pfCandidateColl) |
1035 |
+ |
{ |
1036 |
+ |
|
1037 |
+ |
int pfMuonIndex = -1; |
1038 |
+ |
float ptMax = 0.; |
1039 |
+ |
|
1040 |
+ |
|
1041 |
+ |
for(unsigned icand=0;icand<pfCandidateColl->size(); icand++) { |
1042 |
+ |
const reco::PFCandidate pfCandidate = pfCandidateColl->at(icand); |
1043 |
+ |
|
1044 |
+ |
int id = pfCandidate.particleId(); |
1045 |
+ |
if(abs(id) != 3) continue; |
1046 |
+ |
|
1047 |
+ |
if(reco::deltaR(pfJet,pfCandidate)>0.5) continue; |
1048 |
+ |
|
1049 |
+ |
double pt = pfCandidate.pt(); |
1050 |
+ |
if(pt>ptMax){ |
1051 |
+ |
ptMax = pt; |
1052 |
+ |
pfMuonIndex = (int) icand; |
1053 |
+ |
} |
1054 |
+ |
} |
1055 |
+ |
|
1056 |
+ |
return pfMuonIndex; |
1057 |
+ |
|
1058 |
+ |
} |
1059 |
+ |
|
1060 |
+ |
|
1061 |
+ |
double |
1062 |
+ |
HiInclusiveJetAnalyzer::getPtRel(const reco::PFCandidate lep, const pat::Jet& jet ) |
1063 |
+ |
{ |
1064 |
+ |
|
1065 |
+ |
float lj_x = jet.p4().px(); |
1066 |
+ |
float lj_y = jet.p4().py(); |
1067 |
+ |
float lj_z = jet.p4().pz(); |
1068 |
+ |
|
1069 |
+ |
// absolute values squared |
1070 |
+ |
float lj2 = lj_x*lj_x+lj_y*lj_y+lj_z*lj_z; |
1071 |
+ |
float lep2 = lep.px()*lep.px()+lep.py()*lep.py()+lep.pz()*lep.pz(); |
1072 |
+ |
|
1073 |
+ |
// projection vec(mu) to lepjet axis |
1074 |
+ |
float lepXlj = lep.px()*lj_x+lep.py()*lj_y+lep.pz()*lj_z; |
1075 |
+ |
|
1076 |
+ |
// absolute value squared and normalized |
1077 |
+ |
float pLrel2 = lepXlj*lepXlj/lj2; |
1078 |
+ |
|
1079 |
+ |
// lep2 = pTrel2 + pLrel2 |
1080 |
+ |
float pTrel2 = lep2-pLrel2; |
1081 |
+ |
|
1082 |
+ |
return (pTrel2 > 0) ? std::sqrt(pTrel2) : 0.0; |
1083 |
+ |
} |
1084 |
+ |
|
1085 |
+ |
// Recursive function, but this version gets called only the first time |
1086 |
+ |
void |
1087 |
+ |
HiInclusiveJetAnalyzer::saveDaughters(const reco::GenParticle &gen){ |
1088 |
+ |
|
1089 |
+ |
for(unsigned i=0;i<gen.numberOfDaughters();i++){ |
1090 |
+ |
const reco::Candidate & daughter = *gen.daughter(i); |
1091 |
+ |
double daughterPt = daughter.pt(); |
1092 |
+ |
if(daughterPt<1.) continue; |
1093 |
+ |
double daughterEta = daughter.eta(); |
1094 |
+ |
if(fabs(daughterEta)>3.) continue; |
1095 |
+ |
int daughterPdgId = daughter.pdgId(); |
1096 |
+ |
int daughterStatus = daughter.status(); |
1097 |
+ |
// Special case when b->b+string, both b and string contain all daughters, so only take the string |
1098 |
+ |
if(gen.pdgId()==daughterPdgId && gen.status()==3 && daughterStatus==2) continue; |
1099 |
+ |
|
1100 |
+ |
// cheesy way of finding strings which were already used |
1101 |
+ |
if(daughter.pdgId()==92){ |
1102 |
+ |
for(unsigned ist=0;ist<usedStringPts.size();ist++){ |
1103 |
+ |
if(fabs(daughter.pt() - usedStringPts[ist]) < 0.0001) return; |
1104 |
+ |
} |
1105 |
+ |
usedStringPts.push_back(daughter.pt()); |
1106 |
+ |
} |
1107 |
+ |
jets_.bJetIndex[jets_.bMult] = jets_.nref; |
1108 |
+ |
jets_.bStatus[jets_.bMult] = daughterStatus; |
1109 |
+ |
jets_.bVx[jets_.bMult] = daughter.vx(); |
1110 |
+ |
jets_.bVy[jets_.bMult] = daughter.vy(); |
1111 |
+ |
jets_.bVz[jets_.bMult] = daughter.vz(); |
1112 |
+ |
jets_.bPt[jets_.bMult] = daughterPt; |
1113 |
+ |
jets_.bEta[jets_.bMult] = daughterEta; |
1114 |
+ |
jets_.bPhi[jets_.bMult] = daughter.phi(); |
1115 |
+ |
jets_.bPdg[jets_.bMult] = daughterPdgId; |
1116 |
+ |
jets_.bChg[jets_.bMult] = daughter.charge(); |
1117 |
+ |
jets_.bMult++; |
1118 |
+ |
saveDaughters(daughter); |
1119 |
+ |
} |
1120 |
+ |
} |
1121 |
+ |
|
1122 |
+ |
// This version called for all subsequent calls |
1123 |
+ |
void |
1124 |
+ |
HiInclusiveJetAnalyzer::saveDaughters(const reco::Candidate &gen){ |
1125 |
+ |
|
1126 |
+ |
for(unsigned i=0;i<gen.numberOfDaughters();i++){ |
1127 |
+ |
const reco::Candidate & daughter = *gen.daughter(i); |
1128 |
+ |
double daughterPt = daughter.pt(); |
1129 |
+ |
if(daughterPt<1.) continue; |
1130 |
+ |
double daughterEta = daughter.eta(); |
1131 |
+ |
if(fabs(daughterEta)>3.) continue; |
1132 |
+ |
int daughterPdgId = daughter.pdgId(); |
1133 |
+ |
int daughterStatus = daughter.status(); |
1134 |
+ |
// Special case when b->b+string, both b and string contain all daughters, so only take the string |
1135 |
+ |
if(gen.pdgId()==daughterPdgId && gen.status()==3 && daughterStatus==2) continue; |
1136 |
+ |
|
1137 |
+ |
// cheesy way of finding strings which were already used |
1138 |
+ |
if(daughter.pdgId()==92){ |
1139 |
+ |
for(unsigned ist=0;ist<usedStringPts.size();ist++){ |
1140 |
+ |
if(fabs(daughter.pt() - usedStringPts[ist]) < 0.0001) return; |
1141 |
+ |
} |
1142 |
+ |
usedStringPts.push_back(daughter.pt()); |
1143 |
+ |
} |
1144 |
+ |
|
1145 |
+ |
jets_.bJetIndex[jets_.bMult] = jets_.nref; |
1146 |
+ |
jets_.bStatus[jets_.bMult] = daughterStatus; |
1147 |
+ |
jets_.bVx[jets_.bMult] = daughter.vx(); |
1148 |
+ |
jets_.bVy[jets_.bMult] = daughter.vy(); |
1149 |
+ |
jets_.bVz[jets_.bMult] = daughter.vz(); |
1150 |
+ |
jets_.bPt[jets_.bMult] = daughterPt; |
1151 |
+ |
jets_.bEta[jets_.bMult] = daughterEta; |
1152 |
+ |
jets_.bPhi[jets_.bMult] = daughter.phi(); |
1153 |
+ |
jets_.bPdg[jets_.bMult] = daughterPdgId; |
1154 |
+ |
jets_.bChg[jets_.bMult] = daughter.charge(); |
1155 |
+ |
jets_.bMult++; |
1156 |
+ |
saveDaughters(daughter); |
1157 |
+ |
} |
1158 |
+ |
} |
1159 |
+ |
|
1160 |
+ |
double HiInclusiveJetAnalyzer::getEt(math::XYZPoint pos, double energy){ |
1161 |
+ |
double et = energy*sin(pos.theta()); |
1162 |
+ |
return et; |
1163 |
+ |
} |
1164 |
+ |
|
1165 |
+ |
math::XYZPoint HiInclusiveJetAnalyzer::getPosition(const DetId &id, reco::Vertex::Point vtx){ |
1166 |
+ |
const GlobalPoint& pos=geo->getPosition(id); |
1167 |
+ |
math::XYZPoint posV(pos.x() - vtx.x(),pos.y() - vtx.y(),pos.z() - vtx.z()); |
1168 |
+ |
return posV; |
1169 |
+ |
} |
1170 |
+ |
|
1171 |
|
|
1172 |
|
|
1173 |
|
DEFINE_FWK_MODULE(HiInclusiveJetAnalyzer); |