#include <JetId.h>

Public Member Functions
float	compute (const l1t::PFJet &iJet, float vz, bool useRawPt)

ap_fixed< 16, 6 >	computeFixed (const l1t::PFJet &iJet, float vz, bool useRawPt)

float	EvaluateNN ()

ap_fixed< 16, 6 >	EvaluateNNFixed ()

	JetId (const std::string &iInput, const std::string &iOutput, const std::shared_ptr< hls4mlEmulator::Model > model, int iNParticles)

	JetId (const std::string &iInput, const std::string &iOutput, const BJetTFCache *cache, int iNParticles)

void	setNNVectorVar ()

	~JetId ()=default

Private Attributes
unique_ptr< int[]>	fCharge_

unique_ptr< float[]>	fDX_

unique_ptr< float[]>	fDY_

unique_ptr< float[]>	fDZ_

unique_ptr< float[]>	fEta_

unique_ptr< int[]>	fId_

std::string	fInput_

int	fNParticles_

std::string	fOutput_

unique_ptr< float[]>	fPhi_

unique_ptr< float[]>	fPt_

std::shared_ptr< hls4mlEmulator::Model >	modelRef_

std::vector< float >	NNvectorVar_

tensorflow::Session *	sessionRef_

Detailed Description

Definition at line 23 of file JetId.h.

Constructor & Destructor Documentation

◆ JetId() [1/2]

JetId::JetId	(	const std::string &	iInput,
		const std::string &	iOutput,
		const std::shared_ptr< hls4mlEmulator::Model >	model,
		int	iNParticles
	)

Definition at line 5 of file JetId.cc.

References fCharge_, fDX_, fDY_, fDZ_, fEta_, fId_, fInput_, fNParticles_, fOutput_, fPhi_, fPt_, and NNvectorVar_.

     : modelRef_(model) {
   NNvectorVar_.clear();
   fNParticles_ = iNParticles;
 
   fPt_ = std::make_unique<float[]>(fNParticles_);
   fEta_ = std::make_unique<float[]>(fNParticles_);
   fPhi_ = std::make_unique<float[]>(fNParticles_);
   fId_ = std::make_unique<int[]>(fNParticles_);
   fCharge_ = std::make_unique<int[]>(fNParticles_);
   fDZ_ = std::make_unique<float[]>(fNParticles_);
   fDX_ = std::make_unique<float[]>(fNParticles_);
   fDY_ = std::make_unique<float[]>(fNParticles_);
   fInput_ = iInput;
   fOutput_ = iOutput;
 }

◆ JetId() [2/2]

JetId::JetId	(	const std::string &	iInput,
		const std::string &	iOutput,
		const BJetTFCache *	cache,
		int	iNParticles
	)

Definition at line 26 of file JetId.cc.

References fCharge_, fDX_, fDY_, fDZ_, fEta_, fId_, fInput_, fNParticles_, fOutput_, fPhi_, fPt_, and NNvectorVar_.

     : sessionRef_(cache->session) {
   NNvectorVar_.clear();
   fNParticles_ = iNParticles;
 
   fPt_ = std::make_unique<float[]>(fNParticles_);
   fEta_ = std::make_unique<float[]>(fNParticles_);
   fPhi_ = std::make_unique<float[]>(fNParticles_);
   fId_ = std::make_unique<int[]>(fNParticles_);
   fCharge_ = std::make_unique<int[]>(fNParticles_);
   fDZ_ = std::make_unique<float[]>(fNParticles_);
   fDX_ = std::make_unique<float[]>(fNParticles_);
   fDY_ = std::make_unique<float[]>(fNParticles_);
   fInput_ = iInput;
   fOutput_ = iOutput;
 }

◆ ~JetId()

JetId::~JetId ( )

default

Member Function Documentation

◆ compute()

float JetId::compute	(	const l1t::PFJet &	iJet,
		float	vz,
		bool	useRawPt
	)

Definition at line 90 of file JetId.cc.

