ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitHzz4l/Selection/src/Selection.cc
(Generate patch)

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

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines