GraphData.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONINFERENCEINTERACES_GRAPHNODE_H
#define MUONINFERENCEINTERACES_GRAPHNODE_H
 
#include <onnxruntime_cxx_api.h>
#include <vector>
#include <memory>
 
namespace MuonML{
    class NodeFeatureList;
        
    struct InferenceGraph {
        std::vector<Ort::Value> dataTensor{};
        std::vector<const char*> nameTensor{};
     };
    
    struct GraphRawData {
        using FeatureVec_t = std::vector<float>;
        using NodeConnectVec_t = std::vector<int64_t>;
        using EdgeCounterVec_t = std::vector<int64_t>;
        FeatureVec_t featureLeaves{};
        EdgeCounterVec_t srcEdges{};
        EdgeCounterVec_t desEdges{};
        NodeConnectVec_t spacePointsInBucket{};
        EdgeCounterVec_t edgeIndexPacked{};
        const NodeFeatureList* previousList{};
        std::unique_ptr<InferenceGraph> graph{};
        
        std::vector<float>::iterator currLeave{featureLeaves.begin()};
        unsigned int nodeIndex{0};
    };
}
 
#endif