Emit various signals from Train when it's loaded

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

diff --git a/source/3d/layout.h b/source/3d/layout.h
index a7fbdb98b9dbc7b7751ce2e26470403dd647c85d..2e56ea63bd0195ea8ede83b3219f6f8dde1392f3 100644 (file)
--- a/source/3d/layout.h
+++ b/source/3d/layout.h
@@ -12,6 +12,7 @@ Distributed under the GPL
 #include "libmarklin/layout.h"
 #include "catalogue.h"
 #include "track.h"
+#include "train.h"
 
 namespace Marklin {
 
@@ -21,16 +22,15 @@ private:
        Layout &layout;
        Catalogue3D catalogue;
        std::list<Track3D *> tracks;
+       std::list<Train3D *> trains;
        Msp::GL::Scene scene;
        Msp::GL::Scene ep_scene;
-       unsigned quality;
 
 public:
        Layout3D(Layout &);
        ~Layout3D();
 
        const Catalogue3D &get_catalogue() const { return catalogue; }
-       void set_quality(unsigned);
 
        void add_track(Track3D &);
        void remove_track(Track3D &);
@@ -38,12 +38,18 @@ public:
        Track3D &get_track(const Track &) const;
        Track3D *pick_track(float, float, float) const;
 
+       void set_traffic_manager(TrafficManager &);
+       void add_train(Train3D &);
+       void remove_train(Train3D &);
+       Train3D &get_train(const Train &) const;
+
        Msp::GL::Scene &get_scene() { return scene; }
        Msp::GL::Scene &get_endpoint_scene() { return ep_scene; }
 
 private:
        void track_added(Track &);
        void track_removed(Track &);
+       void train_added(Train &);
 };
 
 } // namespace Marklin