BJetId Class Reference

#include <BJetId.h>

Public Member Functions
	BJetId (const std::string &iInput, const std::string &iOutput, const BJetTFCache *cache, int iNParticles)
float	compute (const l1t::PFJet &iJet, float vz, bool useRawPt)
float	EvaluateNN ()
void	setNNVectorVar ()
	~BJetId ()

Private Attributes
unique_ptr< int[]>	fCharge_
unique_ptr< float[]>	fDX_
unique_ptr< float[]>	fDY_
unique_ptr< float[]>	fDZ_
unique_ptr< float[]>	fEta_
unique_ptr< float[]>	fId_
std::string	fInput_
int	fNParticles_
std::string	fOutput_
unique_ptr< float[]>	fPhi_
unique_ptr< float[]>	fPt_
std::vector< float >	NNvectorVar_
tensorflow::Session *	sessionRef_

Detailed Description

Definition at line 18 of file BJetId.h.

Constructor & Destructor Documentation

BJetId()

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

Definition at line 5 of file BJetId.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<float[]>(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;
}

~BJetId()

BJetId::~BJetId

(

)

Definition at line 22 of file BJetId.cc.

{}

Member Function Documentation

compute()

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

Definition at line 56 of file BJetId.cc.

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

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

EvaluateNN()

float BJetId::EvaluateNN

(

)

Definition at line 46 of file BJetId.cc.

References fInput_, nano_mu_digi_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

setNNVectorVar()

void BJetId::setNNVectorVar

(

)

Definition at line 23 of file BJetId.cc.

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

Referenced by compute().

                            {
  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[]> BJetId::fCharge_

private

Definition at line 36 of file BJetId.h.

Referenced by BJetId(), compute(), and setNNVectorVar().

fDX_

unique_ptr<float[]> BJetId::fDX_

private

Definition at line 38 of file BJetId.h.

Referenced by BJetId(), compute(), and setNNVectorVar().

fDY_

unique_ptr<float[]> BJetId::fDY_

private

Definition at line 39 of file BJetId.h.

Referenced by BJetId(), compute(), and setNNVectorVar().

fDZ_

unique_ptr<float[]> BJetId::fDZ_

private

Definition at line 37 of file BJetId.h.

Referenced by BJetId(), compute(), and setNNVectorVar().

fEta_

unique_ptr<float[]> BJetId::fEta_

private

Definition at line 33 of file BJetId.h.

Referenced by BJetId(), compute(), and setNNVectorVar().

fId_

unique_ptr<float[]> BJetId::fId_

private

Definition at line 35 of file BJetId.h.

Referenced by BJetId(), compute(), and setNNVectorVar().

fInput_

std::string BJetId::fInput_

private

Definition at line 29 of file BJetId.h.

Referenced by BJetId(), and EvaluateNN().

fNParticles_

int BJetId::fNParticles_

private

Definition at line 31 of file BJetId.h.

Referenced by BJetId(), compute(), and setNNVectorVar().

fOutput_

std::string BJetId::fOutput_

private

Definition at line 30 of file BJetId.h.

Referenced by BJetId(), and EvaluateNN().

fPhi_

unique_ptr<float[]> BJetId::fPhi_

private

Definition at line 34 of file BJetId.h.

Referenced by BJetId(), compute(), and setNNVectorVar().

fPt_

unique_ptr<float[]> BJetId::fPt_

private

Definition at line 32 of file BJetId.h.

Referenced by BJetId(), compute(), and setNNVectorVar().

NNvectorVar_

std::vector<float> BJetId::NNvectorVar_

private

Definition at line 28 of file BJetId.h.

Referenced by BJetId(), EvaluateNN(), and setNNVectorVar().

sessionRef_

tensorflow::Session* BJetId::sessionRef_

private

Definition at line 40 of file BJetId.h.

Referenced by EvaluateNN().