Basic support for beam gate sensors

[r2c2.git] / source / libr2c2 / train.cpp

diff --git a/source/libr2c2/train.cpp b/source/libr2c2/train.cpp
index 25dcae97da9580cea1946315674fb9bd36c5c3ed..aa2ba3f528627beb093dc936446b1f2b3bbd4d81 100644 (file)
--- a/source/libr2c2/train.cpp
+++ b/source/libr2c2/train.cpp
@@ -422,13 +422,14 @@ void Train::loco_func_event(unsigned addr, unsigned func, bool state)
 
 void Train::sensor_state_changed(Sensor &sensor, Sensor::State state)
 {
-       Block *block = 0;
-       if(TrackCircuit *tc = dynamic_cast<TrackCircuit *>(&sensor))
-               block = &tc->get_block();
-       else
+       if(state!=Sensor::MAYBE_ACTIVE)
+               return;
+
+       Block *block = sensor.get_block();
+       if(!block || block->get_train()!=this)
                return;
 
-       if(block->get_train()==this && state==Sensor::MAYBE_ACTIVE)
+       if(dynamic_cast<TrackCircuit *>(&sensor))
        {
                if(last_entry_block)
                {