References l1t::PFJet::constituents(), srCondWrite_cfg::deltaPhi, reco::LeafCandidate::eta(), EvaluateNN(), fCharge_, fDX_, fDY_, fDZ_, fEta_, fId_, fNParticles_, fPhi_, fPt_, mps_fire::i, dqmiolumiharvest::j, phi, reco::LeafCandidate::phi(), reco::LeafCandidate::pt(), l1t::PFJet::rawPt(), setNNVectorVar(), jetUpdater_cfi::sort, and btvMC_cff::vz.

                                                                   {
   for (int i0 = 0; i0 < fNParticles_; i0++) {
     fPt_.get()[i0] = 0;
     fEta_.get()[i0] = 0;
     fPhi_.get()[i0] = 0;
     fId_.get()[i0] = 0;
     fCharge_.get()[i0] = 0;
     fDZ_.get()[i0] = 0;
     fDX_.get()[i0] = 0;
     fDY_.get()[i0] = 0;
   }
   auto iParts = iJet.constituents();
   std::sort(iParts.begin(), iParts.end(), [](edm::Ptr<l1t::PFCandidate> i, edm::Ptr<l1t::PFCandidate> j) {
     return (i->pt() > j->pt());
   });
   float jetpt = useRawPt ? iJet.rawPt() : iJet.pt();
   for (unsigned int i0 = 0; i0 < iParts.size(); i0++) {
     if (i0 >= (unsigned int)fNParticles_)
       break;
     fPt_.get()[i0] = iParts[i0]->pt() / jetpt;
     fEta_.get()[i0] = iParts[i0]->eta() - iJet.eta();
     fPhi_.get()[i0] = deltaPhi(iParts[i0]->phi(), iJet.phi());
     fId_.get()[i0] = iParts[i0]->id();
     fCharge_.get()[i0] = iParts[i0]->charge();
     if (iParts[i0]->pfTrack().isNonnull()) {
       fDX_.get()[i0] = iParts[i0]->pfTrack()->vx();
       fDY_.get()[i0] = iParts[i0]->pfTrack()->vy();
       fDZ_.get()[i0] = iParts[i0]->pfTrack()->vz() - vz;
     }
   }
   setNNVectorVar();
   return EvaluateNN();
 }

◆ computeFixed()

ap_fixed< 16, 6 > JetId::computeFixed	(	const l1t::PFJet &	iJet,
		float	vz,
		bool	useRawPt
	)

Definition at line 124 of file JetId.cc.

References l1t::PFJet::constituents(), srCondWrite_cfg::deltaPhi, reco::LeafCandidate::eta(), EvaluateNNFixed(), fCharge_, fDX_, fDY_, fDZ_, fEta_, fId_, fNParticles_, fPhi_, fPt_, mps_fire::i, dqmiolumiharvest::j, phi, reco::LeafCandidate::phi(), reco::LeafCandidate::pt(), l1t::PFJet::rawPt(), setNNVectorVar(), jetUpdater_cfi::sort, and btvMC_cff::vz.

                                                                                  {
   for (int i0 = 0; i0 < fNParticles_; i0++) {
     fPt_.get()[i0] = 0;
     fEta_.get()[i0] = 0;
     fPhi_.get()[i0] = 0;
     fId_.get()[i0] = 0;
     fCharge_.get()[i0] = 0;
     fDZ_.get()[i0] = 0;
     fDX_.get()[i0] = 0;
     fDY_.get()[i0] = 0;
   }
   auto iParts = iJet.constituents();
   std::sort(iParts.begin(), iParts.end(), [](edm::Ptr<l1t::PFCandidate> i, edm::Ptr<l1t::PFCandidate> j) {
     return (i->pt() > j->pt());
   });
   float jetpt = useRawPt ? iJet.rawPt() : iJet.pt();
   for (unsigned int i0 = 0; i0 < iParts.size(); i0++) {
     if (i0 >= (unsigned int)fNParticles_)
       break;
     fPt_.get()[i0] = iParts[i0]->pt() / jetpt;
     fEta_.get()[i0] = iParts[i0]->eta() - iJet.eta();
     fPhi_.get()[i0] = deltaPhi(iParts[i0]->phi(), iJet.phi());
     fId_.get()[i0] = iParts[i0]->id();
     fCharge_.get()[i0] = iParts[i0]->charge();
     if (iParts[i0]->pfTrack().isNonnull()) {
       fDX_.get()[i0] = iParts[i0]->pfTrack()->vx();
       fDY_.get()[i0] = iParts[i0]->pfTrack()->vy();
       fDZ_.get()[i0] = iParts[i0]->pfTrack()->vz() - vz;
     }
   }
   setNNVectorVar();
   return EvaluateNNFixed();
 }

