]> git.tdb.fi Git - r2c2.git/blobdiff - source/engineer/engineer.cpp
Foundations of using physics simulation for trains
[r2c2.git] / source / engineer / engineer.cpp
index 371e82e49e7c4d4fc6b69315be3e0ca1b355d4a6..92f0f0f419e451c98676597f9954b216c0717b42 100644 (file)
@@ -23,9 +23,12 @@ Distributed under the GPL
 #include <msp/io/print.h>
 #include <msp/strings/formatter.h>
 #include <msp/time/units.h>
+#include <msp/time/utils.h>
 #include "libmarklin/driver.h"
 #include "libmarklin/tracktype.h"
 #include "3d/path.h"
+#include "3d/track.h"
+#include "3d/vehicle.h"
 #include "engineer.h"
 #include "mainpanel.h"
 #include "trainpanel.h"
@@ -44,9 +47,10 @@ Engineer::Engineer(int argc, char **argv):
        layout_3d(layout),
        server(0),
        pipeline(window.get_width(), window.get_height(), false),
-       placing_train(0),
-       placing_block(0),
-       placing_entry(0)
+       picking(false),
+       picking_track(0),
+       picking_entry(0),
+       picking_path(0)
 {
        // Setup GUI
        window.set_title("Railroad Engineer");
@@ -72,6 +76,7 @@ Engineer::Engineer(int argc, char **argv):
 
        layout.signal_train_added.connect(sigc::mem_fun(this, &Engineer::train_added));
        layout.signal_block_reserved.connect(sigc::mem_fun(this, &Engineer::block_reserved));
+       layout.get_driver().signal_sensor.connect(sigc::mem_fun(this, &Engineer::sensor_event));
        if(FS::exists("engineer.state"))
                DataFile::load(layout, "engineer.state");
 
@@ -81,8 +86,6 @@ Engineer::Engineer(int argc, char **argv):
                server->use_event_dispatcher(event_disp);
        }
 
-       layout.get_driver().signal_sensor.connect(sigc::mem_fun(this, &Engineer::sensor_event));
-
        // Setup 3D view
        DataFile::load(arrow_mesh, "arrow.mesh");
 
@@ -111,7 +114,7 @@ Engineer::~Engineer()
 {
        const map<unsigned, Train *> &trains = layout.get_trains();
        for(map<unsigned, Train *>::const_iterator i=trains.begin(); i!=trains.end(); ++i)
-               i->second->set_speed(0);
+               layout.get_driver().set_loco_speed(i->first, 0);
        layout.get_driver().flush();
 
        if(!options.simulate)
@@ -123,11 +126,17 @@ Engineer::~Engineer()
        delete server;
 }
 
-void Engineer::place_train(Train &train)
+void Engineer::set_status(const string &text)
 {
-       placing_train = &train;
-       placing_block = 0;
-       main_panel->set_status_text("Select location");
+       main_panel->set_status_text(text);
+       status_timeout = Time::now()+10*Time::sec;
+}
+
+void Engineer::pick(bool with_ep)
+{
+       picking = true;
+       picking_track = 0;
+       picking_entry = (with_ep ? 0 : -1);
 }
 
 int Engineer::main()
@@ -144,22 +153,30 @@ void Engineer::tick()
        layout.tick();
        event_disp.tick(Time::zero);
 
+       if(status_timeout && Time::now()>status_timeout)
+       {
+               main_panel->set_status_text(string());
+               status_timeout = Time::TimeStamp();
+       }
+
        GL::clear(GL::COLOR_BUFFER_BIT|GL::DEPTH_BUFFER_BIT);
 
        pipeline.render_all();
-       layout_3d.get_path_scene().render();
+       {
+               GL::Bind depth(GL::DepthTest::lequal());
+               layout_3d.get_path_scene().render();
+       }
        {
                GL::Bind blend(GL::Blend::alpha());
                overlay->render(0);
        }
 
