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")), |
166 |
|
|
167 |
|
int PFcount = 0; |
168 |
|
int Gencount = 0; |
173 |
– |
int GenPcount = 0; |
169 |
|
int Vtxcount = 0; |
170 |
|
|
171 |
|
if(doPFJets_){ |
240 |
|
|
241 |
|
if(!isRealData){ |
242 |
|
|
248 |
– |
Handle<reco::GenParticleCollection > genpcoll; |
249 |
– |
iEvent.getByLabel(genParticles_, genpcoll); |
243 |
|
|
244 |
|
Handle<reco::GenJetCollection> GenJets; |
245 |
|
iEvent.getByLabel(GenJetHandle_, GenJets); |
256 |
|
++Gencount; |
257 |
|
|
258 |
|
} |
259 |
< |
} |
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 |
< |
|
259 |
> |
} |
260 |
|
} |
261 |
|
|
262 |
|
|
272 |
|
vtxCon->SetPosition(myVtx.x(), myVtx.y(), myVtx.z()); |
273 |
|
vtxCon->SetNDof(myVtx.ndof()); |
274 |
|
vtxCon->SetChi2(myVtx.chi2()); |
275 |
< |
//placeholders |
276 |
< |
// vtxCon->SetNTracks(myVtx.tracksSize()); |
296 |
< |
// vtxCon->SetTracksSumPtSquared(sumPtSquared(*myVtx)); |
297 |
< |
int test1 = myVtx.tracksSize(); |
298 |
< |
double test2 = sumPtSquared(myVtx); |
275 |
> |
vtxCon->SetNTrks(myVtx.tracksSize()); |
276 |
> |
vtxCon->SetSumPt2Trks(sumPtSquared(myVtx)); |
277 |
|
++Vtxcount; |
278 |
|
|
279 |
|
} |
315 |
|
} |
316 |
|
// cout<< "total status: "<<hex<<triggerStatus<<endl; |
317 |
|
|
318 |
< |
if(PFcount > 0 || Gencount > 0 || Vtxcount > 0); sTree -> Fill(); |
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"); |