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.36 by anlevin, Wed Oct 17 01:31:22 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 muonPOG2012CutBasedIDTight(const mithep::Muon *mu, const mithep::Vertex * vtx, const mithep::Array<mithep::PFCandidate> * pfCandidates, const mithep::Array<mithep::PileupEnergyDensity> * puEnergyDensity)
23	+	{
24	+
25	+	Float_t charged_iso = 0;
26	+
27	+	Float_t sumChargedHadronPt = 0;
28	+	Float_t sumNeutralHadronPt = 0;
29	+	Float_t sumPhotonPt = 0;
30	+	Float_t sumPUPt = 0;
31	+
32	+	for(unsigned k=0; k<pfCandidates->GetEntries(); ++k) {
33	+	const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[k]);
34	+	Double_t dr = mithep::MathUtils::DeltaR(mu->Phi(),mu->Eta(), pf->Phi(), pf->Eta());
35	+	if(dr > 0.4) continue;
36	+	if(pf->TrackerTrk() && mu->TrackerTrk() && pf->TrackerTrk() == mu->TrackerTrk()) continue;
37	+
38	+	if(PFnoPUflag[k]){ //pf no pileup
39	+	if(abs(pf->PFType()) == mithep::PFCandidate::eHadron)
40	+	sumChargedHadronPt += pf->Pt();
41	+	else if (pf->Pt() > 0.5 && abs(pf->PFType()) == mithep::PFCandidate::eNeutralHadron)
42	+	sumNeutralHadronPt += pf->Pt();
43	+	else if (pf->Pt() > 0.5 && abs(pf->PFType()) == mithep::PFCandidate::eGamma)
44	+	sumPhotonPt += pf->Pt();
45	+	}
46	+	else {//pf pile up
47	+	if(pf->Charge() != 0)
48	+	sumPUPt += pf->Pt();
49	+	}
50	+
51	+	}
52	+	double iso_sum =  sumChargedHadronPt+ max(0.,sumNeutralHadronPt+sumPhotonPt-0.5*sumPUPt);
53	+
54	+	bool pass = kTRUE;
55	+	if(!mu->IsGlobalMuon()) pass = kFALSE;
56	+
57	+	//there is not necessarily a global track or a tracker track if is global muon is false
58	+	if(!pass) return pass;
59	+
60	+	if(!mu->IsPFMuon()) pass = kFALSE;
61	+	if(mu->GlobalTrk()->RChi2() > 10) pass= kFALSE;
62	+	if(mu->NValidHits() == 0 ) pass= kFALSE;
63	+	if(mu->NMatches() <= 1) pass = kFALSE;
64	+	if(fabs(mu->TrackerTrk()->D0Corrected(*vtx)) > 0.2) pass = kFALSE;
65	+	if(fabs(mu->TrackerTrk()->DzCorrected(*vtx)) > 0.5) pass = kFALSE;
66	+	if(mu->BestTrk()->NPixelHits() == 0) pass = kFALSE;
67	+	if(mu->NTrkLayersHit() <= 5) pass = kFALSE;
68	+	if(iso_sum/mu->Pt() > 0.12 ) pass = kFALSE;
69	+
70	+	return pass;
71	+
72	+
73	+	}
74	+	//----------------------------------------------------------------------------------------
75	+	bool muon2012CutBasedIDTightVersionWithOldIsoThatWorksOn2011(ControlFlags &ctrl, const mithep::Muon *mu, const mithep::Vertex * vtx, const mithep::Array<mithep::PFCandidate> * pfCandidates,
76	+	const mithep::Array<mithep::PileupEnergyDensity> * puEnergyDensity,   mithep::MuonTools::EMuonEffectiveAreaTarget EffectiveAreaVersion
77	+	)
78	+	{
79	+
80	+	mithep::MuonTools   muT;
81	+
82	+	Float_t charged_iso = 0;
83	+	Float_t neutral_iso = 0;
84	+	Float_t gamma_iso = 0;
85	+
86	+	for(int k=0; k<pfCandidates->GetEntries(); ++k) {
87	+
88	+	if( !(PFnoPUflag[k]) ) continue;
89	+	const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[k]);
90	+
91	+	Double_t dr = mithep::MathUtils::DeltaR(mu->Phi(),mu->Eta(), pf->Phi(), pf->Eta());
92	+	if (dr > 0.4) continue;
93	+
94	+	if (pf->Charge() != 0 && (pf->HasTrackerTrk()||pf->HasGsfTrk()) ) {
95	+
96	+	if (abs(pf->PFType()) == mithep::PFCandidate::eElectron || abs(pf->PFType()) == mithep::PFCandidate::eMuon) continue;
97	+
98	+	charged_iso += pf->Pt();
99	+	}
100	+
101	+	else if (abs(pf->PFType()) == mithep::PFCandidate::eGamma) {
102	+
103	+	gamma_iso += pf->Pt();
104	+	}
105	+
106	+	else {
107	+	neutral_iso += pf->Pt();
108	+	}
109	+	}
110	+
111	+	double iso_sum=999;
112	+	double rho;
113	+	if(ctrl.era == 2012) {
114	+	EffectiveAreaVersion  = mithep::MuonTools::kMuEAData2012;
115	+	rho = puEnergyDensity->At(0)->RhoKt6PFJetsCentralNeutral();
116	+
117	+	iso_sum = charged_iso + fmax(0.0,(gamma_iso + neutral_iso
118	+	-rho*muT.MuonEffectiveArea(muT.kMuGammaAndNeutralHadronIso04,
119	+	mu->Eta(),EffectiveAreaVersion)));
120	+	} else if(ctrl.era == 2011) {
121	+	setMuonRhoAndEaVersion(rho,EffectiveAreaVersion,puEnergyDensity);
122	+	iso_sum = charged_iso + fmax(0.0,(gamma_iso + neutral_iso
123	+	-rho*muT.MuonEffectiveArea(muT.kMuGammaAndNeutralHadronIso04,
124	+	mu->Eta(),EffectiveAreaVersion)));
125	+	} else assert(0);
126	+
127	+	bool pass = kTRUE;
128	+	if(!mu->IsGlobalMuon()) pass = kFALSE;
129	+
130	+	//there is not necessarily a global track or a tracker track if is global muon is false
131	+	if(!pass) return pass;
132	+
133	+	if(!mu->IsPFMuon() && ctrl.era==2012)                 pass = kFALSE;
134	+	if(mu->GlobalTrk()->RChi2() > 10)                     pass = kFALSE;
135	+	if(mu->NValidHits() == 0 )                            pass = kFALSE;
136	+	if(mu->NMatches() <= 1)                               pass = kFALSE;
137	+	if(fabs(mu->TrackerTrk()->D0Corrected(*vtx)) > 0.2)   pass = kFALSE;
138	+	if(fabs(mu->TrackerTrk()->DzCorrected(*vtx)) > 0.5)   pass = kFALSE;
139	+	if(mu->BestTrk()->NPixelHits() == 0)                  pass = kFALSE;
140	+	if(mu->NTrkLayersHit() <= 5)                          pass = kFALSE;
141	+	if(iso_sum/mu->Pt() > 0.12 )                          pass = kFALSE;
142	+
143	+	return pass;
144	+
145	+
146	+	}
147	+
148	+	bool trackToPFMuonSelection( const mithep::Track *track, const mithep::MuonCol * muoncol )
149	+	{
150	+	bool pass = false;
151	+	for(UInt_t i=0; i<muoncol->GetEntries(); ++i) {
152	+	const mithep::Muon *mu = muoncol->At(i);
153	+	if(mu->TrackerTrk() == track && (mu->IsGlobalMuon() || mu->IsTrackerMuon()) && mu->IsPFMuon())
154	+	pass = true;
155	+	}
156	+
157	+	return pass;
158	+
159	+	}
160	+
161	+	//--------------------------------------------------------------------------------------------------
162	+	SelectionStatus muonDummyVeto(ControlFlags &ctrl,
163	+	const mithep::Muon *muon,
164	+	const mithep::Vertex * vtx)
165	+	//--------------------------------------------------------------------------------------------------
166	+	{
167	+	SelectionStatus status;
168	+	status.setStatus(SelectionStatus::PRESELECTION);
169	+	return status;
170	+	}
171	+
172	+	//--------------------------------------------------------------------------------------------------
173	+	SelectionStatus muonCutBasedVeto(ControlFlags &ctrl,
174	+	const mithep::Muon *muon,
175	+	const mithep::Vertex *vtx)
176	+	//--------------------------------------------------------------------------------------------------
177	+	{
178	+	//
179	+	// Loose cut-based ID for isolation veto
180	+	//
181	+	bool ret = true;
182	+
183	+	if(!(muon->IsGlobalMuon() || muon->IsTrackerMuon())) ret = false;
184	+	if( muon->NValidHits() < 11 )                        ret = false;
185	+	if( fabs(muon->Ip3dPVSignificance()) >= 4 )          ret = false;
186	+
187	+	SelectionStatus status;
188	+	if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
189	+	return status;
190	+	}
191	+
192	+
193		//--------------------------------------------------------------------------------------------------
194		SelectionStatus noPreSelection( ControlFlags &ctrl,  const mithep::Muon * mu )
195
#	Line 32 | Line 204 | SelectionStatus noPreSelection( ControlF
204		//--------------------------------------------------------------------------------------------------
205		SelectionStatus muonPreSelection( ControlFlags &ctrl,
206		const mithep::Muon * mu,
207	<	const mithep::Vertex & vtx,
207	>	const mithep::Vertex * vtx,
208		const mithep::Array<mithep::PFCandidate> * pfCandidates )
209		//--------------------------------------------------------------------------------------------------
210		{
211		bool ret = true;
212	<	if(ctrl.debug) cout << "inside muonpresel ... " << endl;
213	<	//  bool ret = isMuFO(mu,vtx,pfCandidates);
214	<	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;
212	>
213	>	ret &= ( fabs(mu->Ip3dPVSignificance()) < 100 );
214	>	ret &= ( mu->Pt() >= 5 );
215		ret &= ( fabs(mu->Eta()) <= 2.4 );
216	<	if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;
217	<	ret &=  (mu->IsTrackerMuon() && mu->HasTrackerTrk() &&
218	<	(mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated));
219	<	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;
216	>	ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
217	>	//   if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
218	>	//     ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
219	>	ret &= (  mu->IsoR03SumPt()/mu->Pt() < 0.7 );
220
221		SelectionStatus status;
222		if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
#	Line 60 | Line 225 | SelectionStatus muonPreSelection( Contro
225		}
226
227
228	+
229		//--------------------------------------------------------------------------------------------------
230	<	bool isMuFO( const mithep::Muon * mu,
231	<	const mithep::Vertex & vtx,
232	<	const mithep::Array<mithep::PFCandidate> * pfCandidates )
230	>	SelectionStatus muonReferencePreSelection( ControlFlags &ctrl,
231	>	const mithep::Muon * mu,
232	>	const mithep::Vertex * vtx,
233	>	const mithep::Array<mithep::PFCandidate> * pfCandidates )
234		//--------------------------------------------------------------------------------------------------
235		{
236	<	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	<
236	>	bool ret = true;
237
238	<	// comment for more stats for mu fake shape
239	<	if( mu->Pt() < 20 ) {
240	<	if( reliso > 3 ) isgood=false;
241	<	} else {
242	<	if( reliso > 5 ) isgood=false;
243	<	}
238	>	if(ctrl.debug) cout << "inside muonpresel ... " << endl;
239	>	ret &= ( fabs(mu->Ip3dPVSignificance()) < 100 );
240	>	if(ctrl.debug) cout << "and pass IP (" << mu->Ip3dPVSignificance() << ") ? ... " << ret << endl;
241	>	ret &= ( mu->Pt() >= 5 );
242	>	if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;
243	>	ret &= ( fabs(mu->Eta()) <= 2.4 );
244	>	if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;
245	>	ret &= ( (mu->IsTrackerMuon() && ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated)) || mu->IsGlobalMuon() );
246	>	if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
247	>	ret &= (mu->HasTrackerTrk() &&
248	>	fabs(mu->TrackerTrk()->D0Corrected(*vtx)) < 0.5 &&
249	>	fabs(mu->TrackerTrk()->DzCorrected(*vtx)) < 1.0);
250	>	if(ctrl.debug) cout << "d0 and dz? ... " << ret << endl;
251	>
252	>	//  if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
253	>	//    ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
254	>	//  if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;
255	>	//  ret &= ( mu->BestTrk()->DzCorrected(vtx) < 0.1);
256	>	//  if( ctrl.debug ) cout << "elepresel : ret after dZcorr : " << ret <<  endl;
257
107	–	return isgood;
108	–	};
258
259	+	SelectionStatus status;
260	+	if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
261	+	if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
262	+	return status;
263	+	}
264		//--------------------------------------------------------------------------------------------------
265	<	SelectionStatus passSoftMuonSelection( ControlFlags &ctrl,
266	<	const mithep::Muon * mu ,
267	<	const mithep::Vertex & vtx)
265	>	SelectionStatus muonPreSelectionNoD0DzIP( ControlFlags &ctrl,
266	>	const mithep::Muon * mu,
267	>	const mithep::Vertex * vtx,
268	>	const mithep::Array<mithep::PFCandidate> * pfCandidates )
269		//--------------------------------------------------------------------------------------------------
270		{
271	+	bool ret = true;
272
273	<	int level=0;
274	<	unsigned failmask=0x0;
275	<
276	<	// Use tracker track when available
277	<	const mithep::Track *muTrk=0;
278	<	if(mu->HasStandaloneTrk()) { muTrk = mu->StandaloneTrk(); }
279	<	else if(mu->HasTrackerTrk())         { muTrk = mu->TrackerTrk(); }
280	<	else if(mu->HasGlobalTrk())     { muTrk = mu->GlobalTrk(); }
281	<	assert(muTrk);
282	<
283	<	if(mu->TrackerTrk()->NHits() < 11 )
284	<	failmask |= (1<<level);
285	<	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	<
273	>	if(ctrl.debug) cout << "inside muonpresel ... " << endl;
274	>	ret &= ( mu->Pt() >= 5 );
275	>	if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;
276	>	ret &= ( fabs(mu->Eta()) <= 2.4 );
277	>	if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;
278	>	ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
279	>	if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
280	>
281	>	//  if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
282	>	//    ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
283	>	//  if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;
284	>	//  ret &= ( mu->BestTrk()->DzCorrected(vtx) < 0.1);
285	>	//  if( ctrl.debug ) cout << "elepresel : ret after dZcorr : " << ret <<  endl;
286
186	–	level++;
187	–	if( (mu->IsoR03SumPt()/mu->Pt()) > 0.7 ) {
188	–	failmask |= (1<<level);
189	–	}
287
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	–
288		SelectionStatus status;
289	<	if( !failmask ) status.orStatus(SelectionStatus::LOOSEID);
290	<	if( !failmask ) status.orStatus(SelectionStatus::TIGHTID);
289	>	if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
290	>	if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
291		return status;
292	<
203	<
204	<
205	<	};
206	<
207	<
208	<
209	<
210	<	//
211	<	// Kevin's WW selection
212	<	//
292	>	}
293		//--------------------------------------------------------------------------------------------------
294	<	SelectionStatus passMuonSelection( ControlFlags &ctrl,
295	<	const mithep::Muon * mu,
296	<	const mithep::Vertex & vtx   )
294	>	SelectionStatus muonPreSelectionNoD0IP( ControlFlags &ctrl,
295	>	const mithep::Muon * mu,
296	>	const mithep::Vertex * vtx,
297	>	const mithep::Array<mithep::PFCandidate> * pfCandidates )
298		//--------------------------------------------------------------------------------------------------
299		{
300	<	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	<	*/
300	>	bool ret = true;
301
302	<	if( !failmask ) status.orStatus(SelectionStatus::LOOSEID);
303	<	if( !failmask ) status.orStatus(SelectionStatus::TIGHTID);
302	>	ret &= ( mu->Pt() >= 5 );
303	>	ret &= ( fabs(mu->Eta()) <= 2.4 );
304	>	ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
305	>	// ret &= (mu->HasTrackerTrk() &&
306	>	//      fabs(mu->TrackerTrk()->D0Corrected(*vtx)) < 0.5 &&
307	>	//      fabs(mu->TrackerTrk()->DzCorrected(*vtx)) < 1.0);
308	>	ret &= (mu->HasTrackerTrk() &&
309	>	fabs(mu->TrackerTrk()->DzCorrected(*vtx)) < 1.0);
310	>
311	>	SelectionStatus status;
312	>	if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
313	>	if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
314		return status;
315	<
308	<
309	<	};
310	<
311	<
315	>	}
316		//--------------------------------------------------------------------------------------------------
317		SelectionStatus muonIDMVASelection(ControlFlags &ctrl,
318		const mithep::Muon *mu,
319	<	const mithep::Vertex & vtx   )
319	>	const mithep::Vertex * vtx   )
320		//--------------------------------------------------------------------------------------------------
321		{
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	–
322		double global_rchi2 = (mu->IsGlobalMuon()) ? mu->GlobalTrk()->RChi2() : 0.;
323
324	+	//
325	+	// WARNING !!! KH hacked out to sync
326	+	//
327	+	/*
328		double mvaval = muIDMVA->MVAValue_ID(mu->Pt(),
329		mu->Eta(),
330		mu->IsGlobalMuon(),
#	Line 338 | Line 332 | SelectionStatus muonIDMVASelection(Contr
332		mu->TrackerTrk()->RChi2(),
333		global_rchi2,
334		(Double_t)(mu->NValidHits()),
335	<	(Double_t)(mu->TrackerTrk()->NHits()),
335	>	(Double_t)(mu->TrackerTrk()->NHits()),
336		(Double_t)(mu->TrackerTrk()->NPixelHits()),
337		(Double_t)(mu->NMatches()),
338		mu->TrkKink(),
#	Line 349 | Line 343 | SelectionStatus muonIDMVASelection(Contr
343		mu->HadS9Energy(),
344		mu->EmS9Energy(),
345		(Bool_t)(ctrl.debug) );
346	<
346	>	*/
347	>	double mvaval = 0.0;
348
349		SelectionStatus status;
350		bool pass = false;
#	Line 362 | Line 357 | SelectionStatus muonIDMVASelection(Contr
357		&& fabs(mu->Eta()) > 1.5 && mu->Pt() <= 10 && mvaval >= MUON_IDMVA_CUT_BIN2)  pass = true;
358		else if( mu->IsGlobalMuon() && mu->IsTrackerMuon()
359		&& fabs(mu->Eta()) > 1.5 && mu->Pt() > 10 && mvaval >= MUON_IDMVA_CUT_BIN3)  pass = true;
360	<	else if( !(mu->IsGlobalMuon()) && mu->IsTrackerMuon()
361	<	&& (mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated) && mvaval >= MUON_IDMVA_CUT_BIN4)
362	<	pass = true;
360	>	else if( !(mu->IsGlobalMuon()) && mu->IsTrackerMuon() && mvaval >= MUON_IDMVA_CUT_BIN4 )  pass = true;
361	>	else if( mu->IsGlobalMuon() && !(mu->IsTrackerMuon()) && mvaval >= MUON_IDMVA_CUT_BIN5 )  pass = true;
362	>
363
364
365		if( pass ) {
#	Line 391 | Line 386 | void initMuonIDMVA()
386		mithep::MuonIDMVA::kIDV0,
387		kTRUE, weightFiles);
388		}
389	+
390	+	//--------------------------------------------------------------------------------------------------
391	+	SelectionStatus muonIDPFSelection(ControlFlags &ctrl,
392	+	const mithep::Muon *mu,
393	+	const mithep::Vertex * vtx,
394	+	const mithep::Array<mithep::PFCandidate> * pfCandidates )
395	+	//--------------------------------------------------------------------------------------------------
396	+	{
397	+	bool pass = false;
398	+	SelectionStatus status; // init'ed to FAIL
399	+
400	+	// check that it matches to a PF muon
401	+	for( int i=0; i<pfCandidates->GetEntries(); i++ ) {
402	+	// tmp
403	+	if( !(PFnoPUflag[i]) ) continue; // my PF no PU hack
404	+
405	+	const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[i]);
406	+	if( (pf->TrackerTrk() == mu->TrackerTrk()) && abs(pf->PFType()) == mithep::PFCandidate::eMuon ) {
407	+	pass = true;
408	+	break;
409	+	}
410	+	}
411	+	//
412	+	//      if(mu->IsPFMuon()) pass = true;
413	+	//      else pass = false;
414	+
415	+	if( pass ) {
416	+	status.orStatus(SelectionStatus::LOOSEID);
417	+	status.orStatus(SelectionStatus::TIGHTID);
418	+	}
419	+
420	+	return status;
421	+	}
422	+	//----------------------------------------------------------------------------------------
423	+	SelectionStatus PassWwMuonSel(const mithep::Muon *mu, const mithep::Vertex *vtx,
424	+	const mithep::Array<mithep::PFCandidate> *pfArr
425	+	)
426	+	{
427	+	bool passID=true;
428	+	bool passIso=true;
429	+
430	+	if (mu->Pt() < 5)
431	+	passID = false;
432	+	if (fabs(mu->Eta()) > 2.4)
433	+	passID = false;
434	+	if (mu->BestTrk()->PtErr()/mu->Pt() > 0.1)
435	+	passID = false;
436	+	if (fabs(mu->BestTrk()->DzCorrected(*vtx)) > 0.1)
437	+	passID = false;
438	+	Bool_t isGlobal  = (mu->IsGlobalMuon()) && (mu->BestTrk()->RChi2() < 10) &&
439	+	(mu->NMatches() > 1) && (mu->NValidHits() > 0);
440	+	Bool_t isTracker = (mu->IsTrackerMuon()) &&
441	+	(mu->Quality().Quality(mithep::MuonQuality::TMLastStationTight));
442	+	if (!isGlobal && !isTracker)
443	+	passID = false;
444	+	int ntrkhits = (mu->HasTrackerTrk()) ? mu->TrackerTrk()->NHits() : 0;
445	+	if (ntrkhits < 10)
446	+	passID = false;
447	+	if (mu->BestTrk()->NPixelHits() < 1)
448	+	passID = false;
449	+	if (fabs(mu->BestTrk()->D0Corrected(*vtx)) > 0.02)
450	+	passID = false;
451	+
452	+	// note: this isn't really ww muon isolation
453	+	float pfiso = computePFMuonIso(mu,vtx,pfArr,0.3);
454	+
455	+	if (pfiso > 0.2*mu->Pt())
456	+	passIso = false;
457	+
458	+	SelectionStatus status;
459	+	if(passID)  status |= SelectionStatus::TIGHTIDANDPRE;
460	+	if(passIso) status |= SelectionStatus::TIGHTISOANDPRE;
461	+
462	+	return status;
463	+
464	+	}

Diff Legend

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