2 ///////////////////////////////////////////////////////////////////////////
3 // This program is free software: you can redistribute it and/or modify //
4 // it under the terms of the version 3 of the GNU General Public License //
5 // as published by the Free Software Foundation. //
7 // This program is distributed in the hope that it will be useful, but //
8 // WITHOUT ANY WARRANTY; without even the implied warranty of //
9 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU //
10 // General Public License for more details. //
12 // You should have received a copy of the GNU General Public License //
13 // along with this program. If not, see <http://www.gnu.org/licenses/>. //
15 // Written by and Copyright (C) Francois Fleuret //
16 // Contact <francois.fleuret@idiap.ch> for comments & bug reports //
17 ///////////////////////////////////////////////////////////////////////////
21 // Takes the graph description file as input and produces a dot file.
23 // EXAMPLE: ./mtp ./graph2.txt | dot -T pdf -o- | xpdf -
36 #include "mtp_graph.h"
39 //////////////////////////////////////////////////////////////////////
41 void find_best_paths(int nb_vertices,
42 int nb_edges, int *ea, int *eb, scalar_t *el,
44 int *result_edge_occupation) {
45 MTPGraph graph(nb_vertices, nb_edges, ea, eb, source, sink);
46 graph.find_best_paths(el, result_edge_occupation);
49 //////////////////////////////////////////////////////////////////////
51 int main(int argc, char **argv) {
53 int nb_time_steps = 4;
55 Tracker tracker(nb_time_steps, nb_locations);
57 for(int l = 0; l < nb_locations; l++) {
58 for(int k = 0; k < nb_locations; k++) {
59 if(abs(l - k) <= 1) tracker.set_allowed_motion(l, k);
63 for(int t = 0; t < nb_time_steps; t++) {
64 for(int l = 0; l < nb_locations; l++) {
65 tracker.set_detection_score(t, l, -1.0);
67 tracker.set_detection_score(t, 0, 1.0);
75 cerr << argv[0] << " <graph file>" << endl;
79 ifstream *file = new ifstream(argv[1]);
81 int nb_edges, nb_vertices;
86 (*file) >> nb_vertices >> nb_edges;
87 (*file) >> source >> sink;
89 scalar_t *edge_lengths = new scalar_t[nb_edges];
90 int *vertex_from = new int[nb_edges];
91 int *vertex_to = new int[nb_edges];
92 int *result_edge_occupation = new int[nb_edges];
94 for(int e = 0; e < nb_edges; e++) {
95 (*file) >> vertex_from[e] >> vertex_to[e] >> edge_lengths[e];
98 find_best_paths(nb_vertices, nb_edges,
99 vertex_from, vertex_to, edge_lengths,
101 result_edge_occupation);
103 dot_print(nb_vertices, nb_edges,
104 vertex_from, vertex_to, edge_lengths,
106 result_edge_occupation);
108 delete[] result_edge_occupation;
109 delete[] edge_lengths;
110 delete[] vertex_from;
115 cerr << "Can not open " << argv[1] << endl;