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.11 by khahn, Thu May 3 22:06:47 2012 UTC vs.
Revision 1.22 by anlevin, Mon May 28 12:37:10 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		//--------------------------------------------------------------------------------------------------
23		SelectionStatus muonDummyVeto(ControlFlags &ctrl,
24		const mithep::Muon *muon,
25	<	const mithep::Vertex &vtx)
25	>	const mithep::Vertex * vtx)
26		//--------------------------------------------------------------------------------------------------
27		{
28		SelectionStatus status;
#	Line 32 | Line 33 | SelectionStatus muonDummyVeto(ControlFla
33		//--------------------------------------------------------------------------------------------------
34		SelectionStatus muonCutBasedVeto(ControlFlags &ctrl,
35		const mithep::Muon *muon,
36	<	const mithep::Vertex &vtx)
36	>	const mithep::Vertex *vtx)
37		//--------------------------------------------------------------------------------------------------
38		{
39		//
#	Line 64 | Line 65 | SelectionStatus noPreSelection( ControlF
65		//--------------------------------------------------------------------------------------------------
66		SelectionStatus muonPreSelection( ControlFlags &ctrl,
67		const mithep::Muon * mu,
68	<	const mithep::Vertex & vtx,
68	>	const mithep::Vertex * vtx,
69		const mithep::Array<mithep::PFCandidate> * pfCandidates )
70		//--------------------------------------------------------------------------------------------------
71		{
72		bool ret = true;
73	<	if(ctrl.debug) cout << "inside muonpresel ... " << endl;
74	<	//  bool ret = isMuFO(mu,vtx,pfCandidates);
75	<	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;
77	<	ret &= ( mu->Pt() > 5 );
78	<	if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;
73	>
74	>	ret &= ( fabs(mu->Ip3dPVSignificance()) < 100 );
75	>	ret &= ( mu->Pt() >= 5 );
76		ret &= ( fabs(mu->Eta()) <= 2.4 );
80	–	if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;
77		ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
78	<	if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
79	<	if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
80	<	ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
85	<	if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;
86	<	ret &= ( mu->BestTrk()->DzCorrected(vtx) < 0.1);
87	<	if( ctrl.debug ) cout << "elepresel : ret after dZcorr : " << ret <<  endl;
88	<
78	>	//   if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
79	>	//     ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
80	>	ret &= (  mu->IsoR03SumPt()/mu->Pt() < 0.7 );
81
82		SelectionStatus status;
83		if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
#	Line 94 | Line 86 | SelectionStatus muonPreSelection( Contro
86		}
87
88
97	–	//--------------------------------------------------------------------------------------------------
98	–	bool isMuFO( const mithep::Muon * mu,
99	–	const mithep::Vertex & vtx,
100	–	const mithep::Array<mithep::PFCandidate> * pfCandidates )
101	–	//--------------------------------------------------------------------------------------------------
102	–	{
103	–	bool isgood=true;
104	–	float reliso = computePFMuonIso(mu,vtx,pfCandidates,0.3)/mu->Pt();
105	–
106	–	if( mu->Pt() < 5  )         isgood=false;
107	–	if ( fabs(mu->BestTrk()->DzCorrected(vtx)) > 0.2 ) isgood=false;   // needed to remove cosmics in HF sample
108	–
109	–	//
110	–
111	–	// HF seems to not want tkquality, SS does
112	–	if( mu->Pt()>20 ) {
113	–	if ( !((mu->IsGlobalMuon()) && mu->NSegments()>0  ) ) isgood=false; //&& mu->nValidHits>0
114	–	} else {
115	–	if ( !((mu->IsGlobalMuon()) && mu->NSegments()>0 ) //&& //&& mu->nValidHits>0
116	–	&& !( (mu->IsTrackerMuon()) &&  (mu->Quality().QualityMask().Mask() & muTools.kTMOneStationLoose)
117	–	)
118	–	)   isgood=false;
119	–	}
120	–
121	–
122	–	//
123	–	// for HF MC to agree w/ data, cannot put any Tk quality bits
124	–	// but to get to 5% closure in the SS test, we need at least Tk oneStationLoose
125	–
126	–	/*
127	–	&& ( (mu->qualityBits & kTMOneStationLoose) ||
128	–	(mu->qualityBits & kTMLastStationOptimizedBarrelLowPtLoose) ||
129	–	(mu->qualityBits & kTMLastStationOptimizedLowPtLoose)
130	–	)
131	–	*/
89
133	–
134	–	// comment for more stats for mu fake shape
135	–	if( mu->Pt() < 20 ) {
136	–	if( reliso > 3 ) isgood=false;
137	–	} else {
138	–	if( reliso > 5 ) isgood=false;
139	–	}
140	–
141	–	return isgood;
142	–	};
143	–
144	–	//--------------------------------------------------------------------------------------------------
145	–	SelectionStatus passSoftMuonSelection( ControlFlags &ctrl,
146	–	const mithep::Muon * mu ,
147	–	const mithep::Vertex & vtx)
90		//--------------------------------------------------------------------------------------------------
91	<	{
92	<
93	<	int level=0;
94	<	unsigned failmask=0x0;
153	<
154	<	// Use tracker track when available
155	<	const mithep::Track *muTrk=0;
156	<	if(mu->HasStandaloneTrk()) { muTrk = mu->StandaloneTrk(); }
157	<	else if(mu->HasTrackerTrk())         { muTrk = mu->TrackerTrk(); }
158	<	else if(mu->HasGlobalTrk())     { muTrk = mu->GlobalTrk(); }
159	<	assert(muTrk);
160	<
161	<	if(mu->TrackerTrk()->NHits() < 11 )
162	<	failmask |= (1<<level);
163	<	level++;
164	<
165	<	if(fabs(muTrk->D0Corrected(vtx)) > 0.2)
166	<	failmask |= (1<<level);
167	<	level++;
168	<
169	<	if(fabs(muTrk->DzCorrected(vtx)) > 0.1)
170	<	failmask |= (1<<level);
171	<	level++;
172	<
173	<	if(!(mu->IsTrackerMuon()))
174	<	failmask |= (1<<level);
175	<	level++;
176	<
177	<	//  if(!(mu->Quality().QualityMask().Mask() & muTools.kTMLastStationAngTight))
178	<	if(!(mu->Quality().QualityMask().Mask() & muTools.kTMLastStationTight))
179	<	failmask |= (1<<level);
180	<	level++;
181	<
182	<	Double_t iso = (mu->IsoR03SumPt() + mu->IsoR03EmEt() + mu->IsoR03HadEt())/mu->Pt();
183	<	if(mu->Pt()>20 && iso<0.1)
184	<	failmask |= (1<<level);
185	<
186	<	bool passtight  = !(failmask);
187	<	bool passloose  = !(failmask);
188	<	return SelectionStatus( passtight | passloose );
189	<
190	<	}
191	<
192	<	//--------------------------------------------------------------------------------------------------
193	<	//unsigned passMuonSelectionZZ( const mithep::TMuon * mu ) {
194	<	SelectionStatus passMuonSelectionZZ( ControlFlags &ctrl, const mithep::Muon * mu )
91	>	SelectionStatus muonReferencePreSelection( ControlFlags &ctrl,
92	>	const mithep::Muon * mu,
93	>	const mithep::Vertex * vtx,
94	>	const mithep::Array<mithep::PFCandidate> * pfCandidates )
95		//--------------------------------------------------------------------------------------------------
96		{
97	<	int level=0;
198	<	unsigned failmask=0x0;
199	<
200	<	if(mu->Pt() < 5) {
201	<	failmask |= (1<<level);
202	<	}
203	<
204	<	level++;
205	<	if(fabs(mu->Eta()) > 2.4) {
206	<	failmask |= (1<<level);
207	<	}
208	<
209	<	level++;
210	<	if(!(mu->IsGlobalMuon())) {
211	<	failmask |= (1<<level);
212	<	}
213	<
214	<	level++;
215	<	if(mu->TrackerTrk()->NHits()  < 10) {
216	<	failmask |= (1<<level);
217	<	}
218	<
219	<
220	<	level++;
221	<	if( (mu->IsoR03SumPt()/mu->Pt()) > 0.7 ) {
222	<	failmask |= (1<<level);
223	<	}
224	<
225	<	//  if(muon->nPixHits   < 1)     return false;
226	<	//  if(muon->muNchi2    > 10)    return false;
227	<	//  if(muon->nMatch     < 2)     return false;
228	<	//  if(muon->nValidHits         < 1)     return false;
229	<	//  if(muon->pterr/muon->pt > 0.1)   return false;
230	<	//  if(fabs(muon->dz)       > 0.1)   return false;
231	<
232	<	SelectionStatus status;
233	<	if( !failmask ) status.orStatus(SelectionStatus::LOOSEID);
234	<	if( !failmask ) status.orStatus(SelectionStatus::TIGHTID);
235	<	return status;
236	<
237	<
238	<
239	<	};
240	<
97	>	bool ret = true;
98
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;
102	+	ret &= ( mu->Pt() >= 5 );
103	+	if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;
104	+	ret &= ( fabs(mu->Eta()) <= 2.4 );
105	+	if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;
106	+	ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
107	+	if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
108	+	ret &= (mu->HasTrackerTrk() && mu->TrackerTrk()->D0Corrected(*vtx) < 0.5 && mu->TrackerTrk()->DzCorrected(*vtx) < 1.0);
109	+	if(ctrl.debug) cout << "d0 and dz? ... " << ret << endl;
110	+
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
244	–	//
245	–	// Kevin's WW selection
246	–	//
247	–	//--------------------------------------------------------------------------------------------------
248	–	SelectionStatus passMuonSelection( ControlFlags &ctrl,
249	–	const mithep::Muon * mu,
250	–	const mithep::Vertex & vtx   )
251	–	//--------------------------------------------------------------------------------------------------
252	–	{
118		SelectionStatus status;
119	<	int level=0;
120	<	unsigned failmask=0x0;
256	<
257	<	// Use tracker track when available
258	<	const mithep::Track *muTrk=0;
259	<	if(mu->HasTrackerTrk())         { muTrk = mu->TrackerTrk(); }
260	<	else if(mu->HasGlobalTrk())     { muTrk = mu->GlobalTrk(); }
261	<	else if(mu->HasStandaloneTrk()) { muTrk = mu->StandaloneTrk(); }
262	<	assert(muTrk);
263	<
264	<
265	<
266	<	// 0x1
267	<	if(muTrk->Pt() < 5) {
268	<	failmask |= (1<<level);
269	<	}
270	<
271	<	// 0x2
272	<	level++;
273	<	if(fabs(muTrk->Eta()) > 2.4) {
274	<	failmask |= (1<<level);
275	<	}
276	<
277	<	// 0x4
278	<	level++;
279	<	if(muTrk->PtErr()/muTrk->Pt() > 0.1)   {
280	<	failmask |= (1<<level);
281	<	}
282	<
283	<	// 0x8
284	<	level++;
285	<	if( fabs(muTrk->DzCorrected(vtx)) > 0.1 )   {
286	<	failmask |= (1<<level);
287	<	}
288	<
289	<	double muNchi2;
290	<	if(mu->HasStandaloneTrk())      { muNchi2 = mu->StandaloneTrk()->RChi2(); }
291	<	else if(mu->HasGlobalTrk())     { muNchi2 = mu->GlobalTrk()->RChi2();     }
292	<	else if(mu->HasTrackerTrk())    { muNchi2 = mu->TrackerTrk()->RChi2();    }
293	<
294	<	unsigned qualityBits = mu->Quality().QualityMask().Mask();
295	<
296	<	Bool_t isGlobal  = (mu->IsGlobalMuon()) && (muNchi2 < 10) && (mu->NMatches() > 1) && (mu->NValidHits() > 0);
297	<	Bool_t isTracker = (mu->IsTrackerMuon() ) && (qualityBits & muTools.kTMLastStationTight);
298	<
299	<	// 0x10
300	<	level++;
301	<	if((!isGlobal && !isTracker) || !(mu->HasTrackerTrk())) {
302	<	failmask |= (1<<level);
303	<	status.setStatus(SelectionStatus::FAIL);
304	<	return status;
305	<	}
306	<
307	<	// 0x20
308	<	level++;
309	<	assert(mu->HasTrackerTrk());
310	<	assert(mu->TrackerTrk());
311	<	if( mu->TrackerTrk()->NHits() < 10 ) {
312	<	failmask |= (1<<level);
313	<	}
314	<
315	<	level++;
316	<	if(muTrk->NPixelHits()          < 1) {
317	<	failmask |= (1<<level);
318	<	}
319	<
320	<
321	<	level++;
322	<	if(fabs(muTrk->D0Corrected(vtx))>0.02)   {
323	<	failmask |= (1<<level);
324	<	}
325	<
326	<	/*
327	<	if(mu->pt>20) {
328	<	if(fabs(mu->d0)>0.02)   {
329	<	failmask |= (1<<level);
330	<	}
331	<	} else {
332	<	if(fabs(mu->d0)>0.01)  {
333	<	failmask |= (1<<level);
334	<	}
335	<	}
336	<	*/
337	<
338	<	if( !failmask ) status.orStatus(SelectionStatus::LOOSEID);
339	<	if( !failmask ) status.orStatus(SelectionStatus::TIGHTID);
119	>	if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
120	>	if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
121		return status;
122	+	}
123
124
343	–	};
125
126
127		//--------------------------------------------------------------------------------------------------
128		SelectionStatus muonIDMVASelection(ControlFlags &ctrl,
129		const mithep::Muon *mu,
130	<	const mithep::Vertex & vtx   )
130	>	const mithep::Vertex * vtx   )
131		//--------------------------------------------------------------------------------------------------
132		{
133		double global_rchi2 = (mu->IsGlobalMuon()) ? mu->GlobalTrk()->RChi2() : 0.;
134
135	+	//
136	+	// WARNING !!! KH hacked out to sync
137	+	//
138	+	/*
139		double mvaval = muIDMVA->MVAValue_ID(mu->Pt(),
140		mu->Eta(),
141		mu->IsGlobalMuon(),
#	Line 369 | Line 154 | SelectionStatus muonIDMVASelection(Contr
154		mu->HadS9Energy(),
155		mu->EmS9Energy(),
156		(Bool_t)(ctrl.debug) );
157	<
157	>	*/
158	>	double mvaval = 0.0;
159
160		SelectionStatus status;
161		bool pass = false;
#	Line 415 | Line 201 | void initMuonIDMVA()
201		//--------------------------------------------------------------------------------------------------
202		SelectionStatus muonIDPFSelection(ControlFlags &ctrl,
203		const mithep::Muon *mu,
204	<	const mithep::Vertex & vtx,
204	>	const mithep::Vertex * vtx,
205		const mithep::Array<mithep::PFCandidate> * pfCandidates )
206		//--------------------------------------------------------------------------------------------------
207		{
#	Line 424 | Line 210 | SelectionStatus muonIDPFSelection(Contro
210
211		// check that it matches to a PF muon
212		for( int i=0; i<pfCandidates->GetEntries(); i++ ) {
213	+	// tmp
214	+	if( !(PFnoPUflag[i]) ) continue; // my PF no PU hack
215	+
216		const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[i]);
217	<	if( pf->HasTrackerTrk() ) {
218	<	if( (pf->TrackerTrk() == mu->TrackerTrk()) && abs(pf->PFType()) == mithep::PFCandidate::eMuon ) {
219	<	pass = true;
431	<	break;
432	<	}
217	>	if( (pf->TrackerTrk() == mu->TrackerTrk()) && abs(pf->PFType()) == mithep::PFCandidate::eMuon ) {
218	>	pass = true;
219	>	break;
220		}
221		}
222	+	//
223	+	//      if(mu->IsPFMuon()) pass = true;
224	+	//      else pass = false;
225
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
235		if( pass ) {
236		status.orStatus(SelectionStatus::LOOSEID);
237		status.orStatus(SelectionStatus::TIGHTID);
238		}
239	+	#endif
240
241		return status;
242		}

Diff Legend

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