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.5 by naodell, Sun May 23 02:33:41 2010 UTC vs.
Revision 1.14 by naodell, Tue Oct 5 21:27:59 2010 UTC

#	Line 1 | Line 1
1	–	/*
2	–	Description: <one line class summary>
3	–
4	–	Implementation:
5	–	n-tuple creator for jet studies
6	–	*/
7	–	//
1		// Original Author:  "Nathaniel Odell"
2		// Secondary Author: Steven Won
3		// With contributions from: Andrey Pozdnyakov
4		//         Created:  Thurs April 22  2010
5		// $Id$
13	–	//
14	–	//
15	–
6
7		// system include files
8		#include <memory>
#	Line 25 | Line 15
15		#include "FWCore/Framework/interface/EventSetup.h"
16		#include "FWCore/Framework/interface/Event.h"
17		#include "FWCore/Framework/interface/MakerMacros.h"
28	–
18		#include "FWCore/ParameterSet/interface/ParameterSet.h"
30	–
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"
#	Line 41 | Line 30
30		#include "DataFormats/JetReco/interface/Jet.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	+	//GenParticles
37	+	//#include "DataFormats/HepMCCandidate/interface/GenParticle.h"
38
39		// Need to get correctors
40		#include "JetMETCorrections/Objects/interface/JetCorrector.h"
#	Line 86 | Line 80 | class MPIntuple : public edm::EDAnalyzer
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
#	Line 93 | Line 88 | class MPIntuple : public edm::EDAnalyzer
88		edm::InputTag GenJetHandle_;
89		edm::InputTag PrimaryVtxHandle_;
90		edm::InputTag triggerResultsTag_;
96	–
97	–
91		//  Counting variables   //
92
93	<	int eventNumber, runNumber, lumiSection;
93	>	int eventNumber, runNumber, lumiSection, bunchCross;
94
95	<	TTree *sTree;
95	>	TTree* sTree;
96		TFile* ntupleFile;
97
98		TClonesArray* jet_ak5PF;
#	Line 108 | Line 101 | class MPIntuple : public edm::EDAnalyzer
101
102		bool doGenJets_;
103		bool doPFJets_;
104	<	bool isMC_;
105	<
104	>	bool triggerHLT_;
105	>	bool isRealData;
106		string rootfilename;
107
108		//Triggers
#	Line 129 | Line 122 | MPIntuple::MPIntuple(const edm::Paramete
122		PrimaryVtxHandle_(iConfig.getUntrackedParameter<edm::InputTag>("PrimaryVtxTag")),
123		doGenJets_(iConfig.getUntrackedParameter<bool>("doGenJets")),
124		doPFJets_(iConfig.getUntrackedParameter<bool>("doPFJets")),
125	<	isMC_(iConfig.getUntrackedParameter<bool>("isMC")),
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"))
129		{
130	<	//edm::TriggerNames
138	<	// triggerNames(iConfig);
130	>
131
132		}
133
#	Line 145 | Line 137 | MPIntuple::~MPIntuple()
137
138		}
139
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 153 | Line 158 | MPIntuple::~MPIntuple()
158		void MPIntuple::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup)
159		{
160
161	<	eventNumber = iEvent.id().event();
162	<	runNumber   = iEvent.id().run();
161	>
162	>	eventNumber  = iEvent.id().event();
163	>	runNumber    = iEvent.id().run();
164		lumiSection  = (unsigned int)iEvent.getLuminosityBlock().luminosityBlock();
165	<
166	<	int PFcount = 0;
167	<	int Gencount = 0;
168	<	int Vtxcount = 0;
165	>	bunchCross   = (unsigned int)iEvent.bunchCrossing();
166	>	isRealData   = iEvent.isRealData();
167	>
168	>	int PFcount   = 0;
169	>	int Gencount  = 0;
170	>	int Vtxcount  = 0;
171
172		if(doPFJets_){
173
174	+	edm::Handle<reco::JetTagCollection> bTagHandle1;
175	+	iEvent.getByLabel("trackCountingHighEffBJetTags", bTagHandle1);
176	+	const reco::JetTagCollection & bTags1 = *(bTagHandle1.product());
177	+	reco::JetTagCollection::const_iterator jet_it_1;
178	+
179	+	edm::Handle<reco::JetTagCollection> bTagHandle2;
180	+	iEvent.getByLabel("trackCountingHighPurBJetTags", bTagHandle2);
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
#	Line 174 | Line 193 | void MPIntuple::analyze(const edm::Event
193		for(PFJetCollection::const_iterator jet_iter = PFJets->begin(); jet_iter!= PFJets->end(); ++jet_iter){
194
195		reco::PFJet myJet = reco::PFJet(*jet_iter);
196	+	if(myJet.et() > 5){
197
198	<	float scale2 = correctorL2->correction(myJet);
199	<	float scale3 = correctorL3->correction(myJet);
198	>	for(jet_it_1 = bTags1.begin(); jet_it_1 != bTags1.end(); jet_it_1++) {
199	>	if( (fabs(jet_it_1->first->eta() - myJet.eta()) < .005) && (deltaPhi(jet_it_1->first->phi(),myJet.phi()) < .005)) break;
200	>	}
201	>
202	>	for(jet_it_2 = bTags2.begin(); jet_it_2 != bTags2.end(); jet_it_2++) {
203	>	if( (fabs(jet_it_2->first->eta() - myJet.eta()) < .005) && (deltaPhi(jet_it_2->first->phi(),myJet.phi()) < .005)) break;
204	>	}
205	>
206	>	TCJet* jetCon = new ((*jet_ak5PF)[PFcount]) TCJet;
207	>
208	>	jetCon->SetP4(myJet.px(), myJet.py(), myJet.pz(), myJet.energy());
209	>	jetCon->SetVtx(myJet.vx(), myJet.vy(), myJet.vz());
210
211	<	if(myJet.et() > 5){
212	<
213	<	TCJet jetCon;
214	<
185	<	jetCon.SetP4(myJet.px(), myJet.py(), myJet.pz(), myJet.energy());
186	<	jetCon.SetVtx(myJet.vx(), myJet.vy(), myJet.vz());
211	>	jetCon->SetChHadFrac(myJet.chargedHadronEnergyFraction());
212	>	jetCon->SetNeuHadFrac(myJet.neutralHadronEnergyFraction());
213	>	jetCon->SetChEmFrac(myJet.chargedEmEnergyFraction());
214	>	jetCon->SetNeuEmFrac(myJet.neutralEmEnergyFraction());
215
216	<	jetCon.SetChHadFrac(myJet.chargedHadronEnergyFraction());
217	<	jetCon.SetNeuHadFrac(myJet.neutralHadronEnergyFraction());
190	<	jetCon.SetChEmFrac(myJet.chargedEmEnergyFraction());
191	<	jetCon.SetNeuEmFrac(myJet.neutralEmEnergyFraction());
216	>	jetCon->SetNumConstit(myJet.chargedMultiplicity() + myJet.neutralMultiplicity());
217	>	jetCon->SetNumChPart(myJet.chargedMultiplicity());
218
219	<	jetCon.SetNumConstit(myJet.chargedMultiplicity() + myJet.neutralMultiplicity());
194	<	jetCon.SetNumChPart(myJet.chargedMultiplicity());
219	>	jetCon->SetNumChPart(myJet.chargedMultiplicity());
220
221	<	jetCon.SetNumChPart(myJet.chargedMultiplicity());
221	>	if(jet_it_2 != bTags2.end()) jetCon->SetBDiscrTrkCountHiPure(jet_it_2->second);
222	>	if(jet_it_1 != bTags1.end()) jetCon->SetBDiscrTrkCountHiEff(jet_it_1->second);
223	>
224	>	float scale2 = correctorL2->correction(myJet);
225	>	myJet.scaleEnergy(scale2);
226	>	float scale3 = correctorL3->correction(myJet);
227	>	myJet.scaleEnergy(scale3);
228	>
229	>	//more corrections?
230	>
231	>	jetCon->SetJetCorr(2, scale2);
232	>	jetCon->SetJetCorr(3, scale3);
233
198	–	jetCon.SetJetCorr(2, scale2);
199	–	jetCon.SetJetCorr(3, scale3);
200	–
201	–	new ((*jet_ak5PF)[PFcount]) TCJet(jetCon);
202	–
234		++PFcount;
235		}
236		}
237		}
238
239	<	if(doGenJets_){
239	>	if(!isRealData){
240
241	+
242		Handle<reco::GenJetCollection> GenJets;
243		iEvent.getByLabel(GenJetHandle_, GenJets);
244
#	Line 222 | Line 254 | void MPIntuple::analyze(const edm::Event
254		++Gencount;
255
256		}
257	<	}
257	>	}
258		}
259	<
260	<
259	>
260	>
261		Handle<reco::VertexCollection> primaryVtcs;
262		iEvent.getByLabel(PrimaryVtxHandle_, primaryVtcs);
263	<
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	<
268	<	TCPrimaryVtx vtxCon;
237	<
238	<	vtxCon.SetPosition(myVtx.x(), myVtx.y(), myVtx.z());
239	<	vtxCon.SetNDof(myVtx.ndof());
240	<	vtxCon.SetChi2(myVtx.chi2());
241	<
242	<	new ((*primaryVtx)[Vtxcount]) TCPrimaryVtx(vtxCon);
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
279
280	+	if(triggerHLT_){
281
282	<	//----------  Filling HLT trigger bits!  ------------
251	<
252	<	edm::Handle<TriggerResults> hltR;
253	<	triggerResultsTag_ = InputTag(hlTriggerResults_,"",hltName_);
254	<	iEvent.getByLabel(triggerResultsTag_,hltR);
255	<
256	<	const TriggerNames & triggerNames = iEvent.triggerNames(*hltR);
257	<	hlNames=triggerNames.triggerNames();
258	<
259	<	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"};
282	>	//----------  Filling HLT trigger bits!  ------------
283
284	<	bool triggerPassed = false;
285	<	triggerStatus = 0x0;
286	<
287	<	for (uint iT=0; iT<hlNames.size(); ++iT) {
288	<
289	<	triggerPassed = triggerDecision(hltR, iT);
284	>	edm::Handle<TriggerResults> hltR;
285	>	triggerResultsTag_ = InputTag(hlTriggerResults_,"",hltName_);
286	>	iEvent.getByLabel(triggerResultsTag_,hltR);
287	>
288	>	const TriggerNames & triggerNames = iEvent.triggerNames(*hltR);
289	>	hlNames=triggerNames.triggerNames();
290
291	<	if(triggerPassed){
291	>	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"};
292	>
293	>	bool triggerPassed = false;
294	>	triggerStatus = 0x0;
295	>
296	>	for (uint i=0; i<hlNames.size(); ++i) {
297	>
298	>	triggerPassed = triggerDecision(hltR, i);
299
300	<	for (int j = 0; j != 32; ++j){
300	>	if(triggerPassed){
301
302	<	if (hlNames[iT] == MPI_TriggerNames[j])
303	<	{
304	<	cout<<"trigger name: "<<hlNames[iT]<<"  status: "<<triggerPassed<<endl;
305	<	triggerStatus |= 0x01 << j;
306	<	}
302	>	for (uint j = 0; j != 32; ++j){
303	>
304	>	if (hlNames[i] == MPI_TriggerNames[j])
305	>	{
306	>	//            cout<<"trigger name: "<<hlNames[i]<<"list: "<<dec<<j+1<<endl;
307	>	triggerStatus |= 0x01 << j;
308	>
309	>	}
310	>	}
311		}
312	<	}
313	<	}
314	<
312	>	}
313	>	}
314	>	//  cout<< "total status: "<<hex<<triggerStatus<<endl;
315	>
316
317	<	if(PFcount > 0 || Gencount > 0 || Vtxcount > 0);  sTree -> Fill();
317	>
318	>	if((PFcount > 3 || Gencount > 3) && Vtxcount > 0)  sTree -> Fill();
319
320		jet_ak5PF->Clear();
321		primaryVtx->Clear();
#	Line 306 | Line 342 | void  MPIntuple::beginJob()
342		sTree->Branch("eventNumber",&eventNumber, "eventNumber/I");
343		sTree->Branch("runNumber",&runNumber, "runNumber/I");
344		sTree->Branch("lumiSection",&lumiSection, "lumiSection/I");
345	<	sTree->Branch("triggerStatus",&triggerStatus, "triggerStatus/I");
346	<	sTree->Branch("isMC_",&isMC_,"isMC_/B");
347	<
345	>	sTree->Branch("triggerStatus",&triggerStatus, "triggerStatus/i");
346	>	sTree->Branch("isRealData",&isRealData, "isRealData/i");
347	>	sTree->Branch("bunchCross",&bunchCross, "bunchCross/i");
348
349		}
350

Diff Legend

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