#include <MVAValueMapProducer.h>

Inheritance diagram for MVAValueMapProducer< ParticleType >:

Public Member Functions
	MVAValueMapProducer (const edm::ParameterSet &)

	~MVAValueMapProducer () override

Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default

bool	hasAbilityToProduceInLumis () const final

bool	hasAbilityToProduceInRuns () const final

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)

Private Member Functions
void	produce (edm::Event &, const edm::EventSetup &) override

template<typename T >
void	writeValueMap (edm::Event &iEvent, const edm::Handle< edm::View< ParticleType > > &handle, const std::vector< T > &values, const std::string &label) const

Private Attributes
std::vector< std::string >	mvaCategoriesMapNames_

std::vector< std::unique_ptr< AnyMVAEstimatorRun2Base > >	mvaEstimators_

std::vector< std::string >	mvaRawValueMapNames_

std::vector< std::string >	mvaValueMapNames_

edm::EDGetToken	src_

edm::EDGetToken	srcMiniAOD_

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

template<class ParticleType>
class MVAValueMapProducer< ParticleType >

Definition at line 22 of file MVAValueMapProducer.h.

Constructor & Destructor Documentation

template<class ParticleType >

MVAValueMapProducer< ParticleType >::MVAValueMapProducer ( const edm::ParameterSet & iConfig )

Definition at line 58 of file MVAValueMapProducer.h.

References beamerCreator::create(), Exception, reco::get(), edm::ParameterSet::getParameter(), edm::ParameterSet::getParameterSetVector(), MVAValueMapProducer< ParticleType >::mvaCategoriesMapNames_, MVAValueMapProducer< ParticleType >::mvaEstimators_, MVAValueMapProducer< ParticleType >::mvaRawValueMapNames_, MVAValueMapProducer< ParticleType >::mvaValueMapNames_, MVAValueMapProducer< ParticleType >::src_, MVAValueMapProducer< ParticleType >::srcMiniAOD_, and AlCaHLTBitMon_QueryRunRegistry::string.

 {
 
   //
   // Declare consummables, handle both AOD and miniAOD case
   //
   src_        = mayConsume<edm::View<ParticleType> >(iConfig.getParameter<edm::InputTag>("src"));
   srcMiniAOD_ = mayConsume<edm::View<ParticleType> >(iConfig.getParameter<edm::InputTag>("srcMiniAOD"));
 
   // Loop over the list of MVA configurations passed here from python and
   // construct all requested MVA estimators.
   const std::vector<edm::ParameterSet>& mvaEstimatorConfigs
     = iConfig.getParameterSetVector("mvaConfigurations");
 
   for( auto &imva : mvaEstimatorConfigs ){
 
     // The factory below constructs the MVA of the appropriate type based
     // on the "mvaName" which is the name of the derived MVA class (plugin)
     if( !imva.empty() ) {
 
       mvaEstimators_.emplace_back(AnyMVAEstimatorRun2Factory::get()->create(
                   imva.getParameter<std::string>("mvaName"), imva));
 
     } else
       throw cms::Exception(" MVA configuration not found: ")
         << " failed to find proper configuration for one of the MVAs in the main python script " << std::endl;
 
     mvaEstimators_.back()->setConsumes( consumesCollector() );
 
     //
     // Compose and save the names of the value maps to be produced
     //
 
     const std::string fullName = ( mvaEstimators_.back()->getName() +
                                    mvaEstimators_.back()->getTag()  );
 
     const std::string thisValueMapName      = fullName + "Values";
     const std::string thisRawValueMapName   = fullName + "RawValues";
     const std::string thisCategoriesMapName = fullName + "Categories";
 
     mvaValueMapNames_     .push_back( thisValueMapName      );
     mvaRawValueMapNames_  .push_back( thisRawValueMapName   );
     mvaCategoriesMapNames_.push_back( thisCategoriesMapName );
 
     // Declare the maps to the framework
     produces<edm::ValueMap<float>>(thisValueMapName     );
     produces<edm::ValueMap<float>>(thisRawValueMapName  );
     produces<edm::ValueMap<int>>  (thisCategoriesMapName);
 
   }
 
 }

