]> git.tdb.fi Git - r2c2.git/blob - source/engineer/trainpanel.cpp
Add a UI for editing train timetables
[r2c2.git] / source / engineer / trainpanel.cpp
1 /* $Id$
2
3 This file is part of the MSP Märklin suite
4 Copyright © 2006-2010  Mikkosoft Productions, Mikko Rasa
5 Distributed under the GPL
6 */
7
8 #include <cmath>
9 #include <msp/gltk/button.h>
10 #include <msp/strings/formatter.h>
11 #include "libmarklin/locotype.h"
12 #include "libmarklin/timetable.h"
13 #include "engineer.h"
14 #include "routeselect.h"
15 #include "timetabledialog.h"
16 #include "trainpanel.h"
17 #include "trainproperties.h"
18
19 using namespace std;
20 using namespace Msp;
21 using namespace Marklin;
22
23 TrainPanel::TrainPanel(Engineer &e, const GLtk::Resources &r, Train &t):
24         Widget(r),
25         Panel(r),
26         engineer(e),
27         train(t)
28 {
29         set_size(200, 170);
30
31         train.signal_control_changed.connect(sigc::mem_fun(this, &TrainPanel::train_control_changed));
32
33         add(*(lbl_addr=new GLtk::Label(res, format("%2d", train.get_address()))));
34         lbl_addr->set_style("digital");
35         lbl_addr->set_geometry(GLtk::Geometry(10, geom.h-34, 35, 24));
36
37         add(*(lbl_name=new GLtk::Label(res, train.get_name())));
38         lbl_name->set_style("digital");
39         lbl_name->set_geometry(GLtk::Geometry(45, geom.h-34, geom.w-55, 24));
40         train.signal_name_changed.connect(sigc::mem_fun(lbl_name, &GLtk::Label::set_text));
41
42         add(*(lbl_speed=new GLtk::Label(res, "  0")));
43         lbl_speed->set_style("digital");
44         lbl_speed->set_geometry(GLtk::Geometry(10, geom.h-58, 35, 24));
45
46         add(*(sld_speed=new GLtk::HSlider(res)));
47         sld_speed->set_geometry(GLtk::Geometry(50, geom.h-51, geom.w-80, 10));
48         sld_speed->set_range(0, 200);
49         sld_speed->set_step(5);
50         sld_speed->signal_value_changed.connect(sigc::mem_fun(this, &TrainPanel::speed_slider_changed));
51
52         add(*(tgl_forward=new GLtk::Toggle(res)));
53         tgl_forward->set_text("Fwd");
54         tgl_forward->set_geometry(GLtk::Geometry(geom.w-30, geom.h-59, 20, 27));
55         tgl_forward->set_value(true);
56         tgl_forward->signal_toggled.connect(sigc::mem_fun(this, &TrainPanel::forward_toggled));
57
58         const Route *route = train.get_route();
59         add(*(lbl_route=new GLtk::Label(res, (route ? route->get_name() : "Free run"))));
60         lbl_route->set_style("digital");
61         lbl_route->set_geometry(GLtk::Geometry(10, 58, geom.w-20, 24));
62         train.signal_route_changed.connect(sigc::mem_fun(this, &TrainPanel::train_route_changed));
63
64         add(*(lbl_status=new GLtk::Label(res, train.get_status())));
65         lbl_status->set_style("digital");
66         lbl_status->set_geometry(GLtk::Geometry(10, 34, geom.w-20, 24));
67         train.signal_status_changed.connect(sigc::mem_fun(this, &TrainPanel::train_status_changed));
68
69         const map<unsigned, string> &funcs = train.get_locomotive_type().get_functions();
70         unsigned x = 10;
71         for(map<unsigned, string>::const_iterator i=funcs.begin(); i!=funcs.end(); ++i, x+=36)
72         {
73                 string fname = i->second;
74                 fname[0] = toupper(fname[0]);
75                 GLtk::Toggle *tgl;
76                 add(*(tgl=new GLtk::Toggle(res)));
77                 tgl->set_text(fname);
78                 tgl->set_geometry(GLtk::Geometry(x, geom.h-85, 36, 27));
79                 tgl->set_value(train.get_function(i->first));
80                 tgl->signal_toggled.connect(sigc::bind(sigc::mem_fun(this, &TrainPanel::func_toggled), i->first));
81
82                 tgl_funcs[i->first] = tgl;
83         }
84         train.signal_function_changed.connect(sigc::mem_fun(this, &TrainPanel::train_function_changed));
85
86         GLtk::Button *btn;
87
88         add(*(btn=new GLtk::Button(res, "Edit")));
89         btn->set_geometry(GLtk::Geometry(geom.w-46, 10, 36, 24));
90         btn->signal_clicked.connect(sigc::mem_fun(this, &TrainPanel::edit_clicked));
91
92         add(*(btn=new GLtk::Button(res, "Place")));
93         btn->set_geometry(GLtk::Geometry(geom.w-82, 10, 36, 24));
94         btn->signal_clicked.connect(sigc::mem_fun(this, &TrainPanel::place_clicked));
95
96         add(*(btn=new GLtk::Button(res, "GoTo")));
97         btn->set_geometry(GLtk::Geometry(geom.w-118, 10, 36, 24));
98         btn->signal_clicked.connect(sigc::mem_fun(this, &TrainPanel::goto_clicked));
99
100         add(*(btn=new GLtk::Button(res, "Route")));
101         btn->set_geometry(GLtk::Geometry(geom.w-154, 10, 36, 24));
102         btn->signal_clicked.connect(sigc::mem_fun(this, &TrainPanel::route_clicked));
103
104         add(*(btn=new GLtk::Button(res, "TTbl")));
105         btn->set_geometry(GLtk::Geometry(geom.w-190, 10, 36, 24));
106         btn->signal_clicked.connect(sigc::mem_fun(this, &TrainPanel::timetable_clicked));
107 }
108
109 void TrainPanel::speed_slider_changed(double value)
110 {
111         float speed = value/3.6*engineer.get_layout().get_catalogue().get_scale();
112         if(!tgl_forward->get_value())
113                 speed = -speed;
114         train.set_control("speed", speed);
115 }
116
117 void TrainPanel::train_control_changed(const string &control, float value)
118 {
119         if(control=="speed")
120         {
121                 float speed = abs(value)/engineer.get_layout().get_catalogue().get_scale()*3.6;
122                 sld_speed->set_value(speed);
123                 lbl_speed->set_text(format("%3.0f", speed));
124                 if(value)
125                         tgl_forward->set_value(value>0);
126         }
127 }
128
129 void TrainPanel::train_function_changed(unsigned func, bool value)
130 {
131         map<unsigned, GLtk::Toggle *>::iterator i = tgl_funcs.find(func);
132         if(i!=tgl_funcs.end())
133                 i->second->set_value(value);
134 }
135
136 void TrainPanel::train_route_changed(const Route *r)
137 {
138         if(r)
139                 lbl_route->set_text(r->get_name());
140         else
141                 lbl_route->set_text("Free run");
142 }
143
144 void TrainPanel::train_status_changed(const string &s)
145 {
146         lbl_status->set_text(s);
147 }
148
149 void TrainPanel::place_clicked()
150 {
151         engineer.pick(true);
152         pick_conn = engineer.signal_pick_done.connect(sigc::mem_fun(this, &TrainPanel::place));
153 }
154
155 void TrainPanel::edit_clicked()
156 {
157         TrainProperties *dialog = new TrainProperties(engineer, res, &train);
158         engineer.get_root().add(*dialog);
159         dialog->set_position(geom.x+geom.w, geom.y+geom.h-dialog->get_geometry().h);
160         dialog->set_visible(true);
161 }
162
163 void TrainPanel::route_clicked()
164 {
165         RouteSelect *dialog = new RouteSelect(engineer, res, train);
166         engineer.get_root().add(*dialog);
167         dialog->set_position(geom.x+geom.w, geom.y+geom.h-dialog->get_geometry().h);
168         dialog->set_visible(true);
169 }
170
171 void TrainPanel::goto_clicked()
172 {
173         engineer.pick(false);
174         pick_conn = engineer.signal_pick_done.connect(sigc::mem_fun(this, &TrainPanel::go_to));
175 }
176
177 void TrainPanel::timetable_clicked()
178 {
179         Timetable *timetable = train.get_timetable();
180         if(!timetable)
181         {
182                 timetable = new Timetable(train);
183                 train.set_timetable(timetable);
184         }
185
186         TimetableDialog *dialog = new TimetableDialog(res, *timetable);
187         engineer.get_root().add(*dialog);
188         dialog->set_position(geom.x+geom.w, geom.y+geom.h-dialog->get_geometry().h);
189 }
190
191 void TrainPanel::forward_toggled(bool /*value*/)
192 {
193         train.set_control("speed", 0);
194 }
195
196 void TrainPanel::func_toggled(bool value, unsigned func)
197 {
198         train.set_function(func, value);
199 }
200
201 void TrainPanel::place(Track *track, unsigned ep)
202 {
203         pick_conn.disconnect();
204
205         Block &block = engineer.get_layout().get_block_by_track(*track);
206
207         while(1)
208         {
209                 const vector<Block::Endpoint> &eps = block.get_endpoints();
210                 bool ok = false;
211                 for(unsigned i=0; (!ok && i<eps.size()); ++i)
212                         if(eps[i].track==track && eps[i].track_ep==ep)
213                         {
214                                 train.place(block, i);
215                                 ok = true;
216                         }
217
218                 if(ok)
219                         break;
220                 else
221                 {
222                         Track *next = track->get_links()[ep];
223                         ep = next->traverse(next->get_endpoint_by_link(*track), 0);
224                         track = next;
225                         if(!block.get_tracks().count(track))
226                                 break;
227                 }
228         }
229 }
230
231 void TrainPanel::go_to(Track *track, unsigned)
232 {
233         pick_conn.disconnect();
234
235         try
236         {
237                 train.go_to(*track);
238         }
239         catch(const Exception &e)
240         {
241                 engineer.set_status(e.what());
242         }
243 }