X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;ds=inline;f=tracker.h;h=b56ac61898465358c30c1b3ab3f4f77e17c22ee6;hb=9665f6b8d842f02facda1590200742482d4362fa;hp=40e21abe7f8c76f081fc08cbb295a4d46d963d12;hpb=9dba69873c4b22f885a70740df7f57b0d6a37fd6;p=mtp.git

diff --git a/tracker.h b/tracker.h
index 40e21ab..b56ac61 100644
--- 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();