pat::MuonMvaEstimator Class Reference

#include <MuonMvaEstimator.h>

Public Member Functions
float	computeMva (const pat::Muon &imuon, const reco::Vertex &vertex, const reco::JetTagCollection &bTags, float &jetPtRatio, float &jetPtRel, float &miniIsoValue, const reco::JetCorrector *correctorL1=nullptr, const reco::JetCorrector *correctorL1L2L3Res=nullptr) const
	MuonMvaEstimator (const edm::FileInPath &weightsfile, float dRmax)
	~MuonMvaEstimator ()

Private Attributes
float	dRmax_
std::unique_ptr< const GBRForest >	gbrForest_

Detailed Description

Definition at line 25 of file MuonMvaEstimator.h.

Constructor & Destructor Documentation

MuonMvaEstimator::MuonMvaEstimator	(	const edm::FileInPath &	weightsfile,
		float	dRmax
	)

Definition at line 16 of file MuonMvaEstimator.cc.

References createGBRForest(), and gbrForest_.

                                                                                : dRmax_(dRmax) {
  gbrForest_ = createGBRForest(weightsfile);
}

MuonMvaEstimator::~MuonMvaEstimator

(

)

Definition at line 20 of file MuonMvaEstimator.cc.

{}

Member Function Documentation

float MuonMvaEstimator::computeMva	(	const pat::Muon &	imuon,
		const reco::Vertex &	vertex,
		const reco::JetTagCollection &	bTags,
		float &	jetPtRatio,
		float &	jetPtRel,
		float &	miniIsoValue,
		const reco::JetCorrector *	correctorL1 = nullptr,
		const reco::JetCorrector *	correctorL1L2L3Res = nullptr
	)		const

Definition at line 53 of file MuonMvaEstimator.cc.

