HyPERParser.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef HYPERANALYSISALGORITHMS_HYPERPARSER_H
#define HYPERANALYSISALGORITHMS_HYPERPARSER_H
 
#include <memory>
 
#include "HyPERAnalysisAlgorithms/HyPERGraph.h"
#include "HyPERAnalysisAlgorithms/HyPERModel.h"
 
namespace EventReco {
class HyPERParser {
 public:
  HyPERParser() = default;
  virtual ~HyPERParser() = default;

  virtual void buildONNXInputs(const HyPERGraph& hyperGraph,
                               HyPERModel& hyperModel) = 0;

  virtual void getONNXOutputs(HyPERModel& hyperModel) = 0;

  virtual void reconstructOutputs(const HyPERGraph& hyperGraph) = 0;

  virtual std::vector<std::string> getLabels() = 0;
  virtual std::vector<float> getScores() = 0;
  virtual std::vector<std::vector<int>> getIndices() = 0;
  virtual std::vector<std::vector<int>> getIds() = 0;
  virtual float getClassificationScore() = 0;

  virtual void clear() = 0;
  virtual void printOutputsForValidation() const = 0;
};
}  // namespace EventReco
 
#endif  // HYPERANALYSISALGORITHMS_HYPERPARSER_H