Don't crash if a train has no router

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

diff --git a/source/3d/signal.h b/source/3d/signal.h
index 5777590f0fcf71069e07ce820ab599505fb5e8ce..47505fc94cc5cda4909ef08d99761556d27438e3 100644 (file)
--- a/source/3d/signal.h
+++ b/source/3d/signal.h
@@ -3,21 +3,22 @@
 
 #include <msp/gl/objectinstance.h>
 #include "libr2c2/signal.h"
+#include "object.h"
 
 namespace R2C2 {
 
-class Layout3D;
-
-class Signal3D: public Msp::GL::ObjectInstance
+class Signal3D: public Object3D, public Msp::GL::ObjectInstance
 {
 private:
-       Layout3D &layout;
        Signal &signal;
 
 public:
        Signal3D(Layout3D &, Signal &);
        ~Signal3D();
 
+       virtual Vector get_node() const;
+       virtual bool is_visible() const { return true; }
+
        Signal &get_signal() const { return signal; }
        virtual void setup_render(Msp::GL::Renderer &, const Msp::GL::Tag &) const;
 };