Basic support for beam gate sensors

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

diff --git a/source/3d/layout.cpp b/source/3d/layout.cpp
index 05aaa4205bf85053d7a0c1513253fddf5cef87b7..1bbbed825bb68f493547046fa8051a60fbdb8b1b 100644 (file)
--- a/source/3d/layout.cpp
+++ b/source/3d/layout.cpp
@@ -1,151 +1,79 @@
-/* $Id$
-
-This file is part of the MSP Märklin suite
-Copyright © 2006-2010 Mikkosoft Productions, Mikko Rasa
-Distributed under the GPL
-*/
-
-#include <algorithm>
-#include <limits>
-#include <msp/gl/clip.h>
-#include <msp/gl/matrix.h>
-#include <msp/gl/rendermode.h>
-#include <msp/gl/select.h>
-#include <msp/gl/texture.h>
-#include <msp/datafile/parser.h>
+#include "beamgate.h"
 #include "layout.h"
+#include "signal.h"
 #include "track.h"
 #include "vehicle.h"
 
 using namespace std;
 using namespace Msp;
 
-namespace Marklin {
+namespace R2C2 {
 
 Layout3D::Layout3D(Layout &l):
        layout(l),
        catalogue(layout.get_catalogue())
 {
-       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));
-
-       const set<Track *> &ltracks = layout.get_tracks();
-       for(set<Track *>::iterator i=ltracks.begin(); i!=ltracks.end(); ++i)
-               track_added(**i);
-}
-
-Layout3D::~Layout3D()
-{
-       while(!tracks.empty())
-               delete tracks.begin()->second;
-       while(!vehicles.empty())
-               delete vehicles.begin()->second;
-}
+       // South, 15° from zenith
+       sun.set_position(0, -0.259, 0.966, 0);
+       lighting.attach(0, sun);
 
-void Layout3D::add_track(Track3D &t)
-{
-       if(tracks.count(&t.get_track()))
-               throw KeyError("Duplicate track");
+       layout.signal_object_added.connect(sigc::mem_fun(this, &Layout3D::object_added));
+       layout.signal_object_removed.connect(sigc::mem_fun(this, &Layout3D::object_removed));
 
-       tracks[&t.get_track()] = &t;
+       const set<Object *> &lobjs = layout.get_all<Object>();
+       for(set<Object *>::iterator i=lobjs.begin(); i!=lobjs.end(); ++i)
+               object_added(**i);
 }
 
-void Layout3D::remove_track(Track3D &t)
-{
-       tracks.erase(&t.get_track());
-}
-
-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;
-}
-
-Track3D *Layout3D::pick_track(float x, float y, float size) const
+Layout3D::~Layout3D()
 {
-       vector<GL::SelectRecord> select_buf;
-       GL::select_buffer(select_buf);
-       GL::render_mode(GL::SELECT);
-
-       {
-               GL::PushMatrix push_mat;
-               GL::load_identity();
-
-               GL::ClipPlane(1, 0, x-size, 0).apply_to(0);
-               GL::ClipPlane(-1, 0, -x-size, 0).apply_to(1);
-               GL::ClipPlane(0, 1, y-size, 0).apply_to(2);
-               GL::ClipPlane(0, -1, -y-size, 0).apply_to(3);
-       }
-
-       scene.render(0);
-
-       GL::ClipPlane::disable(0);
-       GL::ClipPlane::disable(1);
-       GL::ClipPlane::disable(2);
-       GL::ClipPlane::disable(3);
-
-       GL::render_mode(GL::RENDER);
-       Track3D *track = 0;
-       unsigned track_depth = numeric_limits<unsigned>::max();
-       for(vector<GL::SelectRecord>::iterator i=select_buf.begin(); i!=select_buf.end(); ++i)
-               if(i->min_depth<track_depth && !i->names.empty())
-               {
-                       track = reinterpret_cast<Track3D *>(i->names.back());
-                       track_depth = i->min_depth;
-               }
-
-       return track;
+       while(!objects.empty())
+               delete objects.begin()->second;
 }
 
-void Layout3D::add_vehicle(Vehicle3D &v)
+void Layout3D::get_bounds(Vector &minp, Vector &maxp) const
 {
-       if(vehicles.count(&v.get_vehicle()))
-               throw KeyError("Duplicate vehicle");
+       Geometry::BoundingBox<float, 3> bbox;
 
-       vehicles[&v.get_vehicle()] = &v;
-}
+       for(ObjectMap::const_iterator i=objects.begin(); i!=objects.end(); ++i)
+               bbox = bbox|i->second->get_object().get_type().get_shape()->get_axis_aligned_bounding_box();
 
-void Layout3D::remove_vehicle(Vehicle3D &v)
-{
-       vehicles.erase(&v.get_vehicle());
+       minp = bbox.get_minimum_point();
+       maxp = bbox.get_maximum_point();
 }
 
-Vehicle3D &Layout3D::get_vehicle(Vehicle &v) const
+void Layout3D::add(Object3D &o)
 {
-       VehicleMap::const_iterator i = vehicles.find(&v);
-       if(i==vehicles.end())
-               throw KeyError("Unknown vehicle");
-
-       return *i->second;
+       insert_unique(objects, &o.get_object(), &o);
 }
 
-void Layout3D::track_added(Track &t)
+Object3D &Layout3D::get(Object &o) const
 {
-       new Track3D(*this, t);
+       return *get_item(objects, &o);
 }
 
-void Layout3D::track_removed(Track &t)
+void Layout3D::remove(Object3D &o)
 {
-       TrackMap::iterator i = tracks.find(&t);
-       if(i!=tracks.end())
-               delete i->second;
+       objects.erase(&o.get_object());
 }
 
-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);
+       else if(BeamGate *g = dynamic_cast<BeamGate *>(&o))
+               new BeamGate3D(*this, *g);
 }
 
-void Layout3D::vehicle_removed(Vehicle &v)
+void Layout3D::object_removed(Object &o)
 {
-       VehicleMap::iterator i = vehicles.find(&v);
-       if(i!=vehicles.end())
+       ObjectMap::iterator i = objects.find(&o);
+       if(i!=objects.end())
                delete i->second;
 }
 
-} // namespace Marklin
+} // namespace R2C2