Add CMSSW part of part 3

part3/cmssw/src/L1Trigger/L1TMLDemo/plugins/BuildFile.xml

+<use name="FWCore/Framework"/>
+<use name="FWCore/ParameterSet"/>
+<use name="FWCore/PluginManager"/>
+<use name="DataFormats/L1Trigger"/>
+<use name="DataFormats/NanoAOD"/>
+<use name="hls"/>
+<use name="hls4mlEmulatorExtras"/>
+<flags EDM_PLUGIN="1"/>
\ No newline at end of file

part3/cmssw/src/L1Trigger/L1TMLDemo/plugins/l1tDemoMLAnalyzer.cc

+// FWCore includes
+#include "FWCore/Framework/interface/global/EDProducer.h"
+#include "FWCore/Framework/interface/Event.h"
+#include "FWCore/Framework/interface/MakerMacros.h"
+#include "FWCore/Framework/interface/ESHandle.h"
+#include "FWCore/Framework/interface/EventSetup.h"
+#include "FWCore/ParameterSet/interface/ParameterSet.h"
+
+// File writing includes
+#include "DataFormats/NanoAOD/interface/FlatTable.h"
+//#include "TTree.h"
+//#include "FWCore/ServiceRegistry/interface/Service.h"
+//#include "CommonTools/UtilAlgos/interface/TFileService.h"
+
+// L1T includes
+#include "DataFormats/L1Trigger/interface/Muon.h"
+#include "DataFormats/L1Trigger/interface/EGamma.h"
+#include "DataFormats/L1Trigger/interface/Tau.h"
+#include "DataFormats/L1Trigger/interface/Jet.h"
+#include "DataFormats/L1Trigger/interface/EtSum.h"
+
+// hls & hls4ml includes
+#include "ap_fixed.h"
+#include "hls4ml/emulator.h"
+
+#include <iostream>
+
+class L1TMLDemoProducer : public edm::global::EDProducer<> {
+public:
+  explicit L1TMLDemoProducer(const edm::ParameterSet& cfg);
+  ~L1TMLDemoProducer();
+
+private:
+  virtual void beginJob() override;
+  virtual void produce(edm::StreamID id, edm::Event& iEvent, const edm::EventSetup& iSetup) const override;
+  virtual void endJob() override;
+
+  edm::EDGetToken muToken;
+  edm::EDGetToken egToken;
+  edm::EDGetToken tauToken;
+  edm::EDGetToken jetToken;
+  edm::EDGetToken sumToken;
+
+  unsigned nMu;
+  unsigned nEG;
+  unsigned nTau;
+  unsigned nJet;
+  unsigned nNNIn;
+
+  // hls4ml emulator model path
+  std::string model_so_path;
+
+};
+
+L1TMLDemoProducer::L1TMLDemoProducer(const edm::ParameterSet& cfg){
+  // consume
+  muToken = consumes<l1t::MuonBxCollection>(cfg.getParameter<edm::InputTag>("muToken"));
+  egToken = consumes<l1t::EGammaBxCollection>(cfg.getParameter<edm::InputTag>("egToken"));
+  tauToken = consumes<l1t::TauBxCollection>(cfg.getParameter<edm::InputTag>("tauToken"));
+  jetToken = consumes<l1t::JetBxCollection>(cfg.getParameter<edm::InputTag>("jetToken"));
+  sumToken = consumes<l1t::EtSumBxCollection>(cfg.getParameter<edm::InputTag>("etSumToken"));
+  nMu = cfg.getParameter<unsigned>("nMu");
+  nEG = cfg.getParameter<unsigned>("nEg");
+  nTau = cfg.getParameter<unsigned>("nTau");
+  nJet = cfg.getParameter<unsigned>("nJet");
+  // total number of inputs to NN
+  nNNIn = 3 * (1 + nMu + nEG + nTau + nJet);
+
+  // store the path to the .so file
+  model_so_path = cfg.getParameter<std::string>("model_so_path");
+
+  // produce
+  produces<nanoaod::FlatTable>("L1TMLDemo");
+
+}
+
+L1TMLDemoProducer::~L1TMLDemoProducer(){
+}
+
+void L1TMLDemoProducer::produce(edm::StreamID id, edm::Event& iEvent, const edm::EventSetup& iSetup) const {
+  using namespace edm;
+  // get input collections
+  // BXVector: first index is BX, second index is object
+  edm::Handle<BXVector<l1t::Muon>> muons;
+  edm::Handle<BXVector<l1t::EGamma>> egammas;
+  edm::Handle<BXVector<l1t::Tau>> taus;
+  edm::Handle<BXVector<l1t::Jet>> jets;
+  edm::Handle<BXVector<l1t::EtSum>> sums;
+  iEvent.getByToken(muToken, muons);
+  iEvent.getByToken(egToken, egammas);
+  iEvent.getByToken(tauToken, taus);
+  iEvent.getByToken(jetToken, jets);
+  iEvent.getByToken(sumToken, sums);
+
+  // The unscaled inputs are hwInts apart from ET that is in GeV with 0.5 GeV LSB
+  // ap_fixed<14,13> is wide enough for all the ET, pT, eta, phi
+  ap_fixed<14,13>* X_unscaled = new ap_fixed<14,13>[nNNIn];
+
+  // fill the inputs
+  unsigned ix = 0;
+  // sums first, just find the MET
+  for(unsigned i = 0; i < sums->size(0); i++){
+    if(sums->at(0, i).getType() == l1t::EtSum::EtSumType::kMissingEt){
+      X_unscaled[ix++] = (float)(sums->at(0,i).hwPt())/2;
+      X_unscaled[ix++] = 0; // for eta
+      X_unscaled[ix++] = sums->at(0,i).hwPhi();
+    }
+  }
+  // muons next
+  ix=3 * ( 1 );
+  for(unsigned i = 0; i < std::min(nMu, muons->size(0)); i++){
+    X_unscaled[ix++] = (float)(muons->at(0, i).hwPt())/2;
+    X_unscaled[ix++] = muons->at(0, i).hwEta();
+    X_unscaled[ix++] = muons->at(0, i).hwPhi();
+  }
+  // egammas next
+  ix = 3 * ( 1 + nMu );
+  for(unsigned i = 0; i < std::min(nEG, egammas->size(0)); i++){
+    X_unscaled[ix++] = (float)(egammas->at(0, i).hwPt())/2;
+    X_unscaled[ix++] = egammas->at(0, i).hwEta();
+    X_unscaled[ix++] = egammas->at(0, i).hwPhi();
+  }
+  // taus next
+  ix = 3 * ( 1 + nMu + nEG );
+  for(unsigned i = 0; i < std::min(nTau, taus->size(0)); i++){
+    X_unscaled[ix++] = (float)(taus->at(0, i).hwPt())/2;
+    X_unscaled[ix++] = taus->at(0, i).hwEta();
+    X_unscaled[ix++] = taus->at(0, i).hwPhi();
+  }
+  // jets last
+  ix = 3 * ( 1 + nMu + nEG + nTau );
+  for(unsigned i = 0; i < std::min(nEG, jets->size(0)); i++){
+    X_unscaled[ix++] = (float)(jets->at(0, i).hwPt())/2;
+    X_unscaled[ix++] = jets->at(0, i).hwEta();
+    X_unscaled[ix++] = jets->at(0, i).hwPhi();
+  }
+
+
+  ap_fixed<16,7,AP_RND,AP_SAT>* X_scaled = new ap_fixed<16,7,AP_RND,AP_SAT>[nNNIn];
+  // scale the inputs
+  for(unsigned i = 0; i < nNNIn; i++){
+    X_scaled[i] = X_unscaled[i]; // placeholder
+  }
+
+  // load the NN emulator object
+  hls4mlEmulator::ModelLoader loader(model_so_path);
+  std::shared_ptr<hls4mlEmulator::Model> model = loader.load_model();
+
+  ap_fixed<13,2,AP_RND,AP_SAT> y; // output object
+
+  // run the actual inference
+  model->prepare_input(X_scaled);
+  model->predict();
+  model->read_result(&y);
+
+  // write the result to the output
+  // note cast from the ap_fixed emulated type to float for convenience
+  auto out = std::make_unique<nanoaod::FlatTable>(1, "L1TMLDemo", false);
+  std::vector<float> y_vec;
+  y_vec.push_back(y);
+  out->addColumn<float>("y", y_vec, "model prediction");
+  iEvent.put(std::move(out), "L1TMLDemo");
+
+}
+
+void L1TMLDemoProducer::beginJob(){
+}
+
+void L1TMLDemoProducer::endJob(){
+}
+
+DEFINE_FWK_MODULE(L1TMLDemoProducer);

