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

Diff Legend

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