Create a more generic version of Train::get_reserved_distance

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

diff --git a/source/3d/layout.h b/source/3d/layout.h
index ba590dd37e2d8482d5180138f2ab6f663a7a6554..a0c77868ae58bc68a3c804843622996d119d2263 100644 (file)
--- a/source/3d/layout.h
+++ b/source/3d/layout.h
@@ -12,20 +12,22 @@ Distributed under the GPL
 #include <msp/gl/scene.h>
 #include "libmarklin/layout.h"
 #include "catalogue.h"
-#include "track.h"
-#include "train.h"
 
 namespace Marklin {
 
+class Track3D;
+class Vehicle3D;
+
 class Layout3D: public sigc::trackable
 {
 private:
        Layout &layout;
        Catalogue3D catalogue;
        std::list<Track3D *> tracks;
-       std::list<Train3D *> trains;
+       std::list<Vehicle3D *> vehicles;
        Msp::GL::Scene scene;
        Msp::GL::Scene ep_scene;
+       Msp::GL::Scene path_scene;
 
 public:
        Layout3D(Layout &);
@@ -39,17 +41,18 @@ public:
        Track3D &get_track(const Track &) const;
        Track3D *pick_track(float, float, float) const;
 
-       void add_train(Train3D &);
-       void remove_train(Train3D &);
-       Train3D &get_train(const Train &) const;
+       void add_vehicle(Vehicle3D &);
+       void remove_vehicle(Vehicle3D &);
+       Vehicle3D &get_vehicle(const Vehicle &) 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; }
 
 private:
        void track_added(Track &);
        void track_removed(Track &);
-       void train_added(Train &);
+       void vehicle_added(Vehicle &);
 };
 
 } // namespace Marklin