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 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_;

  };
}
#endif