18 |
|
#include "FWCore/ParameterSet/interface/ParameterSet.h" |
19 |
|
#include "DataFormats/GeometryVector/interface/GlobalPoint.h" |
20 |
|
#include "Geometry/Records/interface/CaloGeometryRecord.h" |
21 |
+ |
#include "DataFormats/Math/interface/deltaPhi.h" |
22 |
|
|
23 |
|
// Jet and vertex functions |
24 |
|
#include "DataFormats/JetReco/interface/CaloJet.h" |
31 |
|
#include "DataFormats/VertexReco/interface/Vertex.h" |
32 |
|
#include "DataFormats/VertexReco/interface/VertexFwd.h" |
33 |
|
#include "DataFormats/BTauReco/interface/JetTag.h" |
34 |
+ |
//#include "RecoVertex/PrimaryVertexProducer/interface/VertexHigherPtSquared.h" |
35 |
|
|
36 |
< |
#include "DataFormats/Math/interface/deltaPhi.h" |
36 |
> |
//GenParticles |
37 |
> |
#include "DataFormats/HepMCCandidate/interface/GenParticle.h" |
38 |
|
|
39 |
|
// Need to get correctors |
40 |
|
#include "JetMETCorrections/Objects/interface/JetCorrector.h" |
80 |
|
virtual void endJob() ; |
81 |
|
|
82 |
|
bool triggerDecision(edm::Handle<edm::TriggerResults> &hltR, int iTrigger); |
83 |
+ |
double sumPtSquared(const Vertex & v); |
84 |
|
|
85 |
|
// ----------member data --------------------------- |
86 |
|
|
88 |
|
edm::InputTag GenJetHandle_; |
89 |
|
edm::InputTag PrimaryVtxHandle_; |
90 |
|
edm::InputTag triggerResultsTag_; |
91 |
< |
|
91 |
> |
edm::InputTag genParticles_; |
92 |
|
|
93 |
|
// Counting variables // |
94 |
|
|
95 |
< |
int eventNumber, runNumber, lumiSection; |
95 |
> |
int eventNumber, runNumber, lumiSection, bunchCross; |
96 |
|
|
97 |
|
TTree* sTree; |
98 |
|
TFile* ntupleFile; |
99 |
|
|
100 |
|
TClonesArray* jet_ak5PF; |
101 |
|
TClonesArray* jetP4_ak5Gen; |
102 |
+ |
TClonesArray* genParticleP4; |
103 |
|
TClonesArray* primaryVtx; |
104 |
|
|
105 |
|
bool doGenJets_; |
106 |
|
bool doPFJets_; |
107 |
|
bool triggerHLT_; |
108 |
< |
|
108 |
> |
bool isRealData; |
109 |
|
string rootfilename; |
110 |
|
|
111 |
|
//Triggers |
123 |
|
PFJetHandle_(iConfig.getUntrackedParameter<edm::InputTag>("PFJetTag")), |
124 |
|
GenJetHandle_(iConfig.getUntrackedParameter<edm::InputTag>("GenJetTag")), |
125 |
|
PrimaryVtxHandle_(iConfig.getUntrackedParameter<edm::InputTag>("PrimaryVtxTag")), |
126 |
+ |
genParticles_(iConfig.getUntrackedParameter<edm::InputTag>("genParticleTag")), |
127 |
|
doGenJets_(iConfig.getUntrackedParameter<bool>("doGenJets")), |
128 |
|
doPFJets_(iConfig.getUntrackedParameter<bool>("doPFJets")), |
129 |
|
triggerHLT_(iConfig.getUntrackedParameter<bool>("triggerHLT")), |
131 |
|
hlTriggerResults_(iConfig.getUntrackedParameter<string>("HLTriggerResults","TriggerResults")), |
132 |
|
hltName_(iConfig.getUntrackedParameter<string>("hltName","HLT")) |
133 |
|
{ |
134 |
< |
//edm::TriggerNames |
129 |
< |
// triggerNames(iConfig); |
134 |
> |
|
135 |
|
|
136 |
|
} |
137 |
|
|
141 |
|
|
142 |
|
} |
143 |
|
|
144 |
+ |
double MPIntuple::sumPtSquared(const Vertex & v) |
145 |
+ |
{ |
146 |
+ |
double sum = 0.; |
147 |
+ |
double pT; |
148 |
+ |
for (Vertex::trackRef_iterator it = v.tracks_begin(); it != v.tracks_end(); it++) { |
149 |
+ |
pT = (**it).pt(); |
150 |
+ |
double epT=(**it).ptError(); pT=pT>epT ? pT-epT : 0; |
151 |
+ |
|
152 |
+ |
sum += pT*pT; |
153 |
+ |
} |
154 |
+ |
return sum; |
155 |
+ |
} |
156 |
+ |
|
157 |
|
// |
158 |
|
// member functions |
159 |
|
// |
165 |
|
eventNumber = iEvent.id().event(); |
166 |
|
runNumber = iEvent.id().run(); |
167 |
|
lumiSection = (unsigned int)iEvent.getLuminosityBlock().luminosityBlock(); |
168 |
< |
// isMC = iEvent.???(); |
169 |
< |
|
170 |
< |
int PFcount = 0; |
171 |
< |
int Gencount = 0; |
172 |
< |
int Vtxcount = 0; |
168 |
> |
bunchCross = (unsigned int)iEvent.bunchCrossing(); |
169 |
> |
isRealData = iEvent.isRealData(); |
170 |
> |
|
171 |
> |
int PFcount = 0; |
172 |
> |
int Gencount = 0; |
173 |
> |
int GenPcount = 0; |
174 |
> |
int Vtxcount = 0; |
175 |
|
|
176 |
|
if(doPFJets_){ |
177 |
|
|
238 |
|
jetCon->SetJetCorr(2, scale2); |
239 |
|
jetCon->SetJetCorr(3, scale3); |
240 |
|
|
221 |
– |
//if(( (lumiSection==666690) && (eventNumber==2439)) || |
222 |
– |
// ( (lumiSection==666702) && (eventNumber==3629)) || |
223 |
– |
// ( (lumiSection==666702) && (eventNumber==3681)) || |
224 |
– |
// ( (lumiSection==666709) && (eventNumber==4320)) || |
225 |
– |
// ( (lumiSection==666709) && (eventNumber==4328)) || |
226 |
– |
// ( (lumiSection==666724) && (eventNumber==5855)) || |
227 |
– |
// ( (lumiSection==666742) && (eventNumber==7642)) || |
228 |
– |
// ( (lumiSection==666754) && (eventNumber==8869)) || |
229 |
– |
// ( (lumiSection==666670) && (eventNumber==466)) || |
230 |
– |
// ( (lumiSection==666675) && (eventNumber==919)) || |
231 |
– |
// ( (lumiSection==666718) && (eventNumber==5235))) |
232 |
– |
//{ |
233 |
– |
// std::cout << runNumber << " " << lumiSection << " " << eventNumber << "\n"; |
234 |
– |
// std::cout << "(" << myJet.pt() << "," << myJet.eta() << "," << myJet.phi() << ", [" << scale2 << ", " << scale3 << "]) " << std::endl; |
235 |
– |
// std::cout << "(" << jetCon->Pt(3) << "," << jetCon->P4(3).Eta() << "," << jetCon->P4(3).Phi() << "," << ", [" << jetCon->JetCorr(2) << ", " << jetCon->JetCorr(3) << "]) " << std::endl; |
236 |
– |
//} |
237 |
– |
|
238 |
– |
|
241 |
|
++PFcount; |
242 |
|
} |
243 |
|
} |
244 |
|
} |
245 |
|
|
246 |
< |
if(doGenJets_){ |
246 |
> |
if(!isRealData){ |
247 |
|
|
248 |
+ |
Handle<reco::GenParticleCollection > genpcoll; |
249 |
+ |
iEvent.getByLabel(genParticles_, genpcoll); |
250 |
+ |
|
251 |
|
Handle<reco::GenJetCollection> GenJets; |
252 |
|
iEvent.getByLabel(GenJetHandle_, GenJets); |
253 |
|
|
264 |
|
|
265 |
|
} |
266 |
|
} |
262 |
– |
} |
263 |
– |
|
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 |
+ |
|
279 |
+ |
} |
280 |
+ |
|
281 |
+ |
|
282 |
|
Handle<reco::VertexCollection> primaryVtcs; |
283 |
|
iEvent.getByLabel(PrimaryVtxHandle_, primaryVtcs); |
284 |
< |
|
284 |
> |
|
285 |
|
for(VertexCollection::const_iterator vtx_iter = primaryVtcs->begin(); vtx_iter!= primaryVtcs->end(); ++vtx_iter){ |
286 |
|
|
287 |
|
reco::Vertex myVtx = reco::Vertex(*vtx_iter); |
288 |
< |
|
288 |
> |
if(!myVtx.isValid() || myVtx.isFake()) continue; |
289 |
|
TCPrimaryVtx* vtxCon = new ((*primaryVtx)[Vtxcount]) TCPrimaryVtx; |
290 |
|
|
291 |
|
vtxCon->SetPosition(myVtx.x(), myVtx.y(), myVtx.z()); |
292 |
|
vtxCon->SetNDof(myVtx.ndof()); |
293 |
|
vtxCon->SetChi2(myVtx.chi2()); |
294 |
< |
|
294 |
> |
//placeholders |
295 |
> |
// vtxCon->SetNTracks(myVtx.tracksSize()); |
296 |
> |
// vtxCon->SetTracksSumPtSquared(sumPtSquared(*myVtx)); |
297 |
> |
int test1 = myVtx.tracksSize(); |
298 |
> |
double test2 = sumPtSquared(myVtx); |
299 |
|
++Vtxcount; |
300 |
|
|
301 |
|
} |
312 |
|
const TriggerNames & triggerNames = iEvent.triggerNames(*hltR); |
313 |
|
hlNames=triggerNames.triggerNames(); |
314 |
|
|
315 |
< |
string MPI_TriggerNames[32] = {"HLT_PixelTracks_Multiplicity70", "HLT_MinBiasBSC_NoBPTX", "HLT_PixelTracks_Multiplicity40","HLT_L1Tech_HCAL_HF", "HLT_IsoTrackHB_8E29", "HLT_IsoTrackHE_8E29", "HLT_L1Tech_RPC_TTU_RBst1_collisions", "HLT_L1_BscMinBiasOR_BptxPlusORMinus", "HLT_L1Tech_BSC_halo_forPhysicsBackground", "HLT_L1Tech_BSC_HighMultiplicity", "HLT_MinBiasPixel_DoubleIsoTrack5", "HLT_MinBiasPixel_DoubleTrack", "HLT_MinBiasPixel_SingleTrack", "HLT_ZeroBiasPixel_SingleTrack", "HLT_MinBiasBSC", "HLT_StoppedHSCP_8E29", "HLT_Jet15U_HcalNoiseFiltered", "HLT_QuadJet15U", "HLT_DiJetAve30U_8E29", "HLT_DiJetAve15U_8E29", "HLT_FwdJet20U", "HLT_Jet50U", "HLT_Jet30U", "HLT_Jet15U", "HLT_BTagMu_Jet10U", "HLT_DoubleJet15U_ForwardBackward", "HLT_BTagIP_Jet50U", "HLT_DoubleLooseIsoTau15", "HLT_SingleLooseIsoTau20", "HLT_HT100U", "HLT_MET100", "HLT_MET45"}; |
315 |
> |
string MPI_TriggerNames[] = {"HLT_PixelTracks_Multiplicity70", "HLT_MinBiasBSC_NoBPTX", "HLT_PixelTracks_Multiplicity40","HLT_L1Tech_HCAL_HF", "HLT_IsoTrackHB_8E29", "HLT_IsoTrackHE_8E29", "HLT_L1Tech_RPC_TTU_RBst1_collisions", "HLT_L1_BscMinBiasOR_BptxPlusORMinus", "HLT_L1Tech_BSC_halo_forPhysicsBackground", "HLT_L1Tech_BSC_HighMultiplicity", "HLT_MinBiasPixel_DoubleIsoTrack5", "HLT_MinBiasPixel_DoubleTrack", "HLT_MinBiasPixel_SingleTrack", "HLT_ZeroBiasPixel_SingleTrack", "HLT_MinBiasBSC", "HLT_StoppedHSCP_8E29", "HLT_Jet15U_HcalNoiseFiltered", "HLT_QuadJet15U", "HLT_DiJetAve30U_8E29", "HLT_DiJetAve15U_8E29", "HLT_FwdJet20U", "HLT_Jet50U", "HLT_Jet30U", "HLT_Jet15U", "HLT_BTagMu_Jet10U", "HLT_DoubleJet15U_ForwardBackward", "HLT_BTagIP_Jet50U", "HLT_DoubleLooseIsoTau15", "HLT_SingleLooseIsoTau20", "HLT_HT100U", "HLT_MET100", "HLT_MET45"}; |
316 |
|
|
317 |
|
bool triggerPassed = false; |
318 |
|
triggerStatus = 0x0; |
355 |
|
|
356 |
|
jet_ak5PF = new TClonesArray("TCJet"); |
357 |
|
jetP4_ak5Gen = new TClonesArray("TLorentzVector"); |
358 |
+ |
genParticleP4 = new TClonesArray("TLorentzVector"); |
359 |
|
primaryVtx = new TClonesArray("TCPrimaryVtx"); |
360 |
|
|
361 |
|
sTree->Branch("jet_ak5PF",&jet_ak5PF, 6400, 0); |
362 |
|
sTree->Branch("jetP4_ak5Gen",&jetP4_ak5Gen, 6400, 0); |
363 |
+ |
sTree->Branch("genParticleP4",&genParticleP4, 6400, 0); |
364 |
|
sTree->Branch("primaryVtx",&primaryVtx, 6400, 0); |
365 |
|
|
366 |
|
sTree->Branch("eventNumber",&eventNumber, "eventNumber/I"); |
367 |
|
sTree->Branch("runNumber",&runNumber, "runNumber/I"); |
368 |
|
sTree->Branch("lumiSection",&lumiSection, "lumiSection/I"); |
369 |
|
sTree->Branch("triggerStatus",&triggerStatus, "triggerStatus/i"); |
370 |
< |
// sTree->Branch("isMC",&isMC,"isMC/O"); |
371 |
< |
|
370 |
> |
sTree->Branch("isRealData",&isRealData, "isRealData/i"); |
371 |
> |
sTree->Branch("bunchCross",&bunchCross, "bunchCross/i"); |
372 |
|
|
373 |
|
} |
374 |
|
|