ViewVC Help

View File | Revision Log | Show Annotations | Root Listing

root/cvsroot/UserCode/MitHzz4l/Selection/src/Selection.cc

Comparing UserCode/MitHzz4l/Selection/src/Selection.cc (file contents):
Revision 1.2 by dkralph, Wed Sep 14 12:11:57 2011 UTC vs.
Revision 1.21 by khahn, Mon Apr 30 21:42:16 2012 UTC

#	Line 1 | Line 1
1	+	#include "SelectionStatus.h"
2	+	#include "EventData.h"
3	+	#include "SimpleLepton.h"
4	+	#include "EfficiencyWeightsInterface.h"
5	+
6	+	#include "ElectronSelection.h"
7	+	#include "MuonSelection.h"
8	+	#include "IsolationSelection.h"
9	+	//#include "PassHLT.h"
10		#include "Selection.h"
2	–	#include "PassHLT.h"
3	–	#include "HZZCiCElectronSelection.h"
4	–	#include "RunLumiRangeMap.h"
5	–
6	–	RunLumiRangeMap rlrm;
7	–
8	–	void initRunLumiRangeMap() {
9	–	rlrm.AddJSONFile(std::string("./data/Cert_136033-149442_7TeV_Apr21ReReco_Collisions10_JSON.txt"));
10	–	rlrm.AddJSONFile(std::string("./data/Cert_160404-173244_7TeV_PromptReco_Collisions11_JSON_v2.txt"));
11	–	rlrm.AddJSONFile(std::string("./data/Cert_160404-163869_7TeV_May10ReReco_Collisions11_JSON_v3.txt"));
12	–	rlrm.AddJSONFile(std::string("./data/Cert_170249-172619_7TeV_ReReco5Aug_Collisions11_JSON.txt"));
13	–	};
14	–
15	–	unsigned fails_HZZ4L_selection(ControlFlags &ctrl,           // input control
16	–	mithep::TEventInfo *info,     // input event inof
17	–	TClonesArray *electronArr,    // input electrons
18	–	TClonesArray *muonArr,        // input muons
19	–	double eventweight,           // weight
20	–	TNtuple * passtuple ) {
21	–
22	–	fails_HZZ4L_selection( ctrl, info, electronArr, muonArr, eventweight, passtuple, NULL );
23	–
24	–	};
25	–
26	–	unsigned fails_HZZ4L_selection(ControlFlags &ctrl,           // input control
27	–	mithep::TEventInfo *info,     // input event inof
28	–	TClonesArray *electronArr,    // input electrons
29	–	TClonesArray *muonArr,        // input muons
30	–	double eventweight,           // weight
31	–	LabVectors  *l ) {
32	–
33	–	fails_HZZ4L_selection( ctrl, info, electronArr, muonArr, eventweight, NULL, l );
34	–
35	–	};
36	–
37	–
38	–	unsigned fails_HZZ4L_selection(ControlFlags &ctrl,           // input control
39	–	mithep::TEventInfo *info,     // input event inof
40	–	TClonesArray *electronArr,    // input electrons
41	–	TClonesArray *muonArr,        // input muons
42	–	double eventweight,           // weight
43	–	TNtuple * passtuple,
44	–	LabVectors * l ) {       // output ntuple
11
12	+	#include "ExternData.h"
13	+	#include "SelectionDefs.h"
14	+
15	+
16	+	// +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
17	+	// +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
18	+	// +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
19	+	EventData apply_HZZ4L_selection(ControlFlags &ctrl,           // input control
20	+	const mithep::EventHeader *info,     // input event info
21	+	const mithep::Vertex & vtx,
22	+	const mithep::Array<mithep::PFCandidate>  *pfCandidates,
23	+	const mithep::Array<mithep::PileupEnergyDensity>  *puEnergyDensity,
24	+	const mithep::Array<mithep::Electron> *electronArr,    // input electrons
25	+	SelectionStatus (*ElectronPreSelector)( ControlFlags &,
26	+	const mithep::Electron*,
27	+	const mithep::Vertex &),
28	+	SelectionStatus (*ElectronIDSelector)( ControlFlags &,
29	+	const mithep::Electron*,
30	+	const mithep::Vertex &),
31	+	SelectionStatus (*ElectronIsoSelector)( ControlFlags &,
32	+	const mithep::Electron*,
33	+	const mithep::Vertex &,
34	+	const mithep::Array<mithep::PFCandidate> *,
35	+	const mithep::Array<mithep::PileupEnergyDensity> *,
36	+	mithep::ElectronTools::EElectronEffectiveAreaTarget,
37	+	vector<const mithep::Muon*>,
38	+	vector<const mithep::Electron*> ),
39	+	const mithep::Array<mithep::Muon> *muonArr,    // input muons
40	+	SelectionStatus (*MuonPreSelector)( ControlFlags &,
41	+	const mithep::Muon*,
42	+	const mithep::Vertex &,
43	+	const mithep::Array<mithep::PFCandidate> *),
44	+	SelectionStatus (*MuonIDSelector)( ControlFlags &,
45	+	const mithep::Muon*,
46	+	const mithep::Vertex &),
47	+	SelectionStatus (*MuonIsoSelector)( ControlFlags &,
48	+	const mithep::Muon*,
49	+	const mithep::Vertex &,
50	+	const mithep::Array<mithep::PFCandidate> *,
51	+	const mithep::Array<mithep::PileupEnergyDensity> *,
52	+	mithep::MuonTools::EMuonEffectiveAreaTarget,
53	+	vector<const mithep::Muon*>,
54	+	vector<const mithep::Electron*> )
55	+	)
56	+	// +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
57	+	{
58	+
59	+	EventData ret;
60		unsigned evtfail = 0x0;
61	+	TRandom3 r;
62	+
63	+	if( ctrl.debug ) {
64	+	cout << "-----------------------------------------------------------------" << endl;
65	+	cout << "-----------------------------------------------------------------" << endl;
66	+	cout << "Run: " << info->RunNum()
67	+	<< "\tEvt: " << info->EvtNum()
68	+	<< "\tLumi: " << info->LumiSec()
69	+	<< endl;
70	+	cout << "-----------------------------------------------------------------" << endl;
71	+	}
72
73		if( !ctrl.mc ) {
74		// not accounting for overlap atm
75	<	RunLumiRangeMap::RunLumiPairType rl(info->runNum, info->lumiSec);
75	>	RunLumiRangeMap::RunLumiPairType rl(info->RunNum(), info->LumiSec());
76		if( !(rlrm.HasRunLumi(rl)) )  {
77	<	evtfail |= (1<<EVTFAIL_JSON);
78	<	return evtfail;
77	>	if( ctrl.debug ) cout << "\tfails JSON" << endl;
78	>	ret.status.setStatus(0);
79	>	return ret;
80		}
81		}
82	<
83	<
84	<	if( ctrl.debug ) {
85	<	cout << "Run: " << info->runNum
86	<	<< "\tEvt: " << info->evtNum
87	<	<< "\tLumi: " << info->lumiSec
62	<	<< endl;
82	>
83	>	mithep::MuonTools::EMuonEffectiveAreaTarget         eraMu  = mithep::MuonTools::kMuEAFall11MC;
84	>	mithep::ElectronTools::EElectronEffectiveAreaTarget eraEle = mithep::ElectronTools::kEleEAFall11MC;
85	>	if( !ctrl.mc ) {
86	>	eraMu  = mithep::MuonTools::kMuEAData2011;
87	>	eraEle = mithep::ElectronTools::kEleEAData2011;
88		}
89
90
91		//********************************************************
92		// Trigger
93		//********************************************************
94	+	//
95	+	// still have to port this part to bambu
96	+	//
97	+	/*
98		if( !ctrl.mc ) {
99	<	//  if( !(passHLT(info->triggerBits, info->runNum, channel) )  ) {
99	>	//if( !(passHLT(info->triggerBits, info->runNum, channel) )  ) {
100		if( !(passHLT(info->triggerBits, info->runNum, 999) )  ) {
101	+	if( ctrl.debug ) cout << "\tfails trigger" << endl;
102		evtfail |= (1<<EVTFAIL_TRIGGER);
103	<	return evtfail;
103	>	ret.status.setStatus(0);
104	>	return ret;
105		}
106	<	} else {
107	<	if( !(passHLTMC(info->triggerBits)) ) {
77	<	evtfail |= (1<<EVTFAIL_TRIGGER);
78	<	return evtfail;
79	<	}
80	<	//    cout << "MC trigger bits: " << hex << info->triggerBits << dec << endl;
81	<	}
106	>	}
107	>	*/
108
109		if( ctrl.debug ) {
110		cout << "presel nlep: " << muonArr->GetEntries() + electronArr->GetEntries()
#	Line 87 | Line 113 | unsigned fails_HZZ4L_selection(ControlFl
113		<< endl;
114		}
115
116	+
117		//********************************************************
118		// Lepton Selection
119		//********************************************************
120		vector<SimpleLepton> lepvec;
94	–
95	–	//
96	–	cout << "\tnMuons: " << muonArr->GetEntries() << endl;
97	–	//----------------------------------------------------
98	–	for(Int_t i=0; i<muonArr->GetEntries(); i++) {
99	–	const mithep::TMuon *mu = (mithep::TMuon*)((*muonArr)[i]);
100	–	unsigned muonfail = passMuonSelectionZZ(mu);
101	–	if( ctrl.debug ) {
102	–	cout << "muon:: pt: " << mu->pt
103	–	<< "\teta: " << mu->eta
104	–	<< "\tmask: 0x" << hex << muonfail << dec
105	–	<< endl;
106	–	}
107	–	if ( !muonfail ) {
108	–	SimpleLepton tmplep;
109	–	tmplep.vec.SetPtEtaPhiM(mu->pt,
110	–	mu->eta,
111	–	mu->phi,
112	–	105.658369e-3);
113	–	tmplep.type    = 13;
114	–	tmplep.index   = i;
115	–	tmplep.charge  = mu->q;
116	–	tmplep.isoTrk  = mu->trkIso03;
117	–	tmplep.isoEcal = mu->emIso03;
118	–	tmplep.isoHcal = mu->hadIso03;
119	–	tmplep.ip3dSig = mu->ip3dSig;
120	–	tmplep.is4l    = false;
121	–	tmplep.isEB    = (fabs(mu->eta) < 1.479 ? 1 : 0 );
122	–	lepvec.push_back(tmplep);
123	–	if( ctrl.debug ) { cout << "muon passes ... " << endl;}
124	–	}
125	–	}
126	–
127	–	//
128	–	if( ctrl.debug ) { cout << "\tnElectron: " << electronArr->GetEntries() << endl; }
129	–	//----------------------------------------------------
130	–	for(Int_t i=0; i<electronArr->GetEntries(); i++) {
131	–	const mithep::TElectron *ele = (mithep::TElectron*)((*electronArr)[i]);
121
122	<	Bool_t isMuonOverlap = kFALSE;
123	<	for (int k=0; k<lepvec.size(); ++k) {
124	<	TVector3 tmplep;
125	<	tmplep.SetPtEtaPhi(ele->pt, ele->eta, ele->phi);
126	<	if ( lepvec[k].type == 13 && lepvec[k].vec.Vect().DrEtaPhi(tmplep) < 0.1 ) {
127	<	cout << "-----> isMuonOverlap! " << endl;
128	<	isMuonOverlap = kTRUE;
129	<	break;
122	>	// do something hacky for vetos for now
123	>	vector<const mithep::Muon*> muonsToVeto;
124	>	vector<const mithep::Electron*> electronsToVeto;
125	>	if( ctrl.debug ) cout << "looping for isolation ..." << endl;
126	>	for(Int_t i=0; i<muonArr->GetEntries(); i++)
127	>	{
128	>	const mithep::Muon *mu = (mithep::Muon*)((*muonArr)[i]);
129	>	SelectionStatus musel;
130	>	musel |= (*MuonPreSelector)(ctrl,mu,vtx,pfCandidates);
131	>	if( !(musel.getStatus() & SelectionStatus::PRESELECTION) ) continue;
132	>	musel |= (*MuonIDSelector)(ctrl,mu,vtx );
133	>	if(ctrl.debug) cout << "status : " << musel.getStatus() << endl;
134	>	if( musel.tightIDAndPre() ) {
135	>	if(ctrl.debug) cout << "pushing mu for isol veto ... " << endl;
136	>	muonsToVeto.push_back( mu );
137		}
138		}
139	<
140	<	CICStruct tightcuts = getTightCuts();
141	<	unsigned failsCIC = failsCicSelection(ele, tightcuts);
142	<	if( ctrl.debug ){
143	<	cout << "CIC category: " << cicCategory(ele)
144	<	<< "\tfailmask : 0x" << hex << failsCIC  << dec
145	<	<< "\tscEt: " << ele->scEt
146	<	<< "\tscEta: " << ele->scEta
147	<	<< endl;
148	<	}
149	<	if ( !failsCIC && !isMuonOverlap ) {
150	<	SimpleLepton tmplep;
155	<	tmplep.vec.SetPtEtaPhiM( ele->pt,
156	<	ele->eta,
157	<	ele->phi,
158	<	0.51099892e-3 );
159	<	tmplep.type    = 11;
160	<	tmplep.index   = i;
161	<	tmplep.charge  = ele->q;
162	<	tmplep.isoTrk  = ele->trkIso03;
163	<	tmplep.isoEcal = ele->emIso03;
164	<	tmplep.isoHcal = ele->hadIso03;
165	<	tmplep.ip3dSig = ele->ip3dSig;
166	<	tmplep.is4l    = false;
167	<	tmplep.isEB    = ele->isEB;
168	<	lepvec.push_back(tmplep);
169	<	if( ctrl.debug ) { cout << "\telectron passes ... " << endl; }
139	>	for(Int_t i=0; i<electronArr->GetEntries(); i++)
140	>	{
141	>	const mithep::Electron *ele = (mithep::Electron*)((*electronArr)[i]);
142	>	SelectionStatus esel;
143	>	esel |= (*ElectronPreSelector)(ctrl,ele,vtx);
144	>	if( !(esel.getStatus() & SelectionStatus::PRESELECTION) ) continue;
145	>	esel |= (*ElectronIDSelector)(ctrl,ele,vtx );
146	>	if(ctrl.debug) cout << "status : " << esel.getStatus() << endl;
147	>	if( esel.tightIDAndPre() ) {
148	>	if(ctrl.debug) cout << "pushing ele for isol veto ... " << endl;
149	>	electronsToVeto.push_back( ele );
150	>	}
151		}
152	<	}
172	<
173	<	sort( lepvec.begin(), lepvec.end(), SimpleLepton::lep_pt_sort );
152	>	if( ctrl.debug ) cout << "done looping for isolation ..." << endl << endl;;
153
154	<	int nmu=0, nele=0;
155	<	for( int i=0; i<lepvec.size(); i++ ) {
156	<	if( abs(lepvec[i].type) == 11 ) nele++;
157	<	else nmu++;
158	<	}
159	<	if( ctrl.debug ) {
181	<	cout << "postsel nlep: " << lepvec.size()
182	<	<< "\tnmuon: " << nmu
183	<	<< "\tnelectron: " << nele
184	<	<< endl;
185	<	}
186	<
187	<	//******************************************************************************
188	<	//Z1 Selection
189	<	//******************************************************************************
190	<	int Z1LeptonPlusIndex = -1;
191	<	int Z1LeptonMinusIndex = -1;
192	<	double BestZ1Mass = -999;
193	<	if( ctrl.debug ) { cout << "looking for a Z1 ..." << endl; }
194	<	for(int i = 0; i < lepvec.size(); ++i) {
195	<	for(int j = i+1; j < lepvec.size(); ++j) {
196	<	if( ctrl.debug ) { cout << "\tconsidering leptons " << i << " & " << j << endl; }
197	<	if (!(lepvec[i].vec.Pt() > 20.0 || lepvec[j].vec.Pt() > 20.0)) continue;
198	<	if( ctrl.debug ) { cout << "\tat least one is > 20 GeV" << endl; }
199	<	if (!(lepvec[i].vec.Pt() > 10.0 && lepvec[j].vec.Pt() > 10.0)) continue;
200	<	if( ctrl.debug ) { cout << "\tthe other  is > 10 GeV" << endl; }
201	<	if (lepvec[i].charge == lepvec[j].charge) continue;
202	<	if( ctrl.debug ) { cout << "\tthey're opposite charge" << endl; }
203	<	if (fabs(lepvec[i].type) != fabs(lepvec[j].type)) continue;
204	<	if( ctrl.debug ) { cout << "\tthey're same flavor" << endl; }
154	>	//
155	>	if( ctrl.debug ) cout << "\tnMuons: " << muonArr->GetEntries() << endl;
156	>	//----------------------------------------------------
157	>	for(Int_t i=0; i<muonArr->GetEntries(); i++)
158	>	{
159	>	const mithep::Muon *mu = (mithep::Muon*)((*muonArr)[i]);
160
161	<	//Make Z1 hypothesis
162	<	TLorentzVector leptonPlus, leptonMinus;
163	<	if ( lepvec[i].charge > 0 ) {
164	<	leptonPlus  = lepvec[i].vec;
165	<	leptonMinus = lepvec[j].vec;
166	<	} else {
167	<	leptonPlus  = lepvec[j].vec;
168	<	leptonMinus = lepvec[i].vec;
161	>	SelectionStatus musel;
162	>	if(ctrl.debug) cout << "musel.status  before anything: " << musel.getStatus() << endl;
163	>	musel |= (*MuonPreSelector)(ctrl,mu,vtx,pfCandidates);
164	>	if(ctrl.debug) cout << "musel.status  after presel: " << musel.getStatus() << endl;
165	>	if( !(musel.getStatus() & SelectionStatus::PRESELECTION) ) continue;
166	>	musel |= (*MuonIDSelector)(ctrl,mu,vtx );
167	>	if(ctrl.debug) cout << "musel.status  after ID: " << musel.getStatus() << endl;
168	>	musel |= (*MuonIsoSelector)(ctrl,mu,vtx,pfCandidates,puEnergyDensity,eraMu,muonsToVeto,electronsToVeto);
169	>	if(ctrl.debug) cout << "musel.status  after iso: " << musel.getStatus() << endl;
170	>
171	>	if( ctrl.debug ) {
172	>	cout << "muon:: pt: " << mu->Pt()
173	>	<< "\teta: " << mu->Eta()
174	>	<< "\tstatus: " << hex << musel.getStatus() << dec
175	>	<< endl;
176		}
177
178	<	float tmpZ1Mass = (leptonPlus+leptonMinus).M();
179	<	cout << "Z1 selection, tmpZ1Mass: " << tmpZ1Mass << endl;
180	<	if( tmpZ1Mass > 60 ) {
181	<	if (fabs(tmpZ1Mass - 91.1876) < fabs(BestZ1Mass - 91.1876)) {
182	<	BestZ1Mass = tmpZ1Mass;
183	<	cout << "Z1 selection, new BestZ1Mass: " << BestZ1Mass
184	<	<< "\tdM: " << fabs(BestZ1Mass - 91.1876)
185	<	<< endl;
186	<	if (lepvec[i].charge > 0) {
187	<	Z1LeptonPlusIndex = i;
188	<	Z1LeptonMinusIndex = j;
189	<	} else {
190	<	Z1LeptonPlusIndex = j;
191	<	Z1LeptonMinusIndex = i;
230	<	}
178	>	if ( musel.pass() ) {
179	>
180	>	SimpleLepton tmplep;
181	>	float pt = mu->Pt();
182	>	tmplep.vecorig->SetPtEtaPhiM(pt,
183	>	mu->Eta(),
184	>	mu->Phi(),
185	>	MUON_MASS);
186	>
187	>	if( ctrl.do_escale_up ) {
188	>	pt=scale_smear_muon_Up(pt, 1,  r);
189	>	}
190	>	if( ctrl.do_escale_down ) {
191	>	pt=scale_smear_muon_Down(pt, 1,  r);
192		}
193	+
194	+	tmplep.vec->SetPtEtaPhiM(pt,
195	+	mu->Eta(),
196	+	mu->Phi(),
197	+	MUON_MASS);
198	+
199	+	tmplep.type    = 13;
200	+	tmplep.index   = i;
201	+	tmplep.charge  = mu->Charge();
202	+	tmplep.isoTrk  = mu->IsoR03SumPt();
203	+	tmplep.isoEcal = mu->IsoR03EmEt();
204	+	tmplep.isoHcal = mu->IsoR03HadEt();
205	+	tmplep.isoPF03 = computePFMuonIso(mu,vtx,pfCandidates,0.3);
206	+	tmplep.isoPF04 = computePFMuonIso(mu,vtx,pfCandidates,0.4);
207	+	tmplep.ip3dSig = mu->Ip3dPVSignificance();
208	+	tmplep.is4l    = false;
209	+	tmplep.isEB    = (fabs(mu->Eta()) < 1.479 ? 1 : 0 );
210	+	tmplep.isTight = musel.tight();
211	+	tmplep.isLoose = musel.loose();
212	+	lepvec.push_back(tmplep);
213	+	if( ctrl.debug ) { cout << "muon passes ... " << endl;}
214		}
215	<	}
216	<	}
235	<	// stop if no Z1 candidate is found
236	<	if( BestZ1Mass < 0 ) {
237	<	evtfail |= (1<<EVTFAIL_Z1);
238	<	return evtfail;
239	<	}
240	<	cout << "\tgot a Z1 ... run: " << info->runNum << "\tevt: " << info->evtNum  << endl;
241	<	cout << "\tZ1 plusindex: " << Z1LeptonPlusIndex << "\tminusindex: " << Z1LeptonMinusIndex << endl;
242	<	TLorentzVector Z1LeptonPlus  = lepvec[Z1LeptonPlusIndex].vec;
243	<	TLorentzVector Z1LeptonMinus = lepvec[Z1LeptonMinusIndex].vec;
244	<	TLorentzVector Z1Candidate   =  Z1LeptonPlus + Z1LeptonMinus;
245	<	if( l != NULL ) {
246	<	l->vecz1 = Z1Candidate;
247	<	l->vecl1p = Z1LeptonPlus;
248	<	l->vecl1m = Z1LeptonMinus;
249	<	}
215	>	//  }
216	>	}
217
218	<	//******************************************************************************
219	<	// Z1 + l
220	<	//******************************************************************************
221	<	if( lepvec.size() < 3 ) {
222	<	evtfail |= (1<<EVTFAIL_Z1_PLUSL);
223	<	return evtfail;
224	<	}
218	>
219	>
220	>	//
221	>	if( ctrl.debug ) { cout << "\tnElectron: " << electronArr->GetEntries() << endl; }
222	>	// --------------------------------------------------------------------------------
223	>	for(Int_t i=0; i<electronArr->GetEntries(); i++)
224	>	{
225	>	const mithep::Electron *ele = (mithep::Electron*)((*electronArr)[i]);
226
227	<	//******************************************************************************
228	<	// 4l/Z2 Selection
229	<	//******************************************************************************
230	<	Int_t Z2LeptonPlusIndex = -1;
231	<	Int_t Z2LeptonMinusIndex = -1;
232	<	Double_t BestZ2Mass = -1;
233	<	cout << "looking for a Z2 ... out of " << lepvec.size() << " leptons" <<endl;
234	<	for(int i = 0; i < lepvec.size(); ++i) {
235	<	for(int j = i+1; j < lepvec.size(); ++j) {
236	<	//            cout << "i: " << i << "\tj: " << j << endl;
237	<	if (i == Z1LeptonPlusIndex || i == Z1LeptonMinusIndex) {
238	<	//            cout << "\ti matches a Z1 index, skipping ..." << endl;
239	<	continue; //skip Z1 leptons
240	<	}
241	<	if (j == Z1LeptonPlusIndex || j == Z1LeptonMinusIndex) {
242	<	//            cout << "\tj matches a Z1 index, skipping ..." << endl;
243	<	continue; //skip Z1 leptons
227	>	Bool_t isMuonOverlap = kFALSE;
228	>	for (int k=0; k<lepvec.size(); ++k) {
229	>	TVector3 tmplep;
230	>	tmplep.SetPtEtaPhi(ele->Pt(), ele->Eta(), ele->Phi());
231	>	if ( lepvec[k].isLoose && lepvec[k].type == 13 && lepvec[k].vec->Vect().DrEtaPhi(tmplep) < 0.1 ) {
232	>	if( ctrl.debug ) cout << "-----> isMuonOverlap! " << endl;
233	>	isMuonOverlap = kTRUE;
234	>	break;
235	>	}
236	>	}
237	>
238	>	SelectionStatus elesel;
239	>	if( ctrl.debug ) cout << "--> status before anything: " << hex << elesel.getStatus() << dec << endl;
240	>	elesel |= (*ElectronPreSelector)(ctrl,ele,vtx);
241	>	if( ctrl.debug ) cout << "--> status after presel: " << hex << elesel.getStatus() << dec << endl;
242	>	elesel |= (*ElectronIDSelector)(ctrl,ele,vtx);
243	>	if( ctrl.debug ) cout << "--> status after ID: " << hex << elesel.getStatus() << dec << endl;
244	>	elesel |= (*ElectronIsoSelector)(ctrl,ele,vtx,pfCandidates,puEnergyDensity,eraEle,muonsToVeto,electronsToVeto);
245	>	if( ctrl.debug ) cout << "--> status after iso: " << hex << elesel.getStatus() << dec << endl;
246	>
247	>	if( ctrl.debug ){
248	>	cout << "\tscEt: " << ele->SCluster()->Et()
249	>	<< "\tscEta: " << ele->SCluster()->Eta()
250	>	<< "\tstatus: " << hex << elesel.getStatus() << dec
251	>	<< endl;
252	>	}
253	>
254	>	if ( elesel.pass() && !isMuonOverlap )
255	>	{
256	>	SimpleLepton tmplep;
257	>
258	>	float pt = ele->Pt();
259	>	tmplep.vecorig->SetPtEtaPhiM( pt,
260	>	ele->Eta(),
261	>	ele->Phi(),
262	>	ELECTRON_MASS );
263	>
264	>	if( ctrl.do_escale ) {
265	>	pt=scale_smear_electron(pt, ele->IsEB(), r);
266		}
267	<	if (lepvec[i].charge == lepvec[j].charge) {
268	<	//              cout << "\ti and j are same sign, skipping ..." << endl;
279	<	continue;         //require opp sign
267	>	if( ctrl.do_escale_up ) {
268	>	pt=scale_smear_electron_Up(pt, ele->IsEB(),  r);
269		}
270	<	if (fabs(lepvec[i].type) != fabs(lepvec[j].type)) {
271	<	//              cout << "\ti and j are not same flavor, skipping ..." << endl;
283	<	continue; //require same flavor
270	>	if( ctrl.do_escale_down ) {
271	>	pt=scale_smear_electron_Down(pt, ele->IsEB(),  r);
272		}
273	+
274
275	+	tmplep.vec->SetPtEtaPhiM( pt,
276	+	ele->Eta(),
277	+	ele->Phi(),
278	+	ELECTRON_MASS );
279
280	<	//Make Z2 hypothesis
281	<	TLorentzVector leptonPlus, leptonMinus;
282	<
283	<	if (lepvec[i].charge > 0 ) {
284	<	leptonPlus  = lepvec[i].vec;
285	<	leptonMinus = lepvec[j].vec;
286	<	} else {
287	<	leptonPlus  = lepvec[j].vec;
288	<	leptonMinus = lepvec[i].vec;
289	<	}
290	<
291	<	TLorentzVector dilepton = leptonPlus+leptonMinus;
292	<	TLorentzVector fourLepton = Z1Candidate + dilepton;
293	<
294	<	cout << "dilepton.M() : " << dilepton.M() << endl;
295	<	cout << "fourLepton.M() : " << fourLepton.M() << endl;
296	<
297	<	if (!(dilepton.M() > 12.0)) continue;
298	<	if (!(fourLepton.M() > 100.0)) continue;
299	<
300	<
301	<	//for 4e and 4mu, require at least 1 of the other opp sign lepton pairs have mass > 12
302	<	if (fabs(lepvec[i].type) == fabs(lepvec[Z1LeptonPlusIndex].type)) {
303	<	TLorentzVector pair1 = Z1LeptonPlus+leptonMinus;
304	<	TLorentzVector pair2 = Z1LeptonMinus+leptonPlus;
305	<	cout << "pair1: " << pair1.M() << "\tpair2: "<< pair2.M() << endl;
306	<	if (!(pair1.M() > 12 || pair2.M() > 12)) continue;
307	<	}
308	<
309	<	//Disambiguiation is done by choosing the pair with the largest ptMax and largest ptMin
310	<	if (Z2LeptonPlusIndex < 0) {
311	<	if (lepvec[i].charge > 0) {
312	<	Z2LeptonPlusIndex = i;
313	<	Z2LeptonMinusIndex = j;
314	<	} else {
315	<	Z2LeptonPlusIndex = j;
316	<	Z2LeptonMinusIndex = i;
317	<	}
318	<	} else {
319	<	Double_t BestPairPtMax = lepvec[Z2LeptonPlusIndex].vec.Pt();
320	<	Double_t BestPairPtMin = lepvec[Z2LeptonMinusIndex].vec.Pt();
321	<	if (lepvec[Z2LeptonMinusIndex].vec.Pt() > BestPairPtMax) {
322	<	BestPairPtMax = lepvec[Z2LeptonMinusIndex].vec.Pt();
323	<	BestPairPtMin = lepvec[Z2LeptonPlusIndex].vec.Pt();
324	<	}
325	<
326	<	Double_t CurrentPairPtMax = lepvec[i].vec.Pt();
327	<	Double_t CurrentPairPtMin = lepvec[j].vec.Pt();
328	<	if (lepvec[j].vec.Pt() > CurrentPairPtMax) {
329	<	CurrentPairPtMax = lepvec[j].vec.Pt();
330	<	CurrentPairPtMin = lepvec[i].vec.Pt();
331	<	}
332	<
333	<	if (CurrentPairPtMax > BestPairPtMax) {
334	<	if (lepvec[i].charge > 0) {
335	<	Z2LeptonPlusIndex = i;
336	<	Z2LeptonMinusIndex = j;
337	<	} else {
338	<	Z2LeptonPlusIndex = j;
339	<	Z2LeptonMinusIndex = i;
340	<	}
341	<	} else if (CurrentPairPtMax  == BestPairPtMax) {
342	<	if (CurrentPairPtMin > BestPairPtMin) {
343	<	if (lepvec[i].charge > 0) {
344	<	Z2LeptonPlusIndex = i;
345	<	Z2LeptonMinusIndex = j;
346	<	} else {
347	<	Z2LeptonPlusIndex = j;
348	<	Z2LeptonMinusIndex = i;
349	<	}
350	<	}
351	<	}
352	<	}
353	<	}
354	<	}
355	<
356	<	// stop if no Z2 candidate is found
357	<	if (Z2LeptonPlusIndex == -1) {
358	<	evtfail |= ( 1<<EVTFAIL_4L );
359	<	return evtfail;
360	<	//      h_evtfail->Fill( evtfail );
361	<	//      cout << "evtfail: " << hex << evtfail << dec << endl;
362	<	//      continue;
370	<	}
371	<	cout << "\tgot a Z2 ..." << endl;
372	<	cout << "\tZ2 plusindex: " << Z2LeptonPlusIndex << "\tminusindex: " << Z2LeptonMinusIndex << endl;
373	<	TLorentzVector  Z2LeptonPlus  = lepvec[Z2LeptonPlusIndex].vec;
374	<	TLorentzVector  Z2LeptonMinus = lepvec[Z2LeptonMinusIndex].vec;
375	<	TLorentzVector  Z2Candidate   = Z2LeptonPlus+Z2LeptonMinus;
376	<	TLorentzVector  ZZSystem      = Z1Candidate + Z2Candidate;
377	<	if( l != NULL ) {
378	<	l->vecz2 = Z2Candidate;
379	<	l->vecl2p = Z2LeptonPlus;
380	<	l->vecl2m = Z2LeptonMinus;
381	<	l->vec4l = ZZSystem;
382	<	}
383	<	lepvec[Z1LeptonPlusIndex].is4l = true;
384	<	lepvec[Z1LeptonMinusIndex].is4l = true;
385	<	lepvec[Z2LeptonPlusIndex].is4l = true;
386	<	lepvec[Z2LeptonMinusIndex].is4l = true;
387	<
388	<	//***************************************************************
389	<	// Isolation
390	<	//***************************************************************
391	<	bool failiso=false;
392	<
393	<	/*
394	<	int i,j;
395	<	i=Z1LeptonPlusIndex;
396	<	j=Z1LeptonMinusIndex;
397	<	float RIso1 = (lepvec[i].isoTrk+lepvec[i].isoEcal+lepvec[i].isoHcal)/lepvec[i].vec.Pt();
398	<	float RIso2 = (lepvec[j].isoTrk+lepvec[j].isoEcal+lepvec[j].isoHcal)/lepvec[j].vec.Pt();
399	<	float comboIso12 = RIso1 + RIso2;
400	<	i=Z2LeptonPlusIndex;
401	<	j=Z2LeptonMinusIndex;
402	<	float RIso3 = (lepvec[i].isoTrk+lepvec[i].isoEcal+lepvec[i].isoHcal)/lepvec[i].vec.Pt();
403	<	float RIso4 = (lepvec[j].isoTrk+lepvec[j].isoEcal+lepvec[j].isoHcal)/lepvec[j].vec.Pt();
404	<	float comboIso34 = RIso3 + RIso4;
405	<	if( comboIso12 > 0.35 || comboIso34 > 0.35 ) {
406	<	failiso = true;
407	<	}
408	<	*/
409	<
410	<	float rho = info->rho;
411	<	for( int i=0; i<lepvec.size(); i++ ) {
412	<	if( !(lepvec[i].is4l) ) continue;
413	<	float effArea_ecal_i, effArea_hcal_i;
414	<	if( lepvec[i].isEB ) {
415	<	if( lepvec[i].type == 11 ) {
416	<	effArea_ecal_i = 0.101;
417	<	effArea_hcal_i = 0.021;
418	<	} else {
419	<	effArea_ecal_i = 0.074;
420	<	effArea_hcal_i = 0.022;
421	<	}
280	>	tmplep.type    = 11;
281	>	tmplep.index   = i;
282	>	tmplep.charge  = ele->Charge();
283	>	tmplep.isoTrk  = ele->TrackIsolationDr03();
284	>	tmplep.isoEcal = ele->EcalRecHitIsoDr03();
285	>	tmplep.isoHcal = ele->HcalTowerSumEtDr03();
286	>	tmplep.isoPF03 = computePFEleIso(ele,vtx,pfCandidates,0.3);
287	>	tmplep.isoPF04 = computePFEleIso(ele,vtx,pfCandidates,0.4);
288	>	tmplep.ip3dSig = ele->Ip3dPVSignificance();
289	>	tmplep.is4l    = false;
290	>	tmplep.isEB    = ele->IsEB();
291	>	tmplep.scID    = ele->SCluster()->GetUniqueID();
292	>	tmplep.isTight = elesel.tight();
293	>	tmplep.isLoose = elesel.loose();
294	>	lepvec.push_back(tmplep);
295	>	if( ctrl.debug ) { cout << "\telectron passes ... " << endl; }
296	>	}
297	>	}
298	>
299	>
300	>	//********************************************************
301	>	// Dump Stuff
302	>	//********************************************************
303	>	sort( lepvec.begin(), lepvec.end(), SimpleLepton::lep_pt_sort );
304	>	int nmu=0, nele=0;
305	>	for( int i=0; i<lepvec.size(); i++ ) {
306	>	if(ctrl.debug) cout << "lepvec :: index: " << i
307	>	<< "\tpt: " << lepvec[i].vec->Pt()
308	>	<< "\ttype: " << lepvec[i].type
309	>	<< endl;
310	>	if( abs(lepvec[i].type) == 11 ) nele++;
311	>	else nmu++;
312	>	}
313	>	if( ctrl.debug ) {
314	>	cout << "postsel nlep: " << lepvec.size()
315	>	<< "\tnmuon: " << nmu
316	>	<< "\tnelectron: " << nele
317	>	<< endl;
318	>	}
319	>
320	>
321	>	//******************************************************************************
322	>	// Z1 Selection
323	>	//******************************************************************************
324	>	int Z1LeptonPlusIndex = -1;
325	>	int Z1LeptonMinusIndex = -1;
326	>	double BestZ1Mass = -999;
327	>	if( ctrl.debug ) { cout << "looking for a Z1 ..." << endl; }
328	>	for(int i = 0; i < lepvec.size(); ++i) {
329	>	if( !(lepvec[i].isLoose) ) continue;
330	>	for(int j = i+1; j < lepvec.size(); ++j) {
331	>	if( !(lepvec[j].isLoose) ) continue;
332	>	if( ctrl.debug ) { cout << "\tconsidering leptons " << i << " & " << j << endl; }
333	>	if (!(lepvec[i].vec->Pt() > 20.0 || lepvec[j].vec->Pt() > 20.0)) continue;
334	>	if( ctrl.debug ) { cout << "\tat least one is > 20 GeV" << endl; }
335	>	if (!(lepvec[i].vec->Pt() > 10.0 && lepvec[j].vec->Pt() > 10.0)) continue;
336	>	if( ctrl.debug ) { cout << "\tthe other  is > 10 GeV" << endl; }
337	>	if (lepvec[i].charge == lepvec[j].charge) continue;
338	>	if( ctrl.debug ) { cout << "\tthey're opposite charge" << endl; }
339	>	if (fabs(lepvec[i].type) != fabs(lepvec[j].type)) continue;
340	>	if( ctrl.debug ) { cout << "\tthey're same flavor" << endl; }
341	>
342	>	//Make Z1 hypothesis
343	>	TLorentzVector *leptonPlus, *leptonMinus;
344	>	if ( lepvec[i].charge > 0 ) {
345	>	leptonPlus  = lepvec[i].vec;
346	>	leptonMinus = lepvec[j].vec;
347	>	} else {
348	>	leptonPlus  = lepvec[j].vec;
349	>	leptonMinus = lepvec[i].vec;
350	>	}
351	>
352	>	float tmpZ1Mass = (*leptonPlus + *leptonMinus).M();
353	>	if( ctrl.debug ) cout << "Z1 selection, tmpZ1Mass: " << tmpZ1Mass << endl;
354	>	if( tmpZ1Mass > 50 ) {
355	>	if (fabs(tmpZ1Mass - Z_MASS) < fabs(BestZ1Mass - Z_MASS)) {
356	>	BestZ1Mass = tmpZ1Mass;
357	>	if( ctrl.debug ) cout << "Z1 selection, new BestZ1Mass: " << BestZ1Mass
358	>	<< "\tdM: " << fabs(BestZ1Mass - Z_MASS)
359	>	<< endl;
360	>	if (lepvec[i].charge > 0) {
361	>	Z1LeptonPlusIndex = i;
362	>	Z1LeptonMinusIndex = j;
363		} else {
364	<	if( lepvec[i].type == 11 ) {
365	<	effArea_ecal_i = 0.046;
425	<	effArea_hcal_i = 0.040;
426	<	} else {
427	<	effArea_ecal_i = 0.045;
428	<	effArea_hcal_i = 0.030;
429	<	}
430	<	}
431	<	float isoEcal_corr_i = lepvec[i].isoEcal - (effArea_ecal_i*rho);
432	<	float isoHcal_corr_i = lepvec[i].isoHcal - (effArea_hcal_i*rho);
433	<	for( int j=i+1; j<lepvec.size(); j++ ) {
434	<	if( !(lepvec[j].is4l) ) continue;
435	<	float effArea_ecal_j, effArea_hcal_j;
436	<	if( lepvec[j].isEB ) {
437	<	if( lepvec[j].type == 11 ) {
438	<	effArea_ecal_j = 0.101;
439	<	effArea_hcal_j = 0.021;
440	<	} else {
441	<	effArea_ecal_j = 0.074;
442	<	effArea_hcal_j = 0.022;
443	<	}
444	<	} else {
445	<	if( lepvec[j].type == 11 ) {
446	<	effArea_ecal_j = 0.046;
447	<	effArea_hcal_j = 0.040;
448	<	} else {
449	<	effArea_ecal_j = 0.045;
450	<	effArea_hcal_j = 0.030;
451	<	}
452	<	}
453	<	float isoEcal_corr_j = lepvec[j].isoEcal - (effArea_ecal_j*rho);
454	<	float isoHcal_corr_j = lepvec[j].isoHcal - (effArea_hcal_j*rho);
455	<	float RIso_i = (lepvec[i].isoTrk+isoEcal_corr_i+isoHcal_corr_i)/lepvec[i].vec.Pt();
456	<	float RIso_j = (lepvec[j].isoTrk+isoEcal_corr_j+isoHcal_corr_j)/lepvec[j].vec.Pt();
457	<	float comboIso = RIso_i + RIso_j;
458	<	if( info->evtNum == 1038911933 ) {
459	<	float tmpdR = lepvec[i].vec.DrEtaPhi(lepvec[j].vec);
460	<	cout << "i: " << i
461	<	<< "\tdR: " << tmpdR
462	<	<< "\trho: " << rho
463	<	<< "\tRIso_i: " << RIso_i
464	<	<< "\ttkrel: " << lepvec[i].isoTrk/lepvec[i].vec.Pt()
465	<	<< "\tecalrel: " << lepvec[i].isoEcal/lepvec[i].vec.Pt()
466	<	<< "\tecalrelcor: " << isoEcal_corr_i/lepvec[i].vec.Pt()
467	<	<< "\thcalrel: " << lepvec[i].isoHcal/lepvec[i].vec.Pt()
468	<	<< "\thcalrelcor: " << isoHcal_corr_i/lepvec[i].vec.Pt()
469	<	<< "\tpt_i: " << lepvec[i].vec.Pt()
470	<	<< "\tj: " << j
471	<	<< "\tRIso_j: " << RIso_j
472	<	<< "\ttkrel: "  << lepvec[j].isoTrk/lepvec[j].vec.Pt()
473	<	<< "\tecalrel: " << lepvec[j].isoEcal/lepvec[j].vec.Pt()
474	<	<< "\tecalrelcor: " << isoEcal_corr_j/lepvec[j].vec.Pt()
475	<	<< "\thcalrel: " << lepvec[j].isoHcal/lepvec[j].vec.Pt()
476	<	<< "\thcalrelcor: " << isoHcal_corr_j/lepvec[j].vec.Pt()
477	<	<< "\tpt_j: " << lepvec[j].vec.Pt()
478	<	<< "\tcombo: " << comboIso
479	<	<< endl;
480	<	cout.flush();
481	<	}
482	<	if( comboIso > 0.35 ) {
483	<	cout << "combo failing for indices: " << i << "," << j << endl;
484	<	failiso = true;
485	<	//            break;
486	<	}
364	>	Z1LeptonPlusIndex = j;
365	>	Z1LeptonMinusIndex = i;
366		}
488	–	}
489	–	if( failiso ) {
490	–	evtfail |= ( 1<<EVTFAIL_ISOLATION );
491	–	return evtfail;
492	–	//h_evtfail->Fill( evtfail, eventweight );
493	–	//      h_evtfail->Fill( evtfail );
494	–	//      cout << "evtfail: " << hex << evtfail << dec << endl;
495	–	//      continue;
496	–	}
497	–
498	–	//***************************************************************
499	–	// IP significance
500	–	//***************************************************************
501	–	bool failip = false;
502	–	for( int i=0; i<lepvec.size(); i++ ) {
503	–	if( !(lepvec[i].is4l) ) continue;
504	–	if( lepvec[i].ip3dSig > 4 ) {
505	–	failip=true;
506	–	break;
367		}
368		}
369	<	if( failip ) {
370	<	evtfail |= (1<<EVTFAIL_IP );
371	<	return evtfail;
372	<	//h_evtfail->Fill( evtfail, eventweight );
373	<	//      h_evtfail->Fill( evtfail );
374	<	//      cout << "evtfail: " << hex << evtfail << dec << endl;
375	<	//      continue;
376	<	}
377	<
378	<	//***************************************************************
379	<	// remaining kinematic cuts
380	<	//***************************************************************
381	<	if ( Z1Candidate.M() > 120 ||
382	<	Z2Candidate.M() < 20  ||
383	<	Z2Candidate.M() > 120 ||
384	<	!(lepvec[Z1LeptonPlusIndex].vec.Pt() > 20.0 || lepvec[Z1LeptonMinusIndex].vec.Pt() > 20.0) ||
385	<	!(lepvec[Z1LeptonPlusIndex].vec.Pt() > 10.0 && lepvec[Z1LeptonMinusIndex].vec.Pt() > 10.0)
386	<	) {
387	<	evtfail |= (1<<EVTFAIL_KINEMATICS );
388	<	return evtfail;
389	<	//h_evtfail->Fill( evtfail, eventweight );
390	<	//      h_evtfail->Fill( evtfail );
391	<	//      cout << "evtfail: " << hex << evtfail << dec << endl;
392	<	//      continue;
393	<	}
394	<
395	<	int channel;
396	<	if( lepvec[Z1LeptonMinusIndex].type == 11 && lepvec[Z2LeptonMinusIndex].type == 11 ) channel=0;
397	<	if( lepvec[Z1LeptonMinusIndex].type == 13 && lepvec[Z2LeptonMinusIndex].type == 13 ) channel=1;
398	<	if( (lepvec[Z1LeptonMinusIndex].type == 11 && lepvec[Z2LeptonMinusIndex].type == 13) ||
399	<	(lepvec[Z1LeptonMinusIndex].type == 13 && lepvec[Z2LeptonMinusIndex].type == 11)) channel=2;
369	>	}
370	>	}
371	>	// stop if no Z1 candidate is found
372	>	if( BestZ1Mass < 0 ) {
373	>	evtfail |= (1<<EVTFAIL_Z1);
374	>	//ret.status = evtfail;
375	>	ret.status.setStatus(0);
376	>	return ret;
377	>	}
378	>	if( ctrl.debug ) cout << "\tgot a Z1 ... run: " << info->RunNum() << "\tevt: " << info->EvtNum()  << endl;
379	>	if( ctrl.debug ) cout << "\tZ1 plusindex: " << Z1LeptonPlusIndex << "\tminusindex: " << Z1LeptonMinusIndex << endl;
380	>	TLorentzVector Z1LeptonPlus  = *(lepvec[Z1LeptonPlusIndex].vec);
381	>	TLorentzVector Z1LeptonMinus = *(lepvec[Z1LeptonMinusIndex].vec);
382	>	TLorentzVector Z1Candidate   =  Z1LeptonPlus + Z1LeptonMinus;
383	>
384	>
385	>	//******************************************************************************
386	>	// Z1 + l
387	>	//******************************************************************************
388	>	if( lepvec.size() < 3 ) {
389	>	evtfail |= (1<<EVTFAIL_Z1_PLUSL);
390	>	//ret.status = evtfail;
391	>	ret.status.setStatus(0);
392	>	return ret;
393	>	}
394	>
395	>	//******************************************************************************
396	>	// 4l/Z2 Selection
397	>	//******************************************************************************
398	>	Int_t Z2LeptonPlusIndex = -1;
399	>	Int_t Z2LeptonMinusIndex = -1;
400	>	Double_t BestZ2Mass = -1;
401	>	if( ctrl.debug ) cout << "looking for a Z2 ... out of " << lepvec.size() << " leptons" <<endl;
402	>	for(int i = 0; i < lepvec.size(); ++i) {
403	>
404	>	if( ctrl.debug)  cout << "i: " << i
405	>	<< "\tpt: " << lepvec[i].vec->Pt()
406	>	<< "\ttype: " << lepvec[i].type
407	>	<< endl;
408	>
409	>	if( ctrl.eleSeleScheme == "mediumloose" &&
410	>	!(lepvec[i].isTight) ) {
411	>	if( ctrl.debug)  cout << "it's not tight, skipping ... " << endl;
412	>	continue;
413	>	}
414	>
415	>	for(int j = i+1; j < lepvec.size(); ++j) {
416	>	if( ctrl.debug)  cout << "\t\tj: " << j
417	>	<< "\tpt: " << lepvec[j].vec->Pt()
418	>	<< "\ttype: " << lepvec[j].type
419	>	<< endl;
420
421	<
421	>	if( ctrl.eleSeleScheme == "mediumloose" &&
422	>	!(lepvec[j].isTight) ) {
423	>	if( ctrl.debug)  cout << "it's not tight, skipping ... " << endl;
424	>	continue;
425	>	}
426	>
427	>
428	>	if (i == Z1LeptonPlusIndex || i == Z1LeptonMinusIndex) {
429	>	if( ctrl.debug)  cout << "\ti matches a Z1 index, skipping ..." << endl;
430	>	continue; //skip Z1 leptons
431	>	}
432	>	if (j == Z1LeptonPlusIndex || j == Z1LeptonMinusIndex) {
433	>	if( ctrl.debug)  cout << "\tj matches a Z1 index, skipping ..." << endl;
434	>	continue; //skip Z1 leptons
435	>	}
436	>	if (lepvec[i].charge == lepvec[j].charge) {
437	>	if( ctrl.debug)  cout << "\ti and j are same sign, skipping ..." << endl;
438	>	continue;         //require opp sign
439	>	}
440	>	if (fabs(lepvec[i].type) != fabs(lepvec[j].type)) {
441	>	if( ctrl.debug) cout << "\ti and j are not same flavor, skipping ..." << endl;
442	>	continue; //require same flavor
443	>	}
444	>
445	>
446	>	//Make Z2 hypothesis
447	>	TLorentzVector *leptonPlus, *leptonMinus;
448	>
449	>	if (lepvec[i].charge > 0 ) {
450	>	leptonPlus  = lepvec[i].vec;
451	>	leptonMinus = lepvec[j].vec;
452	>	} else {
453	>	leptonPlus  = lepvec[j].vec;
454	>	leptonMinus = lepvec[i].vec;
455	>	}
456	>
457	>	TLorentzVector dilepton = *leptonPlus + *leptonMinus;
458	>	TLorentzVector fourLepton = Z1Candidate + dilepton;
459
460	<	if( passtuple != NULL ) {
461	<	passtuple->Fill( info->runNum,
462	<	info->evtNum,
463	<	info->lumiSec,
464	<	channel,
465	<	Z1Candidate.M(),
466	<	Z2Candidate.M(),
467	<	ZZSystem.M(),
468	<	ZZSystem.Pt(),
469	<	eventweight);
470	<	}
471	<
472	<	cout  << "run: " << info->runNum
473	<	<< "\tevt: " << info->evtNum
474	<	<< "\tZ1channel: " << lepvec[Z1LeptonMinusIndex].type
475	<	<< "\tZ2channel: " << lepvec[Z2LeptonMinusIndex].type
476	<	<< "\tmZ1: " << Z1Candidate.M()
477	<	<< "\tmZ2: " << Z2Candidate.M()
478	<	<< "\tm4l: " << ZZSystem.M()
479	<	<< "\tevtfail: " << hex << evtfail << dec
480	<	<< "\ttrigbits: " << hex << info->triggerBits << dec
481	<	//          << "\ttree: " << inputFiles[q][f]
482	<	<< endl;
460	>	if( ctrl.debug ) cout << "dilepton.M() : " << dilepton.M() << endl;
461	>	if( ctrl.debug ) cout << "fourLepton.M() : " << fourLepton.M() << endl;
462	>
463	>	if (!(dilepton.M() > 12.0)) continue;
464	>	if (!(fourLepton.M() > 100.0)) continue;
465	>
466	>	//for 4e and 4mu, require at least 1 of the other opp sign lepton pairs have mass > 12
467	>	if (fabs(lepvec[i].type) == fabs(lepvec[Z1LeptonPlusIndex].type)) {
468	>	TLorentzVector pair1 = Z1LeptonPlus + *leptonMinus;
469	>	TLorentzVector pair2 = Z1LeptonMinus + *leptonPlus;
470	>	if( ctrl.debug ) cout << "pair1: " << pair1.M() << "\tpair2: "<< pair2.M() << endl;
471	>	if (!(pair1.M() > 12 || pair2.M() > 12)) continue;
472	>	}
473	>
474	>
475	>	//Disambiguiation is done by choosing the pair with the largest ptMax and largest ptMin
476	>	if (Z2LeptonPlusIndex < 0) {
477	>	if (lepvec[i].charge > 0) {
478	>	Z2LeptonPlusIndex = i;
479	>	Z2LeptonMinusIndex = j;
480	>	} else {
481	>	Z2LeptonPlusIndex = j;
482	>	Z2LeptonMinusIndex = i;
483	>	}
484	>	} else {
485	>	Double_t BestPairPtMax = lepvec[Z2LeptonPlusIndex].vec->Pt();
486	>	Double_t BestPairPtMin = lepvec[Z2LeptonMinusIndex].vec->Pt();
487	>	if (lepvec[Z2LeptonMinusIndex].vec->Pt() > BestPairPtMax) {
488	>	BestPairPtMax = lepvec[Z2LeptonMinusIndex].vec->Pt();
489	>	BestPairPtMin = lepvec[Z2LeptonPlusIndex].vec->Pt();
490	>	}
491	>
492	>	Double_t CurrentPairPtMax = lepvec[i].vec->Pt();
493	>	Double_t CurrentPairPtMin = lepvec[j].vec->Pt();
494	>	if (lepvec[j].vec->Pt() > CurrentPairPtMax) {
495	>	CurrentPairPtMax = lepvec[j].vec->Pt();
496	>	CurrentPairPtMin = lepvec[i].vec->Pt();
497	>	}
498	>
499	>	if (CurrentPairPtMax > BestPairPtMax) {
500	>	if (lepvec[i].charge > 0) {
501	>	Z2LeptonPlusIndex = i;
502	>	Z2LeptonMinusIndex = j;
503	>	} else {
504	>	Z2LeptonPlusIndex = j;
505	>	Z2LeptonMinusIndex = i;
506	>	}
507	>	} else if (CurrentPairPtMax  == BestPairPtMax) {
508	>	if (CurrentPairPtMin > BestPairPtMin) {
509	>	if (lepvec[i].charge > 0) {
510	>	Z2LeptonPlusIndex = i;
511	>	Z2LeptonMinusIndex = j;
512	>	} else {
513	>	Z2LeptonPlusIndex = j;
514	>	Z2LeptonMinusIndex = i;
515	>	}
516	>	}
517	>	}
518	>	}
519	>	}
520	>	}
521	>
522	>	// stop if no Z2 candidate is found
523	>	if (Z2LeptonPlusIndex == -1) {
524	>	evtfail |= ( 1<<EVTFAIL_4L );
525	>	//      ret.status = evtfail;
526	>	ret.status.setStatus(0);
527	>	return ret;
528	>	}
529	>	if( ctrl.debug ) cout << "\tgot a Z2 ..." << endl;
530	>	if( ctrl.debug ) cout << "\tZ2 plusindex: " << Z2LeptonPlusIndex
531	>	<< "\tminusindex: " << Z2LeptonMinusIndex << endl;
532	>	TLorentzVector  Z2LeptonPlus  = *(lepvec[Z2LeptonPlusIndex].vec);
533	>	TLorentzVector  Z2LeptonMinus = *(lepvec[Z2LeptonMinusIndex].vec);
534	>	TLorentzVector  Z2Candidate   = Z2LeptonPlus+Z2LeptonMinus;
535	>	TLorentzVector  ZZSystem      = Z1Candidate + Z2Candidate;
536	>	lepvec[Z1LeptonPlusIndex].is4l = true;
537	>	lepvec[Z1LeptonMinusIndex].is4l = true;
538	>	lepvec[Z2LeptonPlusIndex].is4l = true;
539	>	lepvec[Z2LeptonMinusIndex].is4l = true;
540	>
541
542	<	return evtfail;
543	<
544	<	}
542	>
543	>
544	>	//***************************************************************
545	>	// remaining kinematic cuts
546	>	//***************************************************************
547	>	double Z2massCut=0;
548	>	if      ( ctrl.kinematics == "loose" ) Z2massCut = 12;
549	>	else if ( ctrl.kinematics == "tight" ) Z2massCut = 20;
550	>	else { cout <<  "error! kinematic tightness not defined!" << endl; assert(0); }
551	>
552	>	if ( Z1Candidate.M() > 120        ||
553	>	Z2Candidate.M() < Z2massCut  ||
554	>	Z2Candidate.M() > 120        ||
555	>	!(lepvec[Z1LeptonPlusIndex].vec->Pt() > 20.0 || lepvec[Z1LeptonMinusIndex].vec->Pt() > 20.0) ||
556	>	!(lepvec[Z1LeptonPlusIndex].vec->Pt() > 10.0 && lepvec[Z1LeptonMinusIndex].vec->Pt() > 10.0)
557	>	) {
558	>	evtfail |= (1<<EVTFAIL_KINEMATICS );
559	>	//      ret.status = evtfail;
560	>	ret.status.setStatus(0);
561	>	return ret;
562	>	}
563	>
564	>	unsigned channel;
565	>	if( lepvec[Z1LeptonMinusIndex].type == 11 && lepvec[Z2LeptonMinusIndex].type == 11 ) channel=0;
566	>	if( lepvec[Z1LeptonMinusIndex].type == 13 && lepvec[Z2LeptonMinusIndex].type == 13 ) channel=1;
567	>	if( (lepvec[Z1LeptonMinusIndex].type == 11 && lepvec[Z2LeptonMinusIndex].type == 13) ||
568	>	(lepvec[Z1LeptonMinusIndex].type == 13 && lepvec[Z2LeptonMinusIndex].type == 11)) channel=2;
569	>
570	>
571	>
572	>
573	>	if( ctrl.debug ) cout  << "run: " << info->RunNum()
574	>	<< "\tevt: " << info->EvtNum()
575	>	<< "\tZ1channel: " << lepvec[Z1LeptonMinusIndex].type
576	>	<< "\tZ2channel: " << lepvec[Z2LeptonMinusIndex].type
577	>	<< "\tmZ1: " << Z1Candidate.M()
578	>	<< "\tmZ2: " << Z2Candidate.M()
579	>	<< "\tm4l: " << ZZSystem.M()
580	>	<< "\tevtfail: " << hex << evtfail << dec
581	>	//                           << "\ttrigbits: " << hex << info->triggerBits << dec
582	>	//              << "\ttree: " << inputFiles[q][f]
583	>	<< endl;
584	>
585
586
587	+	//***************************************************************
588	+	// finish
589	+	//***************************************************************
590	+
591	+	if( !evtfail ) {
592	+	ret.status.setStatus(SelectionStatus::EVTPASS);
593	+	ret.Z1leptons.push_back(lepvec[Z1LeptonMinusIndex]);
594	+	ret.Z1leptons.push_back(lepvec[Z1LeptonPlusIndex]);
595	+	ret.Z2leptons.push_back(lepvec[Z2LeptonMinusIndex]);
596	+	ret.Z2leptons.push_back(lepvec[Z2LeptonPlusIndex]);
597	+	}
598
599	+	return ret;
600	+	}
601
602

Diff Legend

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