3 This file is part of the MSP Märklin suite
4 Copyright © 2006-2009 Mikkosoft Productions, Mikko Rasa
5 Distributed under the GPL
12 #include <msp/gl/rendermode.h>
13 #include <msp/gl/select.h>
14 #include <msp/gl/texture2d.h>
15 #include <msp/input/keys.h>
16 #include <msp/strings/codec.h>
17 #include <msp/strings/lexicalcast.h>
18 #include <msp/strings/utf8.h>
19 #include <msp/strings/utils.h>
20 #include <msp/time/units.h>
21 #include <msp/time/utils.h>
22 #include "libmarklin/tracktype.h"
25 #include "manipulator.h"
27 #include "selection.h"
30 using namespace Marklin;
33 Application::RegApp<Designer> Designer::reg;
35 Designer::Designer(int argc, char **argv):
43 cam_pos(0, -0.5, 0.5),
51 DataFile::load(catalogue, "tracks.dat");
53 cat_layout = new Layout(catalogue);
54 cat_layout_3d = new Layout3D(*cat_layout);
56 const map<unsigned, TrackType *> &ctracks = catalogue.get_tracks();
58 for(map<unsigned, TrackType *>::const_iterator i=ctracks.begin(); i!=ctracks.end(); ++i, ++n)
60 Track *track = new Track(*i->second);
61 track->set_position(Point((n%11)*0.1-0.5, 0.2-n/11*0.3, 0));
62 track->set_rotation(M_PI/2);
63 cat_layout->add_track(*track);
66 manipulator = new Manipulator(*this);
67 manipulator->signal_status.connect(sigc::mem_fun(this, &Designer::manipulation_status));
68 manipulator->signal_done.connect(sigc::mem_fun(this, &Designer::manipulation_done));
70 layout = new Layout(catalogue);
71 layout_3d = new Layout3D(*layout);
75 DataFile::load(*layout, argv[1]);
76 const list<Track3D *> <racks = layout_3d->get_tracks();
77 for(list<Track3D *>::const_iterator i=ltracks.begin(); i!=ltracks.end(); ++i)
79 if((*i)->get_track().get_sensor_id())
80 (*i)->set_color(GL::Color(1, 1, 0.5));
81 else if((*i)->get_track().get_turnout_id())
82 (*i)->set_color(GL::Color(0.5, 1, 1));
83 else if((*i)->get_track().get_flex())
84 (*i)->set_color(GL::Color(1, 0.5, 1));
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);
119 wnd->signal_close.connect(sigc::bind(sigc::mem_fun(this, &Designer::exit), 0));
120 wnd->signal_key_press.connect(sigc::mem_fun(this, &Designer::key_press));
121 wnd->signal_key_release.connect(sigc::mem_fun(this, &Designer::key_release));
122 wnd->signal_button_press.connect(sigc::mem_fun(this, &Designer::button_press));
123 wnd->signal_pointer_motion.connect(sigc::mem_fun(this, &Designer::pointer_motion));
127 glEnableClientState(GL_VERTEX_ARRAY);
128 glEnable(GL_DEPTH_TEST);
130 glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
131 glEnable(GL_CULL_FACE);
133 GL::Texture2D *font_tex = new GL::Texture2D;
134 font_tex->set_min_filter(GL::LINEAR);
135 font_tex->load_image("dejavu-20.png");
136 font = new GL::Font();
137 font->set_texture(*font_tex);
138 DataFile::load(*font, "dejavu-20.font");
155 void Designer::map_pointer_coords(int x, int y, float &gx, float &gy)
157 float cos_pitch = cos(cam_pitch);
158 float sin_pitch = sin(cam_pitch);
159 float cos_yaw = cos(cam_yaw);
160 float sin_yaw = sin(cam_yaw);
162 float rx = sin_yaw*0.55228;
163 float ry = -cos_yaw*0.55228;
165 float ux = cos_yaw*-sin_pitch*0.41421;
166 float uy = sin_yaw*-sin_pitch*0.41421;
167 float uz = cos_pitch*0.41421;
169 float xf = static_cast<float>(x)*2/screen_w-1;
170 float yf = static_cast<float>(y)*2/screen_h-1;
172 float vx = cos_yaw*cos_pitch + xf*rx + yf*ux;
173 float vy = sin_yaw*cos_pitch + xf*ry + yf*uy;
174 float vz = sin_pitch + yf*uz;
176 gx = cam_pos.x-vx*cam_pos.z/vz;
177 gy = cam_pos.y-vy*cam_pos.z/vz;
180 void Designer::tick()
182 const Msp::Time::TimeStamp t = Msp::Time::now();
183 float dt = (t-last_tick)/Msp::Time::sec;
190 cam_pos.x += cos(cam_yaw)*dt*move_y;
191 cam_pos.y += sin(cam_yaw)*dt*move_y;
195 cam_pos.x += sin(cam_yaw)*dt*move_x;
196 cam_pos.y += -cos(cam_yaw)*dt*move_x;
200 cam_pos.x += cos(cam_yaw)*cos(cam_pitch)*dt*zoom;
201 cam_pos.y += sin(cam_yaw)*cos(cam_pitch)*dt*zoom;
202 cam_pos.z += sin(cam_pitch)*dt*zoom;
206 float vx = cos(cam_yaw)*cos(cam_pitch);
207 float vy = sin(cam_yaw)*cos(cam_pitch);
208 float vz = sin(cam_pitch);
210 float gx = cam_pos.x-vx*cam_pos.z/vz;
211 float gy = cam_pos.y-vy*cam_pos.z/vz;
212 float d = sqrt(vx*vx+vy*vy)*cam_pos.z/vz;
214 cam_yaw += M_PI*dt*rotate;
220 cam_pos.x = gx+cos(cam_yaw)*d;
221 cam_pos.y = gy+sin(cam_yaw)*d;
225 cam_pitch += M_PI/2*dt*pitch;
226 if(cam_pitch>M_PI/12)
228 else if(cam_pitch<-M_PI/2)
232 if(tooltip_timeout && t>tooltip_timeout)
237 t3d = pick_track(pointer_x, pointer_y);
239 t3d = pick_track(pointer_x, pointer_y);
243 const Track &track = t3d->get_track();
244 const TrackType &ttype = track.get_type();
247 ss<<ttype.get_article_number()<<' '<<ttype.get_description();
249 ss<<" (slope "<<track.get_slope()/ttype.get_total_length()*100<<"%)";
250 if(track.get_turnout_id())
251 ss<<" (turnout "<<track.get_turnout_id()<<')';
252 else if(track.get_sensor_id())
253 ss<<" (sensor "<<track.get_sensor_id()<<')';
256 move_tooltip(pointer_x, pointer_y);
261 tooltip_timeout = Msp::Time::TimeStamp();
271 void Designer::key_press(unsigned code, unsigned mod, wchar_t ch)
273 unsigned key = Msp::Input::key_from_sys(code);
277 input->key_press(key, mod, ch);
281 if(key==Msp::Input::KEY_SHIFT_L || key==Msp::Input::KEY_SHIFT_R)
284 if(key==Msp::Input::KEY_N)
286 else if(key==Msp::Input::KEY_G)
288 manipulator->start_move();
291 else if(key==Msp::Input::KEY_R)
293 manipulator->start_rotate();
296 else if(key==Msp::Input::KEY_D)
298 manipulator->duplicate();
299 manipulator->start_move();
302 else if(key==Msp::Input::KEY_W)
304 input = new ::Input(*this, "Filename");
305 input->signal_cancel.connect(sigc::mem_fun(this, &Designer::input_dismiss));
306 input->signal_accept.connect(sigc::mem_fun(this, &Designer::save_accept));
309 else if(key==Msp::Input::KEY_PLUS)
310 selection->select_more();
311 else if(key==Msp::Input::KEY_L && (mod&1))
313 const set<Track *> &tracks = layout->get_tracks();
315 for(set<Track *>::const_iterator i=tracks.begin(); i!=tracks.end(); ++i)
316 len += (*i)->get_type().get_total_length();
317 cout<<"Total length: "<<len<<"m\n";
319 else if(key==Msp::Input::KEY_L)
320 selection->select_linked();
321 else if(key==Msp::Input::KEY_M)
326 else if(key==Msp::Input::KEY_Z)
328 manipulator->start_elevate();
331 else if(key==Msp::Input::KEY_ESC)
334 manipulator->cancel();
335 else if(mode==CATALOGUE)
340 else if(key==Msp::Input::KEY_X)
342 set<Track *> tracks = selection->get_tracks();
344 for(set<Track *>::iterator i=tracks.begin(); i!=tracks.end(); ++i)
346 layout->remove_track(**i);
350 else if(key==Msp::Input::KEY_F && (mod&1))
352 const set<Track *> &tracks = selection->get_tracks();
353 const set<Track *> <racks = layout->get_tracks();
354 for(set<Track *>::const_iterator i=tracks.begin(); i!=tracks.end(); ++i)
356 (*i)->set_flex(!(*i)->get_flex());
358 for(set<Track *>::const_iterator j=ltracks.begin(); j!=ltracks.end(); ++j)
360 (*i)->snap_to(**j, true);
362 Track3D &t3d = layout_3d->get_track(**i);
364 t3d.set_color(GL::Color(1, 0.5, 1));
366 t3d.set_color(GL::Color(1, 1, 1));
369 else if(key==Msp::Input::KEY_F)
370 manipulator->flatten();
371 else if(key==Msp::Input::KEY_E && (mod&1))
372 manipulator->even_slope(true);
373 else if(key==Msp::Input::KEY_E)
374 manipulator->even_slope();
375 else if(key==Msp::Input::KEY_T)
377 Track *track = selection->get_track();
378 if(selection->size()==1 && track->get_type().get_n_routes()>1)
381 ss<<track->get_turnout_id();
382 input = new ::Input(*this, "Turnout ID", ss.str());
383 input->signal_cancel.connect(sigc::mem_fun(this, &Designer::input_dismiss));
384 input->signal_accept.connect(sigc::mem_fun(this, &Designer::turnout_id_accept));
388 else if(key==Msp::Input::KEY_S)
390 const set<Track *> &tracks = selection->get_tracks();
393 for(set<Track *>::const_iterator i=tracks.begin(); i!=tracks.end(); ++i)
395 if((*i)->get_type().get_n_routes()==1)
397 if(static_cast<int>((*i)->get_sensor_id())!=id)
400 id = (*i)->get_sensor_id();
410 input = new ::Input(*this, "Sensor ID", ss.str());
411 input->signal_cancel.connect(sigc::mem_fun(this, &Designer::input_dismiss));
412 input->signal_accept.connect(sigc::mem_fun(this, &Designer::sensor_id_accept));
416 else if(key==Msp::Input::KEY_RIGHT)
418 else if(key==Msp::Input::KEY_LEFT)
420 else if(key==Msp::Input::KEY_UP)
422 else if(key==Msp::Input::KEY_DOWN)
424 else if(key==Msp::Input::KEY_INSERT)
426 else if(key==Msp::Input::KEY_PGUP)
428 else if(key==Msp::Input::KEY_HOME)
430 else if(key==Msp::Input::KEY_END)
432 else if(key==Msp::Input::KEY_DELETE)
434 else if(key==Msp::Input::KEY_PGDN)
438 void Designer::key_release(unsigned code, unsigned)
440 unsigned key = Msp::Input::key_from_sys(code);
445 if(key==Msp::Input::KEY_SHIFT_L || key==Msp::Input::KEY_SHIFT_R)
447 else if(key==Msp::Input::KEY_RIGHT || key==Msp::Input::KEY_LEFT)
449 else if(key==Msp::Input::KEY_UP || key==Msp::Input::KEY_DOWN)
451 else if(key==Msp::Input::KEY_INSERT || key==Msp::Input::KEY_PGUP)
453 else if(key==Msp::Input::KEY_HOME || key==Msp::Input::KEY_END)
455 else if(key==Msp::Input::KEY_DELETE || key==Msp::Input::KEY_PGDN)
459 void Designer::button_press(int x, int y, unsigned btn, unsigned)
464 map_pointer_coords(x, y, gx, gy);
470 Track3D *ctrack = pick_track(x, y);
473 Track *track = ctrack->get_track().copy();
474 track->set_position(Point(gx, gy, 0));
475 layout->add_track(*track);
478 selection->add_track(track);
486 else if(mode==SELECT)
490 Track3D *track = pick_track(x, y);
495 selection->toggle_track(&track->get_track());
499 else if(mode==MANIPULATE)
500 manipulator->button_press(x, y, gx, gy, btn);
501 else if(mode==MEASURE)
502 measure->button_press(x, y, gx, gy, btn);
505 void Designer::pointer_motion(int x, int y)
510 map_pointer_coords(x, y, gx, gy);
512 if(mode==SELECT || mode==CATALOGUE)
516 tooltip_timeout = Msp::Time::now()+100*Msp::Time::msec;
521 manipulator->pointer_motion(x, y, gx, gy);
522 measure->pointer_motion(x, y, gx, gy);
525 if(mode==MEASURE || mode==MANIPULATE)
529 void Designer::project_3d()
531 glMatrixMode(GL_PROJECTION);
533 glFrustum(-0.055228, 0.055228, -0.041421, 0.041421, 0.1, 10);
534 glMatrixMode(GL_MODELVIEW);
537 void Designer::apply_camera()
541 glTranslatef(0, 0, -1);
544 glRotatef(-cam_pitch*180/M_PI-90, 1, 0, 0);
545 glRotatef(90-cam_yaw*180/M_PI, 0, 0, 1);
546 glTranslatef(-cam_pos.x, -cam_pos.y, -cam_pos.z);
550 void Designer::render()
552 glClear(GL_COLOR_BUFFER_BIT|GL_DEPTH_BUFFER_BIT);
553 glEnable(GL_DEPTH_TEST);
558 cat_layout_3d->render();
563 GL::Texture::unbind();
566 layout_3d->render(true);
567 manipulator->render();
572 glMatrixMode(GL_PROJECTION);
574 glOrtho(0, screen_w, 0, screen_h, 0, 1);
575 glMatrixMode(GL_MODELVIEW);
578 glDisable(GL_DEPTH_TEST);
582 glTranslatef(tooltip_x, tooltip_y, 0);
583 glScalef(20, 20, 20);
584 float width = font->get_string_width(tooltip);
585 glColor4f(0, 0, 0, 0.5);
588 glVertex2f(width, 0);
589 glVertex2f(width, 1);
592 glColor4f(1, 1, 1, 1);
593 font->draw_string(tooltip);
600 Track3D *Designer::pick_track(int x, int y)
602 Layout3D *l = layout_3d;
606 float xx = (static_cast<float>(x-static_cast<int>(screen_w)/2)/screen_h)*0.82843;
607 float yy = (static_cast<float>(y)/screen_h-0.5)*0.82843;
608 float size = 4.0/screen_h*0.82843;
613 return l->pick_track(xx, yy, size);
616 void Designer::manipulation_status(const string &status)
621 void Designer::manipulation_done(bool)
626 void Designer::measure_changed()
628 float pard = measure->get_parallel_distance()*1000;
629 float perpd = measure->get_perpendicular_distance()*1000;
630 float d = sqrt(pard*pard+perpd*perpd);
631 float adiff = measure->get_angle_difference()*180/M_PI;
634 ss<<"Par "<<pard<<"mm - Perp "<<perpd<<"mm - Total "<<d<<"mm - Angle "<<adiff<<"°";
638 void Designer::measure_done()
643 void Designer::move_tooltip(int x, int y)
645 int w = static_cast<int>(font->get_string_width(tooltip)*20);
646 tooltip_x = max(min(static_cast<int>(screen_w)-w, x), 0);
647 tooltip_y = max(min(static_cast<int>(screen_h)-20, y), 0);
650 void Designer::save_accept()
652 layout->save(input->get_text());
657 void Designer::turnout_id_accept()
659 Track *track = selection->get_track();
660 unsigned id = lexical_cast<unsigned>(input->get_text());
661 track->set_turnout_id(id);
663 Track3D &t3d = layout_3d->get_track(*track);
665 t3d.set_color(GL::Color(0.5, 1, 1));
667 t3d.set_color(GL::Color(1, 1, 1));
672 void Designer::sensor_id_accept()
674 const set<Track *> &tracks = selection->get_tracks();
675 unsigned id = lexical_cast<unsigned>(input->get_text());
676 for(set<Track *>::const_iterator i=tracks.begin(); i!=tracks.end(); ++i)
678 (*i)->set_sensor_id(id);
680 Track3D &t3d = layout_3d->get_track(**i);
682 t3d.set_color(GL::Color(1, 1, 0.5));
684 t3d.set_color(GL::Color(1, 1, 1));
690 void Designer::input_dismiss()