PNETLeptonProducer< LeptonType > Class Template Reference

Inheritance diagram for PNETLeptonProducer< LeptonType >:

Public Member Functions
	PNETLeptonProducer (const edm::ParameterSet &, const cms::Ort::ONNXRuntime *)
	~PNETLeptonProducer () override
Public Member Functions inherited from edm::stream::EDProducer< edm::GlobalCache< cms::Ort::ONNXRuntime > >
	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 ONNXRuntime *)
static std::unique_ptr< ONNXRuntime >	initializeGlobalCache (const edm::ParameterSet &)

Private Types
using	LeptonTagInfoCollection = DeepBoostedJetFeaturesCollection

Private Member Functions
void	make_inputs (const DeepBoostedJetFeatures &)
void	produce (edm::Event &, const edm::EventSetup &) override

Private Attributes
cms::Ort::FloatArrays	data_
bool	debug_ = false
std::vector< std::string >	flav_names_
std::vector< std::string >	input_names_
std::vector< std::vector< int64_t > >	input_shapes_
std::vector< unsigned >	input_sizes_
edm::EDGetTokenT< edm::View< LeptonType > >	leps_
std::unordered_map< std::string, btagbtvdeep::PreprocessParams >	prep_info_map_
edm::EDGetTokenT< LeptonTagInfoCollection >	src_

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

template<typename LeptonType>
class PNETLeptonProducer< LeptonType >

Definition at line 31 of file PNETLeptonProducer.cc.

Member Typedef Documentation

LeptonTagInfoCollection

template<typename LeptonType >

using PNETLeptonProducer< LeptonType >::LeptonTagInfoCollection = DeepBoostedJetFeaturesCollection

private

Definition at line 42 of file PNETLeptonProducer.cc.

Constructor & Destructor Documentation

PNETLeptonProducer()

template<typename LeptonType >

PNETLeptonProducer< LeptonType >::PNETLeptonProducer ( const edm::ParameterSet &  iConfig, const cms::Ort::ONNXRuntime *  cache  )

explicit

Definition at line 61 of file PNETLeptonProducer.cc.

References gather_cfg::cout, PNETLeptonProducer< LeptonType >::data_, PNETLeptonProducer< LeptonType >::debug_, PNETLeptonProducer< LeptonType >::flav_names_, mps_fire::i, PNETLeptonProducer< LeptonType >::input_names_, PNETLeptonProducer< LeptonType >::input_shapes_, PNETLeptonProducer< LeptonType >::input_sizes_, btagbtvdeep::ParticleNetConstructor(), PNETLeptonProducer< LeptonType >::prep_info_map_, and x.

    : src_(consumes<LeptonTagInfoCollection>(iConfig.getParameter<edm::InputTag>("src"))),
      leps_(consumes<edm::View<LeptonType>>(iConfig.getParameter<edm::InputTag>("srcLeps"))),
      flav_names_(iConfig.getParameter<std::vector<std::string>>("flav_names")),
      debug_(iConfig.getUntrackedParameter<bool>("debugMode", false)) {
  ParticleNetConstructor(iConfig, true, input_names_, prep_info_map_, input_shapes_, input_sizes_, &data_);

  if (debug_) {
    for (unsigned i = 0; i < input_names_.size(); ++i) {
      const auto &group_name = input_names_.at(i);
      std::cout << group_name << std::endl;
      if (!input_shapes_.empty()) {
        std::cout << group_name << "\nshapes: ";
        for (const auto &x : input_shapes_.at(i)) {
          std::cout << x << ", ";
        }
      }
      std::cout << "\nvariables: ";
      for (const auto &x : prep_info_map_.at(group_name).var_names) {
        std::cout << x << ", ";
      }
      std::cout << "\n";
    }
    std::cout << "\n";
  }

  for (const auto &flav_name : flav_names_) {
    produces<edm::ValueMap<float>>(flav_name);
  }
}

~PNETLeptonProducer()

template<typename LeptonType >

PNETLeptonProducer< LeptonType >::~PNETLeptonProducer ( )

inlineoverride

Definition at line 34 of file PNETLeptonProducer.cc.

{}

Member Function Documentation

fillDescriptions()

template<typename LeptonType >

