Inheritance diagram for RecoTauDiscriminantCutMultiplexer:

Classes
struct	DiscriminantCutEntry

Public Member Functions
void	beginEvent (const edm::Event &event, const edm::EventSetup &eventSetup) override

double	discriminate (const reco::PFTauRef &) const override

	RecoTauDiscriminantCutMultiplexer (const edm::ParameterSet &pset)

	~RecoTauDiscriminantCutMultiplexer ()

Public Member Functions inherited from TauDiscriminationProducerBase< TauType, TauDiscriminator >
virtual double	discriminate (const TauRef &tau) const =0

virtual void	endEvent (edm::Event &evt)

void	produce (edm::Event &, const edm::EventSetup &)

	TauDiscriminationProducerBase (const edm::ParameterSet &iConfig)

	TauDiscriminationProducerBase ()

virtual	~TauDiscriminationProducerBase ()

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
	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 ()

Private Types
typedef std::map< int, std::unique_ptr < DiscriminantCutEntry > >	DiscriminantCutMap

Private Attributes
DiscriminantCutMap	cuts_

edm::FileInPath	inputFileName_

bool	isInitialized_

edm::InputTag	key_

edm::EDGetTokenT < reco::PFTauDiscriminator >	key_token

edm::Handle < reco::PFTauDiscriminator >	keyHandle_

bool	loadMVAfromDB_

std::string	moduleLabel_

std::unique_ptr< const TFormula >	mvaOutput_normalization_

std::string	mvaOutputNormalizationName_

edm::InputTag	toMultiplex_

edm::EDGetTokenT < reco::PFTauDiscriminator >	toMultiplex_token

edm::Handle < reco::PFTauDiscriminator >	toMultiplexHandle_

int	verbosity_

Additional Inherited Members
Public Types inherited from TauDiscriminationProducerBase< TauType, TauDiscriminator >
typedef std::vector< TauType >	TauCollection

typedef edm::Ref< TauCollection >	TauRef

typedef edm::RefProd < TauCollection >	TauRefProd

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

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

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

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

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)

Protected Attributes inherited from TauDiscriminationProducerBase< TauType, TauDiscriminator >
std::string	moduleLabel_

double	prediscriminantFailValue_

edm::EDGetTokenT< TauCollection >	Tau_token

size_t	tauIndex_

edm::InputTag	TauProducer_

Detailed Description

Definition at line 34 of file RecoTauDiscriminantCutMultiplexer.cc.

Member Typedef Documentation

typedef std::map<int, std::unique_ptr<DiscriminantCutEntry> > RecoTauDiscriminantCutMultiplexer::DiscriminantCutMap

private

Definition at line 66 of file RecoTauDiscriminantCutMultiplexer.cc.

Constructor & Destructor Documentation

RecoTauDiscriminantCutMultiplexer::RecoTauDiscriminantCutMultiplexer ( const edm::ParameterSet & pset )

explicit

Definition at line 136 of file RecoTauDiscriminantCutMultiplexer.cc.

