Model.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef RingerSelectorTools_tools_onnx_Model_h
#define RingerSelectorTools_tools_onnx_Model_h
 
#include "AsgMessaging/AsgMessaging.h"
#include <onnxruntime_cxx_api.h>
#include "AthOnnxInterfaces/IOnnxRuntimeSvc.h"
#include <assert.h>
#include <vector>
 
 
 
 
namespace Ringer{
 
  namespace onnx{
 
    class Model
    {
      public:

        Model( const std::string& modelPath, AthOnnx::IOnnxRuntimeSvc *svc,
               float etmin, float etmax, float etamin, float etamax,
               unsigned barcode);

        ~Model()=default;
 
        void compile();

        float etMin() const { return m_etmin; };

        float etMax() const { return m_etmax; };

        float etaMin() const { return m_etamin; };

        float etaMax() const { return m_etamax; };

        float predict ( std::vector< std::vector<float> > & ) const;
 
        unsigned barcode() const { return m_barcode; };
 
      private:
 
        // onnx session
        std::shared_ptr<Ort::Session> m_session;
        std::vector<std::vector<int64_t>>   m_input_node_dims;
        std::vector<int64_t>                m_output_node_dims;
        std::vector<const char*>            m_input_node_names;
        std::vector<const char*>            m_output_node_names;
 
        float m_etmin, m_etmax;
        float m_etamin, m_etamax;
        unsigned m_barcode;
 
 
    };
 
  } // namepace onnx
 
}// namespace ringer
 
#endif