X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fnetwork%2Fclient.h;h=3a9cbc0717a659cc042e8874b6333d5f099b0f89;hb=2fe7cbcb761487bc7409b93b372da6f8ab3c581e;hp=ce8dcb66fc15cdb681157b57d3b5e50b968741fc;hpb=e95240551e3c34d2fb21e99c358fa5922a2060c9;p=r2c2.git

diff --git a/source/network/client.h b/source/network/client.h
index ce8dcb6..3a9cbc0 100644
--- a/source/network/client.h
+++ b/source/network/client.h
@@ -20,7 +20,9 @@ namespace Marklin {
 class Client: public Msp::Net::PacketReceiver<TrainInfoPacket>,
 	Msp::Net::PacketReceiver<TrainSpeedPacket>,
 	Msp::Net::PacketReceiver<TrainFunctionPacket>,
-	Msp::Net::PacketReceiver<TrainStatusPacket>
+	Msp::Net::PacketReceiver<TrainStatusPacket>,
+	Msp::Net::PacketReceiver<RouteInfoPacket>,
+	Msp::Net::PacketReceiver<TrainRoutePacket>
 {
 public:
 	sigc::signal<void, NetTrain &> signal_train_added;
@@ -31,6 +33,7 @@ private:
 	Msp::Net::StreamSocket *socket;
 	Msp::Net::Communicator *comm;
 	Msp::IO::EventDispatcher *event_disp;
+	std::list<std::string> routes;
 	std::map<unsigned, NetTrain *> trains;
 
 public:
@@ -45,6 +48,7 @@ public:
 	{ if(comm) comm->send(pkt); }
 
 	const Catalogue &get_catalogue() const { return catalogue; }
+	const std::list<std::string> &get_routes() const { return routes; }
 	NetTrain &get_train(unsigned);
 
 private:
@@ -52,6 +56,8 @@ private:
 	virtual void receive(const TrainSpeedPacket &);
 	virtual void receive(const TrainFunctionPacket &);
 	virtual void receive(const TrainStatusPacket &);
+	virtual void receive(const RouteInfoPacket &);
+	virtual void receive(const TrainRoutePacket &);
 };
 
 } // namespace Marklin