X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fnetwork%2Ftrain.cpp;h=4ba4df08870891e354c10c86489bc3c82a0290ac;hb=b78b49d85fbb9b2901c77e6450cfd41c0a818ac1;hp=468b2200a0aa7d97dfd9e9724169a89e1b348da6;hpb=010d8321e982d1684fcbff5bf6fc2bdec7cb7bae;p=r2c2.git

diff --git a/source/network/train.cpp b/source/network/train.cpp
index 468b220..4ba4df0 100644
--- a/source/network/train.cpp
+++ b/source/network/train.cpp
@@ -1,49 +1,34 @@
-/* $Id$
-
-This file is part of the MSP Märklin suite
-Copyright © 2009  Mikkosoft Productions, Mikko Rasa
-Distributed under the GPL
-*/
-
 #include "client.h"
 #include "train.h"
 
 using namespace std;
 
-namespace Marklin {
+namespace R2C2 {
 
 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<VehicleType>(pkt.loco_type)),
 	address(pkt.address),
 	name(pkt.name),
-	speed(0),
-	reverse(0),
 	functions(0)
 { }
 
-void NetTrain::set_speed(unsigned s)
+void NetTrain::set_control(const string &c, float v)
 {
-	if(s==speed)
+	if(v==controls[c])
 		return;
 
-	TrainSpeedPacket pkt;
+	TrainControlPacket pkt;
 	pkt.address = address;
-	pkt.speed = s;
-	pkt.reverse = reverse;
+	pkt.control = c;
+	pkt.value = v;
 	client.send(pkt);
 }
 
-void NetTrain::set_reverse(bool r)
+float NetTrain::get_control(const string &c) const
 {
-	if(r==reverse)
-		return;
-
-	TrainSpeedPacket pkt;
-	pkt.address = address;
-	pkt.speed = speed;
-	pkt.reverse = r;
-	client.send(pkt);
+	map<string, float>::const_iterator i = controls.find(c);
+	return i!=controls.end() ? i->second : 0;
 }
 
 void NetTrain::set_function(unsigned i, bool set)
@@ -58,18 +43,21 @@ void NetTrain::set_function(unsigned i, bool set)
 	client.send(pkt);
 }
 
-void NetTrain::process_packet(const TrainSpeedPacket &pkt)
+void NetTrain::set_route(const string &r)
+{
+	if(r==route)
+		return;
+
+	TrainRoutePacket pkt;
+	pkt.address = address;
+	pkt.route = r;
+	client.send(pkt);
+}
+
+void NetTrain::process_packet(const TrainControlPacket &pkt)
 {
-	if(pkt.speed!=speed)
-	{
-		speed = pkt.speed;
-		signal_speed_changed.emit(speed);
-	}
-	if(pkt.reverse!=reverse)
-	{
-		reverse = pkt.reverse;
-		signal_reverse_changed.emit(reverse);
-	}
+	controls[pkt.control] = pkt.value;
+	signal_control_changed.emit(pkt.control, pkt.value);
 }
 
 void NetTrain::process_packet(const TrainFunctionPacket &pkt)
@@ -87,4 +75,10 @@ void NetTrain::process_packet(const TrainStatusPacket &pkt)
 	signal_status_changed.emit(status);
 }
 
-} // namespace Marklin
+void NetTrain::process_packet(const TrainRoutePacket &pkt)
+{
+	route = pkt.route;
+	signal_route_changed.emit(route);
+}
+
+} // namespace R2C2