#include <DeepTauBase.h>

Inheritance diagram for deep_tau::DeepTauBase:

Classes
struct	Output

Public Types
using	Cutter = TauWPThreshold

using	CutterPtr = std::unique_ptr< Cutter >

using	ElectronCollection = pat::ElectronCollection

using	LorentzVectorXYZ = ROOT::Math::LorentzVector< ROOT::Math::PxPyPzE4D< double >>

using	MuonCollection = pat::MuonCollection

using	OutputCollection = std::map< std::string, Output >

using	TauCollection = std::vector< TauType >

using	TauDiscriminator = pat::PATTauDiscriminator

using	TauRef = edm::Ref< TauCollection >

using	TauRefProd = edm::RefProd< TauCollection >

using	TauType = pat::Tau

using	WPMap = std::map< std::string, CutterPtr >

Public Types inherited from edm::stream::EDProducer< edm::GlobalCache< DeepTauCache > >
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
	DeepTauBase (const edm::ParameterSet &cfg, const OutputCollection &outputs, const DeepTauCache *cache)

void	produce (edm::Event &event, const edm::EventSetup &es) override

	~DeepTauBase () override

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

bool	hasAbilityToProduceInLumis () const final

bool	hasAbilityToProduceInRuns () const final

Static Public Member Functions
static void	globalEndJob (const DeepTauCache *cache)

static std::unique_ptr< DeepTauCache >	initializeGlobalCache (const edm::ParameterSet &cfg)

Protected Attributes
const DeepTauCache *	cache_

OutputCollection	outputs_

edm::EDGetTokenT< pat::PackedCandidateCollection >	pfcandToken_

edm::EDGetTokenT< TauCollection >	tausToken_

edm::EDGetTokenT< reco::VertexCollection >	vtxToken_

std::map< std::string, WPMap >	workingPoints_

Private Member Functions
virtual void	createOutputs (edm::Event &event, const tensorflow::Tensor &pred, edm::Handle< TauCollection > taus)

virtual tensorflow::Tensor	getPredictions (edm::Event &event, const edm::EventSetup &es, edm::Handle< TauCollection > taus)=0

Detailed Description

Definition at line 58 of file DeepTauBase.h.

Member Typedef Documentation

using deep_tau::DeepTauBase::Cutter = TauWPThreshold

Definition at line 68 of file DeepTauBase.h.

using deep_tau::DeepTauBase::CutterPtr = std::unique_ptr<Cutter>

Definition at line 69 of file DeepTauBase.h.

using deep_tau::DeepTauBase::ElectronCollection = pat::ElectronCollection

Definition at line 65 of file DeepTauBase.h.

using deep_tau::DeepTauBase::LorentzVectorXYZ = ROOT::Math::LorentzVector<ROOT::Math::PxPyPzE4D<double>>

Definition at line 67 of file DeepTauBase.h.

using deep_tau::DeepTauBase::MuonCollection = pat::MuonCollection

Definition at line 66 of file DeepTauBase.h.

using deep_tau::DeepTauBase::OutputCollection = std::map<std::string, Output>

Definition at line 82 of file DeepTauBase.h.

using deep_tau::DeepTauBase::TauCollection = std::vector<TauType>

Definition at line 62 of file DeepTauBase.h.

using deep_tau::DeepTauBase::TauDiscriminator = pat::PATTauDiscriminator

Definition at line 61 of file DeepTauBase.h.

using deep_tau::DeepTauBase::TauRef = edm::Ref<TauCollection>

Definition at line 63 of file DeepTauBase.h.

using deep_tau::DeepTauBase::TauRefProd = edm::RefProd<TauCollection>

Definition at line 64 of file DeepTauBase.h.

using deep_tau::DeepTauBase::TauType = pat::Tau

Definition at line 60 of file DeepTauBase.h.

using deep_tau::DeepTauBase::WPMap = std::map<std::string, CutterPtr>

Definition at line 70 of file DeepTauBase.h.

Constructor & Destructor Documentation

deep_tau::DeepTauBase::DeepTauBase	(	const edm::ParameterSet &	cfg,
		const OutputCollection &	outputs,
		const DeepTauCache *	cache
	)

Definition at line 79 of file DeepTauBase.cc.

References edm::ParameterSet::getParameter(), outputs_, AlCaHLTBitMon_QueryRunRegistry::string, and workingPoints_.

                                                     :
     tausToken_(consumes<TauCollection>(cfg.getParameter<edm::InputTag>("taus"))),
     pfcandToken_(consumes<pat::PackedCandidateCollection>(cfg.getParameter<edm::InputTag>("pfcands"))),
     vtxToken_(consumes<reco::VertexCollection>(cfg.getParameter<edm::InputTag>("vertices"))),
     outputs_(outputCollection),
     cache_(cache)
 {
     for(const auto& output_desc : outputs_) {
         produces<TauDiscriminator>(output_desc.first);
         const auto& cut_pset = cfg.getParameter<edm::ParameterSet>(output_desc.first + "WP");
         for(const std::string& wp_name : cut_pset.getParameterNames()) {
             const auto& cut_str = cut_pset.getParameter<std::string>(wp_name);
             workingPoints_[output_desc.first][wp_name] = std::make_unique<Cutter>(cut_str);
             produces<TauDiscriminator>(output_desc.first + wp_name);
         }
     }
 }

deep_tau::DeepTauBase::~DeepTauBase ( )

inlineoverride

Definition at line 86 of file DeepTauBase.h.

References looper::cfg.

86 {}

Member Function Documentation

void deep_tau::DeepTauBase::createOutputs	(	edm::Event &	event,
		const tensorflow::Tensor &	pred,
		edm::Handle< TauCollection >	taus
	)

privatevirtual

Definition at line 107 of file DeepTauBase.cc.

References eostools::move(), outputs_, mps_fire::result, and workingPoints_.

Referenced by produce().

 {
     for(const auto& output_desc : outputs_) {
         auto result_map = output_desc.second.get_value(taus, pred, workingPoints_.at(output_desc.first));
         for(auto& result : result_map)
             event.put(std::move(result.second), output_desc.first + result.first);
     }
 }

virtual tensorflow::Tensor deep_tau::DeepTauBase::getPredictions	(	edm::Event &	event,
		const edm::EventSetup &	es,
		edm::Handle< TauCollection >	taus
	)

privatepure virtual

Implemented in DeepTauId, and DPFIsolation.

Referenced by produce().

static void deep_tau::DeepTauBase::globalEndJob ( const DeepTauCache * cache )

inlinestatic

Definition at line 91 of file DeepTauBase.h.

References nano_cff::taus.

Referenced by DeepTauId::globalEndJob().

91 { }

std::unique_ptr< DeepTauCache > deep_tau::DeepTauBase::initializeGlobalCache ( const edm::ParameterSet & cfg )

static

Definition at line 116 of file DeepTauBase.cc.

References mps_splice::entry, edm::FileInPath::fullPath(), edm::ParameterSet::getParameter(), and AlCaHLTBitMon_QueryRunRegistry::string.

Referenced by DeepTauId::initializeGlobalCache().

 {
     const auto graph_name_vector = cfg.getParameter<std::vector<std::string>>("graph_file");
     std::map<std::string, std::string> graph_names;
     for(const auto& entry : graph_name_vector) {
         const size_t sep_pos = entry.find(':');
         std::string entry_name, graph_file;
         if(sep_pos != std::string::npos) {
             entry_name = entry.substr(0, sep_pos);
             graph_file = entry.substr(sep_pos + 1);
         } else {
             entry_name = "";
             graph_file = entry;
         }
         graph_file = edm::FileInPath(graph_file).fullPath();
         if(graph_names.count(entry_name))
             throw cms::Exception("DeepTauCache") << "Duplicated graph entries";
         graph_names[entry_name] = graph_file;
     }
     bool mem_mapped = cfg.getParameter<bool>("mem_mapped");
     return std::make_unique<DeepTauCache>(graph_names, mem_mapped);
 }

void deep_tau::DeepTauBase::produce	(	edm::Event &	event,
		const edm::EventSetup &	es
	)

override

Definition at line 98 of file DeepTauBase.cc.

References createOutputs(), getPredictions(), nano_cff::taus, and tausToken_.

 {
     edm::Handle<TauCollection> taus;
     event.getByToken(tausToken_, taus);
 
     const tensorflow::Tensor& pred = getPredictions(event, es, taus);
     createOutputs(event, pred, taus);
 }

Member Data Documentation

const DeepTauCache* deep_tau::DeepTauBase::cache_

protected

Definition at line 103 of file DeepTauBase.h.

OutputCollection deep_tau::DeepTauBase::outputs_

protected

Definition at line 102 of file DeepTauBase.h.

Referenced by createOutputs(), and DeepTauBase().

edm::EDGetTokenT<pat::PackedCandidateCollection> deep_tau::DeepTauBase::pfcandToken_

protected

Definition at line 99 of file DeepTauBase.h.

edm::EDGetTokenT<TauCollection> deep_tau::DeepTauBase::tausToken_

protected

Definition at line 98 of file DeepTauBase.h.

Referenced by produce().

edm::EDGetTokenT<reco::VertexCollection> deep_tau::DeepTauBase::vtxToken_

protected

Definition at line 100 of file DeepTauBase.h.

std::map<std::string, WPMap> deep_tau::DeepTauBase::workingPoints_

protected

Definition at line 101 of file DeepTauBase.h.

Referenced by createOutputs(), and DeepTauBase().

Classes

Public Types

Public Member Functions

Static Public Member Functions

Protected Attributes

Private Member Functions

Detailed Description

Member Typedef Documentation

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation