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.7 by khahn, Fri Oct 14 11:20:25 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"
11  
12 < #include "SiMVAElectronSelection.h"
12 > #include "ExternData.h"
13 > #include "SelectionDefs.h"
14  
6 #include "HZZCiCElectronSelection.h"
7 #include "HZZLikelihoodElectronSelection.h"
8 #include "HZZBDTElectronSelection.h"
9 #include "RunLumiRangeMap.h"
10
11 RunLumiRangeMap rlrm;
12
13 void initRunLumiRangeMap() {
14  rlrm.AddJSONFile(std::string("./data/Cert_136033-149442_7TeV_Apr21ReReco_Collisions10_JSON.txt"));
15  rlrm.AddJSONFile(std::string("./data/Cert_160404-173244_7TeV_PromptReco_Collisions11_JSON_v2.txt"));
16  rlrm.AddJSONFile(std::string("./data/Cert_160404-163869_7TeV_May10ReReco_Collisions11_JSON_v3.txt"));  
17  rlrm.AddJSONFile(std::string("./data/Cert_170249-172619_7TeV_ReReco5Aug_Collisions11_JSON.txt"));  
18 };
19
20 unsigned fails_HZZ4L_selection(ControlFlags &ctrl,           // input control
21                               mithep::TEventInfo *info,     // input event inof
22                               TClonesArray *electronArr,    // input electrons
23                               TClonesArray *muonArr,        // input muons
24                               double eventweight,           // weight
25                               TTree * passtuple ) {
26
27  fails_HZZ4L_selection( ctrl, info, electronArr, muonArr, eventweight, passtuple, NULL );
28
29 };
30
31 unsigned fails_HZZ4L_selection(ControlFlags &ctrl,           // input control
32                               mithep::TEventInfo *info,     // input event inof
33                               TClonesArray *electronArr,    // input electrons
34                               TClonesArray *muonArr,        // input muons
35                               double eventweight,           // weight
36                               LabVectors  *l ) {
37
38  fails_HZZ4L_selection( ctrl, info, electronArr, muonArr, eventweight, NULL, l );
39
40 };
41
42
43 unsigned fails_HZZ4L_selection(ControlFlags &ctrl,           // input control
44                               mithep::TEventInfo *info,     // input event inof
45                               TClonesArray *electronArr,    // input electrons
46                               TClonesArray *muonArr,        // input muons
47                               double eventweight,           // weight
48                               TTree * passtuple,
49                               LabVectors * l) {       // output ntuple
50
51  unsigned evtfail = 0x0;
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 << "-----------------------------------------------------------------" << 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        if( ctrl.debug ) cout << "\tfails JSON" << endl;
78 <      evtfail |= (1<<EVTFAIL_JSON);
79 <      return evtfail;
78 >      ret.status.setStatus(0);
79 >      return ret;
80      }
81    }
70  
71  
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)) ) {
86 <      evtfail |= (1<<EVTFAIL_TRIGGER);
87 <      return evtfail;
88 <    }      
89 <    //    cout << "MC trigger bits: " << hex << info->triggerBits << dec << endl;
90 <  }
106 >  }
107 >  */
108  
109    if( ctrl.debug ) {
110      cout << "presel nlep: " << muonArr->GetEntries() + electronArr->GetEntries()
# Line 96 | Line 113 | unsigned fails_HZZ4L_selection(ControlFl
113           << endl;
114    }
115  
116 +
117    //********************************************************
118    // Lepton Selection
119    //********************************************************
120    vector<SimpleLepton> lepvec;
103  
104  //    
105  if( ctrl.debug ) cout << "\tnMuons: " << muonArr->GetEntries() << endl;
106  //----------------------------------------------------
107  for(Int_t i=0; i<muonArr->GetEntries(); i++) {
108    const mithep::TMuon *mu = (mithep::TMuon*)((*muonArr)[i]);      
109    unsigned muonfail = passMuonSelectionZZ(mu);
110    if( ctrl.debug ) {
111      cout << "muon:: pt: " << mu->pt
112           << "\teta: " << mu->eta
113           << "\tmask: 0x" << hex << muonfail << dec
114           << endl;
115    }
116    if ( !muonfail ) {
117      SimpleLepton tmplep;
118      tmplep.vec.SetPtEtaPhiM(mu->pt,
119                              mu->eta,
120                              mu->phi,
121                              105.658369e-3);
122      tmplep.type    = 13;
123      tmplep.index   = i;
124      tmplep.charge  = mu->q;
125      tmplep.isoTrk  = mu->trkIso03;
126      tmplep.isoEcal = mu->emIso03;
127      tmplep.isoHcal = mu->hadIso03;
128      tmplep.ip3dSig = mu->ip3dSig;
129      tmplep.is4l    = false;
130      tmplep.isEB    = (fabs(mu->eta) < 1.479 ? 1 : 0 );
131      lepvec.push_back(tmplep);
132      if( ctrl.debug ) { cout << "muon passes ... " << endl;}
133    }
134  }
135  
136  if( ctrl.debug ) { cout << "\tnElectron: " << electronArr->GetEntries() << endl; }
121  
122 <  //----------------------------------------------------
123 <  for(Int_t i=0; i<electronArr->GetEntries(); i++) {
124 <    const mithep::TElectron *ele = (mithep::TElectron*)((*electronArr)[i]);
125 <
126 <    Bool_t isMuonOverlap = kFALSE;
127 <    for (int k=0; k<lepvec.size(); ++k) {
128 <      TVector3 tmplep;
129 <      tmplep.SetPtEtaPhi(ele->pt, ele->eta, ele->phi);
130 <      if ( lepvec[k].type == 13 && lepvec[k].vec.Vect().DrEtaPhi(tmplep) < 0.1 ) {
131 <        if( ctrl.debug ) cout << "-----> isMuonOverlap! " << endl;
132 <        isMuonOverlap = kTRUE;
133 <        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 <    unsigned FAIL=0;
141 <    CICStruct ciccuts = getCiCCuts(ctrl.eleSeleScheme);
142 <    unsigned  failsCIC=0;
143 <    if(ctrl.eleSele=="cic") {
144 <      failsCIC = failsCicSelection(ctrl, ele, ciccuts, ctrl.kinematics);
145 <      FAIL = failsCIC;
146 <    }
147 <    LikStruct likcuts;
148 <    unsigned failsLike=0;
149 <    if(ctrl.eleSele=="lik") {
150 <      likcuts = getLikCuts(ctrl.eleSeleScheme);
164 <      failsLike = failsLikelihoodSelection(ele, likcuts, ctrl.kinematics);
165 <      FAIL = failsLike;
166 <    }
167 <    unsigned  failsBDT=0;
168 <    if(ctrl.eleSele=="bdt") {
169 <      failsBDT = failsBDTSelection(ctrl,ele);
170 <      FAIL = failsBDT;
171 <    }
172 <    unsigned  failsSi=0;
173 <    if(ctrl.eleSele=="si") {
174 <      failsSi = failsSiMVAElectronSelection(ctrl, ele, 0.95, ctrl.kinematics);
175 <      FAIL = failsSi;
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 +  if( ctrl.debug ) cout << "done looping for isolation ..." << endl << endl;;
153  
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 +      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 +      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 +    }    
217 +  
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 +        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( ctrl.do_escale_up ) {
268 +              pt=scale_smear_electron_Up(pt, ele->IsEB(),  r);
269 +            }
270 +            if( ctrl.do_escale_down ) {
271 +              pt=scale_smear_electron_Down(pt, ele->IsEB(),  r);
272 +            }
273  
274 <    if( ctrl.debug ){
275 <      cout << "CIC category: " << cicCategory(ele)
276 <           << "\tlikelihood: " << ele->likelihood
277 <           << "\tFAIL:  0x"     << hex << FAIL      << dec
278 <           << "\tfailsCIC:  0x" << hex << failsCIC  << dec
279 <           << "\tfailsLike: 0x" << hex << failsLike << dec
280 <           << "\tfailsBDT:  0x" << hex << failsBDT << dec
281 <           << "\tscEt: " << ele->scEt
282 <           << "\tscEta: " << ele->scEta
283 <           << "\tncluster: " << ele->ncluster
274 >            
275 >            tmplep.vec->SetPtEtaPhiM( pt,
276 >                                      ele->Eta(),
277 >                                      ele->Phi(),
278 >                                      ELECTRON_MASS );
279 >            
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 <    if ( !FAIL && !isMuonOverlap ) {
320 <      SimpleLepton tmplep;
321 <      tmplep.vec.SetPtEtaPhiM( ele->pt,
322 <                               ele->eta,
323 <                               ele->phi,
324 <                               0.51099892e-3 );
325 <      tmplep.type    = 11;
326 <      tmplep.index   = i;
327 <      tmplep.charge  = ele->q;
328 <      tmplep.isoTrk  = ele->trkIso03;
329 <      tmplep.isoEcal = ele->emIso03;
330 <      tmplep.isoHcal = ele->hadIso03;
331 <      tmplep.ip3dSig = ele->ip3dSig;
332 <      tmplep.is4l    = false;
333 <      tmplep.isEB    = ele->isEB;
334 <      lepvec.push_back(tmplep);
335 <      if( ctrl.debug ) { cout << "\telectron passes ... " << endl; }
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 >              Z1LeptonPlusIndex = j;
365 >              Z1LeptonMinusIndex = i;
366 >            }
367 >          }
368 >        }
369 >      }
370      }
371 <  }
372 <  
373 <  sort( lepvec.begin(), lepvec.end(), SimpleLepton::lep_pt_sort );
374 <
375 <  int nmu=0, nele=0;
376 <  for( int i=0; i<lepvec.size(); i++ ) {
377 <    if( abs(lepvec[i].type) == 11 ) nele++;
378 <    else nmu++;
379 <  }
380 <  if( ctrl.debug ) {
381 <    cout << "postsel nlep: " << lepvec.size()
382 <         << "\tnmuon: " << nmu
383 <         << "\tnelectron: " << nele
384 <         << endl;
385 <  }
386 <
387 <  //******************************************************************************
388 <  //Z1 Selection
389 <  //******************************************************************************
390 <  int Z1LeptonPlusIndex = -1;
391 <  int Z1LeptonMinusIndex = -1;
392 <  double BestZ1Mass = -999;
393 <  if( ctrl.debug ) { cout << "looking for a Z1 ..." << endl; }
394 <  for(int i = 0; i < lepvec.size(); ++i) {
395 <    for(int j = i+1; j < lepvec.size(); ++j) {
396 <      if( ctrl.debug ) { cout << "\tconsidering leptons " << i << " & " << j << endl; }
397 <      if (!(lepvec[i].vec.Pt() > 20.0 || lepvec[j].vec.Pt() > 20.0)) continue;
398 <      if( ctrl.debug ) { cout << "\tat least one is > 20 GeV" << endl; }
399 <      if (!(lepvec[i].vec.Pt() > 10.0 && lepvec[j].vec.Pt() > 10.0)) continue;
400 <      if( ctrl.debug ) { cout << "\tthe other  is > 10 GeV" << endl; }
401 <      if (lepvec[i].charge == lepvec[j].charge) continue;          
402 <      if( ctrl.debug ) { cout << "\tthey're opposite charge" << endl; }
243 <      if (fabs(lepvec[i].type) != fabs(lepvec[j].type)) continue;  
244 <      if( ctrl.debug ) { cout << "\tthey're same flavor" << endl; }
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 <      //Make Z1 hypothesis
405 <      TLorentzVector leptonPlus, leptonMinus;
406 <      if ( lepvec[i].charge > 0 ) {
407 <        leptonPlus  = lepvec[i].vec;
408 <        leptonMinus = lepvec[j].vec;
409 <      } else {
410 <        leptonPlus  = lepvec[j].vec;
411 <        leptonMinus = lepvec[i].vec;
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 <      float tmpZ1Mass = (leptonPlus+leptonMinus).M();
416 <      if( ctrl.debug ) cout << "Z1 selection, tmpZ1Mass: " << tmpZ1Mass << endl;
417 <      if( tmpZ1Mass > 60 ) {
418 <        if (fabs(tmpZ1Mass - 91.1876) < fabs(BestZ1Mass - 91.1876)) {
419 <          BestZ1Mass = tmpZ1Mass;
420 <          if( ctrl.debug ) cout << "Z1 selection, new BestZ1Mass: " << BestZ1Mass
421 <               << "\tdM: " << fabs(BestZ1Mass - 91.1876)
422 <               << endl;
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 >        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( 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 <            Z1LeptonPlusIndex = i;
479 <            Z1LeptonMinusIndex = j;
478 >            Z2LeptonPlusIndex = i;
479 >            Z2LeptonMinusIndex = j;
480            } else {
481 <            Z1LeptonPlusIndex = j;
482 <            Z1LeptonMinusIndex = i;
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            }
271        }
272      }
273    }
274  }
275  // stop if no Z1 candidate is found
276  if( BestZ1Mass < 0 ) {
277    evtfail |= (1<<EVTFAIL_Z1);
278    return evtfail;
279  }
280  if( ctrl.debug ) cout << "\tgot a Z1 ... run: " << info->runNum << "\tevt: " << info->evtNum  << endl;
281  if( ctrl.debug ) cout << "\tZ1 plusindex: " << Z1LeptonPlusIndex << "\tminusindex: " << Z1LeptonMinusIndex << endl;
282  TLorentzVector Z1LeptonPlus  = lepvec[Z1LeptonPlusIndex].vec;
283  TLorentzVector Z1LeptonMinus = lepvec[Z1LeptonMinusIndex].vec;
284  TLorentzVector Z1Candidate   =  Z1LeptonPlus + Z1LeptonMinus;
285  if( l != NULL ) {
286    l->vecz1 = Z1Candidate;
287    l->vecl1p = Z1LeptonPlus;
288    l->vecl1m = Z1LeptonMinus;
289  }
290  
291  //******************************************************************************
292  // Z1 + l
293  //******************************************************************************
294  if( lepvec.size() < 3 ) {
295    evtfail |= (1<<EVTFAIL_Z1_PLUSL);
296    return evtfail;
297  }
491          
492 <  //******************************************************************************
493 <  // 4l/Z2 Selection
494 <  //******************************************************************************
495 <  Int_t Z2LeptonPlusIndex = -1;
496 <  Int_t Z2LeptonMinusIndex = -1;
497 <  Double_t BestZ2Mass = -1;
498 <  if( ctrl.debug ) cout << "looking for a Z2 ... out of " << lepvec.size() << " leptons" <<endl;
499 <  for(int i = 0; i < lepvec.size(); ++i) {
500 <    for(int j = i+1; j < lepvec.size(); ++j) {
501 <      //            cout << "i: " << i << "\tj: " << j << endl;
502 <      if (i == Z1LeptonPlusIndex || i == Z1LeptonMinusIndex) {
310 <        //            cout << "\ti matches a Z1 index, skipping ..." << endl;
311 <        continue; //skip Z1 leptons
312 <      }
313 <      if (j == Z1LeptonPlusIndex || j == Z1LeptonMinusIndex) {
314 <        //            cout << "\tj matches a Z1 index, skipping ..." << endl;
315 <              continue; //skip Z1 leptons
316 <            }
317 <            if (lepvec[i].charge == lepvec[j].charge) {
318 <              //              cout << "\ti and j are same sign, skipping ..." << endl;
319 <              continue;         //require opp sign
320 <            }
321 <            if (fabs(lepvec[i].type) != fabs(lepvec[j].type)) {
322 <              //              cout << "\ti and j are not same flavor, skipping ..." << endl;
323 <              continue; //require same flavor
324 <            }
325 <            
326 <            
327 <            //Make Z2 hypothesis
328 <            TLorentzVector leptonPlus, leptonMinus;
329 <            
330 <            if (lepvec[i].charge > 0 ) {
331 <              leptonPlus  = lepvec[i].vec;
332 <              leptonMinus = lepvec[j].vec;
333 <            } else {
334 <              leptonPlus  = lepvec[j].vec;
335 <              leptonMinus = lepvec[i].vec;
336 <            }
337 <
338 <            TLorentzVector dilepton = leptonPlus+leptonMinus;
339 <            TLorentzVector fourLepton = Z1Candidate + dilepton;
340 <
341 <            if( ctrl.debug ) cout << "dilepton.M() : " << dilepton.M() << endl;
342 <            if( ctrl.debug ) cout << "fourLepton.M() : " << fourLepton.M() << endl;
343 <
344 <            if (!(dilepton.M() > 12.0)) continue;
345 <            if (!(fourLepton.M() > 100.0)) continue;
346 <
347 <            
348 <            //for 4e and 4mu, require at least 1 of the other opp sign lepton pairs have mass > 12
349 <            if (fabs(lepvec[i].type) == fabs(lepvec[Z1LeptonPlusIndex].type)) {
350 <              TLorentzVector pair1 = Z1LeptonPlus+leptonMinus;
351 <              TLorentzVector pair2 = Z1LeptonMinus+leptonPlus;
352 <              if( ctrl.debug ) cout << "pair1: " << pair1.M() << "\tpair2: "<< pair2.M() << endl;
353 <              if (!(pair1.M() > 12 || pair2.M() > 12)) continue;
354 <            }
355 <            
356 <            //Disambiguiation is done by choosing the pair with the largest ptMax and largest ptMin
357 <            if (Z2LeptonPlusIndex < 0) {
358 <              if (lepvec[i].charge > 0) {
359 <                Z2LeptonPlusIndex = i;
360 <                Z2LeptonMinusIndex = j;
361 <              } else {
362 <                Z2LeptonPlusIndex = j;
363 <                Z2LeptonMinusIndex = i;
364 <              }
365 <            } else {
366 <              Double_t BestPairPtMax = lepvec[Z2LeptonPlusIndex].vec.Pt();              
367 <              Double_t BestPairPtMin = lepvec[Z2LeptonMinusIndex].vec.Pt();
368 <              if (lepvec[Z2LeptonMinusIndex].vec.Pt() > BestPairPtMax) {
369 <                BestPairPtMax = lepvec[Z2LeptonMinusIndex].vec.Pt();
370 <                BestPairPtMin = lepvec[Z2LeptonPlusIndex].vec.Pt();
371 <              }
372 <
373 <              Double_t CurrentPairPtMax = lepvec[i].vec.Pt();              
374 <              Double_t CurrentPairPtMin = lepvec[j].vec.Pt();
375 <              if (lepvec[j].vec.Pt() > CurrentPairPtMax) {
376 <                CurrentPairPtMax = lepvec[j].vec.Pt();
377 <                CurrentPairPtMin = lepvec[i].vec.Pt();
378 <              }
379 <
380 <              if (CurrentPairPtMax > BestPairPtMax) {
381 <                if (lepvec[i].charge > 0) {
382 <                  Z2LeptonPlusIndex = i;
383 <                  Z2LeptonMinusIndex = j;
384 <                } else {
385 <                  Z2LeptonPlusIndex = j;
386 <                  Z2LeptonMinusIndex = i;
387 <                }
388 <              } else if (CurrentPairPtMax  == BestPairPtMax) {
389 <                if (CurrentPairPtMin > BestPairPtMin) {
390 <                  if (lepvec[i].charge > 0) {
391 <                    Z2LeptonPlusIndex = i;
392 <                    Z2LeptonMinusIndex = j;
393 <                  } else {
394 <                    Z2LeptonPlusIndex = j;
395 <                    Z2LeptonMinusIndex = i;
396 <                  }                  
397 <                }
398 <              }
399 <            }            
400 <          }
401 <        }
402 <
403 <        // stop if no Z2 candidate is found
404 <        if (Z2LeptonPlusIndex == -1) {
405 <          evtfail |= ( 1<<EVTFAIL_4L );
406 <          return evtfail;
407 <          //      h_evtfail->Fill( evtfail );
408 <          //      cout << "evtfail: " << hex << evtfail << dec << endl;
409 <          //      continue;
410 <        }
411 <        if( ctrl.debug ) cout << "\tgot a Z2 ..." << endl;
412 <        if( ctrl.debug ) cout << "\tZ2 plusindex: " << Z2LeptonPlusIndex
413 <                              << "\tminusindex: " << Z2LeptonMinusIndex << endl;
414 <        TLorentzVector  Z2LeptonPlus  = lepvec[Z2LeptonPlusIndex].vec;
415 <        TLorentzVector  Z2LeptonMinus = lepvec[Z2LeptonMinusIndex].vec;
416 <        TLorentzVector  Z2Candidate   = Z2LeptonPlus+Z2LeptonMinus;
417 <        TLorentzVector  ZZSystem      = Z1Candidate + Z2Candidate;
418 <        if( l != NULL ) {
419 <          l->vecz2 = Z2Candidate;
420 <          l->vecl2p = Z2LeptonPlus;
421 <          l->vecl2m = Z2LeptonMinus;
422 <          l->vec4l = ZZSystem;
423 <        }
424 <        lepvec[Z1LeptonPlusIndex].is4l = true;
425 <        lepvec[Z1LeptonMinusIndex].is4l = true;
426 <        lepvec[Z2LeptonPlusIndex].is4l = true;
427 <        lepvec[Z2LeptonMinusIndex].is4l = true;
428 <
429 <        //***************************************************************
430 <        // Isolation
431 <        //***************************************************************
432 <        bool failiso=false;
433 <
434 <        /*
435 <        int i,j;
436 <        i=Z1LeptonPlusIndex;
437 <        j=Z1LeptonMinusIndex;
438 <        float RIso1 = (lepvec[i].isoTrk+lepvec[i].isoEcal+lepvec[i].isoHcal)/lepvec[i].vec.Pt();
439 <        float RIso2 = (lepvec[j].isoTrk+lepvec[j].isoEcal+lepvec[j].isoHcal)/lepvec[j].vec.Pt();
440 <        float comboIso12 = RIso1 + RIso2;
441 <        i=Z2LeptonPlusIndex;
442 <        j=Z2LeptonMinusIndex;
443 <        float RIso3 = (lepvec[i].isoTrk+lepvec[i].isoEcal+lepvec[i].isoHcal)/lepvec[i].vec.Pt();
444 <        float RIso4 = (lepvec[j].isoTrk+lepvec[j].isoEcal+lepvec[j].isoHcal)/lepvec[j].vec.Pt();
445 <        float comboIso34 = RIso3 + RIso4;
446 <        if( comboIso12 > 0.35 || comboIso34 > 0.35 ) {
447 <              failiso = true;
448 <        }
449 <        */
450 <
451 <        float rho = info->rho;
452 <        for( int i=0; i<lepvec.size(); i++ ) {
453 <          if( !(lepvec[i].is4l) ) continue;
454 <            float effArea_ecal_i, effArea_hcal_i;
455 <            if( lepvec[i].isEB ) {
456 <              if( lepvec[i].type == 11 ) {
457 <                effArea_ecal_i = 0.101;
458 <                effArea_hcal_i = 0.021;
459 <              } else {
460 <                effArea_ecal_i = 0.074;
461 <                effArea_hcal_i = 0.022;
462 <              }
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 <              if( lepvec[i].type == 11 ) {
505 <                effArea_ecal_i = 0.046;
466 <                effArea_hcal_i = 0.040;
467 <              } else {
468 <                effArea_ecal_i = 0.045;
469 <                effArea_hcal_i = 0.030;
470 <              }
504 >              Z2LeptonPlusIndex = j;
505 >              Z2LeptonMinusIndex = i;
506              }
507 <            float isoEcal_corr_i = lepvec[i].isoEcal - (effArea_ecal_i*rho);
508 <            float isoHcal_corr_i = lepvec[i].isoHcal - (effArea_hcal_i*rho);
509 <            for( int j=i+1; j<lepvec.size(); j++ ) {
510 <              if( !(lepvec[j].is4l) ) continue;
511 <              float effArea_ecal_j, effArea_hcal_j;
477 <              if( lepvec[j].isEB ) {
478 <                if( lepvec[j].type == 11 ) {
479 <                  effArea_ecal_j = 0.101;
480 <                  effArea_hcal_j = 0.021;
481 <                } else {
482 <                  effArea_ecal_j = 0.074;
483 <                  effArea_hcal_j = 0.022;
484 <                }
507 >          } else if (CurrentPairPtMax  == BestPairPtMax) {
508 >            if (CurrentPairPtMin > BestPairPtMin) {
509 >              if (lepvec[i].charge > 0) {
510 >                Z2LeptonPlusIndex = i;
511 >                Z2LeptonMinusIndex = j;
512                } else {
513 <                if( lepvec[j].type == 11 ) {
514 <                  effArea_ecal_j = 0.046;
515 <                  effArea_hcal_j = 0.040;
489 <                } else {
490 <                  effArea_ecal_j = 0.045;
491 <                  effArea_hcal_j = 0.030;
492 <                }
493 <              }
494 <              float isoEcal_corr_j = lepvec[j].isoEcal - (effArea_ecal_j*rho);
495 <              float isoHcal_corr_j = lepvec[j].isoHcal - (effArea_hcal_j*rho);
496 <              float RIso_i = (lepvec[i].isoTrk+isoEcal_corr_i+isoHcal_corr_i)/lepvec[i].vec.Pt();
497 <              float RIso_j = (lepvec[j].isoTrk+isoEcal_corr_j+isoHcal_corr_j)/lepvec[j].vec.Pt();          
498 <              float comboIso = RIso_i + RIso_j;
499 <              if( info->evtNum == 1038911933 ) {
500 <                float tmpdR = lepvec[i].vec.DrEtaPhi(lepvec[j].vec);
501 <                cout << "i: " << i
502 <                     << "\tdR: " << tmpdR
503 <                     << "\trho: " << rho
504 <                     << "\tRIso_i: " << RIso_i  
505 <                     << "\ttkrel: " << lepvec[i].isoTrk/lepvec[i].vec.Pt()
506 <                     << "\tecalrel: " << lepvec[i].isoEcal/lepvec[i].vec.Pt()
507 <                     << "\tecalrelcor: " << isoEcal_corr_i/lepvec[i].vec.Pt()
508 <                     << "\thcalrel: " << lepvec[i].isoHcal/lepvec[i].vec.Pt()
509 <                     << "\thcalrelcor: " << isoHcal_corr_i/lepvec[i].vec.Pt()
510 <                     << "\tpt_i: " << lepvec[i].vec.Pt()
511 <                     << "\tj: " << j
512 <                     << "\tRIso_j: " << RIso_j  
513 <                     << "\ttkrel: "  << lepvec[j].isoTrk/lepvec[j].vec.Pt()
514 <                     << "\tecalrel: " << lepvec[j].isoEcal/lepvec[j].vec.Pt()
515 <                     << "\tecalrelcor: " << isoEcal_corr_j/lepvec[j].vec.Pt()
516 <                     << "\thcalrel: " << lepvec[j].isoHcal/lepvec[j].vec.Pt()
517 <                     << "\thcalrelcor: " << isoHcal_corr_j/lepvec[j].vec.Pt()
518 <                     << "\tpt_j: " << lepvec[j].vec.Pt()
519 <                     << "\tcombo: " << comboIso
520 <                     << endl;
521 <                cout.flush();
522 <              }
523 <              if( comboIso > 0.35 ) {
524 <                if( ctrl.debug ) cout << "combo failing for indices: " << i << "," << j << endl;
525 <                failiso = true;
526 <                //            break;
527 <              }
513 >                Z2LeptonPlusIndex = j;
514 >                Z2LeptonMinusIndex = i;
515 >              }                  
516              }
529        }
530        if( failiso ) {
531          evtfail |= ( 1<<EVTFAIL_ISOLATION );
532          return evtfail;
533          //h_evtfail->Fill( evtfail, eventweight );
534          //      h_evtfail->Fill( evtfail );
535          //      cout << "evtfail: " << hex << evtfail << dec << endl;
536          //      continue;
537        }
538
539        //***************************************************************
540        // IP significance
541        //***************************************************************
542        bool failip = false;
543        for( int i=0; i<lepvec.size(); i++ ) {
544          if( !(lepvec[i].is4l) ) continue;
545          if( lepvec[i].ip3dSig > 4 ) {
546            failip=true;
547            break;
517            }
518 <        }
519 <        if( failip ) {
520 <          evtfail |= (1<<EVTFAIL_IP );
521 <          return evtfail;
522 <          //h_evtfail->Fill( evtfail, eventweight );
523 <          //      h_evtfail->Fill( evtfail );
524 <          //      cout << "evtfail: " << hex << evtfail << dec << endl;
525 <          //      continue;
526 <        }
527 <
528 <        //***************************************************************
529 <        // remaining kinematic cuts
530 <        //***************************************************************
531 <        double Z2massCut=0;
532 <        if      ( ctrl.kinematics == "loose" ) Z2massCut = 12;
533 <        else if ( ctrl.kinematics == "tight" ) Z2massCut = 20;
534 <        else { cout <<  "error! kinematic tightness not defined!" << endl; assert(0); }
535 <
536 <        if ( Z1Candidate.M() > 120        ||
537 <             Z2Candidate.M() < Z2massCut  ||
538 <             Z2Candidate.M() > 120        ||
539 <             !(lepvec[Z1LeptonPlusIndex].vec.Pt() > 20.0 || lepvec[Z1LeptonMinusIndex].vec.Pt() > 20.0) ||
540 <             !(lepvec[Z1LeptonPlusIndex].vec.Pt() > 10.0 && lepvec[Z1LeptonMinusIndex].vec.Pt() > 10.0)
572 <             ) {
573 <          evtfail |= (1<<EVTFAIL_KINEMATICS );
574 <          return evtfail;
575 <          //h_evtfail->Fill( evtfail, eventweight );
576 <          //      h_evtfail->Fill( evtfail );
577 <          //      cout << "evtfail: " << hex << evtfail << dec << endl;
578 <          //      continue;
579 <        }
580 <        
581 <        int channel;
582 <        if( lepvec[Z1LeptonMinusIndex].type == 11 && lepvec[Z2LeptonMinusIndex].type == 11 ) channel=0;
583 <        if( lepvec[Z1LeptonMinusIndex].type == 13 && lepvec[Z2LeptonMinusIndex].type == 13 ) channel=1;
584 <        if( (lepvec[Z1LeptonMinusIndex].type == 11 && lepvec[Z2LeptonMinusIndex].type == 13) ||
585 <            (lepvec[Z1LeptonMinusIndex].type == 13 && lepvec[Z2LeptonMinusIndex].type == 11)) channel=2;
586 <        
587 <
588 <
589 <        if( passtuple != NULL ) {
590 <          unsigned run   = info->runNum;
591 <          unsigned evt   = info->evtNum;
592 <          unsigned lumi  = info->lumiSec;
593 <          unsigned chan  = channel;
594 <          double   w     = eventweight;
595 <          float mZ1      = Z1Candidate.M() ;
596 <          float mZ2      = Z2Candidate.M() ;
597 <          float m4l      = ZZSystem.M() ;
598 <          float pt4l     = ZZSystem.Pt() ;
599 <          passtuple->SetBranchAddress("run",  &run);
600 <          passtuple->SetBranchAddress("evt",  &evt);
601 <          passtuple->SetBranchAddress("lumi", &lumi);
602 <          passtuple->SetBranchAddress("mZ1",  &mZ1);
603 <          passtuple->SetBranchAddress("mZ2",  &mZ2);
604 <          passtuple->SetBranchAddress("m4l",  &m4l);
605 <          passtuple->SetBranchAddress("pt4l", &pt4l);
606 <          passtuple->SetBranchAddress("w",    &w);
607 <          passtuple->Fill( );
608 <        }
609 <
610 <        if( ctrl.debug ) cout  << "run: " << info->runNum  
611 <              << "\tevt: " << info->evtNum
612 <              << "\tZ1channel: " << lepvec[Z1LeptonMinusIndex].type
613 <              << "\tZ2channel: " << lepvec[Z2LeptonMinusIndex].type
614 <              << "\tmZ1: " << Z1Candidate.M()
615 <              << "\tmZ2: " << Z2Candidate.M()
616 <              << "\tm4l: " << ZZSystem.M()
617 <              << "\tevtfail: " << hex << evtfail << dec
618 <              << "\ttrigbits: " << hex << info->triggerBits << dec
619 <          //          << "\ttree: " << inputFiles[q][f]
620 <              << endl;
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