]> git.tdb.fi Git - r2c2.git/blob - source/engineer/newtraindialog.cpp
Reflect the current state of functions in a newly opened train dialog
[r2c2.git] / source / engineer / newtraindialog.cpp
1 #include <msp/core/maputils.h>
2 #include <msp/strings/format.h>
3 #include "libr2c2/catalogue.h"
4 #include "libr2c2/driver.h"
5 #include "libr2c2/layout.h"
6 #include "libr2c2/vehicletype.h"
7 #include "engineer.h"
8 #include "newtraindialog.h"
9 #include "traindialog.h"
10
11 using namespace std;
12 using namespace Msp;
13 using namespace R2C2;
14
15 static string vehicle_type_name(const VehicleType *const &vehtype)
16 {
17         return format("%s %s", vehtype->get_article_number(), vehtype->get_name());
18 }
19
20 NewTrainDialog::NewTrainDialog(Engineer &e):
21         engineer(e),
22         loco_types(vehicle_type_name)
23 {
24         Loader::WidgetMap widgets;
25         DataFile::load(*this, "data/newtraindialog.ui", widgets);
26
27         drp_protocol = dynamic_cast<GLtk::Dropdown *>(get_item(widgets, "drp_protocol"));
28         drp_protocol->set_data(protocols);
29         ent_address = dynamic_cast<GLtk::Entry *>(get_item(widgets, "ent_address"));
30         ent_name = dynamic_cast<GLtk::Entry *>(get_item(widgets, "ent_name"));
31         drp_loco_type = dynamic_cast<GLtk::Dropdown *>(get_item(widgets, "drp_loco_type"));
32         drp_loco_type->set_data(loco_types);
33
34         const Driver &driver = engineer.get_layout().get_driver();
35         for(unsigned i=0;; ++i)
36         {
37                 if(const char *proto = driver.enumerate_protocols(i))
38                         protocols.append(proto);
39                 else
40                         break;
41         }
42
43         drp_protocol->set_selected_index(0);
44
45         list<VehicleType *> veh_types = engineer.get_layout().get_catalogue().get_list<VehicleType>();
46         for(list<VehicleType *>::iterator i=veh_types.begin(); i!=veh_types.end(); ++i)
47                 if((*i)->is_locomotive())
48                         loco_types.append(*i);
49 }
50
51 void NewTrainDialog::prefill(const Driver::DetectedLocomotive &loco)
52 {
53         drp_protocol->set_selected_index(protocols.find(loco.protocol));
54         ent_address->set_text(lexical_cast<string>(loco.address));
55         ent_name->set_text(loco.name);
56 }
57
58 void NewTrainDialog::on_response(int code)
59 {
60         if(code)
61         {
62                 int index = drp_protocol->get_selected_index();
63                 if(index<0)
64                         return;
65                 string protocol = protocols.get(index);
66
67                 unsigned address = lexical_cast<unsigned>(ent_address->get_text());
68
69                 index = drp_loco_type->get_selected_index();
70                 if(index<0)
71                         return;
72                 const VehicleType &type = *loco_types.get(index);
73
74                 Train *train = new Train(engineer.get_layout(), type, address, protocol);
75                 train->set_name(ent_name->get_text());
76
77                 TrainDialog *dlg = new TrainDialog(engineer, *train);
78                 find_ancestor<GLtk::Root>()->add(*dlg);
79                 dlg->autosize();
80         }
81 }