#include <DeepTauBase.h>

Inheritance diagram for deep_tau::DeepTauBase:

Classes
struct	Output

struct	TauDiscInfo

Public Types
enum	BasicDiscriminator { ChargedIsoPtSum, NeutralIsoPtSum, NeutralIsoPtSumWeight, FootprintCorrection, PhotonPtSumOutsideSignalCone, PUcorrPtSum }

using	CandidateCollection = edm::View< reco::Candidate >

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 = edm::View< reco::BaseTau >

using	TauDiscriminator = reco::TauDiscriminatorContainer

using	TauRef = edm::Ref< TauCollection >

using	TauRefProd = edm::RefProd< TauCollection >

using	WPList = std::vector< CutterPtr >

Public Types inherited from edm::stream::EDProducer< edm::GlobalCache< DeepTauCache > >
using	CacheTypes = CacheContexts< T...>

using	GlobalCache = typename CacheTypes::GlobalCache

using	HasAbility = AbilityChecker< T...>

using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache

using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache

using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >

using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache

using	RunCache = typename CacheTypes::RunCache

using	RunContext = RunContextT< RunCache, GlobalCache >

using	RunSummaryCache = typename CacheTypes::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

	EDProducer (const EDProducer &)=delete

bool	hasAbilityToProduceInBeginLumis () const final

bool	hasAbilityToProduceInBeginProcessBlocks () const final

bool	hasAbilityToProduceInBeginRuns () const final

bool	hasAbilityToProduceInEndLumis () const final

bool	hasAbilityToProduceInEndProcessBlocks () const final

bool	hasAbilityToProduceInEndRuns () const final

const EDProducer &	operator= (const EDProducer &)=delete

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

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

Public Attributes
uint8_t	andPrediscriminants_

std::vector< TauDiscInfo < pat::PATTauDiscriminator > >	patPrediscriminants_

std::vector< TauDiscInfo < reco::PFTauDiscriminator > >	recoPrediscriminants_

Protected Attributes
const DeepTauCache *	cache_

const bool	is_online_

OutputCollection	outputs_

edm::EDGetTokenT < CandidateCollection >	pfcandToken_

edm::EDGetTokenT< TauCollection >	tausToken_

edm::EDGetTokenT < reco::VertexCollection >	vtxToken_

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

Static Protected Attributes
static const std::vector < BasicDiscriminator >	requiredBasicDiscriminators_

static const std::vector < BasicDiscriminator >	requiredBasicDiscriminatorsdR03_

static const std::map < BasicDiscriminator, std::string >	stringFromDiscriminator_

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

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

Detailed Description

Definition at line 66 of file DeepTauBase.h.

Member Typedef Documentation

using deep_tau::DeepTauBase::CandidateCollection = edm::View<reco::Candidate>

Definition at line 70 of file DeepTauBase.h.

using deep_tau::DeepTauBase::Cutter = TauWPThreshold

Definition at line 76 of file DeepTauBase.h.

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

Definition at line 77 of file DeepTauBase.h.

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

Definition at line 73 of file DeepTauBase.h.

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

Definition at line 75 of file DeepTauBase.h.

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

Definition at line 74 of file DeepTauBase.h.

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

Definition at line 91 of file DeepTauBase.h.

using deep_tau::DeepTauBase::TauCollection = edm::View<reco::BaseTau>

Definition at line 69 of file DeepTauBase.h.

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

Definition at line 68 of file DeepTauBase.h.

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

Definition at line 71 of file DeepTauBase.h.

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

Definition at line 72 of file DeepTauBase.h.

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

Definition at line 78 of file DeepTauBase.h.

Member Enumeration Documentation

enum deep_tau::DeepTauBase::BasicDiscriminator

Enumerator
ChargedIsoPtSum
NeutralIsoPtSum
NeutralIsoPtSumWeight
FootprintCorrection
PhotonPtSumOutsideSignalCone
PUcorrPtSum

Definition at line 115 of file DeepTauBase.h.

                             {
       ChargedIsoPtSum,
       NeutralIsoPtSum,
       NeutralIsoPtSumWeight,
       FootprintCorrection,
       PhotonPtSumOutsideSignalCone,
       PUcorrPtSum
     };

Constructor & Destructor Documentation

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

Definition at line 91 of file DeepTauBase.cc.

