Simplifying + cosmetics.

[mtp.git] / mtp.cc

diff --git a/mtp.cc b/mtp.cc
index e07a738..979915b 100644 (file)
--- a/mtp.cc
+++ b/mtp.cc
@@ -44,10 +44,10 @@ int main(int argc, char **argv) {
 
   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();
@@ -57,9 +57,9 @@ int main(int argc, char **argv) {
 
   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();