SeedMvaEstimator Class Reference

#include <SeedMvaEstimator.h>

Public Member Functions
double	computeMva (const TrajectorySeed &, const GlobalVector &, const l1t::MuonBxCollection &, const reco::RecoChargedCandidateCollection &) const
	SeedMvaEstimator (const edm::FileInPath &weightsfile, const std::vector< double > &scale_mean, const std::vector< double > &scale_std, const bool isFromL1, const int minL1Qual)
	~SeedMvaEstimator ()

Private Member Functions
void	getL1MuonVariables (const GlobalVector &, const l1t::MuonBxCollection &, float &, float &) const
void	getL2MuonVariables (const GlobalVector &, const reco::RecoChargedCandidateCollection &, float &, float &) const

Private Attributes
std::unique_ptr< const GBRForest >	gbrForest_
const bool	isFromL1_
const int	minL1Qual_
std::vector< float >	scale_istd_
const std::vector< float >	scale_mean_

Detailed Description

Definition at line 41 of file SeedMvaEstimator.h.

Constructor & Destructor Documentation

SeedMvaEstimator()

SeedMvaEstimator::SeedMvaEstimator	(	const edm::FileInPath &	weightsfile,
		const std::vector< double > &	scale_mean,
		const std::vector< double > &	scale_std,
		const bool	isFromL1,
		const int	minL1Qual
	)

Definition at line 9 of file SeedMvaEstimator.cc.

References createGBRForest(), gbrForest_, scale_istd_, and findQualityFiles::v.

    : scale_mean_(scale_mean.begin(), scale_mean.end()), isFromL1_(isFromL1), minL1Qual_(minL1Qual) {
  scale_istd_.reserve(scale_std.size());
  for (auto v : scale_std)
    scale_istd_.push_back(1. / v);
  gbrForest_ = createGBRForest(weightsfile);
}

~SeedMvaEstimator()

SeedMvaEstimator::~SeedMvaEstimator

(

)

Definition at line 21 of file SeedMvaEstimator.cc.

{}

Member Function Documentation

computeMva()

double SeedMvaEstimator::computeMva	(	const TrajectorySeed &	seed,
		const GlobalVector &	global_p,
		const l1t::MuonBxCollection &	l1Muons,
		const reco::RecoChargedCandidateCollection &	l2Muons
	)		const

Definition at line 58 of file SeedMvaEstimator.cc.

References ALPAKA_ACCELERATOR_NAMESPACE::brokenline::constexpr(), gbrForest_, getL1MuonVariables(), getL2MuonVariables(), isFromL1_, ALPAKA_ACCELERATOR_NAMESPACE::vertexFinder::iv, Phase2::kLast, Phase2::kTsosDxdz, Phase2::kTsosDydz, Phase2::kTsosErr0, Phase2::kTsosErr2, Phase2::kTsosErr5, Phase2::kTsosQbp, scale_istd_, scale_mean_, fileCollector::seed, mathSSE::sqrt(), and trigObjTnPSource_cfi::var.

                                                                                             {
  static constexpr float initDRdPhi(99999.);
  auto kLast = isFromL1_ ? kLastL1 : kLastL2;
  float var[kLast];

  var[kTsosErr0] = seed.startingState().error(0);
  var[kTsosErr2] = seed.startingState().error(2);
  var[kTsosErr5] = seed.startingState().error(5);
  var[kTsosDxdz] = seed.startingState().parameters().dxdz();
  var[kTsosDydz] = seed.startingState().parameters().dydz();
  var[kTsosQbp] = seed.startingState().parameters().qbp();

  float dR2dRL1SeedP = initDRdPhi;
  float dPhidRL1SeedP = initDRdPhi;
  getL1MuonVariables(global_p, l1Muons, dR2dRL1SeedP, dPhidRL1SeedP);

  var[kDRdRL1SeedP] = std::sqrt(dR2dRL1SeedP);
  var[kDPhidRL1SeedP] = dPhidRL1SeedP;

  if (!isFromL1_) {
    float dR2dRL2SeedP = initDRdPhi;
    float dPhidRL2SeedP = initDRdPhi;
    getL2MuonVariables(global_p, l2Muons, dR2dRL2SeedP, dPhidRL2SeedP);

    var[kDRdRL2SeedP] = std::sqrt(dR2dRL2SeedP);
    var[kDPhidRL2SeedP] = dPhidRL2SeedP;
  }

  for (int iv = 0; iv < kLast; ++iv) {
    var[iv] = (var[iv] - scale_mean_[iv]) * scale_istd_[iv];
  }

  return gbrForest_->GetResponse(var);
}

