Add basic support for signals

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

diff --git a/source/3d/layout.h b/source/3d/layout.h
index 76f1b91d7a3c21ee577c183c732f28151415df5e..1e189bd1ce0eedd88a0541672e9d58f60e6f6efc 100644 (file)
--- a/source/3d/layout.h
+++ b/source/3d/layout.h
@@ -11,7 +11,7 @@
 
 namespace R2C2 {
 
-class Object3D;
+class Signal3D;
 class Track3D;
 class Vehicle3D;
 
@@ -20,12 +20,14 @@ class Layout3D: public sigc::trackable
 public:
        typedef std::map<Track *, Track3D *> TrackMap;
        typedef std::map<Vehicle *, Vehicle3D *> VehicleMap;
+       typedef std::map<Signal *, Signal3D *> SignalMap;
 
 private:
        Layout &layout;
        Catalogue3D catalogue;
        TrackMap tracks;
        VehicleMap vehicles;
+       SignalMap signals;
        Msp::GL::InstanceScene scene;
        Msp::GL::SimpleScene ep_scene;
        Msp::GL::InstanceScene path_scene;
@@ -49,6 +51,10 @@ public:
        void remove_vehicle(Vehicle3D &);
        Vehicle3D &get_vehicle(Vehicle &) const;
 
+       void add_signal(Signal3D &);
+       void remove_signal(Signal3D &);
+       Signal3D &get_signal(Signal &) const;
+
        Msp::GL::Scene &get_scene() { return scene; }
        Msp::GL::Scene &get_endpoint_scene() { return ep_scene; }
        Msp::GL::Scene &get_path_scene() { return path_scene; }
@@ -59,6 +65,8 @@ private:
        void track_removed(Track &);
        void vehicle_added(Vehicle &);
        void vehicle_removed(Vehicle &);
+       void signal_added(Signal &);
+       void signal_removed(Signal &);
 };
 
 } // namespace R2C2