X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2F3d%2Fcatalogue.cpp;h=fb4c397d36d153320426b7c3ff71c5ac292a0002;hb=508ee4bfcc0f8fb1373fb7af251c59c873ef896f;hp=994d5f4d82169aa4dd386471373576a365db7afa;hpb=7e382cc3cad8c4f6945b0c9d89e2ca917b42b740;p=r2c2.git

diff --git a/source/3d/catalogue.cpp b/source/3d/catalogue.cpp
index 994d5f4..fb4c397 100644
--- a/source/3d/catalogue.cpp
+++ b/source/3d/catalogue.cpp
@@ -1,33 +1,92 @@
-/* $Id$
-
-This file is part of the MSP Märklin suite
-Copyright © 2010 Mikkosoft Productions, Mikko Rasa
-Distributed under the GPL
-*/
-
+#include <msp/core/maputils.h>
+#include <msp/gl/meshbuilder.h>
+#include "libr2c2/trackappearance.h"
 #include "catalogue.h"
+#include "signaltype.h"
 #include "tracktype.h"
+#include "vehicletype.h"
 
 using namespace std;
 using namespace Msp;
 
-namespace Marklin {
+namespace R2C2 {
 
-Catalogue3D::Catalogue3D(const Catalogue &c):
+Catalogue3D::Catalogue3D(Catalogue &c):
 	catalogue(c)
 {
-	const map<unsigned, TrackType *> &trks = catalogue.get_tracks();
-	for(map<unsigned, TrackType *>::const_iterator i=trks.begin(); i!=trks.end(); ++i)
-		tracks[i->second] = new TrackType3D(*this, *i->second);
+	const list<DataFile::CollectionSource *> &src = catalogue.get_sources();
+	for(list<DataFile::CollectionSource *>::const_iterator i=src.begin(); i!=src.end(); ++i)
+		add_source(**i);
+	catalogue.signal_source_added.connect(sigc::mem_fun(static_cast<DataFile::Collection *>(this), &Catalogue3D::add_source));
+}
+
+Catalogue3D::~Catalogue3D()
+{
+	for(map<const ObjectType *, ObjectType3D *>::iterator i=objects.begin(); i!=objects.end(); ++i)
+		delete i->second;
 }
 
-const TrackType3D &Catalogue3D::get_track(const TrackType &tt) const
+const ObjectType3D &Catalogue3D::get_3d(const ObjectType &ot)
 {
-	map<const TrackType *, TrackType3D *>::const_iterator i = tracks.find(&tt);
-	if(i==tracks.end())
-		throw KeyError("Unknown track type");
+	ObjectMap::iterator i = objects.find(&ot);
+	if(i!=objects.end())
+		return *i->second;
+
+	ObjectType3D *ot3d = 0;
+	if(const TrackType *tt = dynamic_cast<const TrackType *>(&ot))
+		ot3d = new TrackType3D(*this, *tt);
+	else if(const SignalType *st = dynamic_cast<const SignalType *>(&ot))
+		ot3d = new SignalType3D(*this, *st);
+	else if(const VehicleType *vt = dynamic_cast<const VehicleType *>(&ot))
+		ot3d = new VehicleType3D(*this, *vt);
+	else
+		throw key_error(&ot);
+
+	objects[&ot] = ot3d;
+	return *ot3d;
+}
+
+const ObjectType3D &Catalogue3D::get_3d(const ObjectType &ot) const
+{
+	return *get_item(objects, &ot);
+}
 
-	return *i->second;
+const GL::Mesh &Catalogue3D::get_endpoint_mesh(const TrackAppearance &appearance)
+{
+	EndpointMap::iterator i = endpoint_meshes.find(&appearance);
+	if(i!=endpoint_meshes.end())
+		return *i->second;
+
+	const Profile &ballast_profile = appearance.get_ballast_profile();
+	const Vector &ballast_min = ballast_profile.get_min_coords();
+	const Vector &ballast_max = ballast_profile.get_max_coords();
+
+	const Profile &rail_profile = appearance.get_rail_profile();
+	const Vector &rail_min = rail_profile.get_min_coords();
+	const Vector &rail_max = rail_profile.get_max_coords();
+
+	float gauge = appearance.get_gauge();
+
+	float width = max(max(-ballast_min.x, ballast_max.x)*2, gauge+(rail_max.x-rail_min.x)*2)+0.004;
+	float height = ballast_max.y-ballast_min.y+rail_max.y-rail_min.y+0.01;
+
+	GL::Mesh *mesh = new GL::Mesh((GL::NORMAL3, GL::VERTEX3));
+	GL::MeshBuilder bld(*mesh);
+	bld.normal(1, 0, 0);
+	bld.begin(GL::QUADS);
+	bld.vertex(0, width/2, 0);
+	bld.vertex(0, width/2, height);
+	bld.vertex(0, -width/2, height);
+	bld.vertex(0, -width/2, 0);
+	bld.end();
+
+	endpoint_meshes[&appearance] = mesh;
+	return *mesh;
+}
+
+const GL::Mesh &Catalogue3D::get_endpoint_mesh(const TrackAppearance &appearance) const
+{
+	return *get_item(endpoint_meshes, &appearance);
 }
 
-} // namespace Marklin
+} // namespace R2C2