graph_visitor.h

 
#include <string>
 
#include <boost/graph/adjacency_list.hpp>
#include <boost/graph/copy.hpp>
 
#include "graphs.h"
#include "lib/json.h"
 
#ifndef BACKENDS_GRAPHS_GRAPH_VISITOR_H_
#define BACKENDS_GRAPHS_GRAPH_VISITOR_H_
 
namespace P4::graphs {

class Graph_visitor : public Graphs {
 private:
    enum class PrevType { Control, Parser };

    struct fullGraphOpts {
        Graph fg;  // fullGraph

        unsigned long node_i = 0;  // node indexer
        unsigned cluster_i = 0;    // cluster indexer
    };

    struct edge_name_copier {
        edge_name_copier(const Graph &from, Graph &to) : from(from), to(to) {}
        const Graph &from;
        Graph &to;
 
        void operator()(Graph::edge_descriptor input, Graph::edge_descriptor output) const {
            auto label = boost::get(boost::edge_name, from, input);
 
            boost::put(boost::edge_name, to, output, label);
        }
    };
 
 public:
    Graph_visitor(std::filesystem::path graphsDir, const bool graphs, const bool fullGraph,
                  const bool jsonOut, std::filesystem::path filename)
        : graphsDir(std::move(graphsDir)),
          graphs(graphs),
          fullGraph(fullGraph),
          jsonOut(jsonOut),
          filename(std::move(filename)) {}

    const char *getType(const VertexType &v_type);

    const char *getPrevType(const PrevType &prev_type);
    void process(std::vector<Graph *> &controlGraphsArray, std::vector<Graph *> &parserGraphsArray);
    void writeGraphToFile(const Graph &g, const std::string &name);
 
 private:
    void forLoopJson(std::vector<Graph *> &graphsArray, PrevType prev_type);

    void forLoopFullGraph(std::vector<Graph *> &graphsArray, fullGraphOpts *opts,
                          PrevType prev_type);
 
    Util::JsonObject *json = nullptr;          // stores json that will be outputted
    Util::JsonArray *programBlocks = nullptr;  // stores objects in top level array "nodes"
    std::filesystem::path graphsDir;
    const bool graphs;     // output boost graphs to files
    const bool fullGraph;  // merge boost graphs into one CFG, and output to file
    const bool jsonOut;    // iterate over boost graphs, and create json representation of these
                           // graphs
    const std::filesystem::path filename;
};
 
}  // namespace P4::graphs
 
#endif /* BACKENDS_GRAPHS_GRAPH_VISITOR_H_ */