Added the fields force_empty_first_frame and force_empty_last_frame in MTPTracker.

[mtp.git] / mtp_example.cc

diff --git a/mtp_example.cc b/mtp_example.cc
index 0db6054..92d4062 100644 (file)
--- a/mtp_example.cc
+++ b/mtp_example.cc
@@ -27,7 +27,7 @@
 
 using namespace std;
 
-#include "tracker.h"
+#include "mtp_tracker.h"
 
 //////////////////////////////////////////////////////////////////////
 
@@ -45,13 +45,12 @@ int main(int argc, char **argv) {
   int nb_time_steps = 8;
   int motion_amplitude = 1;
 
-  Tracker *tracker = new Tracker();
+  MTPTracker *tracker = new MTPTracker();
 
   tracker->allocate(nb_time_steps, nb_locations);
 
-  // We define the spatial structures by stating what are the possible
-  // motions of targets, and what are the entrances and the
-  // exits.
+  // We define the spatial structure by stating what are the possible
+  // motions of targets, and what are the entrances and the exits.
 
   // Here our example is a 1D space with motions from any location to
   // any location less than motion_amplitude away, entrance at
@@ -65,6 +64,9 @@ int main(int argc, char **argv) {
     tracker->exits[nb_locations - 1] = 1;
   }
 
+  tracker->force_empty_first_frame = 0;
+  tracker->force_empty_last_frame = 0;
+
   // We construct the graph corresponding to this structure
 
   tracker->build_graph();
@@ -84,14 +86,14 @@ int main(int argc, char **argv) {
     }
   }
 
-  // Then we two targets with the typical local minimum:
+  // Then we add two targets with a typical tracking local minimum
   //
   // * Target A moves from location 0 to the middle, stays there for a
-  //   while, and comes back, and is strongly detected on the first
+  //   while, and comes back. It is strongly detected on the first
   //   half
   //
   // * Target B moves from location nb_locations-1 to the middle, stay
-  //   there for a while, and comes back, and is strongly detected on
+  //   there for a while, and comes back. It is strongly detected on
   //   the second half
 
   int la, lb; // Target locations
@@ -116,12 +118,13 @@ int main(int argc, char **argv) {
     tracker->detection_scores[t][lb] = sb;
   }
 
-  { // Write down the tracker setting
+  { // Write down the tracker setting, so that we can use it as an
+    // example for the mtp command line
     ofstream out_tracker("tracker.dat");
     tracker->write(&out_tracker);
   }
 
-  // Does the tracking per se
+  // Performs the tracking per se
 
   tracker->track();