HyPERTtbarDiLeptonParser.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef HYPERANALYSISALGORITHMS_HYPERTTBARDILEPTONPARSER_H
#define HYPERANALYSISALGORITHMS_HYPERTTBARDILEPTONPARSER_H
 
#include <string>
#include <vector>
 
#include "HyPERAnalysisAlgorithms/HyPERParser.h"
#include "HyPERAnalysisAlgorithms/HyPERUtils.h"
 
namespace EventReco {
class HyPERTtbarDiLeptonParser : public HyPERParser {
 public:
  HyPERTtbarDiLeptonParser() = default;
  ~HyPERTtbarDiLeptonParser() = default;
 
  void buildONNXInputs(const HyPERGraph& hyperGraph,
                       HyPERModel& hyperModel) override;
 
  void getONNXOutputs(HyPERModel& hyperModel) override;
 
  void reconstructOutputs(const HyPERGraph& hyperGraph) override;
 
  std::vector<std::string> getLabels() override { return m_reco_labels_out; }
  std::vector<float> getScores() override { return m_reco_scores_out; }
  std::vector<std::vector<int>> getIndices() override {
 
    return vector2DTypeConverter<int>(m_reco_indices_out);
  }
  std::vector<std::vector<int>> getIds() override {
    return vector2DTypeConverter<int>(m_reco_ids_out);
  }
 
  void clear() override;
  virtual void printOutputsForValidation() const override;
 
  float getClassificationScore() override { return m_classification_score; }
 
 public:
  // Outputs
  std::vector<int64_t> m_edge_out_shape{};
  std::vector<int64_t> m_hyperedge_out_shape{};
  std::vector<int64_t> m_classification_out_shape{
      1, 1};  // This is just a single number.
  std::vector<std::vector<float>> m_edge_out{};
  std::vector<std::vector<float>> m_hyperedge_out{};
  std::vector<std::vector<float>> m_classification_out{};
 
  std::vector<std::vector<int64_t>> m_reco_indices_out = {};
  std::vector<float> m_reco_scores_out = {};
  std::vector<std::string> m_reco_labels_out = {};
  std::vector<std::vector<int64_t>> m_reco_ids_out = {};
  float m_classification_score = -1.0f;
 
 private:
  // Input tensors for the ONNX model including shapes and names.
  std::vector<int64_t> m_l_edge_index_ = {{}, {}};
  std::vector<int64_t> m_l_edge_index_shape;
  const std::string m_l_edge_index_name = "edge_index";
  std::vector<float> m_l_edge_attr_s_ = {};
  std::vector<int64_t> m_l_edge_attr_s_shape;
  const std::string m_l_edge_attr_s_name = "edge_attr_s";
  std::vector<float> m_l_x_s_ = {};
  std::vector<int64_t> m_l_x_s_shape;
  const std::string m_l_x_s_name = "x_s";
  std::vector<int64_t> m_l_batch_ = {};
  std::vector<int64_t> m_l_batch_shape;
  const std::string m_l_batch_name = "batch";
  std::vector<int64_t> m_l_edge_index_h_ = {};
  std::vector<int64_t> m_l_edge_index_h_shape;
  const std::string m_l_edge_index_h_name = "edge_index_h";
  std::vector<int64_t> m_l_edge_index_h_batch_ = {};
  std::vector<int64_t> m_l_edge_index_h_batch_shape;
  const std::string m_l_edge_index_h_batch_name = "batch_hyperedge";
  std::vector<float> m_l_u_s_;
  std::vector<int64_t> m_l_u_s_shape;
  const std::string m_l_u_s_name = "u_s";
};
 
}  // namespace EventReco
 
#endif  // HYPERANALYSISALGORITHMS_HYPERTTBARDILEPTONPARSER_H