ViewVC Help

View File | Revision Log | Show Annotations | Root Listing

root/cvsroot/UserCode/VHbbAnalysis/HbbAnalyzer/plugins/HbbAnalyzerNew.cc

Comparing UserCode/VHbbAnalysis/HbbAnalyzer/plugins/HbbAnalyzerNew.cc (file contents):
Revision 1.1 by tboccali, Wed Jun 8 17:25:32 2011 UTC vs.
Revision 1.8 by tboccali, Tue Jun 21 15:49:01 2011 UTC

#	Line 17 | Line 17 | Implementation:
17		//
18		//
19
20	<	#include "UserCode/HbbAnalyzer/interface/HbbAnalyzerNew.h"
21	<	#include "UserCode/HbbAnalyzer/interface/VHbbEvent.h"
20	>	#include "VHbbAnalysis/HbbAnalyzer/interface/HbbAnalyzerNew.h"
21	>	#include "VHbbAnalysis/HbbAnalyzer/interface/VHbbEvent.h"
22	>
23	>	#include "DataFormats/GeometryVector/interface/VectorUtil.h"
24	>
25
26		#define GENPTOLOR(a) TLorentzVector((a).px(), (a).py(), (a).pz(), (a).energy())
27		#define GENPTOLORP(a) TLorentzVector((a)->px(), (a)->py(), (a)->pz(), (a)->energy())
28
29		HbbAnalyzerNew::HbbAnalyzerNew(const edm::ParameterSet& iConfig):
30	<	eleLabel_(iConfig.getUntrackedParameter<edm::InputTag>("electronTag")),
31	<	muoLabel_(iConfig.getUntrackedParameter<edm::InputTag>("muonTag")),
32	<	jetLabel_(iConfig.getUntrackedParameter<edm::InputTag>("jetTag")),
33	<	subjetLabel_(iConfig.getUntrackedParameter<edm::InputTag>("subjetTag")),
34	<	simplejet1Label_(iConfig.getUntrackedParameter<edm::InputTag>("simplejet1Tag")),
35	<	simplejet2Label_(iConfig.getUntrackedParameter<edm::InputTag>("simplejet2Tag")),
36	<	simplejet3Label_(iConfig.getUntrackedParameter<edm::InputTag>("simplejet3Tag")),
37	<	simplejet4Label_(iConfig.getUntrackedParameter<edm::InputTag>("simplejet4Tag")),
38	<	tauLabel_(iConfig.getUntrackedParameter<edm::InputTag>("tauTag")),
39	<	metLabel_(iConfig.getUntrackedParameter<edm::InputTag>("metTag")),
40	<	phoLabel_(iConfig.getUntrackedParameter<edm::InputTag>("photonTag")),
41	<	dimuLabel_(iConfig.getUntrackedParameter<edm::InputTag>("dimuTag")),
42	<	dielecLabel_(iConfig.getUntrackedParameter<edm::InputTag>("dielecTag")),
43	<	hltResults_(iConfig.getUntrackedParameter<edm::InputTag>("hltResultsTag")){
30	>	eleLabel_(iConfig.getParameter<edm::InputTag>("electronTag")),
31	>	muoLabel_(iConfig.getParameter<edm::InputTag>("muonTag")),
32	>	jetLabel_(iConfig.getParameter<edm::InputTag>("jetTag")),
33	>	subjetLabel_(iConfig.getParameter<edm::InputTag>("subjetTag")),
34	>	simplejet1Label_(iConfig.getParameter<edm::InputTag>("simplejet1Tag")),
35	>	simplejet2Label_(iConfig.getParameter<edm::InputTag>("simplejet2Tag")),
36	>	simplejet3Label_(iConfig.getParameter<edm::InputTag>("simplejet3Tag")),
37	>	simplejet4Label_(iConfig.getParameter<edm::InputTag>("simplejet4Tag")),
38	>	tauLabel_(iConfig.getParameter<edm::InputTag>("tauTag")),
39	>	metLabel_(iConfig.getParameter<edm::InputTag>("metTag")),
40	>	phoLabel_(iConfig.getParameter<edm::InputTag>("photonTag")),
41	>	dimuLabel_(iConfig.getParameter<edm::InputTag>("dimuTag")),
42	>	dielecLabel_(iConfig.getParameter<edm::InputTag>("dielecTag")),
43	>	hltResults_(iConfig.getParameter<edm::InputTag>("hltResultsTag")),
44	>	runOnMC_(iConfig.getParameter<bool>("runOnMC")) {
45	>
46		//
47		// put the setwhatproduced etc etc
48
#	Line 311 | Line 316 | HbbAnalyzerNew::produce(edm::Event& iEve
316
317		Handle<GenParticleCollection> genParticles;
318
314	–	bool isMC=1;
319		bool printJet=0;
320
321
322	<	if(isMC){
322	>	if(runOnMC_){
323
324		int Hin=-99,Win=-99,Zin=-99,bin=-99,bbarin=-99;
325
#	Line 513 | Line 517 | HbbAnalyzerNew::produce(edm::Event& iEve
517		VHbbEvent::SimpleJet sj;
518		sj.flavour = jet_iter->partonFlavour();
519
516	–
520		sj.tche=jet_iter->bDiscriminator("trackCountingHighEffBJetTags");
521		sj.tchp=jet_iter->bDiscriminator("trackCountingHighPurBJetTags");
522		sj.jp=jet_iter->bDiscriminator("jetProbabilityBJetTags");
#	Line 524 | Line 527 | HbbAnalyzerNew::produce(edm::Event& iEve
527		sj.charge=jet_iter->jetCharge();
528		sj.ntracks=jet_iter->associatedTracks().size();
529		sj.fourMomentum=GENPTOLORP(jet_iter);
530	+	sj.chargedTracksFourMomentum=(getChargedTracksMomentum(&*(jet_iter)));
531	+
532	+	//
533	+	// add tVector
534	+	//
535	+	sj.tVector = getTvect(&(*jet_iter));
536
537		Particle::LorentzVector p4Jet = jet_iter->p4();
538
539	<	if(isMC){
539	>	if(runOnMC_){
540		double minb1DR=9999.;
541		for(size_t i = 0; i < genParticles->size(); ++ i) {
542		const GenParticle & p = (*genParticles)[i];
#	Line 535 | Line 544 | HbbAnalyzerNew::produce(edm::Event& iEve
544		if(abs(id)<=6 || id==21 || id==23 || abs(id)==24){
545		double bb1DR=TMath::Sqrt((p.eta()-p4Jet.eta())*(p.eta()-p4Jet.eta())+
546		(p.phi()-p4Jet.phi())*(p.phi()-p4Jet.phi()));
547	<	if(bb1DR<minb1DR) {minb1DR=bb1DR; sj.b1BestMCid=id; if(p.mother()!=0) sj.b1BestMCmomid=p.mother()->pdgId();}
547	>	if(bb1DR<minb1DR) {minb1DR=bb1DR; sj.bestMCid=id; if(p.mother()!=0) sj.bestMCmomid=p.mother()->pdgId();}
548		}
549		}
550		} //isMC
#	Line 560 | Line 569 | HbbAnalyzerNew::produce(edm::Event& iEve
569		sj.charge=jet_iter->jetCharge();
570		sj.ntracks=jet_iter->associatedTracks().size();
571		sj.fourMomentum=GENPTOLORP(jet_iter);
572	+	sj.chargedTracksFourMomentum=(getChargedTracksMomentum(&*(jet_iter)));
573	+	sj.tVector = getTvect(&(*jet_iter));
574
575		Particle::LorentzVector p4Jet = jet_iter->p4();
576
577	<	if(isMC){
577	>	if(runOnMC_){
578		double minb2DR=9999.;
579		for(size_t i = 0; i < genParticles->size(); ++ i) {
580		const GenParticle & p = (*genParticles)[i];
#	Line 571 | Line 582 | HbbAnalyzerNew::produce(edm::Event& iEve
582		if(abs(id)<=6 || id==21 || id==23 || abs(id)==24){
583		double bb2DR=TMath::Sqrt((p.eta()-p4Jet.eta())*(p.eta()-p4Jet.eta())+
584		(p.phi()-p4Jet.phi())*(p.phi()-p4Jet.phi()));
585	<	if(bb2DR<minb2DR) {minb2DR=bb2DR; sj.b1BestMCid=id; if(p.mother()!=0) sj.b1BestMCmomid=p.mother()->pdgId();}
585	>	if(bb2DR<minb2DR) {minb2DR=bb2DR; sj.bestMCid=id; if(p.mother()!=0) sj.bestMCmomid=p.mother()->pdgId();}
586		}
587		}
588		}   //isMC
#	Line 623 | Line 634 | HbbAnalyzerNew::produce(edm::Event& iEve
634		hj.subFourMomentum.push_back(GENPTOLORP(icandJet));
635		hj.etaSub.push_back(icandJet->eta());
636		hj.phiSub.push_back(icandJet->phi());
626	–
637
638		}
639		hbbInfo->hardJets.push_back(hj);
#	Line 648 | Line 658 | HbbAnalyzerNew::produce(edm::Event& iEve
658		VHbbEvent::SimpleJet sj;
659
660		sj.flavour = subjet_iter->partonFlavour();
661	<
661	>	sj.tVector = getTvect(&(*subjet_iter));
662		sj.tche=subjet_iter->bDiscriminator("trackCountingHighEffBJetTags");
663		sj.tchp=subjet_iter->bDiscriminator("trackCountingHighPurBJetTags");
664		sj.jp=subjet_iter->bDiscriminator("jetProbabilityBJetTags");
#	Line 659 | Line 669 | HbbAnalyzerNew::produce(edm::Event& iEve
669		sj.charge=subjet_iter->jetCharge();
670		sj.ntracks=subjet_iter->associatedTracks().size();
671		sj.fourMomentum=GENPTOLORP(subjet_iter);
672	<
672	>	sj.fourMomentum=(getChargedTracksMomentum(&*(subjet_iter)));
673		hbbInfo->subJets.push_back(sj);
674
675		}
#	Line 730 | Line 740 | HbbAnalyzerNew::produce(edm::Event& iEve
740		const reco::HitPattern& q = gTrack->hitPattern();
741		mf.globChi2=gTrack.get()->normalizedChi2();
742		mf.globNHits=q.numberOfValidMuonHits();
743	+	mf. validMuStations = q. muonStationsWithValidHits();
744		}else{
745		mf.globChi2=-99;
746		mf.globNHits=-99;
747		}
748	+	//
749	+	// add stamuon
750	+
751	+	//    if (mu->isStandAloneMuon()) {
752	+	//      reco::TrackRef sta = mu->standAloneMuon();
753	+	//
754	+	//    }
755	+
756
757		//     int muInfo[12];
758		//     fillMuBlock(mu,  muInfo);
759	<	if(isMC){
759	>	if(runOnMC_){
760		const GenParticle* muMc = mu->genLepton();
761		if(muMc!=0){
762		mf.mcId=muMc->pdgId();
#	Line 761 | Line 780 | HbbAnalyzerNew::produce(edm::Event& iEve
780		ef.hIso=elec->hcalIso();
781		Geom::Phi<double> deltaphi(elec->superCluster()->phi()-atan2(hbbInfo->calomet.fourMomentum.Py(),hbbInfo->calomet.fourMomentum.Px()));
782		ef.acop = deltaphi.value();
783	<
784	<	if(isMC){
783	>	//
784	>	// fill eleids
785	>	//
786	>	ef.id95 = elec->electronID("simpleEleId95cIso");
787	>	ef.id85 = elec->electronID("simpleEleId85cIso");
788	>	ef.id70 = elec->electronID("simpleEleId70cIso");
789	>	ef.id95r = elec->electronID("simpleEleId95relIso");
790	>	ef.id70r = elec->electronID("simpleEleId70relIso");
791	>	ef.id85r = elec->electronID("simpleEleId85relIso");
792	>
793	>	if(runOnMC_){
794		const GenParticle* elecMc = elec->genLepton();
795		if(elecMc!=0){
796		ef.mcId=elecMc->pdgId();
#	Line 862 | Line 890 | HbbAnalyzerNew::produce(edm::Event& iEve
890		const reco::HitPattern& q = gTrack->hitPattern();
891		df.daughter1.globNHits=q.numberOfValidMuonHits();
892		df.daughter1.globChi2=gTrack.get()->normalizedChi2();
893	+	df.daughter1.validMuStations = q. muonStationsWithValidHits();
894		}
895		if(muonDau1.isGlobalMuon()){
896		TrackRef gTrack = muonDau1.globalTrack();
897		const reco::HitPattern& q = gTrack->hitPattern();
898		df.daughter2.globNHits=q.numberOfValidMuonHits();
899		df.daughter2.globChi2=gTrack.get()->normalizedChi2();
900	+	df.daughter2.validMuStations = q. muonStationsWithValidHits();
901		}
902
903		}
#	Line 900 | Line 930 | HbbAnalyzerNew::produce(edm::Event& iEve
930		df.daughter1.hIso = elecDau0.hcalIso();
931		df.daughter2.hIso = elecDau1.hcalIso();
932
933	<
933	>	// ids
934	>	df.daughter1.id95 = elecDau0.electronID("simpleEleId95cIso");
935	>	df.daughter1.id85 = elecDau0.electronID  ("simpleEleId85cIso");
936	>	df.daughter1.id70 = elecDau0.electronID  ("simpleEleId70cIso");
937	>	df.daughter1.id95r = elecDau0.electronID ("simpleEleId95relIso");
938	>	df.daughter1.id85r = elecDau0.electronID ("simpleEleId85relIso");
939	>	df.daughter1.id70r = elecDau0.electronID ("simpleEleId70relIso");
940	>
941	>
942	>	df.daughter2.id95 = elecDau1.electronID("simpleEleId95cIso");
943	>	df.daughter2.id85 = elecDau1.electronID  ("simpleEleId85cIso");
944	>	df.daughter2.id70 = elecDau1.electronID  ("simpleEleId70cIso");
945	>	df.daughter2.id95r = elecDau1.electronID ("simpleEleId95relIso");
946	>	df.daughter2.id85r = elecDau1.electronID ("simpleEleId85relIso");
947	>	df.daughter2.id70r = elecDau1.electronID ("simpleEleId70relIso");
948	>
949		hbbInfo->diElectronInfo.push_back(df);
950
951		}
#	Line 939 | Line 984 | void
984		HbbAnalyzerNew::endJob() {
985		}
986
987	+	TVector2 HbbAnalyzerNew::getTvect( const pat::Jet* patJet ){
988	+
989	+	TVector2 t_Vect(0,0);
990	+	TVector2 null(0,0);
991	+	TVector2 ci(0,0);
992	+	TLorentzVector pi(0,0,0,0);
993	+	TLorentzVector J(0,0,0,0);
994	+	TVector2 r(0,0);
995	+	double patJetpfcPt = 1e10;
996	+	double r_mag = 1e10;
997	+	unsigned int nOfconst = 0;
998	+
999	+
1000	+	if (patJet->isPFJet() == false) {
1001	+	return t_Vect;
1002	+	}
1003	+
1004	+
1005	+	//re-reconstruct the jet direction with the charged tracks
1006	+	std::vector<reco::PFCandidatePtr>
1007	+	patJetpfc = patJet->getPFConstituents();
1008	+	for(size_t idx = 0; idx < patJetpfc.size(); idx++){
1009	+	if( patJetpfc.at(idx)->charge() != 0 ){
1010	+	pi.SetPtEtaPhiE( patJetpfc.at(idx)->pt(), patJetpfc.at(idx)->eta(), patJetpfc.at(idx)->phi(), patJetpfc.at(idx)->energy() );
1011	+	J += pi;
1012	+	nOfconst++;
1013	+	}
1014	+	}
1015	+	// if there are less than two charged tracks do not calculate the pull (there is not enough info). It returns a null vector
1016	+
1017	+	if( nOfconst < 2 )
1018	+	return null;
1019	+
1020	+
1021	+
1022	+	TVector2 v_J( J.Rapidity(), J.Phi() );
1023	+	//calculate TVector using only charged tracks
1024	+	for(size_t idx = 0; idx < patJetpfc.size(); idx++){
1025	+	if( patJetpfc.at(idx)->charge() != 0  ){
1026	+	patJetpfcPt = patJetpfc.at(idx)->pt();
1027	+	pi.SetPtEtaPhiE( patJetpfc.at(idx)->pt(), patJetpfc.at(idx)->eta(), patJetpfc.at(idx)->phi(), patJetpfc.at(idx)->energy() );
1028	+	r.Set( pi.Rapidity() - J.Rapidity(), Geom::deltaPhi( patJetpfc.at(idx)->phi(), J.Phi() ) );
1029	+	r_mag = r.Mod();
1030	+	t_Vect += ( patJetpfcPt / J.Pt() ) * r_mag * r;
1031	+	}
1032	+	}
1033	+
1034	+
1035	+	return t_Vect;
1036	+
1037	+	}
1038	+
1039	+	TLorentzVector HbbAnalyzerNew::getChargedTracksMomentum(const pat::Jet* patJet ){
1040	+	//  return TLorentzVector();
1041	+	TLorentzVector pi(0,0,0,0);
1042	+	TLorentzVector v_j1(0,0,0,0);
1043	+
1044	+
1045	+	//  std::cout <<"fff ECCCCCCOOOOO "<<patJet->isPFJet()<<std::endl;
1046	+
1047	+	if (patJet->isPFJet() == false ){
1048	+	v_j1 = GENPTOLORP(patJet);
1049	+	return v_j1;
1050	+	}
1051	+	std::vector<reco::PFCandidatePtr>
1052	+	j1pfc = patJet->getPFConstituents();
1053	+	for(size_t idx = 0; idx < j1pfc.size(); idx++){
1054	+	if( j1pfc.at(idx)->charge() != 0 ){
1055	+	pi.SetPtEtaPhiE( j1pfc.at(idx)->pt(), j1pfc.at(idx)->eta(), j1pfc.at(idx)->phi(), j1pfc.at(idx)->energy() );
1056	+	v_j1 += pi;
1057	+	}
1058	+	}
1059	+	return v_j1;
1060	+	//re-
1061	+	}
1062	+
1063		//define this as a plug-in
1064		DEFINE_FWK_MODULE(HbbAnalyzerNew);

Diff Legend

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