pat::SoftMuonMvaEstimator Class Reference

#include <SoftMuonMvaEstimator.h>

Public Member Functions
float	computeMva (const pat::Muon &imuon) const
	SoftMuonMvaEstimator (const edm::FileInPath &weightsfile)
	~SoftMuonMvaEstimator ()

Private Attributes
std::unique_ptr< const GBRForest >	gbrForest_

Detailed Description

Definition at line 18 of file SoftMuonMvaEstimator.h.

Constructor & Destructor Documentation

SoftMuonMvaEstimator::SoftMuonMvaEstimator

(

const edm::FileInPath &

weightsfile

)

Definition at line 12 of file SoftMuonMvaEstimator.cc.

References createGBRForest(), and gbrForest_.

{
  gbrForest_ = createGBRForest( weightsfile );
}

SoftMuonMvaEstimator::~SoftMuonMvaEstimator

(

)

Definition at line 17 of file SoftMuonMvaEstimator.cc.

{ }

Member Function Documentation

float SoftMuonMvaEstimator::computeMva

(

const pat::Muon &

imuon

)

const

Definition at line 43 of file SoftMuonMvaEstimator.cc.

References reco::MuonQuality::chi2LocalMomentum, reco::MuonQuality::chi2LocalPosition, reco::Muon::combinedQuality(), reco::LeafCandidate::eta(), gbrForest_, reco::HitPattern::getHitPattern(), reco::HitPattern::getMuonStation(), reco::MuonQuality::glbKink, reco::MuonQuality::glbTrackProbability, pat::Muon::globalTrack(), mps_fire::i, pat::Muon::innerTrack(), edm::Ref< C, T, F >::isNonnull(), JetComb::kEta, kPt, reco::HitPattern::muonCSCHitFilter(), reco::HitPattern::muonDTHitFilter(), reco::HitPattern::muonRPCHitFilter(), reco::HitPattern::numberOfAllHits(), pat::Muon::outerTrack(), reco::LeafCandidate::pt(), pat::Muon::segmentCompatibility(), relativeConstraints::station, reco::Muon::time(), reco::MuonTime::timeAtIpInOutErr, reco::HitPattern::TRACK_HITS, reco::MuonQuality::trkKink, reco::MuonQuality::trkRelChi2, reco::HitPattern::validHitFilter(), and JetChargeProducer_cfi::var.

{
    float var[kLast]{};

    reco::TrackRef gTrack = muon.globalTrack();
    reco::TrackRef iTrack = muon.innerTrack();
    reco::TrackRef oTrack = muon.outerTrack();

    if(!(muon.innerTrack().isNonnull() and
         muon.outerTrack().isNonnull() and
         muon.globalTrack().isNonnull()))
    {
      return -1;
    }

    //VARIABLE EXTRACTION
    var[kPt] = muon.pt();
    var[kEta] = muon.eta();
    var[kMomID] = -1;
    var[kPID] = -1;

    var[kChi2LocalMomentum] = muon.combinedQuality().chi2LocalMomentum;
    var[kChi2LocalPosition] = muon.combinedQuality().chi2LocalPosition;
    var[kGlbTrackProbability] =  muon.combinedQuality().glbTrackProbability;
    var[kTrkRelChi2] =  muon.combinedQuality().trkRelChi2;

    var[kTrkKink] = muon.combinedQuality().trkKink;
    var[kLog2PlusGlbKink] =  TMath::Log(2+muon.combinedQuality().glbKink);
    var[kSegmentCompatibility] = muon.segmentCompatibility();

    var[kTimeAtIpInOutErr] = muon.time().timeAtIpInOutErr;

    //TRACK RELATED VARIABLES
    
    var[kIValidFraction] =  iTrack->validFraction();
    var[kInnerChi2] =  iTrack->normalizedChi2();
    var[kLayersWithMeasurement] = iTrack->hitPattern().trackerLayersWithMeasurement();

    var[kOuterChi2] = oTrack->normalizedChi2();

    var[kQProd] =  iTrack->charge()*oTrack->charge();

    //vComb Calculation

    const reco::HitPattern &gMpattern = gTrack->hitPattern();

    std::vector<int> fvDThits {0,0,0,0};
    std::vector<int> fvRPChits {0,0,0,0};
    std::vector<int> fvCSChits {0,0,0,0};

    var[kVMuonHitComb] = 0;

    for (int i=0;i<gMpattern.numberOfAllHits(reco::HitPattern::TRACK_HITS);i++){

      uint32_t hit = gMpattern.getHitPattern(reco::HitPattern::TRACK_HITS, i);
      if ( !gMpattern.validHitFilter(hit) ) continue;

      int muStation0 = gMpattern.getMuonStation(hit) - 1;
      if (muStation0 >=0 && muStation0 < 4){
        if (gMpattern.muonDTHitFilter(hit)) fvDThits[muStation0]++;
        if (gMpattern.muonRPCHitFilter(hit)) fvRPChits[muStation0]++;
        if (gMpattern.muonCSCHitFilter(hit)) fvCSChits[muStation0]++;
      }

    }
  

    for (unsigned int station = 0; station < 4; ++station) {

      var[kVMuonHitComb] += (fvDThits[station])/2.;
      var[kVMuonHitComb] += fvRPChits[station];

      if (fvCSChits[station] > 6){
        var[kVMuonHitComb] += 6;
      }else{
        var[kVMuonHitComb] += fvCSChits[station];
      }

    }

    if(var[kChi2LocalMomentum] < 5000 and var[kChi2LocalPosition] < 2000 and
       var[kGlbTrackProbability] < 5000 and var[kTrkKink] < 900 and
       var[kLog2PlusGlbKink] < 50 and var[kTimeAtIpInOutErr] < 4 and
       var[kOuterChi2] < 1000 and var[kInnerChi2] < 10 and var[kTrkRelChi2] < 3)
    {
      return gbrForest_->GetAdaBoostClassifier(var);
    } else {
      return -1;
        }
}

Member Data Documentation

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

private

Definition at line 29 of file SoftMuonMvaEstimator.h.

Referenced by computeMva(), and SoftMuonMvaEstimator().