MuonMvaEstimator.h

Go to the documentation of this file.

#ifndef __PhysicsTools_PatAlgos_MuonMvaEstimator__
#define __PhysicsTools_PatAlgos_MuonMvaEstimator__
 
#include "DataFormats/BTauReco/interface/JetTag.h"
 
#include <memory>
#include <string>
 
class GBRForest;
 
namespace pat {
  class Muon;
}
 
namespace reco {
  class JetCorrector;
  class Vertex;
}  // namespace reco
 
namespace edm {
  class FileInPath;
}
 
namespace pat {
  class MuonMvaEstimator {
  public:
    MuonMvaEstimator(const edm::FileInPath& weightsfile, float dRmax);
 
    ~MuonMvaEstimator();
 
    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;
 
  private:
    std::unique_ptr<const GBRForest> gbrForest_;
    float dRmax_;
  };
}  // namespace pat
#endif