X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fnetwork%2Fclient.cpp;h=50db7df6d817bdcbb738e4182262de9df2fd275f;hb=6ba6af3637c299ab00828c49de9151429488cc17;hp=4eb4b25f5aefdd81498cd06a66b78925426b03f8;hpb=010d8321e982d1684fcbff5bf6fc2bdec7cb7bae;p=r2c2.git

diff --git a/source/network/client.cpp b/source/network/client.cpp
index 4eb4b25..50db7df 100644
--- a/source/network/client.cpp
+++ b/source/network/client.cpp
@@ -43,9 +43,9 @@ void Client::connect(const Net::SockAddr &addr)
 	comm = new Net::Communicator(*socket, proto, *this);
 }
 
-NetTrain &Client::get_train(unsigned addr)
+NetTrain &Client::get_train(unsigned addr) const
 {
-	map<unsigned, NetTrain *>::iterator i = trains.find(addr);
+	map<unsigned, NetTrain *>::const_iterator i = trains.find(addr);
 	if(i==trains.end())
 		throw KeyError("Unknown train");
 	return *i->second;
@@ -58,7 +58,7 @@ void Client::receive(const TrainInfoPacket &pkt)
 	signal_train_added.emit(*train);
 }
 
-void Client::receive(const TrainSpeedPacket &pkt)
+void Client::receive(const TrainControlPacket &pkt)
 {
 	get_train(pkt.address).process_packet(pkt);
 }
@@ -73,4 +73,19 @@ void Client::receive(const TrainStatusPacket &pkt)
 	get_train(pkt.address).process_packet(pkt);
 }
 
+void Client::receive(const RouteInfoPacket &pkt)
+{
+	routes.push_back(pkt.name);
+}
+
+void Client::receive(const TrainRoutePacket &pkt)
+{
+	get_train(pkt.address).process_packet(pkt);
+}
+
+void Client::receive(const ErrorPacket &pkt)
+{
+	signal_error.emit(pkt.message);
+}
+
 } // namespace Marklin