X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fnetwork%2Fclient.h;h=94fa78545da9ef52b977acac9084b694256f7636;hb=3ffe8b384fabeee1c3e636b83dd0006f3f5acfe1;hp=02931e553d69a9efc1a8afed0d4714da924654f2;hpb=7b4155fa0581d7c5fe198e4fb1dea61fa0846eec;p=r2c2.git

diff --git a/source/network/client.h b/source/network/client.h
index 02931e5..94fa785 100644
--- a/source/network/client.h
+++ b/source/network/client.h
@@ -15,6 +15,7 @@ class Client: Msp::Net::PacketReceiver<DriverStatePacket>,
 	Msp::Net::PacketReceiver<TrainInfoPacket>,
 	Msp::Net::PacketReceiver<TrainFunctionPacket>,
 	Msp::Net::PacketReceiver<TrainControlPacket>,
+	Msp::Net::PacketReceiver<TrainAIControlPacket>,
 	Msp::Net::PacketReceiver<TrainStatusPacket>,
 	Msp::Net::PacketReceiver<RouteInfoPacket>,
 	Msp::Net::PacketReceiver<TrainRoutePacket>,
@@ -53,6 +54,8 @@ public:
 	const std::list<std::string> &get_routes() const { return routes; }
 	void set_power(bool);
 	void set_halt(bool);
+	bool get_power() const { return power; }
+	bool get_halt() const { return halt; }
 	NetTrain &get_train(unsigned) const;
 	const std::map<unsigned, NetTrain *> &get_trains() const { return trains; }
 
@@ -62,6 +65,7 @@ private:
 	virtual void receive(const TrainInfoPacket &);
 	virtual void receive(const TrainControlPacket &);
 	virtual void receive(const TrainFunctionPacket &);
+	virtual void receive(const TrainAIControlPacket &);
 	virtual void receive(const TrainStatusPacket &);
 	virtual void receive(const RouteInfoPacket &);
 	virtual void receive(const TrainRoutePacket &);