110 lines
2.4 KiB
C++
110 lines
2.4 KiB
C++
|
/*
|
||
|
|
* graph_balancer implementation
|
||
|
|
*
|
||
|
|
* Copyright (C) Jochen Sprickerhof
|
||
|
|
*
|
||
|
|
* Released under the GPL version 3.
|
||
|
|
*
|
||
|
|
*/
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @file stand alone graph balancer implementation
|
||
|
|
* reads graph in graphviz format and optimizes it
|
||
|
|
* @author Jochen Sprickerhof
|
||
|
|
*/
|
||
|
|
|
||
|
|
#ifndef _MSC_VER
|
||
|
|
#include <getopt.h>
|
||
|
|
#else
|
||
|
|
#include "XGetopt.h"
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#include <iostream>
|
||
|
|
using std::cout;
|
||
|
|
using std::cerr;
|
||
|
|
using std::endl;
|
||
|
|
|
||
|
|
#include <string>
|
||
|
|
using std::string;
|
||
|
|
|
||
|
|
#include <boost/graph/graphviz.hpp>
|
||
|
|
|
||
|
|
#include "slam6d/elch6D.h"
|
||
|
|
|
||
|
|
void usage(string progname) {
|
||
|
|
cout << "Usage: " << progname << " OPTIONS FILE" << endl <<
|
||
|
|
"\t -s \t\t first node in loop" << endl <<
|
||
|
|
"\t -e \t\t last node in loop" << endl <<
|
||
|
|
"\t -o \t\t output filename (must be given)" << endl;
|
||
|
|
}
|
||
|
|
|
||
|
|
int main(int argc, char *argv[])
|
||
|
|
{
|
||
|
|
|
||
|
|
if(argc < 2) {
|
||
|
|
usage(argv[0]);
|
||
|
|
exit(1);
|
||
|
|
}
|
||
|
|
|
||
|
|
int start = 0, end = 4;
|
||
|
|
string out_file;
|
||
|
|
/* options descriptor */
|
||
|
|
// 0: no arguments, 1: required argument, 2: optional argument
|
||
|
|
static struct option longopts[] = {
|
||
|
|
{ "end", required_argument, 0, 'e' },
|
||
|
|
{ "out", required_argument, 0, 'o' },
|
||
|
|
{ "start", required_argument, 0, 's' },
|
||
|
|
//{ "reduce", required_argument, 0, 'r' },
|
||
|
|
{ 0, 0, 0, 0} // needed, cf. getopt.h
|
||
|
|
};
|
||
|
|
|
||
|
|
int c;
|
||
|
|
while((c = getopt_long(argc, argv, "e:o:s:", longopts, NULL)) != -1) {
|
||
|
|
switch (c) {
|
||
|
|
case 'e':
|
||
|
|
end = atoi(optarg);
|
||
|
|
if(end < 0) {
|
||
|
|
cerr << "Error: Cannot end at a negative number." << endl;
|
||
|
|
exit(1);
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
case 'o':
|
||
|
|
out_file = optarg;
|
||
|
|
break;
|
||
|
|
case 's':
|
||
|
|
start = atoi(optarg);
|
||
|
|
if(start < 0) {
|
||
|
|
cerr << "Error: Cannot start at a negative number." << endl;
|
||
|
|
exit(1);
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
case '?':
|
||
|
|
default:
|
||
|
|
exit(1);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if(out_file.empty()) {
|
||
|
|
usage(argv[0]);
|
||
|
|
exit(1);
|
||
|
|
}
|
||
|
|
|
||
|
|
graph_t g(0);
|
||
|
|
boost::dynamic_properties dp(boost::ignore_other_properties);
|
||
|
|
boost::property_map<graph_t, boost::edge_weight_t>::type weight = get(boost::edge_weight, g);
|
||
|
|
dp.property("weight", weight);
|
||
|
|
|
||
|
|
std::ifstream gf(argv[optind]);
|
||
|
|
read_graphviz(gf, g, dp);
|
||
|
|
|
||
|
|
graph_t g_org(g);
|
||
|
|
double *w = new double[num_vertices(g)];
|
||
|
|
|
||
|
|
elch6D::graph_balancer(g, start, end, w);
|
||
|
|
elch6D::graph_weight_out(g_org, start, end, w, out_file);
|
||
|
|
|
||
|
|
delete[] w;
|
||
|
|
exit(0);
|
||
|
|
}
|
||
|
|
|