Fix a segfault in removing vehicles when creating a new train

[r2c2.git] / source / engineer / trainproperties.h

diff --git a/source/engineer/trainproperties.h b/source/engineer/trainproperties.h
index 3c1a372fe606fed9801ce1ea069fbebe38903f10..465154d2948789734672233ad7b0f39d8766978f 100644 (file)
--- a/source/engineer/trainproperties.h
+++ b/source/engineer/trainproperties.h
@@ -21,14 +21,22 @@ private:
        Engineer &engineer;
        Marklin::Train *train;
        Msp::GLtk::Entry *ent_addr;
+       Msp::GLtk::Dropdown *drp_protocol;
        Msp::GLtk::Dropdown *drp_type;
        Msp::GLtk::Entry *ent_name;
        Msp::GLtk::Dropdown *drp_priority;
+       Msp::GLtk::List *lst_vehicles;
+       Msp::GLtk::Dropdown *drp_new_vehicle;
+       std::vector<const Marklin::VehicleType *> add_vehicles;
+       std::set<unsigned> rem_vehicles;
 
 public:
-       TrainProperties(Engineer &, const Msp::GLtk::Resources &, Marklin::Train *);
+       TrainProperties(Engineer &, Marklin::Train *);
 private:
        virtual void on_ok_clicked();
+       void new_vehicle_selected(unsigned, const std::string &);
+       void remove_vehicle_clicked();
+       const Marklin::VehicleType &get_vehicle_type(unsigned, bool);
 };
 
 #endif