source/3d/vehicle.cpp

   1 /* $Id$
   2
   3 This file is part of the MSP Märklin suite
   4 Copyright © 2010  Mikkosoft Productions, Mikko Rasa
   5 Distributed under the GPL
   6 */
   7
   8 #include <cmath>
   9 #include <msp/gl/matrix.h>
  10 #include "layout.h"
  11 #include "vehicle.h"
  12 #include "vehicletype.h"
  13
  14 using namespace std;
  15 using namespace Msp;
  16
  17 namespace Marklin {
  18
  19 Vehicle3D::Vehicle3D(Layout3D &l, Vehicle &v):
  20         layout(l),
  21         vehicle(v),
  22         type(layout.get_catalogue().get_vehicle(vehicle.get_type()))
  23 {
  24         layout.add_vehicle(*this);
  25         layout.get_scene().add(*this);
  26 }
  27
  28 Vehicle3D::~Vehicle3D()
  29 {
  30         layout.remove_vehicle(*this);
  31         layout.get_scene().remove(*this);
  32 }
  33
  34 Point Vehicle3D::get_node() const
  35 {
  36         Point p = vehicle.get_position();
  37         return Point(p.x, p.y, p.z+0.01+vehicle.get_type().get_height());
  38 }
  39
  40 bool Vehicle3D::is_visible() const
  41 {
  42         return vehicle.get_track();
  43 }
  44
  45 void Vehicle3D::render(const GL::Tag &tag) const
  46 {
  47         if(!vehicle.get_track())
  48                 return;
  49
  50         if(tag==0)
  51         {
  52                 GL::PushMatrix push_mat;
  53
  54                 const Point &pos = vehicle.get_position();
  55                 GL::translate(pos.x, pos.y, pos.z);
  56                 GL::rotate(vehicle.get_direction()*180/M_PI, 0, 0, 1);
  57
  58                 if(const GL::Object *obj = type.get_body_object())
  59                         obj->render(tag);
  60
  61                 const vector<VehicleType::Axle> &axles = vehicle.get_type().get_axles();
  62                 for(unsigned i=0; i<axles.size(); ++i)
  63                         if(const GL::Object *obj = type.get_axle_object(i))
  64                         {
  65                                 GL::PushMatrix push_mat2;
  66                                 GL::translate(axles[i].position, 0, axles[i].wheel_dia/2);
  67                                 obj->render(tag);
  68                         }
  69
  70                 const vector<VehicleType::Bogie> &bogies = vehicle.get_type().get_bogies();
  71                 for(unsigned i=0; i<bogies.size(); ++i)
  72                 {
  73                         GL::PushMatrix push_mat2;
  74                         GL::translate(bogies[i].position, 0, 0);
  75                         float angle = vehicle.get_bogie_direction(i)*180/M_PI;
  76                         GL::rotate(angle, 0, 0, 1);
  77
  78                         for(unsigned j=0; j<bogies[i].axles.size(); ++j)
  79                                 if(const GL::Object *obj = type.get_bogie_axle_object(i, j))
  80                                 {
  81                                         GL::PushMatrix push_mat3;
  82                                         GL::translate(bogies[i].axles[j].position, 0, bogies[i].axles[j].wheel_dia/2);
  83                                         obj->render(tag);
  84                                 }
  85
  86                         if(bogies[i].rotate_object)
  87                                 GL::rotate(180, 0, 0, 1);
  88                         if(const GL::Object *obj = type.get_bogie_object(i))
  89                                 obj->render(tag);
  90                 }
  91         }
  92 }
  93
  94 } // namespace Marklin