Inheritance diagram for DeepMETSonicProducer:

Public Member Functions
void	acquire (edm::Event const &iEvent, edm::EventSetup const &iSetup, Input &iInput) override

	DeepMETSonicProducer (const edm::ParameterSet &)

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

Public Member Functions inherited from TritonEDProducerT< G, Capabilities >
void	endStream () final

	TritonEDProducerT (edm::ParameterSet const &cfg)

virtual void	tritonEndStream ()

Public Member Functions inherited from SonicEDProducer< TritonClient, edm::GlobalCache< G >, Capabilities... >
void	produce (edm::Event &iEvent, edm::EventSetup const &iSetup) final

virtual void	produce (edm::Event &iEvent, edm::EventSetup const &iSetup, Output const &iOutput)=0

	SonicEDProducer (edm::ParameterSet const &cfg)

	~SonicEDProducer () override=default

Public Member Functions inherited from SonicAcquirer< TritonClient, edm::stream::EDProducer< edm::ExternalWork, Capabilities... > >
void	acquire (edm::Event const &iEvent, edm::EventSetup const &iSetup, edm::WaitingTaskWithArenaHolder holder) final

virtual void	acquire (edm::Event const &iEvent, edm::EventSetup const &iSetup, Input &iInput)=0

void	beginStream (edm::StreamID) override

	SonicAcquirer (edm::ParameterSet const &cfg)

	~SonicAcquirer () override=default

Public Member Functions inherited from edm::stream::EDProducer< edm::ExternalWork, Capabilities... >
	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 Public Member Functions inherited from TritonEDProducerT< G, Capabilities >
static void	globalEndJob (G *)

static std::unique_ptr< G >	initializeGlobalCache (edm::ParameterSet const &pset)

Private Attributes
const bool	ignore_leptons_

const unsigned int	max_n_pf_

const float	norm_

const edm::EDGetTokenT< std::vector< pat::PackedCandidate > >	pf_token_

float	px_leptons_

float	py_leptons_

const float	scale_

Additional Inherited Members
Public Types inherited from SonicEDProducer< TritonClient, edm::GlobalCache< G >, Capabilities... >
typedef TritonClient ::Output	Output

Public Types inherited from SonicAcquirer< TritonClient, edm::stream::EDProducer< edm::ExternalWork, Capabilities... > >
typedef TritonClient ::Input	Input

Public Types inherited from edm::stream::EDProducer< edm::ExternalWork, Capabilities... >
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

Protected Member Functions inherited from SonicAcquirer< TritonClient, edm::stream::EDProducer< edm::ExternalWork, Capabilities... > >
void	makeClient ()

Protected Attributes inherited from SonicAcquirer< TritonClient, edm::stream::EDProducer< edm::ExternalWork, Capabilities... > >
std::unique_ptr< TritonClient >	client_

edm::ParameterSet	clientPset_

std::string	debugName_

std::chrono::time_point< std::chrono::high_resolution_clock >	t_dispatch_

bool	verbose_

Detailed Description

Definition at line 13 of file DeepMETSonicProducer.cc.

Constructor & Destructor Documentation

◆ DeepMETSonicProducer()

DeepMETSonicProducer::DeepMETSonicProducer ( const edm::ParameterSet & cfg )

explicit

Definition at line 30 of file DeepMETSonicProducer.cc.

     : TritonEDProducer<>(cfg),
       pf_token_(consumes<std::vector<pat::PackedCandidate>>(cfg.getParameter<edm::InputTag>("pf_src"))),
       norm_(cfg.getParameter<double>("norm_factor")),
       ignore_leptons_(cfg.getParameter<bool>("ignore_leptons")),
       max_n_pf_(cfg.getParameter<unsigned int>("max_n_pf")),
       scale_(1.0 / norm_) {
   produces<pat::METCollection>();
 }

Member Function Documentation

◆ acquire()

void DeepMETSonicProducer::acquire	(	edm::Event const &	iEvent,
		edm::EventSetup const &	iSetup,
		Input &	iInput
	)

override

Definition at line 40 of file DeepMETSonicProducer.cc.

