source/engineer/trainpanel.cpp

   1 #include <cmath>
   2 #include <msp/strings/format.h>
   3 #include "libr2c2/aicontrol.h"
   4 #include "libr2c2/timetable.h"
   5 #include "libr2c2/trackiter.h"
   6 #include "libr2c2/trainrouter.h"
   7 #include "libr2c2/vehicletype.h"
   8 #include "engineer.h"
   9 #include "routeselect.h"
  10 #include "timetabledialog.h"
  11 #include "trainpanel.h"
  12 #include "trainproperties.h"
  13 #include "trainview.h"
  14
  15 using namespace std;
  16 using namespace Msp;
  17 using namespace R2C2;
  18
  19 TrainPanel::TrainPanel(Engineer &e, Train &t):
  20         engineer(e),
  21         train(t),
  22         status(train),
  23         expanded(false)
  24 {
  25         set_size(200, 65);
  26
  27         AIControl *ai = new AIControl(train);
  28         ai->signal_event.connect(sigc::mem_fun(this, &TrainPanel::ai_event));
  29
  30         add(*(pnl_basic = new GLtk::Panel));
  31         pnl_basic->set_style("group");
  32         pnl_basic->set_geometry(GLtk::Geometry(0, geom.h-58, geom.w, 45));
  33
  34         pnl_basic->add(*(lbl_addr = new GLtk::Label(format("%2d", train.get_address()))));
  35         lbl_addr->set_style("digital");
  36         lbl_addr->set_geometry(GLtk::Geometry(10, 28, 35, 20));
  37
  38         pnl_basic->add(*(lbl_name = new GLtk::Label(train.get_name())));
  39         lbl_name->set_style("digital");
  40         lbl_name->set_geometry(GLtk::Geometry(50, 28, geom.w-77, 20));
  41         train.signal_name_changed.connect(sigc::mem_fun(lbl_name, &GLtk::Label::set_text));
  42
  43         pnl_basic->add(*(lbl_speed = new GLtk::Label("  0")));
  44         lbl_speed->set_style("digital");
  45         lbl_speed->set_geometry(GLtk::Geometry(10, 3, 35, 20));
  46
  47         pnl_basic->add(*(sld_speed = new GLtk::HSlider));
  48         sld_speed->set_geometry(GLtk::Geometry(50, 8, geom.w-80, 10));
  49         sld_speed->set_range(0, 200);
  50         sld_speed->set_step(5);
  51         sld_speed->signal_value_changed.connect(sigc::mem_fun(this, &TrainPanel::speed_slider_changed));
  52
  53         pnl_basic->add(*(tgl_forward = new GLtk::Toggle));
  54         tgl_forward->set_text("Fwd");
  55         tgl_forward->set_geometry(GLtk::Geometry(geom.w-30, 0, 20, 27));
  56         tgl_forward->set_value(true);
  57         tgl_forward->signal_toggled.connect(sigc::mem_fun(this, &TrainPanel::forward_toggled));
  58
  59         pnl_basic->add(*(btn_expand = new GLtk::Button));
  60         btn_expand->set_style("arrow_down");
  61         btn_expand->set_geometry(GLtk::Geometry(geom.w-22, 28, 12, 20));
  62         btn_expand->signal_clicked.connect(sigc::mem_fun(this, &TrainPanel::expand_clicked));
  63
  64         add(*(pnl_extra = new GLtk::Panel));
  65         pnl_extra->set_style("group");
  66         pnl_extra->set_geometry(GLtk::Geometry(0, 10, geom.w, 135));
  67         pnl_extra->set_visible(false);
  68
  69         TrainRouter *router = dynamic_cast<TrainRouter *>(train.get_tagged_ai("router"));
  70         if(!router)
  71         {
  72                 router = new TrainRouter(train);
  73                 router->set_tag("router");
  74         }
  75
  76         const Route *route = router->get_route();
  77         pnl_extra->add(*(lbl_route = new GLtk::Label((route ? route->get_name() : "Free run"))));
  78         lbl_route->set_style("digital");
  79         lbl_route->set_geometry(GLtk::Geometry(10, 85, geom.w-20, 20));
  80         router->signal_route_changed.connect(sigc::mem_fun(this, &TrainPanel::train_route_changed));
  81
  82         pnl_extra->add(*(lbl_status = new GLtk::Label(status.get_status())));
  83         lbl_status->set_style("digital");
  84         lbl_status->set_geometry(GLtk::Geometry(10, 60, geom.w-20, 20));
  85         status.signal_changed.connect(sigc::mem_fun(this, &TrainPanel::train_status_changed));
  86
  87         const VehicleType::FunctionMap &funcs = train.get_locomotive_type().get_functions();
  88         unsigned x = 10;
  89         for(VehicleType::FunctionMap::const_iterator i=funcs.begin(); i!=funcs.end(); ++i, x+=36)
  90         {
  91                 string fname = i->second;
  92                 fname[0] = toupper(fname[0]);
  93                 GLtk::Toggle *tgl;
  94                 pnl_extra->add(*(tgl = new GLtk::Toggle));
  95                 tgl->set_text(fname);
  96                 tgl->set_geometry(GLtk::Geometry(x, 108, 36, 27));
  97                 tgl->set_value(train.get_function(i->first));
  98                 tgl->signal_toggled.connect(sigc::bind(sigc::mem_fun(this, &TrainPanel::func_toggled), i->first));
  99
 100                 tgl_funcs[i->first] = tgl;
 101         }
 102         train.signal_function_changed.connect(sigc::mem_fun(this, &TrainPanel::train_function_changed));
 103
 104         GLtk::Button *btn;
 105
 106         pnl_extra->add(*(btn = new GLtk::Button("Edit")));
 107         btn->set_geometry(GLtk::Geometry(10, 30, 36, 25));
 108         btn->signal_clicked.connect(sigc::mem_fun(this, &TrainPanel::edit_clicked));
 109
 110         pnl_extra->add(*(btn = new GLtk::Button("Place")));
 111         btn->set_geometry(GLtk::Geometry(10, 0, 36, 25));
 112         btn->signal_clicked.connect(sigc::mem_fun(this, &TrainPanel::place_clicked));
 113
 114         pnl_extra->add(*(btn = new GLtk::Button("Take")));
 115         btn->set_geometry(GLtk::Geometry(46, 0, 36, 25));
 116         btn->signal_clicked.connect(sigc::mem_fun(this, &TrainPanel::take_clicked));
 117
 118         pnl_extra->add(*(btn = new GLtk::Button("GoTo")));
 119         btn->set_geometry(GLtk::Geometry(100, 0, 36, 25));
 120         btn->signal_clicked.connect(sigc::mem_fun(this, &TrainPanel::goto_clicked));
 121
 122         pnl_extra->add(*(btn = new GLtk::Button("Route")));
 123         btn->set_geometry(GLtk::Geometry(100, 30, 36, 25));
 124         btn->signal_clicked.connect(sigc::mem_fun(this, &TrainPanel::route_clicked));
 125
 126         pnl_extra->add(*(btn = new GLtk::Button("TTbl")));
 127         btn->set_geometry(GLtk::Geometry(46, 30, 36, 25));
 128         btn->signal_clicked.connect(sigc::mem_fun(this, &TrainPanel::timetable_clicked));
 129
 130         pnl_extra->add(*(btn = new GLtk::Button("View")));
 131         btn->set_geometry(GLtk::Geometry(geom.w-46, 30, 36, 25));
 132         btn->signal_clicked.connect(sigc::mem_fun(this, &TrainPanel::view_clicked));
 133 }
 134
 135 void TrainPanel::expand(bool e)
 136 {
 137         expanded = e;
 138         pnl_extra->set_visible(expanded);
 139         if(expanded)
 140         {
 141                 set_size(geom.w, 205);
 142                 btn_expand->set_style("arrow_up");
 143         }
 144         else
 145         {
 146                 set_size(geom.w, 65);
 147                 btn_expand->set_style("arrow_down");
 148         }
 149         pnl_basic->set_geometry(GLtk::Geometry(0, geom.h-58, geom.w, 45));
 150         engineer.rearrange_panels();
 151 }
 152
 153 void TrainPanel::ai_event(const TrainAI::Message &msg)
 154 {
 155         if(msg.type=="target-speed-changed")
 156         {
 157                 float speed = msg.value.value<float>()/engineer.get_layout().get_catalogue().get_scale()*3.6;
 158                 sld_speed->set_value(speed);
 159                 lbl_speed->set_text(format("%3.0f", speed));
 160         }
 161         else if(msg.type=="reverse-changed")
 162                 tgl_forward->set_value(!msg.value.value<bool>());
 163 }
 164
 165 void TrainPanel::train_function_changed(unsigned func, bool value)
 166 {
 167         map<unsigned, GLtk::Toggle *>::iterator i = tgl_funcs.find(func);
 168         if(i!=tgl_funcs.end())
 169                 i->second->set_value(value);
 170 }
 171
 172 void TrainPanel::train_route_changed(const Route *r)
 173 {
 174         if(r)
 175                 lbl_route->set_text(r->get_name());
 176         else
 177                 lbl_route->set_text("Free run");
 178 }
 179
 180 void TrainPanel::train_status_changed(const string &s)
 181 {
 182         lbl_status->set_text(s);
 183 }
 184
 185 void TrainPanel::place_clicked()
 186 {
 187         engineer.pick(true);
 188         pick_conn = engineer.signal_pick_done.connect(sigc::mem_fun(this, &TrainPanel::place));
 189 }
 190
 191 void TrainPanel::take_clicked()
 192 {
 193         train.unplace();
 194 }
 195
 196 void TrainPanel::edit_clicked()
 197 {
 198         TrainProperties *dialog = new TrainProperties(engineer, &train);
 199         engineer.get_root().add(*dialog);
 200         dialog->set_position(geom.x+geom.w, geom.y+geom.h-dialog->get_geometry().h);
 201 }
 202
 203 void TrainPanel::route_clicked()
 204 {
 205         RouteSelect *dialog = new RouteSelect(engineer, train);
 206         engineer.get_root().add(*dialog);
 207         dialog->set_position(geom.x+geom.w, geom.y+geom.h-dialog->get_geometry().h);
 208 }
 209
 210 void TrainPanel::goto_clicked()
 211 {
 212         engineer.pick(false);
 213         pick_conn = engineer.signal_pick_done.connect(sigc::mem_fun(this, &TrainPanel::go_to));
 214 }
 215
 216 void TrainPanel::timetable_clicked()
 217 {
 218         Timetable *timetable = dynamic_cast<Timetable *>(train.get_tagged_ai("timetable"));
 219         if(!timetable)
 220         {
 221                 timetable = new Timetable(train);
 222                 timetable->set_tag("timetable");
 223         }
 224
 225         TimetableDialog *dialog = new TimetableDialog(*timetable);
 226         engineer.get_root().add(*dialog);
 227         dialog->set_position(geom.x+geom.w, geom.y+geom.h-dialog->get_geometry().h);
 228 }
 229
 230 void TrainPanel::view_clicked()
 231 {
 232         TrainView *dialog = new TrainView(engineer, train);
 233         engineer.get_root().add(*dialog);
 234         dialog->set_position(geom.x+geom.w, geom.y+geom.h-dialog->get_geometry().h);
 235 }
 236
 237 void TrainPanel::expand_clicked()
 238 {
 239         expand(!expanded);
 240 }
 241
 242 void TrainPanel::speed_slider_changed(double value)
 243 {
 244         float speed = value/3.6*engineer.get_layout().get_catalogue().get_scale();
 245         train.ai_message(TrainAI::Message("set-target-speed", speed));
 246 }
 247
 248 void TrainPanel::forward_toggled(bool value)
 249 {
 250         train.ai_message(TrainAI::Message("set-reverse", !value));
 251 }
 252
 253 void TrainPanel::func_toggled(bool value, unsigned func)
 254 {
 255         train.set_function(func, value);
 256 }
 257
 258 void TrainPanel::place(Track *track, unsigned ep)
 259 {
 260         pick_conn.disconnect();
 261
 262         Block &block = track->get_block();
 263         TrackIter iter(track, ep);
 264
 265         while(block.has_track(*iter))
 266         {
 267                 const vector<Block::Endpoint> &eps = block.get_endpoints();
 268                 bool ok = false;
 269                 for(unsigned i=0; (!ok && i<eps.size()); ++i)
 270                         if(eps[i].track==iter.track() && eps[i].track_ep==iter.entry())
 271                         {
 272                                 train.place(block, i);
 273                                 ok = true;
 274                         }
 275
 276                 if(ok)
 277                         break;
 278
 279                 iter = iter.flip().reverse();
 280         }
 281 }
 282
 283 void TrainPanel::go_to(Track *track, unsigned)
 284 {
 285         pick_conn.disconnect();
 286
 287         TrainRouter *router = dynamic_cast<TrainRouter *>(train.get_tagged_ai("router"));
 288         if(!router || !router->go_to(*track))
 289                 engineer.set_status("Could not set route");
 290 }