IntrospectionGraph.hpp

Go to the documentation of this file.

#pragma once

 
#include <icarus/io/data/DataDictionary.hpp>
#include <nlohmann/json.hpp>
#include <string>
#include <vector>
#include <yaml-cpp/yaml.h>
 
namespace icarus {

enum class EdgeKind {
    Route,  
    Resolve 
};

struct IntrospectionEdge {
    std::string source; 
    std::string target; 
    EdgeKind kind;      
};

struct IntrospectionGraph {
    DataDictionary dictionary;

    std::vector<IntrospectionEdge> edges;

    [[nodiscard]] nlohmann::json ToJSON() const {
        nlohmann::json j = dictionary.ToJSONValue();
        j["summary"]["total_edges"] = edges.size();
 
        // Edges (NEW — additive)
        j["edges"] = nlohmann::json::array();
        for (const auto &edge : edges) {
            nlohmann::json jedge;
            jedge["source"] = edge.source;
            jedge["target"] = edge.target;
            jedge["kind"] = (edge.kind == EdgeKind::Route) ? "route" : "resolve";
            j["edges"].push_back(jedge);
        }
 
        return j;
    }

    void ToJSONFile(const std::string &path) const {
        detail::WriteToFile(path, [&](std::ofstream &file) { file << ToJSON().dump(2); });
    }

    void ToYAMLFile(const std::string &path) const {
        YAML::Emitter out;
        out << YAML::BeginMap;
 
        // Summary
        out << YAML::Key << "summary" << YAML::Value << YAML::BeginMap;
        out << YAML::Key << "total_outputs" << YAML::Value << dictionary.total_outputs;
        out << YAML::Key << "total_inputs" << YAML::Value << dictionary.total_inputs;
        out << YAML::Key << "total_parameters" << YAML::Value << dictionary.total_parameters;
        out << YAML::Key << "total_config" << YAML::Value << dictionary.total_config;
        out << YAML::Key << "integrable_states" << YAML::Value << dictionary.integrable_states;
        out << YAML::Key << "unwired_inputs" << YAML::Value << dictionary.unwired_inputs;
        out << YAML::Key << "total_edges" << YAML::Value << edges.size();
        out << YAML::EndMap;
 
        dictionary.AppendComponentsToYAML(out);
 
        // Edges
        out << YAML::Key << "edges" << YAML::Value << YAML::BeginSeq;
        for (const auto &edge : edges) {
            out << YAML::BeginMap;
            out << YAML::Key << "source" << YAML::Value << edge.source;
            out << YAML::Key << "target" << YAML::Value << edge.target;
            out << YAML::Key << "kind" << YAML::Value
                << ((edge.kind == EdgeKind::Route) ? "route" : "resolve");
            out << YAML::EndMap;
        }
        out << YAML::EndSeq;
 
        out << YAML::EndMap;
 
        detail::WriteToFile(path, [&](std::ofstream &file) { file << out.c_str(); });
    }
};
 
} // namespace icarus