AsgRingerSelectorTool.h

Go to the documentation of this file.

// Dear emacs, this is -*- c++ -*-
/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
#ifndef RINGERSELECTORTOOLS_ASGRINGERSELECTORTOOL_H
#define RINGERSELECTORTOOLS_ASGRINGERSELECTORTOOL_H
 
// STL includes:
#include <string>
 
// Athena includes
#include "PATCore/AcceptData.h"
#include "PATCore/AcceptInfo.h"
#include "AsgTools/AsgTool.h"
 
// xAOD includes:
#include "RingerSelectorTools/IAsgRingerSelectorTool.h"
#include "RingerSelectorTools/tools/onnx/RingerSelector.h"
#include "AthOnnxInterfaces/IOnnxRuntimeSvc.h"
 
 
namespace Ringer {
  
  class AsgRingerSelectorTool : public asg::AsgTool,
                                  virtual public IAsgRingerSelectorTool
  {
 
  public:
 
    // Use ASG_TOOL macro for 2 interface base classes.
    ASG_TOOL_CLASS2(AsgRingerSelectorTool,
                    IAsgRingerSelectorTool,
                    IAsgSelectionTool)
 
 
    
    AsgRingerSelectorTool(const std::string& name);

    virtual ~AsgRingerSelectorTool();

    virtual StatusCode initialize() override;

    virtual StatusCode finalize() override;
 

    virtual const asg::AcceptInfo& getAcceptInfo() const override{ return m_accept; };
 

    virtual float predict(const xAOD::TrigRingerRings * ) const override;

    virtual float predict(const xAOD::TrigRingerRings *, const xAOD::TrigElectron *) const override;
 

    virtual asg::AcceptData  accept(const xAOD::TrigRingerRings *, float discr, float mu ) const override;
    
    
    virtual asg::AcceptData  accept(const xAOD::IParticle *) const override;
 
 
  private:
 
    asg::AcceptInfo m_accept;
    onnx::RingerSelector m_selector;
 
    Gaudi::Property<std::vector<std::string>> m_configFiles{this, "ConfigFiles", {}, "Ringer Run3 Calib Path"};  
    ServiceHandle< AthOnnx::IOnnxRuntimeSvc > m_onnxSvc{ this, "ONNXRuntimeSvc", "AthOnnx::OnnxRuntimeSvc", "Name of the service to use" };
    Gaudi::Property<bool> m_useTansigOutput{this, "UseTansigOutput", {}, "Use tansig output."};  
 
};
 
}
#endif