◆ EvaluateNN()

float JetId::EvaluateNN ( )

Definition at line 66 of file JetId.cc.

References fInput_, ALCARECOEcalPhiSym_cff::float, fOutput_, mps_fire::i, input, createfilelist::int, NNvectorVar_, PatBasicFWLiteJetAnalyzer_Selector_cfg::outputs, tensorflow::run(), and sessionRef_.

Referenced by compute().

                         {
   tensorflow::Tensor input(tensorflow::DT_FLOAT, {1, (unsigned int)NNvectorVar_.size(), 1});
   for (unsigned int i = 0; i < NNvectorVar_.size(); i++) {
     input.tensor<float, 3>()(0, i, 0) = float(NNvectorVar_[i]);
   }
   std::vector<tensorflow::Tensor> outputs;
   tensorflow::run(sessionRef_, {{fInput_, input}}, {fOutput_}, &outputs);
   return outputs[0].matrix<float>()(0, 0);
 }  //end EvaluateNN

◆ EvaluateNNFixed()

ap_fixed< 16, 6 > JetId::EvaluateNNFixed ( )

Definition at line 76 of file JetId.cc.

References mps_fire::i, modelRef_, and NNvectorVar_.

Referenced by computeFixed().

                                        {
   ap_fixed<16, 6> modelInput[140] = {};
   for (unsigned int i = 0; i < NNvectorVar_.size(); i++) {
     modelInput[i] = NNvectorVar_[i];
   }
   ap_fixed<16, 6> modelResult[1] = {-1};
 
   modelRef_->prepare_input(modelInput);
   modelRef_->predict();
   modelRef_->read_result(modelResult);
   ap_fixed<16, 6> modelResult_ = modelResult[0];
   return modelResult_;
 }  //end EvaluateNNFixed

◆ setNNVectorVar()

void JetId::setNNVectorVar ( )

Definition at line 43 of file JetId.cc.

References l1t::PFCandidate::ChargedHadron, l1t::PFCandidate::Electron, fCharge_, fDX_, fDY_, fDZ_, fEta_, fId_, fNParticles_, fPhi_, fPt_, Matriplex::hypot(), testProducerWithPsetDescEmpty_cfi::i1, l1t::PFCandidate::Muon, l1t::PFCandidate::NeutralHadron, NNvectorVar_, and l1t::PFCandidate::Photon.

Referenced by compute(), and computeFixed().

                            {
   NNvectorVar_.clear();
   for (int i0 = 0; i0 < fNParticles_; i0++) {
     if (fPt_.get()[i0] == 0) {
       for (int i1 = 0; i1 < 13; i1++)
         NNvectorVar_.push_back(0);
       continue;
     }
     NNvectorVar_.push_back(fId_.get()[i0] == l1t::PFCandidate::Electron && fCharge_.get()[i0] < 0);       // Electron
     NNvectorVar_.push_back(fId_.get()[i0] == l1t::PFCandidate::Electron && fCharge_.get()[i0] > 0);       // Positron
     NNvectorVar_.push_back(fId_.get()[i0] == l1t::PFCandidate::Muon && fCharge_.get()[i0] < 0);           // Muon
     NNvectorVar_.push_back(fId_.get()[i0] == l1t::PFCandidate::Muon && fCharge_.get()[i0] > 0);           // Anti-Muon
     NNvectorVar_.push_back(fId_.get()[i0] == l1t::PFCandidate::Photon);                                   // Photon
     NNvectorVar_.push_back(fId_.get()[i0] == l1t::PFCandidate::NeutralHadron);                            // Neutral Had
     NNvectorVar_.push_back(fId_.get()[i0] == l1t::PFCandidate::ChargedHadron && fCharge_.get()[i0] < 0);  // Pion
     NNvectorVar_.push_back(fId_.get()[i0] == l1t::PFCandidate::ChargedHadron && fCharge_.get()[i0] > 0);  // Anti-Pion
     NNvectorVar_.push_back(fDZ_.get()[i0]);                                                               //dZ
     NNvectorVar_.push_back(std::hypot(fDX_.get()[i0], fDY_.get()[i0]));                                   //d0
     NNvectorVar_.push_back(fPt_.get()[i0]);   //pT as a fraction of jet pT
     NNvectorVar_.push_back(fEta_.get()[i0]);  //dEta from jet axis
     NNvectorVar_.push_back(fPhi_.get()[i0]);  //dPhi from jet axis
   }
 }

