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.25 by khahn, Sun Jun 3 15:51:33 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 trackToPFMuonSelection( const mithep::Track *track, const mithep::MuonCol * muoncol )
23	+	{
24	+	bool pass = false;
25	+	for(UInt_t i=0; i<muoncol->GetEntries(); ++i) {
26	+	const mithep::Muon *mu = muoncol->At(i);
27	+	if(mu->TrackerTrk() == track && (mu->IsGlobalMuon() || mu->IsTrackerMuon()) && mu->IsPFMuon())
28	+	pass = true;
29	+	}
30	+
31	+	return pass;
32	+
33	+	}
34	+
35	+	//--------------------------------------------------------------------------------------------------
36	+	SelectionStatus muonDummyVeto(ControlFlags &ctrl,
37	+	const mithep::Muon *muon,
38	+	const mithep::Vertex * vtx)
39	+	//--------------------------------------------------------------------------------------------------
40	+	{
41	+	SelectionStatus status;
42	+	status.setStatus(SelectionStatus::PRESELECTION);
43	+	return status;
44	+	}
45	+
46	+	//--------------------------------------------------------------------------------------------------
47	+	SelectionStatus muonCutBasedVeto(ControlFlags &ctrl,
48	+	const mithep::Muon *muon,
49	+	const mithep::Vertex *vtx)
50	+	//--------------------------------------------------------------------------------------------------
51	+	{
52	+	//
53	+	// Loose cut-based ID for isolation veto
54	+	//
55	+	bool ret = true;
56	+
57	+	if(!(muon->IsGlobalMuon() || muon->IsTrackerMuon())) ret = false;
58	+	if( muon->NValidHits() < 11 )                        ret = false;
59	+	if( fabs(muon->Ip3dPVSignificance()) >= 4 )          ret = false;
60	+
61	+	SelectionStatus status;
62	+	if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
63	+	return status;
64	+	}
65	+
66	+
67		//--------------------------------------------------------------------------------------------------
68		SelectionStatus noPreSelection( ControlFlags &ctrl,  const mithep::Muon * mu )
69
#	Line 32 | Line 78 | SelectionStatus noPreSelection( ControlF
78		//--------------------------------------------------------------------------------------------------
79		SelectionStatus muonPreSelection( ControlFlags &ctrl,
80		const mithep::Muon * mu,
81	<	const mithep::Vertex & vtx,
81	>	const mithep::Vertex * vtx,
82		const mithep::Array<mithep::PFCandidate> * pfCandidates )
83		//--------------------------------------------------------------------------------------------------
84		{
85		bool ret = true;
86	<	if(ctrl.debug) cout << "inside muonpresel ... " << endl;
87	<	//  bool ret = isMuFO(mu,vtx,pfCandidates);
88	<	if(ctrl.debug) cout << "iS fo? ... " << ret << endl;
43	<	ret &= ( fabs(mu->Ip3dPVSignificance()) < 4 );
44	<	if(ctrl.debug) cout << "and pass IP (" << mu->Ip3dPVSignificance() << ") ? ... " << ret << endl;
45	<	ret &= ( mu->Pt() > 5 );
46	<	if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;
86	>
87	>	ret &= ( fabs(mu->Ip3dPVSignificance()) < 100 );
88	>	ret &= ( mu->Pt() >= 5 );
89		ret &= ( fabs(mu->Eta()) <= 2.4 );
90	<	if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;
91	<	ret &=  (mu->IsTrackerMuon() && mu->HasTrackerTrk() &&
92	<	(mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated));
93	<	if(ctrl.debug) cout << "and isTrackerMuon ? ... " << ret << endl;
52	<	//  ret &=  (mu->IsoR03SumPt()/mu->Pt() < 0.7 );
53	<	ret &=  (mu->IsoR03SumPt()/mu->Pt() < 0.7 );
54	<	if(ctrl.debug) cout << "and loose trk iso ? ... " << ret << endl;
90	>	ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
91	>	//   if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
92	>	//     ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
93	>	ret &= (  mu->IsoR03SumPt()/mu->Pt() < 0.7 );
94
95		SelectionStatus status;
96		if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
#	Line 60 | Line 99 | SelectionStatus muonPreSelection( Contro
99		}
100
101
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	–	//
102
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)
103		//--------------------------------------------------------------------------------------------------
104	<	{
105	<
106	<	int level=0;
107	<	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 )
104	>	SelectionStatus muonReferencePreSelection( ControlFlags &ctrl,
105	>	const mithep::Muon * mu,
106	>	const mithep::Vertex * vtx,
107	>	const mithep::Array<mithep::PFCandidate> * pfCandidates )
108		//--------------------------------------------------------------------------------------------------
109		{
110	<	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	<	}
110	>	bool ret = true;
111
112	+	if(ctrl.debug) cout << "inside muonpresel ... " << endl;
113	+	ret &= ( fabs(mu->Ip3dPVSignificance()) < 100 );
114	+	if(ctrl.debug) cout << "and pass IP (" << mu->Ip3dPVSignificance() << ") ? ... " << ret << endl;
115	+	ret &= ( mu->Pt() >= 5 );
116	+	if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;
117	+	ret &= ( fabs(mu->Eta()) <= 2.4 );
118	+	if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;
119	+	ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
120	+	if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
121	+	ret &= (mu->HasTrackerTrk() &&
122	+	mu->TrackerTrk()->D0Corrected(*vtx) < 0.5
123	+	&& mu->TrackerTrk()->DzCorrected(*vtx) < 1.0);
124	+	if(ctrl.debug) cout << "d0 and dz? ... " << ret << endl;
125	+
126	+	//  if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
127	+	//    ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
128	+	//  if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;
129	+	//  ret &= ( mu->BestTrk()->DzCorrected(vtx) < 0.1);
130	+	//  if( ctrl.debug ) cout << "elepresel : ret after dZcorr : " << ret <<  endl;
131
186	–	level++;
187	–	if( (mu->IsoR03SumPt()/mu->Pt()) > 0.7 ) {
188	–	failmask |= (1<<level);
189	–	}
132
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	–
133		SelectionStatus status;
134	<	if( !failmask ) status.orStatus(SelectionStatus::LOOSEID);
135	<	if( !failmask ) status.orStatus(SelectionStatus::TIGHTID);
134	>	if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
135	>	if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
136		return status;
137	+	}
138
203	–
204	–
205	–	};
206	–
207	–
208	–
209	–
210	–	//
211	–	// Kevin's WW selection
212	–	//
139		//--------------------------------------------------------------------------------------------------
140	<	SelectionStatus passMuonSelection( ControlFlags &ctrl,
141	<	const mithep::Muon * mu,
142	<	const mithep::Vertex & vtx   )
140	>	SelectionStatus muonPreSelectionNoD0DzIP( ControlFlags &ctrl,
141	>	const mithep::Muon * mu,
142	>	const mithep::Vertex * vtx,
143	>	const mithep::Array<mithep::PFCandidate> * pfCandidates )
144		//--------------------------------------------------------------------------------------------------
145		{
146	<	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	<
146	>	bool ret = true;
147
148	<	level++;
149	<	if(fabs(muTrk->D0Corrected(vtx))>0.02)   {
150	<	failmask |= (1<<level);
151	<	}
148	>	if(ctrl.debug) cout << "inside muonpresel ... " << endl;
149	>	ret &= ( mu->Pt() >= 5 );
150	>	if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;
151	>	ret &= ( fabs(mu->Eta()) <= 2.4 );
152	>	if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;
153	>	ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
154	>	if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
155	>
156	>	//  if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
157	>	//    ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
158	>	//  if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;
159	>	//  ret &= ( mu->BestTrk()->DzCorrected(vtx) < 0.1);
160	>	//  if( ctrl.debug ) cout << "elepresel : ret after dZcorr : " << ret <<  endl;
161
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	–	*/
162
163	<	if( !failmask ) status.orStatus(SelectionStatus::LOOSEID);
164	<	if( !failmask ) status.orStatus(SelectionStatus::TIGHTID);
163	>	SelectionStatus status;
164	>	if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
165	>	if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
166		return status;
167	+	}
168
169
309	–	};
170
171
172		//--------------------------------------------------------------------------------------------------
173		SelectionStatus muonIDMVASelection(ControlFlags &ctrl,
174		const mithep::Muon *mu,
175	<	const mithep::Vertex & vtx   )
175	>	const mithep::Vertex * vtx   )
176		//--------------------------------------------------------------------------------------------------
177		{
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	–
178		double global_rchi2 = (mu->IsGlobalMuon()) ? mu->GlobalTrk()->RChi2() : 0.;
179
180	+	//
181	+	// WARNING !!! KH hacked out to sync
182	+	//
183	+	/*
184		double mvaval = muIDMVA->MVAValue_ID(mu->Pt(),
185		mu->Eta(),
186		mu->IsGlobalMuon(),
#	Line 338 | Line 188 | SelectionStatus muonIDMVASelection(Contr
188		mu->TrackerTrk()->RChi2(),
189		global_rchi2,
190		(Double_t)(mu->NValidHits()),
191	<	(Double_t)(mu->TrackerTrk()->NHits()),
191	>	(Double_t)(mu->TrackerTrk()->NHits()),
192		(Double_t)(mu->TrackerTrk()->NPixelHits()),
193		(Double_t)(mu->NMatches()),
194		mu->TrkKink(),
#	Line 349 | Line 199 | SelectionStatus muonIDMVASelection(Contr
199		mu->HadS9Energy(),
200		mu->EmS9Energy(),
201		(Bool_t)(ctrl.debug) );
202	<
202	>	*/
203	>	double mvaval = 0.0;
204
205		SelectionStatus status;
206		bool pass = false;
#	Line 362 | Line 213 | SelectionStatus muonIDMVASelection(Contr
213		&& fabs(mu->Eta()) > 1.5 && mu->Pt() <= 10 && mvaval >= MUON_IDMVA_CUT_BIN2)  pass = true;
214		else if( mu->IsGlobalMuon() && mu->IsTrackerMuon()
215		&& fabs(mu->Eta()) > 1.5 && mu->Pt() > 10 && mvaval >= MUON_IDMVA_CUT_BIN3)  pass = true;
216	<	else if( !(mu->IsGlobalMuon()) && mu->IsTrackerMuon()
217	<	&& (mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated) && mvaval >= MUON_IDMVA_CUT_BIN4)
218	<	pass = true;
216	>	else if( !(mu->IsGlobalMuon()) && mu->IsTrackerMuon() && mvaval >= MUON_IDMVA_CUT_BIN4 )  pass = true;
217	>	else if( mu->IsGlobalMuon() && !(mu->IsTrackerMuon()) && mvaval >= MUON_IDMVA_CUT_BIN5 )  pass = true;
218	>
219
220
221		if( pass ) {
#	Line 391 | Line 242 | void initMuonIDMVA()
242		mithep::MuonIDMVA::kIDV0,
243		kTRUE, weightFiles);
244		}
245	+
246	+	//--------------------------------------------------------------------------------------------------
247	+	SelectionStatus muonIDPFSelection(ControlFlags &ctrl,
248	+	const mithep::Muon *mu,
249	+	const mithep::Vertex * vtx,
250	+	const mithep::Array<mithep::PFCandidate> * pfCandidates )
251	+	//--------------------------------------------------------------------------------------------------
252	+	{
253	+	bool pass = false;
254	+	SelectionStatus status; // init'ed to FAIL
255	+
256	+	// check that it matches to a PF muon
257	+	for( int i=0; i<pfCandidates->GetEntries(); i++ ) {
258	+	// tmp
259	+	if( !(PFnoPUflag[i]) ) continue; // my PF no PU hack
260	+
261	+	const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[i]);
262	+	if( (pf->TrackerTrk() == mu->TrackerTrk()) && abs(pf->PFType()) == mithep::PFCandidate::eMuon ) {
263	+	pass = true;
264	+	break;
265	+	}
266	+	}
267	+	//
268	+	//      if(mu->IsPFMuon()) pass = true;
269	+	//      else pass = false;
270	+
271	+	//
272	+	// NB : ipsig here for reference synch ...
273	+	//
274	+	#ifdef SYNC
275	+	if( pass && fabs(mu->Ip3dPVSignificance()) < 4 ) {
276	+	status.orStatus(SelectionStatus::LOOSEID);
277	+	status.orStatus(SelectionStatus::TIGHTID);
278	+	}
279	+	#else
280	+	if( pass ) {
281	+	status.orStatus(SelectionStatus::LOOSEID);
282	+	status.orStatus(SelectionStatus::TIGHTID);
283	+	}
284	+	#endif
285	+
286	+	return status;
287	+	}

Diff Legend

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