Add basic support for signals

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

diff --git a/source/3d/catalogue.cpp b/source/3d/catalogue.cpp
index 935219ed75e268347ea654e20d9aa6f56e7215e3..4d7b4a83fea5f3835221198004c6b28dd9c80e37 100644 (file)
--- a/source/3d/catalogue.cpp
+++ b/source/3d/catalogue.cpp
@@ -4,6 +4,7 @@
 #include <msp/gl/program.h>
 #include <msp/gl/technique.h>
 #include "catalogue.h"
+#include "signaltype.h"
 #include "tracktype.h"
 #include "vehicletype.h"
 
@@ -24,10 +25,14 @@ Catalogue3D::Catalogue3D(Catalogue &c):
 
        catalogue.signal_track_added.connect(sigc::mem_fun(this, &Catalogue3D::track_added));
        catalogue.signal_vehicle_added.connect(sigc::mem_fun(this, &Catalogue3D::vehicle_added));
+       catalogue.signal_signal_added.connect(sigc::mem_fun(this, &Catalogue3D::signal_added));
 
        const Catalogue::TrackMap &trks = catalogue.get_tracks();
        for(Catalogue::TrackMap::const_iterator i=trks.begin(); i!=trks.end(); ++i)
                track_added(*i->second);
+       const Catalogue::SignalMap &sigs = catalogue.get_signals();
+       for(Catalogue::SignalMap::const_iterator i=sigs.begin(); i!=sigs.end(); ++i)
+               signal_added(*i->second);
 
        build_endpoint_mesh();
 }
@@ -38,6 +43,8 @@ Catalogue3D::~Catalogue3D()
                delete i->second;
        for(map<const VehicleType *, VehicleType3D *>::iterator i=vehicles.begin(); i!=vehicles.end(); ++i)
                delete i->second;
+       for(map<const SignalType *, SignalType3D *>::iterator i=signals.begin(); i!=signals.end(); ++i)
+               delete i->second;
 }
 
 const TrackType3D &Catalogue3D::get_track(const TrackType &tt) const
@@ -50,6 +57,11 @@ const VehicleType3D &Catalogue3D::get_vehicle(const VehicleType &vt) const
        return *get_item(vehicles, &vt);
 }
 
+const SignalType3D &Catalogue3D::get_signal(const SignalType &vt) const
+{
+       return *get_item(signals, &vt);
+}
+
 void Catalogue3D::track_added(const TrackType &track)
 {
        tracks[&track] = new TrackType3D(*this, track);
@@ -60,6 +72,11 @@ void Catalogue3D::vehicle_added(const VehicleType &veh)
        vehicles[&veh] = new VehicleType3D(*this, veh);
 }
 
+void Catalogue3D::signal_added(const SignalType &sig)
+{
+       signals[&sig] = new SignalType3D(*this, sig);
+}
+
 void Catalogue3D::build_endpoint_mesh()
 {
        const Profile &ballast_profile = catalogue.get_ballast_profile();