source/engineer/trainproperties.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 <msp/gltk/label.h>
   9 #include <msp/strings/formatter.h>
  10 #include <msp/strings/lexicalcast.h>
  11 #include "libmarklin/vehicle.h"
  12 #include "libmarklin/vehicletype.h"
  13 #include "engineer.h"
  14 #include "trainproperties.h"
  15
  16 using namespace std;
  17 using namespace Msp;
  18 using namespace Marklin;
  19
  20 TrainProperties::TrainProperties(Engineer &e, const GLtk::Resources &r, Train *t):
  21         Widget(r),
  22         Dialog(r),
  23         engineer(e),
  24         train(t)
  25 {
  26         set_size(200, 275);
  27
  28         GLtk::Label *label;
  29         add(*(label = new GLtk::Label(res, "Train properties")));
  30         label->set_geometry(GLtk::Geometry(10, geom.h-25, geom.w-20, 20));
  31
  32         add(*(ent_addr = new GLtk::Entry(res)));
  33         ent_addr->set_geometry(GLtk::Geometry(10, geom.h-50, 40, 20));
  34
  35         add(*(drp_type = new GLtk::Dropdown(res)));
  36         drp_type->set_geometry(GLtk::Geometry(60, geom.h-50, geom.w-70, 20));
  37
  38         const map<unsigned, VehicleType *> &vehs = engineer.get_catalogue().get_vehicles();
  39         unsigned n = 0;
  40         for(map<unsigned, VehicleType *>::const_iterator i=vehs.begin(); i!=vehs.end(); ++i, ++n)
  41         {
  42                 if(!i->second->is_locomotive())
  43                         continue;
  44
  45                 drp_type->append(format("%d %s", i->second->get_article_number(), i->second->get_name()));
  46                 if(train && i->second==&train->get_locomotive_type())
  47                         drp_type->set_selected_index(n);
  48         }
  49
  50         add(*(ent_name = new GLtk::Entry(res)));
  51         ent_name->set_geometry(GLtk::Geometry(10, geom.h-75, geom.w-20, 20));
  52
  53         add(*(drp_priority = new GLtk::Dropdown(res)));
  54         drp_priority->set_geometry(GLtk::Geometry(10, geom.h-100, geom.w-20, 20));
  55         drp_priority->append("Standard freight");
  56         drp_priority->append("Express freight");
  57         drp_priority->append("Unspecified");
  58         drp_priority->append("Standard passenger");
  59         drp_priority->append("Express passenger");
  60
  61         add(*(lst_vehicles = new GLtk::List(res)));
  62         lst_vehicles->set_geometry(GLtk::Geometry(10, geom.h-205, geom.w-20, 100));
  63
  64         add(*(drp_new_vehicle = new GLtk::Dropdown(res)));
  65         drp_new_vehicle->set_geometry(GLtk::Geometry(10, geom.h-230, geom.w-20, 20));
  66         drp_new_vehicle->append("(new vehicle)");
  67         drp_new_vehicle->set_selected_index(0);
  68         for(map<unsigned, VehicleType *>::const_iterator i=vehs.begin(); i!=vehs.end(); ++i)
  69         {
  70                 if(i->second->is_locomotive())
  71                         continue;
  72
  73                 drp_new_vehicle->append(format("%d %s", i->second->get_article_number(), i->second->get_name()));
  74         }
  75         drp_new_vehicle->signal_item_selected.connect(sigc::mem_fun(this, &TrainProperties::new_vehicle_selected));
  76
  77         if(train)
  78         {
  79                 ent_addr->set_text(lexical_cast(train->get_address()));
  80                 ent_name->set_text(train->get_name());
  81                 drp_priority->set_selected_index(train->get_priority()+2);
  82
  83                 unsigned n_vehicles = train->get_n_vehicles();
  84                 for(unsigned i=1; i<n_vehicles; ++i)
  85                 {
  86                         const VehicleType &type = train->get_vehicle(i).get_type();
  87                         lst_vehicles->append(format("%d %s", type.get_article_number(), type.get_name()));
  88                 }
  89         }
  90         else
  91         {
  92                 ent_name->set_text(format("Train %d", engineer.get_layout().get_trains().size()+1));
  93                 drp_priority->set_selected_index(2);
  94         }
  95 }
  96
  97 void TrainProperties::on_ok_clicked()
  98 {
  99         if(!train)
 100         {
 101                 const VehicleType &type = get_vehicle_type(drp_type->get_selected_index(), true);
 102                 unsigned addr = lexical_cast<unsigned>(ent_addr->get_text());
 103                 train = new Train(engineer.get_layout(), type, addr);
 104         }
 105
 106         train->set_name(ent_name->get_text());
 107         train->set_priority(drp_priority->get_selected_index()-2);
 108
 109         for(vector<const VehicleType *>::const_iterator i=add_vehicles.begin(); i!=add_vehicles.end(); ++i)
 110                 train->add_vehicle(**i);
 111 }
 112
 113 void TrainProperties::new_vehicle_selected(unsigned n, const string &)
 114 {
 115         if(n==0)
 116                 return;
 117
 118         const VehicleType &type = get_vehicle_type(n-1, false);
 119         add_vehicles.push_back(&type);
 120         lst_vehicles->append(format("%d %s", type.get_article_number(), type.get_name()));
 121
 122         drp_new_vehicle->set_selected_index(0);
 123 }
 124
 125 const VehicleType &TrainProperties::get_vehicle_type(unsigned n, bool loco)
 126 {
 127         const map<unsigned, VehicleType *> &vehs = engineer.get_catalogue().get_vehicles();
 128         map<unsigned, VehicleType *>::const_iterator i = vehs.begin();
 129         while(i!=vehs.end())
 130         {
 131                 if(i->second->is_locomotive()==loco)
 132                 {
 133                         if(!n)
 134                                 return *i->second;
 135                         --n;
 136                 }
 137                 ++i;
 138         }
 139
 140         throw InvalidParameterValue("Vehicle type index out of range");
 141 }