VILLASnode/src/villas-graph.cpp

/** Create a graph representation of the VILLASnode configuration file
 *
 * @file
 * @author Steffen Vogel <stvogel@eonerc.rwth-aachen.de>
 * @copyright 2014-2019, Institute for Automation of Complex Power Systems, EONERC
 * @license GNU General Public License (version 3)
 *
 * VILLASnode
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 * @addtogroup tools Test and debug tools
 * @{
 *********************************************************************************/

#include <iostream>

#include <graphviz/types.h>
#include <graphviz/gvcommon.h>

#include <villas/tool.hpp>
#include <villas/utils.hpp>
#include <villas/super_node.hpp>

using namespace villas;

struct GVC_s {
	GVCOMMON_t common;

	char *config_path;
	boolean config_found;

	/* gvParseArgs */
	char **input_filenames;
};

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()
	{
		std::cout << "Usage: villas-graph [OPTIONS]" << std::endl
			  << "For OPTIONS see dot(1).";

		printCopyright();
	}

	void parse()
	{
		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();
	}

	virtual void handler(int signal, siginfo_t *siginfp, void *)
	{
#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()
	{
		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();
}

/** @} */