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_
const std::vector< double >	scale_mean_
const std::vector< double >	scale_std_

Detailed Description

Definition at line 23 of file SeedMvaEstimator.h.

Constructor & Destructor Documentation

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(), and gbrForest_.

    : scale_mean_(scale_mean), scale_std_(scale_std), isFromL1_(isFromL1), minL1Qual_(minL1Qual) {
  gbrForest_ = createGBRForest(weightsfile);
}

SeedMvaEstimator::~SeedMvaEstimator

(

)

Definition at line 18 of file SeedMvaEstimator.cc.

{}

Member Function Documentation

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

Definition at line 72 of file SeedMvaEstimator.cc.

References LocalTrajectoryParameters::dxdz(), LocalTrajectoryParameters::dydz(), PTrajectoryStateOnDet::error(), gbrForest_, getL1MuonVariables(), getL2MuonVariables(), isFromL1_, gpuVertexFinder::iv, PTrajectoryStateOnDet::parameters(), LocalTrajectoryParameters::qbp(), scale_mean_, scale_std_, mathSSE::sqrt(), TrajectorySeed::startingState(), and ALCARECOEcalPhiSym_cff::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_.at(iv)) / scale_std_.at(iv);
  }

  return gbrForest_->GetResponse(var);
}

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

private

Definition at line 38 of file SeedMvaEstimator.cc.

References BXVector< T >::begin(), reco::deltaPhi(), reco::deltaR2(), BXVector< T >::end(), PV3DBase< T, PVType, FrameType >::eta(), BXVector< T >::getFirstBX(), BXVector< T >::getLastBX(), minL1Qual_, and PV3DBase< T, PVType, FrameType >::phi().

Referenced by computeMva().

                                                                      {
  for (int ibx = l1Muons.getFirstBX(); ibx <= l1Muons.getLastBX(); ++ibx) {
    if (ibx != 0)
      continue;  // -- only take when ibx == 0 -- //

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

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

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

private

Definition at line 59 of file SeedMvaEstimator.cc.

References reco::deltaPhi(), reco::deltaR2(), and PV3DBase< T, PVType, FrameType >::phi().

Referenced by computeMva().

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

Member Data Documentation

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

private

Definition at line 38 of file SeedMvaEstimator.h.

Referenced by computeMva(), and SeedMvaEstimator().

const bool SeedMvaEstimator::isFromL1_

private

Definition at line 41 of file SeedMvaEstimator.h.

Referenced by computeMva().

const int SeedMvaEstimator::minL1Qual_

private

Definition at line 42 of file SeedMvaEstimator.h.

Referenced by getL1MuonVariables().

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

private

Definition at line 39 of file SeedMvaEstimator.h.

Referenced by computeMva().

const std::vector<double> SeedMvaEstimator::scale_std_

private

Definition at line 40 of file SeedMvaEstimator.h.

Referenced by computeMva().