Renamed the Tracker class to MTPTracker.

[mtp.git] / mtp_tracker.h

diff --git a/mtp_tracker.h b/mtp_tracker.h
new file mode 100644 (file)
index 0000000..08de498
--- /dev/null
+++ b/mtp_tracker.h
@@ -0,0 +1,84 @@
+
+/*
+ *  mtp is the ``Multi Tracked Paths'', an implementation of the
+ *  k-shortest paths algorithm for multi-target tracking.
+ *
+ *  Copyright (c) 2012 Idiap Research Institute, http://www.idiap.ch/
+ *  Written by Francois Fleuret <francois.fleuret@idiap.ch>
+ *
+ *  This file is part of mtp.
+ *
+ *  mtp is free software: you can redistribute it and/or modify it
+ *  under the terms of the GNU General Public License version 3 as
+ *  published by the Free Software Foundation.
+ *
+ *  mtp is distributed in the hope that it will be useful, but WITHOUT
+ *  ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
+ *  or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public
+ *  License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with selector.  If not, see <http://www.gnu.org/licenses/>.
+ *
+ */
+
+#ifndef MTP_TRACKER_H
+#define MTP_TRACKER_H
+
+#include <iostream>
+
+using namespace std;
+
+#include "misc.h"
+#include "mtp_graph.h"
+
+class MTPTracker {
+  int _nb_locations, _nb_time_steps;
+  scalar_t **_detection_score;
+  int **_allowed_motion;
+  int *_entrances, *_exits;
+
+  MTPGraph *_graph;
+  scalar_t *_edge_lengths;
+
+  int early_pair_node(int t, int l);
+  int late_pair_node(int t, int l);
+
+public:
+
+  // The spatial structure
+  int **allowed_motion;
+  int *entrances, *exits;
+
+  // The detection scores at each node
+  scalar_t **detection_scores;
+
+  MTPTracker();
+  ~MTPTracker();
+
+  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();
+  void print_graph_dot(ostream *os);
+
+  // Compute the optimal set of trajectories
+
+  void track();
+
+  // Read-out of the optimal trajectories
+
+  int nb_trajectories();
+  scalar_t trajectory_score(int k);
+  int trajectory_entrance_time(int k);
+  int trajectory_duration(int k);
+  int trajectory_location(int k, int time_from_entry);
+};
+
+#endif