3dpcp/src/slam6d/graph_balancer.cc

110 lines
2.4 KiB
C++
Raw Normal View History

2012-09-16 12:33:11 +00:00
/*
* 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);
}