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.17 by khahn, Thu May 10 22:36:12 2012 UTC vs.
Revision 1.28 by dkralph, Wed Jun 13 14:11:22 2012 UTC

#	Line 13 | Line 13
13		mithep::MuonIDMVA * muIDMVA;
14		mithep::MuonTools muTools;
15
16	+	extern vector<bool> PFnoPUflag;
17		extern Float_t computePFMuonIso(const mithep::Muon *mu,
18	<	const mithep::Vertex & vtx,
18	>	const mithep::Vertex * vtx,
19		const mithep::Array<mithep::PFCandidate> * pfCandidates,
20		const Double_t dRMax);
21
22	+	bool muon2012CutBasedIDTight(const mithep::Muon *mu, const mithep::Vertex * vtx, const mithep::Array<mithep::PFCandidate> * pfCandidates, const mithep::Array<mithep::PileupEnergyDensity> * puEnergyDensity)
23	+	{
24	+
25	+	mithep::MuonTools   muT;
26	+
27	+	Float_t charged_iso = 0;
28	+	Float_t neutral_iso = 0;
29	+	Float_t gamma_iso = 0;
30	+
31	+	for(int k=0; k<pfCandidates->GetEntries(); ++k) {
32	+
33	+	if( !(PFnoPUflag[k]) ) continue;
34	+	const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[k]);
35	+
36	+	Double_t dr = mithep::MathUtils::DeltaR(mu->Phi(),mu->Eta(), pf->Phi(), pf->Eta());
37	+	if (dr > 0.4) continue;
38	+
39	+	if (pf->Charge() != 0 && (pf->HasTrackerTrk()||pf->HasGsfTrk()) ) {
40	+
41	+	if (abs(pf->PFType()) == mithep::PFCandidate::eElectron || abs(pf->PFType()) == mithep::PFCandidate::eMuon) continue;
42	+
43	+	charged_iso += pf->Pt();
44	+	}
45	+
46	+	else if (abs(pf->PFType()) == mithep::PFCandidate::eGamma) {
47	+
48	+	gamma_iso += pf->Pt();
49	+	}
50	+
51	+	else {
52	+	neutral_iso += pf->Pt();
53	+	}
54	+	}
55	+
56	+	mithep::MuonTools::EMuonEffectiveAreaTarget EffectiveAreaVersion  = mithep::MuonTools::kMuEAData2012;
57	+	float rho = puEnergyDensity->At(0)->RhoKt6PFJetsCentralNeutral();
58	+
59	+	double iso_sum = charged_iso + fmax(0.0,(gamma_iso + neutral_iso
60	+	-rho*muT.MuonEffectiveArea(muT.kMuGammaAndNeutralHadronIso04,
61	+	mu->Eta(),EffectiveAreaVersion)));
62	+
63	+	bool pass = kTRUE;
64	+	if(!mu->IsGlobalMuon()) pass = kFALSE;
65	+
66	+	//there is not necessarily a global track or a tracker track if is global muon is false
67	+	if(!pass) return pass;
68	+
69	+	if(!mu->IsPFMuon()) pass = kFALSE;
70	+	if(mu->GlobalTrk()->RChi2() > 10) pass= kFALSE;
71	+	if(mu->NValidHits() == 0 ) pass= kFALSE;
72	+	if(mu->NMatches() <= 1) pass = kFALSE;
73	+	if(fabs(mu->TrackerTrk()->D0Corrected(*vtx)) > 0.2) pass = kFALSE;
74	+	if(fabs(mu->TrackerTrk()->DzCorrected(*vtx)) > 0.5) pass = kFALSE;
75	+	if(mu->BestTrk()->NPixelHits() == 0) pass = kFALSE;
76	+	if(mu->NTrkLayersHit() <= 5) pass = kFALSE;
77	+	if(iso_sum/mu->Pt() > 0.12 ) pass = kFALSE;
78	+
79	+	return pass;
80	+
81	+
82	+	}
83	+
84	+	bool trackToPFMuonSelection( const mithep::Track *track, const mithep::MuonCol * muoncol )
85	+	{
86	+	bool pass = false;
87	+	for(UInt_t i=0; i<muoncol->GetEntries(); ++i) {
88	+	const mithep::Muon *mu = muoncol->At(i);
89	+	if(mu->TrackerTrk() == track && (mu->IsGlobalMuon() || mu->IsTrackerMuon()) && mu->IsPFMuon())
90	+	pass = true;
91	+	}
92	+
93	+	return pass;
94	+
95	+	}
96	+
97		//--------------------------------------------------------------------------------------------------
98		SelectionStatus muonDummyVeto(ControlFlags &ctrl,
99		const mithep::Muon *muon,
100	<	const mithep::Vertex &vtx)
100	>	const mithep::Vertex * vtx)
101		//--------------------------------------------------------------------------------------------------
102		{
103		SelectionStatus status;
#	Line 32 | Line 108 | SelectionStatus muonDummyVeto(ControlFla
108		//--------------------------------------------------------------------------------------------------
109		SelectionStatus muonCutBasedVeto(ControlFlags &ctrl,
110		const mithep::Muon *muon,
111	<	const mithep::Vertex &vtx)
111	>	const mithep::Vertex *vtx)
112		//--------------------------------------------------------------------------------------------------
113		{
114		//
#	Line 64 | Line 140 | SelectionStatus noPreSelection( ControlF
140		//--------------------------------------------------------------------------------------------------
141		SelectionStatus muonPreSelection( ControlFlags &ctrl,
142		const mithep::Muon * mu,
143	<	const mithep::Vertex & vtx,
143	>	const mithep::Vertex * vtx,
144		const mithep::Array<mithep::PFCandidate> * pfCandidates )
145		//--------------------------------------------------------------------------------------------------
146		{
147		bool ret = true;
148	<	if(ctrl.debug) cout << "inside muonpresel ... " << endl;
149	<	//  bool ret = isMuFO(mu,vtx,pfCandidates);
74	<	if(ctrl.debug) cout << "iS fo? ... " << ret << endl;
75	<	ret &= ( fabs(mu->Ip3dPVSignificance()) < 4 );
76	<	if(ctrl.debug) cout << "and pass IP (" << mu->Ip3dPVSignificance() << ") ? ... " << ret << endl;
148	>
149	>	ret &= ( fabs(mu->Ip3dPVSignificance()) < 100 );
150		ret &= ( mu->Pt() >= 5 );
78	–	if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;
151		ret &= ( fabs(mu->Eta()) <= 2.4 );
80	–	if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;
152		ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
153	<	if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
154	<	if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
155	<	ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
85	<	if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;
86	<	ret &= ( fabs(mu->BestTrk()->DzCorrected(vtx)) < 0.1 );
87	<	if( ctrl.debug ) cout << "elepresel : ret after dZcorr : " << ret <<  endl;
88	<
153	>	//   if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
154	>	//     ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
155	>	ret &= (  mu->IsoR03SumPt()/mu->Pt() < 0.7 );
156
157		SelectionStatus status;
158		if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
#	Line 98 | Line 165 | SelectionStatus muonPreSelection( Contro
165		//--------------------------------------------------------------------------------------------------
166		SelectionStatus muonReferencePreSelection( ControlFlags &ctrl,
167		const mithep::Muon * mu,
168	<	const mithep::Vertex & vtx,
168	>	const mithep::Vertex * vtx,
169		const mithep::Array<mithep::PFCandidate> * pfCandidates )
170		//--------------------------------------------------------------------------------------------------
171		{
172		bool ret = true;
173	+
174		if(ctrl.debug) cout << "inside muonpresel ... " << endl;
175		ret &= ( fabs(mu->Ip3dPVSignificance()) < 100 );
176		if(ctrl.debug) cout << "and pass IP (" << mu->Ip3dPVSignificance() << ") ? ... " << ret << endl;
#	Line 111 | Line 179 | SelectionStatus muonReferencePreSelectio
179		ret &= ( fabs(mu->Eta()) <= 2.4 );
180		if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;
181		ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
182	<	if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
182	>	if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
183	>	ret &= (mu->HasTrackerTrk() &&
184	>	fabs(mu->TrackerTrk()->D0Corrected(*vtx)) < 0.5 &&
185	>	fabs(mu->TrackerTrk()->DzCorrected(*vtx)) < 1.0);
186	>	if(ctrl.debug) cout << "d0 and dz? ... " << ret << endl;
187	>
188	>	//  if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
189	>	//    ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
190	>	//  if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;
191	>	//  ret &= ( mu->BestTrk()->DzCorrected(vtx) < 0.1);
192	>	//  if( ctrl.debug ) cout << "elepresel : ret after dZcorr : " << ret <<  endl;
193	>
194	>
195	>	SelectionStatus status;
196	>	if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
197	>	if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
198	>	return status;
199	>	}
200	>
201	>	//--------------------------------------------------------------------------------------------------
202	>	SelectionStatus muonPreSelectionNoD0DzIP( ControlFlags &ctrl,
203	>	const mithep::Muon * mu,
204	>	const mithep::Vertex * vtx,
205	>	const mithep::Array<mithep::PFCandidate> * pfCandidates )
206	>	//--------------------------------------------------------------------------------------------------
207	>	{
208	>	bool ret = true;
209	>
210	>	if(ctrl.debug) cout << "inside muonpresel ... " << endl;
211	>	ret &= ( mu->Pt() >= 5 );
212	>	if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;
213	>	ret &= ( fabs(mu->Eta()) <= 2.4 );
214	>	if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;
215	>	ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
216	>	if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
217	>
218		//  if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
219		//    ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
220		//  if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;
#	Line 131 | Line 234 | SelectionStatus muonReferencePreSelectio
234		//--------------------------------------------------------------------------------------------------
235		SelectionStatus muonIDMVASelection(ControlFlags &ctrl,
236		const mithep::Muon *mu,
237	<	const mithep::Vertex & vtx   )
237	>	const mithep::Vertex * vtx   )
238		//--------------------------------------------------------------------------------------------------
239		{
240		double global_rchi2 = (mu->IsGlobalMuon()) ? mu->GlobalTrk()->RChi2() : 0.;
#	Line 205 | Line 308 | void initMuonIDMVA()
308		//--------------------------------------------------------------------------------------------------
309		SelectionStatus muonIDPFSelection(ControlFlags &ctrl,
310		const mithep::Muon *mu,
311	<	const mithep::Vertex & vtx,
311	>	const mithep::Vertex * vtx,
312		const mithep::Array<mithep::PFCandidate> * pfCandidates )
313		//--------------------------------------------------------------------------------------------------
314		{
#	Line 214 | Line 317 | SelectionStatus muonIDPFSelection(Contro
317
318		// check that it matches to a PF muon
319		for( int i=0; i<pfCandidates->GetEntries(); i++ ) {
320	+	// tmp
321	+	if( !(PFnoPUflag[i]) ) continue; // my PF no PU hack
322	+
323		const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[i]);
324	<	//    if( pf->HasTrackerTrk() ) {
219	<	//      if( (pf->TrackerTrk() == mu->TrackerTrk()) && abs(pf->PFType()) == mithep::PFCandidate::eMuon ) {
220	<	if(  abs(pf->PFType()) == mithep::PFCandidate::eMuon ) {
324	>	if( (pf->TrackerTrk() == mu->TrackerTrk()) && abs(pf->PFType()) == mithep::PFCandidate::eMuon ) {
325		pass = true;
326		break;
223	–	//    }
327		}
328		}
329	+	//
330	+	//      if(mu->IsPFMuon()) pass = true;
331	+	//      else pass = false;
332
333		//
334		// NB : ipsig here for reference synch ...
335		//
336	+	#ifdef SYNC
337		if( pass && fabs(mu->Ip3dPVSignificance()) < 4 ) {
338		status.orStatus(SelectionStatus::LOOSEID);
339		status.orStatus(SelectionStatus::TIGHTID);
340		}
341	+	#else
342	+	if( pass ) {
343	+	status.orStatus(SelectionStatus::LOOSEID);
344	+	status.orStatus(SelectionStatus::TIGHTID);
345	+	}
346	+	#endif
347	+
348	+	return status;
349	+	}
350	+	//----------------------------------------------------------------------------------------
351	+	SelectionStatus PassWwMuonSel(const mithep::Muon *mu, const mithep::Vertex *vtx,
352	+	const mithep::Array<mithep::PFCandidate> *pfArr
353	+	)
354	+	{
355	+	bool passID=true;
356	+	bool passIso=true;
357	+
358	+	if (mu->Pt() < 5)
359	+	passID = false;
360	+	if (fabs(mu->Eta()) > 2.4)
361	+	passID = false;
362	+	if (mu->BestTrk()->PtErr()/mu->Pt() > 0.1)
363	+	passID = false;
364	+	if (fabs(mu->BestTrk()->DzCorrected(*vtx)) > 0.1)
365	+	passID = false;
366	+	Bool_t isGlobal  = (mu->IsGlobalMuon()) && (mu->BestTrk()->RChi2() < 10) &&
367	+	(mu->NMatches() > 1) && (mu->NValidHits() > 0);
368	+	Bool_t isTracker = (mu->IsTrackerMuon()) &&
369	+	(mu->Quality().Quality(mithep::MuonQuality::TMLastStationTight));
370	+	if (!isGlobal && !isTracker)
371	+	passID = false;
372	+	int ntrkhits = (mu->HasTrackerTrk()) ? mu->TrackerTrk()->NHits() : 0;
373	+	if (ntrkhits < 10)
374	+	passID = false;
375	+	if (mu->BestTrk()->NPixelHits() < 1)
376	+	passID = false;
377	+	if (fabs(mu->BestTrk()->D0Corrected(*vtx)) > 0.02)
378	+	passID = false;
379	+
380	+	// note: this isn't really ww muon isolation
381	+	float pfiso = computePFMuonIso(mu,vtx,pfArr,0.3);
382
383	+	if (pfiso > 0.2)
384	+	passIso = false;
385	+
386	+	SelectionStatus status;
387	+	if(passID)  status |= SelectionStatus::TIGHTIDANDPRE;
388	+	if(passIso) status |= SelectionStatus::TIGHTISOANDPRE;
389	+
390		return status;
391	+
392		}

Diff Legend

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