X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2F3d%2Flayout.cpp;h=2927e446741df4e1ad2ae194949831f449cc2afc;hb=cb4e53ba1b8ea41b978190de6668e9862b9cfea8;hp=2881e8d33c82ed23d8c5c6edcffa09e684b8f8fa;hpb=495c54e187ddbb7514c983e1829a69d82ccad5f3;p=r2c2.git

diff --git a/source/3d/layout.cpp b/source/3d/layout.cpp
index 2881e8d..2927e44 100644
--- a/source/3d/layout.cpp
+++ b/source/3d/layout.cpp
@@ -1,11 +1,5 @@
-/* $Id$
-
-This file is part of R²C²
-Copyright © 2006-2011 Mikkosoft Productions, Mikko Rasa
-Distributed under the GPL
-*/
-
 #include "layout.h"
+#include "signal.h"
 #include "track.h"
 #include "vehicle.h"
 
@@ -22,18 +16,18 @@ Layout3D::Layout3D(Layout &l):
 	sun.set_position(0, -0.259, 0.966, 0);
 	lighting.attach(0, sun);
 
-	layout.signal_track_added.connect(sigc::mem_fun(this, &Layout3D::track_added));
-	layout.signal_track_removed.connect(sigc::mem_fun(this, &Layout3D::track_removed));
-	layout.signal_vehicle_added.connect(sigc::mem_fun(this, &Layout3D::vehicle_added));
-	layout.signal_vehicle_removed.connect(sigc::mem_fun(this, &Layout3D::vehicle_removed));
+	layout.signal_object_added.connect(sigc::mem_fun(this, &Layout3D::object_added));
+	layout.signal_object_removed.connect(sigc::mem_fun(this, &Layout3D::object_removed));
 
-	const set<Track *> &ltracks = layout.get_tracks();
-	for(set<Track *>::iterator i=ltracks.begin(); i!=ltracks.end(); ++i)
-		track_added(**i);
+	const set<Object *> &lobjs = layout.get_all<Object>();
+	for(set<Object *>::iterator i=lobjs.begin(); i!=lobjs.end(); ++i)
+		object_added(**i);
 }
 
 Layout3D::~Layout3D()
 {
+	while(!signals.empty())
+		delete signals.begin()->second;
 	while(!tracks.empty())
 		delete tracks.begin()->second;
 	while(!vehicles.empty())
@@ -42,26 +36,18 @@ Layout3D::~Layout3D()
 
 void Layout3D::get_bounds(Vector &minp, Vector &maxp) const
 {
-	minp = maxp = Vector();
+	Geometry::BoundingBox<float, 3> bbox;
 
 	for(TrackMap::const_iterator i=tracks.begin(); i!=tracks.end(); ++i)
-	{
-		Vector tmin;
-		Vector tmax;
-		i->second->get_bounds(0, tmin, tmax);
-		minp.x = min(minp.x, tmin.x);
-		minp.y = min(minp.y, tmin.y);
-		maxp.x = max(maxp.x, tmax.x);
-		maxp.y = max(maxp.y, tmax.y);
-	}
+		bbox = bbox|i->second->get_track().get_type().get_shape()->get_axis_aligned_bounding_box();
+
+	minp = bbox.get_minimum_point();
+	maxp = bbox.get_maximum_point();
 }
 
 void Layout3D::add_track(Track3D &t)
 {
-	if(tracks.count(&t.get_track()))
-		throw KeyError("Duplicate track");
-
-	tracks[&t.get_track()] = &t;
+	insert_unique(tracks, &t.get_track(), &t);
 }
 
 void Layout3D::remove_track(Track3D &t)
@@ -71,19 +57,12 @@ void Layout3D::remove_track(Track3D &t)
 
 Track3D &Layout3D::get_track(Track &t) const
 {
-	TrackMap::const_iterator i = tracks.find(&t);
-	if(i==tracks.end())
-		throw KeyError("Unknown track");
-
-	return *i->second;
+	return *get_item(tracks, &t);
 }
 
 void Layout3D::add_vehicle(Vehicle3D &v)
 {
-	if(vehicles.count(&v.get_vehicle()))
-		throw KeyError("Duplicate vehicle");
-
-	vehicles[&v.get_vehicle()] = &v;
+	insert_unique(vehicles, &v.get_vehicle(), &v);
 }
 
 void Layout3D::remove_vehicle(Vehicle3D &v)
@@ -93,35 +72,54 @@ void Layout3D::remove_vehicle(Vehicle3D &v)
 
 Vehicle3D &Layout3D::get_vehicle(Vehicle &v) const
 {
-	VehicleMap::const_iterator i = vehicles.find(&v);
-	if(i==vehicles.end())
-		throw KeyError("Unknown vehicle");
+	return *get_item(vehicles, &v);
+}
 
-	return *i->second;
+void Layout3D::add_signal(Signal3D &s)
+{
+	insert_unique(signals, &s.get_signal(), &s);
 }
 
-void Layout3D::track_added(Track &t)
+void Layout3D::remove_signal(Signal3D &s)
 {
-	new Track3D(*this, t);
+	signals.erase(&s.get_signal());
 }
 
-void Layout3D::track_removed(Track &t)
+Signal3D &Layout3D::get_signal(Signal &s) const
 {
-	TrackMap::iterator i = tracks.find(&t);
-	if(i!=tracks.end())
-		delete i->second;
+	return *get_item(signals, &s);
 }
 
-void Layout3D::vehicle_added(Vehicle &v)
+void Layout3D::object_added(Object &o)
 {
-	new Vehicle3D(*this, v);
+	if(Track *t = dynamic_cast<Track *>(&o))
+		new Track3D(*this, *t);
+	else if(Signal *s = dynamic_cast<Signal *>(&o))
+		new Signal3D(*this, *s);
+	else if(Vehicle *v = dynamic_cast<Vehicle *>(&o))
+		new Vehicle3D(*this, *v);
 }
 
-void Layout3D::vehicle_removed(Vehicle &v)
+void Layout3D::object_removed(Object &o)
 {
-	VehicleMap::iterator i = vehicles.find(&v);
-	if(i!=vehicles.end())
-		delete i->second;
+	if(Track *t = dynamic_cast<Track *>(&o))
+	{
+		TrackMap::iterator i = tracks.find(t);
+		if(i!=tracks.end())
+			delete i->second;
+	}
+	else if(Signal *s = dynamic_cast<Signal *>(&o))
+	{
+		SignalMap::iterator i = signals.find(s);
+		if(i!=signals.end())
+			delete i->second;
+	}
+	else if(Vehicle *v = dynamic_cast<Vehicle *>(&o))
+	{
+		VehicleMap::iterator i = vehicles.find(v);
+		if(i!=vehicles.end())
+			delete i->second;
+	}
 }
 
 } // namespace R2C2