#include <MVAValueMapProducer.h>

Inheritance diagram for MVAValueMapProducer< ParticleType >:

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

	~MVAValueMapProducer ()

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

Public Member Functions inherited from edm::stream::EDProducerBase
	EDProducerBase ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducerBase ()

Public Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

virtual	~ProducerBase ()

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

	EDConsumerBase ()

ProductHolderIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesDependentUpon (const std::string &iProcessName, std::vector< const char * > &oModuleLabels) const

void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

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

Static Public Member Functions inherited from edm::stream::EDProducerBase
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Private Member Functions
virtual 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 >	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

Public Types inherited from edm::stream::EDProducerBase
typedef EDProducerAdaptorBase	ModuleType

Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

template<class ParticleType>
class MVAValueMapProducer< ParticleType >

Definition at line 18 of file MVAValueMapProducer.h.

Constructor & Destructor Documentation

template<class ParticleType >

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

explicit

Definition at line 53 of file MVAValueMapProducer.h.

References HLT_25ns14e33_v1_cff::estimator, edm::hlt::Exception, reco::get(), edm::ParameterSet::getParameter(), edm::ParameterSet::getParameterSetVector(), NULL, 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 esimtators.
   const std::vector<edm::ParameterSet>& mvaEstimatorConfigs
     = iConfig.getParameterSetVector("mvaConfigurations");
   for( auto &imva : mvaEstimatorConfigs ){
 
     std::unique_ptr<AnyMVAEstimatorRun2Base> thisEstimator;
     thisEstimator.reset(NULL);
     if( !imva.empty() ) {
       const std::string& pName = imva.getParameter<std::string>("mvaName");
       // The factory below constructs the MVA of the appropriate type based
       // on the "mvaName" which is the name of the derived MVA class (plugin)
       AnyMVAEstimatorRun2Base *estimator = AnyMVAEstimatorRun2Factory::get()->create(pName, imva);
       // Declare all event content, such as ValueMaps produced upstream or other,
       // original event data pieces, that is needed (if any is implemented in the specific
       // MVA classes)
       //edm::ConsumesCollector &cc = consumesCollector();
       estimator->setConsumes( consumesCollector() );
 
       thisEstimator.reset(estimator);
       
     } 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;
 
     // The unique pointer control is passed to the vector in the line below.
     // Don't use thisEstimator pointer beyond the next line.
     mvaEstimators_.emplace_back( thisEstimator.release() );
 
     //
     // Compose and save the names of the value maps to be produced
     //
     const auto& currentEstimator = mvaEstimators_.back();
     std::string thisValueMapName = currentEstimator->getName() + "Values";
     std::string thisCategoriesMapName = currentEstimator->getName() + "Categories";
     mvaValueMapNames_.push_back( thisValueMapName );
     mvaCategoriesMapNames_.push_back( thisCategoriesMapName );
 
     // Declare the maps to the framework
     produces<edm::ValueMap<float> >(thisValueMapName);  
     produces<edm::ValueMap<int> >(thisCategoriesMapName);  
 
   }
 
 
 }

template<class ParticleType >

MVAValueMapProducer< ParticleType >::~MVAValueMapProducer ( )

Definition at line 110 of file MVAValueMapProducer.h.

110 {

111 }

Member Function Documentation

template<class ParticleType >

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

static

Definition at line 175 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
	)

overrideprivatevirtual

Implements edm::stream::EDProducerBase.

Definition at line 114 of file MVAValueMapProducer.h.

References edm::Event::getByToken(), i, and alcazmumu_cfi::src.

                                                                                              {
 
   using namespace edm;
   
   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++ ){
     
     // Set up all event content, such as ValueMaps produced upstream or other,
     // original event data pieces, that is needed (if any is implemented in the specific
     // MVA classes)
     mvaEstimators_[iEstimator]->getEventContent( iEvent );
 
     std::vector<float> mvaValues;
     std::vector<int> mvaCategories;
 
     // Loop over particles
     for (size_t i = 0; i < src->size(); ++i){
       auto iCand = src->ptrAt(i);
       
       mvaValues.push_back( mvaEstimators_[iEstimator]->mvaValue(  iCand ) );
       mvaCategories.push_back( mvaEstimators_[iEstimator]->findCategory(  iCand ) );
     } // end loop over particles
 
     writeValueMap(iEvent, src, mvaValues, mvaValueMapNames_[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 160 of file MVAValueMapProducer.h.

References edm::helper::Filler< Map >::fill(), patZpeak::handle, edm::helper::Filler< Map >::insert(), and edm::Event::put().

 {
   using namespace edm; 
   using namespace std;
   auto_ptr<ValueMap<T> > valMap(new ValueMap<T>());
   typename edm::ValueMap<T>::Filler filler(*valMap);
   filler.insert(handle, values.begin(), values.end());
   filler.fill();
   iEvent.put(valMap, label);
 }

Member Data Documentation

template<class ParticleType >

std::vector<std::string> MVAValueMapProducer< ParticleType >::mvaCategoriesMapNames_

private

Definition at line 48 of file MVAValueMapProducer.h.

template<class ParticleType >

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

private

Definition at line 44 of file MVAValueMapProducer.h.

template<class ParticleType >

std::vector<std::string> MVAValueMapProducer< ParticleType >::mvaValueMapNames_

private

Definition at line 47 of file MVAValueMapProducer.h.

template<class ParticleType >

edm::EDGetToken MVAValueMapProducer< ParticleType >::src_

private

Definition at line 38 of file MVAValueMapProducer.h.

template<class ParticleType >

edm::EDGetToken MVAValueMapProducer< ParticleType >::srcMiniAOD_

private

Definition at line 41 of file MVAValueMapProducer.h.

Public Member Functions

Static Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

template<class ParticleType> class MVAValueMapProducer< ParticleType >

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation

template<class ParticleType>
class MVAValueMapProducer< ParticleType >