source/3d/layout.cpp

   1 /* $Id$
   2
   3 This file is part of the MSP Märklin suite
   4 Copyright © 2006-2010 Mikkosoft Productions, Mikko Rasa
   5 Distributed under the GPL
   6 */
   7
   8 #include <algorithm>
   9 #include <limits>
  10 #include <msp/gl/clip.h>
  11 #include <msp/gl/matrix.h>
  12 #include <msp/gl/rendermode.h>
  13 #include <msp/gl/select.h>
  14 #include <msp/gl/texture.h>
  15 #include <msp/datafile/parser.h>
  16 #include "libmarklin/trafficmanager.h"
  17 #include "layout.h"
  18
  19 using namespace std;
  20 using namespace Msp;
  21
  22 namespace Marklin {
  23
  24 Layout3D::Layout3D(Layout &l):
  25         layout(l),
  26         catalogue(layout.get_catalogue())
  27 {
  28         layout.signal_track_added.connect(sigc::mem_fun(this, &Layout3D::track_added));
  29         layout.signal_track_removed.connect(sigc::mem_fun(this, &Layout3D::track_removed));
  30
  31         const set<Track *> &ltracks = layout.get_tracks();
  32         for(set<Track *>::iterator i=ltracks.begin(); i!=ltracks.end(); ++i)
  33                 track_added(**i);
  34 }
  35
  36 Layout3D::~Layout3D()
  37 {
  38         while(!tracks.empty())
  39                 delete tracks.front();
  40         while(!trains.empty())
  41                 delete trains.front();
  42 }
  43
  44 void Layout3D::add_track(Track3D &t)
  45 {
  46         tracks.push_back(&t);
  47 }
  48
  49 void Layout3D::remove_track(Track3D &t)
  50 {
  51         list<Track3D *>::iterator i = find(tracks.begin(), tracks.end(), &t);
  52         if(i!=tracks.end())
  53                 tracks.erase(i);
  54 }
  55
  56 Track3D &Layout3D::get_track(const Track &t) const
  57 {
  58         for(list<Track3D *>::const_iterator i=tracks.begin(); i!=tracks.end(); ++i)
  59                 if(&(*i)->get_track()==&t)
  60                         return **i;
  61
  62         throw KeyError("Unknown track");
  63 }
  64
  65 Track3D *Layout3D::pick_track(float x, float y, float size) const
  66 {
  67         vector<GL::SelectRecord> select_buf;
  68         GL::select_buffer(select_buf);
  69         GL::render_mode(GL::SELECT);
  70
  71         {
  72                 GL::PushMatrix push_mat;
  73                 GL::load_identity();
  74
  75                 GL::ClipPlane(1, 0, x-size, 0).apply_to(0);
  76                 GL::ClipPlane(-1, 0, -x-size, 0).apply_to(1);
  77                 GL::ClipPlane(0, 1, y-size, 0).apply_to(2);
  78                 GL::ClipPlane(0, -1, -y-size, 0).apply_to(3);
  79         }
  80
  81         scene.render(0);
  82
  83         GL::ClipPlane::disable(0);
  84         GL::ClipPlane::disable(1);
  85         GL::ClipPlane::disable(2);
  86         GL::ClipPlane::disable(3);
  87
  88         GL::render_mode(GL::RENDER);
  89         Track3D *track = 0;
  90         unsigned track_depth = numeric_limits<unsigned>::max();
  91         for(vector<GL::SelectRecord>::iterator i=select_buf.begin(); i!=select_buf.end(); ++i)
  92                 if(i->min_depth<track_depth)
  93                 {
  94                         track = reinterpret_cast<Track3D *>(i->names.back());
  95                         track_depth = i->min_depth;
  96                 }
  97
  98         return track;
  99 }
 100
 101 void Layout3D::set_traffic_manager(TrafficManager &tm)
 102 {
 103         tm.signal_train_added.connect(sigc::mem_fun(this, &Layout3D::train_added));
 104 }
 105
 106 void Layout3D::add_train(Train3D &t)
 107 {
 108         trains.push_back(&t);
 109 }
 110
 111 void Layout3D::remove_train(Train3D &t)
 112 {
 113         list<Train3D *>::iterator i = find(trains.begin(), trains.end(), &t);
 114         if(i!=trains.end())
 115                 trains.erase(i);
 116 }
 117
 118 Train3D &Layout3D::get_train(const Train &t) const
 119 {
 120         for(list<Train3D *>::const_iterator i=trains.begin(); i!=trains.end(); ++i)
 121                 if(&(*i)->get_train()==&t)
 122                         return **i;
 123
 124         throw KeyError("Unknown train");
 125 }
 126
 127 void Layout3D::track_added(Track &t)
 128 {
 129         new Track3D(*this, t);
 130 }
 131
 132 void Layout3D::track_removed(Track &t)
 133 {
 134         for(list<Track3D *>::iterator i=tracks.begin(); i!=tracks.end(); ++i)
 135                 if(&(*i)->get_track()==&t)
 136                 {
 137                         delete *i;
 138                         return;
 139                 }
 140 }
 141
 142 void Layout3D::train_added(Train &t)
 143 {
 144         new Train3D(*this, t);
 145 }
 146
 147 } // namespace Marklin