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.3 by anlevin, Wed Feb 29 10:10:18 2012 UTC vs.
Revision 1.30 by dkralph, Thu Jul 5 13:43:13 2012 UTC

#	Line 1 | Line 1
1		#include <math.h>
2		#include <iostream>
3
4	–	#include "HiggsAnaDefs.hh"
5	–
4		#include "MuonSelection.h"
5		#include "ParseArgs.h"
6		#include "SelectionStatus.h"
7
8	<	SelectionStatus noPreSelection( ControlFlags &ctrl,  const mithep::TMuon * mu ) {
9	<	SelectionStatus status;
10	<	status.setStatus(SelectionStatus::PRESELECTION);
11	<	if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
12	<	return status;
13	<	}
8	>	#include "Track.h"
9	>	#include "MuonTools.h"
10	>	#include "MuonQuality.h"
11	>	#include "MuonIDMVA.h"
12	>
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,
19	>	const mithep::Array<mithep::PFCandidate> * pfCandidates,
20	>	const Double_t dRMax);
21	>
22	>	bool muon2012CutBasedIDTight(ControlFlags &ctrl, const mithep::Muon *mu, const mithep::Vertex * vtx, const mithep::Array<mithep::PFCandidate> * pfCandidates,
23	>	const mithep::Array<mithep::PileupEnergyDensity> * puEnergyDensity,   mithep::MuonTools::EMuonEffectiveAreaTarget EffectiveAreaVersion
24	>	)
25	>	{
26	>
27	>	mithep::MuonTools   muT;
28	>
29	>	Float_t charged_iso = 0;
30	>	Float_t neutral_iso = 0;
31	>	Float_t gamma_iso = 0;
32	>
33	>	for(int k=0; k<pfCandidates->GetEntries(); ++k) {
34	>
35	>	if( !(PFnoPUflag[k]) ) continue;
36	>	const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[k]);
37	>
38	>	Double_t dr = mithep::MathUtils::DeltaR(mu->Phi(),mu->Eta(), pf->Phi(), pf->Eta());
39	>	if (dr > 0.4) continue;
40	>
41	>	if (pf->Charge() != 0 && (pf->HasTrackerTrk()||pf->HasGsfTrk()) ) {
42	>
43	>	if (abs(pf->PFType()) == mithep::PFCandidate::eElectron || abs(pf->PFType()) == mithep::PFCandidate::eMuon) continue;
44	>
45	>	charged_iso += pf->Pt();
46	>	}
47	>
48	>	else if (abs(pf->PFType()) == mithep::PFCandidate::eGamma) {
49	>
50	>	gamma_iso += pf->Pt();
51	>	}
52	>
53	>	else {
54	>	neutral_iso += pf->Pt();
55	>	}
56	>	}
57
58	<	SelectionStatus muonPreSelection( ControlFlags &ctrl,  const mithep::TMuon * mu ) {
59	<	if(ctrl.debug) cout << "inside muonpresel ... " << endl;
60	<	bool ret = isMuFO(mu);
61	<	if(ctrl.debug) cout << "iS fo? ... " << ret << endl;
62	<	ret &= ( fabs(mu->ip3dSig) < 4 );
63	<	if(ctrl.debug) cout << "and pass IP (" << mu->ip3dSig << ") ? ... " << ret << endl;
64	<	ret &= ( mu->pt > 5 );
65	<	if(ctrl.debug) cout << "and >5 GeV ? ... " << ret << endl;
66	<	ret &= ( fabs(mu->eta) < 2.4 );
67	<	if(ctrl.debug) cout << "and < 2.4 eta ? ... " << ret << endl;
58	>	double iso_sum=999;
59	>	double rho;
60	>	if(ctrl.era == 2012) {
61	>	EffectiveAreaVersion  = mithep::MuonTools::kMuEAData2012;
62	>	rho = puEnergyDensity->At(0)->RhoKt6PFJetsCentralNeutral();
63	>
64	>	iso_sum = charged_iso + fmax(0.0,(gamma_iso + neutral_iso
65	>	-rho*muT.MuonEffectiveArea(muT.kMuGammaAndNeutralHadronIso04,
66	>	mu->Eta(),EffectiveAreaVersion)));
67	>	} else if(ctrl.era == 2011) {
68	>	setMuonRhoAndEaVersion(rho,EffectiveAreaVersion,puEnergyDensity);
69	>	iso_sum = charged_iso + fmax(0.0,(gamma_iso + neutral_iso
70	>	-rho*muT.MuonEffectiveArea(muT.kMuGammaAndNeutralHadronIso04,
71	>	mu->Eta(),EffectiveAreaVersion)));
72	>	} else assert(0);
73	>
74	>	bool pass = kTRUE;
75	>	if(!mu->IsGlobalMuon()) pass = kFALSE;
76	>
77	>	//there is not necessarily a global track or a tracker track if is global muon is false
78	>	if(!pass) return pass;
79	>
80	>	if(!mu->IsPFMuon() && ctrl.era==2012)                 pass = kFALSE;
81	>	if(mu->GlobalTrk()->RChi2() > 10)                     pass = kFALSE;
82	>	if(mu->NValidHits() == 0 )                            pass = kFALSE;
83	>	if(mu->NMatches() <= 1)                               pass = kFALSE;
84	>	if(fabs(mu->TrackerTrk()->D0Corrected(*vtx)) > 0.2)   pass = kFALSE;
85	>	if(fabs(mu->TrackerTrk()->DzCorrected(*vtx)) > 0.5)   pass = kFALSE;
86	>	if(mu->BestTrk()->NPixelHits() == 0)                  pass = kFALSE;
87	>	if(mu->NTrkLayersHit() <= 5)                          pass = kFALSE;
88	>	if(iso_sum/mu->Pt() > 0.12 )                          pass = kFALSE;
89	>
90	>	return pass;
91	>
92	>
93	>	}
94	>
95	>	bool trackToPFMuonSelection( const mithep::Track *track, const mithep::MuonCol * muoncol )
96	>	{
97	>	bool pass = false;
98	>	for(UInt_t i=0; i<muoncol->GetEntries(); ++i) {
99	>	const mithep::Muon *mu = muoncol->At(i);
100	>	if(mu->TrackerTrk() == track && (mu->IsGlobalMuon() || mu->IsTrackerMuon()) && mu->IsPFMuon())
101	>	pass = true;
102	>	}
103	>
104	>	return pass;
105	>
106	>	}
107
108	+	//--------------------------------------------------------------------------------------------------
109	+	SelectionStatus muonDummyVeto(ControlFlags &ctrl,
110	+	const mithep::Muon *muon,
111	+	const mithep::Vertex * vtx)
112	+	//--------------------------------------------------------------------------------------------------
113	+	{
114		SelectionStatus status;
115	<	if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
116	<	if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
117	<	return status;
32	<	}
33	<
34	<	bool isMuFO( const mithep::TMuon * mu ) {
35	<	bool isgood=true;
36	<	float reliso = mu->pfIso03/mu->pt;
37	<
38	<	if( mu->pt < 5  )         isgood=false;
39	<	if ( fabs(mu->dz) > 0.2 ) isgood=false;   // needed to remove cosmics in HF sample
40	<
41	<	//
42	<
43	<	// HF seems to not want tkquality, SS does
44	<	if( mu->pt>20 ) {
45	<	if ( !((mu->typeBits & kGlobal) && mu->nSeg>0  ) ) isgood=false; //&& mu->nValidHits>0
46	<	} else {
47	<	if ( !((mu->typeBits & kGlobal) && mu->nSeg>0 ) //&& //&& mu->nValidHits>0
48	<	&& !( (mu->typeBits & kTracker) &&  (mu->qualityBits & kTMOneStationLoose)
49	<	)
50	<	)   isgood=false;
51	<	}
52	<
115	>	status.setStatus(SelectionStatus::PRESELECTION);
116	>	return status;
117	>	}
118
119	+	//--------------------------------------------------------------------------------------------------
120	+	SelectionStatus muonCutBasedVeto(ControlFlags &ctrl,
121	+	const mithep::Muon *muon,
122	+	const mithep::Vertex *vtx)
123	+	//--------------------------------------------------------------------------------------------------
124	+	{
125		//
126	<	// for HF MC to agree w/ data, cannot put any Tk quality bits
127	<	// but to get to 5% closure in the SS test, we need at least Tk oneStationLoose
128	<
58	<	/*
59	<	&& ( (mu->qualityBits & kTMOneStationLoose) ||
60	<	(mu->qualityBits & kTMLastStationOptimizedBarrelLowPtLoose) ||
61	<	(mu->qualityBits & kTMLastStationOptimizedLowPtLoose)
62	<	)
63	<	*/
64	<
65	<
66	<	// comment for more stats for mu fake shape
67	<	if( mu->pt < 20 ) {
68	<	if( reliso > 3 ) isgood=false;
69	<	} else {
70	<	if( reliso > 5 ) isgood=false;
71	<	}
72	<
73	<	return isgood;
74	<	};
75	<
76	<
77	<	SelectionStatus passSoftMuonSelection( ControlFlags &ctrl, const mithep::TMuon * mu ) {
78	<
79	<	int level=0;
80	<	unsigned failmask=0x0;
81	<
82	<	if(mu->nTkHits < 11 )
83	<	failmask |= (1<<level);
84	<	level++;
85	<
86	<	if(fabs(mu->d0) > 0.2)
87	<	failmask |= (1<<level);
88	<	level++;
89	<
90	<	if(fabs(mu->dz) > 0.1)
91	<	failmask |= (1<<level);
92	<	level++;
93	<
94	<	if(!(mu->typeBits & kTracker))
95	<	failmask |= (1<<level);
96	<	level++;
97	<
98	<	if(!(mu->qualityBits & kTMLastStationAngTight))
99	<	failmask |= (1<<level);
100	<	level++;
101	<
102	<	Double_t iso = (mu->trkIso03 + mu->emIso03 + mu->hadIso03)/mu->pt;
103	<	if(mu->pt>20 && iso<0.1)
104	<	failmask |= (1<<level);
105	<
106	<	bool passtight  = !(failmask);
107	<	bool passloose  = !(failmask);
108	<	return SelectionStatus( passtight | passloose );
126	>	// Loose cut-based ID for isolation veto
127	>	//
128	>	bool ret = true;
129
130	<	}
131	<
132	<
113	<	//unsigned passMuonSelectionZZ( const mithep::TMuon * mu ) {
114	<	SelectionStatus passMuonSelectionZZ( ControlFlags &ctrl, const mithep::TMuon * mu ) {
115	<	int level=0;
116	<	unsigned failmask=0x0;
117	<
118	<	if(mu->pt < 5) {
119	<	failmask |= (1<<level);
120	<	}
121	<
122	<	level++;
123	<	if(fabs(mu->eta) > 2.4) {
124	<	failmask |= (1<<level);
125	<	}
126	<
127	<	level++;
128	<	if(!(mu->typeBits & kGlobal)) {
129	<	failmask |= (1<<level);
130	<	}
131	<
132	<	level++;
133	<	if(mu->nTkHits          < 10) {
134	<	failmask |= (1<<level);
135	<	}
130	>	if(!(muon->IsGlobalMuon() || muon->IsTrackerMuon())) ret = false;
131	>	if( muon->NValidHits() < 11 )                        ret = false;
132	>	if( fabs(muon->Ip3dPVSignificance()) >= 4 )          ret = false;
133
137	–
138	–	level++;
139	–	if( (mu->trkIso03/mu->pt) > 0.7 ) {
140	–	failmask |= (1<<level);
141	–	}
142	–
143	–	//  if(muon->nPixHits   < 1)     return false;
144	–	//  if(muon->muNchi2    > 10)    return false;
145	–	//  if(muon->nMatch     < 2)     return false;
146	–	//  if(muon->nValidHits         < 1)     return false;
147	–	//  if(muon->pterr/muon->pt > 0.1)   return false;
148	–	//  if(fabs(muon->dz)       > 0.1)   return false;
149	–
134		SelectionStatus status;
135	<	if( !failmask ) status.setStatus(SelectionStatus::LOOSEID);
152	<	if( !failmask ) status.setStatus(SelectionStatus::TIGHTID);
135	>	if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
136		return status;
137	+	}
138
139
140	+	//--------------------------------------------------------------------------------------------------
141	+	SelectionStatus noPreSelection( ControlFlags &ctrl,  const mithep::Muon * mu )
142
143	<	};
144	<
143	>	//--------------------------------------------------------------------------------------------------
144	>	{
145	>	SelectionStatus status;
146	>	status.setStatus(SelectionStatus::PRESELECTION);
147	>	if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
148	>	return status;
149	>	}
150
151	+	//--------------------------------------------------------------------------------------------------
152	+	SelectionStatus muonPreSelection( ControlFlags &ctrl,
153	+	const mithep::Muon * mu,
154	+	const mithep::Vertex * vtx,
155	+	const mithep::Array<mithep::PFCandidate> * pfCandidates )
156	+	//--------------------------------------------------------------------------------------------------
157	+	{
158	+	bool ret = true;
159	+
160	+	ret &= ( fabs(mu->Ip3dPVSignificance()) < 100 );
161	+	ret &= ( mu->Pt() >= 5 );
162	+	ret &= ( fabs(mu->Eta()) <= 2.4 );
163	+	ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
164	+	//   if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
165	+	//     ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
166	+	ret &= (  mu->IsoR03SumPt()/mu->Pt() < 0.7 );
167
168	+	SelectionStatus status;
169	+	if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
170	+	if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
171	+	return status;
172	+	}
173
162	–	//
163	–	// Kevin's WW selection
164	–	//
165	–	SelectionStatus passMuonSelection( ControlFlags &ctrl, const mithep::TMuon * mu ) {
166	–	int level=0;
167	–	unsigned failmask=0x0;
174
175
176	<	// 0x1
177	<	if(mu->pt < 5) {
178	<	failmask |= (1<<level);
179	<	}
176	>	//--------------------------------------------------------------------------------------------------
177	>	SelectionStatus muonReferencePreSelection( ControlFlags &ctrl,
178	>	const mithep::Muon * mu,
179	>	const mithep::Vertex * vtx,
180	>	const mithep::Array<mithep::PFCandidate> * pfCandidates )
181	>	//--------------------------------------------------------------------------------------------------
182	>	{
183	>	bool ret = true;
184
185	<	// 0x2
186	<	level++;
187	<	if(fabs(mu->eta) > 2.4) {
188	<	failmask |= (1<<level);
189	<	}
185	>	if(ctrl.debug) cout << "inside muonpresel ... " << endl;
186	>	ret &= ( fabs(mu->Ip3dPVSignificance()) < 100 );
187	>	if(ctrl.debug) cout << "and pass IP (" << mu->Ip3dPVSignificance() << ") ? ... " << ret << endl;
188	>	ret &= ( mu->Pt() >= 5 );
189	>	if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;
190	>	ret &= ( fabs(mu->Eta()) <= 2.4 );
191	>	if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;
192	>	ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
193	>	if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
194	>	ret &= (mu->HasTrackerTrk() &&
195	>	fabs(mu->TrackerTrk()->D0Corrected(*vtx)) < 0.5 &&
196	>	fabs(mu->TrackerTrk()->DzCorrected(*vtx)) < 1.0);
197	>	if(ctrl.debug) cout << "d0 and dz? ... " << ret << endl;
198	>
199	>	//  if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
200	>	//    ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
201	>	//  if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;
202	>	//  ret &= ( mu->BestTrk()->DzCorrected(vtx) < 0.1);
203	>	//  if( ctrl.debug ) cout << "elepresel : ret after dZcorr : " << ret <<  endl;
204
181	–	// 0x4
182	–	level++;
183	–	if(mu->ptErr/mu->pt > 0.1)   {
184	–	failmask |= (1<<level);
185	–	}
205
206	<	// 0x8
207	<	level++;
208	<	if( fabs(mu->dz) > 0.1 )   {
209	<	failmask |= (1<<level);
210	<	}
206	>	SelectionStatus status;
207	>	if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
208	>	if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
209	>	return status;
210	>	}
211	>	//--------------------------------------------------------------------------------------------------
212	>	SelectionStatus muonPreSelectionNoD0DzIP( ControlFlags &ctrl,
213	>	const mithep::Muon * mu,
214	>	const mithep::Vertex * vtx,
215	>	const mithep::Array<mithep::PFCandidate> * pfCandidates )
216	>	//--------------------------------------------------------------------------------------------------
217	>	{
218	>	bool ret = true;
219
220	+	if(ctrl.debug) cout << "inside muonpresel ... " << endl;
221	+	ret &= ( mu->Pt() >= 5 );
222	+	if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;
223	+	ret &= ( fabs(mu->Eta()) <= 2.4 );
224	+	if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;
225	+	ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
226	+	if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
227	+
228	+	//  if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
229	+	//    ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
230	+	//  if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;
231	+	//  ret &= ( mu->BestTrk()->DzCorrected(vtx) < 0.1);
232	+	//  if( ctrl.debug ) cout << "elepresel : ret after dZcorr : " << ret <<  endl;
233
194	–	Bool_t isGlobal  = (mu->typeBits & kGlobal) && (mu->muNchi2 < 10) && (mu->nMatch > 1) && (mu->nValidHits > 0);
195	–	Bool_t isTracker = (mu->typeBits & kTracker) && (mu->qualityBits & kTMLastStationTight);
234
235	<	// 0x10
236	<	level++;
237	<	if(!isGlobal && !isTracker) {
238	<	failmask |= (1<<level);
239	<	}
235	>	SelectionStatus status;
236	>	if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
237	>	if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
238	>	return status;
239	>	}
240
203	–	// 0x20
204	–	level++;
205	–	if( mu->nTkHits < 10 ) {
206	–	failmask |= (1<<level);
207	–	}
241
209	–	level++;
210	–	if(mu->nPixHits         < 1) {
211	–	failmask |= (1<<level);
212	–	}
242
243
244	<	level++;
245	<	if(fabs(mu->d0)>0.02)   {
246	<	failmask |= (1<<level);
247	<	}
244	>	//--------------------------------------------------------------------------------------------------
245	>	SelectionStatus muonIDMVASelection(ControlFlags &ctrl,
246	>	const mithep::Muon *mu,
247	>	const mithep::Vertex * vtx   )
248	>	//--------------------------------------------------------------------------------------------------
249	>	{
250	>	double global_rchi2 = (mu->IsGlobalMuon()) ? mu->GlobalTrk()->RChi2() : 0.;
251
252	+	//
253	+	// WARNING !!! KH hacked out to sync
254	+	//
255		/*
256	<	if(mu->pt>20) {
257	<	if(fabs(mu->d0)>0.02)   {
258	<	failmask |= (1<<level);
259	<	}
260	<	} else {
261	<	if(fabs(mu->d0)>0.01)  {
262	<	failmask |= (1<<level);
263	<	}
264	<	}
256	>	double mvaval = muIDMVA->MVAValue_ID(mu->Pt(),
257	>	mu->Eta(),
258	>	mu->IsGlobalMuon(),
259	>	mu->IsTrackerMuon(),
260	>	mu->TrackerTrk()->RChi2(),
261	>	global_rchi2,
262	>	(Double_t)(mu->NValidHits()),
263	>	(Double_t)(mu->TrackerTrk()->NHits()),
264	>	(Double_t)(mu->TrackerTrk()->NPixelHits()),
265	>	(Double_t)(mu->NMatches()),
266	>	mu->TrkKink(),
267	>	muTools.GetSegmentCompatability(mu),
268	>	muTools.GetCaloCompatability(mu,true,true),
269	>	mu->HadEnergy(),
270	>	mu->EmEnergy(),
271	>	mu->HadS9Energy(),
272	>	mu->EmS9Energy(),
273	>	(Bool_t)(ctrl.debug) );
274		*/
275	+	double mvaval = 0.0;
276
277		SelectionStatus status;
278	<	if( !failmask ) status.setStatus(SelectionStatus::LOOSEID);
279	<	if( !failmask ) status.setStatus(SelectionStatus::TIGHTID);
280	<	return status;
281	<
278	>	bool pass = false;
279	>
280	>	if( mu->IsGlobalMuon() && mu->IsTrackerMuon()
281	>	&& fabs(mu->Eta()) <= 1.5 && mu->Pt() <= 10 && mvaval >= MUON_IDMVA_CUT_BIN0)  pass = true;
282	>	else if( mu->IsGlobalMuon() && mu->IsTrackerMuon()
283	>	&& fabs(mu->Eta()) <= 1.5 && mu->Pt() > 10 && mvaval >= MUON_IDMVA_CUT_BIN1)  pass = true;
284	>	else if( mu->IsGlobalMuon() && mu->IsTrackerMuon()
285	>	&& fabs(mu->Eta()) > 1.5 && mu->Pt() <= 10 && mvaval >= MUON_IDMVA_CUT_BIN2)  pass = true;
286	>	else if( mu->IsGlobalMuon() && mu->IsTrackerMuon()
287	>	&& fabs(mu->Eta()) > 1.5 && mu->Pt() > 10 && mvaval >= MUON_IDMVA_CUT_BIN3)  pass = true;
288	>	else if( !(mu->IsGlobalMuon()) && mu->IsTrackerMuon() && mvaval >= MUON_IDMVA_CUT_BIN4 )  pass = true;
289	>	else if( mu->IsGlobalMuon() && !(mu->IsTrackerMuon()) && mvaval >= MUON_IDMVA_CUT_BIN5 )  pass = true;
290	>
291	>
292	>
293	>	if( pass ) {
294	>	status.orStatus(SelectionStatus::LOOSEID);
295	>	status.orStatus(SelectionStatus::TIGHTID);
296	>	}
297
298	<	};
298	>	return status;
299	>	}
300
301	<	SelectionStatus passMuonSelectionBackDoor( ControlFlags &ctrl, const mithep::TMuon * mu ) {
302	<	int level=0;
303	<	unsigned failmask=0x0;
301	>	//--------------------------------------------------------------------------------------------------
302	>	void initMuonIDMVA()
303	>	//--------------------------------------------------------------------------------------------------
304	>	{
305	>	muIDMVA = new mithep::MuonIDMVA();
306	>	vector<string> weightFiles;
307	>	weightFiles.push_back("./data/MuonIDMVAWeights/MuonIDMVA_BDTG_barrel_lowpt_V2.weights.xml");
308	>	weightFiles.push_back("./data/MuonIDMVAWeights/MuonIDMVA_BDTG_barrel_highpt_V2.weights.xml");
309	>	weightFiles.push_back("./data/MuonIDMVAWeights/MuonIDMVA_BDTG_endcap_lowpt_V2.weights.xml");
310	>	weightFiles.push_back("./data/MuonIDMVAWeights/MuonIDMVA_BDTG_endcap_highpt_V2.weights.xml");
311	>	weightFiles.push_back("./data/MuonIDMVAWeights/MuonIDMVA_BDTG_tracker_V2.weights.xml");
312	>	weightFiles.push_back("./data/MuonIDMVAWeights/MuonIDMVA_BDTG_global_V2.weights.xml");
313	>	muIDMVA->Initialize( "MuonIDMVA",
314	>	mithep::MuonIDMVA::kIDV0,
315	>	kTRUE, weightFiles);
316	>	}
317
318	<	if(mu->pt < 5) {
319	<	failmask |= (1<<level);
318	>	//--------------------------------------------------------------------------------------------------
319	>	SelectionStatus muonIDPFSelection(ControlFlags &ctrl,
320	>	const mithep::Muon *mu,
321	>	const mithep::Vertex * vtx,
322	>	const mithep::Array<mithep::PFCandidate> * pfCandidates )
323	>	//--------------------------------------------------------------------------------------------------
324	>	{
325	>	bool pass = false;
326	>	SelectionStatus status; // init'ed to FAIL
327	>
328	>	// check that it matches to a PF muon
329	>	for( int i=0; i<pfCandidates->GetEntries(); i++ ) {
330	>	// tmp
331	>	if( !(PFnoPUflag[i]) ) continue; // my PF no PU hack
332	>
333	>	const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[i]);
334	>	if( (pf->TrackerTrk() == mu->TrackerTrk()) && abs(pf->PFType()) == mithep::PFCandidate::eMuon ) {
335	>	pass = true;
336	>	break;
337	>	}
338		}
339	<
340	<	level++;
341	<	if(fabs(mu->eta) > 2.5) {
342	<	failmask |= (1<<level);
339	>	//
340	>	//      if(mu->IsPFMuon()) pass = true;
341	>	//      else pass = false;
342	>
343	>	if( pass ) {
344	>	status.orStatus(SelectionStatus::LOOSEID);
345	>	status.orStatus(SelectionStatus::TIGHTID);
346		}
347
348	<	level++;
349	<	if(!(mu->typeBits & kGlobal) || !(mu->typeBits & kTracker)) {
350	<	failmask |= (1<<level);
351	<	}
348	>	return status;
349	>	}
350	>	//----------------------------------------------------------------------------------------
351	>	SelectionStatus PassWwMuonSel(const mithep::Muon *mu, const mithep::Vertex *vtx,
352	>	const mithep::Array<mithep::PFCandidate> *pfArr
353	>	)
354	>	{
355	>	bool passID=true;
356	>	bool passIso=true;
357	>
358	>	if (mu->Pt() < 5)
359	>	passID = false;
360	>	if (fabs(mu->Eta()) > 2.4)
361	>	passID = false;
362	>	if (mu->BestTrk()->PtErr()/mu->Pt() > 0.1)
363	>	passID = false;
364	>	if (fabs(mu->BestTrk()->DzCorrected(*vtx)) > 0.1)
365	>	passID = false;
366	>	Bool_t isGlobal  = (mu->IsGlobalMuon()) && (mu->BestTrk()->RChi2() < 10) &&
367	>	(mu->NMatches() > 1) && (mu->NValidHits() > 0);
368	>	Bool_t isTracker = (mu->IsTrackerMuon()) &&
369	>	(mu->Quality().Quality(mithep::MuonQuality::TMLastStationTight));
370	>	if (!isGlobal && !isTracker)
371	>	passID = false;
372	>	int ntrkhits = (mu->HasTrackerTrk()) ? mu->TrackerTrk()->NHits() : 0;
373	>	if (ntrkhits < 10)
374	>	passID = false;
375	>	if (mu->BestTrk()->NPixelHits() < 1)
376	>	passID = false;
377	>	if (fabs(mu->BestTrk()->D0Corrected(*vtx)) > 0.02)
378	>	passID = false;
379	>
380	>	// note: this isn't really ww muon isolation
381	>	float pfiso = computePFMuonIso(mu,vtx,pfArr,0.3);
382
383	<	level++;
384	<	if(mu->nTkHits          <= 10) {
260	<	failmask |= (1<<level);
261	<	}
383	>	if (pfiso > 0.2*mu->Pt())
384	>	passIso = false;
385
263	–	level++;
264	–	if(mu->ip3dSig > 4) {
265	–	failmask |= (1<<level);
266	–	}
267	–
268	–	level++;
269	–	if(mu->muNchi2          >= 10)  {
270	–	failmask |= (1<<level);
271	–	}
272	–
273	–	level++;
274	–	if(mu->nValidHits       < 1)   {
275	–	failmask |= (1<<level);
276	–	}
277	–
278	–	level++;
279	–	if(mu->nMatch   < 2)   {
280	–	failmask |= (1<<level);
281	–	}
282	–
386		SelectionStatus status;
387	<
388	<	if( !failmask ) status.setStatus(SelectionStatus::LOOSEID);
389	<	if( !failmask ) status.setStatus(SelectionStatus::TIGHTID);
390	<
288	<	return status;
387	>	if(passID)  status |= SelectionStatus::TIGHTIDANDPRE;
388	>	if(passIso) status |= SelectionStatus::TIGHTISOANDPRE;
389	>
390	>	return status;
391
392	<	};
392	>	}

Diff Legend

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