References andPrediscriminants_, GOODCOLL_filter_cfg::cut, deep_tau::DeepTauBase::TauDiscInfo< ConsumeType >::cut, deep_tau::DeepTauBase::TauDiscInfo< ConsumeType >::disc_token, Exception, edm::ParameterSet::getParameter(), edm::ParameterSet::getParameterNamesForType(), is_online_, label, deep_tau::DeepTauBase::TauDiscInfo< ConsumeType >::label, outputs_, patPrediscriminants_, recoPrediscriminants_, AlCaHLTBitMon_QueryRunRegistry::string, HcalDetIdTransform::transform(), and workingPoints_.

       : tausToken_(consumes<TauCollection>(cfg.getParameter<edm::InputTag>("taus"))),
         pfcandToken_(consumes<CandidateCollection>(cfg.getParameter<edm::InputTag>("pfcands"))),
         vtxToken_(consumes<reco::VertexCollection>(cfg.getParameter<edm::InputTag>("vertices"))),
         is_online_(cfg.getParameter<bool>("is_online")),
         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));
       }
     }
 
     // prediscriminant operator
     // require the tau to pass the following prediscriminants
     const edm::ParameterSet& prediscriminantConfig = cfg.getParameter<edm::ParameterSet>("Prediscriminants");
 
     // determine boolean operator used on the prediscriminants
     std::string pdBoolOperator = prediscriminantConfig.getParameter<std::string>("BooleanOperator");
     // convert string to lowercase
     transform(pdBoolOperator.begin(), pdBoolOperator.end(), pdBoolOperator.begin(), ::tolower);
 
     if (pdBoolOperator == "and") {
       andPrediscriminants_ = 0x1;  //use chars instead of bools so we can do a bitwise trick later
     } else if (pdBoolOperator == "or") {
       andPrediscriminants_ = 0x0;
     } else {
       throw cms::Exception("TauDiscriminationProducerBase")
           << "PrediscriminantBooleanOperator defined incorrectly, options are: AND,OR";
     }
 
     // get the list of prediscriminants
     std::vector<std::string> prediscriminantsNames =
         prediscriminantConfig.getParameterNamesForType<edm::ParameterSet>();
 
     for (auto const& iDisc : prediscriminantsNames) {
       const edm::ParameterSet& iPredisc = prediscriminantConfig.getParameter<edm::ParameterSet>(iDisc);
       const edm::InputTag& label = iPredisc.getParameter<edm::InputTag>("Producer");
       double cut = iPredisc.getParameter<double>("cut");
 
       if (is_online_) {
         TauDiscInfo<reco::PFTauDiscriminator> thisDiscriminator;
         thisDiscriminator.label = label;
         thisDiscriminator.cut = cut;
         thisDiscriminator.disc_token = consumes<reco::PFTauDiscriminator>(label);
         recoPrediscriminants_.push_back(thisDiscriminator);
       } else {
         TauDiscInfo<pat::PATTauDiscriminator> thisDiscriminator;
         thisDiscriminator.label = label;
         thisDiscriminator.cut = cut;
         thisDiscriminator.disc_token = consumes<pat::PATTauDiscriminator>(label);
         patPrediscriminants_.push_back(thisDiscriminator);
       }
     }
   }

deep_tau::DeepTauBase::~DeepTauBase ( )

inlineoverride

Definition at line 94 of file DeepTauBase.h.

94 {}

Member Function Documentation

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

privatevirtual

Definition at line 182 of file DeepTauBase.cc.

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

Referenced by produce().

                                                                                                               {
     for (const auto& output_desc : outputs_) {
       const WPList* working_points = nullptr;
       if (workingPoints_.find(output_desc.first) != workingPoints_.end()) {
         working_points = &workingPoints_.at(output_desc.first);
       }
       auto result = output_desc.second.get_value(taus, pred, working_points, is_online_);
       event.put(std::move(result), output_desc.first);
     }
   }

