source/3d/catalogue.cpp

   1 #include <msp/fs/stat.h>
   2 #include <msp/gl/meshbuilder.h>
   3 #include <msp/gl/object.h>
   4 #include <msp/gl/program.h>
   5 #include <msp/gl/technique.h>
   6 #include "catalogue.h"
   7 #include "tracktype.h"
   8 #include "vehicletype.h"
   9
  10 using namespace std;
  11 using namespace Msp;
  12
  13 namespace R2C2 {
  14
  15 Catalogue3D::Catalogue3D(Catalogue &c):
  16         catalogue(c),
  17         endpoint_mesh((GL::NORMAL3, GL::VERTEX3))
  18 {
  19         add_creator<GL::Material>(&Catalogue3D::create<GL::Material>);
  20         add_creator<GL::Mesh>(&Catalogue3D::create<GL::Mesh>);
  21         add_creator<GL::Object>(&Catalogue3D::create2<GL::Object>);
  22         add_creator<GL::Program>(&Catalogue3D::create<GL::Program>);
  23         add_creator<GL::Technique>(&Catalogue3D::create2<GL::Technique>);
  24
  25         catalogue.signal_track_added.connect(sigc::mem_fun(this, &Catalogue3D::track_added));
  26         catalogue.signal_vehicle_added.connect(sigc::mem_fun(this, &Catalogue3D::vehicle_added));
  27
  28         const Catalogue::TrackMap &trks = catalogue.get_tracks();
  29         for(Catalogue::TrackMap::const_iterator i=trks.begin(); i!=trks.end(); ++i)
  30                 track_added(*i->second);
  31
  32         build_endpoint_mesh();
  33 }
  34
  35 Catalogue3D::~Catalogue3D()
  36 {
  37         for(map<const TrackType *, TrackType3D *>::iterator i=tracks.begin(); i!=tracks.end(); ++i)
  38                 delete i->second;
  39         for(map<const VehicleType *, VehicleType3D *>::iterator i=vehicles.begin(); i!=vehicles.end(); ++i)
  40                 delete i->second;
  41 }
  42
  43 const TrackType3D &Catalogue3D::get_track(const TrackType &tt) const
  44 {
  45         map<const TrackType *, TrackType3D *>::const_iterator i = tracks.find(&tt);
  46         if(i==tracks.end())
  47                 throw KeyError("Unknown track type");
  48
  49         return *i->second;
  50 }
  51
  52 const VehicleType3D &Catalogue3D::get_vehicle(const VehicleType &vt) const
  53 {
  54         map<const VehicleType *, VehicleType3D *>::const_iterator i = vehicles.find(&vt);
  55         if(i==vehicles.end())
  56                 throw KeyError("Unknown vehicle type");
  57
  58         return *i->second;
  59 }
  60
  61 void Catalogue3D::track_added(const TrackType &track)
  62 {
  63         tracks[&track] = new TrackType3D(*this, track);
  64 }
  65
  66 void Catalogue3D::vehicle_added(const VehicleType &veh)
  67 {
  68         vehicles[&veh] = new VehicleType3D(*this, veh);
  69 }
  70
  71 void Catalogue3D::build_endpoint_mesh()
  72 {
  73         const Profile &ballast_profile = catalogue.get_ballast_profile();
  74         const Vector &ballast_min = ballast_profile.get_min_coords();
  75         const Vector &ballast_max = ballast_profile.get_max_coords();
  76
  77         const Profile &rail_profile = catalogue.get_rail_profile();
  78         const Vector &rail_min = rail_profile.get_min_coords();
  79         const Vector &rail_max = rail_profile.get_max_coords();
  80
  81         float gauge = catalogue.get_gauge();
  82
  83         float width = max(max(-ballast_min.x, ballast_max.x)*2, gauge+(rail_max.x-rail_min.x)*2)+0.004;
  84         float height = ballast_max.y-ballast_min.y+rail_max.y-rail_min.y+0.01;
  85
  86         GL::MeshBuilder bld(endpoint_mesh);
  87         bld.normal(1, 0, 0);
  88         bld.begin(GL::QUADS);
  89         bld.vertex(0, width/2, 0);
  90         bld.vertex(0, width/2, height);
  91         bld.vertex(0, -width/2, height);
  92         bld.vertex(0, -width/2, 0);
  93         bld.end();
  94 }
  95
  96 FS::Path Catalogue3D::locate_file(const string &name)
  97 {
  98         if(FS::exists(name))
  99                 return name;
 100
 101         FS::Path path = FS::Path("data")/name;
 102         if(FS::exists(path))
 103                 return path;
 104
 105         throw Exception("Can't locate "+name);
 106 }
 107
 108 template<typename T>
 109 T *Catalogue3D::create(const string &name)
 110 {
 111         RefPtr<T> obj = new T;
 112         DataFile::load(*obj, locate_file(name).str());
 113         return obj.release();
 114 }
 115
 116 template<typename T>
 117 T *Catalogue3D::create2(const string &name)
 118 {
 119         RefPtr<T> obj = new T;
 120         DataFile::load(*obj, locate_file(name).str(), *this);
 121         return obj.release();
 122 }
 123
 124 } // namespace R2C2