X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fnetwork%2Ftrain.cpp;h=4b6c8e86e1233955d3d4d04b19b6e1d27c522670;hb=8e3009751ee81e0e5312cae53ee844bdd9e49628;hp=5c4b457bad9bb3093edb962887d567798e6d8777;hpb=9ddcd066e37e4c72685817c042c30897786ece05;p=r2c2.git

diff --git a/source/network/train.cpp b/source/network/train.cpp
index 5c4b457..4b6c8e8 100644
--- a/source/network/train.cpp
+++ b/source/network/train.cpp
@@ -14,7 +14,7 @@ namespace Marklin {
 
 NetTrain::NetTrain(Client &c, const TrainInfoPacket &pkt):
 	client(c),
-	loco_type(client.get_catalogue().get_locomotive(pkt.loco_type)),
+	loco_type(client.get_catalogue().get_vehicle(pkt.loco_type)),
 	address(pkt.address),
 	name(pkt.name),
 	functions(0)
@@ -32,6 +32,12 @@ void NetTrain::set_control(const string &c, float v)
 	client.send(pkt);
 }
 
+float NetTrain::get_control(const string &c) const
+{
+	map<string, float>::const_iterator i = controls.find(c);
+	return i!=controls.end() ? i->second : 0;
+}
+
 void NetTrain::set_function(unsigned i, bool set)
 {
 	TrainFunctionPacket pkt;