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	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () 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(), begin, looper::cfg, filterCSVwithJSON::copy, 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 edm::ParameterSetDescription::add(), L1TMuonDQMOffline_cfi::cuts, and HLT_2018_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, PDWG_EXOHSCP_cff::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().