References python.rootplot.argparse::category, gather_cfg::cout, GOODCOLL_filter_cfg::cut, cuts_, edm::hlt::Exception, edm::ParameterSet::exists(), edm::ParameterSet::getParameter(), inputFileName_, key_, key_token, RecoTauDiscriminantCutMultiplexer::DiscriminantCutEntry::kFixedCut, RecoTauDiscriminantCutMultiplexer::DiscriminantCutEntry::kVariableCut, loadMVAfromDB_, moduleLabel_, eostools::move(), mvaOutputNormalizationName_, AlCaHLTBitMon_QueryRunRegistry::string, toMultiplex_, toMultiplex_token, and verbosity_.

   : PFTauDiscriminationProducerBase(cfg),
     moduleLabel_(cfg.getParameter<std::string>("@module_label")),
     mvaOutput_normalization_(),
     isInitialized_(false)
 {
   
   toMultiplex_ = cfg.getParameter<edm::InputTag>("toMultiplex");
   toMultiplex_token = consumes<reco::PFTauDiscriminator>(toMultiplex_);
   key_ = cfg.getParameter<edm::InputTag>("key");
   key_token = consumes<reco::PFTauDiscriminator>(key_);
 
   verbosity_ = ( cfg.exists("verbosity") ) ?
     cfg.getParameter<int>("verbosity") : 0;
 
 
   loadMVAfromDB_ = cfg.exists("loadMVAfromDB") ? cfg.getParameter<bool>("loadMVAfromDB") : false;
   if ( !loadMVAfromDB_ ) {
     if(cfg.exists("inputFileName")){
       inputFileName_ = cfg.getParameter<edm::FileInPath>("inputFileName");
     }else throw cms::Exception("MVA input not defined") << "Requested to load tau MVA input from ROOT file but no file provided in cfg file";
   }
   if(verbosity_)  std::cout << moduleLabel_ << " loadMVA = " << loadMVAfromDB_ << std::endl;
   if ( cfg.exists("mvaOutput_normalization") ) {
     mvaOutputNormalizationName_ = cfg.getParameter<std::string>("mvaOutput_normalization"); 
   }
 
   // Setup our cut map
   typedef std::vector<edm::ParameterSet> VPSet;
   VPSet mapping = cfg.getParameter<VPSet>("mapping");
   for ( VPSet::const_iterator mappingEntry = mapping.begin();
     mappingEntry != mapping.end(); ++mappingEntry ) {
     unsigned category = mappingEntry->getParameter<uint32_t>("category");
     std::unique_ptr<DiscriminantCutEntry> cut{new DiscriminantCutEntry()};
     if ( mappingEntry->existsAs<double>("cut") ) {
       cut->cutValue_ = mappingEntry->getParameter<double>("cut");
       cut->mode_ = DiscriminantCutEntry::kFixedCut;
     } else if ( mappingEntry->existsAs<std::string>("cut") ) {
       cut->cutName_ = mappingEntry->getParameter<std::string>("cut");
       std::string cutVariable_string = mappingEntry->getParameter<std::string>("variable");
       cut->cutVariable_.reset( new StringObjectFunction<reco::PFTau>(cutVariable_string.data()) );
       cut->mode_ = DiscriminantCutEntry::kVariableCut;
     } else {
       throw cms::Exception("RecoTauDiscriminantCutMultiplexer") 
         << " Undefined Configuration Parameter 'cut' !!\n";
     }
     cuts_[category] = std::move(cut);
   }
 
   verbosity_ = ( cfg.exists("verbosity") ) ?
     cfg.getParameter<int>("verbosity") : 0;
   if(verbosity_) std::cout << "constructed " << moduleLabel_ << std::endl;
 }

RecoTauDiscriminantCutMultiplexer::~RecoTauDiscriminantCutMultiplexer ( )

Definition at line 190 of file RecoTauDiscriminantCutMultiplexer.cc.

191 {

192 }

Member Function Documentation

void RecoTauDiscriminantCutMultiplexer::beginEvent	(	const edm::Event &	event,
		const edm::EventSetup &	eventSetup
	)

overridevirtual

Reimplemented from TauDiscriminationProducerBase< TauType, TauDiscriminator >.

Definition at line 194 of file RecoTauDiscriminantCutMultiplexer.cc.

References gather_cfg::cout, GOODCOLL_filter_cfg::cut, cuts_, edm::Event::getByToken(), analyzePatCleaning_cfg::inputFile, inputFileName_, isInitialized_, key_token, keyHandle_, RecoTauDiscriminantCutMultiplexer::DiscriminantCutEntry::kVariableCut, loadMVAfromDB_, moduleLabel_, mvaOutput_normalization_, mvaOutputNormalizationName_, groupFilesInBlocks::temp, toMultiplex_token, toMultiplexHandle_, and verbosity_.

 {
   if(verbosity_) std::cout << " begin! " << moduleLabel_ << " " << isInitialized_ << std::endl;
   if ( !isInitialized_ ) {
     //Only open the file once and we can close it when this routine is done
     // since all objects gotten from the file will have been copied
     std::unique_ptr<TFile> inputFile;
     if ( mvaOutputNormalizationName_ != "" ) {
       if ( !loadMVAfromDB_ ) {
     inputFile = openInputFile(inputFileName_);
     mvaOutput_normalization_ = loadObjectFromFile<TFormula>(*inputFile, mvaOutputNormalizationName_);
       } else {
     auto temp = loadTFormulaFromDB(es, mvaOutputNormalizationName_, Form("%s_mvaOutput_normalization", moduleLabel_.data()), verbosity_);
     mvaOutput_normalization_.reset(temp.release());
       }
     }
     for ( DiscriminantCutMap::iterator cut = cuts_.begin();
       cut != cuts_.end(); ++cut ) {
       if ( cut->second->mode_ == DiscriminantCutEntry::kVariableCut ) {
     if ( !loadMVAfromDB_ ) {
       if(not inputFile) {
         inputFile = openInputFile(inputFileName_);
       }
       if(verbosity_) std::cout << "Loading from file" << inputFileName_ << std::endl;
       cut->second->cutFunction_ = loadObjectFromFile<TGraph>(*inputFile, cut->second->cutName_);
     } else {
       if(verbosity_) std::cout << "Loading from DB" << std::endl;
       cut->second->cutFunction_ = loadTGraphFromDB(es, cut->second->cutName_, verbosity_);
     }
       }
     }
     isInitialized_ = true;
   }
 
   evt.getByToken(toMultiplex_token, toMultiplexHandle_);
   evt.getByToken(key_token, keyHandle_);
 }

