TrackMVAClassifierBase Class Reference

#include <TrackMVAClassifier.h>

Inheritance diagram for TrackMVAClassifierBase:

Public Types
using	MVACollection = std::vector< float >
using	MVAPairCollection = std::vector< std::pair< float, bool > >
using	QualityMaskCollection = std::vector< unsigned char >
Public Types inherited from edm::stream::EDProducer<>
using	CacheTypes = CacheContexts< T... >
using	GlobalCache = typename CacheTypes::GlobalCache
using	HasAbility = AbilityChecker< T... >
using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache
using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache
using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >
using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache
using	RunCache = typename CacheTypes::RunCache
using	RunContext = RunContextT< RunCache, GlobalCache >
using	RunSummaryCache = typename CacheTypes::RunSummaryCache

Public Member Functions
	TrackMVAClassifierBase (const edm::ParameterSet &cfg)
	~TrackMVAClassifierBase () override
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
	EDProducer (const EDProducer &)=delete
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginProcessBlocks () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndProcessBlocks () const final
bool	hasAbilityToProduceInEndRuns () const final
const EDProducer &	operator= (const EDProducer &)=delete

Protected Member Functions
virtual void	computeMVA (reco::TrackCollection const &tracks, reco::BeamSpot const &beamSpot, reco::VertexCollection const &vertices, MVAPairCollection &mvas) const =0
virtual void	initEvent (const edm::EventSetup &es)=0

Static Protected Member Functions
static void	fill (edm::ParameterSetDescription &desc)

Private Member Functions
void	produce (edm::Event &evt, const edm::EventSetup &es) final

Private Attributes
edm::EDGetTokenT< reco::BeamSpot >	beamspot_
bool	ignoreVertices_
float	qualityCuts [3]
edm::EDGetTokenT< reco::TrackCollection >	src_
	source collection label More...
edm::EDGetTokenT< reco::VertexCollection >	vertices_

Detailed Description

Definition at line 23 of file TrackMVAClassifier.h.

Member Typedef Documentation

MVACollection

using TrackMVAClassifierBase::MVACollection = std::vector<float>

Definition at line 28 of file TrackMVAClassifier.h.

MVAPairCollection

using TrackMVAClassifierBase::MVAPairCollection = std::vector<std::pair<float, bool> >

Definition at line 32 of file TrackMVAClassifier.h.

QualityMaskCollection

using TrackMVAClassifierBase::QualityMaskCollection = std::vector<unsigned char>

Definition at line 29 of file TrackMVAClassifier.h.

Constructor & Destructor Documentation

TrackMVAClassifierBase()

TrackMVAClassifierBase::TrackMVAClassifierBase ( const edm::ParameterSet &  cfg )

explicit

Definition at line 22 of file TrackMVAClassifierBase.cc.

References cms::cuda::assert(), looper::cfg, filterCSVwithJSON::copy, and qualityCuts.

    : src_(consumes<reco::TrackCollection>(cfg.getParameter<edm::InputTag>("src"))),
      beamspot_(consumes<reco::BeamSpot>(cfg.getParameter<edm::InputTag>("beamspot"))),
      vertices_(mayConsume<reco::VertexCollection>(cfg.getParameter<edm::InputTag>("vertices"))),
      ignoreVertices_(cfg.getParameter<bool>("ignoreVertices")) {
  auto const& qv = cfg.getParameter<std::vector<double>>("qualityCuts");
  assert(qv.size() == 3);
  std::copy(std::begin(qv), std::end(qv), std::begin(qualityCuts));

  produces<MVACollection>("MVAValues");
  produces<QualityMaskCollection>("QualityMasks");
}

~TrackMVAClassifierBase()

TrackMVAClassifierBase::~TrackMVAClassifierBase ( )

override

Definition at line 20 of file TrackMVAClassifierBase.cc.

{}

Member Function Documentation

computeMVA()

virtual void TrackMVAClassifierBase::computeMVA ( reco::TrackCollection const &  tracks, reco::BeamSpot const &  beamSpot, reco::VertexCollection const &  vertices, MVAPairCollection &  mvas  ) const

protectedpure virtual

Implemented in TrackMVAClassifier< MVA, EventCache >.

Referenced by produce().

fill()

void TrackMVAClassifierBase::fill ( edm::ParameterSetDescription &  desc )

staticprotected

Definition at line 10 of file TrackMVAClassifierBase.cc.

References submitPVResolutionJobs::desc, and ProducerED_cfi::InputTag.

