]> git.tdb.fi Git - r2c2.git/blob - source/3d/layout.cpp
Avoid crash if an object does not have a shape
[r2c2.git] / source / 3d / layout.cpp
1 #include "beamgate.h"
2 #include "layout.h"
3 #include "signal.h"
4 #include "track.h"
5 #include "utility.h"
6 #include "vehicle.h"
7
8 using namespace std;
9 using namespace Msp;
10
11 namespace R2C2 {
12
13 Layout3D::Layout3D(Layout &l):
14         layout(l),
15         catalogue(layout.get_catalogue())
16 {
17         // South, 15° from zenith
18         sun.set_position(0, -0.259, 0.966, 0);
19         sun.set_diffuse(GL::Color(0.9));
20         lighting.set_ambient(GL::Color(0.4));
21         lighting.attach(0, sun);
22
23         layout.signal_object_added.connect(sigc::mem_fun(this, &Layout3D::object_added));
24         layout.signal_object_removed.connect(sigc::mem_fun(this, &Layout3D::object_removed));
25
26         const set<Object *> &lobjs = layout.get_all<Object>();
27         for(set<Object *>::iterator i=lobjs.begin(); i!=lobjs.end(); ++i)
28                 object_added(**i);
29 }
30
31 Layout3D::~Layout3D()
32 {
33         while(!utilities.empty())
34                 delete *utilities.begin();
35         while(!objects.empty())
36                 delete objects.begin()->second;
37 }
38
39 void Layout3D::get_bounds(Vector &minp, Vector &maxp) const
40 {
41         Geometry::BoundingBox<float, 3> bbox;
42
43         for(ObjectMap::const_iterator i=objects.begin(); i!=objects.end(); ++i)
44         {
45                 const Shape *shape = i->second->get_object().get_shape();
46                 if(shape)
47                         bbox = bbox|shape->get_axis_aligned_bounding_box();
48         }
49
50         minp = bbox.get_minimum_point();
51         maxp = bbox.get_maximum_point();
52 }
53
54 void Layout3D::add(Object3D &o)
55 {
56         insert_unique(objects, &o.get_object(), &o);
57 }
58
59 Object3D &Layout3D::get_3d(Object &o) const
60 {
61         return *get_item(objects, &o);
62 }
63
64 void Layout3D::remove(Object3D &o)
65 {
66         objects.erase(&o.get_object());
67 }
68
69 void Layout3D::add(Utility3D &u)
70 {
71         utilities.insert(&u);
72 }
73
74 void Layout3D::remove(Utility3D &u)
75 {
76         utilities.erase(&u);
77 }
78
79 void Layout3D::object_added(Object &o)
80 {
81         if(Track *t = dynamic_cast<Track *>(&o))
82                 new Track3D(*this, *t);
83         else if(Signal *s = dynamic_cast<Signal *>(&o))
84                 new Signal3D(*this, *s);
85         else if(Vehicle *v = dynamic_cast<Vehicle *>(&o))
86                 new Vehicle3D(*this, *v);
87         else if(BeamGate *g = dynamic_cast<BeamGate *>(&o))
88                 new BeamGate3D(*this, *g);
89 }
90
91 void Layout3D::object_removed(Object &o)
92 {
93         ObjectMap::iterator i = objects.find(&o);
94         if(i!=objects.end())
95                 delete i->second;
96 }
97
98 } // namespace R2C2