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<>
typedef CacheContexts< T... >	CacheTypes
typedef CacheTypes::GlobalCache	GlobalCache
typedef AbilityChecker< T... >	HasAbility
typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache
typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext
typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache
typedef CacheTypes::RunCache	RunCache
typedef RunContextT< RunCache, GlobalCache >	RunContext
typedef CacheTypes::RunSummaryCache	RunSummaryCache

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

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 21 of file TrackMVAClassifier.h.

Member Typedef Documentation

MVACollection

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

Definition at line 26 of file TrackMVAClassifier.h.

MVAPairCollection

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

Definition at line 30 of file TrackMVAClassifier.h.

QualityMaskCollection

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

Definition at line 27 of file TrackMVAClassifier.h.

Constructor & Destructor Documentation

TrackMVAClassifierBase()

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

explicit

Definition at line 22 of file TrackMVAClassifierBase.cc.

    : 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");
}

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

~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.

                                                                  {
  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);
}

References cuts, submitPVResolutionJobs::desc, and HLT_FULL_cff::InputTag.

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

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.

                                                                           {
  // 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");
}

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

Member Data Documentation

beamspot_

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

private

Definition at line 47 of file TrackMVAClassifier.h.

Referenced by produce().

ignoreVertices_

bool TrackMVAClassifierBase::ignoreVertices_

private

Definition at line 50 of file TrackMVAClassifier.h.

Referenced by produce().

qualityCuts

float TrackMVAClassifierBase::qualityCuts[3]

private

Definition at line 55 of file TrackMVAClassifier.h.

Referenced by produce(), and TrackMVAClassifierBase().

src_

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

private

source collection label

Definition at line 46 of file TrackMVAClassifier.h.

Referenced by produce().

vertices_

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

private

Definition at line 48 of file TrackMVAClassifier.h.

Referenced by produce().