SeedMvaEstimator.h

Go to the documentation of this file.

#ifndef RecoMuon_TrackerSeedGenerator_SeedMvaEstimator_h
#define RecoMuon_TrackerSeedGenerator_SeedMvaEstimator_h
 
#include "DataFormats/TrajectorySeed/interface/TrajectorySeed.h"
#include "DataFormats/TrajectorySeed/interface/TrajectorySeedCollection.h"
#include "DataFormats/TrajectorySeed/interface/PropagationDirection.h"
#include "DataFormats/TrajectoryState/interface/PTrajectoryStateOnDet.h"
#include "DataFormats/TrajectoryState/interface/LocalTrajectoryParameters.h"
 
#include "DataFormats/L1Trigger/interface/Muon.h"
 
#include "DataFormats/RecoCandidate/interface/RecoChargedCandidate.h"
#include "DataFormats/RecoCandidate/interface/RecoChargedCandidateFwd.h"
 
#include <memory>
 
class GBRForest;
 
namespace edm {
  class FileInPath;
}
 
class SeedMvaEstimator {
public:
  SeedMvaEstimator(const edm::FileInPath& weightsfile,
                   const std::vector<double>& scale_mean,
                   const std::vector<double>& scale_std,
                   const bool isFromL1,
                   const int minL1Qual);
  ~SeedMvaEstimator();
 
  double computeMva(const TrajectorySeed&,
                    const GlobalVector&,
                    const l1t::MuonBxCollection&,
                    const reco::RecoChargedCandidateCollection&) const;
 
private:
  std::unique_ptr<const GBRForest> gbrForest_;
  const std::vector<double> scale_mean_;
  const std::vector<double> scale_std_;
  const bool isFromL1_;
  const int minL1Qual_;
 
  void getL1MuonVariables(const GlobalVector&, const l1t::MuonBxCollection&, float&, float&) const;
  void getL2MuonVariables(const GlobalVector&, const reco::RecoChargedCandidateCollection&, float&, float&) const;
};
#endif