RingerSelector.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
#ifndef RingerSelectorTools_tools_onnx_RingerSelector_h
#define RingerSelectorTools_tools_onnx_RingerSelector_h
 
 
 
#include "GaudiKernel/StatusCode.h"
#include "GaudiKernel/ServiceHandle.h"
#include "AsgMessaging/AsgMessaging.h"
#include "RingerSelectorTools/tools/onnx/Model.h"
#include "RingerSelectorTools/tools/onnx/Threshold.h"
#include "AthOnnxInterfaces/IOnnxRuntimeSvc.h"
#include <onnxruntime_cxx_api.h>
#include "xAODTrigRinger/TrigRingerRings.h"
#include "xAODTrigCalo/TrigEMCluster.h"
#include "xAODTrigEgamma/TrigElectron.h"
#include "GaudiKernel/SystemOfUnits.h"
#include <memory>
#include "TEnv.h"
 
namespace Ringer {
 
  namespace onnx{
 
    class RingerSelector : public asg::AsgMessaging
 
    {
      public:
 
        RingerSelector( const std::string& name );
 
        ~RingerSelector()=default;
 
 
        StatusCode read_from( const std::string&, AthOnnx::IOnnxRuntimeSvc *);
 
 
        std::vector<std::vector<float>>  prepare_inputs( unsigned barcode, const xAOD::TrigRingerRings *, const xAOD::TrigElectron *) const;
 
 
        float predict(const xAOD::TrigRingerRings *, const xAOD::TrigElectron * ) const;
 
 
        bool accept(const xAOD::TrigRingerRings *, float discr, float mu ) const;
 
 
      private:
 
        std::vector< Ringer::onnx::Model > m_models;
        std::vector< Ringer::onnx::Threshold > m_thresholds;
 
        template <typename T> bool strtof(const std::string& input, T& f);
        template <typename T>  std::vector<T> GetValues (const std::string& input,  TEnv& env);
        std::vector<std::string> GetPaths(const std::string& input, TEnv& env);
        std::string GetBasePath(const std::string &path) const;
 
    };
 
  } // namespace onnx
} //namespace Ringer
 
#endif