Add basic support for signals

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

diff --git a/source/3d/catalogue.h b/source/3d/catalogue.h
index 6b819b71b61561c2a2db2589bd74df859934f4e7..ecac7ddca27b97ca05ac3a71c8c37fecdb0fd56f 100644 (file)
--- a/source/3d/catalogue.h
+++ b/source/3d/catalogue.h
@@ -9,6 +9,7 @@
 
 namespace R2C2 {
 
+class SignalType3D;
 class TrackType3D;
 class VehicleType3D;
 
@@ -18,6 +19,7 @@ private:
        Catalogue &catalogue;
        std::map<const TrackType *, TrackType3D *> tracks;
        std::map<const VehicleType *, VehicleType3D *> vehicles;
+       std::map<const SignalType *, SignalType3D *> signals;
        Msp::GL::Mesh endpoint_mesh;
 
 public:
@@ -27,10 +29,12 @@ public:
        const Catalogue &get_catalogue() const { return catalogue; }
        const TrackType3D &get_track(const TrackType &) const;
        const VehicleType3D &get_vehicle(const VehicleType &) const;
+       const SignalType3D &get_signal(const SignalType &) const;
        const Msp::GL::Mesh &get_endpoint_mesh() const { return endpoint_mesh; }
 private:
        void track_added(const TrackType &);
        void vehicle_added(const VehicleType &);
+       void signal_added(const SignalType &);
        void build_endpoint_mesh();
 
        Msp::FS::Path locate_file(const std::string &);