21 |
|
#include "DataFormats/HeavyIonEvent/interface/Centrality.h" |
22 |
|
|
23 |
|
#include "DataFormats/CaloTowers/interface/CaloTowerCollection.h" |
24 |
– |
#include "DataFormats/PatCandidates/interface/Jet.h" |
24 |
|
#include "DataFormats/JetReco/interface/CaloJetCollection.h" |
25 |
|
#include "SimDataFormats/HiGenData/interface/GenHIEvent.h" |
26 |
|
#include "SimDataFormats/GeneratorProducts/interface/GenEventInfoProduct.h" |
27 |
+ |
#include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h" |
28 |
|
#include "DataFormats/JetReco/interface/GenJetCollection.h" |
29 |
|
#include "DataFormats/VertexReco/interface/Vertex.h" |
30 |
|
#include "DataFormats/Math/interface/deltaPhi.h" |
43 |
|
using namespace reco; |
44 |
|
|
45 |
|
HiInclusiveJetAnalyzer::HiInclusiveJetAnalyzer(const edm::ParameterSet& iConfig) { |
46 |
– |
|
46 |
|
|
47 |
|
jetTag_ = iConfig.getParameter<InputTag>("jetTag"); |
48 |
< |
vtxTag_ = iConfig.getUntrackedParameter<edm::InputTag>("vtxTag",edm::InputTag("hiSelectedVertex")); |
48 |
> |
matchTag_ = iConfig.getUntrackedParameter<InputTag>("matchTag",jetTag_); |
49 |
> |
|
50 |
> |
vtxTag_ = iConfig.getUntrackedParameter<edm::InputTag>("vtxTag",edm::InputTag("hiSelectedVertex")); |
51 |
> |
trackTag_ = iConfig.getParameter<InputTag>("trackTag"); |
52 |
> |
useQuality_ = iConfig.getUntrackedParameter<bool>("useQuality",0); |
53 |
> |
string trackQuality_ = iConfig.getUntrackedParameter<string>("trackQuality","highPurity"); |
54 |
|
|
55 |
|
isMC_ = iConfig.getUntrackedParameter<bool>("isMC",false); |
56 |
+ |
fillGenJets_ = iConfig.getUntrackedParameter<bool>("fillGenJets",false); |
57 |
+ |
|
58 |
+ |
doTrigger_ = iConfig.getUntrackedParameter<bool>("doTrigger",false); |
59 |
+ |
|
60 |
+ |
rParam = iConfig.getParameter<double>("rParam"); |
61 |
+ |
hardPtMin_ = iConfig.getUntrackedParameter<double>("hardPtMin",4); |
62 |
|
|
63 |
|
if(isMC_){ |
64 |
|
genjetTag_ = iConfig.getParameter<InputTag>("genjetTag"); |
67 |
|
verbose_ = iConfig.getUntrackedParameter<bool>("verbose",false); |
68 |
|
|
69 |
|
useCentrality_ = iConfig.getUntrackedParameter<bool>("useCentrality",false); |
70 |
< |
useVtx_ = iConfig.getUntrackedParameter<bool>("useVtx",true); |
70 |
> |
useVtx_ = iConfig.getUntrackedParameter<bool>("useVtx",false); |
71 |
|
useJEC_ = iConfig.getUntrackedParameter<bool>("useJEC",true); |
72 |
|
usePat_ = iConfig.getUntrackedParameter<bool>("usePAT",true); |
73 |
|
|
74 |
< |
if(!isMC_){ |
74 |
> |
doLifeTimeTagging_ = iConfig.getUntrackedParameter<bool>("doLifeTimeTagging",false); |
75 |
> |
doLifeTimeTaggingExtras_ = iConfig.getUntrackedParameter<bool>("doLifeTimeTaggingExtras",true); |
76 |
> |
saveBfragments_ = iConfig.getUntrackedParameter<bool>("saveBfragments",false); |
77 |
> |
|
78 |
> |
pfCandidateLabel_ = iConfig.getUntrackedParameter<edm::InputTag>("pfCandidateLabel",edm::InputTag("particleFlowTmp")); |
79 |
> |
|
80 |
> |
if(doTrigger_){ |
81 |
|
L1gtReadout_ = iConfig.getParameter<edm::InputTag>("L1gtReadout"); |
82 |
|
hltResName_ = iConfig.getUntrackedParameter<string>("hltTrgResults","TriggerResults::HLT"); |
83 |
|
|
94 |
|
} |
95 |
|
|
96 |
|
cout<<" jet collection : "<<jetTag_<<endl; |
97 |
< |
if(isMC_)cout<<" genjet collection : "<<genjetTag_<<endl; |
97 |
> |
doSubEvent_ = 0; |
98 |
|
|
99 |
+ |
if(isMC_){ |
100 |
+ |
cout<<" genjet collection : "<<genjetTag_<<endl; |
101 |
+ |
genPtMin_ = iConfig.getUntrackedParameter<double>("genPtMin",0); |
102 |
+ |
doSubEvent_ = iConfig.getUntrackedParameter<bool>("doSubEvent",1); |
103 |
+ |
} |
104 |
|
|
105 |
|
|
106 |
|
} |
125 |
|
t = fs1->make<TTree>("t",jetTagTitle.c_str()); |
126 |
|
|
127 |
|
// TTree* t= new TTree("t","Jet Response Analyzer"); |
128 |
< |
t->Branch("run",&jets_.run,"run/I"); |
128 |
> |
//t->Branch("run",&jets_.run,"run/I"); |
129 |
|
t->Branch("evt",&jets_.evt,"evt/I"); |
130 |
< |
t->Branch("lumi",&jets_.lumi,"lumi/I"); |
130 |
> |
//t->Branch("lumi",&jets_.lumi,"lumi/I"); |
131 |
|
t->Branch("b",&jets_.b,"b/F"); |
132 |
< |
t->Branch("vx",&jets_.vx,"vx/F"); |
133 |
< |
t->Branch("vy",&jets_.vy,"vy/F"); |
134 |
< |
t->Branch("vz",&jets_.vz,"vz/F"); |
135 |
< |
t->Branch("hf",&jets_.hf,"hf/F"); |
132 |
> |
if (useVtx_) { |
133 |
> |
t->Branch("vx",&jets_.vx,"vx/F"); |
134 |
> |
t->Branch("vy",&jets_.vy,"vy/F"); |
135 |
> |
t->Branch("vz",&jets_.vz,"vz/F"); |
136 |
> |
} |
137 |
> |
if (useCentrality_) { |
138 |
> |
t->Branch("hf",&jets_.hf,"hf/F"); |
139 |
> |
t->Branch("bin",&jets_.bin,"bin/I"); |
140 |
> |
} |
141 |
> |
|
142 |
|
t->Branch("nref",&jets_.nref,"nref/I"); |
116 |
– |
t->Branch("bin",&jets_.bin,"bin/I"); |
143 |
|
t->Branch("rawpt",jets_.rawpt,"rawpt[nref]/F"); |
144 |
|
t->Branch("jtpt",jets_.jtpt,"jtpt[nref]/F"); |
145 |
|
t->Branch("jteta",jets_.jteta,"jteta[nref]/F"); |
147 |
|
t->Branch("jtphi",jets_.jtphi,"jtphi[nref]/F"); |
148 |
|
t->Branch("jtpu",jets_.jtpu,"jtpu[nref]/F"); |
149 |
|
|
150 |
+ |
// jet ID information, jet composition |
151 |
+ |
t->Branch("discr_fr01", jets_.discr_fr01,"discr_fr01[nref]/F"); |
152 |
+ |
|
153 |
+ |
t->Branch("trackMax", jets_.trackMax,"trackMax[nref]/F"); |
154 |
+ |
t->Branch("trackSum", jets_.trackSum,"trackSum[nref]/F"); |
155 |
+ |
t->Branch("trackN", jets_.trackN,"trackN[nref]/I"); |
156 |
+ |
t->Branch("trackHardSum", jets_.trackHardSum,"trackHardSum[nref]/F"); |
157 |
+ |
t->Branch("trackHardN", jets_.trackHardN,"trackHardN[nref]/I"); |
158 |
+ |
|
159 |
+ |
t->Branch("chargedMax", jets_.chargedMax,"chargedMax[nref]/F"); |
160 |
+ |
t->Branch("chargedSum", jets_.chargedSum,"chargedSum[nref]/F"); |
161 |
+ |
t->Branch("chargedN", jets_.chargedN,"chargedN[nref]/I"); |
162 |
+ |
t->Branch("chargedHardSum", jets_.chargedHardSum,"chargedHardSum[nref]/F"); |
163 |
+ |
t->Branch("chargedHardN", jets_.chargedHardN,"chargedHardN[nref]/I"); |
164 |
+ |
|
165 |
+ |
t->Branch("photonMax", jets_.photonMax,"photonMax[nref]/F"); |
166 |
+ |
t->Branch("photonSum", jets_.photonSum,"photonSum[nref]/F"); |
167 |
+ |
t->Branch("photonN", jets_.photonN,"photonN[nref]/I"); |
168 |
+ |
t->Branch("photonHardSum", jets_.photonHardSum,"photonHardSum[nref]/F"); |
169 |
+ |
t->Branch("photonHardN", jets_.photonHardN,"photonHardN[nref]/I"); |
170 |
+ |
|
171 |
+ |
t->Branch("neutralMax", jets_.neutralMax,"neutralMax[nref]/F"); |
172 |
+ |
t->Branch("neutralSum", jets_.neutralSum,"neutralSum[nref]/F"); |
173 |
+ |
t->Branch("neutralN", jets_.neutralN,"neutralN[nref]/I"); |
174 |
+ |
|
175 |
+ |
t->Branch("eMax", jets_.eMax,"eMax[nref]/F"); |
176 |
+ |
t->Branch("eSum", jets_.eSum,"eSum[nref]/F"); |
177 |
+ |
t->Branch("eN", jets_.eN,"eN[nref]/I"); |
178 |
+ |
|
179 |
+ |
t->Branch("muMax", jets_.muMax,"muMax[nref]/F"); |
180 |
+ |
t->Branch("muSum", jets_.muSum,"muSum[nref]/F"); |
181 |
+ |
t->Branch("muN", jets_.muN,"muN[nref]/I"); |
182 |
+ |
|
183 |
+ |
t->Branch("matchedPt", jets_.matchedPt,"matchedPt[nref]/F"); |
184 |
+ |
t->Branch("matchedR", jets_.matchedR,"matchedR[nref]/F"); |
185 |
+ |
|
186 |
+ |
// b-jet discriminators |
187 |
+ |
if (doLifeTimeTagging_) { |
188 |
+ |
|
189 |
+ |
t->Branch("discr_csvMva",jets_.discr_csvMva,"discr_csvMva[nref]/F"); |
190 |
+ |
t->Branch("discr_csvSimple",jets_.discr_csvSimple,"discr_csvSimple[nref]/F"); |
191 |
+ |
t->Branch("discr_muByIp3",jets_.discr_muByIp3,"discr_muByIp3[nref]/F"); |
192 |
+ |
t->Branch("discr_muByPt",jets_.discr_muByPt,"discr_muByPt[nref]/F"); |
193 |
+ |
t->Branch("discr_prob",jets_.discr_prob,"discr_prob[nref]/F"); |
194 |
+ |
t->Branch("discr_probb",jets_.discr_probb,"discr_probb[nref]/F"); |
195 |
+ |
t->Branch("discr_tcHighEff",jets_.discr_tcHighEff,"discr_tcHighEff[nref]/F"); |
196 |
+ |
t->Branch("discr_tcHighPur",jets_.discr_tcHighPur,"discr_tcHighPur[nref]/F"); |
197 |
+ |
|
198 |
+ |
t->Branch("nsvtx", jets_.nsvtx, "nsvtx[nref]/I"); |
199 |
+ |
t->Branch("svtxntrk", jets_.svtxntrk, "svtxntrk[nref]/I"); |
200 |
+ |
t->Branch("svtxdl", jets_.svtxdl, "svtxdl[nref]/F"); |
201 |
+ |
t->Branch("svtxdls", jets_.svtxdls, "svtxdls[nref]/F"); |
202 |
+ |
t->Branch("svtxm", jets_.svtxm, "svtxm[nref]/F"); |
203 |
+ |
t->Branch("svtxpt", jets_.svtxpt, "svtxpt[nref]/F"); |
204 |
+ |
|
205 |
+ |
t->Branch("nIPtrk",jets_.nIPtrk,"nIPtrk[nref]/I"); |
206 |
+ |
t->Branch("nselIPtrk",jets_.nselIPtrk,"nselIPtrk[nref]/I"); |
207 |
+ |
|
208 |
+ |
if (doLifeTimeTaggingExtras_) { |
209 |
+ |
t->Branch("nIP",&jets_.nIP,"nIP/I"); |
210 |
+ |
t->Branch("ipJetIndex",jets_.ipJetIndex,"ipJetIndex[nIP]/I"); |
211 |
+ |
t->Branch("ipPt",jets_.ipPt,"ipPt[nIP]/F"); |
212 |
+ |
t->Branch("ipProb0",jets_.ipProb0,"ipProb0[nIP]/F"); |
213 |
+ |
t->Branch("ipProb1",jets_.ipProb1,"ipProb1[nIP]/F"); |
214 |
+ |
t->Branch("ip2d",jets_.ip2d,"ip2d[nIP]/F"); |
215 |
+ |
t->Branch("ip2dSig",jets_.ip2dSig,"ip2dSig[nIP]/F"); |
216 |
+ |
t->Branch("ip3d",jets_.ip3d,"ip3d[nIP]/F"); |
217 |
+ |
t->Branch("ip3dSig",jets_.ip3dSig,"ip3dSig[nIP]/F"); |
218 |
+ |
t->Branch("ipDist2Jet",jets_.ipDist2Jet,"ipDist2Jet[nIP]/F"); |
219 |
+ |
t->Branch("ipDist2JetSig",jets_.ipDist2JetSig,"ipDist2JetSig[nIP]/F"); |
220 |
+ |
t->Branch("ipClosest2Jet",jets_.ipClosest2Jet,"ipClosest2Jet[nIP]/F"); |
221 |
+ |
|
222 |
+ |
} |
223 |
+ |
|
224 |
+ |
t->Branch("mue", jets_.mue, "mue[nref]/F"); |
225 |
+ |
t->Branch("mupt", jets_.mupt, "mupt[nref]/F"); |
226 |
+ |
t->Branch("mueta", jets_.mueta, "mueta[nref]/F"); |
227 |
+ |
t->Branch("muphi", jets_.muphi, "muphi[nref]/F"); |
228 |
+ |
t->Branch("mudr", jets_.mudr, "mudr[nref]/F"); |
229 |
+ |
t->Branch("muptrel", jets_.muptrel, "muptrel[nref]/F"); |
230 |
+ |
t->Branch("muchg", jets_.muchg, "muchg[nref]/I"); |
231 |
+ |
} |
232 |
+ |
|
233 |
+ |
|
234 |
|
if(isMC_){ |
235 |
+ |
t->Branch("beamId1",&jets_.beamId1,"beamId1/I"); |
236 |
+ |
t->Branch("beamId2",&jets_.beamId2,"beamId2/I"); |
237 |
+ |
|
238 |
|
t->Branch("pthat",&jets_.pthat,"pthat/F"); |
239 |
|
|
240 |
|
// Only matched gen jets |
246 |
|
t->Branch("refdrjt",jets_.refdrjt,"refdrjt[nref]/F"); |
247 |
|
// matched parton |
248 |
|
t->Branch("refparton_pt",jets_.refparton_pt,"refparton_pt[nref]/F"); |
249 |
< |
t->Branch("refparton_flavor",jets_.refparton_flavor,"refparton_flavor[nref]/F"); |
249 |
> |
t->Branch("refparton_flavor",jets_.refparton_flavor,"refparton_flavor[nref]/I"); |
250 |
> |
t->Branch("refparton_flavorForB",jets_.refparton_flavorForB,"refparton_flavorForB[nref]/I"); |
251 |
> |
|
252 |
> |
if(fillGenJets_){ |
253 |
> |
// For all gen jets, matched or unmatched |
254 |
> |
t->Branch("ngen",&jets_.ngen,"ngen/I"); |
255 |
> |
t->Branch("genmatchindex",jets_.genmatchindex,"genmatchindex[ngen]/I"); |
256 |
> |
t->Branch("genpt",jets_.genpt,"genpt[ngen]/F"); |
257 |
> |
t->Branch("geneta",jets_.geneta,"geneta[ngen]/F"); |
258 |
> |
t->Branch("geny",jets_.geny,"geny[ngen]/F"); |
259 |
> |
t->Branch("genphi",jets_.genphi,"genphi[ngen]/F"); |
260 |
> |
t->Branch("gendphijt",jets_.gendphijt,"gendphijt[ngen]/F"); |
261 |
> |
t->Branch("gendrjt",jets_.gendrjt,"gendrjt[ngen]/F"); |
262 |
> |
|
263 |
> |
if(doSubEvent_){ |
264 |
> |
t->Branch("gensubid",jets_.gensubid,"gensubid[ngen]/I"); |
265 |
> |
} |
266 |
> |
} |
267 |
> |
|
268 |
> |
if(saveBfragments_ ) { |
269 |
> |
t->Branch("bMult",&jets_.bMult,"bMult/I"); |
270 |
> |
t->Branch("bJetIndex",jets_.bJetIndex,"bJetIndex[bMult]/I"); |
271 |
> |
t->Branch("bStatus",jets_.bStatus,"bStatus[bMult]/I"); |
272 |
> |
t->Branch("bVx",jets_.bVx,"bVx[bMult]/F"); |
273 |
> |
t->Branch("bVy",jets_.bVy,"bVy[bMult]/F"); |
274 |
> |
t->Branch("bVz",jets_.bVz,"bVz[bMult]/F"); |
275 |
> |
t->Branch("bPt",jets_.bPt,"bPt[bMult]/F"); |
276 |
> |
t->Branch("bEta",jets_.bEta,"bEta[bMult]/F"); |
277 |
> |
t->Branch("bPhi",jets_.bPhi,"bPhi[bMult]/F"); |
278 |
> |
t->Branch("bPdg",jets_.bPdg,"bPdg[bMult]/I"); |
279 |
> |
t->Branch("bChg",jets_.bChg,"bChg[bMult]/I"); |
280 |
> |
} |
281 |
|
|
138 |
– |
// For all gen jets, matched or unmatched |
139 |
– |
t->Branch("ngen",&jets_.ngen,"ngen/I"); |
140 |
– |
t->Branch("genmatchindex",jets_.genmatchindex,"genmatchindex[ngen]/I"); |
141 |
– |
t->Branch("genpt",jets_.genpt,"genpt[ngen]/F"); |
142 |
– |
t->Branch("geneta",jets_.geneta,"geneta[ngen]/F"); |
143 |
– |
t->Branch("geny",jets_.geny,"geny[ngen]/F"); |
144 |
– |
t->Branch("genphi",jets_.genphi,"genphi[ngen]/F"); |
145 |
– |
t->Branch("gendphijt",jets_.gendphijt,"gendphijt[ngen]/F"); |
146 |
– |
t->Branch("gendrjt",jets_.gendrjt,"gendrjt[ngen]/F"); |
282 |
|
} |
283 |
< |
|
283 |
> |
/* |
284 |
|
if(!isMC_){ |
285 |
|
t->Branch("nL1TBit",&jets_.nL1TBit,"nL1TBit/I"); |
286 |
|
t->Branch("l1TBit",jets_.l1TBit,"l1TBit[nL1TBit]/O"); |
292 |
|
t->Branch("hltBit",jets_.hltBit,"hltBit[nHLTBit]/O"); |
293 |
|
|
294 |
|
} |
295 |
< |
|
295 |
> |
*/ |
296 |
|
TH1D::SetDefaultSumw2(); |
297 |
|
|
298 |
|
|
313 |
|
|
314 |
|
LogDebug("HiInclusiveJetAnalyzer")<<"START event: "<<event<<" in run "<<run<<endl; |
315 |
|
|
181 |
– |
|
316 |
|
int bin = -1; |
317 |
|
double hf = 0.; |
318 |
|
double b = 999.; |
319 |
|
|
320 |
|
if(useCentrality_){ |
187 |
– |
//if(!isMC_){ |
321 |
|
if(!centrality_) centrality_ = new CentralityProvider(iSetup); |
322 |
|
centrality_->newEvent(iEvent,iSetup); // make sure you do this first in every event |
323 |
|
//double c = centrality_->centralityValue(); |
327 |
|
|
328 |
|
bin = centrality_->getBin(); |
329 |
|
b = centrality_->bMean(); |
197 |
– |
//} |
198 |
– |
/* |
199 |
– |
else{ |
200 |
– |
|
201 |
– |
TFile * centFile = new TFile("/net/hidsk0001/d00/scratch/mnguyen/CMSSW_3_9_1_patch1/src/macros/Hydjet_CentTable.root"); |
202 |
– |
|
203 |
– |
edm::Handle<reco::Centrality> cent; |
204 |
– |
iEvent.getByLabel(edm::InputTag("hiCentrality"),cent); |
205 |
– |
//cout<<" grabbed centrality "<<endl; |
206 |
– |
CentralityBins::RunMap cmap = getCentralityFromFile(centFile, "makeCentralityTableTFile", "HFhitsHydjet_2760GeV", 1, 1); |
207 |
– |
|
208 |
– |
// Still getting cent from hits. When tower tables become available, we need to switch |
209 |
– |
hf = cent->EtHFhitSum(); |
210 |
– |
//cout<<" hf "<<hf<<endl; |
211 |
– |
bin = cmap[run]->getBin(hf); |
212 |
– |
b = cmap[run]->bMeanOfBin(bin); |
213 |
– |
} |
214 |
– |
*/ |
330 |
|
} |
331 |
|
|
332 |
|
|
333 |
|
|
334 |
|
|
220 |
– |
//double jetPtMin = 35.; |
221 |
– |
|
335 |
|
|
336 |
|
// loop the events |
337 |
|
|
352 |
|
|
353 |
|
edm::Handle<pat::JetCollection> patjets; |
354 |
|
if(usePat_)iEvent.getByLabel(jetTag_, patjets); |
355 |
+ |
|
356 |
+ |
edm::Handle<pat::JetCollection> matchedjets; |
357 |
+ |
iEvent.getByLabel(matchTag_, matchedjets); |
358 |
|
|
359 |
|
edm::Handle<reco::JetView> jets; |
360 |
|
iEvent.getByLabel(jetTag_, jets); |
361 |
|
|
362 |
+ |
edm::Handle<reco::PFCandidateCollection> pfCandidates; |
363 |
+ |
iEvent.getByLabel(pfCandidateLabel_,pfCandidates); |
364 |
+ |
|
365 |
+ |
edm::Handle<reco::TrackCollection> tracks; |
366 |
+ |
iEvent.getByLabel(trackTag_,tracks); |
367 |
+ |
|
368 |
|
// FILL JRA TREE |
369 |
|
|
370 |
|
jets_.b = b; |
371 |
|
jets_.nref = 0; |
372 |
|
|
373 |
< |
if(!isMC_){ |
373 |
> |
if(doTrigger_){ |
374 |
|
fillL1Bits(iEvent); |
375 |
|
fillHLTBits(iEvent); |
376 |
|
} |
378 |
|
for(unsigned int j = 0 ; j < jets->size(); ++j){ |
379 |
|
const reco::Jet& jet = (*jets)[j]; |
380 |
|
|
381 |
< |
//cout<<" jet pt "<<jet.pt()<<endl; |
260 |
< |
//if(jet.pt() < jetPtMin) continue; |
381 |
> |
|
382 |
|
if (useJEC_ && usePat_){ |
383 |
|
jets_.rawpt[jets_.nref]=(*patjets)[j].correctedJet("Uncorrected").pt(); |
384 |
|
} |
385 |
+ |
|
386 |
+ |
if(doLifeTimeTagging_){ |
387 |
+ |
|
388 |
+ |
if(jetTag_.label()=="icPu5patJets"){ |
389 |
+ |
jets_.discr_csvMva[jets_.nref] = (*patjets)[j].bDiscriminator("icPu5CombinedSecondaryVertexMVABJetTags"); |
390 |
+ |
jets_.discr_csvSimple[jets_.nref] = (*patjets)[j].bDiscriminator("icPu5CombinedSecondaryVertexBJetTags"); |
391 |
+ |
jets_.discr_muByIp3[jets_.nref] = (*patjets)[j].bDiscriminator("icPu5SoftMuonByIP3dBJetTags"); |
392 |
+ |
jets_.discr_muByPt[jets_.nref] = (*patjets)[j].bDiscriminator("icPu5SoftMuonByPtBJetTags"); |
393 |
+ |
jets_.discr_prob[jets_.nref] = (*patjets)[j].bDiscriminator("icPu5JetBProbabilityBJetTags"); |
394 |
+ |
jets_.discr_probb[jets_.nref] = (*patjets)[j].bDiscriminator("icPu5JetProbabilityBJetTags"); |
395 |
+ |
jets_.discr_tcHighEff[jets_.nref] = (*patjets)[j].bDiscriminator("icPu5TrackCountingHighEffBJetTags"); |
396 |
+ |
jets_.discr_tcHighPur[jets_.nref] = (*patjets)[j].bDiscriminator("icPu5TrackCountingHighPurBJetTags"); |
397 |
+ |
} |
398 |
+ |
else if(jetTag_.label()=="akPu3PFpatJets"){ |
399 |
+ |
jets_.discr_csvMva[jets_.nref] = (*patjets)[j].bDiscriminator("akPu3PFCombinedSecondaryVertexMVABJetTags"); |
400 |
+ |
jets_.discr_csvSimple[jets_.nref] = (*patjets)[j].bDiscriminator("akPu3PFCombinedSecondaryVertexBJetTags"); |
401 |
+ |
jets_.discr_muByIp3[jets_.nref] = (*patjets)[j].bDiscriminator("akPu3PFSoftMuonByIP3dBJetTags"); |
402 |
+ |
jets_.discr_muByPt[jets_.nref] = (*patjets)[j].bDiscriminator("akPu3PFSoftMuonByPtBJetTags"); |
403 |
+ |
jets_.discr_prob[jets_.nref] = (*patjets)[j].bDiscriminator("akPu3PFJetBProbabilityBJetTags"); |
404 |
+ |
jets_.discr_probb[jets_.nref] = (*patjets)[j].bDiscriminator("akPu3PFJetProbabilityBJetTags"); |
405 |
+ |
jets_.discr_tcHighEff[jets_.nref] = (*patjets)[j].bDiscriminator("akPu3PFTrackCountingHighEffBJetTags"); |
406 |
+ |
jets_.discr_tcHighPur[jets_.nref] = (*patjets)[j].bDiscriminator("akPu3PFTrackCountingHighPurBJetTags"); |
407 |
+ |
} |
408 |
+ |
else{ |
409 |
+ |
cout<<" b-tagging variables not filled for this collection, turn of doLifeTimeTagging "<<endl; |
410 |
+ |
} |
411 |
+ |
|
412 |
+ |
const reco::SecondaryVertexTagInfo& tagInfoSV=*(*patjets)[j].tagInfoSecondaryVertex(); |
413 |
+ |
|
414 |
+ |
jets_.nsvtx[jets_.nref] = tagInfoSV.nVertices(); |
415 |
+ |
|
416 |
+ |
if (tagInfoSV.nVertices()>0) { |
417 |
+ |
jets_.svtxntrk[jets_.nref] = tagInfoSV.nVertexTracks(0); |
418 |
+ |
// this is the 3d flight distance, for 2-D use (0,true) |
419 |
+ |
Measurement1D m1D = tagInfoSV.flightDistance(0); |
420 |
+ |
jets_.svtxdl[jets_.nref] = m1D.value(); |
421 |
+ |
jets_.svtxdls[jets_.nref] = m1D.significance(); |
422 |
+ |
|
423 |
+ |
const reco::Vertex& svtx = tagInfoSV.secondaryVertex(0); |
424 |
+ |
//cout<<" SV: vx: "<<svtx.x()<<" vy "<<svtx.y()<<" vz "<<svtx.z()<<endl; |
425 |
+ |
//cout<<" PV: vx: "<<jet.vx()<<" vy "<<jet.vy()<<" vz "<<jet.vz()<<endl; |
426 |
+ |
jets_.svtxm[jets_.nref] = svtx.p4().mass(); |
427 |
+ |
jets_.svtxpt[jets_.nref] = svtx.p4().pt(); |
428 |
+ |
//cout<<" chi2 "<<svtx.chi2()<<" ndof "<<svtx.ndof()<<endl; |
429 |
+ |
} |
430 |
+ |
|
431 |
+ |
const reco::TrackIPTagInfo& tagInfoIP=*(*patjets)[j].tagInfoTrackIP(); |
432 |
+ |
|
433 |
+ |
jets_.nIPtrk[jets_.nref] = tagInfoIP.tracks().size(); |
434 |
+ |
jets_.nselIPtrk[jets_.nref] = tagInfoIP.selectedTracks().size(); |
435 |
+ |
|
436 |
+ |
if (doLifeTimeTaggingExtras_) { |
437 |
+ |
|
438 |
+ |
TrackRefVector selTracks=tagInfoIP.selectedTracks(); |
439 |
+ |
|
440 |
+ |
GlobalPoint pv(tagInfoIP.primaryVertex()->position().x(),tagInfoIP.primaryVertex()->position().y(),tagInfoIP.primaryVertex()->position().z()); |
441 |
+ |
|
442 |
+ |
for(int it=0;it<jets_.nselIPtrk[jets_.nref] ;it++) |
443 |
+ |
{ |
444 |
+ |
jets_.ipJetIndex[jets_.nIP + it]= jets_.nref; |
445 |
+ |
TrackIPTagInfo::TrackIPData data = tagInfoIP.impactParameterData()[it]; |
446 |
+ |
jets_.ipPt[jets_.nIP + it] = selTracks[it]->pt(); |
447 |
+ |
jets_.ipProb0[jets_.nIP + it] = tagInfoIP.probabilities(0)[it]; |
448 |
+ |
jets_.ipProb1[jets_.nIP + it] = tagInfoIP.probabilities(1)[it]; |
449 |
+ |
jets_.ip2d[jets_.nIP + it] = data.ip2d.value(); |
450 |
+ |
jets_.ip2dSig[jets_.nIP + it] = data.ip2d.significance(); |
451 |
+ |
jets_.ip3d[jets_.nIP + it] = data.ip3d.value(); |
452 |
+ |
jets_.ip3dSig[jets_.nIP + it] = data.ip3d.significance(); |
453 |
+ |
jets_.ipDist2Jet[jets_.nIP + it] = data.distanceToJetAxis.value(); |
454 |
+ |
jets_.ipDist2JetSig[jets_.nIP + it] = data.distanceToJetAxis.significance(); |
455 |
+ |
jets_.ipClosest2Jet[jets_.nIP + it] = (data.closestToJetAxis - pv).mag(); //decay length |
456 |
+ |
} |
457 |
+ |
|
458 |
+ |
jets_.nIP += jets_.nselIPtrk[jets_.nref]; |
459 |
+ |
|
460 |
+ |
} |
461 |
+ |
|
462 |
+ |
const reco::PFCandidateCollection *pfCandidateColl = &(*pfCandidates); |
463 |
+ |
int pfMuonIndex = getPFJetMuon(jet, pfCandidateColl); |
464 |
+ |
if(pfMuonIndex >=0){ |
465 |
+ |
const reco::PFCandidate muon = pfCandidateColl->at(pfMuonIndex); |
466 |
+ |
jets_.mupt[jets_.nref] = muon.pt(); |
467 |
+ |
jets_.mueta[jets_.nref] = muon.eta(); |
468 |
+ |
jets_.muphi[jets_.nref] = muon.phi(); |
469 |
+ |
jets_.mue[jets_.nref] = muon.energy(); |
470 |
+ |
jets_.mudr[jets_.nref] = reco::deltaR(jet,muon); |
471 |
+ |
jets_.muptrel[jets_.nref] = getPtRel(muon, jet); |
472 |
+ |
jets_.muchg[jets_.nref] = muon.charge(); |
473 |
+ |
}else{ |
474 |
+ |
jets_.mupt[jets_.nref] = 0.0; |
475 |
+ |
jets_.mueta[jets_.nref] = 0.0; |
476 |
+ |
jets_.muphi[jets_.nref] = 0.0; |
477 |
+ |
jets_.mue[jets_.nref] = 0.0; |
478 |
+ |
jets_.mudr[jets_.nref] = 9.9; |
479 |
+ |
jets_.muptrel[jets_.nref] = 0.0; |
480 |
+ |
jets_.muchg[jets_.nref] = 0; |
481 |
+ |
} |
482 |
+ |
|
483 |
+ |
} |
484 |
+ |
|
485 |
+ |
|
486 |
+ |
|
487 |
+ |
// Jet ID variables |
488 |
+ |
|
489 |
+ |
jets_.muMax[jets_.nref] = 0; |
490 |
+ |
jets_.muSum[jets_.nref] = 0; |
491 |
+ |
jets_.muN[jets_.nref] = 0; |
492 |
+ |
|
493 |
+ |
jets_.eMax[jets_.nref] = 0; |
494 |
+ |
jets_.eSum[jets_.nref] = 0; |
495 |
+ |
jets_.eN[jets_.nref] = 0; |
496 |
+ |
|
497 |
+ |
jets_.neutralMax[jets_.nref] = 0; |
498 |
+ |
jets_.neutralSum[jets_.nref] = 0; |
499 |
+ |
jets_.neutralN[jets_.nref] = 0; |
500 |
+ |
|
501 |
+ |
jets_.photonMax[jets_.nref] = 0; |
502 |
+ |
jets_.photonSum[jets_.nref] = 0; |
503 |
+ |
jets_.photonN[jets_.nref] = 0; |
504 |
+ |
jets_.photonHardSum[jets_.nref] = 0; |
505 |
+ |
jets_.photonHardN[jets_.nref] = 0; |
506 |
+ |
|
507 |
+ |
jets_.chargedMax[jets_.nref] = 0; |
508 |
+ |
jets_.chargedSum[jets_.nref] = 0; |
509 |
+ |
jets_.chargedN[jets_.nref] = 0; |
510 |
+ |
jets_.chargedHardSum[jets_.nref] = 0; |
511 |
+ |
jets_.chargedHardN[jets_.nref] = 0; |
512 |
+ |
|
513 |
+ |
jets_.trackMax[jets_.nref] = 0; |
514 |
+ |
jets_.trackSum[jets_.nref] = 0; |
515 |
+ |
jets_.trackN[jets_.nref] = 0; |
516 |
+ |
jets_.trackHardSum[jets_.nref] = 0; |
517 |
+ |
jets_.trackHardN[jets_.nref] = 0; |
518 |
+ |
|
519 |
+ |
|
520 |
+ |
for(unsigned int icand = 0; icand < tracks->size(); ++icand){ |
521 |
+ |
const reco::Track& track = (*tracks)[icand]; |
522 |
+ |
if(useQuality_ && !(track.quality(reco::TrackBase::qualityByName(trackQuality_)))) continue; |
523 |
+ |
|
524 |
+ |
double dr = deltaR(jet,track); |
525 |
+ |
if(dr < rParam){ |
526 |
+ |
double ptcand = track.pt(); |
527 |
+ |
jets_.trackSum[jets_.nref] += ptcand; |
528 |
+ |
jets_.trackN[jets_.nref] += 1; |
529 |
+ |
|
530 |
+ |
if(ptcand > hardPtMin_){ |
531 |
+ |
jets_.trackHardSum[jets_.nref] += ptcand; |
532 |
+ |
jets_.trackHardN[jets_.nref] += 1; |
533 |
+ |
|
534 |
+ |
} |
535 |
+ |
if(ptcand > jets_.trackMax[jets_.nref]) jets_.trackMax[jets_.nref] = ptcand; |
536 |
+ |
|
537 |
+ |
} |
538 |
+ |
} |
539 |
+ |
|
540 |
+ |
for(unsigned int icand = 0; icand < pfCandidates->size(); ++icand){ |
541 |
+ |
const reco::PFCandidate& track = (*pfCandidates)[icand]; |
542 |
+ |
double dr = deltaR(jet,track); |
543 |
+ |
if(dr < rParam){ |
544 |
+ |
double ptcand = track.pt(); |
545 |
+ |
int pfid = track.particleId(); |
546 |
+ |
|
547 |
+ |
switch(pfid){ |
548 |
+ |
|
549 |
+ |
case 1: |
550 |
+ |
jets_.chargedSum[jets_.nref] += ptcand; |
551 |
+ |
jets_.chargedN[jets_.nref] += 1; |
552 |
+ |
if(ptcand > hardPtMin_){ |
553 |
+ |
jets_.chargedHardSum[jets_.nref] += ptcand; |
554 |
+ |
jets_.chargedHardN[jets_.nref] += 1; |
555 |
+ |
} |
556 |
+ |
if(ptcand > jets_.chargedMax[jets_.nref]) jets_.chargedMax[jets_.nref] = ptcand; |
557 |
+ |
break; |
558 |
+ |
|
559 |
+ |
case 2: |
560 |
+ |
jets_.eSum[jets_.nref] += ptcand; |
561 |
+ |
jets_.eN[jets_.nref] += 1; |
562 |
+ |
if(ptcand > jets_.eMax[jets_.nref]) jets_.eMax[jets_.nref] = ptcand; |
563 |
+ |
break; |
564 |
+ |
|
565 |
+ |
case 3: |
566 |
+ |
jets_.muSum[jets_.nref] += ptcand; |
567 |
+ |
jets_.muN[jets_.nref] += 1; |
568 |
+ |
if(ptcand > jets_.muMax[jets_.nref]) jets_.muMax[jets_.nref] = ptcand; |
569 |
+ |
break; |
570 |
+ |
|
571 |
+ |
case 4: |
572 |
+ |
jets_.photonSum[jets_.nref] += ptcand; |
573 |
+ |
jets_.photonN[jets_.nref] += 1; |
574 |
+ |
if(ptcand > hardPtMin_){ |
575 |
+ |
jets_.photonHardSum[jets_.nref] += ptcand; |
576 |
+ |
jets_.photonHardN[jets_.nref] += 1; |
577 |
+ |
} |
578 |
+ |
if(ptcand > jets_.photonMax[jets_.nref]) jets_.photonMax[jets_.nref] = ptcand; |
579 |
+ |
break; |
580 |
+ |
|
581 |
+ |
case 5: |
582 |
+ |
jets_.neutralSum[jets_.nref] += ptcand; |
583 |
+ |
jets_.neutralN[jets_.nref] += 1; |
584 |
+ |
if(ptcand > jets_.neutralMax[jets_.nref]) jets_.neutralMax[jets_.nref] = ptcand; |
585 |
+ |
break; |
586 |
+ |
|
587 |
+ |
default: |
588 |
+ |
break; |
589 |
+ |
|
590 |
+ |
} |
591 |
+ |
} |
592 |
+ |
} |
593 |
+ |
|
594 |
+ |
double drMin = 100; |
595 |
+ |
for(unsigned int imatch = 0 ; imatch < matchedjets->size(); ++imatch){ |
596 |
+ |
const reco::Jet& mjet = (*matchedjets)[imatch]; |
597 |
+ |
|
598 |
+ |
double dr = deltaR(jet,mjet); |
599 |
+ |
if(dr < drMin){ |
600 |
+ |
jets_.matchedPt[jets_.nref] = mjet.pt(); |
601 |
+ |
jets_.matchedR[jets_.nref] = dr; |
602 |
+ |
|
603 |
+ |
} |
604 |
+ |
} |
605 |
+ |
|
606 |
+ |
|
607 |
+ |
|
608 |
+ |
|
609 |
+ |
// if(etrk.quality(reco::TrackBase::qualityByName(qualityString_))) pev_.trkQual[pev_.nTrk]=1; |
610 |
+ |
|
611 |
+ |
///////////////////////////////////////////////////////////////// |
612 |
+ |
// Jet core pt^2 discriminant for fake jets |
613 |
+ |
// Edited by Yue Shi Lai <ylai@mit.edu> |
614 |
+ |
|
615 |
+ |
// Initial value is 0 |
616 |
+ |
jets_.discr_fr01[jets_.nref] = 0; |
617 |
+ |
// Start with no directional adaption, i.e. the fake rejection |
618 |
+ |
// axis is the jet axis |
619 |
+ |
float pseudorapidity_adapt = jets_.jteta[jets_.nref]; |
620 |
+ |
float azimuth_adapt = jets_.jtphi[jets_.nref]; |
621 |
+ |
|
622 |
+ |
// Unadapted discriminant with adaption search |
623 |
+ |
for (size_t iteration = 0; iteration < 2; iteration++) { |
624 |
+ |
float pseudorapidity_adapt_new = pseudorapidity_adapt; |
625 |
+ |
float azimuth_adapt_new = azimuth_adapt; |
626 |
+ |
float max_weighted_perp = 0; |
627 |
+ |
float perp_square_sum = 0; |
628 |
+ |
|
629 |
+ |
for (size_t index_pf_candidate = 0; |
630 |
+ |
index_pf_candidate < pfCandidates->size(); |
631 |
+ |
index_pf_candidate++) { |
632 |
+ |
const reco::PFCandidate &p = |
633 |
+ |
(*pfCandidates)[index_pf_candidate]; |
634 |
+ |
|
635 |
+ |
switch (p.particleId()) { |
636 |
+ |
//case 1: // Charged hadron |
637 |
+ |
//case 3: // Muon |
638 |
+ |
case 4: // Photon |
639 |
+ |
{ |
640 |
+ |
const float dpseudorapidity = |
641 |
+ |
p.eta() - pseudorapidity_adapt; |
642 |
+ |
const float dazimuth = |
643 |
+ |
reco::deltaPhi(p.phi(), azimuth_adapt); |
644 |
+ |
// The Gaussian scale factor is 0.5 / (0.1 * 0.1) |
645 |
+ |
// = 50 |
646 |
+ |
const float angular_weight = |
647 |
+ |
exp(-50.0F * (dpseudorapidity * dpseudorapidity + |
648 |
+ |
dazimuth * dazimuth)); |
649 |
+ |
const float weighted_perp = |
650 |
+ |
angular_weight * p.pt() * p.pt(); |
651 |
+ |
const float weighted_perp_square = |
652 |
+ |
weighted_perp * p.pt(); |
653 |
+ |
|
654 |
+ |
perp_square_sum += weighted_perp_square; |
655 |
+ |
if (weighted_perp >= max_weighted_perp) { |
656 |
+ |
pseudorapidity_adapt_new = p.eta(); |
657 |
+ |
azimuth_adapt_new = p.phi(); |
658 |
+ |
max_weighted_perp = weighted_perp; |
659 |
+ |
} |
660 |
+ |
} |
661 |
+ |
default: |
662 |
+ |
break; |
663 |
+ |
} |
664 |
+ |
} |
665 |
+ |
// Update the fake rejection value |
666 |
+ |
jets_.discr_fr01[jets_.nref] = std::max( |
667 |
+ |
jets_.discr_fr01[jets_.nref], perp_square_sum); |
668 |
+ |
// Update the directional adaption |
669 |
+ |
pseudorapidity_adapt = pseudorapidity_adapt_new; |
670 |
+ |
azimuth_adapt = azimuth_adapt_new; |
671 |
+ |
} |
672 |
+ |
|
673 |
+ |
|
674 |
|
jets_.jtpt[jets_.nref] = jet.pt(); |
675 |
|
jets_.jteta[jets_.nref] = jet.eta(); |
676 |
|
jets_.jtphi[jets_.nref] = jet.phi(); |
678 |
|
jets_.jtpu[jets_.nref] = jet.pileup(); |
679 |
|
|
680 |
|
if(isMC_ && usePat_){ |
681 |
< |
const reco::GenJet* genjet = (*patjets)[j].genJet(); |
681 |
> |
|
682 |
> |
const reco::GenJet * genjet = (*patjets)[j].genJet(); |
683 |
> |
|
684 |
|
if(genjet){ |
685 |
|
jets_.refpt[jets_.nref] = genjet->pt(); |
686 |
|
jets_.refeta[jets_.nref] = genjet->eta(); |
697 |
|
jets_.refdrjt[jets_.nref] = -999.; |
698 |
|
} |
699 |
|
|
700 |
< |
// matched partons |
701 |
< |
const reco::GenParticle * parton = (*patjets)[j].genParton(); |
702 |
< |
if(parton){ |
703 |
< |
jets_.refparton_pt[jets_.nref] = parton->pt(); |
704 |
< |
jets_.refparton_flavor[jets_.nref] = parton->pdgId(); |
705 |
< |
} else { |
706 |
< |
jets_.refparton_pt[jets_.nref] = -999; |
707 |
< |
jets_.refparton_flavor[jets_.nref] = -999; |
708 |
< |
} |
700 |
> |
jets_.refparton_flavorForB[jets_.nref] = (*patjets)[j].partonFlavour(); |
701 |
> |
|
702 |
> |
// matched partons |
703 |
> |
const reco::GenParticle & parton = *(*patjets)[j].genParton(); |
704 |
> |
|
705 |
> |
if((*patjets)[j].genParton()){ |
706 |
> |
jets_.refparton_pt[jets_.nref] = parton.pt(); |
707 |
> |
jets_.refparton_flavor[jets_.nref] = parton.pdgId(); |
708 |
> |
|
709 |
> |
if(saveBfragments_ && abs(jets_.refparton_flavorForB[jets_.nref])==5){ |
710 |
> |
|
711 |
> |
usedStringPts.clear(); |
712 |
> |
|
713 |
> |
// uncomment this if you want to know the ugly truth about parton matching -matt |
714 |
> |
//if(jet.pt() > 50 &&abs(parton.pdgId())!=5 && parton.pdgId()!=21) |
715 |
> |
// cout<<" Identified as a b, but doesn't match b or gluon, id = "<<parton.pdgId()<<endl; |
716 |
> |
|
717 |
> |
jets_.bJetIndex[jets_.bMult] = jets_.nref; |
718 |
> |
jets_.bStatus[jets_.bMult] = parton.status(); |
719 |
> |
jets_.bVx[jets_.bMult] = parton.vx(); |
720 |
> |
jets_.bVy[jets_.bMult] = parton.vy(); |
721 |
> |
jets_.bVz[jets_.bMult] = parton.vz(); |
722 |
> |
jets_.bPt[jets_.bMult] = parton.pt(); |
723 |
> |
jets_.bEta[jets_.bMult] = parton.eta(); |
724 |
> |
jets_.bPhi[jets_.bMult] = parton.phi(); |
725 |
> |
jets_.bPdg[jets_.bMult] = parton.pdgId(); |
726 |
> |
jets_.bChg[jets_.bMult] = parton.charge(); |
727 |
> |
jets_.bMult++; |
728 |
> |
saveDaughters(parton); |
729 |
> |
} |
730 |
> |
|
731 |
> |
|
732 |
> |
} else { |
733 |
> |
jets_.refparton_pt[jets_.nref] = -999; |
734 |
> |
jets_.refparton_flavor[jets_.nref] = -999; |
735 |
> |
} |
736 |
> |
|
737 |
> |
|
738 |
|
} |
739 |
|
|
740 |
|
jets_.nref++; |
745 |
|
|
746 |
|
if(isMC_){ |
747 |
|
|
748 |
+ |
edm::Handle<HepMCProduct> hepMCProduct; |
749 |
+ |
iEvent.getByLabel(eventInfoTag_,hepMCProduct); |
750 |
+ |
const HepMC::GenEvent* MCEvt = hepMCProduct->GetEvent(); |
751 |
+ |
std::pair<HepMC::GenParticle*,HepMC::GenParticle*> beamParticles = MCEvt->beam_particles(); |
752 |
+ |
jets_.beamId1 = beamParticles.first->pdg_id(); |
753 |
+ |
jets_.beamId2 = beamParticles.second->pdg_id(); |
754 |
+ |
|
755 |
|
edm::Handle<GenEventInfoProduct> hEventInfo; |
756 |
|
iEvent.getByLabel(eventInfoTag_,hEventInfo); |
757 |
|
//jets_.pthat = hEventInfo->binningValues()[0]; |
769 |
|
float genjet_pt = genjet.pt(); |
770 |
|
|
771 |
|
// threshold to reduce size of output in minbias PbPb |
772 |
< |
if(genjet_pt>20.){ |
772 |
> |
if(genjet_pt>genPtMin_){ |
773 |
|
|
774 |
|
jets_.genpt[jets_.ngen] = genjet_pt; |
775 |
|
jets_.geneta[jets_.ngen] = genjet.eta(); |
776 |
|
jets_.genphi[jets_.ngen] = genjet.phi(); |
777 |
|
jets_.geny[jets_.ngen] = genjet.eta(); |
778 |
|
|
779 |
+ |
if(doSubEvent_){ |
780 |
+ |
const GenParticle* gencon = genjet.getGenConstituent(0); |
781 |
+ |
jets_.gensubid[jets_.ngen] = gencon->collisionId(); |
782 |
+ |
} |
783 |
|
|
784 |
|
// find matching patJet if there is one |
785 |
|
|
786 |
|
jets_.gendrjt[jets_.ngen] = -1.0; |
787 |
|
jets_.genmatchindex[jets_.ngen] = -1; |
788 |
|
|
337 |
– |
|
789 |
|
for(unsigned int ijet = 0 ; ijet < jets->size(); ++ijet){ |
790 |
|
const pat::Jet& jet = (*jets)[ijet]; |
791 |
< |
|
792 |
< |
if(jet.genJet()){ |
793 |
< |
if(fabs(genjet.pt()-jet.genJet()->pt())<0.0001 && |
794 |
< |
fabs(genjet.eta()-jet.genJet()->eta())<0.0001 && |
795 |
< |
(fabs(genjet.phi()-jet.genJet()->phi())<0.0001 || fabs(genjet.phi()-jet.genJet()->phi() - 2.0*TMath::Pi()) < 0.0001 )){ |
791 |
> |
const reco::GenJet* matchedGenJet = (*patjets)[ijet].genJet(); |
792 |
> |
if(matchedGenJet){ |
793 |
> |
// poor man's matching, someone fix please |
794 |
> |
if(fabs(genjet.pt()-matchedGenJet->pt())<0.0001 && |
795 |
> |
fabs(genjet.eta()-matchedGenJet->eta())<0.0001 && |
796 |
> |
(fabs(genjet.phi()-matchedGenJet->phi())<0.0001 || fabs(genjet.phi()-matchedGenJet->phi() - 2.0*TMath::Pi()) < 0.0001 )){ |
797 |
|
|
798 |
|
jets_.genmatchindex[jets_.ngen] = (int)ijet; |
799 |
|
jets_.gendphijt[jets_.ngen] = reco::deltaPhi(jet.phi(),genjet.phi()); |
810 |
|
|
811 |
|
} |
812 |
|
t->Fill(); |
813 |
+ |
memset(&jets_,0,sizeof jets_); |
814 |
|
} |
815 |
|
|
816 |
|
|
910 |
|
} |
911 |
|
|
912 |
|
|
913 |
+ |
int |
914 |
+ |
HiInclusiveJetAnalyzer::getPFJetMuon(const pat::Jet& pfJet, const reco::PFCandidateCollection *pfCandidateColl) |
915 |
+ |
{ |
916 |
+ |
|
917 |
+ |
int pfMuonIndex = -1; |
918 |
+ |
float ptMax = 0.; |
919 |
+ |
|
920 |
+ |
|
921 |
+ |
for(unsigned icand=0;icand<pfCandidateColl->size(); icand++) { |
922 |
+ |
const reco::PFCandidate pfCandidate = pfCandidateColl->at(icand); |
923 |
+ |
|
924 |
+ |
int id = pfCandidate.particleId(); |
925 |
+ |
if(abs(id) != 3) continue; |
926 |
+ |
|
927 |
+ |
if(reco::deltaR(pfJet,pfCandidate)>0.5) continue; |
928 |
+ |
|
929 |
+ |
double pt = pfCandidate.pt(); |
930 |
+ |
if(pt>ptMax){ |
931 |
+ |
ptMax = pt; |
932 |
+ |
pfMuonIndex = (int) icand; |
933 |
+ |
} |
934 |
+ |
} |
935 |
+ |
|
936 |
+ |
return pfMuonIndex; |
937 |
+ |
|
938 |
+ |
} |
939 |
|
|
940 |
+ |
|
941 |
+ |
double |
942 |
+ |
HiInclusiveJetAnalyzer::getPtRel(const reco::PFCandidate lep, const pat::Jet& jet ) |
943 |
+ |
{ |
944 |
+ |
|
945 |
+ |
float lj_x = jet.p4().px(); |
946 |
+ |
float lj_y = jet.p4().py(); |
947 |
+ |
float lj_z = jet.p4().pz(); |
948 |
+ |
|
949 |
+ |
// absolute values squared |
950 |
+ |
float lj2 = lj_x*lj_x+lj_y*lj_y+lj_z*lj_z; |
951 |
+ |
float lep2 = lep.px()*lep.px()+lep.py()*lep.py()+lep.pz()*lep.pz(); |
952 |
+ |
|
953 |
+ |
// projection vec(mu) to lepjet axis |
954 |
+ |
float lepXlj = lep.px()*lj_x+lep.py()*lj_y+lep.pz()*lj_z; |
955 |
+ |
|
956 |
+ |
// absolute value squared and normalized |
957 |
+ |
float pLrel2 = lepXlj*lepXlj/lj2; |
958 |
+ |
|
959 |
+ |
// lep2 = pTrel2 + pLrel2 |
960 |
+ |
float pTrel2 = lep2-pLrel2; |
961 |
+ |
|
962 |
+ |
return (pTrel2 > 0) ? std::sqrt(pTrel2) : 0.0; |
963 |
+ |
} |
964 |
+ |
|
965 |
+ |
// Recursive function, but this version gets called only the first time |
966 |
+ |
void |
967 |
+ |
HiInclusiveJetAnalyzer::saveDaughters(const reco::GenParticle &gen){ |
968 |
+ |
|
969 |
+ |
for(unsigned i=0;i<gen.numberOfDaughters();i++){ |
970 |
+ |
const reco::Candidate & daughter = *gen.daughter(i); |
971 |
+ |
double daughterPt = daughter.pt(); |
972 |
+ |
if(daughterPt<1.) continue; |
973 |
+ |
double daughterEta = daughter.eta(); |
974 |
+ |
if(fabs(daughterEta)>3.) continue; |
975 |
+ |
int daughterPdgId = daughter.pdgId(); |
976 |
+ |
int daughterStatus = daughter.status(); |
977 |
+ |
// Special case when b->b+string, both b and string contain all daughters, so only take the string |
978 |
+ |
if(gen.pdgId()==daughterPdgId && gen.status()==3 && daughterStatus==2) continue; |
979 |
+ |
|
980 |
+ |
// cheesy way of finding strings which were already used |
981 |
+ |
if(daughter.pdgId()==92){ |
982 |
+ |
for(unsigned ist=0;ist<usedStringPts.size();ist++){ |
983 |
+ |
if(fabs(daughter.pt() - usedStringPts[ist]) < 0.0001) return; |
984 |
+ |
} |
985 |
+ |
usedStringPts.push_back(daughter.pt()); |
986 |
+ |
} |
987 |
+ |
jets_.bJetIndex[jets_.bMult] = jets_.nref; |
988 |
+ |
jets_.bStatus[jets_.bMult] = daughterStatus; |
989 |
+ |
jets_.bVx[jets_.bMult] = daughter.vx(); |
990 |
+ |
jets_.bVy[jets_.bMult] = daughter.vy(); |
991 |
+ |
jets_.bVz[jets_.bMult] = daughter.vz(); |
992 |
+ |
jets_.bPt[jets_.bMult] = daughterPt; |
993 |
+ |
jets_.bEta[jets_.bMult] = daughterEta; |
994 |
+ |
jets_.bPhi[jets_.bMult] = daughter.phi(); |
995 |
+ |
jets_.bPdg[jets_.bMult] = daughterPdgId; |
996 |
+ |
jets_.bChg[jets_.bMult] = daughter.charge(); |
997 |
+ |
jets_.bMult++; |
998 |
+ |
saveDaughters(daughter); |
999 |
+ |
} |
1000 |
+ |
} |
1001 |
+ |
|
1002 |
+ |
// This version called for all subsequent calls |
1003 |
+ |
void |
1004 |
+ |
HiInclusiveJetAnalyzer::saveDaughters(const reco::Candidate &gen){ |
1005 |
+ |
|
1006 |
+ |
for(unsigned i=0;i<gen.numberOfDaughters();i++){ |
1007 |
+ |
const reco::Candidate & daughter = *gen.daughter(i); |
1008 |
+ |
double daughterPt = daughter.pt(); |
1009 |
+ |
if(daughterPt<1.) continue; |
1010 |
+ |
double daughterEta = daughter.eta(); |
1011 |
+ |
if(fabs(daughterEta)>3.) continue; |
1012 |
+ |
int daughterPdgId = daughter.pdgId(); |
1013 |
+ |
int daughterStatus = daughter.status(); |
1014 |
+ |
// Special case when b->b+string, both b and string contain all daughters, so only take the string |
1015 |
+ |
if(gen.pdgId()==daughterPdgId && gen.status()==3 && daughterStatus==2) continue; |
1016 |
+ |
|
1017 |
+ |
// cheesy way of finding strings which were already used |
1018 |
+ |
if(daughter.pdgId()==92){ |
1019 |
+ |
for(unsigned ist=0;ist<usedStringPts.size();ist++){ |
1020 |
+ |
if(fabs(daughter.pt() - usedStringPts[ist]) < 0.0001) return; |
1021 |
+ |
} |
1022 |
+ |
usedStringPts.push_back(daughter.pt()); |
1023 |
+ |
} |
1024 |
+ |
|
1025 |
+ |
jets_.bJetIndex[jets_.bMult] = jets_.nref; |
1026 |
+ |
jets_.bStatus[jets_.bMult] = daughterStatus; |
1027 |
+ |
jets_.bVx[jets_.bMult] = daughter.vx(); |
1028 |
+ |
jets_.bVy[jets_.bMult] = daughter.vy(); |
1029 |
+ |
jets_.bVz[jets_.bMult] = daughter.vz(); |
1030 |
+ |
jets_.bPt[jets_.bMult] = daughterPt; |
1031 |
+ |
jets_.bEta[jets_.bMult] = daughterEta; |
1032 |
+ |
jets_.bPhi[jets_.bMult] = daughter.phi(); |
1033 |
+ |
jets_.bPdg[jets_.bMult] = daughterPdgId; |
1034 |
+ |
jets_.bChg[jets_.bMult] = daughter.charge(); |
1035 |
+ |
jets_.bMult++; |
1036 |
+ |
saveDaughters(daughter); |
1037 |
+ |
} |
1038 |
+ |
} |
1039 |
|
|
1040 |
|
DEFINE_FWK_MODULE(HiInclusiveJetAnalyzer); |