#include <SoftMuonMvaEstimator.h>

Public Member Functions
void	computeMva (const pat::Muon &imuon)

void	initialize (std::string weightsfile)

float	mva () const

	SoftMuonMvaEstimator ()

Private Attributes
float	chi2LocalMomentum_

float	chi2LocalPosition_

float	dummy_

float	eta_

float	glbTrackProbability_

bool	initialized_

float	innerChi2_

float	iValidFraction_

float	layersWithMeasurement_

float	log2PlusGlbKink_

float	momID_

float	mva_

float	outerChi2_

float	pID_

float	pt_

float	qProd_

float	segmentCompatibility_

float	timeAtIpInOutErr_

TMVA::Reader	tmvaReader_

float	trkKink_

float	trkRelChi2_

float	vMuonHitComb_

Detailed Description

Definition at line 8 of file SoftMuonMvaEstimator.h.

Constructor & Destructor Documentation

SoftMuonMvaEstimator::SoftMuonMvaEstimator ( )

Definition at line 7 of file SoftMuonMvaEstimator.cc.

                                           :
     tmvaReader_("!Color:!Silent:Error"),
     initialized_(false),
     mva_(0)
 {}

Member Function Documentation

void SoftMuonMvaEstimator::computeMva ( const pat::Muon & imuon )

Definition at line 40 of file SoftMuonMvaEstimator.cc.

References reco::MuonQuality::chi2LocalMomentum, chi2LocalMomentum_, reco::MuonQuality::chi2LocalPosition, chi2LocalPosition_, reco::Muon::combinedQuality(), dummy_, reco::LeafCandidate::eta(), eta_, Exception, reco::HitPattern::getHitPattern(), reco::HitPattern::getMuonStation(), reco::MuonQuality::glbKink, reco::MuonQuality::glbTrackProbability, glbTrackProbability_, pat::Muon::globalTrack(), mps_fire::i, initialized_, innerChi2_, pat::Muon::innerTrack(), edm::Ref< C, T, F >::isNonnull(), iValidFraction_, layersWithMeasurement_, log2PlusGlbKink_, momID_, reco::HitPattern::muonCSCHitFilter(), reco::HitPattern::muonDTHitFilter(), reco::HitPattern::muonRPCHitFilter(), mva_, reco::HitPattern::numberOfAllHits(), outerChi2_, pat::Muon::outerTrack(), pID_, reco::LeafCandidate::pt(), pt_, qProd_, pat::Muon::segmentCompatibility(), segmentCompatibility_, relativeConstraints::station, reco::Muon::time(), reco::MuonTime::timeAtIpInOutErr, timeAtIpInOutErr_, tmvaReader_, reco::HitPattern::TRACK_HITS, reco::MuonQuality::trkKink, trkKink_, reco::MuonQuality::trkRelChi2, trkRelChi2_, reco::HitPattern::validHitFilter(), and vMuonHitComb_.

Referenced by pat::PATMuonProducer::produce().

 {
     if (not initialized_)
         throw cms::Exception("FatalError") << "SoftMuonMVA is not initialized";
 
     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()))
     {
       mva_ = -1; 
       return;
     }
 
     //VARIABLE EXTRACTION
     pt_ = muon.pt();
     eta_ = muon.eta();
     dummy_ = -1;
     momID_ = dummy_;
     pID_ = dummy_;
 
 
     chi2LocalMomentum_ = muon.combinedQuality().chi2LocalMomentum;
     chi2LocalPosition_ = muon.combinedQuality().chi2LocalPosition;
     glbTrackProbability_ =  muon.combinedQuality().glbTrackProbability;
     trkRelChi2_ =  muon.combinedQuality().trkRelChi2;
 
     trkKink_ = muon.combinedQuality().trkKink;
     log2PlusGlbKink_ =  TMath::Log(2+muon.combinedQuality().glbKink);
     segmentCompatibility_ = muon.segmentCompatibility();
 
     timeAtIpInOutErr_ = muon.time().timeAtIpInOutErr;
 
     //TRACK RELATED VARIABLES
     
     iValidFraction_ =  iTrack->validFraction();
     innerChi2_ =  iTrack->normalizedChi2();
     layersWithMeasurement_ = iTrack->hitPattern().trackerLayersWithMeasurement();
 
     outerChi2_ = oTrack->normalizedChi2();
 
     qProd_ =  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};
 
     vMuonHitComb_ = 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) {
 
       vMuonHitComb_ += (fvDThits[station])/2.;
       vMuonHitComb_ += fvRPChits[station];
 
       if (fvCSChits[station] > 6){
         vMuonHitComb_ += 6; 
       }else{
         vMuonHitComb_ += fvCSChits[station];
       }
 
     }
 
     if(chi2LocalMomentum_ < 5000 and chi2LocalPosition_ < 2000 and
        glbTrackProbability_ < 5000 and trkKink_ < 900 and
        log2PlusGlbKink_ < 50 and timeAtIpInOutErr_ < 4 and
        outerChi2_ < 1000 and innerChi2_ < 10 and trkRelChi2_ < 3) 
     {
       mva_ = tmvaReader_.EvaluateMVA("BDT");
     }else{ mva_ = -1; }
 
 }

void SoftMuonMvaEstimator::initialize ( std::string weightsfile )

Definition at line 13 of file SoftMuonMvaEstimator.cc.

References chi2LocalMomentum_, chi2LocalPosition_, eta_, glbTrackProbability_, initialized_, innerChi2_, iValidFraction_, layersWithMeasurement_, log2PlusGlbKink_, momID_, outerChi2_, pID_, pt_, qProd_, segmentCompatibility_, timeAtIpInOutErr_, tmvaReader_, trkKink_, trkRelChi2_, and vMuonHitComb_.

Referenced by pat::PATMuonProducer::PATMuonProducer().

 {
     if (initialized_) return;
     tmvaReader_.AddVariable("segComp",          &segmentCompatibility_);
     tmvaReader_.AddVariable("chi2LocMom",           &chi2LocalMomentum_);
     tmvaReader_.AddVariable("chi2LocPos",           &chi2LocalPosition_);
     tmvaReader_.AddVariable("glbTrackTailProb",     &glbTrackProbability_);
     tmvaReader_.AddVariable("iValFrac",         &iValidFraction_);
     tmvaReader_.AddVariable("LWH",              &layersWithMeasurement_);
     tmvaReader_.AddVariable("kinkFinder",           &trkKink_);
     tmvaReader_.AddVariable("TMath::Log(2+glbKinkFinder)",  &log2PlusGlbKink_);
     tmvaReader_.AddVariable("timeAtIpInOutErr",     &timeAtIpInOutErr_);
     tmvaReader_.AddVariable("outerChi2",            &outerChi2_);
     tmvaReader_.AddVariable("innerChi2",            &innerChi2_);
     tmvaReader_.AddVariable("trkRelChi2",           &trkRelChi2_);
     tmvaReader_.AddVariable("vMuonHitComb",         &vMuonHitComb_);
     tmvaReader_.AddVariable("Qprod",            &qProd_);
 
     tmvaReader_.AddSpectator("pID",                 &pID_);
     tmvaReader_.AddSpectator("pt",              &pt_);
     tmvaReader_.AddSpectator("eta",             &eta_);
     tmvaReader_.AddSpectator("MomID",           &momID_);
 
     tmvaReader_.BookMVA("BDT", weightsfile);
     initialized_ = true;    
 }