[ViewVC] Diff of: cvsroot/UserCode/MitHzz4l/LeptonSelection/src/MuonSelection.cc

Comparing UserCode/MitHzz4l/LeptonSelection/src/MuonSelection.cc (file contents):
Revision 1.22 by anlevin, Mon May 28 12:37:10 2012 UTC vs.
Revision 1.35 by dkralph, Tue Jul 17 10:17:22 2012 UTC

+                        const mithep::Array<mithep::PFCandidate> * pfCandidates,
+                        const Double_t dRMax);
-+
+bool muonPOG2012CutBasedIDTight(const mithep::Muon *mu, const mithep::Vertex * vtx, const mithep::Array<mithep::PFCandidate> * pfCandidates, const mithep::Array<mithep::PileupEnergyDensity> * puEnergyDensity)
-+
+{
-+
-+
+  Float_t charged_iso = 0;
-+
-+
+  Float_t sumChargedHadronPt = 0;
-+
+  Float_t sumNeutralHadronPt = 0;
-+
+  Float_t sumPhotonPt = 0;
-+
+  Float_t sumPUPt = 0;
-+
-+
+  for(unsigned k=0; k<pfCandidates->GetEntries(); ++k) {
-+
+    const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[k]);
-+
+    Double_t dr = mithep::MathUtils::DeltaR(mu->Phi(),mu->Eta(), pf->Phi(), pf->Eta());
-+
+    if(dr > 0.4) continue;
-+
+    if(pf->TrackerTrk() && mu->TrackerTrk() && pf->TrackerTrk() == mu->TrackerTrk()) continue;
-+
-+
+    if(PFnoPUflag[k]){ //pf no pileup
-+
+      if(abs(pf->PFType()) == mithep::PFCandidate::eHadron)
-+
+        sumChargedHadronPt += pf->Pt();
-+
+      else if (pf->Pt() > 0.5 && abs(pf->PFType()) == mithep::PFCandidate::eNeutralHadron)
-+
+        sumNeutralHadronPt += pf->Pt();
-+
+      else if (pf->Pt() > 0.5 && abs(pf->PFType()) == mithep::PFCandidate::eGamma)
-+
+        sumPhotonPt += pf->Pt();
-+
+    }
-+
+    else {//pf pile up
-+
+      if(pf->Charge() != 0)
-+
+        sumPUPt += pf->Pt();
-+
+    }
-+
-+
+  }
-+
+  double iso_sum =  sumChargedHadronPt+ max(0.,sumNeutralHadronPt+sumPhotonPt-0.5*sumPUPt);
-+
-+
+  bool pass = kTRUE;
-+
+  if(!mu->IsGlobalMuon()) pass = kFALSE;
-+
-+
+  //there is not necessarily a global track or a tracker track if is global muon is false
-+
+  if(!pass) return pass;
-+
-+
+  if(!mu->IsPFMuon()) pass = kFALSE;
-+
+  if(mu->GlobalTrk()->RChi2() > 10) pass= kFALSE;
-+
+  if(mu->NValidHits() == 0 ) pass= kFALSE;
-+
+  if(mu->NMatches() <= 1) pass = kFALSE;
-+
+  if(fabs(mu->TrackerTrk()->D0Corrected(*vtx)) > 0.2) pass = kFALSE;
-+
+  if(fabs(mu->TrackerTrk()->DzCorrected(*vtx)) > 0.5) pass = kFALSE;
-+
+  if(mu->BestTrk()->NPixelHits() == 0) pass = kFALSE;
-+
+  if(mu->NTrkLayersHit() <= 5) pass = kFALSE;
-+
+  if(iso_sum/mu->Pt() > 0.12 ) pass = kFALSE;
-+
-+
+  return pass;
-+
-+
-+
+}
-+
+//----------------------------------------------------------------------------------------
-+
+bool muon2012CutBasedIDTightVersionWithOldIsoThatWorksOn2011(ControlFlags &ctrl, const mithep::Muon *mu, const mithep::Vertex * vtx, const mithep::Array<mithep::PFCandidate> * pfCandidates,
-+
+                                                             const mithep::Array<mithep::PileupEnergyDensity> * puEnergyDensity,   mithep::MuonTools::EMuonEffectiveAreaTarget EffectiveAreaVersion
-+
+                                                             )
-+
+{
-+
-+
+  mithep::MuonTools   muT;
-+
-+
+  Float_t charged_iso = 0;
-+
+  Float_t neutral_iso = 0;
-+
+  Float_t gamma_iso = 0;
-+
-+
+  for(int k=0; k<pfCandidates->GetEntries(); ++k) {
-+
-+
+    if( !(PFnoPUflag[k]) ) continue;
-+
+    const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[k]);
-+
-+
+    Double_t dr = mithep::MathUtils::DeltaR(mu->Phi(),mu->Eta(), pf->Phi(), pf->Eta());
-+
+    if (dr > 0.4) continue;
-+
-+
+    if (pf->Charge() != 0 && (pf->HasTrackerTrk()||pf->HasGsfTrk()) ) {
-+
-+
+      if (abs(pf->PFType()) == mithep::PFCandidate::eElectron || abs(pf->PFType()) == mithep::PFCandidate::eMuon) continue;
-+
-+
+      charged_iso += pf->Pt();
-+
+    }
-+
-+
+    else if (abs(pf->PFType()) == mithep::PFCandidate::eGamma) {
-+
-+
+      gamma_iso += pf->Pt();
-+
+    }
-+
-+
+    else {
-+
+      neutral_iso += pf->Pt();
-+
+    }
-+
+  }
-+
-+
+  double iso_sum=999;
-+
+  double rho;
-+
+  if(ctrl.era == 2012) {
-+
+    EffectiveAreaVersion  = mithep::MuonTools::kMuEAData2012;
-+
+    rho = puEnergyDensity->At(0)->RhoKt6PFJetsCentralNeutral();
-+
-+
+    iso_sum = charged_iso + fmax(0.0,(gamma_iso + neutral_iso
-+
+                                      -rho*muT.MuonEffectiveArea(muT.kMuGammaAndNeutralHadronIso04,
-+
+                                                                 mu->Eta(),EffectiveAreaVersion)));
-+
+  } else if(ctrl.era == 2011) {
-+
+    setMuonRhoAndEaVersion(rho,EffectiveAreaVersion,puEnergyDensity);
-+
+    iso_sum = charged_iso + fmax(0.0,(gamma_iso + neutral_iso
-+
+                                      -rho*muT.MuonEffectiveArea(muT.kMuGammaAndNeutralHadronIso04,
-+
+                                                                 mu->Eta(),EffectiveAreaVersion)));
-+
+  } else assert(0);
-+
-+
+  bool pass = kTRUE;
-+
+  if(!mu->IsGlobalMuon()) pass = kFALSE;
-+
-+
+  //there is not necessarily a global track or a tracker track if is global muon is false
-+
+  if(!pass) return pass;
-+
-+
+  if(!mu->IsPFMuon() && ctrl.era==2012)                 pass = kFALSE;
-+
+  if(mu->GlobalTrk()->RChi2() > 10)                     pass = kFALSE;
-+
+  if(mu->NValidHits() == 0 )                            pass = kFALSE;
-+
+  if(mu->NMatches() <= 1)                               pass = kFALSE;
-+
+  if(fabs(mu->TrackerTrk()->D0Corrected(*vtx)) > 0.2)   pass = kFALSE;
-+
+  if(fabs(mu->TrackerTrk()->DzCorrected(*vtx)) > 0.5)   pass = kFALSE;
-+
+  if(mu->BestTrk()->NPixelHits() == 0)                  pass = kFALSE;
-+
+  if(mu->NTrkLayersHit() <= 5)                          pass = kFALSE;
-+
+  if(iso_sum/mu->Pt() > 0.12 )                          pass = kFALSE;
-+
-+
+  return pass;
-+
-+
-+
+ }
-+
-+
+bool trackToPFMuonSelection( const mithep::Track *track, const mithep::MuonCol * muoncol )
-+
+{
-+
+        bool pass = false;
-+
+  for(UInt_t i=0; i<muoncol->GetEntries(); ++i) {
-+
+    const mithep::Muon *mu = muoncol->At(i);
-+
+                if(mu->TrackerTrk() == track && (mu->IsGlobalMuon() || mu->IsTrackerMuon()) && mu->IsPFMuon())
-+
+                        pass = true;
-+
+        }
-+
-+
+        return pass;
-+
-+
+}
-+
+//--------------------------------------------------------------------------------------------------
+SelectionStatus muonDummyVeto(ControlFlags &ctrl,
+                              const mithep::Muon *muon,
+  if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;
+  ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
+  if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
-<
+        ret &= (mu->HasTrackerTrk() && mu->TrackerTrk()->D0Corrected(*vtx) < 0.5 && mu->TrackerTrk()->DzCorrected(*vtx) < 1.0);
->
+  ret &= (mu->HasTrackerTrk() &&
->
+          fabs(mu->TrackerTrk()->D0Corrected(*vtx)) < 0.5 &&
->
+          fabs(mu->TrackerTrk()->DzCorrected(*vtx)) < 1.0);
+  if(ctrl.debug) cout << "d0 and dz? ... " << ret << endl;
+  //  if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
+  if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
+  return status;
-+
+//--------------------------------------------------------------------------------------------------
-+
+SelectionStatus muonPreSelectionNoD0DzIP( ControlFlags &ctrl,
-+
+                                          const mithep::Muon * mu,
-+
+                                          const mithep::Vertex * vtx,
-+
+                                          const mithep::Array<mithep::PFCandidate> * pfCandidates )
-+
+//--------------------------------------------------------------------------------------------------
-+
+{
-+
+  bool ret = true;
-+
+  if(ctrl.debug) cout << "inside muonpresel ... " << endl;
-+
+  ret &= ( mu->Pt() >= 5 );
-+
+  if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;
-+
+  ret &= ( fabs(mu->Eta()) <= 2.4 );
-+
+  if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;
-+
+  ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
-+
+  if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
-+
-+
+  //  if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
-+
+  //    ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
-+
+  //  if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;
-+
+  //  ret &= ( mu->BestTrk()->DzCorrected(vtx) < 0.1);
-+
+  //  if( ctrl.debug ) cout << "elepresel : ret after dZcorr : " << ret <<  endl;
-+
+  SelectionStatus status;
-+
+  if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
-+
+  if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
-+
+  return status;
-+
+}
-+
+//--------------------------------------------------------------------------------------------------
-+
+SelectionStatus muonPreSelectionNoD0IP( ControlFlags &ctrl,
-+
+                                          const mithep::Muon * mu,
-+
+                                          const mithep::Vertex * vtx,
-+
+                                          const mithep::Array<mithep::PFCandidate> * pfCandidates )
-+
+//--------------------------------------------------------------------------------------------------
-+
+{
-+
+  bool ret = true;
-+
+  ret &= ( mu->Pt() >= 5 );
-+
+  ret &= ( fabs(mu->Eta()) <= 2.4 );
-+
+  ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
-+
+  // ret &= (mu->HasTrackerTrk() &&
-+
+  //      fabs(mu->TrackerTrk()->D0Corrected(*vtx)) < 0.5 &&
-+
+  //      fabs(mu->TrackerTrk()->DzCorrected(*vtx)) < 1.0);
-+
+  ret &= (mu->HasTrackerTrk() &&
-+
+          fabs(mu->TrackerTrk()->DzCorrected(*vtx)) < 1.0);
-+
-+
+  SelectionStatus status;
-+
+  if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
-+
+  if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
-+
+  return status;
-+
+}
+//--------------------------------------------------------------------------------------------------
+SelectionStatus muonIDMVASelection(ControlFlags &ctrl,
+                                 const mithep::Muon *mu,
+//      if(mu->IsPFMuon()) pass = true;
+//      else pass = false;
-–
+  //
-–
+  // NB : ipsig here for reference synch ...
-–
+  //
-–
+#ifdef SYNC
-–
+  if( pass && fabs(mu->Ip3dPVSignificance()) < 4 ) {
-–
+    status.orStatus(SelectionStatus::LOOSEID);
-–
+    status.orStatus(SelectionStatus::TIGHTID);
-–
+  }
-–
+#else
+  if( pass ) {
+    status.orStatus(SelectionStatus::LOOSEID);
+    status.orStatus(SelectionStatus::TIGHTID);
-–
+#endif
+  return status;
-+
+//----------------------------------------------------------------------------------------
-+
+SelectionStatus PassWwMuonSel(const mithep::Muon *mu, const mithep::Vertex *vtx,
-+
+                              const mithep::Array<mithep::PFCandidate> *pfArr
-+
+)
-+
+{
-+
+  bool passID=true;
-+
+  bool passIso=true;
-+
-+
+  if (mu->Pt() < 5)
-+
+    passID = false;
-+
+  if (fabs(mu->Eta()) > 2.4)
-+
+    passID = false;
-+
+  if (mu->BestTrk()->PtErr()/mu->Pt() > 0.1)
-+
+    passID = false;
-+
+  if (fabs(mu->BestTrk()->DzCorrected(*vtx)) > 0.1)
-+
+    passID = false;
-+
+  Bool_t isGlobal  = (mu->IsGlobalMuon()) && (mu->BestTrk()->RChi2() < 10) &&
-+
+                      (mu->NMatches() > 1) && (mu->NValidHits() > 0);
-+
+  Bool_t isTracker = (mu->IsTrackerMuon()) &&
-+
+                      (mu->Quality().Quality(mithep::MuonQuality::TMLastStationTight));
-+
+  if (!isGlobal && !isTracker)
-+
+    passID = false;
-+
+  int ntrkhits = (mu->HasTrackerTrk()) ? mu->TrackerTrk()->NHits() : 0;
-+
+  if (ntrkhits < 10)
-+
+    passID = false;
-+
+  if (mu->BestTrk()->NPixelHits() < 1)
-+
+    passID = false;
-+
+  if (fabs(mu->BestTrk()->D0Corrected(*vtx)) > 0.02)
-+
+    passID = false;
-+
-+
+  // note: this isn't really ww muon isolation
-+
+  float pfiso = computePFMuonIso(mu,vtx,pfArr,0.3);
-+
-+
+  if (pfiso > 0.2*mu->Pt())
-+
+    passIso = false;
-+
-+
+  SelectionStatus status;
-+
+  if(passID)  status |= SelectionStatus::TIGHTIDANDPRE;
-+
+  if(passIso) status |= SelectionStatus::TIGHTISOANDPRE;
-+
-+
+  return status;
-+
-+
+}

Diff Legend

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

Comparing UserCode/MitHzz4l/LeptonSelection/src/MuonSelection.cc (file contents): Revision 1.22 by anlevin, Mon May 28 12:37:10 2012 UTC vs. Revision 1.35 by dkralph, Tue Jul 17 10:17:22 2012 UTC

Diff Legend

Comparing UserCode/MitHzz4l/LeptonSelection/src/MuonSelection.cc (file contents):
Revision 1.22 by anlevin, Mon May 28 12:37:10 2012 UTC vs.
Revision 1.35 by dkralph, Tue Jul 17 10:17:22 2012 UTC