Use the TrackChain base class as destination in Router

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

diff --git a/source/libr2c2/trainrouter.h b/source/libr2c2/trainrouter.h
index 28fc80246f522cb48d6f1db88db7e66c917b37b9..c518522590bc4fdda11cf330b8a5532ffc4d7d56 100644 (file)
--- a/source/libr2c2/trainrouter.h
+++ b/source/libr2c2/trainrouter.h
@@ -8,9 +8,9 @@
 namespace R2C2 {
 
 class Block;
-class Route;
+class Layout;
 class Track;
-class Zone;
+class TrackChain;
 
 class TrainRouter: public TrainAI
 {
@@ -27,27 +27,43 @@ public:
        sigc::signal<void> signal_arrived;
 
 private:
+       struct Wait
+       {
+               Block *block;
+               Train *train;
+
+               Wait();
+       };
+
        typedef std::list<const Route *> RouteList;
 
        int priority;
        RouteList routes;
-       bool arriving;
-       const Train *yielding_to;
+       unsigned arriving;
+       const TrackChain *destination;
+       std::list<Wait> waits;
+       Msp::Time::TimeDelta delay;
+
+       bool update_pending;
 
 public:
        TrainRouter(Train &);
 
        void set_priority(int);
        int get_priority() const { return priority; }
-       void yield_to(const Train &);
 
        bool set_route(const Route *);
-       bool go_to(Track &);
-       bool go_to(const Zone &);
+       void add_wait(Block &, Train *);
        const Route *get_route() const;
 
+       void set_destination(const TrackChain &);
+       const TrackChain *get_destination() const { return destination; }
+       bool is_destination(Track &) const;
+       void set_departure_delay(const Msp::Time::TimeDelta &);
+       const Msp::Time::TimeDelta &get_departure_delay() const { return delay; }
+
        virtual void message(const Message &);
-       virtual void tick(const Msp::Time::TimeStamp &, const Msp::Time::TimeDelta &);
+       virtual void tick(const Msp::Time::TimeDelta &);
 
        void save(std::list<Msp::DataFile::Statement> &) const;
 
@@ -59,6 +75,8 @@ private:
        Route *create_lead_route(Route *, const Route *);
        bool advance_route(RouteList::iterator &, const Block &);
        bool is_on_route(const Block &);
+
+       static void create_plans(Layout &);
 };
 
 } // namespace R2C2