void PNETLeptonProducer< LeptonType >::fillDescriptions ( edm::ConfigurationDescriptions &  descriptions )

static

Definition at line 137 of file PNETLeptonProducer.cc.

References edm::ConfigurationDescriptions::add(), submitPVResolutionJobs::desc, timingPdfMaker::modname, and AlCaHLTBitMon_QueryRunRegistry::string.

                                                                                                {
  edm::ParameterSetDescription desc;
  desc.add<edm::InputTag>("src")->setComment("input variables");
  desc.add<edm::InputTag>("srcLeps")->setComment("input physics object collection. src and srcLeps must be in synch");
  desc.add<std::vector<std::string>>("flav_names")->setComment("Names of the oputput classes");
  desc.add<std::string>("preprocess_json", "PhysicsTools/NanoAOD/data/PNetMuonId/preprocess.json");
  desc.add<edm::FileInPath>("model_path", edm::FileInPath("PhysicsTools/NanoAOD/data/PNetMuonId/model.onnx"));
  desc.addOptionalUntracked<bool>("debugMode", false);

  std::string modname;
  if (typeid(LeptonType) == typeid(pat::Muon))
    modname += "muon";
  else if (typeid(LeptonType) == typeid(pat::Electron))
    modname += "electron";
  modname += "PNetTags";
  descriptions.add(modname, desc);
}

globalEndJob()

template<typename LeptonType >

void PNETLeptonProducer< LeptonType >::globalEndJob ( const ONNXRuntime *  cache )

static

Definition at line 134 of file PNETLeptonProducer.cc.

{}

initializeGlobalCache()

template<typename LeptonType >

std::unique_ptr< cms::Ort::ONNXRuntime > PNETLeptonProducer< LeptonType >::initializeGlobalCache ( const edm::ParameterSet &  cfg )

static

Definition at line 128 of file PNETLeptonProducer.cc.

References looper::cfg, and contentValuesFiles::fullPath.

                                {
  return std::make_unique<cms::Ort::ONNXRuntime>(cfg.getParameter<edm::FileInPath>("model_path").fullPath());
}

make_inputs()

template<typename LeptonType >

void PNETLeptonProducer< LeptonType >::make_inputs ( const DeepBoostedJetFeatures &  taginfo )

private

Definition at line 156 of file PNETLeptonProducer.cc.

References btagbtvdeep::center_norm_pad(), EcalPhiSymFlatTableProducers_cfi::fill, btagbtvdeep::DeepBoostedJetFeatures::get(), mps_fire::i, and info().

                                                                                      {
  for (unsigned igroup = 0; igroup < input_names_.size(); ++igroup) {
    const auto &group_name = input_names_[igroup];
    const auto &prep_params = prep_info_map_.at(group_name);
    auto &group_values = data_[igroup];
    group_values.resize(input_sizes_[igroup]);

    // first reset group_values to 0
    std::fill(group_values.begin(), group_values.end(), 0);
    unsigned curr_pos = 0;

    // transform/pad
    for (unsigned i = 0; i < prep_params.var_names.size(); ++i) {
      const auto &varname = prep_params.var_names[i];
      const auto &raw_value = taginfo.get(varname);
      const auto &info = prep_params.info(varname);
      int insize = btagbtvdeep::center_norm_pad(raw_value,
                                                info.center,
                                                info.norm_factor,
                                                prep_params.min_length,
                                                prep_params.max_length,
                                                group_values,
                                                curr_pos,
                                                info.pad,
                                                info.replace_inf_value,
                                                info.lower_bound,
                                                info.upper_bound);
      curr_pos += insize;
      if (i == 0 && (!input_shapes_.empty())) {
        input_shapes_[igroup][2] = insize;
      }
    }
    group_values.resize(curr_pos);
  }
}

produce()

template<typename LeptonType >

void PNETLeptonProducer< LeptonType >::produce ( edm::Event &  iEvent, const edm::EventSetup &  iSetup  )

overrideprivate

Definition at line 93 of file PNETLeptonProducer.cc.

References cms::cuda::assert(), SplitLinear::begin, mps_fire::end, trigObjTnPSource_cfi::filler, iEvent, TtSemiLepHitFitProducer_Electrons_cfi::leps, eostools::move(), PatBasicFWLiteJetAnalyzer_Selector_cfg::outputs, and TrackRefitter_38T_cff::src.

                                                                                          {
  edm::Handle<LeptonTagInfoCollection> src;
  iEvent.getByToken(src_, src);

  edm::Handle<edm::View<LeptonType>> leps;
  iEvent.getByToken(leps_, leps);

  std::vector<std::vector<float>> mvaScores(flav_names_.size(), std::vector<float>(leps->size(), -1));

  // tagInfo src could be empty if the event has no PV
  if (!src->empty()) {
    assert(src->size() == leps->size());
    for (size_t ilep = 0; ilep < src->size(); ilep++) {
      const auto &taginfo = (*src)[ilep];
      make_inputs(taginfo);
      auto outputs = globalCache()->run(input_names_, data_, input_shapes_)[0];
      // std::cout<<"outputs.size(): "<<outputs.size()<<std::endl;
      // std::cout<<"flav_names_.size(): "<<flav_names_.size()<<std::endl;
      assert(outputs.size() == flav_names_.size());
      for (unsigned int iflav = 0; iflav < flav_names_.size(); ++iflav) {
        mvaScores[iflav][ilep] = outputs.at(iflav);
      }
    }
  }

  for (unsigned int iflav = 0; iflav < flav_names_.size(); ++iflav) {
    std::unique_ptr<edm::ValueMap<float>> pnScore(new edm::ValueMap<float>());
    edm::ValueMap<float>::Filler filler(*pnScore);
    filler.insert(leps, mvaScores[iflav].begin(), mvaScores[iflav].end());
    filler.fill();
    iEvent.put(std::move(pnScore), flav_names_[iflav]);
  }
}

Member Data Documentation

data_

template<typename LeptonType >

cms::Ort::FloatArrays PNETLeptonProducer< LeptonType >::data_

private

Definition at line 56 of file PNETLeptonProducer.cc.

Referenced by PNETLeptonProducer< LeptonType >::PNETLeptonProducer().

debug_

template<typename LeptonType >

bool PNETLeptonProducer< LeptonType >::debug_ = false

private

Definition at line 57 of file PNETLeptonProducer.cc.

Referenced by PNETLeptonProducer< LeptonType >::PNETLeptonProducer().

flav_names_

template<typename LeptonType >

std::vector<std::string> PNETLeptonProducer< LeptonType >::flav_names_

private

Definition at line 49 of file PNETLeptonProducer.cc.

Referenced by PNETLeptonProducer< LeptonType >::PNETLeptonProducer().

input_names_

template<typename LeptonType >

std::vector<std::string> PNETLeptonProducer< LeptonType >::input_names_

private

Definition at line 50 of file PNETLeptonProducer.cc.

Referenced by PNETLeptonProducer< LeptonType >::PNETLeptonProducer().

input_shapes_

template<typename LeptonType >

std::vector<std::vector<int64_t> > PNETLeptonProducer< LeptonType >::input_shapes_

private

Definition at line 51 of file PNETLeptonProducer.cc.

Referenced by PNETLeptonProducer< LeptonType >::PNETLeptonProducer().

input_sizes_

template<typename LeptonType >

std::vector<unsigned> PNETLeptonProducer< LeptonType >::input_sizes_

private

Definition at line 52 of file PNETLeptonProducer.cc.

Referenced by PNETLeptonProducer< LeptonType >::PNETLeptonProducer().

leps_

template<typename LeptonType >

edm::EDGetTokenT<edm::View<LeptonType> > PNETLeptonProducer< LeptonType >::leps_

private

Definition at line 48 of file PNETLeptonProducer.cc.

prep_info_map_

template<typename LeptonType >

std::unordered_map<std::string, btagbtvdeep::PreprocessParams> PNETLeptonProducer< LeptonType >::prep_info_map_

private

Definition at line 54 of file PNETLeptonProducer.cc.

Referenced by PNETLeptonProducer< LeptonType >::PNETLeptonProducer().

src_

template<typename LeptonType >

edm::EDGetTokenT<LeptonTagInfoCollection> PNETLeptonProducer< LeptonType >::src_

private

Definition at line 47 of file PNETLeptonProducer.cc.