Add a virtual get_block function to Sensor

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

diff --git a/source/libr2c2/layout.cpp b/source/libr2c2/layout.cpp
index 40863237081c9677a6451eadf29de9584f51bf80..6cb0eb3634ae334cca8714b87f5d6fb7ae5251bd 100644 (file)
--- a/source/libr2c2/layout.cpp
+++ b/source/libr2c2/layout.cpp
@@ -435,10 +435,7 @@ void Layout::sensor_state_changed(Sensor &sensor, Sensor::State state)
 {
        if(state==Sensor::ACTIVE)
        {
-               Block *block = 0;
-               if(TrackCircuit *tc = dynamic_cast<TrackCircuit *>(&sensor))
-                       block = &tc->get_block();
-
+               Block *block = sensor.get_block();
                if(block && !block->get_train())
                        emergency(format("Unreserved sensor %d triggered", sensor.get_address()));
        }