ViewVC Help

View File | Revision Log | Show Annotations | Root Listing

root/cvsroot/UserCode/MPIAnalyzer/src/MPIntuple.cc

Comparing UserCode/MPIAnalyzer/src/MPIntuple.cc (file contents):
Revision 1.13 by naodell, Fri Jul 23 12:45:32 2010 UTC vs.
Revision 1.18 by naodell, Wed Oct 27 14:21:20 2010 UTC

#	Line 20 | Line 20
20		#include "Geometry/Records/interface/CaloGeometryRecord.h"
21		#include "DataFormats/Math/interface/deltaPhi.h"
22
23	<	// Jet and vertex functions
23	>	// Libraries for objects
24		#include "DataFormats/JetReco/interface/CaloJet.h"
25		#include "DataFormats/JetReco/interface/CaloJetCollection.h"
26		#include "DataFormats/JetReco/interface/PFJet.h"
#	Line 31 | Line 31
31		#include "DataFormats/VertexReco/interface/Vertex.h"
32		#include "DataFormats/VertexReco/interface/VertexFwd.h"
33		#include "DataFormats/BTauReco/interface/JetTag.h"
34	+	#include "DataFormats/TrackReco/interface/Track.h"
35	+	#include "DataFormats/BeamSpot/interface/BeamSpot.h"
36	+	#include "SimDataFormats/GeneratorProducts/interface/GenEventInfoProduct.h"
37	+
38		//#include "RecoVertex/PrimaryVertexProducer/interface/VertexHigherPtSquared.h"
39
40		//GenParticles
#	Line 50 | Line 54
54
55		//Root  stuff
56		#include "TROOT.h"
57	+	#include "TH1.h"
58	+	#include "TH2.h"
59	+	#include "TProfile.h"
60		#include "TFile.h"
61		#include "TTree.h"
62		#include "TString.h"
#	Line 91 | Line 98 | class MPIntuple : public edm::EDAnalyzer
98		//  Counting variables   //
99
100		int eventNumber, runNumber, lumiSection, bunchCross;
101	<
101	>	double ptHat, qScale;
102	>
103		TTree* sTree;
104		TFile* ntupleFile;
105
#	Line 112 | Line 120 | class MPIntuple : public edm::EDAnalyzer
120		vector<string>  hlNames;
121		unsigned int triggerStatus;
122
123	+	//Histograms
124	+	TH1F * h1_nAssociatedTracks;
125	+	TH1F * h1_trackZ;
126	+	TH1F * h1_trackDxy;
127	+	TH1F * h1_allTrackDeltaZ_PV;
128	+	TH1F * h1_allTrackDeltaZ_LJ;
129	+	TH1F * h1_ptHat;
130	+	TH2F * h2_nAssTracksVsJetPt;
131	+	TProfile * p1_nVtcs;
132
116	–	};
133
134	<	MPIntuple::MPIntuple(const edm::ParameterSet& iConfig):
134	>	};
135
136	<	PFJetHandle_(iConfig.getUntrackedParameter<edm::InputTag>("PFJetTag")),
121	<	GenJetHandle_(iConfig.getUntrackedParameter<edm::InputTag>("GenJetTag")),
122	<	PrimaryVtxHandle_(iConfig.getUntrackedParameter<edm::InputTag>("PrimaryVtxTag")),
123	<	doGenJets_(iConfig.getUntrackedParameter<bool>("doGenJets")),
124	<	doPFJets_(iConfig.getUntrackedParameter<bool>("doPFJets")),
125	<	triggerHLT_(iConfig.getUntrackedParameter<bool>("triggerHLT")),
126	<	rootfilename(iConfig.getUntrackedParameter<string>("rootfilename")),
127	<	hlTriggerResults_(iConfig.getUntrackedParameter<string>("HLTriggerResults","TriggerResults")),
128	<	hltName_(iConfig.getUntrackedParameter<string>("hltName","HLT"))
136	>	MPIntuple::MPIntuple(const edm::ParameterSet& iConfig)
137		{
138	<
139	<
138	>	PFJetHandle_ = iConfig.getUntrackedParameter<edm::InputTag>("PFJetTag");
139	>	GenJetHandle_ = iConfig.getUntrackedParameter<edm::InputTag>("GenJetTag");
140	>	PrimaryVtxHandle_ = iConfig.getUntrackedParameter<edm::InputTag>("PrimaryVtxTag");
141	>	doGenJets_ = iConfig.getUntrackedParameter<bool>("doGenJets");
142	>	doPFJets_ = iConfig.getUntrackedParameter<bool>("doPFJets");
143	>	triggerHLT_ = iConfig.getUntrackedParameter<bool>("triggerHLT");
144	>	rootfilename = iConfig.getUntrackedParameter<string>("rootfilename");
145	>	hlTriggerResults_ = iConfig.getUntrackedParameter<string>("HLTriggerResults","TriggerResults");
146	>	hltName_ = iConfig.getUntrackedParameter<string>("hltName","RECO");
147		}
148
134	–
149		MPIntuple::~MPIntuple()
150		{
151
152		}
153
140	–	double MPIntuple::sumPtSquared(const Vertex & v)
141	–	{
142	–	double sum = 0.;
143	–	double pT;
144	–	for (Vertex::trackRef_iterator it = v.tracks_begin(); it != v.tracks_end(); it++) {
145	–	pT = (**it).pt();
146	–	double epT=(**it).ptError(); pT=pT>epT ? pT-epT : 0;
147	–
148	–	sum += pT*pT;
149	–	}
150	–	return sum;
151	–	}
152	–
154		//
155		// member functions
156		//
#	Line 164 | Line 165 | void MPIntuple::analyze(const edm::Event
165		bunchCross   = (unsigned int)iEvent.bunchCrossing();
166		isRealData   = iEvent.isRealData();
167
168	<	int PFcount   = 0;
169	<	int Gencount  = 0;
170	<	int Vtxcount  = 0;
168	>
169	>	edm::Handle<reco::BeamSpot> beamSpotHandle;
170	>	iEvent.getByLabel("offlineBeamSpot", beamSpotHandle);
171	>	reco::BeamSpot vertexBeamSpot = *beamSpotHandle;
172	>
173	>	int pfCount   = 0;
174	>	int genCount  = 0;
175	>	int vtxCount  = 0;
176	>	float ljVertexZ = -999;
177	>	float primaryVertexZ = -999;
178	>
179	>	//////////////////////////
180	>	//Get vertex information//
181	>	//////////////////////////
182	>
183	>	Handle<reco::VertexCollection> primaryVtcs;
184	>	iEvent.getByLabel(PrimaryVtxHandle_, primaryVtcs);
185	>
186	>	for(VertexCollection::const_iterator vtx_iter = primaryVtcs->begin(); vtx_iter!= primaryVtcs->end(); ++vtx_iter){
187	>
188	>	reco::Vertex myVtx = reco::Vertex(*vtx_iter);
189	>	if(!myVtx.isValid() || myVtx.isFake()) continue;
190	>	TCPrimaryVtx* vtxCon = new ((*primaryVtx)[vtxCount]) TCPrimaryVtx;
191	>
192	>	vtxCon->SetPosition(myVtx.x(), myVtx.y(), myVtx.z());
193	>	vtxCon->SetNDof(myVtx.ndof());
194	>	vtxCon->SetChi2(myVtx.chi2());
195	>	vtxCon->SetNTrks(myVtx.tracksSize());
196	>	vtxCon->SetSumPt2Trks(sumPtSquared(myVtx));
197	>
198	>	if(vtxCount == 0) primaryVertexZ = myVtx.z();
199	>
200	>	++vtxCount;
201	>
202	>	}
203	>
204	>	p1_nVtcs->Fill(runNumber, vtxCount);
205	>
206	>	///////////////////////
207	>	//get jet information//
208	>	///////////////////////
209
210		if(doPFJets_){
211
#	Line 180 | Line 219 | void MPIntuple::analyze(const edm::Event
219		const reco::JetTagCollection & bTags2 = *(bTagHandle2.product());
220		reco::JetTagCollection::const_iterator jet_it_2;
221
222	+
223		const JetCorrector* correctorL2 = JetCorrector::getJetCorrector ("ak5PFL2Relative",iSetup);
224		const JetCorrector* correctorL3 = JetCorrector::getJetCorrector ("ak5PFL3Absolute",iSetup);
225
226		Handle<reco::PFJetCollection> PFJets;
227		iEvent.getByLabel(PFJetHandle_, PFJets);
228	<
189	<	PFcount = 0;
190	<
228	>
229		for(PFJetCollection::const_iterator jet_iter = PFJets->begin(); jet_iter!= PFJets->end(); ++jet_iter){
230
231		reco::PFJet myJet = reco::PFJet(*jet_iter);
194	–
195	–	//      float scale2 = correctorL2->correction(myJet);
196	–	//      float scale3 = correctorL3->correction(myJet);
197	–
232		if(myJet.et() > 5){
233
234		for(jet_it_1 = bTags1.begin(); jet_it_1 != bTags1.end(); jet_it_1++) {
#	Line 205 | Line 239 | void MPIntuple::analyze(const edm::Event
239		if( (fabs(jet_it_2->first->eta() - myJet.eta()) < .005) && (deltaPhi(jet_it_2->first->phi(),myJet.phi()) < .005)) break;
240		}
241
242	<	TCJet* jetCon = new ((*jet_ak5PF)[PFcount]) TCJet;
242	>	TCJet* jetCon = new ((*jet_ak5PF)[pfCount]) TCJet;
243
244		jetCon->SetP4(myJet.px(), myJet.py(), myJet.pz(), myJet.energy());
245	<	jetCon->SetVtx(myJet.vx(), myJet.vy(), myJet.vz());
245	>	jetCon->SetVtx(-999.0, -999.0, -999.0);
246
247		jetCon->SetChHadFrac(myJet.chargedHadronEnergyFraction());
248		jetCon->SetNeuHadFrac(myJet.neutralHadronEnergyFraction());
#	Line 233 | Line 267 | void MPIntuple::analyze(const edm::Event
267		jetCon->SetJetCorr(2, scale2);
268		jetCon->SetJetCorr(3, scale3);
269
270	<	++PFcount;
270	>	/////////////////////////
271	>	//get associated tracks//
272	>	/////////////////////////
273	>
274	>	const reco::TrackRefVector &tracks = myJet.getTrackRefs();
275	>
276	>	float sumTrackZ, sumTrackIP;
277	>	float meanTrackZ, meanTrackIP;
278	>	int   nAssociatedTracks = -1;
279	>
280	>	sumTrackZ  = sumTrackIP  = 0;
281	>	meanTrackZ = meanTrackIP = -999;
282	>
283	>	if(myJet.pt() > 20 && fabs(myJet.eta()) < 2.4){
284	>
285	>	nAssociatedTracks = 0;
286	>
287	>	for(TrackRefVector::const_iterator iTrack = tracks.begin(); iTrack != tracks.end(); ++iTrack){
288	>
289	>	const reco::Track& myTrack = **iTrack;
290	>
291	>	sumTrackZ += myTrack.vz();
292	>	sumTrackIP += myTrack.dxy(vertexBeamSpot.position());
293	>	++nAssociatedTracks;
294	>
295	>	}
296	>
297	>	meanTrackZ = sumTrackZ/nAssociatedTracks;
298	>	h1_nAssociatedTracks->Fill(nAssociatedTracks);
299	>	h2_nAssTracksVsJetPt->Fill(myJet.pt(), nAssociatedTracks);
300	>
301	>	for(TrackRefVector::const_iterator iTrack = tracks.begin(); iTrack != tracks.end(); ++iTrack){
302	>
303	>	const reco::Track& myTrack = **iTrack;
304	>
305	>	h1_trackZ->Fill(meanTrackZ - myTrack.vz());
306	>	h1_trackDxy->Fill(myTrack.dxy(vertexBeamSpot.position()));
307	>
308	>	}
309	>
310	>	if(pfCount == 0){
311	>
312	>	ljVertexZ = meanTrackZ;
313	>
314	>	}else{
315	>
316	>	h1_allTrackDeltaZ_LJ->Fill(ljVertexZ - meanTrackZ);
317	>
318	>	}
319	>
320	>	h1_allTrackDeltaZ_PV->Fill(primaryVertexZ - meanTrackZ);
321	>
322	>	}
323	>
324	>	jetCon->SetVtx(0, 0, meanTrackZ);
325	>
326	>	++pfCount;
327		}
328		}
329		}
330
331		if(!isRealData){
332
333	+	Handle<GenEventInfoProduct> GenInfoHandle;
334	+	iEvent.getByLabel("generator", GenInfoHandle);
335
336		Handle<reco::GenJetCollection> GenJets;
337		iEvent.getByLabel(GenJetHandle_, GenJets);
338	<
339	<
338	>
339	>	ptHat = qScale = -1;
340	>
341	>	if(GenInfoHandle.isValid()){
342	>
343	>	qScale = GenInfoHandle->qScale();
344	>	ptHat  = (GenInfoHandle->hasBinningValues() ? GenInfoHandle->binningValues()[0] : 0.0);
345	>
346	>	h1_ptHat->Fill(ptHat);
347	>
348	>	}
349	>
350		for(GenJetCollection::const_iterator jet_iter = GenJets->begin(); jet_iter!= GenJets->end(); ++jet_iter){
351
352		reco::GenJet myJet = reco::GenJet(*jet_iter);
353
354		if(myJet.pt() > 5){
355
356	<	new ((*jetP4_ak5Gen)[Gencount]) TLorentzVector(myJet.px(), myJet.py(), myJet.pz(), myJet.energy());
356	>	new ((*jetP4_ak5Gen)[genCount]) TLorentzVector(myJet.px(), myJet.py(), myJet.pz(), myJet.energy());
357
358	<	++Gencount;
358	>	++genCount;
359
360		}
361		}
362		}
363
364	<
365	<	Handle<reco::VertexCollection> primaryVtcs;
366	<	iEvent.getByLabel(PrimaryVtxHandle_, primaryVtcs);
265	<
266	<	for(VertexCollection::const_iterator vtx_iter = primaryVtcs->begin(); vtx_iter!= primaryVtcs->end(); ++vtx_iter){
267	<
268	<	reco::Vertex myVtx = reco::Vertex(*vtx_iter);
269	<	if(!myVtx.isValid() || myVtx.isFake()) continue;
270	<	TCPrimaryVtx* vtxCon = new ((*primaryVtx)[Vtxcount]) TCPrimaryVtx;
271	<
272	<	vtxCon->SetPosition(myVtx.x(), myVtx.y(), myVtx.z());
273	<	vtxCon->SetNDof(myVtx.ndof());
274	<	vtxCon->SetChi2(myVtx.chi2());
275	<	vtxCon->SetNTrks(myVtx.tracksSize());
276	<	vtxCon->SetSumPt2Trks(sumPtSquared(myVtx));
277	<	++Vtxcount;
278	<
279	<	}
280	<
364	>	///////////////////////////
365	>	//get trigger information//
366	>	///////////////////////////
367
368		if(triggerHLT_){
369
284	–	//----------  Filling HLT trigger bits!  ------------
285	–
370		edm::Handle<TriggerResults> hltR;
371		triggerResultsTag_ = InputTag(hlTriggerResults_,"",hltName_);
372		iEvent.getByLabel(triggerResultsTag_,hltR);
#	Line 305 | Line 389 | void MPIntuple::analyze(const edm::Event
389
390		if (hlNames[i] == MPI_TriggerNames[j])
391		{
392	<	//            cout<<"trigger name: "<<hlNames[i]<<"list: "<<dec<<j+1<<endl;
392	>	//cout<<"trigger name: "<<hlNames[i]<<"list: "<<dec<<j+1<<endl;
393		triggerStatus |= 0x01 << j;
394
395		}
#	Line 313 | Line 397 | void MPIntuple::analyze(const edm::Event
397		}
398		}
399		}
400	<	//  cout<< "total status: "<<hex<<triggerStatus<<endl;
401	<
402	<	if((PFcount > 3 || Gencount > 3) && Vtxcount > 0)  sTree -> Fill();
403	<
400	>	//cout<< "total status: "<<hex<<triggerStatus<<endl;
401	>
402	>
403	>
404	>	if((pfCount > 3 || genCount > 3) && vtxCount > 0) sTree -> Fill();
405	>
406		jet_ak5PF->Clear();
407		primaryVtx->Clear();
408		jetP4_ak5Gen->Clear();
#	Line 330 | Line 416 | void  MPIntuple::beginJob()
416
417		ntupleFile           = new TFile(rootfilename.c_str(), "RECREATE");
418		sTree                = new TTree("mpiTree", "Tree for Jets");
419	<
419	>
420	>	h1_trackZ            = new TH1F("h1_trackZ", "#Delta z for associated tracks", 60, -0.3, 0.3);
421	>	h1_nAssociatedTracks = new TH1F("h1_nAssociatedTracks", "Number of tracks associated to jet", 20, -0.5, 19.5);
422	>	h1_trackDxy          = new TH1F("h1_trackDxy", "dxy for all associated tracks", 50, -0.4, 0.4);
423	>	h1_allTrackDeltaZ_PV = new TH1F("h1_allTrackDeltaZ_PV", "#Delta z between jet and primary vertex", 60, -0.3, 0.3);
424	>	h1_allTrackDeltaZ_LJ = new TH1F("h1_allTrackDeltaZ_LJ", "#Delta z between leading jet and other jet vertices", 60, -0.3, 0.3);
425	>	h1_ptHat             = new TH1F("h1_ptHat", "ptHat", 50, 10, 100);
426	>	h2_nAssTracksVsJetPt = new TH2F("h2_nAssTracksVsJetPt", "nTracks vs Jet p_{T}", 50, 0, 100, 20, -0.5, 19.5);
427	>	p1_nVtcs             = new TProfile("p1_nVtcs", "<N> per run", 5200.0, 132300.0, 137500.0, 0.0, 6.0);
428	>
429		jet_ak5PF            = new TClonesArray("TCJet");
430		jetP4_ak5Gen         = new TClonesArray("TLorentzVector");
431		primaryVtx           = new TClonesArray("TCPrimaryVtx");
432
433	+
434		sTree->Branch("jet_ak5PF",&jet_ak5PF, 6400, 0);
435		sTree->Branch("jetP4_ak5Gen",&jetP4_ak5Gen, 6400, 0);
436		sTree->Branch("primaryVtx",&primaryVtx, 6400, 0);
#	Line 345 | Line 441 | void  MPIntuple::beginJob()
441		sTree->Branch("triggerStatus",&triggerStatus, "triggerStatus/i");
442		sTree->Branch("isRealData",&isRealData, "isRealData/i");
443		sTree->Branch("bunchCross",&bunchCross, "bunchCross/i");
444	+	sTree->Branch("ptHat",&ptHat, "ptHat/d");
445
446		}
447
448		// ------------ method called once each job just after ending the event loop  ------------
449		void MPIntuple::endJob()
450		{
451	+
452	+	ntupleFile->cd();
453	+
454	+	h1_trackZ->Write();
455	+	h1_nAssociatedTracks->Write();
456	+	h1_trackDxy->Write();
457	+	h1_allTrackDeltaZ_PV->Write();
458	+	h1_allTrackDeltaZ_LJ->Write();
459	+	h1_ptHat->Write();
460	+	h2_nAssTracksVsJetPt->Write();
461	+	p1_nVtcs->Write();
462	+
463		ntupleFile->Write();
464		ntupleFile->Close();
465
#	Line 366 | Line 475 | bool MPIntuple::triggerDecision(edm::Han
475		triggerPassed = true;
476		}
477		return triggerPassed;
478	<	}
478	>	}
479	>
480	>	double MPIntuple::sumPtSquared(const Vertex & v)
481	>	{
482	>	double sum = 0.;
483	>	double pT;
484	>	for (Vertex::trackRef_iterator it = v.tracks_begin(); it != v.tracks_end(); it++) {
485	>	pT = (**it).pt();
486	>	double epT=(**it).ptError(); pT=pT>epT ? pT-epT : 0;
487
488	+	sum += pT*pT;
489	+	}
490	+	return sum;
491	+	}
492
493		//define this as a plug-in
494		DEFINE_FWK_MODULE(MPIntuple);

Diff Legend

–	Removed lines
+	Added lines
<	Changed lines
>	Changed lines