3 This file is part of R²C²
4 Copyright © 2009-2010 Mikkosoft Productions, Mikko Rasa
5 Distributed under the GPL
8 #include <msp/net/inet.h>
9 #include "libr2c2/route.h"
10 #include "libr2c2/train.h"
11 #include "libr2c2/vehicletype.h"
19 Server::Server(Layout &l):
21 listen_sock(Net::INET),
24 layout.signal_train_added.connect(sigc::mem_fun(this, &Server::train_added));
26 const map<unsigned, Train *> &trains = layout.get_trains();
27 for(map<unsigned, Train *>::const_iterator i=trains.begin(); i!=trains.end(); ++i)
28 train_added(*i->second);
30 listen_sock.listen(Net::InetAddr(0, 8315), 4);
31 listen_sock.signal_data_available.connect(sigc::mem_fun(this, &Server::incoming_connection));
34 void Server::use_event_dispatcher(IO::EventDispatcher &ed)
37 event_disp->add(listen_sock);
40 void Server::incoming_connection()
42 Net::StreamSocket *sock = listen_sock.accept();
44 event_disp->add(*sock);
45 connections.push_back(new Connection(*this, sock));
48 void Server::train_added(Train &train)
50 train.signal_control_changed.connect(sigc::bind<0>(sigc::mem_fun(this, &Server::train_control_changed), sigc::ref(train)));
51 train.signal_function_changed.connect(sigc::bind<0>(sigc::mem_fun(this, &Server::train_function_changed), sigc::ref(train)));
52 train.signal_route_changed.connect(sigc::bind<0>(sigc::mem_fun(this, &Server::train_route_changed), sigc::ref(train)));
53 train.signal_status_changed.connect(sigc::bind<0>(sigc::mem_fun(this, &Server::train_status_changed), sigc::ref(train)));
56 pkt.address = train.get_address();
57 pkt.loco_type = train.get_locomotive_type().get_article_number().str();
58 pkt.name = train.get_name();
62 void Server::train_control_changed(const Train &train, const string &control, float value)
64 TrainControlPacket pkt;
65 pkt.address = train.get_address();
66 pkt.control = control;
71 void Server::train_function_changed(const Train &train, unsigned, bool)
73 TrainFunctionPacket pkt;
74 pkt.address = train.get_address();
75 pkt.functions = train.get_functions();
79 void Server::train_route_changed(const Train &train, const Route *route)
82 pkt.address = train.get_address();
84 pkt.route = route->get_name();
88 void Server::train_status_changed(const Train &train, const string &status)
90 TrainStatusPacket pkt;
91 pkt.address = train.get_address();
97 void Server::send(const P &pkt)
99 for(vector<Connection *>::const_iterator i=connections.begin(); i!=connections.end(); ++i)
101 (*i)->comm.send(pkt);
105 Server::Connection::Connection(Server &s, Net::StreamSocket *o):
108 comm(*socket, server.proto, *this)
110 socket->signal_end_of_file.connect(sigc::mem_fun(this, &Connection::end_of_file));
111 comm.signal_handshake_done.connect(sigc::mem_fun(this, &Connection::handshake_done));
112 comm.initiate_handshake();
115 Server::Connection::~Connection()
120 void Server::Connection::handshake_done()
122 const set<Route *> &routes = server.layout.get_routes();
123 for(set<Route *>::const_iterator i=routes.begin(); i!=routes.end(); ++i)
126 pkt.name = (*i)->get_name();
130 const map<unsigned, Train *> &trains = server.layout.get_trains();
131 for(map<unsigned, Train *>::const_iterator i=trains.begin(); i!=trains.end(); ++i)
133 const Train &train = *i->second;
137 pkt.address = train.get_address();
138 pkt.loco_type = train.get_locomotive_type().get_article_number().str();
139 pkt.name = train.get_name();
142 for(unsigned j=0;; ++j)
144 const char *name = train.get_controller().enumerate_controls(j);
148 TrainControlPacket pkt;
149 pkt.address = train.get_address();
151 pkt.value = train.get_control(name);
155 TrainFunctionPacket pkt;
156 pkt.address = train.get_address();
157 pkt.functions = train.get_functions();
161 TrainStatusPacket pkt;
162 pkt.address = train.get_address();
163 pkt.status = train.get_status();
166 if(train.get_route())
168 TrainRoutePacket pkt;
169 pkt.address = train.get_address();
170 pkt.route = train.get_route()->get_name();
176 void Server::Connection::end_of_file()
182 void Server::Connection::receive(const TrainControlPacket &pkt)
186 Train &train = server.layout.get_train(pkt.address);
187 train.set_control(pkt.control, pkt.value);
189 catch(const Exception &e)
195 void Server::Connection::receive(const TrainFunctionPacket &pkt)
199 Train &train = server.layout.get_train(pkt.address);
200 for(unsigned i=0; i<9; ++i)
201 if(((pkt.functions^train.get_functions())>>i)&1)
202 train.set_function(i, (pkt.functions>>i)&1);
204 catch(const Exception &e)
210 void Server::Connection::receive(const TrainRoutePacket &pkt)
214 Train &train = server.layout.get_train(pkt.address);
215 if(pkt.route.empty())
219 Route &route = server.layout.get_route(pkt.route);
220 train.set_route(&route);
223 catch(const Exception &e)
229 void Server::Connection::error(const string &msg)