3 This file is part of the MSP Märklin suite
4 Copyright © 2006-2009 Mikkosoft Productions, Mikko Rasa
5 Distributed under the GPL
11 #include <msp/gl/matrix.h>
12 #include <msp/gl/misc.h>
13 #include <msp/gl/projection.h>
14 #include <msp/gl/rendermode.h>
15 #include <msp/gl/select.h>
16 #include <msp/gl/tests.h>
17 #include <msp/gl/texture2d.h>
18 #include <msp/input/keys.h>
19 #include <msp/io/print.h>
20 #include <msp/strings/codec.h>
21 #include <msp/strings/lexicalcast.h>
22 #include <msp/strings/utf8.h>
23 #include <msp/strings/utils.h>
24 #include <msp/time/units.h>
25 #include <msp/time/utils.h>
26 #include "libmarklin/tracktype.h"
29 #include "manipulator.h"
31 #include "selection.h"
35 using namespace Marklin;
38 Application::RegApp<Designer> Designer::reg;
40 Designer::Designer(int argc, char **argv):
49 cam_pos(0, -0.5, 0.5),
57 DataFile::load(catalogue, "tracks.dat");
59 cat_layout = new Layout(catalogue);
60 cat_layout_3d = new Layout3D(*cat_layout);
62 const map<unsigned, TrackType *> &ctracks = catalogue.get_tracks();
64 for(map<unsigned, TrackType *>::const_iterator i=ctracks.begin(); i!=ctracks.end(); ++i, ++n)
66 Track *track = new Track(*i->second);
67 track->set_position(Point((n%11)*0.1-0.5, 0.2-n/11*0.3, 0));
68 track->set_rotation(M_PI/2);
69 cat_layout->add_track(*track);
72 manipulator = new Manipulator(*this);
73 manipulator->signal_status.connect(sigc::mem_fun(this, &Designer::manipulation_status));
74 manipulator->signal_done.connect(sigc::mem_fun(this, &Designer::manipulation_done));
76 layout = new Layout(catalogue);
77 layout_3d = new Layout3D(*layout);
82 DataFile::load(*layout, argv[1]);
83 const list<Track3D *> <racks = layout_3d->get_tracks();
84 for(list<Track3D *>::const_iterator i=ltracks.begin(); i!=ltracks.end(); ++i)
85 update_track_color(**i);
87 if(!layout->get_base().empty())
89 base_mesh = new GL::Mesh;
90 DataFile::load(*base_mesh, layout->get_base());
94 selection = new Selection;
95 manipulator->set_selection(selection);
97 measure = new Measure(*this);
98 measure->signal_changed.connect(sigc::mem_fun(this, &Designer::measure_changed));
99 measure->signal_done.connect(sigc::mem_fun(this, &Designer::measure_done));
102 Designer::~Designer()
109 delete cat_layout_3d;
115 dpy = new Graphics::Display;
116 wnd = new Graphics::Window(*dpy, screen_w, screen_h);
117 glc = new Graphics::GLContext(*wnd);
118 wnd->set_title("Railway Designer");
120 wnd->signal_close.connect(sigc::bind(sigc::mem_fun(this, &Designer::exit), 0));
123 glEnableClientState(GL_VERTEX_ARRAY);
124 glEnable(GL_DEPTH_TEST);
126 glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
127 glEnable(GL_CULL_FACE);
129 DataFile::load(ui_res, "marklin.res");
130 root = new GLtk::Root(ui_res, *wnd);
132 lbl_tooltip = new GLtk::Label(ui_res);
133 lbl_tooltip->set_style("tooltip");
134 root->add(*lbl_tooltip);
135 lbl_tooltip->set_visible(false);
137 toolbar = new Toolbar(*this);
139 toolbar->set_position(0, screen_h-toolbar->get_geometry().h);
140 toolbar->set_visible(true);
143 wnd->signal_key_press.connect(sigc::mem_fun(this, &Designer::key_press));
144 wnd->signal_key_release.connect(sigc::mem_fun(this, &Designer::key_release));
145 wnd->signal_button_press.connect(sigc::mem_fun(this, &Designer::button_press));
146 wnd->signal_pointer_motion.connect(sigc::mem_fun(this, &Designer::pointer_motion));
161 void Designer::save()
163 input = new ::Input(*this, "Save layout", filename);
164 input->signal_cancel.connect(sigc::mem_fun(this, &Designer::input_dismiss));
165 input->signal_accept.connect(sigc::mem_fun(this, &Designer::save_accept));
169 void Designer::quit()
174 void Designer::edit_route(Route &r)
179 void Designer::add_selection_to_route()
186 const set<Track *> &stracks = selection->get_tracks();
187 set<const Track *> tracks(stracks.begin(), stracks.end());
188 cur_route->add_tracks(tracks);
190 catch(const Exception &e)
192 IO::print("%s\n", e.what());
196 void Designer::map_pointer_coords(int x, int y, float &gx, float &gy)
198 float cos_pitch = cos(cam_pitch);
199 float sin_pitch = sin(cam_pitch);
200 float cos_yaw = cos(cam_yaw);
201 float sin_yaw = sin(cam_yaw);
203 float rx = sin_yaw*0.55228;
204 float ry = -cos_yaw*0.55228;
206 float ux = cos_yaw*-sin_pitch*0.41421;
207 float uy = sin_yaw*-sin_pitch*0.41421;
208 float uz = cos_pitch*0.41421;
210 float xf = static_cast<float>(x)*2/screen_w-1;
211 float yf = static_cast<float>(y)*2/screen_h-1;
213 float vx = cos_yaw*cos_pitch + xf*rx + yf*ux;
214 float vy = sin_yaw*cos_pitch + xf*ry + yf*uy;
215 float vz = sin_pitch + yf*uz;
217 gx = cam_pos.x-vx*cam_pos.z/vz;
218 gy = cam_pos.y-vy*cam_pos.z/vz;
221 void Designer::tick()
223 const Msp::Time::TimeStamp t = Msp::Time::now();
224 float dt = (t-last_tick)/Msp::Time::sec;
231 cam_pos.x += cos(cam_yaw)*dt*move_y;
232 cam_pos.y += sin(cam_yaw)*dt*move_y;
236 cam_pos.x += sin(cam_yaw)*dt*move_x;
237 cam_pos.y += -cos(cam_yaw)*dt*move_x;
241 cam_pos.x += cos(cam_yaw)*cos(cam_pitch)*dt*zoom;
242 cam_pos.y += sin(cam_yaw)*cos(cam_pitch)*dt*zoom;
243 cam_pos.z += sin(cam_pitch)*dt*zoom;
247 float vx = cos(cam_yaw)*cos(cam_pitch);
248 float vy = sin(cam_yaw)*cos(cam_pitch);
249 float vz = sin(cam_pitch);
251 float gx = cam_pos.x-vx*cam_pos.z/vz;
252 float gy = cam_pos.y-vy*cam_pos.z/vz;
253 float d = sqrt(vx*vx+vy*vy)*cam_pos.z/vz;
255 cam_yaw += M_PI*dt*rotate;
261 cam_pos.x = gx+cos(cam_yaw)*d;
262 cam_pos.y = gy+sin(cam_yaw)*d;
266 cam_pitch += M_PI/2*dt*pitch;
267 if(cam_pitch>M_PI/12)
269 else if(cam_pitch<-M_PI/2)
273 if(tooltip_timeout && t>tooltip_timeout)
278 t3d = pick_track(pointer_x, pointer_y);
280 t3d = pick_track(pointer_x, pointer_y);
284 const Track &track = t3d->get_track();
285 const TrackType &ttype = track.get_type();
286 string info = format("%d %s", ttype.get_article_number(), ttype.get_description());
287 if(mode!=CATALOGUE && abs(track.get_slope())>1e-4)
288 info += format(" (slope %.1f%%)", abs(track.get_slope()/ttype.get_total_length()*100));
289 if(track.get_turnout_id())
290 info += format(" (turnout %d)", track.get_turnout_id());
291 else if(track.get_sensor_id())
292 info += format(" (sensor %d)", track.get_sensor_id());
294 set_tooltip(pointer_x, pointer_y, info);
299 tooltip_timeout = Msp::Time::TimeStamp();
307 void Designer::key_press(unsigned code, unsigned mod, wchar_t)
309 unsigned key = Msp::Input::key_from_sys(code);
314 if(key==Msp::Input::KEY_SHIFT_L || key==Msp::Input::KEY_SHIFT_R)
317 if(key==Msp::Input::KEY_N)
319 else if(key==Msp::Input::KEY_G)
321 manipulator->start_move();
324 else if(key==Msp::Input::KEY_R)
326 manipulator->start_rotate();
329 else if(key==Msp::Input::KEY_D)
331 manipulator->duplicate();
332 manipulator->start_move();
335 else if(key==Msp::Input::KEY_W)
337 else if(key==Msp::Input::KEY_PLUS)
338 selection->select_more();
339 else if(key==Msp::Input::KEY_L && (mod&1))
341 const set<Track *> &tracks = layout->get_tracks();
343 for(set<Track *>::const_iterator i=tracks.begin(); i!=tracks.end(); ++i)
344 len += (*i)->get_type().get_total_length();
345 IO::print("Total length: %.1fm\n", len);
347 else if(key==Msp::Input::KEY_L)
348 selection->select_linked();
349 else if(key==Msp::Input::KEY_M)
354 else if(key==Msp::Input::KEY_Z)
356 manipulator->start_elevate();
359 else if(key==Msp::Input::KEY_ESC)
362 manipulator->cancel();
363 else if(mode==CATALOGUE)
368 else if(key==Msp::Input::KEY_X)
370 set<Track *> tracks = selection->get_tracks();
372 for(set<Track *>::iterator i=tracks.begin(); i!=tracks.end(); ++i)
374 layout->remove_track(**i);
378 else if(key==Msp::Input::KEY_F && (mod&1))
380 const set<Track *> &tracks = selection->get_tracks();
381 const set<Track *> <racks = layout->get_tracks();
382 for(set<Track *>::const_iterator i=tracks.begin(); i!=tracks.end(); ++i)
384 (*i)->set_flex(!(*i)->get_flex());
386 for(set<Track *>::const_iterator j=ltracks.begin(); j!=ltracks.end(); ++j)
388 (*i)->snap_to(**j, true);
390 update_track_color(layout_3d->get_track(**i));
393 else if(key==Msp::Input::KEY_F)
394 manipulator->flatten();
395 else if(key==Msp::Input::KEY_E && (mod&1))
396 manipulator->even_slope(true);
397 else if(key==Msp::Input::KEY_E)
398 manipulator->even_slope();
399 else if(key==Msp::Input::KEY_T)
401 Track *track = selection->get_track();
402 if(selection->size()==1 && track->get_type().get_n_paths()>1)
404 input = new ::Input(*this, "Turnout ID", lexical_cast(track->get_turnout_id()));
405 input->signal_cancel.connect(sigc::mem_fun(this, &Designer::input_dismiss));
406 input->signal_accept.connect(sigc::mem_fun(this, &Designer::turnout_id_accept));
410 else if(key==Msp::Input::KEY_S)
412 const set<Track *> &tracks = selection->get_tracks();
415 for(set<Track *>::const_iterator i=tracks.begin(); i!=tracks.end(); ++i)
417 if((*i)->get_type().get_n_paths()==1)
419 if(static_cast<int>((*i)->get_sensor_id())!=id)
422 id = (*i)->get_sensor_id();
429 input = new ::Input(*this, "Sensor ID", (id>=0 ? lexical_cast(id) : string()));
430 input->signal_cancel.connect(sigc::mem_fun(this, &Designer::input_dismiss));
431 input->signal_accept.connect(sigc::mem_fun(this, &Designer::sensor_id_accept));
435 else if(key==Msp::Input::KEY_A)
436 add_selection_to_route();
437 else if(key==Msp::Input::KEY_RIGHT)
439 else if(key==Msp::Input::KEY_LEFT)
441 else if(key==Msp::Input::KEY_UP)
443 else if(key==Msp::Input::KEY_DOWN)
445 else if(key==Msp::Input::KEY_INSERT)
447 else if(key==Msp::Input::KEY_PGUP)
449 else if(key==Msp::Input::KEY_HOME)
451 else if(key==Msp::Input::KEY_END)
453 else if(key==Msp::Input::KEY_DELETE)
455 else if(key==Msp::Input::KEY_PGDN)
459 void Designer::key_release(unsigned code, unsigned)
461 unsigned key = Msp::Input::key_from_sys(code);
466 if(key==Msp::Input::KEY_SHIFT_L || key==Msp::Input::KEY_SHIFT_R)
468 else if(key==Msp::Input::KEY_RIGHT || key==Msp::Input::KEY_LEFT)
470 else if(key==Msp::Input::KEY_UP || key==Msp::Input::KEY_DOWN)
472 else if(key==Msp::Input::KEY_INSERT || key==Msp::Input::KEY_PGUP)
474 else if(key==Msp::Input::KEY_HOME || key==Msp::Input::KEY_END)
476 else if(key==Msp::Input::KEY_DELETE || key==Msp::Input::KEY_PGDN)
480 void Designer::button_press(int x, int y, unsigned btn, unsigned)
485 map_pointer_coords(x, y, gx, gy);
491 Track3D *ctrack = pick_track(x, y);
494 Track *track = ctrack->get_track().copy();
495 track->set_position(Point(gx, gy, 0));
496 layout->add_track(*track);
499 selection->add_track(track);
507 else if(mode==SELECT)
511 Track3D *track = pick_track(x, y);
516 selection->toggle_track(&track->get_track());
520 else if(mode==MANIPULATE)
521 manipulator->button_press(x, y, gx, gy, btn);
522 else if(mode==MEASURE)
523 measure->button_press(x, y, gx, gy, btn);
526 void Designer::pointer_motion(int x, int y)
533 if(mode==SELECT || mode==CATALOGUE)
534 tooltip_timeout = Msp::Time::now()+100*Msp::Time::msec;
541 map_pointer_coords(x, y, gx, gy);
542 manipulator->pointer_motion(x, y, gx, gy);
543 measure->pointer_motion(x, y, gx, gy);
547 void Designer::project_3d()
549 glMatrixMode(GL_PROJECTION);
551 glFrustum(-0.055228, 0.055228, -0.041421, 0.041421, 0.1, 10);
552 glMatrixMode(GL_MODELVIEW);
555 void Designer::apply_camera()
559 glTranslatef(0, 0, -1);
562 glRotatef(-cam_pitch*180/M_PI-90, 1, 0, 0);
563 glRotatef(90-cam_yaw*180/M_PI, 0, 0, 1);
564 glTranslatef(-cam_pos.x, -cam_pos.y, -cam_pos.z);
568 void Designer::render()
570 glClear(GL_COLOR_BUFFER_BIT|GL_DEPTH_BUFFER_BIT);
571 GL::enable(GL::DEPTH_TEST);
576 cat_layout_3d->render();
581 GL::Texture::unbind();
584 layout_3d->render(true);
587 glColor4f(0.5, 0.8, 1.0, 1.0);
588 const set<const Track *> &rtracks = cur_route->get_tracks();
589 const map<unsigned, int> &turnouts = cur_route->get_turnouts();
590 for(set<const Track *>::const_iterator i=rtracks.begin(); i!=rtracks.end(); ++i)
593 if(unsigned tid=(*i)->get_turnout_id())
595 map<unsigned, int>::const_iterator j = turnouts.find(tid);
596 if(j!=turnouts.end())
599 layout_3d->get_track(**i).render_path(path);
602 manipulator->render();
607 GL::matrix_mode(GL::PROJECTION);
609 GL::ortho_bottomleft(screen_w, screen_h);
610 GL::matrix_mode(GL::MODELVIEW);
613 GL::disable(GL::DEPTH_TEST);
618 Track3D *Designer::pick_track(int x, int y)
620 Layout3D *l = layout_3d;
624 float xx = (static_cast<float>(x-static_cast<int>(screen_w)/2)/screen_h)*0.82843;
625 float yy = (static_cast<float>(y)/screen_h-0.5)*0.82843;
626 float size = 4.0/screen_h*0.82843;
631 return l->pick_track(xx, yy, size);
634 void Designer::update_track_color(Track3D &track)
636 if(track.get_track().get_sensor_id())
638 if(track.get_track().get_flex())
639 track.set_color(GL::Color(1, 0.6, 1));
641 track.set_color(GL::Color(0.7, 0.7, 1));
643 else if(track.get_track().get_turnout_id())
644 track.set_color(GL::Color(0.8, 1, 0.8));
645 else if(track.get_track().get_flex())
646 track.set_color(GL::Color(1, 0.8, 0.8));
648 track.set_color(GL::Color(1, 1, 1));
651 void Designer::manipulation_status(const string &status)
653 set_tooltip(pointer_x, pointer_y, status);
656 void Designer::manipulation_done(bool)
661 void Designer::measure_changed()
663 float pard = measure->get_parallel_distance()*1000;
664 float perpd = measure->get_perpendicular_distance()*1000;
665 float d = sqrt(pard*pard+perpd*perpd);
666 float adiff = measure->get_angle_difference()*180/M_PI;
667 string info = format("Par %.1fmm - Perp %.1fmm - Total %.1fmm - Angle %.1f°", pard, perpd, d, adiff);
668 set_tooltip(pointer_x, pointer_y, info);
671 void Designer::measure_done()
676 void Designer::set_tooltip(int x, int y, const std::string &text)
678 int fontsize = ui_res.get_default_font().get_default_size();
680 int w = static_cast<int>(ui_res.get_default_font().get_string_width(text)*fontsize)+6;
681 x = max(min(static_cast<int>(screen_w)-w, x), 0);
682 y = max(min(static_cast<int>(screen_h)-h, y), 0);
683 lbl_tooltip->set_text(text);
684 lbl_tooltip->set_geometry(GLtk::Geometry(x, y, w, h));
685 lbl_tooltip->set_visible(true);
688 void Designer::clear_tooltip()
690 lbl_tooltip->set_visible(false);
693 void Designer::save_accept()
695 layout->save(input->get_text());
700 void Designer::turnout_id_accept()
702 Track *track = selection->get_track();
703 unsigned id = lexical_cast<unsigned>(input->get_text());
704 track->set_turnout_id(id);
706 update_track_color(layout_3d->get_track(*track));
711 void Designer::sensor_id_accept()
713 const set<Track *> &tracks = selection->get_tracks();
714 unsigned id = lexical_cast<unsigned>(input->get_text());
715 for(set<Track *>::const_iterator i=tracks.begin(); i!=tracks.end(); ++i)
717 (*i)->set_sensor_id(id);
719 update_track_color(layout_3d->get_track(**i));
725 void Designer::input_dismiss()