X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fengineer%2Ftrainproperties.cpp;h=9686dee485527574cd2b1f78b3c111f3b46a1f09;hb=1ff06c5bc46a677fa389ef86c6b26664368f1653;hp=be3ba8daa410091746b9efb3cde1f7407b30aa1b;hpb=38fb8d56efde037a71c46a58bda314655e68ab6c;p=r2c2.git diff --git a/source/engineer/trainproperties.cpp b/source/engineer/trainproperties.cpp index be3ba8d..9686dee 100644 --- a/source/engineer/trainproperties.cpp +++ b/source/engineer/trainproperties.cpp @@ -1,67 +1,182 @@ /* $Id$ -This file is part of the MSP Märklin suite -Copyright © 2006-2008 Mikkosoft Productions, Mikko Rasa +This file is part of R²C² +Copyright © 2006-2010 Mikkosoft Productions, Mikko Rasa Distributed under the GPL */ -#include +#include #include #include +#include "libr2c2/driver.h" +#include "libr2c2/vehicle.h" +#include "libr2c2/vehicletype.h" #include "engineer.h" #include "trainproperties.h" using namespace std; using namespace Msp; -using namespace Marklin; +using namespace R2C2; -TrainProperties::TrainProperties(Engineer &e, GLtk::Resources &r, Train *t): - Panel(r), +TrainProperties::TrainProperties(Engineer &e, Train *t): engineer(e), train(t) { - set_size(200, 95); + set_size(250, 305); - add(*(ent_addr=new GLtk::Entry(res))); - ent_addr->set_geometry(GLtk::Geometry(10, geom.h-30, 40, 20)); + GLtk::Label *label; + add(*(label = new GLtk::Label("Train properties"))); + label->set_geometry(GLtk::Geometry(10, geom.h-25, geom.w-20, 20)); - add(*(drp_type=new GLtk::Dropdown(res))); - drp_type->set_geometry(GLtk::Geometry(60, geom.h-30, geom.w-70, 20)); + add(*(ent_addr = new GLtk::Entry)); + ent_addr->set_geometry(GLtk::Geometry(10, geom.h-50, 40, 20)); + + add(*(drp_protocol = new GLtk::Dropdown)); + drp_protocol->set_geometry(GLtk::Geometry(60, geom.h-50, 60, 20)); + for(unsigned i=0;; ++i) + { + if(const char *proto = engineer.get_layout().get_driver().enumerate_protocols(i)) + drp_protocol->append(proto); + else + break; + } + drp_protocol->set_selected_index(0); + + add(*(drp_type = new GLtk::Dropdown)); + drp_type->set_geometry(GLtk::Geometry(130, geom.h-50, geom.w-140, 20)); + + const Catalogue::VehicleMap &vehs = engineer.get_catalogue().get_vehicles(); + unsigned n = 0; + for(Catalogue::VehicleMap::const_iterator i=vehs.begin(); i!=vehs.end(); ++i) + { + if(!i->second->is_locomotive()) + continue; - const map &locos = engineer.get_catalogue().get_locomotives(); - for(map::const_iterator i=locos.begin(); i!=locos.end(); ++i) drp_type->append(format("%d %s", i->second->get_article_number(), i->second->get_name())); + if(train && i->second==&train->get_locomotive_type()) + drp_type->set_selected_index(n); + + ++n; + } + + add(*(ent_name = new GLtk::Entry)); + ent_name->set_geometry(GLtk::Geometry(10, geom.h-75, geom.w-20, 20)); - add(*(ent_name=new GLtk::Entry(res, "Train"))); - ent_name->set_geometry(GLtk::Geometry(10, geom.h-55, geom.w-20, 20)); + add(*(drp_priority = new GLtk::Dropdown)); + drp_priority->set_geometry(GLtk::Geometry(10, geom.h-100, geom.w-20, 20)); + drp_priority->append("Standard freight"); + drp_priority->append("Express freight"); + drp_priority->append("Unspecified"); + drp_priority->append("Standard passenger"); + drp_priority->append("Express passenger"); + + add(*(lst_vehicles = new GLtk::List)); + lst_vehicles->set_geometry(GLtk::Geometry(10, 100, geom.w-20, geom.h-205)); GLtk::Button *btn; - add(*(btn=new GLtk::Button(res, "OK"))); - btn->set_style("green"); - btn->set_geometry(GLtk::Geometry(geom.w-40, 10, 30, 25)); - btn->signal_clicked.connect(sigc::mem_fun(this, &TrainProperties::ok_clicked)); + add(*(btn = new GLtk::Button("Rem"))); + btn->set_geometry(GLtk::Geometry(10, 70, 40, 25)); + btn->signal_clicked.connect(sigc::mem_fun(this, &TrainProperties::remove_vehicle_clicked)); + + add(*(drp_new_vehicle = new GLtk::Dropdown)); + drp_new_vehicle->set_geometry(GLtk::Geometry(10, 45, geom.w-20, 20)); + drp_new_vehicle->append("(new vehicle)"); + drp_new_vehicle->set_selected_index(0); + for(Catalogue::VehicleMap::const_iterator i=vehs.begin(); i!=vehs.end(); ++i) + { + if(i->second->is_locomotive()) + continue; + + drp_new_vehicle->append(format("%d %s", i->second->get_article_number(), i->second->get_name())); + } + drp_new_vehicle->signal_item_selected.connect(sigc::mem_fun(this, &TrainProperties::new_vehicle_selected)); - add(*(btn=new GLtk::Button(res, "Cncl"))); - btn->set_style("red"); - btn->set_geometry(GLtk::Geometry(geom.w-80, 10, 30, 25)); - btn->signal_clicked.connect(sigc::mem_fun(this, &TrainProperties::cancel_clicked)); + if(train) + { + ent_addr->set_text(lexical_cast(train->get_address())); + ent_name->set_text(train->get_name()); + drp_priority->set_selected_index(train->get_priority()+2); + + unsigned n_vehicles = train->get_n_vehicles(); + for(unsigned i=1; iget_vehicle(i).get_type(); + lst_vehicles->append(format("%d %s", type.get_article_number(), type.get_name())); + } + } + else + { + ent_name->set_text(format("Train %d", engineer.get_layout().get_trains().size()+1)); + drp_priority->set_selected_index(2); + } } -void TrainProperties::ok_clicked() +void TrainProperties::on_ok_clicked() { if(!train) { - const map &locos = engineer.get_catalogue().get_locomotives(); - map::const_iterator i = locos.begin(); - advance(i, drp_type->get_selected_index()); - train = engineer.add_train(*i->second, lexical_cast(ent_addr->get_text())); + const VehicleType &type = get_vehicle_type(drp_type->get_selected_index(), true); + unsigned addr = lexical_cast(ent_addr->get_text()); + train = new Train(engineer.get_layout(), type, addr, drp_protocol->get_selected()); } train->set_name(ent_name->get_text()); - signal_ok.emit(); + train->set_priority(drp_priority->get_selected_index()-2); + + // The locomotive is vehicle 0 so we need to add 1 + for(set::const_iterator i=rem_vehicles.end(); i!=rem_vehicles.begin();) + train->remove_vehicle(*--i+1); + for(vector::const_iterator i=add_vehicles.begin(); i!=add_vehicles.end(); ++i) + train->add_vehicle(**i); } -void TrainProperties::cancel_clicked() +void TrainProperties::new_vehicle_selected(unsigned n, const string &) { + if(n==0) + return; + + const VehicleType &type = get_vehicle_type(n-1, false); + add_vehicles.push_back(&type); + lst_vehicles->append(format("%d %s", type.get_article_number(), type.get_name())); + + drp_new_vehicle->set_selected_index(0); +} + +void TrainProperties::remove_vehicle_clicked() +{ + int selected = lst_vehicles->get_selected_index(); + if(selected<0) + return; + + lst_vehicles->remove(selected); + + unsigned n_vehicles = (train ? train->get_n_vehicles()-rem_vehicles.size() : 0); + if(static_cast(selected)>=n_vehicles) + add_vehicles.erase(add_vehicles.begin()+(selected-n_vehicles)); + else + { + for(set::const_iterator i=rem_vehicles.begin(); i!=rem_vehicles.end(); ++i) + if(*i<=static_cast(selected)) + ++selected; + rem_vehicles.insert(selected); + } +} + +const VehicleType &TrainProperties::get_vehicle_type(unsigned n, bool loco) +{ + const Catalogue::VehicleMap &vehs = engineer.get_catalogue().get_vehicles(); + Catalogue::VehicleMap::const_iterator i = vehs.begin(); + while(i!=vehs.end()) + { + if(i->second->is_locomotive()==loco) + { + if(!n) + return *i->second; + --n; + } + ++i; + } + + throw InvalidParameterValue("Vehicle type index out of range"); }