X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2F3d%2Flayout.h;h=2e56ea63bd0195ea8ede83b3219f6f8dde1392f3;hb=bc8ac89bbe774bb133b758416182aa18e5e0a5a5;hp=09189ed82969df8b69d3bf1682b6387fb1119211;hpb=6c61179fe09af2f5366d50f10aadbf5f83438087;p=r2c2.git

diff --git a/source/3d/layout.h b/source/3d/layout.h
index 09189ed..2e56ea6 100644
--- a/source/3d/layout.h
+++ b/source/3d/layout.h
@@ -1,29 +1,55 @@
+/* $Id$
+
+This file is part of the MSP Märklin suite
+Copyright © 2006-2010 Mikkosoft Productions, Mikko Rasa
+Distributed under the GPL
+*/
+
 #ifndef MARKLIN3D_LAYOUT_H_
 #define MARKLIN3D_LAYOUT_H_
 
+#include <msp/gl/scene.h>
 #include "libmarklin/layout.h"
+#include "catalogue.h"
 #include "track.h"
+#include "train.h"
 
 namespace Marklin {
 
 class Layout3D
 {
+private:
+	Layout &layout;
+	Catalogue3D catalogue;
+	std::list<Track3D *> tracks;
+	std::list<Train3D *> trains;
+	Msp::GL::Scene scene;
+	Msp::GL::Scene ep_scene;
+
 public:
 	Layout3D(Layout &);
 	~Layout3D();
 
-	void set_quality(unsigned);
-	const Track3DSeq &get_tracks() const { return tracks; }
-	void  render(bool =false);
-	Track3D *get_track(const Track *);
-	Track3D *pick_track(float, float, float);
-private:
-	Layout     &layout;
-	Track3DSeq tracks;
-	unsigned   quality;
+	const Catalogue3D &get_catalogue() const { return catalogue; }
 
-	void track_added(Track *);
-	void track_removed(Track *);
+	void add_track(Track3D &);
+	void remove_track(Track3D &);
+	const std::list<Track3D *> &get_tracks() const { return tracks; }
+	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