ViewVC Help

View File | Revision Log | Show Annotations | Root Listing

root/cvsroot/UserCode/MitHzz4l/LeptonSelection/src/MuonSelection.cc

Comparing UserCode/MitHzz4l/LeptonSelection/src/MuonSelection.cc (file contents):
Revision 1.22 by anlevin, Mon May 28 12:37:10 2012 UTC vs.
Revision 1.37 by dkralph, Tue Oct 23 10:42:07 2012 UTC

#	Line 19 | Line 19 | extern Float_t computePFMuonIso(const mi
19		const mithep::Array<mithep::PFCandidate> * pfCandidates,
20		const Double_t dRMax);
21
22	+	bool muonPOG2012CutBasedIDTight(const mithep::Muon *mu, const mithep::Vertex * vtx, const mithep::Array<mithep::PFCandidate> * pfCandidates, const mithep::Array<mithep::PileupEnergyDensity> * puEnergyDensity)
23	+	{
24	+
25	+	Float_t charged_iso = 0;
26	+
27	+	Float_t sumChargedHadronPt = 0;
28	+	Float_t sumNeutralHadronPt = 0;
29	+	Float_t sumPhotonPt = 0;
30	+	Float_t sumPUPt = 0;
31	+
32	+	for(unsigned k=0; k<pfCandidates->GetEntries(); ++k) {
33	+	const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[k]);
34	+	Double_t dr = mithep::MathUtils::DeltaR(mu->Phi(),mu->Eta(), pf->Phi(), pf->Eta());
35	+	if(dr > 0.4) continue;
36	+	if(pf->TrackerTrk() && mu->TrackerTrk() && pf->TrackerTrk() == mu->TrackerTrk()) continue;
37	+
38	+	if(PFnoPUflag[k]){ //pf no pileup
39	+	if(abs(pf->PFType()) == mithep::PFCandidate::eHadron)
40	+	sumChargedHadronPt += pf->Pt();
41	+	else if (pf->Pt() > 0.5 && abs(pf->PFType()) == mithep::PFCandidate::eNeutralHadron)
42	+	sumNeutralHadronPt += pf->Pt();
43	+	else if (pf->Pt() > 0.5 && abs(pf->PFType()) == mithep::PFCandidate::eGamma)
44	+	sumPhotonPt += pf->Pt();
45	+	}
46	+	else {//pf pile up
47	+	if(pf->Charge() != 0)
48	+	sumPUPt += pf->Pt();
49	+	}
50	+
51	+	}
52	+	double iso_sum =  sumChargedHadronPt+ max(0.,sumNeutralHadronPt+sumPhotonPt-0.5*sumPUPt);
53	+
54	+	bool pass = kTRUE;
55	+	if(!mu->IsGlobalMuon()) pass = kFALSE;
56	+
57	+	//there is not necessarily a global track or a tracker track if is global muon is false
58	+	if(!pass) return pass;
59	+
60	+	if(!mu->IsPFMuon()) pass = kFALSE;
61	+	if(mu->GlobalTrk()->RChi2() > 10) pass= kFALSE;
62	+	if(mu->NValidHits() == 0 ) pass= kFALSE;
63	+	if(mu->NMatches() <= 1) pass = kFALSE;
64	+	if(fabs(mu->TrackerTrk()->D0Corrected(*vtx)) > 0.2) pass = kFALSE;
65	+	if(fabs(mu->TrackerTrk()->DzCorrected(*vtx)) > 0.5) pass = kFALSE;
66	+	if(mu->BestTrk()->NPixelHits() == 0) pass = kFALSE;
67	+	if(mu->NTrkLayersHit() <= 5) pass = kFALSE;
68	+	if(iso_sum/mu->Pt() > 0.12 ) pass = kFALSE;
69	+
70	+	return pass;
71	+
72	+
73	+	}
74	+	//----------------------------------------------------------------------------------------
75	+	bool muon2012CutBasedIDTightVersionWithOldIsoThatWorksOn2011(ControlFlags &ctrl, const mithep::Muon *mu, const mithep::Vertex * vtx, const mithep::Array<mithep::PFCandidate> * pfCandidates,
76	+	const mithep::Array<mithep::PileupEnergyDensity> * puEnergyDensity,   mithep::MuonTools::EMuonEffectiveAreaTarget EffectiveAreaVersion
77	+	)
78	+	{
79	+
80	+	mithep::MuonTools   muT;
81	+
82	+	Float_t charged_iso = 0;
83	+	Float_t neutral_iso = 0;
84	+	Float_t gamma_iso = 0;
85	+
86	+	for(int k=0; k<pfCandidates->GetEntries(); ++k) {
87	+
88	+	if( !(PFnoPUflag[k]) ) continue;
89	+	const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[k]);
90	+
91	+	Double_t dr = mithep::MathUtils::DeltaR(mu->Phi(),mu->Eta(), pf->Phi(), pf->Eta());
92	+	if (dr > 0.4) continue;
93	+
94	+	if (pf->Charge() != 0 && (pf->HasTrackerTrk()||pf->HasGsfTrk()) ) {
95	+
96	+	if (abs(pf->PFType()) == mithep::PFCandidate::eElectron || abs(pf->PFType()) == mithep::PFCandidate::eMuon) continue;
97	+
98	+	charged_iso += pf->Pt();
99	+	}
100	+
101	+	else if (abs(pf->PFType()) == mithep::PFCandidate::eGamma) {
102	+
103	+	gamma_iso += pf->Pt();
104	+	}
105	+
106	+	else {
107	+	neutral_iso += pf->Pt();
108	+	}
109	+	}
110	+
111	+	double iso_sum=999;
112	+	double rho;
113	+	if(ctrl.era == 2012) {
114	+	EffectiveAreaVersion  = mithep::MuonTools::kMuEAData2012;
115	+	rho = puEnergyDensity->At(0)->RhoKt6PFJetsCentralNeutral();
116	+
117	+	iso_sum = charged_iso + fmax(0.0,(gamma_iso + neutral_iso
118	+	-rho*muT.MuonEffectiveArea(muT.kMuGammaAndNeutralHadronIso04,
119	+	mu->Eta(),EffectiveAreaVersion)));
120	+	} else if(ctrl.era == 2011) {
121	+	setMuonRhoAndEaVersion(rho,EffectiveAreaVersion,puEnergyDensity);
122	+	iso_sum = charged_iso + fmax(0.0,(gamma_iso + neutral_iso
123	+	-rho*muT.MuonEffectiveArea(muT.kMuGammaAndNeutralHadronIso04,
124	+	mu->Eta(),EffectiveAreaVersion)));
125	+	} else assert(0);
126	+
127	+	bool pass = kTRUE;
128	+	if(!mu->IsGlobalMuon()) pass = kFALSE;
129	+
130	+	//there is not necessarily a global track or a tracker track if is global muon is false
131	+	if(!pass) return pass;
132	+
133	+	if(!mu->IsPFMuon() && ctrl.era==2012)                 pass = kFALSE;
134	+	if(mu->GlobalTrk()->RChi2() > 10)                     pass = kFALSE;
135	+	if(mu->NValidHits() == 0 )                            pass = kFALSE;
136	+	if(mu->NMatches() <= 1)                               pass = kFALSE;
137	+	if(fabs(mu->TrackerTrk()->D0Corrected(*vtx)) > 0.2)   pass = kFALSE;
138	+	if(fabs(mu->TrackerTrk()->DzCorrected(*vtx)) > 0.5)   pass = kFALSE;
139	+	if(mu->BestTrk()->NPixelHits() == 0)                  pass = kFALSE;
140	+	if(mu->NTrkLayersHit() <= 5)                          pass = kFALSE;
141	+	if(iso_sum/mu->Pt() > 0.12 )                          pass = kFALSE;
142	+
143	+	return pass;
144	+
145	+
146	+	}
147	+
148	+	bool trackToPFMuonSelection( const mithep::Track *track, const mithep::MuonCol * muoncol )
149	+	{
150	+	bool pass = false;
151	+	for(UInt_t i=0; i<muoncol->GetEntries(); ++i) {
152	+	const mithep::Muon *mu = muoncol->At(i);
153	+	if(mu->TrackerTrk() == track && (mu->IsGlobalMuon() || mu->IsTrackerMuon()) && mu->IsPFMuon())
154	+	pass = true;
155	+	}
156	+
157	+	return pass;
158	+
159	+	}
160	+
161		//--------------------------------------------------------------------------------------------------
162		SelectionStatus muonDummyVeto(ControlFlags &ctrl,
163		const mithep::Muon *muon,
#	Line 61 | Line 200 | SelectionStatus noPreSelection( ControlF
200		if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
201		return status;
202		}
64	–
203		//--------------------------------------------------------------------------------------------------
204	<	SelectionStatus muonPreSelection( ControlFlags &ctrl,
205	<	const mithep::Muon * mu,
206	<	const mithep::Vertex * vtx,
207	<	const mithep::Array<mithep::PFCandidate> * pfCandidates )
204	>	SelectionStatus muonReferencePreSelection( ControlFlags &ctrl,
205	>	const mithep::Muon * mu,
206	>	const mithep::Vertex * vtx,
207	>	const mithep::Array<mithep::PFCandidate> * pfCandidates )
208		//--------------------------------------------------------------------------------------------------
209		{
210		bool ret = true;
211
212	+	if(ctrl.debug) cout << "inside muonpresel ... " << endl;
213		ret &= ( fabs(mu->Ip3dPVSignificance()) < 100 );
214	+	if(ctrl.debug) cout << "and pass IP (" << mu->Ip3dPVSignificance() << ") ? ... " << ret << endl;
215		ret &= ( mu->Pt() >= 5 );
216	+	if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;
217		ret &= ( fabs(mu->Eta()) <= 2.4 );
218	<	ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
219	<	//   if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
220	<	//     ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
221	<	ret &= (  mu->IsoR03SumPt()/mu->Pt() < 0.7 );
222	<
218	>	if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;
219	>	ret &= ( (mu->IsTrackerMuon() && ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated)) || mu->IsGlobalMuon() );
220	>	if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
221	>	ret &= (mu->HasTrackerTrk() &&
222	>	fabs(mu->TrackerTrk()->D0Corrected(*vtx)) < 0.5 &&
223	>	fabs(mu->TrackerTrk()->DzCorrected(*vtx)) < 1.0);
224	>	if(ctrl.debug) cout << "d0 and dz? ... " << ret << endl;
225	>
226		SelectionStatus status;
227		if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
228		if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
229		return status;
230		}
87	–
88	–
89	–
231		//--------------------------------------------------------------------------------------------------
232	<	SelectionStatus muonReferencePreSelection( ControlFlags &ctrl,
233	<	const mithep::Muon * mu,
234	<	const mithep::Vertex * vtx,
235	<	const mithep::Array<mithep::PFCandidate> * pfCandidates )
232	>	SelectionStatus muonPreSelectionNoD0IP( ControlFlags &ctrl,
233	>	const mithep::Muon * mu,
234	>	const mithep::Vertex * vtx,
235	>	const mithep::Array<mithep::PFCandidate> * pfCandidates )
236		//--------------------------------------------------------------------------------------------------
237		{
238		bool ret = true;
239
99	–	if(ctrl.debug) cout << "inside muonpresel ... " << endl;
100	–	ret &= ( fabs(mu->Ip3dPVSignificance()) < 100 );
101	–	if(ctrl.debug) cout << "and pass IP (" << mu->Ip3dPVSignificance() << ") ? ... " << ret << endl;
240		ret &= ( mu->Pt() >= 5 );
103	–	if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;
241		ret &= ( fabs(mu->Eta()) <= 2.4 );
105	–	if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;
242		ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
243	<	if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
244	<	ret &= (mu->HasTrackerTrk() && mu->TrackerTrk()->D0Corrected(*vtx) < 0.5 && mu->TrackerTrk()->DzCorrected(*vtx) < 1.0);
245	<	if(ctrl.debug) cout << "d0 and dz? ... " << ret << endl;
243	>	// they put this back in for the h.c.p. sync...
244	>	if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
245	>	ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
246	>	ret &= (mu->HasTrackerTrk() &&
247	>	fabs(mu->TrackerTrk()->DzCorrected(*vtx)) < 1.0);
248
111	–	//  if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
112	–	//    ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
113	–	//  if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;
114	–	//  ret &= ( mu->BestTrk()->DzCorrected(vtx) < 0.1);
115	–	//  if( ctrl.debug ) cout << "elepresel : ret after dZcorr : " << ret <<  endl;
116	–
117	–
249		SelectionStatus status;
250		if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
251		if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
252		return status;
253		}
123	–
124	–
125	–
126	–
254		//--------------------------------------------------------------------------------------------------
255		SelectionStatus muonIDMVASelection(ControlFlags &ctrl,
256		const mithep::Muon *mu,
#	Line 206 | Line 333 | SelectionStatus muonIDPFSelection(Contro
333		//--------------------------------------------------------------------------------------------------
334		{
335		bool pass = false;
336	<	SelectionStatus status; // init'ed to FAIL
336	>	SelectionStatus status;
337
338		// check that it matches to a PF muon
339		for( int i=0; i<pfCandidates->GetEntries(); i++ ) {
340	<	// tmp
214	<	if( !(PFnoPUflag[i]) ) continue; // my PF no PU hack
340	>	if( !(PFnoPUflag[i]) ) continue;
341
342		const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[i]);
343		if( (pf->TrackerTrk() == mu->TrackerTrk()) && abs(pf->PFType()) == mithep::PFCandidate::eMuon ) {
#	Line 219 | Line 345 | SelectionStatus muonIDPFSelection(Contro
345		break;
346		}
347		}
222	–	//
223	–	//      if(mu->IsPFMuon()) pass = true;
224	–	//      else pass = false;
348
226	–	//
227	–	// NB : ipsig here for reference synch ...
228	–	//
229	–	#ifdef SYNC
230	–	if( pass && fabs(mu->Ip3dPVSignificance()) < 4 ) {
231	–	status.orStatus(SelectionStatus::LOOSEID);
232	–	status.orStatus(SelectionStatus::TIGHTID);
233	–	}
234	–	#else
349		if( pass ) {
350		status.orStatus(SelectionStatus::LOOSEID);
351		status.orStatus(SelectionStatus::TIGHTID);
352		}
239	–	#endif
353
354		return status;
355		}
356	+	//----------------------------------------------------------------------------------------
357	+	SelectionStatus PassWwMuonSel(const mithep::Muon *mu, const mithep::Vertex *vtx,
358	+	const mithep::Array<mithep::PFCandidate> *pfArr
359	+	)
360	+	{
361	+	bool passID=true;
362	+	bool passIso=true;
363	+
364	+	if (mu->Pt() < 5)
365	+	passID = false;
366	+	if (fabs(mu->Eta()) > 2.4)
367	+	passID = false;
368	+	if (mu->BestTrk()->PtErr()/mu->Pt() > 0.1)
369	+	passID = false;
370	+	if (fabs(mu->BestTrk()->DzCorrected(*vtx)) > 0.1)
371	+	passID = false;
372	+	Bool_t isGlobal  = (mu->IsGlobalMuon()) && (mu->BestTrk()->RChi2() < 10) &&
373	+	(mu->NMatches() > 1) && (mu->NValidHits() > 0);
374	+	Bool_t isTracker = (mu->IsTrackerMuon()) &&
375	+	(mu->Quality().Quality(mithep::MuonQuality::TMLastStationTight));
376	+	if (!isGlobal && !isTracker)
377	+	passID = false;
378	+	int ntrkhits = (mu->HasTrackerTrk()) ? mu->TrackerTrk()->NHits() : 0;
379	+	if (ntrkhits < 10)
380	+	passID = false;
381	+	if (mu->BestTrk()->NPixelHits() < 1)
382	+	passID = false;
383	+	if (fabs(mu->BestTrk()->D0Corrected(*vtx)) > 0.02)
384	+	passID = false;
385	+
386	+	// note: this isn't really ww muon isolation
387	+	float pfiso = computePFMuonIso(mu,vtx,pfArr,0.3);
388	+
389	+	if (pfiso > 0.2*mu->Pt())
390	+	passIso = false;
391	+
392	+	SelectionStatus status;
393	+	if(passID)  status |= SelectionStatus::TIGHTIDANDPRE;
394	+	if(passIso) status |= SelectionStatus::TIGHTISOANDPRE;
395	+
396	+	return status;
397	+
398	+	}
399	+	// //--------------------------------------------------------------------------------------------------
400	+	// SelectionStatus muonPreSelectionNoD0DzIP( ControlFlags &ctrl,
401	+	//                                        const mithep::Muon * mu,
402	+	//                                        const mithep::Vertex * vtx,
403	+	//                                        const mithep::Array<mithep::PFCandidate> * pfCandidates )
404	+	// //--------------------------------------------------------------------------------------------------
405	+	// {
406	+	//   bool ret = true;
407	+
408	+	//   if(ctrl.debug) cout << "inside muonpresel ... " << endl;
409	+	//   ret &= ( mu->Pt() >= 5 );
410	+	//   if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;
411	+	//   ret &= ( fabs(mu->Eta()) <= 2.4 );
412	+	//   if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;
413	+	//   ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
414	+	//   if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
415	+
416	+	//   //  if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
417	+	//   //    ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
418	+	//   //  if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;
419	+	//   //  ret &= ( mu->BestTrk()->DzCorrected(vtx) < 0.1);
420	+	//   //  if( ctrl.debug ) cout << "elepresel : ret after dZcorr : " << ret <<  endl;
421	+
422	+
423	+	//   SelectionStatus status;
424	+	//   if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
425	+	//   if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
426	+	//   return status;
427	+	// }

Diff Legend

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