References funct::abs(), deepmet_helper::charge_embedding, SonicAcquirer< TritonClient, edm::stream::EDProducer< edm::ExternalWork, Capabilities... > >::client_, iEvent, ignore_leptons_, input, max_n_pf_, HiggsValidation_cfi::pdg_id, deepmet_helper::pdg_id_embedding, packedPFCandidateRefMixer_cfi::pf, pf_token_, px_leptons_, py_leptons_, deepmet_helper::rm_outlier(), scale_, and deepmet_helper::scale_and_rm_outlier().

                                                                                                      {
   // one event per batch
   client_->setBatchSize(1);
   px_leptons_ = 0.;
   py_leptons_ = 0.;
 
   auto const& pfs = iEvent.get(pf_token_);
 
   auto& input = iInput.at("input");
   auto pfdata = input.allocate<float>();
   auto& vpfdata = (*pfdata)[0];
 
   auto& input_cat0 = iInput.at("input_cat0");
   auto pfchg = input_cat0.allocate<float>();
   auto& vpfchg = (*pfchg)[0];
 
   auto& input_cat1 = iInput.at("input_cat1");
   auto pfpdgId = input_cat1.allocate<float>();
   auto& vpfpdgId = (*pfpdgId)[0];
 
   auto& input_cat2 = iInput.at("input_cat2");
   auto pffromPV = input_cat2.allocate<float>();
   auto& vpffromPV = (*pffromPV)[0];
 
   size_t i_pf = 0;
   for (const auto& pf : pfs) {
     if (ignore_leptons_) {
       int pdg_id = std::abs(pf.pdgId());
       if (pdg_id == 11 || pdg_id == 13) {
         px_leptons_ += pf.px();
         py_leptons_ += pf.py();
         continue;
       }
     }
 
     // PF keys [b'PF_dxy', b'PF_dz', b'PF_eta', b'PF_mass', b'PF_pt', b'PF_puppiWeight', b'PF_px', b'PF_py']
     vpfdata.push_back(rm_outlier(pf.dxy()));
     vpfdata.push_back(rm_outlier(pf.dz()));
     vpfdata.push_back(rm_outlier(pf.eta()));
     vpfdata.push_back(rm_outlier(pf.mass()));
     vpfdata.push_back(scale_and_rm_outlier(pf.pt(), scale_));
     vpfdata.push_back(rm_outlier(pf.puppiWeight()));
     vpfdata.push_back(scale_and_rm_outlier(pf.px(), scale_));
     vpfdata.push_back(scale_and_rm_outlier(pf.py(), scale_));
 
     vpfchg.push_back(charge_embedding.at(pf.charge()));
 
     vpfpdgId.push_back(pdg_id_embedding.at(pf.pdgId()));
 
     vpffromPV.push_back(pf.fromPV());
 
     ++i_pf;
     if (i_pf == max_n_pf_) {
       edm::LogWarning("acquire")
           << "<DeepMETSonicProducer::acquire>:" << std::endl
           << " The number of particles is equal to or exceeds the maximum considerable for DeepMET" << std::endl;
       break;
     }
   }
 
   // fill the remaining with zeros
   // resize the vector to 4500 for zero-padding
   vpfdata.resize(8 * max_n_pf_);
   vpfchg.resize(max_n_pf_);
   vpfpdgId.resize(max_n_pf_);
   vpffromPV.resize(max_n_pf_);
 
   input.toServer(pfdata);
   input_cat0.toServer(pfchg);
   input_cat1.toServer(pfpdgId);
   input_cat2.toServer(pffromPV);
 }

◆ fillDescriptions()

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

static

Definition at line 134 of file DeepMETSonicProducer.cc.

References edm::ConfigurationDescriptions::add(), submitPVResolutionJobs::desc, TritonClient::fillPSetDescription(), and ProducerED_cfi::InputTag.

                                                                                       {
   edm::ParameterSetDescription desc;
   TritonClient::fillPSetDescription(desc);
   desc.add<edm::InputTag>("pf_src", edm::InputTag("packedPFCandidates"));
   desc.add<bool>("ignore_leptons", false);
   desc.add<double>("norm_factor", 50.);
   desc.add<unsigned int>("max_n_pf", 4500);
   descriptions.add("deepMETSonicProducer", desc);
 }

◆ produce()

void DeepMETSonicProducer::produce	(	edm::Event &	iEvent,
		edm::EventSetup const &	iSetup,
		Output const &	iOutput
	)

override

Definition at line 113 of file DeepMETSonicProducer.cc.

References Matriplex::hypot(), iEvent, LogDebug, eostools::move(), norm_, skim900GeV_StreamA_MinBiasPD_cfg::output1, PatBasicFWLiteJetAnalyzer_Selector_cfg::outputs, multPhiCorr_741_25nsDY_cfi::px, px_leptons_, multPhiCorr_741_25nsDY_cfi::py, and py_leptons_.

                                                                                                        {
   const auto& output1 = iOutput.begin()->second;
   const auto& outputs = output1.fromServer<float>();
 
   // outputs are px and py
   float px = outputs[0][0] * norm_;
   float py = outputs[0][1] * norm_;
 
   // subtract the lepton pt contribution
   px -= px_leptons_;
   py -= py_leptons_;
 
   LogDebug("produce") << "<DeepMETSonicProducer::produce>:" << std::endl
                       << " MET from DeepMET Sonic Producer is MET_x " << px << " and MET_y " << py << std::endl;
 
   auto pf_mets = std::make_unique<pat::METCollection>();
   const reco::Candidate::LorentzVector p4(px, py, 0., std::hypot(px, py));
   pf_mets->emplace_back(reco::MET(p4, {}));
   iEvent.put(std::move(pf_mets));
 }

Member Data Documentation

◆ ignore_leptons_

const bool DeepMETSonicProducer::ignore_leptons_

private

Definition at line 23 of file DeepMETSonicProducer.cc.

Referenced by acquire().

◆ max_n_pf_

const unsigned int DeepMETSonicProducer::max_n_pf_

private

Definition at line 24 of file DeepMETSonicProducer.cc.

Referenced by acquire().

◆ norm_

const float DeepMETSonicProducer::norm_

private

Definition at line 22 of file DeepMETSonicProducer.cc.

Referenced by produce().

◆ pf_token_

const edm::EDGetTokenT<std::vector<pat::PackedCandidate> > DeepMETSonicProducer::pf_token_

private

Definition at line 21 of file DeepMETSonicProducer.cc.

Referenced by acquire().

◆ px_leptons_

float DeepMETSonicProducer::px_leptons_

private

Definition at line 26 of file DeepMETSonicProducer.cc.

Referenced by acquire(), and produce().

◆ py_leptons_

float DeepMETSonicProducer::py_leptons_

private

Definition at line 27 of file DeepMETSonicProducer.cc.

Referenced by acquire(), and produce().

◆ scale_

const float DeepMETSonicProducer::scale_

private

Definition at line 25 of file DeepMETSonicProducer.cc.

Referenced by acquire().

Public Member Functions

Static Public Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

◆ DeepMETSonicProducer()

Member Function Documentation

◆ acquire()

◆ fillDescriptions()

◆ produce()

Member Data Documentation

◆ ignore_leptons_

◆ max_n_pf_

◆ norm_

◆ pf_token_

◆ px_leptons_

◆ py_leptons_

◆ scale_