TrigClusterNNProducer.h

Go to the documentation of this file.

 
#ifndef TRIGGER_TRIGCLUSTERNNPRODUCER_H
#define TRIGGER_TRIGCLUSTERNNPRODUCER_H
 
#include <cstddef>
#include <memory>
#include <string>
#include <vector>
 
// LDMX
#include "Framework/Configure/Parameters.h"
#include "Framework/EventProcessor.h"
#include "Tools/ONNXRuntime.h"
#include "Trigger/Event/TrigCaloCluster.h"
#include "Trigger/Event/TrigEnergySum.h"
 
namespace trigger {
 
class TrigClusterNNProducer : public framework::Producer {
 public:
  TrigClusterNNProducer(const std::string& name, framework::Process& process);
  virtual ~TrigClusterNNProducer() = default;
  void configure(framework::config::Parameters& ps) override;
  void produce(framework::Event& event) override;
  void onProcessEnd() override;
 
  static void fillInputs(const TrigCaloClusterCollection& clusters,
                         const TrigEnergySumCollection& ecal_sums,
                         const TrigEnergySumCollection& hcal_sums,
                         std::vector<float>& inputs);
 
  static constexpr std::size_t MAX_CLUSTERS = 3;
  static constexpr std::size_t N_FEATURES = 9;
  static constexpr std::size_t N_CLASSES = 9;
 
  static constexpr std::size_t FEAT_X = 0;
  static constexpr std::size_t FEAT_Y = 1;
  static constexpr std::size_t FEAT_Z = 2;
  static constexpr std::size_t FEAT_E = 3;
  static constexpr std::size_t FEAT_DEPTH = 4;
  static constexpr std::size_t FEAT_N_TP = 5;
  static constexpr std::size_t FEAT_N_CLUSTERS = 6;
  static constexpr std::size_t FEAT_ECAL_SUM_E = 7;
  static constexpr std::size_t FEAT_HCAL_SUM_ADC = 8;
  static_assert(FEAT_HCAL_SUM_ADC + 1 == N_FEATURES);
 
  static constexpr std::size_t BKG_CLASS = 0;
 
 private:
  std::string cluster_coll_name_;
  std::string ecal_sum_coll_name_;
  std::string hcal_sum_coll_name_;
  std::string input_pass_;
  std::string score_coll_name_;
 
  ldmx::ort::FloatArrays data_;
  std::unique_ptr<ldmx::ort::ONNXRuntime> rt_;
 
  int nevents_{0};
  double processing_time_{0.};
  double inference_time_{0.};
};
 
}  // namespace trigger
 
#endif  // TRIGGER_TRIGCLUSTERNNPRODUCER_H