Allow rotating the body object of a vehicle

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

diff --git a/source/3d/vehicle.cpp b/source/3d/vehicle.cpp
index 37619ebd339461f104dc0bf54aeb6fcfe71e7a67..0bd57e95da2b80d276fdf82cef0015649dc1d53d 100644 (file)
--- a/source/3d/vehicle.cpp
+++ b/source/3d/vehicle.cpp
@@ -1,26 +1,60 @@
-/* $Id$
-
-This file is part of the MSP Märklin suite
-Copyright © 2010  Mikkosoft Productions, Mikko Rasa
-Distributed under the GPL
-*/
-
 #include <cmath>
 #include <msp/gl/matrix.h>
+#include <msp/gl/renderer.h>
+#include "axle.h"
+#include "bogie.h"
 #include "layout.h"
+#include "rod.h"
 #include "vehicle.h"
 #include "vehicletype.h"
 
 using namespace std;
 using namespace Msp;
 
-namespace Marklin {
+namespace R2C2 {
 
 Vehicle3D::Vehicle3D(Layout3D &l, Vehicle &v):
+       GL::ObjectInstance(*l.get_catalogue().get_vehicle(v.get_type()).get_body_object()),
        layout(l),
        vehicle(v),
        type(layout.get_catalogue().get_vehicle(vehicle.get_type()))
 {
+       unsigned n_axles = vehicle.get_type().get_fixed_axles().size();
+       for(unsigned i=0; i<n_axles; ++i)
+               if(type.get_fixed_axle_object(i))
+               {
+                       Axle3D *a = new Axle3D(*this, i);
+                       axles.push_back(a);
+                       layout.get_scene().add(*a);
+               }
+
+       unsigned n_bogies = vehicle.get_type().get_bogies().size();
+       for(unsigned i=0; i<n_bogies; ++i)
+               if(type.get_bogie_object(i))
+               {
+                       Bogie3D *b = new Bogie3D(*this, i);
+                       bogies.push_back(b);
+                       layout.get_scene().add(*b);
+
+                       n_axles = vehicle.get_type().get_bogie(i).axles.size();
+                       for(unsigned j=0; j<n_axles; ++j)
+                               if(type.get_bogie_axle_object(i, j))
+                               {
+                                       Axle3D *a = new Axle3D(*this, i, j);
+                                       axles.push_back(a);
+                                       layout.get_scene().add(*a);
+                               }
+               }
+
+       unsigned n_rods = vehicle.get_type().get_rods().size();
+       for(unsigned i=0; i<n_rods; ++i)
+               if(type.get_rod_object(i))
+               {
+                       Rod3D *r = new Rod3D(*this, i);
+                       rods.push_back(r);
+                       layout.get_scene().add(*r);
+               }
+
        layout.add_vehicle(*this);
        layout.get_scene().add(*this);
 }
@@ -29,12 +63,18 @@ Vehicle3D::~Vehicle3D()
 {
        layout.remove_vehicle(*this);
        layout.get_scene().remove(*this);
+       for(vector<Axle3D *>::iterator i=axles.begin(); i!=axles.end(); ++i)
+               delete *i;
+       for(vector<Bogie3D *>::iterator i=bogies.begin(); i!=bogies.end(); ++i)
+               delete *i;
+       for(vector<Rod3D *>::iterator i=rods.begin(); i!=rods.end(); ++i)
+               delete *i;
 }
 
-Point Vehicle3D::get_node() const
+Vector Vehicle3D::get_node() const
 {
-       Point p = vehicle.get_position();
-       return Point(p.x, p.y, p.z+0.01+vehicle.get_type().get_height());
+       Vector p = vehicle.get_position();
+       return Vector(p.x, p.y, p.z+0.01+vehicle.get_type().get_height());
 }
 
 bool Vehicle3D::is_visible() const
@@ -42,35 +82,24 @@ bool Vehicle3D::is_visible() const
        return vehicle.get_track();
 }
 
-void Vehicle3D::render(const GL::Tag &tag) const
+void Vehicle3D::render(GL::Renderer &renderer, const GL::Tag &tag) const
 {
        if(!vehicle.get_track())
                return;
 
-       if(tag==0)
-       {
-               GL::PushMatrix push_mat;
-
-               const Point &pos = vehicle.get_position();
-               GL::translate(pos.x, pos.y, pos.z+0.01);
-               GL::rotate(vehicle.get_direction()*180/M_PI, 0, 0, 1);
-
-               if(type.get_body_object())
-                       type.get_body_object()->render(tag);
+       ObjectInstance::render(renderer, tag);
+}
 
-               const vector<VehicleType::Bogie> &bogies = vehicle.get_type().get_bogies();
-               for(unsigned i=0; i<bogies.size(); ++i)
-               {
-                       GL::PushMatrix push_mat2;
-                       GL::translate(bogies[i].position, 0, 0);
-                       float angle = vehicle.get_bogie_direction(i)*180/M_PI;
-                       if(bogies[i].rotate_object)
-                               angle += 180;
-                       GL::rotate(angle, 0, 0, 1);
-                       if(type.get_bogie_object(i))
-                               type.get_bogie_object(i)->render(tag);
-               }
-       }
+void Vehicle3D::setup_render(GL::Renderer &renderer, const GL::Tag &) const
+{
+       GL::Matrix matrix;
+       const Vector &pos = vehicle.get_position();
+       matrix.translate(pos.x, pos.y, pos.z);
+       float dir = vehicle.get_direction();
+       if(vehicle.get_type().get_rotate_object())
+               dir += M_PI;
+       matrix.rotate(dir, 0, 0, 1);
+       renderer.matrix_stack() *= matrix;
 }
 
-} // namespace Marklin
+} // namespace R2C2