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.2 by khahn, Tue Feb 14 01:29:01 2012 UTC vs.
Revision 1.17 by khahn, Thu May 10 22:36:12 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	+	#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 Float_t computePFMuonIso(const mithep::Muon *mu,
17	+	const mithep::Vertex & vtx,
18	+	const mithep::Array<mithep::PFCandidate> * pfCandidates,
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	<	SelectionStatus muonPreSelection( ControlFlags &ctrl,  const mithep::TMuon * mu ) {
33	<	if(ctrl.debug) cout << "inside muonpresel ... " << endl;
34	<	bool ret = isMuFO(mu);
35	<	if(ctrl.debug) cout << "iS fo? ... " << ret << endl;
36	<	ret &= ( fabs(mu->ip3dSig) < 4 );
37	<	if(ctrl.debug) cout << "and pass IP (" << mu->ip3dSig << ") ? ... " << ret << endl;
38	<	ret &= ( mu->pt > 5 );
39	<	if(ctrl.debug) cout << "and >5 GeV ? ... " << ret << endl;
40	<	ret &= ( fabs(mu->eta) < 2.4 );
41	<	if(ctrl.debug) cout << "and < 2.4 eta ? ... " << ret << endl;
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);
24	–	if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
49		return status;
26	–	}
27	–
28	–	bool isMuFO( const mithep::TMuon * mu ) {
29	–	bool isgood=true;
30	–	float reliso = mu->pfIso03/mu->pt;
31	–
32	–	if( mu->pt < 5  )         isgood=false;
33	–	if ( fabs(mu->dz) > 0.2 ) isgood=false;   // needed to remove cosmics in HF sample
34	–
35	–	//
36	–
37	–	// HF seems to not want tkquality, SS does
38	–	if( mu->pt>20 ) {
39	–	if ( !((mu->typeBits & kGlobal) && mu->nSeg>0  ) ) isgood=false; //&& mu->nValidHits>0
40	–	} else {
41	–	if ( !((mu->typeBits & kGlobal) && mu->nSeg>0 ) //&& //&& mu->nValidHits>0
42	–	&& !( (mu->typeBits & kTracker) &&  (mu->qualityBits & kTMOneStationLoose)
43	–	)
44	–	)   isgood=false;
45	–	}
46	–
47	–
48	–	//
49	–	// for HF MC to agree w/ data, cannot put any Tk quality bits
50	–	// but to get to 5% closure in the SS test, we need at least Tk oneStationLoose
51	–
52	–	/*
53	–	&& ( (mu->qualityBits & kTMOneStationLoose) ||
54	–	(mu->qualityBits & kTMLastStationOptimizedBarrelLowPtLoose) ||
55	–	(mu->qualityBits & kTMLastStationOptimizedLowPtLoose)
56	–	)
57	–	*/
58	–
59	–
60	–	// comment for more stats for mu fake shape
61	–	if( mu->pt < 20 ) {
62	–	if( reliso > 3 ) isgood=false;
63	–	} else {
64	–	if( reliso > 5 ) isgood=false;
65	–	}
66	–
67	–	return isgood;
68	–	};
69	–
70	–
71	–	SelectionStatus passSoftMuonSelection( ControlFlags &ctrl, const mithep::TMuon * mu ) {
72	–
73	–	int level=0;
74	–	unsigned failmask=0x0;
75	–
76	–	if(mu->nTkHits < 11 )
77	–	failmask |= (1<<level);
78	–	level++;
79	–
80	–	if(fabs(mu->d0) > 0.2)
81	–	failmask |= (1<<level);
82	–	level++;
83	–
84	–	if(fabs(mu->dz) > 0.1)
85	–	failmask |= (1<<level);
86	–	level++;
87	–
88	–	if(!(mu->typeBits & kTracker))
89	–	failmask |= (1<<level);
90	–	level++;
91	–
92	–	if(!(mu->qualityBits & kTMLastStationAngTight))
93	–	failmask |= (1<<level);
94	–	level++;
95	–
96	–	Double_t iso = (mu->trkIso03 + mu->emIso03 + mu->hadIso03)/mu->pt;
97	–	if(mu->pt>20 && iso<0.1)
98	–	failmask |= (1<<level);
99	–
100	–	bool passtight  = !(failmask);
101	–	bool passloose  = !(failmask);
102	–	return SelectionStatus( passtight | passloose );
103	–
50		}
51
106	–
107	–	//unsigned passMuonSelectionZZ( const mithep::TMuon * mu ) {
108	–	SelectionStatus passMuonSelectionZZ( ControlFlags &ctrl, const mithep::TMuon * mu ) {
109	–	int level=0;
110	–	unsigned failmask=0x0;
52
53	<	if(mu->pt < 5) {
54	<	failmask |= (1<<level);
114	<	}
115	<
116	<	level++;
117	<	if(fabs(mu->eta) > 2.4) {
118	<	failmask |= (1<<level);
119	<	}
120	<
121	<	level++;
122	<	if(!(mu->typeBits & kGlobal)) {
123	<	failmask |= (1<<level);
124	<	}
53	>	//--------------------------------------------------------------------------------------------------
54	>	SelectionStatus noPreSelection( ControlFlags &ctrl,  const mithep::Muon * mu )
55
56	<	level++;
57	<	if(mu->nTkHits          < 10) {
58	<	failmask |= (1<<level);
59	<	}
56	>	//--------------------------------------------------------------------------------------------------
57	>	{
58	>	SelectionStatus status;
59	>	status.setStatus(SelectionStatus::PRESELECTION);
60	>	if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
61	>	return status;
62	>	}
63
64	+	//--------------------------------------------------------------------------------------------------
65	+	SelectionStatus muonPreSelection( ControlFlags &ctrl,
66	+	const mithep::Muon * mu,
67	+	const mithep::Vertex & vtx,
68	+	const mithep::Array<mithep::PFCandidate> * pfCandidates )
69	+	//--------------------------------------------------------------------------------------------------
70	+	{
71	+	bool ret = true;
72	+	if(ctrl.debug) cout << "inside muonpresel ... " << endl;
73	+	//  bool ret = isMuFO(mu,vtx,pfCandidates);
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 (" << 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
132	–	level++;
133	–	if( (mu->trkIso03/mu->pt) > 0.7 ) {
134	–	failmask |= (1<<level);
135	–	}
89
137	–	//  if(muon->nPixHits   < 1)     return false;
138	–	//  if(muon->muNchi2    > 10)    return false;
139	–	//  if(muon->nMatch     < 2)     return false;
140	–	//  if(muon->nValidHits         < 1)     return false;
141	–	//  if(muon->pterr/muon->pt > 0.1)   return false;
142	–	//  if(fabs(muon->dz)       > 0.1)   return false;
143	–
90		SelectionStatus status;
91	<	if( !failmask ) status.setStatus(SelectionStatus::LOOSEID);
92	<	if( !failmask ) status.setStatus(SelectionStatus::TIGHTID);
91	>	if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
92	>	if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
93		return status;
94	+	}
95
96
97
98	<	};
99	<
100	<
101	<
98	>	//--------------------------------------------------------------------------------------------------
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	>	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
156	–	//
157	–	// Kevin's WW selection
158	–	//
159	–	SelectionStatus passMuonSelection( ControlFlags &ctrl, const mithep::TMuon * mu ) {
160	–	int level=0;
161	–	unsigned failmask=0x0;
121
122	<	// 0x1
123	<	if(mu->pt < 5) {
124	<	failmask |= (1<<level);
125	<	}
126	<
168	<	// 0x2
169	<	level++;
170	<	if(fabs(mu->eta) > 2.4) {
171	<	failmask |= (1<<level);
172	<	}
122	>	SelectionStatus status;
123	>	if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
124	>	if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
125	>	return status;
126	>	}
127
174	–	// 0x4
175	–	level++;
176	–	if(mu->ptErr/mu->pt > 0.1)   {
177	–	failmask |= (1<<level);
178	–	}
128
180	–	// 0x8
181	–	level++;
182	–	if( fabs(mu->dz) > 0.1 )   {
183	–	failmask |= (1<<level);
184	–	}
129
130
131	<	Bool_t isGlobal  = (mu->typeBits & kGlobal) && (mu->muNchi2 < 10) && (mu->nMatch > 1) && (mu->nValidHits > 0);
132	<	Bool_t isTracker = (mu->typeBits & kTracker) && (mu->qualityBits & kTMLastStationTight);
131	>	//--------------------------------------------------------------------------------------------------
132	>	SelectionStatus muonIDMVASelection(ControlFlags &ctrl,
133	>	const mithep::Muon *mu,
134	>	const mithep::Vertex & vtx   )
135	>	//--------------------------------------------------------------------------------------------------
136	>	{
137	>	double global_rchi2 = (mu->IsGlobalMuon()) ? mu->GlobalTrk()->RChi2() : 0.;
138
139	<	// 0x10
140	<	level++;
141	<	if(!isGlobal && !isTracker) {
142	<	failmask |= (1<<level);
143	<	}
139	>	//
140	>	// WARNING !!! KH hacked out to sync
141	>	//
142	>	/*
143	>	double mvaval = muIDMVA->MVAValue_ID(mu->Pt(),
144	>	mu->Eta(),
145	>	mu->IsGlobalMuon(),
146	>	mu->IsTrackerMuon(),
147	>	mu->TrackerTrk()->RChi2(),
148	>	global_rchi2,
149	>	(Double_t)(mu->NValidHits()),
150	>	(Double_t)(mu->TrackerTrk()->NHits()),
151	>	(Double_t)(mu->TrackerTrk()->NPixelHits()),
152	>	(Double_t)(mu->NMatches()),
153	>	mu->TrkKink(),
154	>	muTools.GetSegmentCompatability(mu),
155	>	muTools.GetCaloCompatability(mu,true,true),
156	>	mu->HadEnergy(),
157	>	mu->EmEnergy(),
158	>	mu->HadS9Energy(),
159	>	mu->EmS9Energy(),
160	>	(Bool_t)(ctrl.debug) );
161	>	*/
162	>	double mvaval = 0.0;
163
164	<	// 0x20
165	<	level++;
166	<	if( mu->nTkHits < 10 ) {
167	<	failmask |= (1<<level);
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;
169	>	else if( mu->IsGlobalMuon() && mu->IsTrackerMuon()
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;
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() && 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 ) {
181	>	status.orStatus(SelectionStatus::LOOSEID);
182	>	status.orStatus(SelectionStatus::TIGHTID);
183		}
184
185	<	level++;
186	<	if(mu->nPixHits         < 1) {
204	<	failmask |= (1<<level);
205	<	}
185	>	return status;
186	>	}
187
188	+	//--------------------------------------------------------------------------------------------------
189	+	void initMuonIDMVA()
190	+	//--------------------------------------------------------------------------------------------------
191	+	{
192	+	muIDMVA = new mithep::MuonIDMVA();
193	+	vector<string> weightFiles;
194	+	weightFiles.push_back("./data/MuonIDMVAWeights/MuonIDMVA_BDTG_barrel_lowpt_V2.weights.xml");
195	+	weightFiles.push_back("./data/MuonIDMVAWeights/MuonIDMVA_BDTG_barrel_highpt_V2.weights.xml");
196	+	weightFiles.push_back("./data/MuonIDMVAWeights/MuonIDMVA_BDTG_endcap_lowpt_V2.weights.xml");
197	+	weightFiles.push_back("./data/MuonIDMVAWeights/MuonIDMVA_BDTG_endcap_highpt_V2.weights.xml");
198	+	weightFiles.push_back("./data/MuonIDMVAWeights/MuonIDMVA_BDTG_tracker_V2.weights.xml");
199	+	weightFiles.push_back("./data/MuonIDMVAWeights/MuonIDMVA_BDTG_global_V2.weights.xml");
200	+	muIDMVA->Initialize( "MuonIDMVA",
201	+	mithep::MuonIDMVA::kIDV0,
202	+	kTRUE, weightFiles);
203	+	}
204
205	<	level++;
206	<	if(fabs(mu->d0)>0.02)   {
207	<	failmask |= (1<<level);
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	<	if(mu->pt>20) {
229	<	if(fabs(mu->d0)>0.02)   {
230	<	failmask |= (1<<level);
231	<	}
232	<	} else {
219	<	if(fabs(mu->d0)>0.01)  {
220	<	failmask |= (1<<level);
221	<	}
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		}
223	–	*/
234
225	–	SelectionStatus status;
226	–	if( !failmask ) status.setStatus(SelectionStatus::LOOSEID);
227	–	if( !failmask ) status.setStatus(SelectionStatus::TIGHTID);
235		return status;
236	<
230	<
231	<	};
232	<
236	>	}

Diff Legend

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