L1NNTauProducer Class Reference

Inheritance diagram for L1NNTauProducer:

Public Member Functions
	L1NNTauProducer (const edm::ParameterSet &, const TauNNTFCache *)
	~L1NNTauProducer () override
Public Member Functions inherited from edm::stream::EDProducer< edm::GlobalCache< TauNNTFCache > >
	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 TauNNTFCache *)
static std::unique_ptr < TauNNTFCache >	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< TauNNTFCache > >
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 ( const edm::ParameterSet &  cfg, const TauNNTFCache *  cache  )

explicit

Definition at line 43 of file L1NNTauProducer.cc.

References utilities::cache(), fNParticles_, fTauNNId_, edm::ParameterSet::getParameter(), 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, lNNFile, fNParticles_);
  produces<l1t::PFTauCollection>("L1PFTausNN");
}

L1NNTauProducer::~L1NNTauProducer ( )

override

Definition at line 159 of file L1NNTauProducer.cc.

{}

Member Function Documentation

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

private

Definition at line 118 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 reco::LeafCandidate::pt().

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

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

static

Definition at line 147 of file L1NNTauProducer.cc.

References edm::ConfigurationDescriptions::add(), edm::ParameterSetDescription::add(), submitPVResolutionJobs::desc, HLT_FULL_cff::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);
}

void L1NNTauProducer::globalEndJob ( const TauNNTFCache *  cache )

static

Definition at line 63 of file L1NNTauProducer.cc.

References TauNNTFCache::graphDef.

                                                            {
  if (cache->graphDef != nullptr) {
    delete cache->graphDef;
  }
}

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

static

Definition at line 55 of file L1NNTauProducer.cc.

References utilities::cache(), personalPlayback::fp, edm::ParameterSet::getParameter(), TauNNTFCache::graphDef, tensorflow::loadGraphDef(), tensorflow::setLogging(), and AlCaHLTBitMon_QueryRunRegistry::string.

                                                                                             {
  tensorflow::setLogging("3");
  std::string lNNFile = cfg.getParameter<std::string>("NNFileName");
  edm::FileInPath fp(lNNFile);
  TauNNTFCache* cache = new TauNNTFCache();
  cache->graphDef = tensorflow::loadGraphDef(fp.fullPath());
  return std::unique_ptr<TauNNTFCache>(cache);
}

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

overrideprivate

Definition at line 68 of file L1NNTauProducer.cc.

References funct::abs(), addTau(), l1t::PFCandidate::ChargedHadron, reco::deltaR2(), l1t::PFCandidate::Electron, fConeSize_, fL1PFToken_, fMaxTaus_, edm::Event::getByToken(), mps_fire::i, dqmiolumiharvest::j, eostools::move(), reco::LeafCandidate::pt(), edm::Event::put(), 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

double L1NNTauProducer::fConeSize_

private

Definition at line 34 of file L1NNTauProducer.cc.

Referenced by addTau(), and produce().

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

private

Definition at line 38 of file L1NNTauProducer.cc.

Referenced by produce().

int L1NNTauProducer::fMaxTaus_

private

Definition at line 36 of file L1NNTauProducer.cc.

Referenced by produce().

int L1NNTauProducer::fNParticles_

private

Definition at line 37 of file L1NNTauProducer.cc.

Referenced by L1NNTauProducer().

double L1NNTauProducer::fSeedPt_

private

Definition at line 33 of file L1NNTauProducer.cc.

Referenced by addTau().

std::unique_ptr<TauNNId> L1NNTauProducer::fTauNNId_

private

Definition at line 27 of file L1NNTauProducer.cc.

Referenced by addTau(), and L1NNTauProducer().

double L1NNTauProducer::fTauSize_

private

Definition at line 35 of file L1NNTauProducer.cc.

Referenced by addTau().