Inheritance diagram for L1NNTauProducer:

Public Member Functions
	L1NNTauProducer (const edm::ParameterSet &, const tensorflow::SessionCache *)

	~L1NNTauProducer () override

Public Member Functions inherited from edm::stream::EDProducer< edm::GlobalCache< tensorflow::SessionCache > >
	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	fillDescriptions (edm::ConfigurationDescriptions &descriptions)

static void	globalEndJob (const tensorflow::SessionCache *)

static std::unique_ptr< tensorflow::SessionCache >	initializeGlobalCache (const edm::ParameterSet &)

Private Member Functions
void	addTau (const l1t::PFCandidate &iCand, const l1t::PFCandidateCollection &iParts, std::unique_ptr< PFTauCollection > &outputTaus)

void	produce (edm::Event &iEvent, const edm::EventSetup &iSetup) override

Private Attributes
double	fConeSize_

edm::EDGetTokenT< vector< l1t::PFCandidate > >	fL1PFToken_

int	fMaxTaus_

int	fNParticles_

double	fSeedPt_

std::unique_ptr< TauNNId >	fTauNNId_

double	fTauSize_

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer< edm::GlobalCache< tensorflow::SessionCache > >
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

Detailed Description

Definition at line 17 of file L1NNTauProducer.cc.

Constructor & Destructor Documentation

◆ L1NNTauProducer()

L1NNTauProducer::L1NNTauProducer	(	const edm::ParameterSet &	cfg,
		const tensorflow::SessionCache *	cache
	)

explicit

Definition at line 43 of file L1NNTauProducer.cc.

References utilities::cache(), looper::cfg, fNParticles_, fTauNNId_, and AlCaHLTBitMon_QueryRunRegistry::string.

     : fSeedPt_(cfg.getParameter<double>("seedpt")),
       fConeSize_(cfg.getParameter<double>("conesize")),
       fTauSize_(cfg.getParameter<double>("tausize")),
       fMaxTaus_(cfg.getParameter<int>("maxtaus")),
       fNParticles_(cfg.getParameter<int>("nparticles")),
       fL1PFToken_(consumes<vector<l1t::PFCandidate>>(cfg.getParameter<edm::InputTag>("L1PFObjects"))) {
   std::string lNNFile = cfg.getParameter<std::string>("NNFileName");  //,"L1Trigger/Phase2L1Taus/data/tau_3layer.pb");
   fTauNNId_ = std::make_unique<TauNNId>(lNNFile.find("v0") == std::string::npos ? "input_1:0" : "dense_1_input:0",
                                         cache->getSession(),
                                         lNNFile,
                                         fNParticles_);
   produces<l1t::PFTauCollection>("L1PFTausNN");
 }

◆ ~L1NNTauProducer()

L1NNTauProducer::~L1NNTauProducer ( )

override

Definition at line 155 of file L1NNTauProducer.cc.

155 {}

Member Function Documentation

◆ addTau()

void L1NNTauProducer::addTau	(	const l1t::PFCandidate &	iCand,
		const l1t::PFCandidateCollection &	iParts,
		std::unique_ptr< PFTauCollection > &	outputTaus
	)

private

Definition at line 114 of file L1NNTauProducer.cc.

References l1t::PFCandidate::ChargedHadron, reco::deltaR2(), l1t::PFCandidate::Electron, fConeSize_, fSeedPt_, fTauNNId_, fTauSize_, mps_fire::i, dqmiolumiharvest::j, mkfit::NN, l1t::PFCandidate::Photon, and jetUpdater_cfi::sort.