double RecoTauDiscriminantCutMultiplexer::discriminate ( const reco::PFTauRef & tau ) const

override

Definition at line 233 of file RecoTauDiscriminantCutMultiplexer.cc.

References assert(), gather_cfg::cout, cuts_, geometryDiff::epsilon, RecoTauDiscriminantCutMultiplexer::DiscriminantCutEntry::kFixedCut, RecoTauDiscriminantCutMultiplexer::DiscriminantCutEntry::kVariableCut, moduleLabel_, mvaOutput_normalization_, TauDiscriminationProducerBase< TauType, TauDiscriminator >::prediscriminantFailValue_, metsig::tau, and verbosity_.

 {
   if ( verbosity_ ) {
     std::cout << "<RecoTauDiscriminantCutMultiplexer::discriminate>:" << std::endl;
     std::cout << " moduleLabel = " << moduleLabel_ << std::endl;
   }
 
   double disc_result = (*toMultiplexHandle_)[tau];
   if ( verbosity_ ) {
     std::cout << "disc_result = " <<  disc_result << std::endl;
   }
   if ( mvaOutput_normalization_ ) {
     disc_result = mvaOutput_normalization_->Eval(disc_result);
     //if ( disc_result > 1. ) disc_result = 1.;
     //if ( disc_result < 0. ) disc_result = 0.;
     if ( verbosity_ ) {
       std::cout << "disc_result (normalized) = " <<  disc_result << std::endl;
     }
   }
   double key_result = (*keyHandle_)[tau];
   DiscriminantCutMap::const_iterator cutIter = cuts_.find(TMath::Nint(key_result));
 
   
   // Return null if it doesn't exist
   if ( cutIter == cuts_.end() ) {
     return prediscriminantFailValue_;
   }
   // See if the discriminator passes our cuts
   bool passesCuts = false;
   if ( cutIter->second->mode_ == DiscriminantCutEntry::kFixedCut ) {
     passesCuts = (disc_result > cutIter->second->cutValue_);
     if ( verbosity_ ) {
       std::cout << "cutValue (fixed) = " << cutIter->second->cutValue_ << " --> passesCuts = " << passesCuts << std::endl;
     }
   } else if ( cutIter->second->mode_ == DiscriminantCutEntry::kVariableCut ) {
     double cutVariable = (*cutIter->second->cutVariable_)(*tau);
     double xMin, xMax, dummy;
     cutIter->second->cutFunction_->GetPoint(0, xMin, dummy);
     cutIter->second->cutFunction_->GetPoint(cutIter->second->cutFunction_->GetN() - 1, xMax, dummy);
     const double epsilon = 1.e-3;
     if      ( cutVariable < (xMin + epsilon) ) cutVariable = xMin + epsilon;
     else if ( cutVariable > (xMax - epsilon) ) cutVariable = xMax - epsilon;
     double cutValue = cutIter->second->cutFunction_->Eval(cutVariable);
     passesCuts = (disc_result > cutValue);
     if ( verbosity_ ) {
       std::cout << "cutValue (@" << cutVariable << ") = " << cutValue << " --> passesCuts = " << passesCuts << std::endl;
     }
   } else assert(0);
 
   return passesCuts;
 }