TrackMVAClassifierBase Class Reference

#include <TrackMVAClassifier.h>

Inheritance diagram for TrackMVAClassifierBase:

Public Member Functions
	TrackMVAClassifierBase (const edm::ParameterSet &cfg)
	~TrackMVAClassifierBase () override
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
bool	hasAbilityToProduceInLumis () const final
bool	hasAbilityToProduceInRuns () const final

Protected Types
using	MVACollection = std::vector< float >
using	QualityMaskCollection = std::vector< unsigned char >

Protected Member Functions
virtual void	computeMVA (reco::TrackCollection const &tracks, reco::BeamSpot const &beamSpot, reco::VertexCollection const &vertices, MVACollection &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_

Additional Inherited Members
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

Detailed Description

Definition at line 23 of file TrackMVAClassifier.h.

Member Typedef Documentation

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

protected

Definition at line 32 of file TrackMVAClassifier.h.

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

protected

Definition at line 33 of file TrackMVAClassifier.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 24 of file TrackMVAClassifierBase.cc.

References begin, popcon2dropbox::copy(), end, edm::ParameterSet::getParameter(), 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 ( )

override

Definition at line 22 of file TrackMVAClassifierBase.cc.

{}

Member Function Documentation

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

protectedpure virtual

Implemented in TrackMVAClassifier< MVA >.

Referenced by produce().

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

staticprotected

Definition at line 11 of file TrackMVAClassifierBase.cc.

References edm::ParameterSetDescription::add(), and egammaForCoreTracking_cff::cuts.

Referenced by TrackMVAClassifier< MVA >::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);
}

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

protectedpure virtual

Implemented in TrackMVAClassifier< MVA >.

Referenced by produce().

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

finalprivate

Definition at line 39 of file TrackMVAClassifierBase.cc.

References beamspot_, computeMVA(), edm::Event::getByToken(), reco::TrackBase::highPurity, ignoreVertices_, initEvent(), edm::HandleBase::isValid(), gen::k, reco::TrackBase::loose, eostools::move(), DetachedQuadStep_cff::mva, RecoTauHPSTancTauProdcuer_cfi::mvas, edm::Event::put(), qualityCuts, src_, reco::TrackBase::tight, l1t::tracks, electrons_cff::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 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,*mvas);
  } 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,*mvas);
  }    
  assert((*mvas).size()==tracks.size());

  unsigned int k=0;
  for (auto mva : *mvas) {
    (*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

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

private

Definition at line 47 of file TrackMVAClassifier.h.

Referenced by produce().

bool TrackMVAClassifierBase::ignoreVertices_

private

Definition at line 50 of file TrackMVAClassifier.h.

Referenced by produce().

float TrackMVAClassifierBase::qualityCuts[3]

private

Definition at line 55 of file TrackMVAClassifier.h.

Referenced by produce(), and TrackMVAClassifierBase().

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

private

source collection label

Definition at line 46 of file TrackMVAClassifier.h.

Referenced by produce().

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

private

Definition at line 48 of file TrackMVAClassifier.h.

Referenced by produce().