X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fnetwork%2Ftrain.h;h=c908beea77c1b8f909e41f191a2597cbdaf50b4f;hb=f8873062b146028c07f55ad625d2767e45133c27;hp=3de4f7d38303bc067f55f4a83a1e070f01ccc07a;hpb=378f04522aa762729c0d5651fb60ad7e4d8ba9f3;p=r2c2.git

diff --git a/source/network/train.h b/source/network/train.h
index 3de4f7d..c908bee 100644
--- a/source/network/train.h
+++ b/source/network/train.h
@@ -1,18 +1,11 @@
-/* $Id$
-
-This file is part of the MSP Märklin suite
-Copyright © 2009  Mikkosoft Productions, Mikko Rasa
-Distributed under the GPL
-*/
-
-#ifndef MARKLINNET_TRAIN_H_
-#define MARKLINNET_TRAIN_H_
+#ifndef R2C2_NET_TRAIN_H_
+#define R2C2_NET_TRAIN_H_
 
 #include <sigc++/signal.h>
-#include "libmarklin/vehicletype.h"
+#include "libr2c2/vehicletype.h"
 #include "packets.h"
 
-namespace Marklin {
+namespace R2C2 {
 
 class Client;
 
@@ -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,22 +35,29 @@ 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 &);
 };
 
-} // namespace Marklin
+} // namespace R2C2
 
 #endif