#include <iostream>
#include <fstream>
-#include <stdlib.h>
using namespace std;
}
int main(int argc, char **argv) {
- int nb_locations = 1000;
- int nb_time_steps = 1000;
+ int nb_locations = 6;
+ int nb_time_steps = 5;
int motion_amplitude = 1;
Tracker *tracker = new Tracker(nb_time_steps, nb_locations);
for(int l = 0; l < nb_locations; l++) {
for(int k = 0; k < nb_locations; k++) {
- tracker->set_allowed_motion(l, k, abs(l - k) <= motion_amplitude);
+ tracker->allowed_motion[l][k] = abs(l - k) <= motion_amplitude;
}
- tracker->set_as_entrance(0, 1);
- tracker->set_as_exit(nb_locations - 1, 1);
+ tracker->entrances[0] = 1;
+ tracker->exits[nb_locations - 1] = 1;
}
tracker->build_graph();
for(int t = 0; t < nb_time_steps; t++) {
for(int l = 0; l < nb_locations; l++) {
- tracker->set_detection_score(t, l, detection_score(-1, 0.95));
+ tracker->detection_score[t][l] = detection_score(-1, 0.95);
}
- tracker->set_detection_score(t, nb_locations/2, detection_score(1, 0.95));
+ tracker->detection_score[t][nb_locations/2] = detection_score(1, 0.95);
}
tracker->track();
for(int t = 0; t < tracker->nb_trajectories(); t++) {
cout << "TRAJECTORY "
<< t
- << " [starting " << tracker->trajectory_entrance_time(t) << "]";
+ << " [starting " << tracker->trajectory_entrance_time(t)
+ << ", score " << tracker->trajectory_score(t) << "]";
for(int u = 0; u < tracker->trajectory_duration(t); u++) {
cout << " " << tracker->trajectory_location(t, u);
}
cout << endl;
}
+ {
+ ofstream dot("graph.dot");
+ tracker->print_graph_dot(&dot);
+ cout << "Wrote graph.dot." << endl;
+ }
+
delete tracker;
exit(EXIT_SUCCESS);