X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Flibr2c2%2Ftrainrouter.h;h=457bdf78cf59c5dd5fcfb4c49db644f991ec888b;hb=8066ed50f94d046b0e009e24315ca7980439bb8c;hp=9fab2892efe15803da1f413277f7a6bd25678122;hpb=83c6923396fcfa42e6adf949b9476663f322748b;p=r2c2.git

diff --git a/source/libr2c2/trainrouter.h b/source/libr2c2/trainrouter.h
index 9fab289..457bdf7 100644
--- a/source/libr2c2/trainrouter.h
+++ b/source/libr2c2/trainrouter.h
@@ -3,6 +3,7 @@
 
 #include <list>
 #include <msp/datafile/objectloader.h>
+#include "trackchain.h"
 #include "trainai.h"
 
 namespace R2C2 {
@@ -10,7 +11,6 @@ namespace R2C2 {
 class Block;
 class Layout;
 class Track;
-class TrackChain;
 class TrainRouteMetric;
 class TrainRoutePlanner;
 
@@ -25,6 +25,14 @@ public:
 		void route(const std::string &);
 	};
 
+	struct Waypoint
+	{
+		const TrackChain *chain;
+		TrackChain::Direction direction;
+
+		Waypoint(const TrackChain &, TrackChain::Direction = TrackChain::UNSPECIFIED);
+	};
+
 	struct SequencePoint
 	{
 		Block *block;
@@ -40,10 +48,11 @@ public:
 private:
 	typedef std::list<const Route *> RouteList;
 
-	enum ArrivalState
+	enum TravelState
 	{
 		ON_THE_WAY,
 		WAITING_FOR_SEQUENCE,
+		SEQUENCE_CHECK_PENDING,
 		RESERVED_TO_END,
 		ADVANCED_TO_END,
 		ARRIVED
@@ -58,14 +67,13 @@ private:
 	int priority;
 	RouteList routes;
 	RouteList::iterator reserving_route;
-	ArrivalState arrival;
-	std::vector<const TrackChain *> waypoints;
+	TravelState state;
+	std::vector<Waypoint> waypoints;
 	bool waypoints_changed;
 	std::vector<TrainRouteMetric *> metrics;
 	bool metrics_stale;
 	std::list<SequencePoint> sequence_points;
 	unsigned current_sequence;
-	bool sequence_check_pending;
 	Msp::Time::TimeDelta delay;
 	Msp::Time::TimeDelta duration;
 	Msp::RefPtr<TrainRoutePlanner> planner;
@@ -86,10 +94,10 @@ private:
 
 public:
 	void set_destination(const TrackChain &);
-	const TrackChain *get_destination() const { return waypoints.empty() ? 0 : waypoints.back(); }
-	void add_waypoint(const TrackChain &);
+	const TrackChain *get_destination() const { return waypoints.empty() ? 0 : waypoints.back().chain; }
+	void add_waypoint(const TrackChain &, TrackChain::Direction = TrackChain::UNSPECIFIED);
 	unsigned get_n_waypoints() const { return waypoints.size(); }
-	const TrackChain &get_waypoint(unsigned) const;
+	const Waypoint &get_waypoint(unsigned) const;
 	const TrainRouteMetric &get_metric(int = -1) const;
 	void set_departure_delay(const Msp::Time::TimeDelta &);
 	const Msp::Time::TimeDelta &get_departure_delay() const { return delay; }
@@ -110,7 +118,7 @@ private:
 	bool create_lead_route();
 	bool advance_to_track(RouteList::iterator &, const TrackIter &);
 
-	static void get_routers(Layout &, std::vector<TrainRouter *> &);
+	static void get_routers(Layout &, std::vector<TrainRouter *> &, TrainRoutePlanner * = 0);
 	static void start_planning(Layout &);
 	static void apply_plan(Layout &, TrainRoutePlanner &);
 };