1 #include <msp/gltk/label.h>
2 #include <msp/strings/format.h>
3 #include <msp/strings/lexicalcast.h>
4 #include "libr2c2/driver.h"
5 #include "libr2c2/trainrouter.h"
6 #include "libr2c2/vehicle.h"
7 #include "libr2c2/vehicletype.h"
9 #include "trainproperties.h"
15 TrainProperties::TrainProperties(Engineer &e, Train *t):
22 add(*(label = new GLtk::Label("Train properties")));
23 label->set_geometry(GLtk::Geometry(10, geom.h-25, geom.w-20, 20));
25 add(*(ent_addr = new GLtk::Entry));
26 ent_addr->set_geometry(GLtk::Geometry(10, geom.h-50, 40, 20));
28 add(*(drp_protocol = new GLtk::Dropdown));
29 drp_protocol->set_geometry(GLtk::Geometry(60, geom.h-50, 60, 20));
30 for(unsigned i=0;; ++i)
32 if(const char *proto = engineer.get_layout().get_driver().enumerate_protocols(i))
33 drp_protocol->append(proto);
37 drp_protocol->set_selected_index(0);
39 add(*(drp_type = new GLtk::Dropdown));
40 drp_type->set_geometry(GLtk::Geometry(130, geom.h-50, geom.w-140, 20));
42 const Catalogue::VehicleMap &vehs = engineer.get_catalogue().get_vehicles();
44 for(Catalogue::VehicleMap::const_iterator i=vehs.begin(); i!=vehs.end(); ++i)
46 if(!i->second->is_locomotive())
49 drp_type->append(format("%d %s", i->second->get_article_number(), i->second->get_name()));
50 if(train && i->second==&train->get_locomotive_type())
51 drp_type->set_selected_index(n);
56 add(*(ent_name = new GLtk::Entry));
57 ent_name->set_geometry(GLtk::Geometry(10, geom.h-75, geom.w-20, 20));
59 add(*(drp_priority = new GLtk::Dropdown));
60 drp_priority->set_geometry(GLtk::Geometry(10, geom.h-100, geom.w-20, 20));
61 drp_priority->append("Standard freight");
62 drp_priority->append("Express freight");
63 drp_priority->append("Unspecified");
64 drp_priority->append("Standard passenger");
65 drp_priority->append("Express passenger");
67 add(*(lst_vehicles = new GLtk::List));
68 lst_vehicles->set_geometry(GLtk::Geometry(10, 100, geom.w-20, geom.h-205));
72 add(*(btn = new GLtk::Button("Rem")));
73 btn->set_geometry(GLtk::Geometry(10, 70, 40, 25));
74 btn->signal_clicked.connect(sigc::mem_fun(this, &TrainProperties::remove_vehicle_clicked));
76 add(*(drp_new_vehicle = new GLtk::Dropdown));
77 drp_new_vehicle->set_geometry(GLtk::Geometry(10, 45, geom.w-20, 20));
78 drp_new_vehicle->append("(new vehicle)");
79 drp_new_vehicle->set_selected_index(0);
80 for(Catalogue::VehicleMap::const_iterator i=vehs.begin(); i!=vehs.end(); ++i)
82 if(i->second->is_locomotive())
85 drp_new_vehicle->append(format("%d %s", i->second->get_article_number(), i->second->get_name()));
87 drp_new_vehicle->signal_item_selected.connect(sigc::mem_fun(this, &TrainProperties::new_vehicle_selected));
91 ent_addr->set_text(lexical_cast<string>(train->get_address()));
92 ent_name->set_text(train->get_name());
93 if(TrainRouter *router = train->get_ai_of_type<TrainRouter>())
94 drp_priority->set_selected_index(router->get_priority()+2);
96 unsigned n_vehicles = train->get_n_vehicles();
97 for(unsigned i=1; i<n_vehicles; ++i)
99 const VehicleType &type = train->get_vehicle(i).get_type();
100 lst_vehicles->append(format("%d %s", type.get_article_number(), type.get_name()));
105 ent_name->set_text(format("Train %d", engineer.get_layout().get_trains().size()+1));
106 drp_priority->set_selected_index(2);
110 void TrainProperties::on_ok_clicked()
114 const VehicleType &type = get_vehicle_type(drp_type->get_selected_index(), true);
115 unsigned addr = lexical_cast<unsigned>(ent_addr->get_text());
116 train = new Train(engineer.get_layout(), type, addr, drp_protocol->get_selected());
119 train->set_name(ent_name->get_text());
120 if(TrainRouter *router = train->get_ai_of_type<TrainRouter>())
121 router->set_priority(drp_priority->get_selected_index()-2);
123 // The locomotive is vehicle 0 so we need to add 1
124 for(set<unsigned>::const_iterator i=rem_vehicles.end(); i!=rem_vehicles.begin();)
125 train->remove_vehicle(*--i+1);
126 for(vector<const VehicleType *>::const_iterator i=add_vehicles.begin(); i!=add_vehicles.end(); ++i)
127 train->add_vehicle(**i);
130 void TrainProperties::new_vehicle_selected(unsigned n, const string &)
135 const VehicleType &type = get_vehicle_type(n-1, false);
136 add_vehicles.push_back(&type);
137 lst_vehicles->append(format("%d %s", type.get_article_number(), type.get_name()));
139 drp_new_vehicle->set_selected_index(0);
142 void TrainProperties::remove_vehicle_clicked()
144 int selected = lst_vehicles->get_selected_index();
148 lst_vehicles->remove(selected);
150 unsigned n_vehicles = (train ? train->get_n_vehicles()-rem_vehicles.size() : 0);
151 if(static_cast<unsigned>(selected)>=n_vehicles)
152 add_vehicles.erase(add_vehicles.begin()+(selected-n_vehicles));
155 for(set<unsigned>::const_iterator i=rem_vehicles.begin(); i!=rem_vehicles.end(); ++i)
156 if(*i<=static_cast<unsigned>(selected))
158 rem_vehicles.insert(selected);
162 const VehicleType &TrainProperties::get_vehicle_type(unsigned n, bool loco)
164 const Catalogue::VehicleMap &vehs = engineer.get_catalogue().get_vehicles();
165 Catalogue::VehicleMap::const_iterator i = vehs.begin();
168 if(i->second->is_locomotive()==loco)
177 throw out_of_range("TrainProperties::get_vehicle_type");