ViewVC Help

View File | Revision Log | Show Annotations | Root Listing

root/cvsroot/UserCode/UHHAnalysis/NtupleWriter/src/NtupleWriter.cc

Comparing UserCode/UHHAnalysis/NtupleWriter/src/NtupleWriter.cc (file contents):
Revision 1.5 by peiffer, Thu Apr 5 09:48:20 2012 UTC vs.
Revision 1.12 by peiffer, Thu Apr 19 12:03:28 2012 UTC

#	Line 53 | Line 53 | NtupleWriter::NtupleWriter(const edm::Pa
53		doPhotons = iConfig.getParameter<bool>("doPhotons");
54		doMET = iConfig.getParameter<bool>("doMET");
55		doGenInfo = iConfig.getParameter<bool>("doGenInfo");
56	+	doAllGenParticles = iConfig.getParameter<bool>("doAllGenParticles");
57	+	doLumiInfo = iConfig.getParameter<bool>("doLumiInfo");
58		doPV = iConfig.getParameter<bool>("doPV");
59		doTopJets = iConfig.getParameter<bool>("doTopJets");
60	+	doTrigger = iConfig.getParameter<bool>("doTrigger");
61
62		// initialization of tree variables
63
#	Line 63 | Line 66 | NtupleWriter::NtupleWriter(const edm::Pa
66		tr->Branch("luminosityBlock",&luminosityBlock);
67		tr->Branch("isRealData",&isRealData);
68		tr->Branch("HBHENoiseFilterResult",&HBHENoiseFilterResult);
69	<	tr->Branch("beamspot_x0",&beamspot_x0);
70	<	tr->Branch("beamspot_y0",&beamspot_y0);
71	<	tr->Branch("beamspot_z0",&beamspot_z0);
72	<
69	>	if(doLumiInfo){
70	>	tr->Branch("intgRecLumi",&intgRecLumi);
71	>	tr->Branch("intgDelLumi",&intgDelLumi);
72	>	}
73	>	if(doPV){
74	>	tr->Branch("beamspot_x0",&beamspot_x0);
75	>	tr->Branch("beamspot_y0",&beamspot_y0);
76	>	tr->Branch("beamspot_z0",&beamspot_z0);
77	>	}
78		if(doElectrons){
79		electron_sources = iConfig.getParameter<std::vector<std::string> >("electron_sources");
80		for(size_t j=0; j< electron_sources.size(); ++j){
#	Line 125 | Line 133 | NtupleWriter::NtupleWriter(const edm::Pa
133		tr->Branch("genInfo","GenInfo",&genInfo);
134		tr->Branch("GenParticles","std::vector<GenParticle>", &genps);
135		}
136	<
137	<	trigger_prefixes = iConfig.getParameter<std::vector<std::string> >("trigger_prefixes");
138	<	//tr->Branch("triggerResults","std::map<std::string, bool>",&triggerResults);
139	<	tr->Branch("triggerNames", "std::vector<std::string>", &triggerNames);
140	<	tr->Branch("triggerResults", "std::vector<bool>", &triggerResults);
141	<	tr->Branch("L1_prescale", "std::vector<int>", &L1_prescale);
142	<	tr->Branch("HLT_prescale", "std::vector<int>", &HLT_prescale);
143	<
136	>	if(doTrigger){
137	>	trigger_prefixes = iConfig.getParameter<std::vector<std::string> >("trigger_prefixes");
138	>	//tr->Branch("triggerResults","std::map<std::string, bool>",&triggerResults);
139	>	tr->Branch("triggerNames", "std::vector<std::string>", &triggerNames);
140	>	tr->Branch("triggerResults", "std::vector<bool>", &triggerResults);
141	>	tr->Branch("L1_prescale", "std::vector<int>", &L1_prescale);
142	>	tr->Branch("HLT_prescale", "std::vector<int>", &HLT_prescale);
143	>	}
144		newrun = true;
145		}
146
#	Line 196 | Line 204 | NtupleWriter::analyze(const edm::Event&
204		pvs[j].push_back(pv);
205		}
206		}
207	+
208	+	edm::Handle<reco::BeamSpot> beamSpot;
209	+	iEvent.getByLabel(edm::InputTag("offlineBeamSpot"), beamSpot);
210	+	const reco::BeamSpot & bsp = *beamSpot;
211	+
212	+	beamspot_x0 = bsp.x0();
213	+	beamspot_y0 = bsp.y0();
214	+	beamspot_z0 = bsp.z0();
215		}
200	–
201	–	edm::Handle<reco::BeamSpot> beamSpot;
202	–	iEvent.getByLabel(edm::InputTag("offlineBeamSpot"), beamSpot);
203	–	const reco::BeamSpot & bsp = *beamSpot;
204	–
205	–	beamspot_x0 = bsp.x0();
206	–	beamspot_y0 = bsp.y0();
207	–	beamspot_z0 = bsp.z0();
216
217		// ------------- electrons -------------
218		if(doElectrons){
219	+
220	+	edm::Handle<reco::ConversionCollection> hConversions;
221	+	iEvent.getByLabel("allConversions", hConversions);
222	+
223	+	edm::Handle<reco::BeamSpot> beamSpot;
224	+	iEvent.getByLabel(edm::InputTag("offlineBeamSpot"), beamSpot);
225	+	const reco::BeamSpot & bsp = *beamSpot;
226	+
227		for(size_t j=0; j< electron_sources.size(); ++j){
228		eles[j].clear();
229		edm::Handle< std::vector<pat::Electron> > ele_handle;
#	Line 241 | Line 257 | NtupleWriter::analyze(const edm::Event&
257		ele.gsfTrack_vx= pat_ele.gsfTrack()->vx();
258		ele.gsfTrack_vy= pat_ele.gsfTrack()->vy();
259		ele.gsfTrack_vz= pat_ele.gsfTrack()->vz();
260	+	ele.passconversionveto = !ConversionTools::hasMatchedConversion(pat_ele,hConversions,bsp.position());
261	+	ele.dEtaIn=pat_ele.deltaEtaSuperClusterTrackAtVtx();
262	+	ele.dPhiIn=pat_ele.deltaPhiSuperClusterTrackAtVtx();
263	+	ele.sigmaIEtaIEta=pat_ele.sigmaIetaIeta();
264	+	ele.HoverE=pat_ele.hadronicOverEm();
265	+	ele.fbrem=pat_ele.fbrem();
266	+	ele.EoverPIn=pat_ele.eSuperClusterOverP();
267	+	ele.EcalEnergy=pat_ele.ecalEnergy();
268	+	ele.mvaTrigV0=pat_ele.electronID("mvaTrigV0");
269	+	ele.mvaNonTrigV0=pat_ele.electronID("mvaNonTrigV0");
270	+
271		eles[j].push_back(ele);
272		}
273		}
#	Line 302 | Line 329 | NtupleWriter::analyze(const edm::Event&
329		mu.innerTrack_d0Error = innerTrack->d0Error();
330		mu.innerTrack_numberOfValidHits = innerTrack->numberOfValidHits();
331		mu.innerTrack_numberOfLostHits = innerTrack->numberOfLostHits();
332	+	mu.innerTrack_trackerLayersWithMeasurement = innerTrack->hitPattern().trackerLayersWithMeasurement();
333	+	mu.innerTrack_numberOfValidPixelHits = innerTrack->hitPattern().numberOfValidPixelHits();
334		}
335		else{
336		mu.innerTrack_chi2 = 0;
#	Line 310 | Line 339 | NtupleWriter::analyze(const edm::Event&
339		mu.innerTrack_d0Error = 0;
340		mu.innerTrack_numberOfValidHits = 0;
341		mu.innerTrack_numberOfLostHits = 0;
342	+	mu.innerTrack_trackerLayersWithMeasurement = 0;
343	+	mu.innerTrack_numberOfValidPixelHits = 0;
344		}
345		reco::TrackRef outerTrack = pat_mu.outerTrack();
346		if(!outerTrack.isNull()){
#	Line 355 | Line 386 | NtupleWriter::analyze(const edm::Event&
386		tau.eta =  pat_tau.eta();
387		tau.phi =  pat_tau.phi();
388		tau.energy =  pat_tau.energy();
389	<
389	>	tau.decayModeFinding = pat_tau.tauID("decayModeFinding")>0.5;
390	>	tau.byVLooseCombinedIsolationDeltaBetaCorr  = pat_tau.tauID("byVLooseCombinedIsolationDeltaBetaCorr")>0.5;
391	>	tau.byLooseCombinedIsolationDeltaBetaCorr = pat_tau.tauID("byLooseCombinedIsolationDeltaBetaCorr")>0.5;
392	>	tau.byMediumCombinedIsolationDeltaBetaCorr = pat_tau.tauID("byMediumCombinedIsolationDeltaBetaCorr")>0.5;
393	>	tau.byTightCombinedIsolationDeltaBetaCorr = pat_tau.tauID("byTightCombinedIsolationDeltaBetaCorr")>0.5;
394	>	tau.againstElectronLoose  = pat_tau.tauID("againstElectronLoose")>0.5;
395	>	tau.againstElectronMedium = pat_tau.tauID("againstElectronMedium")>0.5;
396	>	tau.againstElectronTight = pat_tau.tauID("againstElectronTight")>0.5;
397	>	tau.againstElectronMVA  = pat_tau.tauID("againstElectronMVA")>0.5;
398	>	tau.againstMuonLoose = pat_tau.tauID("againstMuonLoose")>0.5;
399	>	tau.againstMuonMedium = pat_tau.tauID("againstMuonMedium")>0.5;
400	>	tau.againstMuonTight = pat_tau.tauID("againstMuonTight")>0.5;
401	>
402	>	reco::PFCandidateRef leadPFCand = pat_tau.leadPFCand();
403	>	if(!leadPFCand.isNull()){
404	>	tau.leadPFCand_px = leadPFCand->px();
405	>	tau.leadPFCand_py = leadPFCand->py();
406	>	tau.leadPFCand_pz = leadPFCand->pz();
407	>	}
408	>	else{
409	>	tau.leadPFCand_px = 0;
410	>	tau.leadPFCand_py = 0;
411	>	tau.leadPFCand_pz = 0;
412	>	}
413		taus[j].push_back(tau);
414		}
415		}
#	Line 409 | Line 463 | NtupleWriter::analyze(const edm::Event&
463		jecUnc->setJetEta(pat_jet.eta());
464		jecUnc->setJetPt(pat_jet.pt());
465		jet.JEC_uncertainty = jecUnc->getUncertainty(true);
466	+	jet.JEC_factor_raw = pat_jet.jecFactor("Uncorrected");
467
468		jet.btag_simpleSecondaryVertexHighEff=pat_jet.bDiscriminator("simpleSecondaryVertexHighEffBJetTags");
469		jet.btag_simpleSecondaryVertexHighPur=pat_jet.bDiscriminator("simpleSecondaryVertexHighPurBJetTags");
#	Line 417 | Line 472 | NtupleWriter::analyze(const edm::Event&
472		jet.btag_jetBProbability=pat_jet.bDiscriminator("jetBProbabilityBJetTags");
473		jet.btag_jetProbability=pat_jet.bDiscriminator("jetProbabilityBJetTags");
474
475	+	const reco::GenJet *genj = pat_jet.genJet();
476	+	if(genj){
477	+	jet.genjet_pt = genj->pt();
478	+	jet.genjet_eta = genj->eta();
479	+	jet.genjet_phi = genj->phi();
480	+	jet.genjet_energy = genj->energy();
481	+	}
482	+
483		jets[j].push_back(jet);
484		}
485		}
#	Line 463 | Line 526 | NtupleWriter::analyze(const edm::Event&
526		jecUnc->setJetEta(pat_topjet.eta());
527		jecUnc->setJetPt(pat_topjet.pt());
528		topjet.JEC_uncertainty = jecUnc->getUncertainty(true);
529	+	topjet.JEC_factor_raw = pat_topjet.jecFactor("Uncorrected");
530
531		topjet.btag_simpleSecondaryVertexHighEff=pat_topjet.bDiscriminator("simpleSecondaryVertexHighEffBJetTags");
532		topjet.btag_simpleSecondaryVertexHighPur=pat_topjet.bDiscriminator("simpleSecondaryVertexHighPurBJetTags");
#	Line 471 | Line 535 | NtupleWriter::analyze(const edm::Event&
535		topjet.btag_jetBProbability=pat_topjet.bDiscriminator("jetBProbabilityBJetTags");
536		topjet.btag_jetProbability=pat_topjet.bDiscriminator("jetProbabilityBJetTags");
537
538	+	const reco::GenJet *genj = pat_topjet.genJet();
539	+	if(genj){
540	+	topjet.genjet_pt = genj->pt();
541	+	topjet.genjet_eta = genj->eta();
542	+	topjet.genjet_phi = genj->phi();
543	+	topjet.genjet_energy = genj->energy();
544	+	}
545	+
546		for (unsigned int k = 0; k < pat_topjet.numberOfDaughters(); k++) {
547		Particle subjet_v4;
548		subjet_v4.pt = pat_topjet.daughter(k)->p4().pt();
#	Line 537 | Line 609 | NtupleWriter::analyze(const edm::Event&
609		}
610
611		// ------------- trigger -------------
612	<
613	<	edm::InputTag triggerEvent = edm::InputTag("hltTriggerSummaryAOD");
614	<	edm::Handle< trigger::TriggerEvent > dummy_TriggerEvent;
615	<	iEvent.getByLabel( edm::InputTag(triggerEvent.label(), triggerEvent.instance()), dummy_TriggerEvent );
616	<
617	<	const edm::Provenance *meta = dummy_TriggerEvent.provenance();
618	<	std::string nameProcess = meta->processName();
619	<	edm::InputTag triggerResultTag = edm::InputTag("TriggerResults");
620	<	triggerResultTag = edm::InputTag( triggerResultTag.label(), triggerResultTag.instance(), nameProcess );
621	<
622	<	edm::Handle<edm::TriggerResults> trigger;
623	<	iEvent.getByLabel(triggerResultTag, trigger);
624	<	const edm::TriggerResults& trig = *(trigger.product());
625	<
626	<	triggerResults.clear();
627	<	triggerNames.clear();
628	<	L1_prescale.clear();
629	<	HLT_prescale.clear();
630	<
631	<	edm::Service<edm::service::TriggerNamesService> tns;
632	<	std::vector<std::string> triggerNames_all;
633	<	tns->getTrigPaths(trig,triggerNames_all);
634	<
635	<	if (trig.size()!=triggerNames_all.size()) std::cout <<"ERROR: length of names and paths not the same: "<<triggerNames_all.size()<<","<<trig.size()<< std::endl;
636	<	for(unsigned int i=0; i<trig.size(); ++i){
637	<	std::vector<std::string>::const_iterator it = trigger_prefixes.begin();
638	<	for(; it!=trigger_prefixes.end(); ++it){
639	<	if(triggerNames_all[i].substr(0, it->size()) == *it)break;
640	<	}
641	<	if(it==trigger_prefixes.end()) continue;
642	<
643	<	//triggerResults.insert(std::pair<std::string, bool>(triggerNames[i],trig.accept(i)));
644	<	triggerResults.push_back(trig.accept(i));
645	<	if(newrun) triggerNames.push_back(triggerNames_all[i]);
646	<	if(isRealData){
647	<	std::pair<int, int> pre=hlt_cfg.prescaleValues(iEvent, iSetup, triggerNames_all[i]);
648	<	L1_prescale.push_back(pre.first);
649	<	HLT_prescale.push_back(pre.second);
650	<	}
651	<	}
652	<	//    for(std::map<std::string, bool>::const_iterator iter = triggerResults.begin(); iter!=triggerResults.end(); iter++){
653	<	//      std::cout << (*iter).first << "   " << (*iter).second << std::endl;
654	<	//    }
655	<	newrun=false;
612	>	if(doTrigger){
613	>	edm::InputTag triggerEvent = edm::InputTag("hltTriggerSummaryAOD");
614	>	edm::Handle< trigger::TriggerEvent > dummy_TriggerEvent;
615	>	iEvent.getByLabel( edm::InputTag(triggerEvent.label(), triggerEvent.instance()), dummy_TriggerEvent );
616	>
617	>	const edm::Provenance *meta = dummy_TriggerEvent.provenance();
618	>	std::string nameProcess = meta->processName();
619	>	edm::InputTag triggerResultTag = edm::InputTag("TriggerResults");
620	>	triggerResultTag = edm::InputTag( triggerResultTag.label(), triggerResultTag.instance(), nameProcess );
621	>
622	>	edm::Handle<edm::TriggerResults> trigger;
623	>	iEvent.getByLabel(triggerResultTag, trigger);
624	>	const edm::TriggerResults& trig = *(trigger.product());
625	>
626	>	triggerResults.clear();
627	>	triggerNames.clear();
628	>	L1_prescale.clear();
629	>	HLT_prescale.clear();
630	>
631	>	edm::Service<edm::service::TriggerNamesService> tns;
632	>	std::vector<std::string> triggerNames_all;
633	>	tns->getTrigPaths(trig,triggerNames_all);
634	>
635	>	if (trig.size()!=triggerNames_all.size()) std::cout <<"ERROR: length of names and paths not the same: "<<triggerNames_all.size()<<","<<trig.size()<< std::endl;
636	>	for(unsigned int i=0; i<trig.size(); ++i){
637	>	std::vector<std::string>::const_iterator it = trigger_prefixes.begin();
638	>	for(; it!=trigger_prefixes.end(); ++it){
639	>	if(triggerNames_all[i].substr(0, it->size()) == *it)break;
640	>	}
641	>	if(it==trigger_prefixes.end()) continue;
642	>
643	>	//triggerResults.insert(std::pair<std::string, bool>(triggerNames[i],trig.accept(i)));
644	>	triggerResults.push_back(trig.accept(i));
645	>	if(newrun) triggerNames.push_back(triggerNames_all[i]);
646	>	if(isRealData){
647	>	std::pair<int, int> pre=hlt_cfg.prescaleValues(iEvent, iSetup, triggerNames_all[i]);
648	>	L1_prescale.push_back(pre.first);
649	>	HLT_prescale.push_back(pre.second);
650	>	//std::cout <<  triggerNames_all[i] << " " << pre.first << " " <<pre.second << "   " << hlt_cfg.prescaleValue(iEvent, iSetup, triggerNames_all[i]) << std::endl;
651	>	}
652	>	}
653	>	//    for(std::map<std::string, bool>::const_iterator iter = triggerResults.begin(); iter!=triggerResults.end(); iter++){
654	>	//      std::cout << (*iter).first << "   " << (*iter).second << std::endl;
655	>	//    }
656	>	newrun=false;
657	>	}
658
659		// ------------- generator info -------------
660
#	Line 645 | Line 719 | NtupleWriter::analyze(const edm::Event&
719		index++;
720
721		//write out only top quarks and status 3 particles (works fine only for MadGraph)
722	<	if(abs(iter->pdgId())==6 || iter->status()==3){
722	>	if(abs(iter->pdgId())==6 || iter->status()==3 || doAllGenParticles){
723		GenParticle genp;
724		genp.charge = iter->charge();;
725		genp.pt = iter->p4().pt();
#	Line 675 | Line 749 | NtupleWriter::analyze(const edm::Event&
749
750		}
751
678	–
752		tr->Fill();
753	<
753	>	if(doLumiInfo)
754	>	previouslumiblockwasfilled=true;
755		}
756
757
#	Line 685 | Line 759 | NtupleWriter::analyze(const edm::Event&
759		void
760		NtupleWriter::beginJob()
761		{
762	+	if(doLumiInfo){
763	+	totalRecLumi=0;
764	+	totalDelLumi=0;
765	+	previouslumiblockwasfilled=false;
766	+	}
767		}
768
769		// ------------ method called once each job just after ending the event loop  ------------
#	Line 700 | Line 779 | NtupleWriter::endJob()
779		void
780		NtupleWriter::beginRun(edm::Run const& iRun, edm::EventSetup const&  iSetup)
781		{
782	<	bool setup_changed = false;
783	<	hlt_cfg.init(iRun, iSetup, "HLT", setup_changed);
784	<	newrun=true;
785	<
786	<	edm::ESHandle<JetCorrectorParametersCollection> JetCorParColl;
708	<	iSetup.get<JetCorrectionsRecord>().get("AK5PF",JetCorParColl);
709	<	JetCorrectorParameters const & JetCorPar = (*JetCorParColl)["Uncertainty"];
710	<	jecUnc = new JetCorrectionUncertainty(JetCorPar);
782	>	if(doTrigger){
783	>	bool setup_changed = false;
784	>	hlt_cfg.init(iRun, iSetup, "HLT", setup_changed);
785	>	newrun=true;
786	>	}
787
788	+	if(doJets || doTopJets){
789	+	edm::ESHandle<JetCorrectorParametersCollection> JetCorParColl;
790	+	iSetup.get<JetCorrectionsRecord>().get("AK5PF",JetCorParColl);
791	+	JetCorrectorParameters const & JetCorPar = (*JetCorParColl)["Uncertainty"];
792	+	jecUnc = new JetCorrectionUncertainty(JetCorPar);
793	+	}
794		}
795
796		// ------------ method called when ending the processing of a run  ------------
797		void
798		NtupleWriter::endRun(edm::Run const&, edm::EventSetup const&)
799		{
800	+	if(doLumiInfo)
801	+	std::cout << "total integ. luminosity: " << totalDelLumi <<"(del) " << totalRecLumi << "(rec)" << std::endl;
802		}
803
804		// ------------ method called when starting to processes a luminosity block  ------------
805		void
806	<	NtupleWriter::beginLuminosityBlock(edm::LuminosityBlock const&, edm::EventSetup const&)
806	>	NtupleWriter::beginLuminosityBlock(edm::LuminosityBlock const& lumi, edm::EventSetup const&)
807		{
808	+	if(doLumiInfo){
809	+	edm::Handle<LumiSummary> l;
810	+	lumi.getByLabel("lumiProducer", l);
811	+
812	+	//add lumi of lumi blocks without any event to next lumiblock
813	+	if(previouslumiblockwasfilled){
814	+	intgRecLumi=0;
815	+	intgDelLumi=0;
816	+	}
817	+	previouslumiblockwasfilled=false;
818	+
819	+	if (l.isValid()){;
820	+	intgRecLumi+=l->intgRecLumi()*6.37;
821	+	intgDelLumi+=l->intgDelLumi()*6.37;
822	+	totalRecLumi+=l->intgRecLumi()*6.37;
823	+	totalDelLumi+=l->intgDelLumi()*6.37;
824	+	}
825	+	//std::cout << "this lb: " <<l->intgRecLumi()*6.37 <<"   " << l->intgDelLumi()*6.37<<std::endl;
826	+	//std::cout << "summed:  "<< intgRecLumi << "   " << intgDelLumi << std::endl;
827	+	}
828		}
829
830		// ------------ method called when ending the processing of a luminosity block  ------------

Diff Legend

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