source/designer/designer.cpp

   1 /* $Id$
   2
   3 This file is part of R²C²
   4 Copyright © 2006-2010 Mikkosoft Productions, Mikko Rasa
   5 Distributed under the GPL
   6 */
   7
   8 #include <signal.h>
   9 #include <algorithm>
  10 #include <cmath>
  11 #include <msp/fs/utils.h>
  12 #include <msp/gl/blend.h>
  13 #include <msp/gl/framebuffer.h>
  14 #include <msp/gl/matrix.h>
  15 #include <msp/gl/misc.h>
  16 #include <msp/gl/projection.h>
  17 #include <msp/gl/rendermode.h>
  18 #include <msp/gl/select.h>
  19 #include <msp/gl/tests.h>
  20 #include <msp/gl/texture2d.h>
  21 #include <msp/input/keys.h>
  22 #include <msp/io/print.h>
  23 #include <msp/strings/codec.h>
  24 #include <msp/strings/lexicalcast.h>
  25 #include <msp/strings/utf8.h>
  26 #include <msp/strings/utils.h>
  27 #include <msp/time/units.h>
  28 #include <msp/time/utils.h>
  29 #include "libr2c2/route.h"
  30 #include "libr2c2/tracktype.h"
  31 #include "3d/path.h"
  32 #include "designer.h"
  33 #include "input.h"
  34 #include "layoutbar.h"
  35 #include "manipulator.h"
  36 #include "measure.h"
  37 #include "routebar.h"
  38 #include "selection.h"
  39 #include "svgexporter.h"
  40 #include "trackbar.h"
  41 #include "trackproperties.h"
  42
  43 using namespace std;
  44 using namespace R2C2;
  45 using namespace Msp;
  46
  47 Application::RegApp<Designer> Designer::reg;
  48
  49 Designer::Designer(int argc, char **argv):
  50         window(1280, 960),
  51         ui_res("r2c2.res"),
  52         root(ui_res, window),
  53         base_object(0),
  54         cur_route(0),
  55         mode(SELECT),
  56         manipulator(*this, root, selection),
  57         measure(*this),
  58         camera_ctl(*this, root, camera),
  59         track_wrap(*this, selection)
  60 {
  61         window.set_title("Railway Designer");
  62         window.signal_close.connect(sigc::bind(sigc::mem_fun(this, &Designer::exit), 0));
  63
  64         selection.signal_changed.connect(sigc::mem_fun(this, &Designer::selection_changed));
  65         manipulator.signal_status.connect(sigc::mem_fun(this, &Designer::manipulation_status));
  66         manipulator.signal_done.connect(sigc::mem_fun(this, &Designer::manipulation_done));
  67         measure.signal_changed.connect(sigc::mem_fun(this, &Designer::measure_changed));
  68         measure.signal_done.connect(sigc::mem_fun(this, &Designer::measure_done));
  69
  70         // Setup catalogue and layout
  71         DataFile::load(catalogue, "tracks.dat");
  72
  73         cat_layout_3d = new Layout3D(catalogue.get_layout());
  74
  75         layout = new Layout(catalogue);
  76         layout_3d = new Layout3D(*layout);
  77
  78         layout->signal_track_added.connect(sigc::mem_fun(this, &Designer::track_added));
  79         layout->signal_track_removed.connect(sigc::mem_fun(this, &Designer::track_removed));
  80
  81         if(argc>1)
  82         {
  83                 filename = argv[1];
  84                 DataFile::load(*layout, argv[1]);
  85
  86                 if(!layout->get_base().empty())
  87                 {
  88                         base_object = new GL::Object;
  89                         DataFile::load(*base_object, layout->get_base());
  90                 }
  91         }
  92
  93         // Setup OpenGL
  94         GL::enable(GL_CULL_FACE);
  95
  96         pipeline = new GL::Pipeline(window.get_width(), window.get_height(), false);
  97         pipeline->set_camera(&camera);
  98         pipeline->add_renderable_for_pass(layout_3d->get_scene(), 0);
  99         if(base_object)
 100                 pipeline->add_renderable(*base_object);
 101         pipeline->add_renderable_for_pass(track_wrap, "unlit");
 102         pipeline->add_renderable_for_pass(layout_3d->get_path_scene(), "unlit");
 103         pipeline->add_renderable_for_pass(layout_3d->get_endpoint_scene(), "unlit");
 104
 105         light.set_position(0, -0.259, 0.966, 0);
 106         lighting.attach(0, light);
 107
 108         GL::PipelinePass *pass = &pipeline->add_pass(0);
 109         pass->lighting = &lighting;
 110         pass->depth_test = &GL::DepthTest::lequal();
 111
 112         pass = &pipeline->add_pass("unlit");
 113         pass->depth_test = &GL::DepthTest::lequal();
 114         pass->blend = &GL::Blend::alpha();
 115
 116         pass = &pipeline->add_pass("blended");
 117         pass->lighting = &lighting;
 118         pass->depth_test = &GL::DepthTest::lequal();
 119         pass->blend = &GL::Blend::alpha();
 120
 121         camera.set_up_direction(GL::Vector3(0, 0, 1));
 122         camera.set_look_direction(GL::Vector3(0, 0.5, -0.866));
 123
 124         // Setup UI
 125         root.signal_key_press.connect(sigc::mem_fun(this, &Designer::key_press));
 126         root.signal_button_press.connect(sigc::mem_fun(this, &Designer::button_press));
 127         root.signal_pointer_motion.connect(sigc::mem_fun(this, &Designer::pointer_motion));
 128         root.signal_tooltip.connect(sigc::mem_fun(this, &Designer::tooltip));
 129
 130         toolbars.push_back(new Layoutbar(*this));
 131         toolbars.push_back(new Trackbar(*this));
 132         toolbars.push_back(new Routebar(*this));
 133         for(vector<Toolbar *>::iterator i=toolbars.begin(); i!=toolbars.end(); ++i)
 134         {
 135                 root.add(**i);
 136                 if(i!=toolbars.begin())
 137                         (*i)->expand(false);
 138                 (*i)->signal_expanded.connect(sigc::hide(sigc::mem_fun(this, &Designer::arrange_toolbars)));
 139         }
 140         arrange_toolbars();
 141
 142         GLtk::Panel *statusbar = new GLtk::Panel;
 143         root.add(*statusbar);
 144         statusbar->set_size(window.get_width(), 20);
 145         statusbar->set_focusable(false);
 146
 147         lbl_status = new GLtk::Label;
 148         statusbar->add(*lbl_status);
 149         lbl_status->set_geometry(GLtk::Geometry(20, 2, 300, 16));
 150
 151         overlay = new Overlay3D(window, camera, ui_res.get_default_font());
 152
 153         camera_ctl.view_all();
 154
 155         const Layout3D::TrackMap &tracks = layout_3d->get_tracks();
 156         for(Layout3D::TrackMap::const_iterator i=tracks.begin(); i!=tracks.end(); ++i)
 157                 update_track_icon(*i->second);
 158 }
 159
 160 Designer::~Designer()
 161 {
 162         delete overlay;
 163         delete pipeline;
 164         delete base_object;
 165         delete layout_3d;
 166         delete layout;
 167         delete cat_layout_3d;
 168 }
 169
 170 int Designer::main()
 171 {
 172         window.show();
 173
 174         mode = SELECT;
 175
 176         return Application::main();
 177 }
 178
 179 void Designer::save()
 180 {
 181         InputDialog *input = new InputDialog(*this, "Save layout", filename);
 182         input->signal_accept.connect(sigc::mem_fun(layout, &Layout::save));
 183 }
 184
 185 void Designer::quit()
 186 {
 187         exit(0);
 188 }
 189
 190 void Designer::new_track()
 191 {
 192         if(mode!=SELECT)
 193                 return;
 194
 195         mode = CATALOGUE;
 196         lbl_status->set_text("Select new track or press Esc to cancel");
 197 }
 198
 199 void Designer::erase_tracks()
 200 {
 201         if(mode!=SELECT)
 202                 return;
 203
 204         set<Track *> tracks = selection.get_tracks();
 205         selection.clear();
 206         for(set<Track *>::iterator i=tracks.begin(); i!=tracks.end(); ++i)
 207         {
 208                 overlay->clear(layout_3d->get_track(**i));
 209                 delete *i;
 210         }
 211 }
 212
 213 void Designer::track_properties()
 214 {
 215         if(mode!=SELECT || selection.empty())
 216                 return;
 217
 218         TrackProperties *track_prop = new TrackProperties(selection);
 219         root.add(*track_prop);
 220         const GLtk::Geometry &root_geom = root.get_geometry();
 221         const GLtk::Geometry &dlg_geom = track_prop->get_geometry();
 222         track_prop->set_position((root_geom.w-dlg_geom.w)/2, (root_geom.h-dlg_geom.h)/2);
 223
 224         track_prop->signal_response.connect(sigc::mem_fun(this, &Designer::track_properties_response));
 225 }
 226
 227 void Designer::extend_track()
 228 {
 229         if(mode!=SELECT)
 230                 return;
 231
 232         if(manipulator.start_extend())
 233                 mode = MANIPULATE;
 234 }
 235
 236 void Designer::connect_tracks()
 237 {
 238         if(mode!=SELECT)
 239                 return;
 240
 241         manipulator.connect();
 242 }
 243
 244 void Designer::flatten_tracks()
 245 {
 246         if(mode!=SELECT)
 247                 return;
 248
 249         manipulator.flatten();
 250 }
 251
 252 void Designer::rename_route()
 253 {
 254         if(mode!=SELECT || !cur_route)
 255                 return;
 256
 257         InputDialog *input = new InputDialog(*this, "Route name", cur_route->get_name());
 258         input->signal_accept.connect(sigc::mem_fun(this, &Designer::route_name_accept));
 259 }
 260
 261 void Designer::svg_export()
 262 {
 263         InputDialog *input = new InputDialog(*this, "SVG export", FS::basepart(filename)+".svg");
 264         input->signal_accept.connect(sigc::mem_fun(this, &Designer::svg_export_accept));
 265 }
 266
 267 void Designer::edit_route(Route *r)
 268 {
 269         cur_route = r;
 270         show_route(r);
 271 }
 272
 273 void Designer::add_selection_to_route()
 274 {
 275         if(!cur_route)
 276                 return;
 277
 278         try
 279         {
 280                 cur_route->add_tracks(selection.get_tracks());
 281         }
 282         catch(const Exception &e)
 283         {
 284                 lbl_status->set_text(e.what());
 285         }
 286
 287         show_route(cur_route);
 288 }
 289
 290 Point Designer::map_pointer_to_ground(int x, int y)
 291 {
 292         float xf = x*2.0/window.get_width()-1.0;
 293         float yf = y*2.0/window.get_height()-1.0;
 294         GL::Vector4 vec = camera.unproject(GL::Vector4(xf, yf, 0, 0));
 295         const GL::Vector3 &pos = camera.get_position();
 296
 297         return Point(pos.x-vec.x*pos.z/vec.z, pos.y-vec.y*pos.z/vec.z);
 298 }
 299
 300 void Designer::tick()
 301 {
 302         const Msp::Time::TimeStamp t = Msp::Time::now();
 303         float dt = (t-last_tick)/Msp::Time::sec;
 304         last_tick = t;
 305
 306         if(mode==MANIPULATE_DONE)
 307                 mode = SELECT;
 308
 309         window.tick();
 310         root.tick();
 311         camera_ctl.tick(dt);
 312
 313         for(list<Track *>::iterator i=new_tracks.begin(); i!=new_tracks.end(); ++i)
 314                 layout_3d->get_track(**i).get_path().set_mask(0);
 315         new_tracks.clear();
 316
 317         render();
 318
 319         window.swap_buffers();
 320 }
 321
 322 void Designer::key_press(unsigned key, unsigned mod, wchar_t)
 323 {
 324         key = Input::key_from_sys(key);
 325         mod = Input::mod_from_sys(mod);
 326
 327         if(key==Msp::Input::KEY_N && (mod&Input::MOD_SHIFT))
 328                 extend_track();
 329         else if(key==Msp::Input::KEY_N)
 330                 new_track();
 331         else if(key==Msp::Input::KEY_G)
 332         {
 333                 manipulator.start_move();
 334                 mode = MANIPULATE;
 335         }
 336         else if(key==Msp::Input::KEY_R)
 337         {
 338                 manipulator.start_rotate();
 339                 mode = MANIPULATE;
 340         }
 341         else if(key==Msp::Input::KEY_D)
 342         {
 343                 manipulator.duplicate();
 344                 manipulator.start_move();
 345                 mode = MANIPULATE;
 346         }
 347         else if(key==Msp::Input::KEY_W)
 348                 save();
 349         else if(key==Msp::Input::KEY_PLUS)
 350                 selection.select_more();
 351         else if(key==Msp::Input::KEY_L && (mod&Input::MOD_SHIFT))
 352                 selection.select_blocks();
 353         else if(key==Msp::Input::KEY_L)
 354                 selection.select_linked();
 355         else if(key==Msp::Input::KEY_M)
 356         {
 357                 measure.start();
 358                 mode = MEASURE;
 359         }
 360         else if(key==Msp::Input::KEY_Z)
 361         {
 362                 manipulator.start_elevate();
 363                 mode = MANIPULATE;
 364         }
 365         else if(key==Msp::Input::KEY_ESC)
 366         {
 367                 if(mode==MANIPULATE)
 368                         manipulator.cancel();
 369                 else if(mode==CATALOGUE)
 370                         mode = SELECT;
 371                 else
 372                         selection.clear();
 373         }
 374         else if(key==Msp::Input::KEY_X)
 375                 erase_tracks();
 376         else if(key==Msp::Input::KEY_F && (mod&Input::MOD_SHIFT))
 377         {
 378                 const set<Track *> &tracks = selection.get_tracks();
 379                 const set<Track *> &ltracks = layout->get_tracks();
 380                 for(set<Track *>::const_iterator i=tracks.begin(); i!=tracks.end(); ++i)
 381                 {
 382                         (*i)->set_flex(!(*i)->get_flex());
 383                         (*i)->break_links();
 384                         for(set<Track *>::const_iterator j=ltracks.begin(); j!=ltracks.end(); ++j)
 385                                 if(*j!=*i)
 386                                         (*i)->snap_to(**j, true);
 387
 388                         update_track_icon(layout_3d->get_track(**i));
 389                 }
 390         }
 391         else if(key==Msp::Input::KEY_F)
 392                 flatten_tracks();
 393         else if(key==Msp::Input::KEY_E && (mod&Input::MOD_SHIFT))
 394                 manipulator.even_slope(true);
 395         else if(key==Msp::Input::KEY_E)
 396                 manipulator.even_slope();
 397         else if(key==Msp::Input::KEY_A)
 398                 add_selection_to_route();
 399         else if(key==Msp::Input::KEY_C)
 400                 connect_tracks();
 401         else if(key==Msp::Input::KEY_V)
 402                 svg_export();
 403         else if(key==Msp::Input::KEY_P)
 404                 track_properties();
 405 }
 406
 407 void Designer::button_press(int x, int y, unsigned btn, unsigned mod)
 408 {
 409         y = window.get_height()-y-1;
 410         mod = Input::mod_from_sys(mod);
 411
 412         Point ground = map_pointer_to_ground(x, y);
 413
 414         if(mode==CATALOGUE)
 415         {
 416                 if(btn==1)
 417                 {
 418                         Track3D *ctrack = pick_track(x, y);
 419                         if(ctrack)
 420                         {
 421                                 Track *track = new Track(*layout, ctrack->get_track().get_type());
 422                                 track->set_position(ground);
 423
 424                                 selection.clear();
 425                                 selection.add_track(track);
 426
 427                                 mode = SELECT;
 428                         }
 429                 }
 430                 else
 431                         mode = SELECT;
 432         }
 433         else if(mode==SELECT)
 434         {
 435                 if(btn==1)
 436                 {
 437                         Track3D *track = pick_track(x, y);
 438                         if(track)
 439                         {
 440                                 if(!(mod&Input::MOD_SHIFT))
 441                                         selection.clear();
 442                                 selection.toggle_track(&track->get_track());
 443                         }
 444                 }
 445         }
 446         else if(mode==MEASURE)
 447                 measure.button_press(x, y, ground.x, ground.y, btn);
 448 }
 449
 450 void Designer::pointer_motion(int x, int y)
 451 {
 452         y = window.get_height()-y-1;
 453
 454         if(!root.get_child_at(x, y))
 455         {
 456                 Point ground = map_pointer_to_ground(x, y);
 457                 measure.pointer_motion(x, y, ground.x, ground.y);
 458         }
 459 }
 460
 461 void Designer::apply_camera()
 462 {
 463         if(mode==CATALOGUE)
 464         {
 465                 GL::matrix_mode(GL::PROJECTION);
 466                 GL::load_identity();
 467                 GL::frustum_centered(0.11046, 0.082843, 0.1, 10);
 468                 GL::matrix_mode(GL::MODELVIEW);
 469                 GL::load_identity();
 470                 GL::translate(0, 0, -1);
 471         }
 472         else
 473                 camera.apply();
 474 }
 475
 476 void Designer::render()
 477 {
 478         GL::Framebuffer::system().clear(GL::COLOR_BUFFER_BIT|GL::DEPTH_BUFFER_BIT);
 479
 480         if(mode==CATALOGUE)
 481         {
 482                 apply_camera();
 483                 cat_layout_3d->get_scene().render();
 484         }
 485         else
 486         {
 487                 pipeline->render_all();
 488                 GL::enable(GL_CULL_FACE);
 489                 {
 490                         GL::Bind bind_blend(GL::Blend::alpha());
 491                         overlay->render(0);
 492                 }
 493
 494                 GL::Bind bind_depth(GL::DepthTest::lequal());
 495                 if(mode==MEASURE)
 496                         measure.render();
 497         }
 498
 499         GL::matrix_mode(GL::PROJECTION);
 500         GL::load_identity();
 501         GL::ortho_bottomleft(window.get_width(), window.get_height());
 502         GL::matrix_mode(GL::MODELVIEW);
 503         GL::load_identity();
 504
 505         GL::disable(GL::DEPTH_TEST);
 506
 507         GL::Bind bind_blend(GL::Blend::alpha());
 508         root.render();
 509         GL::Texture::unbind();
 510 }
 511
 512 void Designer::track_added(Track &trk)
 513 {
 514         new_tracks.push_back(&trk);
 515 }
 516
 517 void Designer::track_removed(Track &trk)
 518 {
 519         list<Track *>::iterator i = find(new_tracks.begin(), new_tracks.end(), &trk);
 520         if(i!=new_tracks.end())
 521                 new_tracks.erase(i);
 522 }
 523
 524 Track3D *Designer::pick_track(int x, int y)
 525 {
 526         Layout3D *l = layout_3d;
 527         if(mode==CATALOGUE)
 528                 l = cat_layout_3d;
 529
 530         float xx = ((float(x)-window.get_width()/2)/window.get_height())*0.82843;
 531         float yy = (float(y)/window.get_height()-0.5)*0.82843;
 532         float size = 4.0/window.get_height()*0.82843;
 533
 534         apply_camera();
 535
 536         return l->pick_track(xx, yy, size);
 537 }
 538
 539 void Designer::update_track_icon(Track3D &track)
 540 {
 541         overlay->clear(track);
 542
 543         if(track.get_track().get_flex())
 544                 overlay->add_graphic(track, "flex");
 545
 546         if(unsigned sid = track.get_track().get_sensor_id())
 547         {
 548                 overlay->add_graphic(track, "sensor");
 549                 overlay->set_label(track, lexical_cast(sid));
 550         }
 551         else if(unsigned tid = track.get_track().get_turnout_id())
 552         {
 553                 if(tid<0x800)
 554                 {
 555                         overlay->add_graphic(track, "turnout");
 556                         overlay->set_label(track, lexical_cast(tid));
 557                 }
 558         }
 559 }
 560
 561 void Designer::selection_changed()
 562 {
 563         const set<Track *> &tracks = selection.get_tracks();
 564         if(tracks.empty())
 565                 lbl_status->set_text(string());
 566         else
 567         {
 568                 float len = 0;
 569                 for(set<Track *>::const_iterator i=tracks.begin(); i!=tracks.end(); ++i)
 570                         len += (*i)->get_type().get_total_length();
 571                 lbl_status->set_text(format("%.2fm of track selected\n", len));
 572         }
 573 }
 574
 575 void Designer::manipulation_status(const string &status)
 576 {
 577         lbl_status->set_text(status);
 578 }
 579
 580 void Designer::manipulation_done(bool)
 581 {
 582         mode = MANIPULATE_DONE;
 583         selection_changed();
 584 }
 585
 586 void Designer::measure_changed()
 587 {
 588         float pard = measure.get_parallel_distance()*1000;
 589         float perpd = measure.get_perpendicular_distance()*1000;
 590         float d = sqrt(pard*pard+perpd*perpd);
 591         float adiff = measure.get_angle_difference()*180/M_PI;
 592         string info = format("Par %.1fmm - Perp %.1fmm - Total %.1fmm - Angle %.1f°", pard, perpd, d, adiff);
 593         lbl_status->set_text(info);
 594 }
 595
 596 void Designer::measure_done()
 597 {
 598         mode = SELECT;
 599         selection_changed();
 600 }
 601
 602 void Designer::arrange_toolbars()
 603 {
 604         unsigned x = 0;
 605         for(vector<Toolbar *>::iterator i=toolbars.begin(); i!=toolbars.end(); ++i)
 606         {
 607                 const GLtk::Geometry &geom = (*i)->get_geometry();
 608                 (*i)->set_position(x, window.get_height()-geom.h);
 609                 x += geom.w;
 610         }
 611 }
 612
 613 void Designer::track_properties_response(int)
 614 {
 615         const set<Track *> &tracks = selection.get_tracks();
 616         for(set<Track *>::const_iterator i=tracks.begin(); i!=tracks.end(); ++i)
 617                 update_track_icon(layout_3d->get_track(**i));
 618 }
 619
 620 void Designer::route_name_accept(const string &text)
 621 {
 622         if(cur_route)
 623                 cur_route->set_name(text);
 624 }
 625
 626 void Designer::svg_export_accept(const string &text)
 627 {
 628         SvgExporter svg_exp(*layout);
 629         svg_exp.save(text);
 630 }
 631
 632 string Designer::tooltip(int x, int y)
 633 {
 634         if(Track3D *t3d = pick_track(x, y))
 635         {
 636                 const Track &track = t3d->get_track();
 637                 const TrackType &ttype = track.get_type();
 638                 string info = format("%d %s", ttype.get_article_number(), ttype.get_description());
 639                 if(mode!=CATALOGUE && abs(track.get_slope())>1e-4)
 640                         info += format(" (slope %.1f%%)", abs(track.get_slope()/ttype.get_total_length()*100));
 641                 if(track.get_turnout_id())
 642                         info += format(" (turnout %d)", track.get_turnout_id());
 643                 else if(track.get_sensor_id())
 644                         info += format(" (sensor %d)", track.get_sensor_id());
 645
 646                 return info;
 647         }
 648
 649         return string();
 650 }
 651
 652 void Designer::show_route(const Route *route)
 653 {
 654         const set<Track *> &ltracks = layout->get_tracks();
 655         for(set<Track *>::iterator i=ltracks.begin(); i!=ltracks.end(); ++i)
 656         {
 657                 Track3D &t3d = layout_3d->get_track(**i);
 658                 if(route && route->has_track(**i))
 659                 {
 660                         t3d.get_path().set_color(GL::Color(0.5, 0.8, 1.0));
 661                         if((*i)->get_type().is_turnout())
 662                         {
 663                                 unsigned tid = (*i)->get_turnout_id();
 664                                 int path = (tid ? route->get_turnout(tid) : -1);
 665                                 if(path>=0)
 666                                         t3d.get_path().set_path(path);
 667                                 else
 668                                         t3d.get_path().set_mask((*i)->get_type().get_paths());
 669                         }
 670                         else
 671                                 t3d.get_path().set_path(0);
 672                 }
 673                 else
 674                         t3d.get_path().set_mask(0);
 675         }
 676 }