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.7 by khahn, Mon Apr 30 21:42:16 2012 UTC vs.
Revision 1.12 by khahn, Sat May 5 13:09:05 2012 UTC

#	Line 19 | Line 19 | extern Float_t computePFMuonIso(const mi
19		const Double_t dRMax);
20
21		//--------------------------------------------------------------------------------------------------
22	+	SelectionStatus muonDummyVeto(ControlFlags &ctrl,
23	+	const mithep::Muon *muon,
24	+	const mithep::Vertex &vtx)
25	+	//--------------------------------------------------------------------------------------------------
26	+	{
27	+	SelectionStatus status;
28	+	status.setStatus(SelectionStatus::PRESELECTION);
29	+	return status;
30	+	}
31	+
32	+	//--------------------------------------------------------------------------------------------------
33	+	SelectionStatus muonCutBasedVeto(ControlFlags &ctrl,
34	+	const mithep::Muon *muon,
35	+	const mithep::Vertex &vtx)
36	+	//--------------------------------------------------------------------------------------------------
37	+	{
38	+	//
39	+	// Loose cut-based ID for isolation veto
40	+	//
41	+	bool ret = true;
42	+
43	+	if(!(muon->IsGlobalMuon() || muon->IsTrackerMuon())) ret = false;
44	+	if( muon->NValidHits() < 11 )                        ret = false;
45	+	if( fabs(muon->Ip3dPVSignificance()) >= 4 )          ret = false;
46	+
47	+	SelectionStatus status;
48	+	if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
49	+	return status;
50	+	}
51	+
52	+
53	+	//--------------------------------------------------------------------------------------------------
54		SelectionStatus noPreSelection( ControlFlags &ctrl,  const mithep::Muon * mu )
55
56		//--------------------------------------------------------------------------------------------------
#	Line 46 | Line 78 | SelectionStatus muonPreSelection( Contro
78		if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;
79		ret &= ( fabs(mu->Eta()) <= 2.4 );
80		if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;
81	<	ret &=  (mu->IsTrackerMuon() && mu->HasTrackerTrk() &&
82	<	(mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated));
83	<	if(ctrl.debug) cout << "and isTrackerMuon ? ... " << ret << endl;
84	<	//  ret &=  (mu->IsoR03SumPt()/mu->Pt() < 0.7 );
85	<	ret &=  (mu->IsoR03SumPt()/mu->Pt() < 0.7 );
86	<	if(ctrl.debug) cout << "and loose trk iso ? ... " << ret << endl;
81	>	ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
82	>	if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
83	>	if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
84	>	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	>
89
90		SelectionStatus status;
91		if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
#	Line 60 | Line 94 | SelectionStatus muonPreSelection( Contro
94		}
95
96
63	–	//--------------------------------------------------------------------------------------------------
64	–	bool isMuFO( const mithep::Muon * mu,
65	–	const mithep::Vertex & vtx,
66	–	const mithep::Array<mithep::PFCandidate> * pfCandidates )
67	–	//--------------------------------------------------------------------------------------------------
68	–	{
69	–	bool isgood=true;
70	–	float reliso = computePFMuonIso(mu,vtx,pfCandidates,0.3)/mu->Pt();
71	–
72	–	if( mu->Pt() < 5  )         isgood=false;
73	–	if ( fabs(mu->BestTrk()->DzCorrected(vtx)) > 0.2 ) isgood=false;   // needed to remove cosmics in HF sample
74	–
75	–	//
76	–
77	–	// HF seems to not want tkquality, SS does
78	–	if( mu->Pt()>20 ) {
79	–	if ( !((mu->IsGlobalMuon()) && mu->NSegments()>0  ) ) isgood=false; //&& mu->nValidHits>0
80	–	} else {
81	–	if ( !((mu->IsGlobalMuon()) && mu->NSegments()>0 ) //&& //&& mu->nValidHits>0
82	–	&& !( (mu->IsTrackerMuon()) &&  (mu->Quality().QualityMask().Mask() & muTools.kTMOneStationLoose)
83	–	)
84	–	)   isgood=false;
85	–	}
86	–
87	–
88	–	//
89	–	// for HF MC to agree w/ data, cannot put any Tk quality bits
90	–	// but to get to 5% closure in the SS test, we need at least Tk oneStationLoose
91	–
92	–	/*
93	–	&& ( (mu->qualityBits & kTMOneStationLoose) ||
94	–	(mu->qualityBits & kTMLastStationOptimizedBarrelLowPtLoose) ||
95	–	(mu->qualityBits & kTMLastStationOptimizedLowPtLoose)
96	–	)
97	–	*/
98	–
99	–
100	–	// comment for more stats for mu fake shape
101	–	if( mu->Pt() < 20 ) {
102	–	if( reliso > 3 ) isgood=false;
103	–	} else {
104	–	if( reliso > 5 ) isgood=false;
105	–	}
106	–
107	–	return isgood;
108	–	};
109	–
110	–	//--------------------------------------------------------------------------------------------------
111	–	SelectionStatus passSoftMuonSelection( ControlFlags &ctrl,
112	–	const mithep::Muon * mu ,
113	–	const mithep::Vertex & vtx)
114	–	//--------------------------------------------------------------------------------------------------
115	–	{
116	–
117	–	int level=0;
118	–	unsigned failmask=0x0;
119	–
120	–	// Use tracker track when available
121	–	const mithep::Track *muTrk=0;
122	–	if(mu->HasStandaloneTrk()) { muTrk = mu->StandaloneTrk(); }
123	–	else if(mu->HasTrackerTrk())         { muTrk = mu->TrackerTrk(); }
124	–	else if(mu->HasGlobalTrk())     { muTrk = mu->GlobalTrk(); }
125	–	assert(muTrk);
126	–
127	–	if(mu->TrackerTrk()->NHits() < 11 )
128	–	failmask |= (1<<level);
129	–	level++;
130	–
131	–	if(fabs(muTrk->D0Corrected(vtx)) > 0.2)
132	–	failmask |= (1<<level);
133	–	level++;
134	–
135	–	if(fabs(muTrk->DzCorrected(vtx)) > 0.1)
136	–	failmask |= (1<<level);
137	–	level++;
138	–
139	–	if(!(mu->IsTrackerMuon()))
140	–	failmask |= (1<<level);
141	–	level++;
142	–
143	–	//  if(!(mu->Quality().QualityMask().Mask() & muTools.kTMLastStationAngTight))
144	–	if(!(mu->Quality().QualityMask().Mask() & muTools.kTMLastStationTight))
145	–	failmask |= (1<<level);
146	–	level++;
147	–
148	–	Double_t iso = (mu->IsoR03SumPt() + mu->IsoR03EmEt() + mu->IsoR03HadEt())/mu->Pt();
149	–	if(mu->Pt()>20 && iso<0.1)
150	–	failmask |= (1<<level);
151	–
152	–	bool passtight  = !(failmask);
153	–	bool passloose  = !(failmask);
154	–	return SelectionStatus( passtight | passloose );
155	–
156	–	}
157	–
158	–	//--------------------------------------------------------------------------------------------------
159	–	//unsigned passMuonSelectionZZ( const mithep::TMuon * mu ) {
160	–	SelectionStatus passMuonSelectionZZ( ControlFlags &ctrl, const mithep::Muon * mu )
161	–	//--------------------------------------------------------------------------------------------------
162	–	{
163	–	int level=0;
164	–	unsigned failmask=0x0;
165	–
166	–	if(mu->Pt() < 5) {
167	–	failmask |= (1<<level);
168	–	}
169	–
170	–	level++;
171	–	if(fabs(mu->Eta()) > 2.4) {
172	–	failmask |= (1<<level);
173	–	}
174	–
175	–	level++;
176	–	if(!(mu->IsGlobalMuon())) {
177	–	failmask |= (1<<level);
178	–	}
179	–
180	–	level++;
181	–	if(mu->TrackerTrk()->NHits()  < 10) {
182	–	failmask |= (1<<level);
183	–	}
184	–
185	–
186	–	level++;
187	–	if( (mu->IsoR03SumPt()/mu->Pt()) > 0.7 ) {
188	–	failmask |= (1<<level);
189	–	}
190	–
191	–	//  if(muon->nPixHits   < 1)     return false;
192	–	//  if(muon->muNchi2    > 10)    return false;
193	–	//  if(muon->nMatch     < 2)     return false;
194	–	//  if(muon->nValidHits         < 1)     return false;
195	–	//  if(muon->pterr/muon->pt > 0.1)   return false;
196	–	//  if(fabs(muon->dz)       > 0.1)   return false;
197	–
198	–	SelectionStatus status;
199	–	if( !failmask ) status.orStatus(SelectionStatus::LOOSEID);
200	–	if( !failmask ) status.orStatus(SelectionStatus::TIGHTID);
201	–	return status;
202	–
203	–
204	–
205	–	};
206	–
207	–
208	–
209	–
210	–	//
211	–	// Kevin's WW selection
212	–	//
213	–	//--------------------------------------------------------------------------------------------------
214	–	SelectionStatus passMuonSelection( ControlFlags &ctrl,
215	–	const mithep::Muon * mu,
216	–	const mithep::Vertex & vtx   )
217	–	//--------------------------------------------------------------------------------------------------
218	–	{
219	–	SelectionStatus status;
220	–	int level=0;
221	–	unsigned failmask=0x0;
222	–
223	–	// Use tracker track when available
224	–	const mithep::Track *muTrk=0;
225	–	if(mu->HasTrackerTrk())         { muTrk = mu->TrackerTrk(); }
226	–	else if(mu->HasGlobalTrk())     { muTrk = mu->GlobalTrk(); }
227	–	else if(mu->HasStandaloneTrk()) { muTrk = mu->StandaloneTrk(); }
228	–	assert(muTrk);
229	–
230	–
231	–
232	–	// 0x1
233	–	if(muTrk->Pt() < 5) {
234	–	failmask |= (1<<level);
235	–	}
236	–
237	–	// 0x2
238	–	level++;
239	–	if(fabs(muTrk->Eta()) > 2.4) {
240	–	failmask |= (1<<level);
241	–	}
242	–
243	–	// 0x4
244	–	level++;
245	–	if(muTrk->PtErr()/muTrk->Pt() > 0.1)   {
246	–	failmask |= (1<<level);
247	–	}
248	–
249	–	// 0x8
250	–	level++;
251	–	if( fabs(muTrk->DzCorrected(vtx)) > 0.1 )   {
252	–	failmask |= (1<<level);
253	–	}
254	–
255	–	double muNchi2;
256	–	if(mu->HasStandaloneTrk())      { muNchi2 = mu->StandaloneTrk()->RChi2(); }
257	–	else if(mu->HasGlobalTrk())     { muNchi2 = mu->GlobalTrk()->RChi2();     }
258	–	else if(mu->HasTrackerTrk())    { muNchi2 = mu->TrackerTrk()->RChi2();    }
259	–
260	–	unsigned qualityBits = mu->Quality().QualityMask().Mask();
261	–
262	–	Bool_t isGlobal  = (mu->IsGlobalMuon()) && (muNchi2 < 10) && (mu->NMatches() > 1) && (mu->NValidHits() > 0);
263	–	Bool_t isTracker = (mu->IsTrackerMuon() ) && (qualityBits & muTools.kTMLastStationTight);
264	–
265	–	// 0x10
266	–	level++;
267	–	if((!isGlobal && !isTracker) || !(mu->HasTrackerTrk())) {
268	–	failmask |= (1<<level);
269	–	status.setStatus(SelectionStatus::FAIL);
270	–	return status;
271	–	}
272	–
273	–	// 0x20
274	–	level++;
275	–	assert(mu->HasTrackerTrk());
276	–	assert(mu->TrackerTrk());
277	–	if( mu->TrackerTrk()->NHits() < 10 ) {
278	–	failmask |= (1<<level);
279	–	}
280	–
281	–	level++;
282	–	if(muTrk->NPixelHits()          < 1) {
283	–	failmask |= (1<<level);
284	–	}
285	–
286	–
287	–	level++;
288	–	if(fabs(muTrk->D0Corrected(vtx))>0.02)   {
289	–	failmask |= (1<<level);
290	–	}
291	–
292	–	/*
293	–	if(mu->pt>20) {
294	–	if(fabs(mu->d0)>0.02)   {
295	–	failmask |= (1<<level);
296	–	}
297	–	} else {
298	–	if(fabs(mu->d0)>0.01)  {
299	–	failmask |= (1<<level);
300	–	}
301	–	}
302	–	*/
303	–
304	–	if( !failmask ) status.orStatus(SelectionStatus::LOOSEID);
305	–	if( !failmask ) status.orStatus(SelectionStatus::TIGHTID);
306	–	return status;
307	–
308	–
309	–	};
97
98
99		//--------------------------------------------------------------------------------------------------
#	Line 315 | Line 102 | SelectionStatus muonIDMVASelection(Contr
102		const mithep::Vertex & vtx   )
103		//--------------------------------------------------------------------------------------------------
104		{
318	–	assert(mu->IsTrackerMuon()); // must be
319	–	assert(mu->HasTrackerTrk());
320	–	assert(mu->TrackerTrk());
321	–
322	–	/*
323	–	cerr << "TrackerTrk: " << mu->TrackerTrk() << endl;
324	–	flush(cerr);
325	–	cerr << "Rchi2: " << mu->TrackerTrk()->RChi2() << endl;
326	–	flush(cerr);
327	–
328	–	cerr << "calling MVa ... "<< endl;
329	–	flush(cerr);
330	–	*/
331	–
105		double global_rchi2 = (mu->IsGlobalMuon()) ? mu->GlobalTrk()->RChi2() : 0.;
106
107		double mvaval = muIDMVA->MVAValue_ID(mu->Pt(),
#	Line 338 | Line 111 | SelectionStatus muonIDMVASelection(Contr
111		mu->TrackerTrk()->RChi2(),
112		global_rchi2,
113		(Double_t)(mu->NValidHits()),
114	<	(Double_t)(mu->TrackerTrk()->NHits()),
114	>	(Double_t)(mu->TrackerTrk()->NHits()),
115		(Double_t)(mu->TrackerTrk()->NPixelHits()),
116		(Double_t)(mu->NMatches()),
117		mu->TrkKink(),
#	Line 362 | Line 135 | SelectionStatus muonIDMVASelection(Contr
135		&& fabs(mu->Eta()) > 1.5 && mu->Pt() <= 10 && mvaval >= MUON_IDMVA_CUT_BIN2)  pass = true;
136		else if( mu->IsGlobalMuon() && mu->IsTrackerMuon()
137		&& fabs(mu->Eta()) > 1.5 && mu->Pt() > 10 && mvaval >= MUON_IDMVA_CUT_BIN3)  pass = true;
138	<	else if( !(mu->IsGlobalMuon()) && mu->IsTrackerMuon()
139	<	&& (mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated) && mvaval >= MUON_IDMVA_CUT_BIN4)
140	<	pass = true;
138	>	else if( !(mu->IsGlobalMuon()) && mu->IsTrackerMuon() && mvaval >= MUON_IDMVA_CUT_BIN4 )  pass = true;
139	>	else if( mu->IsGlobalMuon() && !(mu->IsTrackerMuon()) && mvaval >= MUON_IDMVA_CUT_BIN5 )  pass = true;
140	>
141
142
143		if( pass ) {
#	Line 391 | Line 164 | void initMuonIDMVA()
164		mithep::MuonIDMVA::kIDV0,
165		kTRUE, weightFiles);
166		}
167	+
168	+	//--------------------------------------------------------------------------------------------------
169	+	SelectionStatus muonIDPFSelection(ControlFlags &ctrl,
170	+	const mithep::Muon *mu,
171	+	const mithep::Vertex & vtx,
172	+	const mithep::Array<mithep::PFCandidate> * pfCandidates )
173	+	//--------------------------------------------------------------------------------------------------
174	+	{
175	+	bool pass = false;
176	+	SelectionStatus status; // init'ed to FAIL
177	+
178	+	// check that it matches to a PF muon
179	+	for( int i=0; i<pfCandidates->GetEntries(); i++ ) {
180	+	const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[i]);
181	+	if( pf->HasTrackerTrk() ) {
182	+	if( (pf->TrackerTrk() == mu->TrackerTrk()) && abs(pf->PFType()) == mithep::PFCandidate::eMuon ) {
183	+	pass = true;
184	+	break;
185	+	}
186	+	}
187	+	}
188	+
189	+	if( pass ) {
190	+	status.orStatus(SelectionStatus::LOOSEID);
191	+	status.orStatus(SelectionStatus::TIGHTID);
192	+	}
193	+
194	+	return status;
195	+	}

Diff Legend

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