getL1MuonVariables()

void SeedMvaEstimator::getL1MuonVariables ( const GlobalVector &  global_p, const l1t::MuonBxCollection &  l1Muons, float &  dR2dRL1SeedP, float &  dPhidRL1SeedP  ) const

private

Definition at line 23 of file SeedMvaEstimator.cc.

References PV3DBase< T, PVType, FrameType >::barePhi(), BXVector< T >::begin(), reco::deltaPhi(), reco::deltaR2(), BXVector< T >::end(), PVValHelper::eta, PV3DBase< T, PVType, FrameType >::eta(), ALPAKA_ACCELERATOR_NAMESPACE::vertexFinder::it, minL1Qual_, and phi.

Referenced by computeMva().

                                                                      {
  int ibx = 0;  // -- only take when ibx == 0 -- //

  auto eta = global_p.eta();
  auto phi = global_p.barePhi();
  for (auto it = l1Muons.begin(ibx); it != l1Muons.end(ibx); it++) {
    if (it->hwQual() < minL1Qual_)
      continue;

    float dR2tmp = reco::deltaR2(it->etaAtVtx(), it->phiAtVtx(), eta, phi);
    if (dR2tmp < dR2dRL1SeedP) {
      dR2dRL1SeedP = dR2tmp;
      dPhidRL1SeedP = reco::deltaPhi(it->phiAtVtx(), phi);
    }
  }
}

getL2MuonVariables()

void SeedMvaEstimator::getL2MuonVariables ( const GlobalVector &  global_p, const reco::RecoChargedCandidateCollection &  l2Muons, float &  dR2dRL2SeedP, float &  dPhidRL2SeedP  ) const

private

Definition at line 43 of file SeedMvaEstimator.cc.

References PV3DBase< T, PVType, FrameType >::barePhi(), reco::deltaPhi(), reco::deltaR2(), PVValHelper::eta, PV3DBase< T, PVType, FrameType >::eta(), ALPAKA_ACCELERATOR_NAMESPACE::vertexFinder::it, and phi.

Referenced by computeMva().

                                                                      {
  auto eta = global_p.eta();
  auto phi = global_p.barePhi();
  for (auto it = l2Muons.begin(); it != l2Muons.end(); it++) {
    float dR2tmp = reco::deltaR2(it->eta(), it->phi(), eta, phi);
    if (dR2tmp < dR2dRL2SeedP) {
      dR2dRL2SeedP = dR2tmp;
      dPhidRL2SeedP = reco::deltaPhi(it->phi(), phi);
    }
  }
}

Member Data Documentation

gbrForest_

std::unique_ptr<const GBRForest> SeedMvaEstimator::gbrForest_

private

Definition at line 56 of file SeedMvaEstimator.h.

Referenced by computeMva(), and SeedMvaEstimator().

isFromL1_

const bool SeedMvaEstimator::isFromL1_

private

Definition at line 59 of file SeedMvaEstimator.h.

Referenced by computeMva().

minL1Qual_

const int SeedMvaEstimator::minL1Qual_

private

Definition at line 60 of file SeedMvaEstimator.h.

Referenced by getL1MuonVariables().

scale_istd_

std::vector<float> SeedMvaEstimator::scale_istd_

private

Definition at line 58 of file SeedMvaEstimator.h.

Referenced by computeMva(), and SeedMvaEstimator().

scale_mean_

const std::vector<float> SeedMvaEstimator::scale_mean_

private

Definition at line 57 of file SeedMvaEstimator.h.

Referenced by computeMva().