X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2F3d%2Fcatalogue.h;h=586ee66758f617778af53d6a489b0c0bd1bd69bc;hb=d06a1e1ab34e5ecb4bb7011d44298a0b5e6162be;hp=689756d1b59775ad03e757b2b01101311922b4cf;hpb=7e382cc3cad8c4f6945b0c9d89e2ca917b42b740;p=r2c2.git

diff --git a/source/3d/catalogue.h b/source/3d/catalogue.h
index 689756d..586ee66 100644
--- a/source/3d/catalogue.h
+++ b/source/3d/catalogue.h
@@ -1,32 +1,49 @@
-/* $Id$
+#ifndef R2C2_3D_CATALOGUE_H_
+#define R2C2_3D_CATALOGUE_H_
 
-This file is part of the MSP Märklin suite
-Copyright © 2010 Mikkosoft Productions, Mikko Rasa
-Distributed under the GPL
-*/
+#include <msp/datafile/collection.h>
+#include <msp/datafile/directorysource.h>
+#include <msp/gl/mesh.h>
+#include <msp/gl/resources.h>
+#include "libr2c2/catalogue.h"
+#include "typemap.h"
 
-#ifndef MARKLIN3D_CATALOGUE_H_
-#define MARKLIN3D_CATALOGUE_H_
+namespace R2C2 {
 
-#include <msp/gl/material.h>
-#include "libmarklin/catalogue.h"
+class ObjectType3D;
 
-namespace Marklin {
-
-class TrackType3D;
-
-class Catalogue3D
+class Catalogue3D: public Msp::GL::Resources
 {
 private:
-	const Catalogue &catalogue;
-	std::map<const TrackType *, TrackType3D *> tracks;
-	Msp::GL::Material *ballast_material;
+	typedef std::map<const ObjectType *, ObjectType3D *> ObjectMap;
+	typedef std::map<const TrackAppearance *, Msp::GL::Mesh *> EndpointMap;
+
+	Catalogue &catalogue;
+	ObjectMap objects;
+	EndpointMap endpoint_meshes;
 
 public:
-	Catalogue3D(const Catalogue &);
+	Catalogue3D(Catalogue &);
+	~Catalogue3D();
 
 	const Catalogue &get_catalogue() const { return catalogue; }
-	const TrackType3D &get_track(const TrackType &) const;
+
+	const ObjectType3D &get_3d(const ObjectType &);
+	const ObjectType3D &get_3d(const ObjectType &) const;
+
+	template<typename T>
+	const typename TypeMap3D<T>::Type3D &get_3d(const T &ot) const
+	{ return dynamic_cast<const typename TypeMap3D<T>::Type3D &>(get_3d(static_cast<const ObjectType &>(ot))); }
+
+	template<typename T>
+	const typename TypeMap3D<T>::Type3D &get_3d(const T &ot)
+	{ return dynamic_cast<const typename TypeMap3D<T>::Type3D &>(get_3d(static_cast<const ObjectType &>(ot))); }
+
+private:
+	void build_endpoint_mesh(const TrackAppearance &);
+public:
+	const Msp::GL::Mesh &get_endpoint_mesh(const TrackAppearance &);
+	const Msp::GL::Mesh &get_endpoint_mesh(const TrackAppearance &) const;
 };
 
 }