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, const reco::JetCorrector *correctorL1=0, const reco::JetCorrector *correctorL1L2L3Res=0) 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 22 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,
		const reco::JetCorrector *	correctorL1 = 0,
		const reco::JetCorrector *	correctorL1L2L3Res = 0
	)		const

Definition at line 54 of file MuonMvaEstimator.cc.

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

{
  float var[kLast]{};

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

  double dB2D  = fabs(muon.dB(pat::Muon::BS2D));
  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 43 of file MuonMvaEstimator.h.

Referenced by computeMva().

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

private

Definition at line 42 of file MuonMvaEstimator.h.

Referenced by computeMva(), and MuonMvaEstimator().