-       if(placing_train && placing_block)
+       if(picking && picking_track && picking_entry>=0)
        {
                GL::PushMatrix push_mat;
 
-               const Marklin::Block::Endpoint &bep = placing_block->get_endpoints()[placing_entry];
-               float rot = bep.track->get_endpoint_direction(bep.track_ep);
-               Point pos = bep.track->get_endpoint_position(bep.track_ep);
+               float rot = picking_track->get_endpoint_direction(picking_entry);
+               Point pos = picking_track->get_endpoint_position(picking_entry);
 
                GL::translate(pos.x, pos.y, pos.z+0.03);
                GL::rotate(rot*180/M_PI+180, 0, 0, 1);
@@ -185,20 +202,19 @@ void Engineer::tick()
 
 void Engineer::button_press(int x, int y, unsigned btn, unsigned)
 {
-       if(placing_train)
+       if(picking)
        {
-               if(btn==1 && placing_block && !placing_block->get_train())
+               if(btn==1 && picking_track)
                {
-                       reset_block_color(*placing_block);
-
-                       placing_train->place(*placing_block, placing_entry);
-                       placing_train = 0;
-                       main_panel->set_status_text(string());
+                       picking = false;
+                       delete picking_path;
+                       picking_path = 0;
+                       signal_pick_done.emit(picking_track, picking_entry);
                }
-               else if(btn==3)
+               else if(btn==3 && picking_entry>=0)
                {
-                       const vector<Block::Endpoint> &endpoints = placing_block->get_endpoints();
-                       placing_entry = (placing_entry+1)%endpoints.size();
+                       picking_entry = (picking_entry+1)%picking_track->get_type().get_endpoints().size();
+                       picking_path->set_mask(picking_track->get_type().get_endpoints()[picking_entry].paths);
                }
        }
        else
@@ -211,7 +227,7 @@ void Engineer::button_press(int x, int y, unsigned btn, unsigned)
                        {
                                Block &block = layout.get_block_by_track(track);
                                if(block.get_train() && !block.get_train()->free_block(block))
-                                       main_panel->set_status_text("Turnout busy");
+                                       set_status("Turnout is busy");
                                else
                                {
                                        unsigned paths = track.get_type().get_paths();
@@ -226,35 +242,34 @@ void Engineer::button_press(int x, int y, unsigned btn, unsigned)
                                        track.set_active_path(i);
                                }
                        }
-                       /*else if(simulate)
+                       else if(options.simulate)
                        {
-                               if(unsigned sid=track->get_track().get_sensor_id())
-                               {
-                                       Sensor &sensor = control.get_sensor(sid);
-                                       control.signal_sensor_event.emit(sid, !sensor.get_state());
-                               }
-                       }*/
+                               if(unsigned sid=track.get_sensor_id())
+                                       layout.get_driver().set_sensor(sid, !layout.get_driver().get_sensor(sid));
+                       }
                }
        }
 }
 
 void Engineer::pointer_motion(int x, int y)
 {
-       if(placing_train)
+       if(picking)
        {
                Track3D *track = pick_track(x, window.get_height()-y-1);
-               if(track)
+               if(track && &track->get_track()!=picking_track)
                {
-                       Block &block = layout.get_block_by_track(track->get_track());
-                       if(&block!=placing_block)
-                       {
-                               if(placing_block)
-                                       reset_block_color(*placing_block);
-                               placing_block = &block;
-                               placing_entry = 0;
-                               if(!placing_block->get_train())
-                                       set_block_color(*placing_block, GL::Color(0));
-                       }
+                       picking_track = &track->get_track();
+                       if(picking_entry>=0)
+                               picking_entry = 0;
+
+                       delete picking_path;
+                       picking_path = new Path3D(*track);
+                       if(picking_entry>=0)
+                               picking_path->set_mask(picking_track->get_type().get_endpoints()[picking_entry].paths);
+                       else
+                               picking_path->set_mask(picking_track->get_type().get_paths());
+                       picking_path->set_color(GL::Color(0));
+                       picking_path->set_layer(1);
                }
        }
 }
@@ -308,6 +323,7 @@ void Engineer::view_all()
        camera.set_position(pos);
        camera.set_up_direction(up);
        camera.set_look_direction(GL::Vector3(0, 0, -1));
+       camera.set_depth_clip(pos.z*0.5, pos.z*1.5);
 }
 
 void Engineer::set_block_color(const Block &block, const GL::Color &color)
@@ -377,9 +393,9 @@ void Engineer::train_added(Train &train)
        train_panels.push_back(tpanel);
        tpanel->set_visible(true);
 
-       Train3D &t3d = layout_3d.get_train(train);
-       overlay->set_label(t3d, train.get_name());
-       train.signal_name_changed.connect(sigc::bind<0>(sigc::mem_fun(overlay, &Overlay3D::set_label), sigc::ref(t3d)));
+       Vehicle3D &loco3d = layout_3d.get_vehicle(train.get_vehicle(0));
+       overlay->set_label(loco3d, train.get_name());
+       train.signal_name_changed.connect(sigc::bind<0>(sigc::mem_fun(overlay, &Overlay3D::set_label), sigc::ref(loco3d)));
 
        GL::Color best_color;
        float best_d_sq = 0;
@@ -417,7 +433,7 @@ void Engineer::sighandler(int sig)
                IO::print(IO::cerr, "Fatal signal received, terminating\n");
                const map<unsigned, Train *> &trains = layout.get_trains();
                for(map<unsigned, Train *>::const_iterator i=trains.begin(); i!=trains.end(); ++i)
-                       i->second->set_speed(0);
+                       layout.get_driver().set_loco_speed(i->first, 0);
                layout.get_driver().flush();
                raise(sig);
        }