]> git.tdb.fi Git - r2c2.git/blobdiff - source/network/server.cpp
Rename the project to R²C²
[r2c2.git] / source / network / server.cpp
index 17142f8dcdaeacf5340e710ff8258c00773846ae..404faa94a6fa83826f46403fd7ff08110d1d231d 100644 (file)
@@ -1,34 +1,31 @@
 /* $Id$
 
-This file is part of the MSP Märklin suite
-Copyright © 2009  Mikkosoft Productions, Mikko Rasa
+This file is part of R²C²
+Copyright © 2009-2010  Mikkosoft Productions, Mikko Rasa
 Distributed under the GPL
 */
 
 #include <msp/net/inet.h>
-#include "libmarklin/control.h"
-#include "libmarklin/locomotive.h"
-#include "libmarklin/locotype.h"
+#include "libr2c2/route.h"
+#include "libr2c2/train.h"
+#include "libr2c2/vehicletype.h"
 #include "server.h"
 
 using namespace std;
 using namespace Msp;
 
-namespace Marklin {
+namespace R2C2 {
 
-Server::Server(TrafficManager &tm):
-       trfc_mgr(tm),
+Server::Server(Layout &l):
+       layout(l),
        listen_sock(Net::INET),
        event_disp(0)
 {
-       const list<Train *> &trains = trfc_mgr.get_trains();
-       for(list<Train *>::const_iterator i=trains.begin(); i!=trains.end(); ++i)
-       {
-               Locomotive &loco = (*i)->get_locomotive();
-               (*i)->signal_target_speed_changed.connect(sigc::bind<0>(sigc::mem_fun(this, &Server::train_speed_changed), sigc::ref(**i)));
-               loco.signal_function_changed.connect(sigc::bind<0>(sigc::mem_fun(this, &Server::train_function_changed), sigc::ref(**i)));
-               (*i)->signal_status_changed.connect(sigc::bind<0>(sigc::mem_fun(this, &Server::train_status_changed), sigc::ref(**i)));
-       }
+       layout.signal_train_added.connect(sigc::mem_fun(this, &Server::train_added));
+
+       const map<unsigned, Train *> &trains = layout.get_trains();
+       for(map<unsigned, Train *>::const_iterator i=trains.begin(); i!=trains.end(); ++i)
+               train_added(*i->second);
 
        listen_sock.listen(Net::InetAddr(0, 8315), 4);
        listen_sock.signal_data_available.connect(sigc::mem_fun(this, &Server::incoming_connection));
@@ -50,39 +47,48 @@ void Server::incoming_connection()
 
 void Server::train_added(Train &train)
 {
-       Locomotive &loco = train.get_locomotive();
-       train.signal_target_speed_changed.connect(sigc::bind<0>(sigc::mem_fun(this, &Server::train_speed_changed), sigc::ref(train)));
-       loco.signal_function_changed.connect(sigc::bind<0>(sigc::mem_fun(this, &Server::train_function_changed), sigc::ref(train)));
+       train.signal_control_changed.connect(sigc::bind<0>(sigc::mem_fun(this, &Server::train_control_changed), sigc::ref(train)));
+       train.signal_function_changed.connect(sigc::bind<0>(sigc::mem_fun(this, &Server::train_function_changed), sigc::ref(train)));
+       train.signal_route_changed.connect(sigc::bind<0>(sigc::mem_fun(this, &Server::train_route_changed), sigc::ref(train)));
        train.signal_status_changed.connect(sigc::bind<0>(sigc::mem_fun(this, &Server::train_status_changed), sigc::ref(train)));
 
        TrainInfoPacket pkt;
-       pkt.address = loco.get_address();
-       pkt.loco_type = loco.get_type().get_article_number();
+       pkt.address = train.get_address();
+       pkt.loco_type = train.get_locomotive_type().get_article_number().str();
        pkt.name = train.get_name();
        send(pkt);
 }
 
-void Server::train_speed_changed(const Train &train, unsigned speed)
+void Server::train_control_changed(const Train &train, const string &control, float value)
 {
-       TrainSpeedPacket pkt;
-       pkt.address = train.get_locomotive().get_address();
-       pkt.speed = speed;
-       pkt.reverse = train.get_locomotive().get_reverse();
+       TrainControlPacket pkt;
+       pkt.address = train.get_address();
+       pkt.control = control;
+       pkt.value = value;
        send(pkt);
 }
 
 void Server::train_function_changed(const Train &train, unsigned, bool)
 {
        TrainFunctionPacket pkt;
-       pkt.address = train.get_locomotive().get_address();
-       pkt.functions = train.get_locomotive().get_functions();
+       pkt.address = train.get_address();
+       pkt.functions = train.get_functions();
+       send(pkt);
+}
+
+void Server::train_route_changed(const Train &train, const Route *route)
+{
+       TrainRoutePacket pkt;
+       pkt.address = train.get_address();
+       if(route)
+               pkt.route = route->get_name();
        send(pkt);
 }
 
 void Server::train_status_changed(const Train &train, const string &status)
 {
        TrainStatusPacket pkt;
-       pkt.address = train.get_locomotive().get_address();
+       pkt.address = train.get_address();
        pkt.status = status;
        send(pkt);
 }
@@ -113,35 +119,44 @@ Server::Connection::~Connection()
 
 void Server::Connection::handshake_done()
 {
-       const list<Train *> &trains = server.trfc_mgr.get_trains();
-       for(list<Train *>::const_iterator i=trains.begin(); i!=trains.end(); ++i)
+       const set<Route *> &routes = server.layout.get_routes();
+       for(set<Route *>::const_iterator i=routes.begin(); i!=routes.end(); ++i)
        {
-               Locomotive &loco = (*i)->get_locomotive();
+               RouteInfoPacket pkt;
+               pkt.name = (*i)->get_name();
+               comm.send(pkt);
+       }
+
+       const map<unsigned, Train *> &trains = server.layout.get_trains();
+       for(map<unsigned, Train *>::const_iterator i=trains.begin(); i!=trains.end(); ++i)
+       {
+               const Train &train = *i->second;
 
                {
                        TrainInfoPacket pkt;
-                       pkt.address = loco.get_address();
-                       pkt.loco_type = loco.get_type().get_article_number();
-                       pkt.name = (*i)->get_name();
+                       pkt.address = train.get_address();
+                       pkt.loco_type = train.get_locomotive_type().get_article_number().str();
+                       pkt.name = train.get_name();
                        comm.send(pkt);
                }
+               // XXX Need control enumeration to send control packets
                {
-                       TrainSpeedPacket pkt;
-                       pkt.address = loco.get_address();
-                       pkt.speed = (*i)->get_target_speed();
-                       pkt.reverse = loco.get_reverse();
+                       TrainFunctionPacket pkt;
+                       pkt.address = train.get_address();
+                       pkt.functions = train.get_functions();
                        comm.send(pkt);
                }
                {
-                       TrainFunctionPacket pkt;
-                       pkt.address = loco.get_address();
-                       pkt.functions = loco.get_functions();
+                       TrainStatusPacket pkt;
+                       pkt.address = train.get_address();
+                       pkt.status = train.get_status();
                        comm.send(pkt);
                }
+               if(train.get_route())
                {
-                       TrainStatusPacket pkt;
-                       pkt.address = loco.get_address();
-                       pkt.status = (*i)->get_status();
+                       TrainRoutePacket pkt;
+                       pkt.address = train.get_address();
+                       pkt.route = train.get_route()->get_name();
                        comm.send(pkt);
                }
        }
@@ -153,22 +168,58 @@ void Server::Connection::end_of_file()
        stale = true;
 }
 
-void Server::Connection::receive(const TrainSpeedPacket &pkt)
+void Server::Connection::receive(const TrainControlPacket &pkt)
 {
-       Locomotive &loco = server.trfc_mgr.get_control().get_locomotive(pkt.address);
-       Train &train = server.trfc_mgr.get_train_by_locomotive(loco);
-       if(pkt.reverse!=loco.get_reverse())
-               train.set_reverse(pkt.reverse);
-       else
-               train.set_speed(pkt.speed);
+       try
+       {
+               Train &train = server.layout.get_train(pkt.address);
+               train.set_control(pkt.control, pkt.value);
+       }
+       catch(const Exception &e)
+       {
+               error(e.what());
+       }
 }
 
 void Server::Connection::receive(const TrainFunctionPacket &pkt)
 {
-       Locomotive &loco = server.trfc_mgr.get_control().get_locomotive(pkt.address);
-       for(unsigned i=0; i<9; ++i)
-               if(((pkt.functions^loco.get_functions())>>i)&1)
-                       loco.set_function(i, (pkt.functions>>i)&1);
+       try
+       {
+               Train &train = server.layout.get_train(pkt.address);
+               for(unsigned i=0; i<9; ++i)
+                       if(((pkt.functions^train.get_functions())>>i)&1)
+                               train.set_function(i, (pkt.functions>>i)&1);
+       }
+       catch(const Exception &e)
+       {
+               error(e.what());
+       }
+}
+
+void Server::Connection::receive(const TrainRoutePacket &pkt)
+{
+       try
+       {
+               Train &train = server.layout.get_train(pkt.address);
+               if(pkt.route.empty())
+                       train.set_route(0);
+               else
+               {
+                       Route &route = server.layout.get_route(pkt.route);
+                       train.set_route(&route);
+               }
+       }
+       catch(const Exception &e)
+       {
+               error(e.what());
+       }
+}
+
+void Server::Connection::error(const string &msg)
+{
+       ErrorPacket pkt;
+       pkt.message = msg;
+       comm.send(pkt);
 }
 
-} // namespace Marklin
+} // namespace R2C2