ViewVC Help

View File | Revision Log | Show Annotations | Root Listing

root/cvsroot/UserCode/MitHzz4l/LeptonSelection/interface/IsolationSelection.h

Comparing UserCode/MitHzz4l/LeptonSelection/interface/IsolationSelection.h (file contents):
Revision 1.2 by khahn, Fri Feb 17 14:48:59 2012 UTC vs.
Revision 1.15 by khahn, Sun Jun 3 15:51:56 2012 UTC

#	Line 1 | Line 1
1	<	#include "TElectron.hh"
2	<	#include "TMuon.hh"
1	>	#include "Array.h"
2	>	#include "Electron.h"
3	>	#include "Muon.h"
4	>	#include "Vertex.h"
5	>	#include "PFCandidate.h"
6	>	#include "PFCandidateCol.h"
7	>	#include "PileupEnergyDensity.h"
8	>
9	>	#include "TLorentzVector.h"
10	>
11	>	#include "MuonTools.h"
12	>	#include "ElectronTools.h"
13
14		#include "ParseArgs.h"
15		#include "SelectionStatus.h"
16		#include "SimpleLepton.h"
17
18	<	SelectionStatus failEleIso(ControlFlags &, const mithep::TElectron *  );
19	<	SelectionStatus passMuonIsoSelection( ControlFlags &, const mithep::TMuon *  );
18	>	#define MUON_ISOMVA_CUT_BIN0  -0.615
19	>	#define MUON_ISOMVA_CUT_BIN1   0.235
20	>	#define MUON_ISOMVA_CUT_BIN2  -0.825
21	>	#define MUON_ISOMVA_CUT_BIN3   0.155
22	>	#define MUON_ISOMVA_CUT_BIN4  -0.965
23	>	#define MUON_ISOMVA_CUT_BIN5  -0.989
24	>
25	>	#define MUON_ISOMVA_FORPFID_CUT_BIN0  -0.5934
26	>	#define MUON_ISOMVA_FORPFID_CUT_BIN1  0.3374
27	>	#define MUON_ISOMVA_FORPFID_CUT_BIN2  -0.7666
28	>	#define MUON_ISOMVA_FORPFID_CUT_BIN3  0.4102
29	>	#define MUON_ISOMVA_FORPFID_CUT_BIN4  -0.989
30	>	#define MUON_ISOMVA_FORPFID_CUT_BIN5  -0.9946
31	>
32	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_BIN0  -0.5934
33	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_BIN1  0.3374
34	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_BIN2  -0.7666
35	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_BIN3  0.4102
36	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_BIN4  -0.989
37	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_BIN5  -0.9946
38	>
39	>	#define MUON_ISOMVA_TIGHT_FORPFID_CUT_BIN0  -0.5934
40	>	#define MUON_ISOMVA_TIGHT_FORPFID_CUT_BIN1  0.3374
41	>	#define MUON_ISOMVA_TIGHT_FORPFID_CUT_BIN2  -0.7666
42	>	#define MUON_ISOMVA_TIGHT_FORPFID_CUT_BIN3  0.4102
43	>	#define MUON_ISOMVA_TIGHT_FORPFID_CUT_BIN4  -0.989
44	>	#define MUON_ISOMVA_TIGHT_FORPFID_CUT_BIN5  -0.9946
45	>
46	>	#define ELECTRON_LOOSE_ISOMVA_CUT_BIN0  0.385
47	>	#define ELECTRON_LOOSE_ISOMVA_CUT_BIN1  -0.083
48	>	#define ELECTRON_LOOSE_ISOMVA_CUT_BIN2  -0.573
49	>	#define ELECTRON_LOOSE_ISOMVA_CUT_BIN3  0.413
50	>	#define ELECTRON_LOOSE_ISOMVA_CUT_BIN4  0.271
51	>	#define ELECTRON_LOOSE_ISOMVA_CUT_BIN5  0.135
52	>
53	>	#define ELECTRON_TIGHT_ISOMVA_CUT_BIN0  0.533
54	>	#define ELECTRON_TIGHT_ISOMVA_CUT_BIN1  -0.237
55	>	#define ELECTRON_TIGHT_ISOMVA_CUT_BIN2  -0.573
56	>	#define ELECTRON_TIGHT_ISOMVA_CUT_BIN3  0.521
57	>	#define ELECTRON_TIGHT_ISOMVA_CUT_BIN4  0.531
58	>	#define ELECTRON_TIGHT_ISOMVA_CUT_BIN5  0.493
59	>
60	>
61	>	//
62	>	// optimization test
63	>	//
64	>	#define ELECTRON_LOOSE_ISOMVA_CUT_OPT_BIN0  0.385
65	>	#define ELECTRON_LOOSE_ISOMVA_CUT_OPT_BIN1  -0.083
66	>	#define ELECTRON_LOOSE_ISOMVA_CUT_OPT_BIN2  -0.573
67	>	#define ELECTRON_LOOSE_ISOMVA_CUT_OPT_BIN3  0.413
68	>	#define ELECTRON_LOOSE_ISOMVA_CUT_OPT_BIN4  0.271
69	>	#define ELECTRON_LOOSE_ISOMVA_CUT_OPT_BIN5  0.135
70	>
71	>	#ifdef   MUONISO_WP0
72	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN0  0.03735
73	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN1  0.86195
74	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN2  -0.15025
75	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN3  0.87945
76	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN4  -0.88455
77	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN5  -0.90835
78	>	#elif MUONISO_WP1
79	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN0  -0.19505
80	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN1  0.77445
81	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN2  -0.46945
82	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN3  0.64635
83	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN4  -0.93635
84	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN5  -0.96505
85	>	#elif MUONISO_WP2
86	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN0  -0.33995
87	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN1  0.60785
88	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN2  -0.52125
89	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN3  0.54135
90	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN4  -0.97765
91	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN5  -0.97485
92	>	#elif MUONISO_WP3
93	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN0  -0.42045
94	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN1  0.37195
95	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN2  -0.59335
96	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN3  0.41045
97	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN4  -0.98465
98	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN5  -0.98185
99	>	#elif MUONISO_WP4
100	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN0  -0.48065
101	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN1  0.26765
102	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN2  -0.64095
103	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN3  0.14865
104	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN4  -0.98675
105	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN5  -0.98815
106	>	#elif MUONISO_WP5
107	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN0  -0.52195
108	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN1  0.12905
109	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN2  -0.68715
110	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN3  0.00305
111	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN4  -0.99165
112	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN5  -0.99235
113	>	#elif MUONISO_WP6
114	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN0  -0.60315
115	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN1  -0.13415
116	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN2  -0.75855
117	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN3  -0.18805
118	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN4  -0.99235
119	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN5  -0.99445
120	>	#elif MUONISO_WP7
121	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN0  -0.65565
122	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN1  -0.34345
123	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN2  -0.79985
124	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN3  -0.31475
125	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN4  -0.99445
126	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN5  -0.99655
127	>	#elif MUONISO_WP8
128	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN0  -0.70535
129	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN1  -0.39945
130	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN2  -0.82645
131	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN3  -0.64865
132	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN4  -0.99515
133	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN5  -0.99795
134	>	#elif MUONISO_WP9
135	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN0  -0.79145
136	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN1  -0.74245
137	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN2  -0.89085
138	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN3  -0.78585
139	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN4  -0.99725
140	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN5  -0.99865
141	>	#else
142	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN0  -0.5934
143	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN1  0.3374
144	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN2  -0.7666
145	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN3  0.4102
146	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN4  -0.989
147	>	#define MUON_ISOMVA_LOOSE_FORPFID_CUT_OPT_BIN5  -0.9946
148	>	#endif
149	>
150	>	/* #define ELECTRON_REFERENCE_PFISO_CUT  0.25 */
151	>	/* #define MUON_REFERENCE_PFISO_CUT      0.25 */
152	>	#define ELECTRON_REFERENCE_PFISO_CUT  0.4
153	>	#define MUON_REFERENCE_PFISO_CUT      0.4
154	>
155	>	Float_t computePFMuonIso(const mithep::Muon *muon,
156	>	const mithep::Vertex * vtx,
157	>	const mithep::Array<mithep::PFCandidate> * fPFCandidates,
158	>	const Double_t dRMax);
159	>
160	>	Float_t computePFEleIso(const mithep::Electron *electron,
161	>	const mithep::Vertex * fVertex,
162	>	const mithep::Array<mithep::PFCandidate> * fPFCandidates,
163	>	const Double_t dRMax);
164	>
165	>	SelectionStatus electronIsoSelection(ControlFlags &, const mithep::Electron *, const mithep::Vertex &, const mithep::Array<mithep::PFCandidate> * fPFCandidateCol   );
166	>	SelectionStatus muonIsoSelection(ControlFlags &, const mithep::Muon *, const mithep::Vertex *, const mithep::Array<mithep::PFCandidate> * fPFCandidateCol   );
167		bool pairwiseIsoSelection( ControlFlags &, vector<SimpleLepton> &, float rho );
168	+	bool noIso(ControlFlags &, vector<SimpleLepton> &, float rho);
169	+
170	+	void initMuonIsoMVA();
171	+	SelectionStatus muonIsoMVASelection(ControlFlags &ctrl,
172	+	const mithep::Muon * mu,
173	+	const mithep::Vertex * vtx,
174	+	const mithep::Array<mithep::PFCandidate> * fPFCandidates,
175	+	const mithep::Array<mithep::PileupEnergyDensity> * fPUEnergyDensity,
176	+	mithep::MuonTools::EMuonEffectiveAreaTarget EffectiveAreaVersion,
177	+	vector<const mithep::Muon*> muonsToVeto,
178	+	vector<const mithep::Electron*> electronsToVeto);
179	+	//hacked version
180	+	SelectionStatus muonIsoMVASelection(ControlFlags &ctrl,
181	+	const mithep::Muon * mu,
182	+	const mithep::Vertex * vtx,
183	+	const mithep::Array<mithep::PFCandidate> * fPFCandidates,
184	+	float rho,
185	+	mithep::MuonTools::EMuonEffectiveAreaTarget EffectiveAreaVersion,
186	+	vector<const mithep::Muon*> muonsToVeto,
187	+	vector<const mithep::Electron*> electronsToVeto);
188	+	double muonPFIso04(ControlFlags &ctrl,
189	+	const mithep::Muon * mu,
190	+	const mithep::Vertex * vtx,
191	+	const mithep::Array<mithep::PFCandidate> * fPFCandidates,
192	+	const mithep::Array<mithep::PileupEnergyDensity> * fPUEnergyDensity,
193	+	mithep::MuonTools::EMuonEffectiveAreaTarget EffectiveAreaVersion,
194	+	vector<const mithep::Muon*> muonsToVeto,
195	+	vector<const mithep::Electron*> electronsToVeto);
196	+	// hacked version
197	+	double muonPFIso04(ControlFlags &ctrl,
198	+	const mithep::Muon * mu,
199	+	const mithep::Vertex * vtx,
200	+	const mithep::Array<mithep::PFCandidate> * fPFCandidates,
201	+	float rho,  // the hack
202	+	mithep::MuonTools::EMuonEffectiveAreaTarget EffectiveAreaVersion,
203	+	vector<const mithep::Muon*> muonsToVeto,
204	+	vector<const mithep::Electron*> electronsToVeto);
205	+	//
206	+	SelectionStatus muonReferenceIsoSelection(ControlFlags &ctrl,
207	+	const mithep::Muon * mu,
208	+	const mithep::Vertex * vtx,
209	+	const mithep::Array<mithep::PFCandidate> * fPFCandidates,
210	+	const mithep::Array<mithep::PileupEnergyDensity> * fPUEnergyDensity,
211	+	mithep::MuonTools::EMuonEffectiveAreaTarget EffectiveAreaVersion,
212	+	vector<const mithep::Muon*> muonsToVeto,
213	+	vector<const mithep::Electron*> electronsToVeto);
214	+	// hacked version
215	+	SelectionStatus muonReferenceIsoSelection(ControlFlags &ctrl,
216	+	const mithep::Muon * mu,
217	+	const mithep::Vertex * vtx,
218	+	const mithep::Array<mithep::PFCandidate> * fPFCandidates,
219	+	float,
220	+	mithep::MuonTools::EMuonEffectiveAreaTarget EffectiveAreaVersion,
221	+	vector<const mithep::Muon*> muonsToVeto,
222	+	vector<const mithep::Electron*> electronsToVeto);
223	+
224	+	void initElectronIsoMVA();
225	+	SelectionStatus electronIsoMVASelection(ControlFlags &ctrl,
226	+	const mithep::Electron * ele,
227	+	const mithep::Vertex * vtx,
228	+	const mithep::Array<mithep::PFCandidate> * fPFCandidates,
229	+	const mithep::Array<mithep::PileupEnergyDensity> * fPUEnergyDensity,
230	+	mithep::ElectronTools::EElectronEffectiveAreaTarget EffectiveAreaVersion,
231	+	vector<const mithep::Muon*> muonsToVeto,
232	+	vector<const mithep::Electron*> electronsToVeto);
233	+	//hacked version
234	+	SelectionStatus electronIsoMVASelection(ControlFlags &ctrl,
235	+	const mithep::Electron * ele,
236	+	const mithep::Vertex * vtx,
237	+	const mithep::Array<mithep::PFCandidate> * fPFCandidates,
238	+	float rho,
239	+	mithep::ElectronTools::EElectronEffectiveAreaTarget EffectiveAreaVersion,
240	+	vector<const mithep::Muon*> muonsToVeto,
241	+	vector<const mithep::Electron*> electronsToVeto);
242	+	float electronPFIso04(ControlFlags &ctrl,
243	+	const mithep::Electron * ele,
244	+	const mithep::Vertex & vtx,
245	+	const mithep::Array<mithep::PFCandidate> * fPFCandidates,
246	+	const mithep::Array<mithep::PileupEnergyDensity> * fPUEnergyDensity,
247	+	mithep::ElectronTools::EElectronEffectiveAreaTarget EffectiveAreaVersion,
248	+	vector<const mithep::Muon*> muonsToVeto,
249	+	vector<const mithep::Electron*> electronsToVeto) ;
250	+	// hacked version
251	+	float electronPFIso04(ControlFlags &ctrl,
252	+	const mithep::Electron * ele,
253	+	const mithep::Vertex * vtx,
254	+	const mithep::Array<mithep::PFCandidate> * fPFCandidates,
255	+	float rho, // the hack
256	+	mithep::ElectronTools::EElectronEffectiveAreaTarget EffectiveAreaVersion,
257	+	vector<const mithep::Muon*> muonsToVeto,
258	+	vector<const mithep::Electron*> electronsToVeto) ;
259	+	//
260	+	SelectionStatus electronReferenceIsoSelection(ControlFlags &ctrl,
261	+	const mithep::Electron * ele,
262	+	const mithep::Vertex * vtx,
263	+	const mithep::Array<mithep::PFCandidate> * fPFCandidates,
264	+	const mithep::Array<mithep::PileupEnergyDensity> * fPUEnergyDensity,
265	+	mithep::ElectronTools::EElectronEffectiveAreaTarget EffectiveAreaVersion,
266	+	vector<const mithep::Muon*> muonsToVeto,
267	+	vector<const mithep::Electron*> electronsToVeto);
268	+	// hacked version
269	+	SelectionStatus electronReferenceIsoSelection(ControlFlags &ctrl,
270	+	const mithep::Electron * ele,
271	+	const mithep::Vertex * vtx,
272	+	const mithep::Array<mithep::PFCandidate> * fPFCandidates,
273	+	float,
274	+	mithep::ElectronTools::EElectronEffectiveAreaTarget EffectiveAreaVersion,
275	+	vector<const mithep::Muon*> muonsToVeto,
276	+	vector<const mithep::Electron*> electronsToVeto);
277	+
278	+	double  dbetaCorrectedIsoDr03(ControlFlags & ctrl,
279	+	const mithep::PFCandidate * photon,
280	+	const mithep::Muon * mu, // need to veto
281	+	const mithep::Array<mithep::PFCandidate> * fPFCandidates);
282	+
283	+	double  dbetaCorrectedIsoDr03(ControlFlags & ctrl,
284	+	const mithep::PFCandidate * photon,
285	+	const mithep::Electron * el, // need to veto
286	+	const mithep::Array<mithep::PFCandidate> * fPFCandidates);
287	+
288	+	double  betaCorrectedIsoDr03(ControlFlags & ctrl,
289	+	const mithep::PFCandidate * photon,
290	+	const mithep::Muon * mu, // need to veto
291	+	const mithep::Array<mithep::PFCandidate> * fPFCandidates);
292	+
293	+	double  betaCorrectedIsoDr03(ControlFlags & ctrl,
294	+	const mithep::PFCandidate * photon,
295	+	const mithep::Electron * el, // need to veto
296	+	const mithep::Array<mithep::PFCandidate> * fPFCandidates);

Diff Legend

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