X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Flibmarklin%2Ftrain.h;h=3e7a56b7e423da7e9d4157d88138fc7e4e4ede85;hb=3d38bcbe8add00b558012fa0cd4e0be26464dbeb;hp=e8785f5e5994deb4ae275e670a0187676dd03534;hpb=378f04522aa762729c0d5651fb60ad7e4d8ba9f3;p=r2c2.git

diff --git a/source/libmarklin/train.h b/source/libmarklin/train.h
index e8785f5..3e7a56b 100644
--- a/source/libmarklin/train.h
+++ b/source/libmarklin/train.h
@@ -12,10 +12,10 @@ Distributed under the GPL
 #include <sigc++/trackable.h>
 #include <msp/time/timestamp.h>
 #include "block.h"
+#include "controller.h"
 
 namespace Marklin {
 
-class ControlModel;
 class Route;
 class Timetable;
 class Vehicle;
@@ -33,6 +33,7 @@ public:
 	public:
 		Loader(Train &);
 	private:
+		virtual void finish();
 		void block(unsigned);
 		void block_hint(unsigned);
 		void name(const std::string &);
@@ -77,7 +78,9 @@ private:
 	std::list<BlockRef> cur_blocks;
 	std::list<BlockRef> rsv_blocks;
 	Block *pending_block;
-	ControlModel *control;
+	bool reserving;
+	bool advancing;
+	Controller *controller;
 	Timetable *timetable;
 	bool active;
 	unsigned current_speed;
@@ -108,7 +111,7 @@ public:
 	const std::string &get_name() const { return name; }
 	void set_priority(int);
 	int get_priority() const { return priority; }
-	ControlModel &get_control() const { return *control; }
+	Controller &get_controller() const { return *controller; }
 
 	void add_vehicle(const VehicleType &);
 	void remove_vehicle(unsigned);
@@ -120,6 +123,7 @@ public:
 	void set_active(bool);
 	void set_function(unsigned, bool);
 	float get_control(const std::string &) const;
+	float get_speed() const;
 	bool is_active() const { return active; }
 	bool get_function(unsigned) const;
 	unsigned get_functions() const { return functions; }
@@ -142,6 +146,7 @@ public:
 
 	void save(std::list<Msp::DataFile::Statement> &) const;
 private:
+	void control_changed(const Controller::Control &);
 	void loco_speed_event(unsigned, unsigned, bool);
 	void loco_func_event(unsigned, unsigned, bool);
 	void sensor_event(unsigned, bool);