Properly detect blocks as critical when a train is not moving

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

diff --git a/source/libr2c2/blockallocator.cpp b/source/libr2c2/blockallocator.cpp
index a054c157331c9b5e9ed074da4fc99469e2b0ee8e..347960ee2ec584029a7bc66dffaaabb56de0cdad 100644 (file)
--- a/source/libr2c2/blockallocator.cpp
+++ b/source/libr2c2/blockallocator.cpp
@@ -257,10 +257,17 @@ bool BlockAllocator::reserve_block(const BlockIter &block)
        /* Add it to the list first to present a consistent state in block_reserved
        signal. */
        blocks.push_back(block);
+
+       bool first_reserve = (cur_blocks_end==blocks.end());
+       if(first_reserve)
+               --cur_blocks_end;
+
        try
        {
                if(!block->reserve(&train))
                {
+                       if(first_reserve)
+                               cur_blocks_end = blocks.end();
                        blocks.pop_back();
                        return false;
                }
@@ -269,6 +276,8 @@ bool BlockAllocator::reserve_block(const BlockIter &block)
        }
        catch(...)
        {
+               if(first_reserve)
+                       cur_blocks_end = blocks.end();
                blocks.pop_back();
                throw;
        }
@@ -413,7 +422,7 @@ void BlockAllocator::reverse()
 
 void BlockAllocator::turnout_path_changing(Track &track)
 {
-       BlockList::iterator i = find_if(blocks.begin(), blocks.end(), BlockMatch(track.get_block()));
+       BlockList::iterator i = find_if(cur_blocks_end, blocks.end(), BlockMatch(track.get_block()));
        if(i!=blocks.end())
        {
                ++i;