X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fnetwork%2Fclient.cpp;fp=source%2Fnetwork%2Fclient.cpp;h=41cd28e900a712a5f5cc5693777bec615381e4f7;hb=4236c30c0a7a48854e9634e9ad7d8d6e0fd243a3;hp=7c583255b2b7b29f04b714c0590257f959c78afb;hpb=ce29d0ccf89840c9581314301de0ea2b5b261c43;p=r2c2.git

diff --git a/source/network/client.cpp b/source/network/client.cpp
index 7c58325..41cd28e 100644
--- a/source/network/client.cpp
+++ b/source/network/client.cpp
@@ -1,3 +1,4 @@
+#include <msp/core/maputils.h>
 #include "client.h"
 
 using namespace std;
@@ -56,10 +57,7 @@ void Client::set_halt(bool h)
 
 NetTrain &Client::get_train(unsigned addr) const
 {
-	map<unsigned, NetTrain *>::const_iterator i = trains.find(addr);
-	if(i==trains.end())
-		throw KeyError("Unknown train");
-	return *i->second;
+	return *get_item(trains, addr);
 }
 
 void Client::receive(const DriverStatePacket &pkt)