Cosmetics.

[mtp.git] / tracker.h

diff --git a/tracker.h b/tracker.h
index 40e21ab..b56ac61 100644 (file)
--- a/tracker.h
+++ b/tracker.h
@@ -40,13 +40,23 @@ class Tracker {
 
 public:
 
-  scalar_t **detection_score;
+  // The spatial structure
   int **allowed_motion;
   int *entrances, *exits;
 
-  Tracker(int nb_time_steps, int nb_locations);
+  // The detection scores at each node
+  scalar_t **detection_scores;
+
+  Tracker();
   ~Tracker();
 
+  void allocate(int nb_time_steps, int nb_locations);
+  void free();
+
+  void write(ostream *os);
+  void read(istream *is);
+  void write_trajectories(ostream *os);
+
   // Build or print the graph needed for the tracking per se
 
   void build_graph();