Use maps instead of lists in Layout3D

[r2c2.git] / source / 3d / layout.h

diff --git a/source/3d/layout.h b/source/3d/layout.h
index f09603de5b00dad0d9ad06178a7e1eef71d98aa0..f2de127a0cc19d96117b4a63f3fb3be8c042921c 100644 (file)
--- a/source/3d/layout.h
+++ b/source/3d/layout.h
@@ -20,11 +20,15 @@ class Vehicle3D;
 
 class Layout3D: public sigc::trackable
 {
+public:
+       typedef std::map<Track *, Track3D *> TrackMap;
+       typedef std::map<Vehicle *, Vehicle3D *> VehicleMap;
+
 private:
        Layout &layout;
        Catalogue3D catalogue;
-       std::list<Track3D *> tracks;
-       std::list<Vehicle3D *> vehicles;
+       TrackMap tracks;
+       VehicleMap vehicles;
        Msp::GL::Scene scene;
        Msp::GL::Scene ep_scene;
        Msp::GL::Scene path_scene;
@@ -37,13 +41,13 @@ public:
 
        void add_track(Track3D &);
        void remove_track(Track3D &);
-       const std::list<Track3D *> &get_tracks() const { return tracks; }
+       const TrackMap &get_tracks() const { return tracks; }
        Track3D &get_track(Track &) const;
        Track3D *pick_track(float, float, float) const;
 
        void add_vehicle(Vehicle3D &);
        void remove_vehicle(Vehicle3D &);
-       Vehicle3D &get_vehicle(const Vehicle &) const;
+       Vehicle3D &get_vehicle(Vehicle &) const;
 
        Msp::GL::Scene &get_scene() { return scene; }
        Msp::GL::Scene &get_endpoint_scene() { return ep_scene; }