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

Comparing UserCode/MitHzz4l/NonMCBackground/src/SelectionEMU.cc (file contents):
Revision 1.3 by khahn, Fri Feb 17 14:48:01 2012 UTC vs.
Revision 1.7 by dkralph, Mon Jun 18 06:56:17 2012 UTC

# Line 1 | Line 1
1 + //***************************************************************************************************
2 + //
3 + // selection sync'ed with https://twiki.cern.ch/twiki/bin/viewauth/CMS/HiggsZZ4l2012SummerSync
4 + //
5 + //***************************************************************************************************
6 +
7 + // system headers
8 + #include <map>
9 + #include <utility>
10 +
11 + // mit headers
12 + #include "Vertex.h"
13 +
14 + // 4L stuff
15   #include "SelectionStatus.h"
16   #include "EventData.h"
17   #include "SimpleLepton.h"
18   #include "EfficiencyWeightsInterface.h"
19 <
20 < #include "HZZBDTElectronSelection.h"
19 > #include "ElectronSelection.h"
20 > #include "MuonSelection.h"
21   #include "IsolationSelection.h"
8 #include "PassHLT.h"
22   #include "SelectionEMU.h"
23 <
24 < #include "ExternData.h"
23 > #include "ReferenceSelection.h"
24 > #include "Selection.h"
25 > #include "CommonDefs.h"
26   #include "SelectionDefs.h"
27 + #include "FSR.h"
28 + #include "SelectionFuncs.h"
29  
30  
31 + extern vector<SimpleLepton> failingLeptons;
32 + extern vector<SimpleLepton> passingLeptons;
33  
34 < // +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
35 < // +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
36 < EventData apply_EMU_selection(ControlFlags &ctrl,           // input control
37 <                              mithep::TEventInfo *info,     // input event info
38 <                              TClonesArray *electronArr,    // input electrons
39 <                              SelectionStatus (*ElectronPreSelector)( ControlFlags &, const mithep::TElectron*),
40 <                              SelectionStatus (*ElectronIDSelector)( ControlFlags &, const mithep::TElectron*),
41 <                              SelectionStatus (*ElectronIsoSelector)( ControlFlags &, const mithep::TElectron*),
42 <                              TClonesArray *muonArr,         // input muons
43 <                              SelectionStatus (*MuonPreSelector)( ControlFlags &, const mithep::TMuon*),
44 <                              SelectionStatus (*MuonIDSelector)( ControlFlags &, const mithep::TMuon*),
45 <                              SelectionStatus (*MuonIsoSelector)( ControlFlags &, const mithep::TMuon*) )
46 < // +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
47 < // +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
34 > //--------------------------------------------------------------------------------------------------
35 > EventData apply_HZZ4L_EMU_selection(ControlFlags &ctrl,           // input control
36 >                                        const mithep::EventHeader *info,     // input event info
37 >                                        const mithep::Array<mithep::Vertex>       * vtxArr ,
38 >                                        const mithep::Array<mithep::PFCandidate>  *pfCandidates,
39 >                                        const mithep::Array<mithep::PileupEnergyDensity>  *puEnergyDensity,
40 >                                        const mithep::Array<mithep::Electron> *electronArr,    // input electrons
41 >                                        SelectionStatus (*ElectronPreSelector)( ControlFlags &,
42 >                                                                                const mithep::Electron*,
43 >                                                                                const mithep::Vertex *),
44 >                                        SelectionStatus (*ElectronIDSelector)( ControlFlags &,
45 >                                                                               const mithep::Electron*,
46 >                                                                               const mithep::Vertex *),
47 >                                        SelectionStatus (*ElectronIsoSelector)( ControlFlags &,
48 >                                                                                const mithep::Electron*,
49 >                                                                                const mithep::Vertex *,
50 >                                                                                const mithep::Array<mithep::PFCandidate> *,
51 >                                                                                const mithep::Array<mithep::PileupEnergyDensity> *,
52 >                                                                                mithep::ElectronTools::EElectronEffectiveAreaTarget,
53 >                                                                                vector<const mithep::PFCandidate*>),
54 >                                        const mithep::Array<mithep::Muon> *muonArr,    // input muons
55 >                                        SelectionStatus (*MuonPreSelector)( ControlFlags &,
56 >                                                                            const mithep::Muon*,
57 >                                                                            const mithep::Vertex *,
58 >                                                                            const mithep::Array<mithep::PFCandidate> *),
59 >                                        SelectionStatus (*MuonIDSelector)( ControlFlags &,
60 >                                                                           const mithep::Muon*,
61 >                                                                           // const mithep::Vertex &),
62 >                                                                           const mithep::Vertex *,
63 >                                                                           const mithep::Array<mithep::PFCandidate> *),
64 >                                        SelectionStatus (*MuonIsoSelector)( ControlFlags &,
65 >                                                                            const mithep::Muon*,
66 >                                                                            const mithep::Vertex *,
67 >                                                                            const mithep::Array<mithep::PFCandidate> *,
68 >                                                                            const mithep::Array<mithep::PileupEnergyDensity> *,
69 >                                                                            mithep::MuonTools::EMuonEffectiveAreaTarget,
70 >                                                                            vector<const mithep::PFCandidate*>)
71 >                                        )
72 > //--------------------------------------------------------------------------------------------------
73   {      
74  
75    EventData ret;
76    unsigned evtfail = 0x0;
77    TRandom3 r;
35  
36  if( ctrl.debug ) {
37    cout << "Run: " << info->runNum
38         << "\tEvt: " << info->evtNum
39         << "\tLumi: " << info->lumiSec
40         << endl;
41  }
78  
79 <  if( !ctrl.mc ) {
80 <    // not accounting for overlap atm
45 <    RunLumiRangeMap::RunLumiPairType rl(info->runNum, info->lumiSec);      
46 <    if( !(rlrm.HasRunLumi(rl)) )  {
47 <      if( ctrl.debug ) cout << "\tfails JSON" << endl;
48 <      ret.status.setStatus(0);
49 <      return ret;
50 <    }
51 <  }
52 <  
53 <  
54 <  //********************************************************
55 <  // Trigger
56 <  //********************************************************
57 < //   if( !ctrl.mc ) {
58 < //     if( !(passHLTSingleMuon(info->triggerBits) )  ) {
59 < //       if( ctrl.debug ) cout << "\tfails trigger" << endl;
60 < //       evtfail |= (1<<EVTFAIL_TRIGGER);
61 < //       ret.status.setStatus(0);
62 < //       return ret;
63 < //     }        
64 < //   }
65 <  if( ctrl.debug ) {
66 <    cout << "presel nlep: " << muonArr->GetEntries() + electronArr->GetEntries()
67 <         << "\tnmuon: "    << muonArr->GetEntries()
68 <         << "\tnelectron: " << electronArr->GetEntries()
69 <         << endl;
70 <  }
79 >  failingLeptons.clear();
80 >  passingLeptons.clear();
81  
82 <  //********************************************************
83 <  // Lepton Selection
84 <  //********************************************************
85 <  vector<SimpleLepton> lepvec;
86 <  if( muonArr->GetEntries() != 1 && electronArr->GetEntries() != 1 ) {
87 <    ret.status.setStatus(0);
82 >  mithep::MuonTools::EMuonEffectiveAreaTarget eraMu;
83 >  mithep::ElectronTools::EElectronEffectiveAreaTarget eraEle;
84 >  getEATargets(ctrl,eraMu,eraEle);
85 >
86 >  const mithep::Vertex * vtx;
87 >  bool goodVertex = setPV( ctrl, vtxArr, vtx );
88 >  if(goodVertex) {
89 >    ret.status.selectionBits.flip(PASS_SKIM2);
90 >  } else {
91 >    if(ctrl.debug) cout << "found bad vertex" << endl;
92 >    ret.status.setStatus(SelectionStatus::FAIL);
93      return ret;
94    }
95  
96 <  //    
96 >  //***********************************************************
97 >  // Lepton Selection
98 >  //***********************************************************
99 >  vector<SimpleLepton> lepvec;
100 >  vector<const mithep::PFCandidate*> photonsToVeto;
101 >
102 >
103    if( ctrl.debug ) cout << "\tnMuons: " << muonArr->GetEntries() << endl;
104    //----------------------------------------------------
105 <  for(Int_t i=0; i<muonArr->GetEntries(); i++)
105 >  for(int i=0; i<muonArr->GetEntries(); i++)
106      {
107 <      const mithep::TMuon *mu = (mithep::TMuon*)((*muonArr)[i]);      
107 >      const mithep::Muon *mu = (mithep::Muon*)((*muonArr)[i]);      
108        
109        SelectionStatus musel;
110 <      if(ctrl.debug) cout << "musel.status  before anything: " << musel.getStatus() << endl;
111 <      musel |= (*MuonPreSelector)(ctrl,mu);
112 <      if(ctrl.debug) cout << "musel.status  after presel: " << musel.getStatus() << endl;
113 <      musel |= (*MuonIDSelector)(ctrl,mu);
114 <      if(ctrl.debug) cout << "musel.status  after ID: " << musel.getStatus() << endl;
115 <      musel |= (*MuonIsoSelector)(ctrl,mu);
116 <      if(ctrl.debug) cout << "musel.status  after iso: " << musel.getStatus() << endl;
117 <
97 <      if( ctrl.debug ) {
98 <        cout << "muon:: pt: " << mu->pt
99 <             << "\teta: " << mu->eta
100 <             << "\tisorel: " <<  mu->pfIso03/mu->pt
101 <             << "\tstatus: " << hex << musel.getStatus() << dec
102 <             << endl;
110 >
111 >      musel |= (*MuonPreSelector)(ctrl,mu,vtx,pfCandidates);
112 >      if( !(musel.getStatus() & SelectionStatus::PRESELECTION) ) continue;
113 >
114 >      musel |= (*MuonIDSelector)(ctrl,mu,vtx,pfCandidates );
115 >
116 >      if( !(ctrl.doFSR) ) {
117 >        musel |=  (*MuonIsoSelector)(ctrl,mu,vtx,pfCandidates,puEnergyDensity,eraMu,photonsToVeto);
118        }
119 +
120 +      SimpleLepton tmplep;
121 +      float pt = mu->Pt();
122 +      tmplep.vec.SetPtEtaPhiM(pt,
123 +                               mu->Eta(),
124 +                               mu->Phi(),
125 +                               MUON_MASS);
126        
127 <      if ( musel.pass() ) {
127 >      tmplep.type    = 13;
128 >      tmplep.index   = i;
129 >      tmplep.charge  = mu->Charge();
130 >      tmplep.isoTrk  = mu->IsoR03SumPt();
131 >      tmplep.isoEcal = mu->IsoR03EmEt();
132 >      tmplep.isoHcal = mu->IsoR03HadEt();
133 >      tmplep.isoPF04 = musel.isoPF04;
134 >      tmplep.chisoPF04 = musel.chisoPF04;
135 >      tmplep.gaisoPF04 = musel.gaisoPF04;
136 >      tmplep.neisoPF04 = musel.neisoPF04;
137 >      // tmplep.isoPF03 = computePFMuonIso(mu,vtx,pfCandidates,0.3);
138 >      // tmplep.isoPF04 = computePFMuonIso(mu,vtx,pfCandidates,0.4);
139 >      tmplep.ip3dSig = mu->Ip3dPVSignificance();
140 >      tmplep.is4l    = false;
141 >      tmplep.isEB    = (fabs(mu->Eta()) < 1.479 ? 1 : 0 );
142 >      tmplep.isoMVA  = musel.isoMVA;
143 >      tmplep.isTight = musel.tight();
144 >      tmplep.isLoose = musel.loose();
145 >      tmplep.status  = musel;    
146 >      tmplep.fsrRecoveryAttempted = false;
147 >      SelectionStatus tmpstat = PassWwMuonSel(mu,vtx,pfCandidates);
148 >      tmplep.tightCutsApplied = tmpstat.tight();
149 >      lepvec.push_back(tmplep);
150 >      if( ctrl.debug ) cout << endl;
151 >    }
152  
153 <        SimpleLepton tmplep;
154 <        float pt = mu->pt;
155 <        tmplep.vecorig->SetPtEtaPhiM(pt,
156 <                                    mu->eta,
157 <                                    mu->phi,
158 <                                    MUON_MASS);
153 >    if( ctrl.debug ) { cout << "\tnElectron: " << electronArr->GetEntries() << endl; }
154 >    // --------------------------------------------------------------------------------
155 >    for(int i=0; i<electronArr->GetEntries(); i++)
156 >      {
157 >        const mithep::Electron *ele = (mithep::Electron*)((*electronArr)[i]);
158 >
159 >        SelectionStatus elesel;
160          
161 <        if( ctrl.do_escale_up ) {
162 <          pt=scale_smear_muon_Up(pt, 1,  r);
163 <        }
164 <        if( ctrl.do_escale_down ) {
165 <          pt=scale_smear_muon_Down(pt, 1,  r);
161 >        elesel |= (*ElectronPreSelector)(ctrl,ele,vtx);
162 >        if( !(elesel.getStatus() & SelectionStatus::PRESELECTION) ) continue;
163 >        
164 >        elesel |= (*ElectronIDSelector)(ctrl,ele,vtx);
165 >        
166 >        if( !(ctrl.doFSR) ) {
167 >          elesel |= (*ElectronIsoSelector)(ctrl,ele,vtx,pfCandidates,puEnergyDensity,eraEle,photonsToVeto);
168          }
169          
170 <        tmplep.vec->SetPtEtaPhiM(pt,
171 <                                mu->eta,
172 <                                mu->phi,
173 <                                MUON_MASS);
170 >        SimpleLepton tmplep;
171 >        float pt = ele->Pt();
172 >        tmplep.vec.SetPtEtaPhiM( pt,
173 >                                 ele->Eta(),
174 >                                 ele->Phi(),
175 >                                 ELECTRON_MASS );
176          
177 <        tmplep.type    = 13;
177 >        tmplep.type    = 11;
178          tmplep.index   = i;
179 <        tmplep.charge  = mu->q;
180 <        tmplep.isoTrk  = mu->trkIso03;
181 <        tmplep.isoEcal = mu->emIso03;
182 <        tmplep.isoHcal = mu->hadIso03;
183 <        tmplep.isoPF03 = mu->pfIso03;
184 <        tmplep.isoPF04 = mu->pfIso04;
185 <        tmplep.ip3dSig = mu->ip3dSig;
179 >        tmplep.charge  = ele->Charge();
180 >        tmplep.isoTrk  = ele->TrackIsolationDr03();
181 >        tmplep.isoEcal = ele->EcalRecHitIsoDr03();
182 >        tmplep.isoHcal = ele->HcalTowerSumEtDr03();
183 >        tmplep.isoPF04 = elesel.isoPF04;
184 >        tmplep.chisoPF04 = elesel.chisoPF04;
185 >        tmplep.gaisoPF04 = elesel.gaisoPF04;
186 >        tmplep.neisoPF04 = elesel.neisoPF04;
187 >        // tmplep.isoPF03 = computePFEleIso(ele,vtx,pfCandidates,0.3);
188 >        // tmplep.isoPF04 = computePFEleIso(ele,vtx,pfCandidates,0.4);
189 >        tmplep.ip3dSig = ele->Ip3dPVSignificance();
190          tmplep.is4l    = false;
191 <        tmplep.isEB    = (fabs(mu->eta) < 1.479 ? 1 : 0 );
192 <        tmplep.isTight = musel.tight();
193 <        tmplep.isLoose = musel.loose();
191 >        tmplep.isEB    = ele->IsEB();
192 >        tmplep.scID    = ele->SCluster()->GetUniqueID();
193 >        tmplep.isTight = elesel.tight();
194 >        tmplep.isLoose = elesel.loose();
195 >        tmplep.status  = elesel;
196 >        tmplep.idMVA   = elesel.idMVA;
197 >        tmplep.isoMVA  = elesel.isoMVA;
198 >        tmplep.fsrRecoveryAttempted = false;
199 >        SelectionStatus tmpstat = electronTagSelection(ele,vtx,pfCandidates);
200 >        tmplep.tightCutsApplied   = tmpstat.tight();
201          lepvec.push_back(tmplep);
202 <        if( ctrl.debug ) { cout << "muon passes ... " << endl;}
202 >        if( ctrl.debug ) cout << endl;
203        }
142      //  }
143    }    
144  
145  
146    
147    //
148    if( ctrl.debug ) { cout << "\tnElectron: " << electronArr->GetEntries() << endl; }
149    // --------------------------------------------------------------------------------
150    for(Int_t i=0; i<electronArr->GetEntries(); i++)
151      {
152        const mithep::TElectron *ele = (mithep::TElectron*)((*electronArr)[i]);
153        
154        Bool_t isMuonOverlap = kFALSE;
155        for (int k=0; k<lepvec.size(); ++k) {
156          TVector3 tmplep;
157          tmplep.SetPtEtaPhi(ele->pt, ele->eta, ele->phi);
158          if ( abs(lepvec[k].type == 13) && lepvec[k].vec->Vect().DrEtaPhi(tmplep) < 0.15 ) {
159            if( ctrl.debug ) cout << "-----> isMuonOverlap! " << endl;
160            isMuonOverlap = kTRUE;
161            break;
162          }
163        }
204  
205 <        // kick out non fiducial, low pT eles
166 <        if ( ele->pt < 7 || fabs(ele->eta) > 2.5 )
167 <          continue;
168 <        
169 <        SelectionStatus elesel;
170 <        elesel |= (*ElectronPreSelector)(ctrl,ele);
171 <        elesel |= (*ElectronIDSelector)(ctrl,ele);
172 <        elesel |= (*ElectronIsoSelector)(ctrl,ele);
173 <
174 < //      if( elesel.getStatus() & SelectionStatus::PRESELECTION )
175 < //        elesel.setStatus(SelectionStatus::LOOSESELECTION);
176 <
177 <        if( ctrl.debug ){
178 <          cout << "\tscEt: " << ele->scEt
179 <               << "\tscEta: " << ele->scEta
180 <               << "\tncluster: " << ele->ncluster
181 <               << "\tisorel: " <<  ele->pfIso04/ele->pt
182 <               << "\tstatus: " << hex << elesel.getStatus() << dec
183 <               << endl;
184 <        }
205 >    sort( lepvec.begin(), lepvec.end(), SimpleLepton::lep_pt_sort );
206  
186        //      if ( elesel.pass() && !isMuonOverlap )
187        if ( !isMuonOverlap )
188          {
189            SimpleLepton tmplep;
190            
191            float pt = ele->pt;
192            tmplep.vecorig->SetPtEtaPhiM( pt,
193                                          ele->eta,
194                                          ele->phi,
195                                          ELECTRON_MASS );
196            
197            if( ctrl.do_escale ) {
198              pt=scale_smear_electron(pt, ele->isEB, r);
199            }
200            if( ctrl.do_escale_up ) {
201              pt=scale_smear_electron_Up(pt, ele->isEB,  r);
202            }
203            if( ctrl.do_escale_down ) {
204              pt=scale_smear_electron_Down(pt, ele->isEB,  r);
205            }
206
207            
208            tmplep.vec->SetPtEtaPhiM( pt,
209                                      ele->eta,
210                                      ele->phi,
211                                      ELECTRON_MASS );
212            
213            tmplep.type    = 11;
214            tmplep.index   = i;
215            tmplep.charge  = ele->q;
216            tmplep.isoTrk  = ele->trkIso03;
217            tmplep.isoEcal = ele->emIso03;
218            tmplep.isoHcal = ele->hadIso03;
219            tmplep.isoPF03 = ele->pfIso03;
220            tmplep.isoPF04 = ele->pfIso04;
221            tmplep.ip3dSig = ele->ip3dSig;
222            tmplep.is4l    = false;
223            tmplep.isEB    = ele->isEB;
224            tmplep.scID    = ele->scID;
225            tmplep.isTight = elesel.tight();
226            tmplep.isLoose = elesel.loose();
227            //      tmplep.isLoose = elesel.pass();
228            lepvec.push_back(tmplep);
229            if( ctrl.debug ) { cout << "\telectron passes ... " << endl; }
230          }
231      }
232    
233    
207      //********************************************************
208 <    // Dump Stuff
208 >    // Step 2: Lepton Cleaning
209      //********************************************************
210 <    sort( lepvec.begin(), lepvec.end(), SimpleLepton::lep_pt_sort );
211 <    int nmu=0, nele=0;
212 <    for( int i=0; i<lepvec.size(); i++ ) {
213 <      if(ctrl.debug) cout << "lepvec :: index: " << i
214 <                          << "\tpt: " << lepvec[i].vec->Pt()
215 <                          << "\ttype: " << lepvec[i].type
216 <                          << endl;
217 <      if( abs(lepvec[i].type) == 11 ) nele++;
218 <      else nmu++;
210 >    vector<vector<SimpleLepton>::iterator> electrons_to_erase;
211 >    for (vector<SimpleLepton>::iterator it1=lepvec.begin(); it1 != lepvec.end(); it1++ ) {
212 >      if ( abs(it1->type) != 11 ) continue;
213 >      TVector3 evec = it1->vec.Vect();
214 >      
215 >      bool erase_this_electron=false;
216 >      for (vector<SimpleLepton>::iterator it2=lepvec.begin(); it2 != lepvec.end(); it2++ ) {
217 >        if ( it2 == it1 )                       continue;
218 >        if ( abs(it2->type) != 13 )             continue;
219 >        // if( !(it2->status.looseIDAndPre()) )    continue;
220 >        TVector3 mvec = it2->vec.Vect();
221 >        
222 >        if ( evec.DrEtaPhi(mvec) < 0.05 ) {
223 >          erase_this_electron=true;
224 >          break;
225 >        }
226 >      }
227 >      if( erase_this_electron )
228 >        electrons_to_erase.push_back(it1);
229      }
230 <    if( ctrl.debug ) {
231 <      cout << "postsel nlep: " << lepvec.size()
249 <           << "\tnmuon: " << nmu
250 <           << "\tnelectron: " << nele
251 <           << endl;
230 >    for( int i=0; i<electrons_to_erase.size(); i++ ) {
231 >      lepvec.erase(electrons_to_erase[i]);
232      }
253    
254    
255    //******************************************************************************
256    // Selection
257    //******************************************************************************
258    float bestMass=-1; int best_mu_index=-1, best_ele_index=-1;
259    for(int i = 0; i<lepvec.size(); i++) {      // get a tight muon
260      if( abs(lepvec[i].type) != 13 ) continue;
261      if( !(lepvec[i].isTight) ) continue;
262      //      if( lepvec[i].vec->Pt() < 35 ) continue;
263      if( ctrl.debug ) cout << "got a muon, index: " << i << endl;
264
265      for(int j = 0; j<lepvec.size(); j++) {     // get a loose electron
266        if( j == i ) continue;
267        if( abs(lepvec[j].type) != 11 ) continue;
268        //      if( !(lepvec[j].isLoose) ) continue;
269        //      if (lepvec[i].charge == lepvec[j].charge) continue;          
270        if (lepvec[i].charge != lepvec[j].charge) continue;          
271        if( ctrl.debug ) cout << "got a electron, index: " << j << endl;
272
273        float tmpMass = ( *(lepvec[i].vec) + *(lepvec[j].vec) ).M();
274        if( ctrl.debug ) cout << "tmp vs best :: " << tmpMass << "," << bestMass << endl;
275        if( tmpMass > bestMass ) {
276          bestMass = tmpMass;
277          best_mu_index=i;
278          best_ele_index=j;
279        }
233  
234 +    //********************************************************
235 +    // Step 3: Good Leptons
236 +    //********************************************************
237 +    vector<double> pt_of_leptons_to_erase;
238 +    for (int i=0; i<lepvec.size(); i++ ) {
239 +      bool already_pushed=false;
240 +      if( !(lepvec[i].status.loose()) ) {
241 +        pt_of_leptons_to_erase.push_back(lepvec[i].vec.Pt());
242 +        already_pushed = true;
243 +        failingLeptons.push_back(lepvec[i]); // these should pass preselection
244 +      } else {
245 +        passingLeptons.push_back(lepvec[i]);
246        }
247 + #ifndef SYNC
248 +      if( !already_pushed && fabs(lepvec[i].ip3dSig)>4 )  
249 +        pt_of_leptons_to_erase.push_back(lepvec[i].vec.Pt());
250 + #endif      
251      }
252 <
253 <    if( bestMass > 0 ) {
254 <      if( ctrl.debug ) cout << "EMU candidate, mass : " << bestMass << endl;
255 <
256 < //       if( lepvec[best_ele_index].isTight )
257 < //      ret.status.setStatus(SelectionStatus::TIGHTSELECTION);
258 < //       else
290 < //      ret.status.setStatus(SelectionStatus::LOOSESELECTION);
291 <
292 <      if( lepvec[best_ele_index].isTight )
293 <        ret.status.setStatus(SelectionStatus::TIGHTSELECTION);
294 <      if( lepvec[best_ele_index].isLoose )
295 <        ret.status.setStatus(SelectionStatus::LOOSESELECTION);
296 <      if( !(lepvec[best_ele_index].isTight) && !(lepvec[best_ele_index].isLoose)  )
297 <        ret.status.setStatus(SelectionStatus::UNDEFINED); // using preselection here to mean that muon
298 <                                                          // is good & ele pass neither loose or tight
299 <
300 <      ret.Z1leptons.push_back(lepvec[best_mu_index]);
301 <      ret.Z1leptons.push_back(lepvec[best_ele_index]);
302 <      ret.Z2leptons.push_back(lepvec[best_mu_index]);
303 <      ret.Z2leptons.push_back(lepvec[best_ele_index]);
304 <      return ret;
252 >    for( int i=0; i<pt_of_leptons_to_erase.size(); i++ ) {
253 >      for( vector<SimpleLepton>::iterator it=lepvec.begin(); it != lepvec.end(); it++ ) {
254 >        SimpleLepton flep = *it;
255 >        if( flep.vec.Pt() != pt_of_leptons_to_erase[i] ) continue;
256 >        lepvec.erase(it);
257 >        break;
258 >      }
259      }
260  
261 <    ret.status.setStatus(SelectionStatus::FAIL);
261 >    //******************************************************************************
262 >    // W + (OF SS lepton) Selection
263 >    //******************************************************************************
264 >    if(has_ssof_lepton(ctrl)) {
265 >      ret.status.setStatus(SelectionStatus::EVTPASS);
266 >      ret.Z1leptons.push_back(passingLeptons[0]);
267 >      ret.Z1leptons.push_back(passingLeptons[0]);
268 >      ret.Z2leptons.push_back(passingLeptons[0]);
269 >      ret.Z2leptons.push_back(passingLeptons[0]);
270 >    } else {
271 >      ret.status.setStatus(SelectionStatus::FAIL);
272 >    }
273      return ret;
309
274   }
275 + //----------------------------------------------------------------------------------------
276 + bool has_ssof_lepton(ControlFlags &ctrl)
277 + {
278 +  bool has_ssof=false;
279 +  for(unsigned iw=0; iw<passingLeptons.size(); iw++) {
280 +    SimpleLepton w_lep = passingLeptons[iw];
281 +    //????????????????????????????????????????????????????????????????????????????????????????
282 +    // this is applied in skim (skim also applies ww muon id)
283 +    // if(abs(w_lep.type) == 11) {
284 +    //   if( !(w_lep.tightCutsApplied) )
285 +    //  continue;
286 +    // }
287 +    //????????????????????????????????????????????????????????????????????????????????????????
288 +    for(unsigned ifake=0; ifake<failingLeptons.size(); ifake++) {
289 +      SimpleLepton fake_lep = failingLeptons[ifake];
290 +      if(abs(fake_lep.type) == abs(w_lep.type)) continue;
291 +      if(fake_lep.charge != w_lep.charge) continue;
292 +      has_ssof = true;
293 +    }
294 +    for(unsigned ipass=0; ipass<passingLeptons.size(); ipass++) {
295 +      if(ipass == iw) continue;
296 +      SimpleLepton pass_lep = passingLeptons[ipass];
297 +      if(abs(pass_lep.type) == abs(w_lep.type)) continue;
298 +      if(pass_lep.charge != w_lep.charge) continue;
299 +      has_ssof = true;
300 +    }
301 +  }
302  
303 <
303 >  return has_ssof;
304 > }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines