#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 = reco::TauDiscriminatorContainer

using	TauRef = edm::Ref< TauCollection >

using	TauRefProd = edm::RefProd< TauCollection >

using	TauType = pat::Tau

using	WPList = std::vector< 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	hasAbilityToProduceInBeginLumis () const final

bool	hasAbilityToProduceInBeginRuns () const final

bool	hasAbilityToProduceInEndLumis () const final

bool	hasAbilityToProduceInEndRuns () 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, WPList >	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

◆ Cutter

using deep_tau::DeepTauBase::Cutter = TauWPThreshold

Definition at line 68 of file DeepTauBase.h.

◆ CutterPtr

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

Definition at line 69 of file DeepTauBase.h.

◆ ElectronCollection

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

Definition at line 65 of file DeepTauBase.h.

◆ LorentzVectorXYZ

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

Definition at line 67 of file DeepTauBase.h.

◆ MuonCollection

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

Definition at line 66 of file DeepTauBase.h.

◆ OutputCollection

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

Definition at line 82 of file DeepTauBase.h.

◆ TauCollection

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

Definition at line 62 of file DeepTauBase.h.

◆ TauDiscriminator

using deep_tau::DeepTauBase::TauDiscriminator = reco::TauDiscriminatorContainer

Definition at line 61 of file DeepTauBase.h.

◆ TauRef

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

Definition at line 63 of file DeepTauBase.h.

◆ TauRefProd

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

Definition at line 64 of file DeepTauBase.h.

◆ TauType

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

Definition at line 60 of file DeepTauBase.h.

◆ WPList

using deep_tau::DeepTauBase::WPList = std::vector<CutterPtr>

Definition at line 70 of file DeepTauBase.h.

Constructor & Destructor Documentation

◆ DeepTauBase()

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

Definition at line 76 of file DeepTauBase.cc.

       : 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_list = cfg.getParameter<std::vector<std::string>>(output_desc.first + "WP");
       for (const std::string& cut_str : cut_list) {
         workingPoints_[output_desc.first].push_back(std::make_unique<Cutter>(cut_str));
       }
     }
   }

References looper::cfg, outputs_, AlCaHLTBitMon_QueryRunRegistry::string, and workingPoints_.

◆ ~DeepTauBase()

deep_tau::DeepTauBase::~DeepTauBase ( )

inlineoverride

Definition at line 85 of file DeepTauBase.h.

85 {}

Member Function Documentation

◆ createOutputs()

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

privatevirtual

Definition at line 101 of file DeepTauBase.cc.

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

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

Referenced by produce().

◆ getPredictions()

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

◆ globalEndJob()

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

inlinestatic

Definition at line 90 of file DeepTauBase.h.

90 {}

◆ initializeGlobalCache()

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

static

Definition at line 108 of file DeepTauBase.cc.

                                                                                            {
     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);
   }

References looper::cfg, mps_splice::entry, edm::FileInPath::fullPath(), and AlCaHLTBitMon_QueryRunRegistry::string.

◆ produce()

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

override

Definition at line 93 of file DeepTauBase.cc.

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

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

Member Data Documentation

◆ cache_

const DeepTauCache* deep_tau::DeepTauBase::cache_

protected

Definition at line 104 of file DeepTauBase.h.

Referenced by DeepTauId::DeepTauId(), DPFIsolation::DPFIsolation(), DeepTauId::getPartialPredictions(), DPFIsolation::getPredictions(), DeepTauId::getPredictionsV1(), DeepTauId::getPredictionsV2(), and DeepTauId::globalEndJob().

◆ outputs_

OutputCollection deep_tau::DeepTauBase::outputs_

protected

Definition at line 103 of file DeepTauBase.h.

Referenced by createOutputs(), DeepTauBase(), DPFIsolation::GetOutputs(), DeepTauId::GetOutputs(), and DPFIsolation::getPredictions().

◆ pfcandToken_

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

protected

Definition at line 100 of file DeepTauBase.h.

Referenced by DPFIsolation::getPredictions(), and DeepTauId::getPredictions().

◆ tausToken_

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

protected

Definition at line 99 of file DeepTauBase.h.

Referenced by produce().

◆ vtxToken_

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

protected

Definition at line 101 of file DeepTauBase.h.

Referenced by DPFIsolation::getPredictions(), and DeepTauId::getPredictions().

◆ workingPoints_

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

protected

Definition at line 102 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

◆ Cutter

◆ CutterPtr

◆ ElectronCollection

◆ LorentzVectorXYZ

◆ MuonCollection

◆ OutputCollection

◆ TauCollection

◆ TauDiscriminator

◆ TauRef

◆ TauRefProd

◆ TauType

◆ WPList

Constructor & Destructor Documentation

◆ DeepTauBase()

◆ ~DeepTauBase()

Member Function Documentation

◆ createOutputs()

◆ getPredictions()

◆ globalEndJob()

◆ initializeGlobalCache()

◆ produce()

Member Data Documentation

◆ cache_

◆ outputs_

◆ pfcandToken_

◆ tausToken_

◆ vtxToken_

◆ workingPoints_