part3/cmssw/src/L1Trigger/L1TMLDemo/test/demoL1TMLNtuple.py

+# import of standard configurations
+import FWCore.ParameterSet.Config as cms
+
+process = cms.Process("l1tMLDemo")
+
+process.load('Configuration.StandardSequences.Services_cff')
+process.load('FWCore.MessageService.MessageLogger_cfi')
+process.load('Configuration.StandardSequences.GeometryRecoDB_cff')
+process.load('Configuration.Geometry.GeometryDB_cff')
+process.load('Configuration.StandardSequences.MagneticField_38T_cff')
+process.load('Configuration.StandardSequences.SimL1Emulator_cff')
+process.load('Configuration.StandardSequences.EndOfProcess_cff')
+process.load('Configuration.StandardSequences.FrontierConditions_GlobalTag_cff')
+
+process.maxEvents = cms.untracked.PSet(
+    input = cms.untracked.int32(-1)
+)
+
+process.source = cms.Source (
+    "PoolSource",
+    fileNames = cms.untracked.vstring('/store/relval/CMSSW_13_3_0_pre3/RelValMinBias_14TeV/GEN-SIM-DIGI-RAW/132X_mcRun3_2023_realistic_v4-v1/2580000/0911bb55-82fb-4a51-bb8f-be79f61b020d.root'),
+)
+
+from Configuration.AlCa.GlobalTag import GlobalTag
+process.GlobalTag = GlobalTag(process.GlobalTag, 'auto:startup', '')
+
+process.l1tDemoMLProducer = cms.EDProducer('L1TMLDemoProducer',
+    muToken    = cms.InputTag("simGmtStage2Digis"),
+    egToken    = cms.InputTag("simCaloStage2Digis"),
+    tauToken   = cms.InputTag("simCaloStage2Digis"),
+    jetToken   = cms.InputTag("simCaloStage2Digis"),
+    etSumToken = cms.InputTag("simCaloStage2Digis"),
+    nMu = cms.uint32(2),
+    nEg = cms.uint32(2),
+    nTau = cms.uint32(0),
+    nJet = cms.uint32(4),
+    model_so_path = cms.string("../data/L1TMLDemo_v1")
+)
+
+process.path = cms.Path(
+    process.l1tDemoMLProducer
+)
+
+process.outnano = cms.OutputModule("NanoAODOutputModule",
+    fileName = cms.untracked.string("L1TMLDemo_NanoAOD.root"),
+    outputCommands = cms.untracked.vstring("drop *", "keep nanoaodFlatTable_*_*_*"),
+    compressionLevel = cms.untracked.int32(4),
+    compressionAlgorithm = cms.untracked.string("ZLIB"),
+)
+process.end = cms.EndPath(process.outnano)
\ No newline at end of file