template<class ParticleType >

MVAValueMapProducer< ParticleType >::~MVAValueMapProducer ( )

override

Definition at line 112 of file MVAValueMapProducer.h.

112 {

113 }

Member Function Documentation

template<class ParticleType >

void MVAValueMapProducer< ParticleType >::fillDescriptions ( edm::ConfigurationDescriptions & descriptions )

static

Definition at line 170 of file MVAValueMapProducer.h.

References edm::ConfigurationDescriptions::addDefault(), and edm::ParameterSetDescription::setUnknown().

                                                                                                      {
   //The following says we do not know what parameters are allowed so do no validation
   // Please change this to state exactly what you do use, even if it is no parameters
   edm::ParameterSetDescription desc;
   desc.setUnknown();
   descriptions.addDefault(desc);
 }

template<class ParticleType >

void MVAValueMapProducer< ParticleType >::produce	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

overrideprivate

Definition at line 116 of file MVAValueMapProducer.h.

References JetChargeProducer_cfi::exp, edm::Event::getByToken(), mps_fire::i, edm::HandleBase::isValid(), MVAValueMapProducer< ParticleType >::mvaCategoriesMapNames_, MVAValueMapProducer< ParticleType >::mvaEstimators_, MVAValueMapProducer< ParticleType >::mvaRawValueMapNames_, MVAValueMapProducer< ParticleType >::mvaValueMapNames_, TrackRefitter_38T_cff::src, MVAValueMapProducer< ParticleType >::src_, MVAValueMapProducer< ParticleType >::srcMiniAOD_, and MVAValueMapProducer< ParticleType >::writeValueMap().

                                                                                              {
 
   edm::Handle<edm::View<ParticleType> > src;
 
   // Retrieve the collection of particles from the event.
   // If we fail to retrieve the collection with the standard AOD
   // name, we next look for the one with the stndard miniAOD name.
   iEvent.getByToken(src_, src);
   if( !src.isValid() ){
     iEvent.getByToken(srcMiniAOD_,src);
     if( !src.isValid() )
       throw cms::Exception(" Collection not found: ")
         << " failed to find a standard AOD or miniAOD particle collection " << std::endl;
   }
 
 
   // Loop over MVA estimators
   for( unsigned iEstimator = 0; iEstimator < mvaEstimators_.size(); iEstimator++ ){
 
     std::vector<float> mvaValues;
     std::vector<float> mvaRawValues;
     std::vector<int>   mvaCategories;
 
     // Loop over particles
     for (size_t i = 0; i < src->size(); ++i){
       auto iCand = src->ptrAt(i);
       const float response = mvaEstimators_[iEstimator]->mvaValue( iCand, iEvent );
       mvaRawValues.push_back( response ); // The MVA score
       mvaValues.push_back( 2.0/(1.0+exp(-2.0*response))-1 ); // MVA output between -1 and 1
       mvaCategories.push_back( mvaEstimators_[iEstimator]->findCategory( iCand ) );
     } // end loop over particles
 
     writeValueMap(iEvent, src, mvaValues    , mvaValueMapNames_     [iEstimator] );
     writeValueMap(iEvent, src, mvaRawValues , mvaRawValueMapNames_  [iEstimator] );
     writeValueMap(iEvent, src, mvaCategories, mvaCategoriesMapNames_[iEstimator] );
 
   } // end loop over estimators
 
 }

template<class ParticleType >

template<typename T >

void MVAValueMapProducer< ParticleType >::writeValueMap	(	edm::Event &	iEvent,
		const edm::Handle< edm::View< ParticleType > > &	handle,
		const std::vector< T > &	values,
		const std::string &	label
	)		const

private

Definition at line 157 of file MVAValueMapProducer.h.

References objects.autophobj::filler, cmsBatch::handle, eostools::move(), and edm::Event::put().

Referenced by MVAValueMapProducer< ParticleType >::produce().

 {
   auto valMap = std::make_unique<edm::ValueMap<T>>();
   typename edm::ValueMap<T>::Filler filler(*valMap);
   filler.insert(handle, values.begin(), values.end());
   filler.fill();
   iEvent.put(std::move(valMap), label);
 }