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.12 by khahn, Sat May 5 13:09:05 2012 UTC vs.
Revision 1.26 by khahn, Tue Jun 5 15:00:20 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)
38	>	const mithep::Vertex * vtx)
39		//--------------------------------------------------------------------------------------------------
40		{
41		SelectionStatus status;
#	Line 32 | Line 46 | SelectionStatus muonDummyVeto(ControlFla
46		//--------------------------------------------------------------------------------------------------
47		SelectionStatus muonCutBasedVeto(ControlFlags &ctrl,
48		const mithep::Muon *muon,
49	<	const mithep::Vertex &vtx)
49	>	const mithep::Vertex *vtx)
50		//--------------------------------------------------------------------------------------------------
51		{
52		//
#	Line 64 | 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	+
87	+	ret &= ( fabs(mu->Ip3dPVSignificance()) < 100 );
88	+	ret &= ( mu->Pt() >= 5 );
89	+	ret &= ( fabs(mu->Eta()) <= 2.4 );
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);
97	+	if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
98	+	return status;
99	+	}
100	+
101	+
102	+
103	+	//--------------------------------------------------------------------------------------------------
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	+	bool ret = true;
111	+
112		if(ctrl.debug) cout << "inside muonpresel ... " << endl;
113	<	//  bool ret = isMuFO(mu,vtx,pfCandidates);
74	<	if(ctrl.debug) cout << "iS fo? ... " << ret << endl;
75	<	ret &= ( fabs(mu->Ip3dPVSignificance()) < 4 );
113	>	ret &= ( fabs(mu->Ip3dPVSignificance()) < 100 );
114		if(ctrl.debug) cout << "and pass IP (" << mu->Ip3dPVSignificance() << ") ? ... " << ret << endl;
115	<	ret &= ( mu->Pt() > 5 );
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	>	fabs(mu->TrackerTrk()->D0Corrected(*vtx)) < 0.5 &&
123	>	fabs(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	>
132	>
133	>	SelectionStatus status;
134	>	if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
135	>	if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
136	>	return status;
137	>	}
138	>
139	>	//--------------------------------------------------------------------------------------------------
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	>	bool ret = true;
147	>
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	<	if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
156	<	ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
157	<	if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;
158	<	ret &= ( mu->BestTrk()->DzCorrected(vtx) < 0.1);
159	<	if( ctrl.debug ) cout << "elepresel : ret after dZcorr : " << ret <<  endl;
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
162
163		SelectionStatus status;
#	Line 99 | Line 172 | SelectionStatus muonPreSelection( Contro
172		//--------------------------------------------------------------------------------------------------
173		SelectionStatus muonIDMVASelection(ControlFlags &ctrl,
174		const mithep::Muon *mu,
175	<	const mithep::Vertex & vtx   )
175	>	const mithep::Vertex * vtx   )
176		//--------------------------------------------------------------------------------------------------
177		{
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 122 | 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 168 | Line 246 | void initMuonIDMVA()
246		//--------------------------------------------------------------------------------------------------
247		SelectionStatus muonIDPFSelection(ControlFlags &ctrl,
248		const mithep::Muon *mu,
249	<	const mithep::Vertex & vtx,
249	>	const mithep::Vertex * vtx,
250		const mithep::Array<mithep::PFCandidate> * pfCandidates )
251		//--------------------------------------------------------------------------------------------------
252		{
#	Line 177 | Line 255 | SelectionStatus muonIDPFSelection(Contro
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->HasTrackerTrk() ) {
263	<	if( (pf->TrackerTrk() == mu->TrackerTrk()) && abs(pf->PFType()) == mithep::PFCandidate::eMuon ) {
264	<	pass = true;
184	<	break;
185	<	}
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