virtual tensorflow::Tensor deep_tau::DeepTauBase::getPredictions	(	edm::Event &	event,
		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 99 of file DeepTauBase.h.

99 {}

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

static

Definition at line 193 of file DeepTauBase.cc.

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

                                                                                            {
     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 151 of file DeepTauBase.cc.

References createOutputs(), Exception, getPredictions(), edm::HandleBase::id(), is_online_, patPrediscriminants_, recoPrediscriminants_, runTauDisplay::taus, and tausToken_.

                                                                     {
     edm::Handle<TauCollection> taus;
     event.getByToken(tausToken_, taus);
     edm::ProductID tauProductID = taus.id();
 
     // load prediscriminators
     size_t nPrediscriminants =
         patPrediscriminants_.empty() ? recoPrediscriminants_.size() : patPrediscriminants_.size();
     for (size_t iDisc = 0; iDisc < nPrediscriminants; ++iDisc) {
       edm::ProductID discKeyId;
       if (is_online_) {
         recoPrediscriminants_[iDisc].fill(event);
         discKeyId = recoPrediscriminants_[iDisc].handle->keyProduct().id();
       } else {
         patPrediscriminants_[iDisc].fill(event);
         discKeyId = patPrediscriminants_[iDisc].handle->keyProduct().id();
       }
 
       // Check to make sure the product is correct for the discriminator.
       // If not, throw a more informative exception.
       if (tauProductID != discKeyId) {
         throw cms::Exception("MisconfiguredPrediscriminant")
             << "The tau collection has product ID: " << tauProductID
             << " but the pre-discriminator is keyed with product ID: " << discKeyId << std::endl;
       }
     }
 
     const tensorflow::Tensor& pred = getPredictions(event, taus);
     createOutputs(event, pred, taus);
   }

Member Data Documentation

uint8_t deep_tau::DeepTauBase::andPrediscriminants_

Definition at line 111 of file DeepTauBase.h.

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

const DeepTauCache* deep_tau::DeepTauBase::cache_

protected

Definition at line 135 of file DeepTauBase.h.

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

const bool deep_tau::DeepTauBase::is_online_

protected

Definition at line 133 of file DeepTauBase.h.

Referenced by createOutputs(), DeepTauId::createTauBlockInputs(), DeepTauBase(), DeepTauId::getPredictions(), and produce().

OutputCollection deep_tau::DeepTauBase::outputs_

protected

Definition at line 134 of file DeepTauBase.h.

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

std::vector<TauDiscInfo<pat::PATTauDiscriminator> > deep_tau::DeepTauBase::patPrediscriminants_

Definition at line 112 of file DeepTauBase.h.

Referenced by DeepTauBase(), DeepTauId::getPredictions(), and produce().

edm::EDGetTokenT<CandidateCollection> deep_tau::DeepTauBase::pfcandToken_

protected

Definition at line 130 of file DeepTauBase.h.

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

std::vector<TauDiscInfo<reco::PFTauDiscriminator> > deep_tau::DeepTauBase::recoPrediscriminants_

Definition at line 113 of file DeepTauBase.h.

Referenced by DeepTauBase(), DeepTauId::getPredictions(), and produce().

const std::vector< bd > deep_tau::DeepTauBase::requiredBasicDiscriminators_

staticprotected

Initial value:

= {bd::ChargedIsoPtSum,
                                                                             bd::NeutralIsoPtSum,
                                                                             bd::NeutralIsoPtSumWeight,
                                                                             bd::PhotonPtSumOutsideSignalCone,
                                                                             bd::PUcorrPtSum}

Definition at line 138 of file DeepTauBase.h.

Referenced by DeepTauId::getPredictions().

const std::vector< bd > deep_tau::DeepTauBase::requiredBasicDiscriminatorsdR03_

staticprotected

Initial value:

= {bd::ChargedIsoPtSum,
                                                                                 bd::NeutralIsoPtSum,
                                                                                 bd::NeutralIsoPtSumWeight,
                                                                                 bd::PhotonPtSumOutsideSignalCone,
                                                                                 bd::FootprintCorrection}

Definition at line 139 of file DeepTauBase.h.

Referenced by DeepTauId::getPredictions().

const std::map< bd, std::string > deep_tau::DeepTauBase::stringFromDiscriminator_

staticprotected

Initial value:

{
    {bd::ChargedIsoPtSum, "ChargedIsoPtSum"},
    {bd::NeutralIsoPtSum, "NeutralIsoPtSum"},
    {bd::NeutralIsoPtSumWeight, "NeutralIsoPtSumWeight"},
    {bd::FootprintCorrection, "TauFootprintCorrection"},
    {bd::PhotonPtSumOutsideSignalCone, "PhotonPtSumOutsideSignalCone"},
    {bd::PUcorrPtSum, "PUcorrPtSum"}}