/* Create a graph representation of the VILLASnode configuration file.

*

* Author: Steffen Vogel <post@steffenvogel.de>

* SPDX-FileCopyrightText: 2014-2023 Institute for Automation of Complex Power Systems, RWTH Aachen University

* SPDX-License-Identifier: Apache-2.0

*/

#include <iostream>

// clang-format off

#include <graphviz/types.h>

#include <graphviz/gvcommon.h>

// clang-format on

#include <villas/super_node.hpp>

#include <villas/tool.hpp>

#include <villas/utils.hpp>

using namespace villas;

struct GVC_s {

GVCOMMON_t common;

char *config_path;

bool config_found;

char **input_filenames; // gvParseArgs

};

namespace villas {

namespace node {

namespace tools {

class Graph : public Tool {

public:

Graph(int argc, char *argv[])

: Tool(argc, argv, "graph", {SIGUSR1, SIGINT}), gvc(nullptr),

graph(nullptr) {

int ret;

ret = memory::init(DEFAULT_NR_HUGEPAGES);

if (ret)

throw RuntimeError("Failed to initialize memory");

this->argv[0] = (char *)"neato"; // Default layout engine

gvc = gvContext();

}

~Graph() { gvFreeContext(gvc); }

protected:

GVC_t *gvc;

graph_t *graph;

std::string configFilename;

void usage() override {

std::cout << "Usage: villas-graph [OPTIONS]" << std::endl

<< "For OPTIONS see dot(1).";

printCopyright();

}

void parse() override {

gvParseArgs(gvc, argc, argv);

std::list<std::string> filenames;

int i;

for (i = 0; gvc->input_filenames[i]; i++)

filenames.emplace_back(gvc->input_filenames[i]);

if (i == 0)

throw RuntimeError("No configuration file given!");

configFilename = filenames.front();

}

void handler(int signal, siginfo_t *siginfp, void *) override {

#ifndef _WIN32

switch (signal) {

case SIGINT:

// If interrupted we try to produce a partial rendering before exiting

if (graph)

gvRenderJobs(gvc, graph);

break;

case SIGUSR1:

// Note that we don't call gvFinalize() so that we don't start event-driven

// devices like -Tgtk or -Txlib */

exit(gvFreeContext(gvc));

break;

default: {

}

}

#endif // _WIN32

}

int main() override {

int ret;

villas::node::SuperNode sn;

sn.parse(configFilename);

sn.check();

sn.prepare();

graph = sn.getGraph();

ret = gvLayoutJobs(gvc, graph); // Take layout engine from command line

if (ret)

return ret;

return gvRenderJobs(gvc, graph);

}

};

} // namespace tools

} // namespace node

} // namespace villas

int main(int argc, char *argv[]) {

node::tools::Graph t(argc, argv);

return t.run();

}