Member Data Documentation

◆ fCharge_

unique_ptr<int[]> JetId::fCharge_

private

Definition at line 47 of file JetId.h.

Referenced by compute(), computeFixed(), JetId(), and setNNVectorVar().

◆ fDX_

unique_ptr<float[]> JetId::fDX_

private

Definition at line 49 of file JetId.h.

Referenced by compute(), computeFixed(), JetId(), and setNNVectorVar().

◆ fDY_

unique_ptr<float[]> JetId::fDY_

private

Definition at line 50 of file JetId.h.

Referenced by compute(), computeFixed(), JetId(), and setNNVectorVar().

◆ fDZ_

unique_ptr<float[]> JetId::fDZ_

private

Definition at line 48 of file JetId.h.

Referenced by compute(), computeFixed(), JetId(), and setNNVectorVar().

◆ fEta_

unique_ptr<float[]> JetId::fEta_

private

Definition at line 44 of file JetId.h.

Referenced by compute(), computeFixed(), JetId(), and setNNVectorVar().

◆ fId_

unique_ptr<int[]> JetId::fId_

private

Definition at line 46 of file JetId.h.

Referenced by compute(), computeFixed(), JetId(), and setNNVectorVar().

◆ fInput_

std::string JetId::fInput_

private

Definition at line 40 of file JetId.h.

Referenced by EvaluateNN(), and JetId().

◆ fNParticles_

int JetId::fNParticles_

private

Definition at line 42 of file JetId.h.

Referenced by compute(), computeFixed(), JetId(), and setNNVectorVar().

◆ fOutput_

std::string JetId::fOutput_

private

Definition at line 41 of file JetId.h.

Referenced by EvaluateNN(), and JetId().

◆ fPhi_

unique_ptr<float[]> JetId::fPhi_

private

Definition at line 45 of file JetId.h.

Referenced by compute(), computeFixed(), JetId(), and setNNVectorVar().

◆ fPt_

unique_ptr<float[]> JetId::fPt_

private

Definition at line 43 of file JetId.h.

Referenced by compute(), computeFixed(), JetId(), and setNNVectorVar().

◆ modelRef_

std::shared_ptr<hls4mlEmulator::Model> JetId::modelRef_

private

Definition at line 52 of file JetId.h.

Referenced by EvaluateNNFixed().

◆ NNvectorVar_

std::vector<float> JetId::NNvectorVar_

private

Definition at line 39 of file JetId.h.

Referenced by EvaluateNN(), EvaluateNNFixed(), JetId(), and setNNVectorVar().

◆ sessionRef_

tensorflow::Session* JetId::sessionRef_

private

Definition at line 51 of file JetId.h.

Referenced by EvaluateNN().

Public Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

◆ JetId() [1/2]

◆ JetId() [2/2]

◆ ~JetId()

Member Function Documentation

◆ compute()

◆ computeFixed()

◆ EvaluateNN()

◆ EvaluateNNFixed()

◆ setNNVectorVar()

Member Data Documentation

◆ fCharge_

◆ fDX_

◆ fDY_

◆ fDZ_

◆ fEta_

◆ fId_

◆ fInput_

◆ fNParticles_

◆ fOutput_

◆ fPhi_

◆ fPt_

◆ modelRef_

◆ NNvectorVar_

◆ sessionRef_