Support routing through multiple waypoints

[r2c2.git] / source / libr2c2 / trainrouter.cpp

diff --git a/source/libr2c2/trainrouter.cpp b/source/libr2c2/trainrouter.cpp
index b5a5bef11f674f3301a36c1304558bc63b2bc8fa..da3b7d515fe253b79db3412ea0e98d2bd062d6ea 100644 (file)
--- a/source/libr2c2/trainrouter.cpp
+++ b/source/libr2c2/trainrouter.cpp
@@ -2,9 +2,9 @@
 #include "route.h"
 #include "trackiter.h"
 #include "train.h"
+#include "trackchain.h"
 #include "trainrouteplanner.h"
 #include "trainrouter.h"
-#include "zone.h"
 
 using namespace std;
 using namespace Msp;
@@ -15,8 +15,7 @@ TrainRouter::TrainRouter(Train &t):
        TrainAI(t),
        priority(0),
        arriving(0),
-       dest_zone(0),
-       dest_block(0),
+       destination(0),
        update_pending(false)
 {
        train.get_layout().signal_block_reserved.connect(sigc::mem_fun(this, &TrainRouter::block_reserved));
@@ -57,6 +56,14 @@ bool TrainRouter::set_route(const Route *r)
        train.stop_at(0);
        arriving = 0;
 
+       /* TODO destination should also be cleared when manually setting a different
+       route, but not when the planner calls this. */
+       if(!r)
+       {
+               destination = 0;
+               waypoints.clear();
+       }
+
        train.refresh_blocks_from(*fncb);
 
        const Route *route = get_route();
@@ -81,33 +88,38 @@ const Route *TrainRouter::get_route() const
        return routes.front();
 }
 
-void TrainRouter::set_destination(const Zone &zone)
+void TrainRouter::set_destination(const TrackChain &d)
 {
-       dest_zone = &zone;
-       dest_block = 0;
+       destination = &d;
        update_pending = true;
 }
 
-void TrainRouter::set_destination(const Block &block)
+bool TrainRouter::is_destination(Track &track) const
+{
+       if(destination)
+               return destination->has_track(track);
+       else
+               return false;
+}
+
+void TrainRouter::add_waypoint(const TrackChain &wp)
 {
-       dest_zone = 0;
-       dest_block = █
+       waypoints.push_back(&wp);
        update_pending = true;
 }
 
-bool TrainRouter::has_destination() const
+bool TrainRouter::is_waypoint(unsigned index, Track &track) const
 {
-       return dest_zone || dest_block;
+       if(index>=waypoints.size())
+               throw out_of_range("TrainRouter::is_waypoint");
+
+       return waypoints[index]->has_track(track);
 }
 
-bool TrainRouter::is_destination(Track &track) const
+void TrainRouter::set_departure_delay(const Time::TimeDelta &d)
 {
-       if(dest_zone)
-               return dest_zone->has_track(track);
-       else if(dest_block)
-               return dest_block->has_track(track);
-       else
-               return false;
+       delay = d;
+       update_pending = true;
 }
 
 void TrainRouter::message(const Message &msg)
@@ -121,24 +133,33 @@ void TrainRouter::message(const Message &msg)
        }
        else if(msg.type=="clear-route")
                set_route(0);
-       else if(msg.type=="set-destination-block")
+       else if(msg.type=="set-destination")
        {
-               if(msg.value.check_type<Block *>())
-                       set_destination(*msg.value.value<Block *>());
+               if(msg.value.check_type<TrackChain *>())
+                       set_destination(*msg.value.value<TrackChain *>());
                else
-                       set_destination(*msg.value.value<const Block *>());
+                       set_destination(*msg.value.value<const TrackChain *>());
        }
-       else if(msg.type=="set-destination-zone")
+       else if(msg.type=="add-waypoint")
        {
-               if(msg.value.check_type<Zone *>())
-                       set_destination(*msg.value.value<Zone *>());
+               if(msg.value.check_type<TrackChain *>())
+                       add_waypoint(*msg.value.value<TrackChain *>());
                else
-                       set_destination(*msg.value.value<const Zone *>());
+                       add_waypoint(*msg.value.value<const TrackChain *>());
        }
+       else if(msg.type=="set-departure-delay")
+               set_departure_delay(msg.value.value<Time::TimeDelta>());
 }
 
-void TrainRouter::tick(const Time::TimeDelta &)
+void TrainRouter::tick(const Time::TimeDelta &dt)
 {
+       if(delay)
+       {
+               delay -= dt;
+               if(delay<=Time::zero)
+                       delay = Time::zero;
+       }
+
        if(update_pending)
                create_plans(train.get_layout());