Add an actual heuristic for wait time

[r2c2.git] / source / libr2c2 / trainroutemetric.h

diff --git a/source/libr2c2/trainroutemetric.h b/source/libr2c2/trainroutemetric.h
index c63f25e77783adea16d1baf950ec2c359f96e7cc..cef08cb2ae67ba9c6fd33f5411850371b8f45128 100644 (file)
--- a/source/libr2c2/trainroutemetric.h
+++ b/source/libr2c2/trainroutemetric.h
@@ -3,12 +3,12 @@
 
 #include <map>
 #include <vector>
+#include "trackchain.h"
 
 namespace R2C2 {
 
 class Layout;
 class Track;
-class TrackChain;
 
 // Metrics store iterators facing away from the goal
 class TrainRouteMetric
@@ -39,9 +39,10 @@ private:
 
 public:
        TrainRouteMetric() { }
-       TrainRouteMetric(const TrackChain &);
+       TrainRouteMetric(const TrackChain &, TrackChain::Direction = TrackChain::UNSPECIFIED);
 
        void chain_to(const TrainRouteMetric &);
+       float get_distance_from(const Track &) const;
        float get_distance_from(const Track &, unsigned) const;
 };