X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fnetwork%2Ftrain.h;h=c908beea77c1b8f909e41f191a2597cbdaf50b4f;hb=HEAD;hp=bffa8a24fbdff372130c822d6abaf363f7e7cf0b;hpb=1ff06c5bc46a677fa389ef86c6b26664368f1653;p=r2c2.git

diff --git a/source/network/train.h b/source/network/train.h
index bffa8a2..c908bee 100644
--- a/source/network/train.h
+++ b/source/network/train.h
@@ -1,10 +1,3 @@
-/* $Id$
-
-This file is part of R²C²
-Copyright © 2009  Mikkosoft Productions, Mikko Rasa
-Distributed under the GPL
-*/
-
 #ifndef R2C2_NET_TRAIN_H_
 #define R2C2_NET_TRAIN_H_
 
@@ -22,6 +15,8 @@ public:
 	sigc::signal<void, const std::string &> signal_name_changed;
 	sigc::signal<void, const std::string &, float> signal_control_changed;
 	sigc::signal<void, unsigned, bool> signal_function_changed;
+	sigc::signal<void, float> signal_target_speed_changed;
+	sigc::signal<void, bool> signal_reverse_changed;
 	sigc::signal<void, const std::string &> signal_route_changed;
 	sigc::signal<void, const std::string &> signal_status_changed;
 
@@ -31,6 +26,8 @@ private:
 	unsigned address;
 	std::string name;
 	std::map<std::string, float> controls;
+	float target_speed;
+	bool reverse;
 	unsigned functions;
 	std::string route;
 	std::string status;
@@ -38,18 +35,25 @@ private:
 public:
 	NetTrain(Client &, const TrainInfoPacket &);
 
+	Client &get_client() const { return client; }
 	const VehicleType &get_loco_type() const { return loco_type; }
 	unsigned get_address() const { return address; }
 	const std::string &get_name() const { return name; }
 	void set_control(const std::string &, float);
 	float get_control(const std::string &) const;
+	void set_target_speed(float);
+	void set_reverse(bool);
+	float get_target_speed() const { return target_speed; }
+	bool get_reverse() const { return reverse; }
 	void set_function(unsigned, bool);
 	bool get_function(unsigned i) const { return (functions>>i)&1; }
 	void set_route(const std::string &);
 	const std::string &get_route() const { return route; }
+	const std::string &get_status() const { return status; }
 
 	void process_packet(const TrainControlPacket &);
 	void process_packet(const TrainFunctionPacket &);
+	void process_packet(const TrainAIControlPacket &);
 	void process_packet(const TrainRoutePacket &);
 	void process_packet(const TrainStatusPacket &);
 };