IJSSMLTool.h

Go to the documentation of this file.

// Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
 
#ifndef IJSSMLTool_H
#define IJSSMLTool_H
 
#include "AsgTools/IAsgTool.h"
 
// System include(s).
#include <memory>
#include <string>
#include <iostream> 
#include <fstream>
#include <arpa/inet.h>
#include <numeric>
 
// root
#include "TH2.h"
 
// xAOD
#include "xAODJet/JetContainer.h"
#include "xAODPFlow/TrackCaloClusterContainer.h"
 
namespace AthONNX {
 
  class IJSSMLTool : virtual public asg::IAsgTool {
    ASG_TOOL_INTERFACE(IJSSMLTool)
 
      public:

    virtual StatusCode initialize() = 0;
    
    virtual double retrieveConstituentsScore(std::vector<TH2D> Images) const = 0;
    virtual double retrieveConstituentsScore(std::vector<std::vector<float>> constituents) const = 0;
    virtual double retrieveConstituentsScore(std::vector<std::vector<float>> constituents, std::vector<std::vector<std::vector<float>>> interactions) const = 0;
    virtual double retrieveConstituentsScore(std::vector<std::vector<float>> constituents, std::vector<std::vector<std::vector<float>>> interactions, std::vector<std::vector<float>> mask) const = 0;
    virtual double retrieveHighLevelScore(std::map<std::string, double> JSSVars) const = 0;
 
    virtual std::vector<float> retrieveConstituentsScoreMultiClass(const std::vector<std::vector<float>>& constituents, const std::vector<std::vector<std::vector<float>>>& interactions, const std::vector<std::vector<float>>& mask) const = 0;
 
    virtual StatusCode SetScaler(const std::map<std::string, std::vector<double>> & scaler) = 0;
 
  };
 
} // end name space AthONNX
 
#endif