Referenced by produce().

                                                                             {
   l1t::PFCandidateCollection pfTauCands;
   math::PtEtaPhiMLorentzVector lTot(0, 0, 0, 0);
   math::PtEtaPhiMLorentzVector lCand(0, 0, 0, 0);
   int lId = 0;
   for (const auto& l1PFCand : iParts) {
     if (reco::deltaR2(iCand, l1PFCand) > fConeSize_ * fConeSize_)
       continue;
     math::PtEtaPhiMLorentzVector pVec(l1PFCand.pt(), l1PFCand.eta(), l1PFCand.phi(), 0);
     lTot += pVec;
     if (reco::deltaR2(iCand, l1PFCand) < fTauSize_ * fTauSize_ &&
         (l1PFCand.id() == l1t::PFCandidate::Electron || l1PFCand.id() == l1t::PFCandidate::ChargedHadron ||
          l1PFCand.id() == l1t::PFCandidate::Photon)) {
       lId++;
       lCand += pVec;
     }
     pfTauCands.push_back(l1PFCand);
   }
   if (lTot.Pt() < fSeedPt_)
     return;
   std::sort(
       pfTauCands.begin(), pfTauCands.end(), [](l1t::PFCandidate i, l1t::PFCandidate j) { return (i.pt() > j.pt()); });
   float NN = fTauNNId_->compute(iCand, pfTauCands);
   math::PtEtaPhiMLorentzVector tempP4(lCand.Pt(), lCand.Eta(), lCand.Phi(), lCand.M());
   l1t::PFTau l1PFTau(tempP4, NN, 0, lId);
   outputTaus->push_back(l1PFTau);
 }

◆ fillDescriptions()

void L1NNTauProducer::fillDescriptions ( edm::ConfigurationDescriptions & descriptions )

static

Definition at line 143 of file L1NNTauProducer.cc.

References edm::ConfigurationDescriptions::add(), submitPVResolutionJobs::desc, ProducerED_cfi::InputTag, and AlCaHLTBitMon_QueryRunRegistry::string.

                                                                                  {
   // L1NNTauProducer
   edm::ParameterSetDescription desc;
   desc.add<std::string>("NNFileName", "L1Trigger/Phase2L1ParticleFlow/data/tau_3layer.pb");
   desc.add<double>("tausize", 0.1);
   desc.add<int>("maxtaus", 5);
   desc.add<int>("nparticles", 10);
   desc.add<double>("conesize", 0.4);
   desc.add<double>("seedpt", 20);
   desc.add<edm::InputTag>("L1PFObjects", edm::InputTag("L1PFProducer", "l1pfCandidates"));
   descriptions.add("L1NNTauProducer", desc);
 }

◆ globalEndJob()

static void L1NNTauProducer::globalEndJob ( const tensorflow::SessionCache * )

inlinestatic

Definition at line 24 of file L1NNTauProducer.cc.

24 {};

◆ initializeGlobalCache()

std::unique_ptr< tensorflow::SessionCache > L1NNTauProducer::initializeGlobalCache ( const edm::ParameterSet & cfg )

static

Definition at line 58 of file L1NNTauProducer.cc.

References looper::cfg, edm::FileInPath::fullPath(), HLT_2023v12_cff::graphPath, tensorflow::setLogging(), and AlCaHLTBitMon_QueryRunRegistry::string.

                                                                                                        {
   tensorflow::setLogging("3");
   std::string graphPath = edm::FileInPath(cfg.getParameter<std::string>("NNFileName")).fullPath();
   return std::make_unique<tensorflow::SessionCache>(graphPath);
 }

◆ produce()

void L1NNTauProducer::produce	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

overrideprivate

Definition at line 64 of file L1NNTauProducer.cc.