Referenced by TrackMVAClassifier< MVA, EventCache >::fillDescriptions().

                                                                  {
  desc.add<edm::InputTag>("src", edm::InputTag());
  desc.add<edm::InputTag>("beamspot", edm::InputTag("offlineBeamSpot"));
  desc.add<edm::InputTag>("vertices", edm::InputTag("firstStepPrimaryVertices"));
  desc.add<bool>("ignoreVertices", false);
  // default cuts for "cut based classification"
  std::vector<double> cuts = {-.7, 0.1, .7};
  desc.add<std::vector<double>>("qualityCuts", cuts);
}

initEvent()

virtual void TrackMVAClassifierBase::initEvent ( const edm::EventSetup &  es )

protectedpure virtual

Implemented in TrackMVAClassifier< MVA, EventCache >.

Referenced by produce().

produce()

void TrackMVAClassifierBase::produce ( edm::Event &  evt, const edm::EventSetup &  es  )

finalprivate

Definition at line 35 of file TrackMVAClassifierBase.cc.

References cms::cuda::assert(), beamspot_, computeMVA(), f, edm::Event::getByToken(), reco::TrackBase::highPurity, ignoreVertices_, initEvent(), edm::HandleBase::isValid(), isotrackApplyRegressor::k, reco::TrackBase::loose, WZElectronSkims53X_cff::max, eostools::move(), beam_dqm_sourceclient-live_cfg::mva, edm::Event::put(), qualityCuts, src_, reco::TrackBase::tight, DiMuonV_cfg::tracks, AlignmentTracksFromVertexSelector_cfi::vertices, and vertices_.

                                                                           {
  // Get tracks
  edm::Handle<reco::TrackCollection> hSrcTrack;
  evt.getByToken(src_, hSrcTrack);
  auto const& tracks(*hSrcTrack);

  // looking for the beam spot
  edm::Handle<reco::BeamSpot> hBsp;
  evt.getByToken(beamspot_, hBsp);

  // Select good primary vertices for use in subsequent track selection
  edm::Handle<reco::VertexCollection> hVtx;
  evt.getByToken(vertices_, hVtx);

  initEvent(es);

  // products
  auto mvaPairs = std::make_unique<MVAPairCollection>(tracks.size(), std::make_pair(-99.f, true));
  auto mvas = std::make_unique<MVACollection>(tracks.size(), -99.f);
  auto quals = std::make_unique<QualityMaskCollection>(tracks.size(), 0);

  if (hVtx.isValid() && !ignoreVertices_) {
    computeMVA(tracks, *hBsp, *hVtx, *mvaPairs);
  } else {
    if (!ignoreVertices_)
      edm::LogWarning("TrackMVAClassifierBase")
          << "ignoreVertices is set to False in the configuration, but the vertex collection is not valid";
    std::vector<reco::Vertex> vertices;
    computeMVA(tracks, *hBsp, vertices, *mvaPairs);
  }
  assert((*mvaPairs).size() == tracks.size());

  unsigned int k = 0;
  for (auto const& output : *mvaPairs) {
    if (output.second) {
      (*mvas)[k] = output.first;
    } else {
      // If the MVA value is known to be unreliable, force into generalTracks collection
      (*mvas)[k] = std::max(output.first, float(qualityCuts[0] + 0.001));
    }
    float mva = (*mvas)[k];
    (*quals)[k++] = (mva > qualityCuts[0]) << reco::TrackBase::loose |
                    (mva > qualityCuts[1]) << reco::TrackBase::tight |
                    (mva > qualityCuts[2]) << reco::TrackBase::highPurity;
  }

  evt.put(std::move(mvas), "MVAValues");
  evt.put(std::move(quals), "QualityMasks");
}

Member Data Documentation

beamspot_

edm::EDGetTokenT<reco::BeamSpot> TrackMVAClassifierBase::beamspot_

private

Definition at line 49 of file TrackMVAClassifier.h.

Referenced by produce().

ignoreVertices_

bool TrackMVAClassifierBase::ignoreVertices_

private

Definition at line 52 of file TrackMVAClassifier.h.

Referenced by produce().

qualityCuts

float TrackMVAClassifierBase::qualityCuts[3]

private

Definition at line 57 of file TrackMVAClassifier.h.

Referenced by produce(), and TrackMVAClassifierBase().

src_

edm::EDGetTokenT<reco::TrackCollection> TrackMVAClassifierBase::src_

private

source collection label

Definition at line 48 of file TrackMVAClassifier.h.

Referenced by produce().

vertices_

edm::EDGetTokenT<reco::VertexCollection> TrackMVAClassifierBase::vertices_

private

Definition at line 50 of file TrackMVAClassifier.h.

Referenced by produce().