34 |
|
//#include "RecoVertex/PrimaryVertexProducer/interface/VertexHigherPtSquared.h" |
35 |
|
|
36 |
|
//GenParticles |
37 |
< |
#include "DataFormats/HepMCCandidate/interface/GenParticle.h" |
37 |
> |
//#include "DataFormats/HepMCCandidate/interface/GenParticle.h" |
38 |
|
|
39 |
|
// Need to get correctors |
40 |
|
#include "JetMETCorrections/Objects/interface/JetCorrector.h" |
88 |
|
edm::InputTag GenJetHandle_; |
89 |
|
edm::InputTag PrimaryVtxHandle_; |
90 |
|
edm::InputTag triggerResultsTag_; |
91 |
– |
edm::InputTag genParticles_; |
92 |
– |
|
91 |
|
// Counting variables // |
92 |
|
|
93 |
|
int eventNumber, runNumber, lumiSection, bunchCross; |
97 |
|
|
98 |
|
TClonesArray* jet_ak5PF; |
99 |
|
TClonesArray* jetP4_ak5Gen; |
102 |
– |
TClonesArray* genParticleP4; |
100 |
|
TClonesArray* primaryVtx; |
101 |
|
|
102 |
|
bool doGenJets_; |
120 |
|
PFJetHandle_(iConfig.getUntrackedParameter<edm::InputTag>("PFJetTag")), |
121 |
|
GenJetHandle_(iConfig.getUntrackedParameter<edm::InputTag>("GenJetTag")), |
122 |
|
PrimaryVtxHandle_(iConfig.getUntrackedParameter<edm::InputTag>("PrimaryVtxTag")), |
126 |
– |
genParticles_(iConfig.getUntrackedParameter<edm::InputTag>("genParticleTag")), |
123 |
|
doGenJets_(iConfig.getUntrackedParameter<bool>("doGenJets")), |
124 |
|
doPFJets_(iConfig.getUntrackedParameter<bool>("doPFJets")), |
125 |
|
triggerHLT_(iConfig.getUntrackedParameter<bool>("triggerHLT")), |
158 |
|
void MPIntuple::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup) |
159 |
|
{ |
160 |
|
|
161 |
+ |
|
162 |
|
eventNumber = iEvent.id().event(); |
163 |
|
runNumber = iEvent.id().run(); |
164 |
|
lumiSection = (unsigned int)iEvent.getLuminosityBlock().luminosityBlock(); |
167 |
|
|
168 |
|
int PFcount = 0; |
169 |
|
int Gencount = 0; |
173 |
– |
int GenPcount = 0; |
170 |
|
int Vtxcount = 0; |
171 |
|
|
172 |
|
if(doPFJets_){ |
181 |
|
const reco::JetTagCollection & bTags2 = *(bTagHandle2.product()); |
182 |
|
reco::JetTagCollection::const_iterator jet_it_2; |
183 |
|
|
184 |
+ |
|
185 |
|
const JetCorrector* correctorL2 = JetCorrector::getJetCorrector ("ak5PFL2Relative",iSetup); |
186 |
|
const JetCorrector* correctorL3 = JetCorrector::getJetCorrector ("ak5PFL3Absolute",iSetup); |
187 |
|
|
193 |
|
for(PFJetCollection::const_iterator jet_iter = PFJets->begin(); jet_iter!= PFJets->end(); ++jet_iter){ |
194 |
|
|
195 |
|
reco::PFJet myJet = reco::PFJet(*jet_iter); |
199 |
– |
|
200 |
– |
// float scale2 = correctorL2->correction(myJet); |
201 |
– |
// float scale3 = correctorL3->correction(myJet); |
202 |
– |
|
196 |
|
if(myJet.et() > 5){ |
197 |
|
|
198 |
|
for(jet_it_1 = bTags1.begin(); jet_it_1 != bTags1.end(); jet_it_1++) { |
238 |
|
|
239 |
|
if(!isRealData){ |
240 |
|
|
248 |
– |
Handle<reco::GenParticleCollection > genpcoll; |
249 |
– |
iEvent.getByLabel(genParticles_, genpcoll); |
241 |
|
|
242 |
|
Handle<reco::GenJetCollection> GenJets; |
243 |
|
iEvent.getByLabel(GenJetHandle_, GenJets); |
254 |
|
++Gencount; |
255 |
|
|
256 |
|
} |
257 |
< |
} |
267 |
< |
|
268 |
< |
for(reco::GenParticleCollection::const_iterator iGen = genpcoll->begin(); iGen!=genpcoll->end(); ++iGen) { |
269 |
< |
|
270 |
< |
reco::GenParticle p = reco::GenParticle(*iGen); |
271 |
< |
|
272 |
< |
new ((*genParticleP4)[GenPcount]) TLorentzVector(p.px(), p.py(), p.pz(), p.energy()); |
273 |
< |
|
274 |
< |
++GenPcount; |
275 |
< |
|
276 |
< |
} |
277 |
< |
|
278 |
< |
|
257 |
> |
} |
258 |
|
} |
259 |
|
|
260 |
|
|
270 |
|
vtxCon->SetPosition(myVtx.x(), myVtx.y(), myVtx.z()); |
271 |
|
vtxCon->SetNDof(myVtx.ndof()); |
272 |
|
vtxCon->SetChi2(myVtx.chi2()); |
273 |
< |
//placeholders |
274 |
< |
// vtxCon->SetNTracks(myVtx.tracksSize()); |
296 |
< |
// vtxCon->SetTracksSumPtSquared(sumPtSquared(*myVtx)); |
297 |
< |
int test1 = myVtx.tracksSize(); |
298 |
< |
double test2 = sumPtSquared(myVtx); |
273 |
> |
vtxCon->SetNTrks(myVtx.tracksSize()); |
274 |
> |
vtxCon->SetSumPt2Trks(sumPtSquared(myVtx)); |
275 |
|
++Vtxcount; |
276 |
|
|
277 |
|
} |
312 |
|
} |
313 |
|
} |
314 |
|
// cout<< "total status: "<<hex<<triggerStatus<<endl; |
315 |
< |
|
316 |
< |
if(PFcount > 0 || Gencount > 0 || Vtxcount > 0); sTree -> Fill(); |
315 |
> |
|
316 |
> |
|
317 |
> |
|
318 |
> |
if((PFcount > 3 || Gencount > 3) && Vtxcount > 0) sTree -> Fill(); |
319 |
|
|
320 |
|
jet_ak5PF->Clear(); |
321 |
|
primaryVtx->Clear(); |
333 |
|
|
334 |
|
jet_ak5PF = new TClonesArray("TCJet"); |
335 |
|
jetP4_ak5Gen = new TClonesArray("TLorentzVector"); |
358 |
– |
genParticleP4 = new TClonesArray("TLorentzVector"); |
336 |
|
primaryVtx = new TClonesArray("TCPrimaryVtx"); |
337 |
|
|
338 |
|
sTree->Branch("jet_ak5PF",&jet_ak5PF, 6400, 0); |
339 |
|
sTree->Branch("jetP4_ak5Gen",&jetP4_ak5Gen, 6400, 0); |
363 |
– |
sTree->Branch("genParticleP4",&genParticleP4, 6400, 0); |
340 |
|
sTree->Branch("primaryVtx",&primaryVtx, 6400, 0); |
341 |
|
|
342 |
|
sTree->Branch("eventNumber",&eventNumber, "eventNumber/I"); |