Measure and display train speeds

[r2c2.git] / source / engineer / engineer.cpp

diff --git a/source/engineer/engineer.cpp b/source/engineer/engineer.cpp
index 3ac7c7682a605cc06942c23cfee7607db157259c..ec5610396f45f3c7530be5bd5f143fbbfafdfc3c 100644 (file)
--- a/source/engineer/engineer.cpp
+++ b/source/engineer/engineer.cpp
@@ -119,14 +119,19 @@ Train *Engineer::add_train(unsigned addr)
                train_panels.push_back(tpanel);
                tpanel->set_visible(true);
 
-               placing_train=train;
-               placing_block=0;
-               main_panel->set_status_text("Select location");
+               place_train(*train);
 
                return train;
        }
 }
 
+void Engineer::place_train(Train &train)
+{
+       placing_train=&train;
+       placing_block=0;
+       main_panel->set_status_text("Select location");
+}
+
 int Engineer::main()
 {
        dpy=new Graphics::Display;
@@ -284,7 +289,7 @@ void Engineer::button_press(int x, int y, unsigned btn, unsigned)
 {
        if(placing_train)
        {
-               if(btn==1 && placing_block)
+               if(btn==1 && placing_block && !placing_block->get_train())
                {
                        set_block_color(*placing_block, GL::Color(1, 1, 1));
 
@@ -335,7 +340,7 @@ void Engineer::pointer_motion(int x, int y)
                        if(&block!=placing_block)
                        {
                                if(placing_block)
-                                       set_block_color(*placing_block, GL::Color(1, 1, 1));
+                                       reset_block_color(*placing_block);
                                placing_block=█
                                placing_entry=0;
                                set_block_color(*placing_block, GL::Color(0.5, 1, 0.7));
@@ -399,37 +404,37 @@ void Engineer::set_block_color(const Block &block, const GL::Color &color)
                layout_3d.get_track(**i).set_color(color);
 }
 
-void Engineer::sensor_event(bool state, Sensor *sensor)
-{
-       const list<Track3D *> &ltracks=layout_3d.get_tracks();
-       for(list<Track3D *>::const_iterator i=ltracks.begin(); i!=ltracks.end(); ++i)
-               if((*i)->get_track().get_sensor_id()==sensor->get_address())
-               {
-                       Block &block=trfc_mgr->get_block_by_track((*i)->get_track());
-                       if(state)
-                               (*i)->set_color(GL::Color(1, 0.5, 0.3));
-                       else if(block.get_train())
-                               set_block_color(block, GL::Color(1, 1, 0.3));
-                       else
-                               (*i)->set_color(GL::Color(1, 1, 1));
-               }
-}
-
-void Engineer::block_reserved(const Block &block, const Train *train)
+void Engineer::reset_block_color(const Block &block)
 {
        if(unsigned sid=block.get_sensor_id())
        {
                Sensor &sensor=control.get_sensor(sid);
                if(sensor.get_state())
+               {
+                       set_block_color(block, GL::Color(1, 0.5, 0.3));
                        return;
+               }
        }
 
-       if(train)
+       if(block.get_train())
                set_block_color(block, GL::Color(1, 1, 0.3));
        else
                set_block_color(block, GL::Color(1, 1, 1));
 }
 
+void Engineer::sensor_event(bool, Sensor *sensor)
+{
+       const list<Block *> &blocks=trfc_mgr->get_blocks();
+       for(list<Block *>::const_iterator i=blocks.begin(); i!=blocks.end(); ++i)
+               if((*i)->get_sensor_id()==sensor->get_address())
+                       reset_block_color(**i);
+}
+
+void Engineer::block_reserved(const Block &block, const Train *)
+{
+       reset_block_color(block);
+}
+
 void Engineer::project_3d()
 {
        glMatrixMode(GL_PROJECTION);