References reco::PFCandidate::bestTrack(), reco::LeafCandidate::charge(), pat::PFIsolation::chargedHadronIso(), reco::JetCorrector::correction(), pat::Muon::dB(), HLT_FULL_cff::deltaR, runTauDisplay::dr, dRmax_, PVValHelper::dz, pat::Muon::edB(), reco::LeafCandidate::eta(), Exception, gbrForest_, reco::TrackBase::highPurity, metsig::jet, JetComb::kEta, kPt, log, SiStripPI::max, pat::Lepton< LeptonType >::miniPFIsolation(), pat::Muon::muonBestTrack(), reco::RecoCandidate::p4, reco::Muon::pfIsolationR04(), reco::Vertex::position(), reco::LeafCandidate::pt(), reco::TrackBase::pt(), pat::Muon::PV2D, pat::Muon::PV3D, pat::Muon::segmentCompatibility(), reco::MuonPFIsolation::sumChargedHadronPt, reco::MuonPFIsolation::sumNeutralHadronEt, reco::MuonPFIsolation::sumPhotonEt, reco::MuonPFIsolation::sumPUPt, and isotrackApplyRegressor::var.

                                                                                     {
  float var[kLast]{};

  var[kPt] = muon.pt();
  var[kEta] = muon.eta();
  var[kSegmentCompatibility] = muon.segmentCompatibility();
  var[kMiniRelIsoCharged] = muon.miniPFIsolation().chargedHadronIso() / muon.pt();
  var[kMiniRelIsoNeutral] = miniIsoValue - var[kMiniRelIsoCharged];

  double dB2D = fabs(muon.dB(pat::Muon::PV2D));
  double dB3D = muon.dB(pat::Muon::PV3D);
  double edB3D = muon.edB(pat::Muon::PV3D);
  double dz = fabs(muon.muonBestTrack()->dz(vertex.position()));
  var[kSip] = edB3D > 0 ? fabs(dB3D / edB3D) : 0.0;
  var[kLog_abs_dxyBS] = dB2D > 0 ? log(dB2D) : 0;
  var[kLog_abs_dzPV] = dz > 0 ? log(dz) : 0;

  //Initialise loop variables
  double minDr = 9999;
  double jecL1L2L3Res = 1.;
  double jecL1 = 1.;

  // Compute corrected isolation variables
  double chIso = muon.pfIsolationR04().sumChargedHadronPt;
  double nIso = muon.pfIsolationR04().sumNeutralHadronEt;
  double phoIso = muon.pfIsolationR04().sumPhotonEt;
  double puIso = muon.pfIsolationR04().sumPUPt;
  double dbCorrectedIsolation = chIso + std::max(nIso + phoIso - .5 * puIso, 0.);
  double dbCorrectedRelIso = dbCorrectedIsolation / muon.pt();

  var[kJetPtRatio] = 1. / (1 + dbCorrectedRelIso);
  var[kJetPtRel] = 0;
  var[kJetBTagCSV] = -999;
  var[kJetNDauCharged] = -1;

  for (const auto& tagI : bTags) {
    // for each muon with the lepton
    double dr = deltaR(*(tagI.first), muon);
    if (dr > minDr)
      continue;
    minDr = dr;

    const reco::Candidate::LorentzVector& muP4(muon.p4());
    reco::Candidate::LorentzVector jetP4(tagI.first->p4());

    if (correctorL1 && correctorL1L2L3Res) {
      jecL1L2L3Res = correctorL1L2L3Res->correction(*(tagI.first));
      jecL1 = correctorL1->correction(*(tagI.first));
    }

    // Get b-jet info
    var[kJetBTagCSV] = tagI.second;
    var[kJetNDauCharged] = 0;
    for (auto jet : tagI.first->getJetConstituentsQuick()) {
      const reco::PFCandidate* pfcand = dynamic_cast<const reco::PFCandidate*>(jet);
      if (pfcand == nullptr)
        throw cms::Exception("ConfigurationError") << "Cannot get jet constituents";
      if (pfcand->charge() == 0)
        continue;
      auto bestTrackPtr = pfcand->bestTrack();
      if (!bestTrackPtr)
        continue;
      if (!bestTrackPtr->quality(reco::Track::highPurity))
        continue;
      if (bestTrackPtr->pt() < 1.)
        continue;
      if (bestTrackPtr->hitPattern().numberOfValidHits() < 8)
        continue;
      if (bestTrackPtr->hitPattern().numberOfValidPixelHits() < 2)
        continue;
      if (bestTrackPtr->normalizedChi2() >= 5)
        continue;

      if (std::fabs(bestTrackPtr->dxy(vertex.position())) > 0.2)
        continue;
      if (std::fabs(bestTrackPtr->dz(vertex.position())) > 17)
        continue;
      var[kJetNDauCharged]++;
    }

    if (minDr < dRmax_) {
      if ((jetP4 - muP4).Rho() < 0.0001) {
        var[kJetPtRel] = 0;
        var[kJetPtRatio] = 1;
      } else {
        jetP4 -= muP4 / jecL1;
        jetP4 *= jecL1L2L3Res;
        jetP4 += muP4;

        var[kJetPtRatio] = muP4.pt() / jetP4.pt();
        var[kJetPtRel] = ptRel(muP4, jetP4);
      }
    }
  }

  if (var[kJetPtRatio] > 1.5)
    var[kJetPtRatio] = 1.5;
  if (var[kJetBTagCSV] < 0)
    var[kJetBTagCSV] = 0;
  jetPtRatio = var[kJetPtRatio];
  jetPtRel = var[kJetPtRel];
  return gbrForest_->GetClassifier(var);
};

Member Data Documentation

float pat::MuonMvaEstimator::dRmax_

private

Definition at line 42 of file MuonMvaEstimator.h.

Referenced by computeMva().

std::unique_ptr<const GBRForest> pat::MuonMvaEstimator::gbrForest_

private

Definition at line 41 of file MuonMvaEstimator.h.

Referenced by computeMva(), and MuonMvaEstimator().