References funct::abs(), addTau(), l1t::PFCandidate::ChargedHadron, reco::deltaR2(), l1t::PFCandidate::Electron, fConeSize_, fL1PFToken_, fMaxTaus_, mps_fire::i, testProducerWithPsetDescEmpty_cfi::i1, iEvent, dqmiolumiharvest::j, eostools::move(), jetUpdater_cfi::sort, and track_trigger_eta_max.

                                                                            {
   edm::Handle<l1t::PFCandidateCollection> l1PFCandidates;
   iEvent.getByToken(fL1PFToken_, l1PFCandidates);
 
   std::vector<unique_ptr<l1t::PFCandidate>> pfChargedHadrons_sort_v;
   std::vector<unique_ptr<l1t::PFCandidate>> pfChargedHadrons_seeds_v;
   for (const auto& l1PFCand : *l1PFCandidates)
     if ((l1PFCand.id() == l1t::PFCandidate::ChargedHadron || l1PFCand.id() == l1t::PFCandidate::Electron) &&
         std::abs(l1PFCand.eta()) < track_trigger_eta_max)
       pfChargedHadrons_sort_v.push_back(std::make_unique<l1t::PFCandidate>(l1PFCand));
 
   std::sort(
       pfChargedHadrons_sort_v.begin(),
       pfChargedHadrons_sort_v.end(),
       [](std::unique_ptr<l1t::PFCandidate>& i, std::unique_ptr<l1t::PFCandidate>& j) { return (i->pt() > j->pt()); });
 
   auto lTaus = std::make_unique<l1t::PFTauCollection>();
   if (pfChargedHadrons_sort_v.empty()) {
     if (lTaus->empty()) {
       PFTau dummy;
       lTaus->push_back(dummy);
     }
     iEvent.put(std::move(lTaus), "L1PFTausNN");
     return;
   }
   pfChargedHadrons_seeds_v.push_back(std::move(pfChargedHadrons_sort_v[0]));
   for (unsigned int i0 = 1; i0 < pfChargedHadrons_sort_v.size(); i0++) {
     bool pMatch = false;
     for (unsigned int i1 = 0; i1 < pfChargedHadrons_seeds_v.size(); i1++) {
       if (reco::deltaR2(*(pfChargedHadrons_seeds_v[i1]), *(pfChargedHadrons_sort_v[i0])) < fConeSize_ * fConeSize_)
         pMatch = true;
     }
     if (pMatch)
       continue;
     pfChargedHadrons_seeds_v.push_back(std::move(pfChargedHadrons_sort_v[i0]));
     if (int(pfChargedHadrons_seeds_v.size()) > fMaxTaus_ - 1)
       break;
   }
   for (unsigned int i0 = 0; i0 < pfChargedHadrons_seeds_v.size(); i0++) {
     addTau(*(pfChargedHadrons_seeds_v[i0]), (*l1PFCandidates), lTaus);
   }
   if (lTaus->empty()) {
     PFTau dummy;
     lTaus->push_back(dummy);
   }
   std::sort(lTaus->begin(), lTaus->end(), [](l1t::PFTau i, l1t::PFTau j) { return (i.pt() > j.pt()); });
   iEvent.put(std::move(lTaus), "L1PFTausNN");
 }

Member Data Documentation

◆ fConeSize_

double L1NNTauProducer::fConeSize_

private

Definition at line 34 of file L1NNTauProducer.cc.

Referenced by addTau(), and produce().

◆ fL1PFToken_

edm::EDGetTokenT<vector<l1t::PFCandidate> > L1NNTauProducer::fL1PFToken_

private

Definition at line 38 of file L1NNTauProducer.cc.

Referenced by produce().

◆ fMaxTaus_

int L1NNTauProducer::fMaxTaus_

private

Definition at line 36 of file L1NNTauProducer.cc.

Referenced by produce().

◆ fNParticles_

int L1NNTauProducer::fNParticles_

private

Definition at line 37 of file L1NNTauProducer.cc.

Referenced by L1NNTauProducer().

◆ fSeedPt_

double L1NNTauProducer::fSeedPt_

private

Definition at line 33 of file L1NNTauProducer.cc.

Referenced by addTau().

◆ fTauNNId_

std::unique_ptr<TauNNId> L1NNTauProducer::fTauNNId_

private

Definition at line 24 of file L1NNTauProducer.cc.

Referenced by addTau(), and L1NNTauProducer().

◆ fTauSize_

double L1NNTauProducer::fTauSize_

private

Definition at line 35 of file L1NNTauProducer.cc.

Referenced by addTau().

Public Member Functions

Static Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

◆ L1NNTauProducer()

◆ ~L1NNTauProducer()

Member Function Documentation

◆ addTau()

◆ fillDescriptions()

◆ globalEndJob()

◆ initializeGlobalCache()

◆ produce()

Member Data Documentation

◆ fConeSize_

◆ fL1PFToken_

◆ fMaxTaus_

◆ fNParticles_

◆ fSeedPt_

◆ fTauNNId_

◆ fTauSize_