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.14 by naodell, Tue Oct 5 21:27:59 2010 UTC vs.
Revision 1.21 by naodell, Wed Nov 3 09:53:54 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	+	TH2F * h2_nAssTracksVsJetPt;
130	+	TProfile * p1_nVtcs;
131
116	–	};
132
133	<	MPIntuple::MPIntuple(const edm::ParameterSet& iConfig):
133	>	};
134
135	<	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"))
135	>	MPIntuple::MPIntuple(const edm::ParameterSet& iConfig)
136		{
137	<
138	<
137	>	PFJetHandle_ = iConfig.getUntrackedParameter<edm::InputTag>("PFJetTag");
138	>	GenJetHandle_ = iConfig.getUntrackedParameter<edm::InputTag>("GenJetTag");
139	>	PrimaryVtxHandle_ = iConfig.getUntrackedParameter<edm::InputTag>("PrimaryVtxTag");
140	>	doGenJets_ = iConfig.getUntrackedParameter<bool>("doGenJets");
141	>	doPFJets_ = iConfig.getUntrackedParameter<bool>("doPFJets");
142	>	triggerHLT_ = iConfig.getUntrackedParameter<bool>("triggerHLT");
143	>	rootfilename = iConfig.getUntrackedParameter<string>("rootfilename");
144	>	hlTriggerResults_ = iConfig.getUntrackedParameter<string>("HLTriggerResults","TriggerResults");
145	>	hltName_ = iConfig.getUntrackedParameter<string>("hltName","HLT");
146		}
147
134	–
148		MPIntuple::~MPIntuple()
149		{
150
151		}
152
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	–
153		//
154		// member functions
155		//
#	Line 158 | Line 158 | double MPIntuple::sumPtSquared(const Ver
158		void MPIntuple::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup)
159		{
160
161	–
161		eventNumber  = iEvent.id().event();
162		runNumber    = iEvent.id().run();
163		lumiSection  = (unsigned int)iEvent.getLuminosityBlock().luminosityBlock();
164		bunchCross   = (unsigned int)iEvent.bunchCrossing();
165		isRealData   = iEvent.isRealData();
166
167	<	int PFcount   = 0;
168	<	int Gencount  = 0;
169	<	int Vtxcount  = 0;
167	>
168	>	edm::Handle<reco::BeamSpot> beamSpotHandle;
169	>	iEvent.getByLabel("offlineBeamSpot", beamSpotHandle);
170	>	reco::BeamSpot vertexBeamSpot = *beamSpotHandle;
171	>
172	>	int pfCount   = 0;
173	>	int genCount  = 0;
174	>	int vtxCount  = 0;
175	>	float ljVertexZ = -999;
176	>	float primaryVertexZ = -999;
177	>
178	>	//////////////////////////
179	>	//Get vertex information//
180	>	//////////////////////////
181	>
182	>	Handle<reco::VertexCollection> primaryVtcs;
183	>	iEvent.getByLabel(PrimaryVtxHandle_, primaryVtcs);
184	>
185	>	for(VertexCollection::const_iterator vtx_iter = primaryVtcs->begin(); vtx_iter!= primaryVtcs->end(); ++vtx_iter){
186	>
187	>	reco::Vertex myVtx = reco::Vertex(*vtx_iter);
188	>	if(!myVtx.isValid() || myVtx.isFake()) continue;
189	>	TCPrimaryVtx* vtxCon = new ((*primaryVtx)[vtxCount]) TCPrimaryVtx;
190	>
191	>	cout<<myVtx.nTracks()<<endl;
192	>
193	>	vtxCon->SetPosition(myVtx.x(), myVtx.y(), myVtx.z());
194	>	vtxCon->SetNDof(myVtx.ndof());
195	>	vtxCon->SetChi2(myVtx.chi2());
196	>	vtxCon->SetNTrks(myVtx.tracksSize());
197	>	vtxCon->SetSumPt2Trks(sumPtSquared(myVtx));
198	>
199	>	if(vtxCount == 0) primaryVertexZ = myVtx.z();
200	>
201	>	++vtxCount;
202	>
203	>	}
204	>
205	>	p1_nVtcs->Fill(runNumber, vtxCount);
206	>
207	>	///////////////////////
208	>	//get jet information//
209	>	///////////////////////
210
211		if(doPFJets_){
212
#	Line 187 | Line 226 | void MPIntuple::analyze(const edm::Event
226
227		Handle<reco::PFJetCollection> PFJets;
228		iEvent.getByLabel(PFJetHandle_, PFJets);
229	<
191	<	PFcount = 0;
192	<
229	>
230		for(PFJetCollection::const_iterator jet_iter = PFJets->begin(); jet_iter!= PFJets->end(); ++jet_iter){
231
232		reco::PFJet myJet = reco::PFJet(*jet_iter);
#	Line 203 | Line 240 | void MPIntuple::analyze(const edm::Event
240		if( (fabs(jet_it_2->first->eta() - myJet.eta()) < .005) && (deltaPhi(jet_it_2->first->phi(),myJet.phi()) < .005)) break;
241		}
242
243	<	TCJet* jetCon = new ((*jet_ak5PF)[PFcount]) TCJet;
243	>	TCJet* jetCon = new ((*jet_ak5PF)[pfCount]) TCJet;
244
245		jetCon->SetP4(myJet.px(), myJet.py(), myJet.pz(), myJet.energy());
246	<	jetCon->SetVtx(myJet.vx(), myJet.vy(), myJet.vz());
246	>	jetCon->SetVtx(-999.0, -999.0, -999.0);
247
248		jetCon->SetChHadFrac(myJet.chargedHadronEnergyFraction());
249		jetCon->SetNeuHadFrac(myJet.neutralHadronEnergyFraction());
#	Line 231 | Line 268 | void MPIntuple::analyze(const edm::Event
268		jetCon->SetJetCorr(2, scale2);
269		jetCon->SetJetCorr(3, scale3);
270
271	<	++PFcount;
271	>	/////////////////////////
272	>	//get associated tracks//
273	>	/////////////////////////
274	>
275	>	const reco::TrackRefVector &tracks = myJet.getTrackRefs();
276	>
277	>	float sumTrackZ, sumTrackIP;
278	>	float meanTrackZ, meanTrackIP;
279	>	int   nAssociatedTracks = -1;
280	>
281	>	sumTrackZ  = sumTrackIP  = 0;
282	>	meanTrackZ = meanTrackIP = -999;
283	>
284	>	if(myJet.pt() > 20 && fabs(myJet.eta()) < 2.4){
285	>
286	>	nAssociatedTracks = 0;
287	>
288	>	for(TrackRefVector::const_iterator iTrack = tracks.begin(); iTrack != tracks.end(); ++iTrack){
289	>
290	>	const reco::Track    &myJetTrack = **iTrack;
291	>	//const reco::TrackRef &myTrackRef = *iTrack;
292	>
293	>	sumTrackZ += myJetTrack.vz();
294	>	sumTrackIP += myJetTrack.dxy(vertexBeamSpot.position());
295	>	++nAssociatedTracks;
296	>
297	>	for(VertexCollection::const_iterator vtx_iter = primaryVtcs->begin(); vtx_iter!= primaryVtcs->end(); ++vtx_iter){
298	>
299	>	reco::Vertex myVtx = reco::Vertex(*vtx_iter);
300	>	if(!myVtx.isValid() || myVtx.isFake()) continue;
301	>
302	>	for(Vertex::trackRef_iterator iTrackRef = myVtx.tracks_begin(); iTrackRef != myVtx.tracks_end(); ++iTrackRef){
303	>	const edm::RefToBase<reco::Track> &myTrackRef = *iTrackRef;
304	>
305	>	if(myTrackRef.isAvailable()){
306	>
307	>	const reco::Track &myVertexTrack = *myTrackRef.get();
308	>
309	>	cout<<"Vertex track z = "<<myVertexTrack.vz()<<endl;
310	>	cout<<"Jet track z = "<<myJetTrack.vz()<<endl;
311	>
312	>	}
313	>	}
314	>	}
315	>	}
316	>
317	>	meanTrackZ = sumTrackZ/nAssociatedTracks;
318	>	h1_nAssociatedTracks->Fill(nAssociatedTracks);
319	>	h2_nAssTracksVsJetPt->Fill(myJet.pt(), nAssociatedTracks);
320	>
321	>	for(TrackRefVector::const_iterator iTrack = tracks.begin(); iTrack != tracks.end(); ++iTrack){
322	>
323	>	const reco::Track &myJetTrack = **iTrack;
324	>
325	>	if(nAssociatedTracks == 3) h1_trackZ->Fill(meanTrackZ - myJetTrack.vz());
326	>	h1_trackDxy->Fill(myJetTrack.dxy(vertexBeamSpot.position()));
327	>
328	>	}
329	>
330	>	if(pfCount == 0){
331	>
332	>	ljVertexZ = meanTrackZ;
333	>
334	>	}else{
335	>
336	>	h1_allTrackDeltaZ_LJ->Fill(ljVertexZ - meanTrackZ);
337	>
338	>	}
339	>
340	>	h1_allTrackDeltaZ_PV->Fill(primaryVertexZ - meanTrackZ);
341	>
342	>	}
343	>
344	>	jetCon->SetVtx(0, 0, meanTrackZ);
345	>
346	>	++pfCount;
347		}
348		}
349		}
350
351		if(!isRealData){
352
353	+	Handle<GenEventInfoProduct> GenInfoHandle;
354	+	iEvent.getByLabel("generator", GenInfoHandle);
355
356		Handle<reco::GenJetCollection> GenJets;
357		iEvent.getByLabel(GenJetHandle_, GenJets);
358	<
359	<
358	>
359	>	ptHat = qScale = -1;
360	>
361	>	if(GenInfoHandle.isValid()){
362	>
363	>	qScale = GenInfoHandle->qScale();
364	>	ptHat  = (GenInfoHandle->hasBinningValues() ? GenInfoHandle->binningValues()[0] : 0.0);
365	>
366	>	}
367	>
368		for(GenJetCollection::const_iterator jet_iter = GenJets->begin(); jet_iter!= GenJets->end(); ++jet_iter){
369
370		reco::GenJet myJet = reco::GenJet(*jet_iter);
371
372		if(myJet.pt() > 5){
373
374	<	new ((*jetP4_ak5Gen)[Gencount]) TLorentzVector(myJet.px(), myJet.py(), myJet.pz(), myJet.energy());
374	>	new ((*jetP4_ak5Gen)[genCount]) TLorentzVector(myJet.px(), myJet.py(), myJet.pz(), myJet.energy());
375
376	<	++Gencount;
376	>	++genCount;
377
378		}
379		}
380		}
381
382	<
383	<	Handle<reco::VertexCollection> primaryVtcs;
384	<	iEvent.getByLabel(PrimaryVtxHandle_, primaryVtcs);
263	<
264	<	for(VertexCollection::const_iterator vtx_iter = primaryVtcs->begin(); vtx_iter!= primaryVtcs->end(); ++vtx_iter){
265	<
266	<	reco::Vertex myVtx = reco::Vertex(*vtx_iter);
267	<	if(!myVtx.isValid() || myVtx.isFake()) continue;
268	<	TCPrimaryVtx* vtxCon = new ((*primaryVtx)[Vtxcount]) TCPrimaryVtx;
269	<
270	<	vtxCon->SetPosition(myVtx.x(), myVtx.y(), myVtx.z());
271	<	vtxCon->SetNDof(myVtx.ndof());
272	<	vtxCon->SetChi2(myVtx.chi2());
273	<	vtxCon->SetNTrks(myVtx.tracksSize());
274	<	vtxCon->SetSumPt2Trks(sumPtSquared(myVtx));
275	<	++Vtxcount;
276	<
277	<	}
278	<
382	>	///////////////////////////
383	>	//get trigger information//
384	>	///////////////////////////
385
386		if(triggerHLT_){
387
282	–	//----------  Filling HLT trigger bits!  ------------
283	–
388		edm::Handle<TriggerResults> hltR;
389		triggerResultsTag_ = InputTag(hlTriggerResults_,"",hltName_);
390		iEvent.getByLabel(triggerResultsTag_,hltR);
#	Line 303 | Line 407 | void MPIntuple::analyze(const edm::Event
407
408		if (hlNames[i] == MPI_TriggerNames[j])
409		{
410	<	//            cout<<"trigger name: "<<hlNames[i]<<"list: "<<dec<<j+1<<endl;
410	>	//cout<<"trigger name: "<<hlNames[i]<<"list: "<<dec<<j+1<<endl;
411		triggerStatus |= 0x01 << j;
412
413		}
#	Line 311 | Line 415 | void MPIntuple::analyze(const edm::Event
415		}
416		}
417		}
418	<	//  cout<< "total status: "<<hex<<triggerStatus<<endl;
419	<
418	>	//cout<< "total status: "<<hex<<triggerStatus<<endl;
419	>
420
421
422	<	if((PFcount > 3 || Gencount > 3) && Vtxcount > 0)  sTree -> Fill();
423	<
422	>	if((pfCount > 3 || genCount > 3) && vtxCount > 0) sTree -> Fill();
423	>
424		jet_ak5PF->Clear();
425		primaryVtx->Clear();
426		jetP4_ak5Gen->Clear();
#	Line 330 | Line 434 | void  MPIntuple::beginJob()
434
435		ntupleFile           = new TFile(rootfilename.c_str(), "RECREATE");
436		sTree                = new TTree("mpiTree", "Tree for Jets");
437	<
437	>
438	>	h1_trackZ            = new TH1F("h1_trackZ", "#Delta z for associated tracks", 60, -0.3, 0.3);
439	>	h1_nAssociatedTracks = new TH1F("h1_nAssociatedTracks", "Number of tracks associated to jet", 20, -0.5, 19.5);
440	>	h1_trackDxy          = new TH1F("h1_trackDxy", "dxy for all associated tracks", 50, -0.4, 0.4);
441	>	h1_allTrackDeltaZ_PV = new TH1F("h1_allTrackDeltaZ_PV", "#Delta z between jet and primary vertex", 60, -0.3, 0.3);
442	>	h1_allTrackDeltaZ_LJ = new TH1F("h1_allTrackDeltaZ_LJ", "#Delta z between leading jet and other jet vertices", 60, -0.3, 0.3);
443	>	h2_nAssTracksVsJetPt = new TH2F("h2_nAssTracksVsJetPt", "nTracks vs Jet p_{T}", 50, 0, 100, 20, -0.5, 19.5);
444	>	p1_nVtcs             = new TProfile("p1_nVtcs", "<N> per run", 5200.0, 132300.0, 137500.0, 0.0, 6.0);
445	>
446		jet_ak5PF            = new TClonesArray("TCJet");
447		jetP4_ak5Gen         = new TClonesArray("TLorentzVector");
448		primaryVtx           = new TClonesArray("TCPrimaryVtx");
449
450	+
451		sTree->Branch("jet_ak5PF",&jet_ak5PF, 6400, 0);
452		sTree->Branch("jetP4_ak5Gen",&jetP4_ak5Gen, 6400, 0);
453		sTree->Branch("primaryVtx",&primaryVtx, 6400, 0);
#	Line 345 | Line 458 | void  MPIntuple::beginJob()
458		sTree->Branch("triggerStatus",&triggerStatus, "triggerStatus/i");
459		sTree->Branch("isRealData",&isRealData, "isRealData/i");
460		sTree->Branch("bunchCross",&bunchCross, "bunchCross/i");
461	+	sTree->Branch("ptHat",&ptHat, "ptHat/d");
462
463		}
464
465		// ------------ method called once each job just after ending the event loop  ------------
466		void MPIntuple::endJob()
467		{
468	+
469	+	ntupleFile->cd();
470	+
471	+	h1_trackZ->Write();
472	+	h1_nAssociatedTracks->Write();
473	+	h1_trackDxy->Write();
474	+	h1_allTrackDeltaZ_PV->Write();
475	+	h1_allTrackDeltaZ_LJ->Write();
476	+	h2_nAssTracksVsJetPt->Write();
477	+	p1_nVtcs->Write();
478	+
479		ntupleFile->Write();
480		ntupleFile->Close();
481
#	Line 366 | Line 491 | bool MPIntuple::triggerDecision(edm::Han
491		triggerPassed = true;
492		}
493		return triggerPassed;
494	<	}
494	>	}
495	>
496	>	double MPIntuple::sumPtSquared(const Vertex & v)
497	>	{
498	>	double sum = 0.;
499	>	double pT;
500	>	for (Vertex::trackRef_iterator it = v.tracks_begin(); it != v.tracks_end(); it++) {
501	>	pT = (**it).pt();
502	>	double epT=(**it).ptError(); pT=pT>epT ? pT-epT : 0;
503
504	+	sum += pT*pT;
505	+	}
506	+	return sum;
507	+	}
508
509		//define this as a plug-in
510		DEFINE_FWK_MODULE(MPIntuple);

Diff Legend

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