X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;ds=sidebyside;f=source%2Flibr2c2%2Ftrain.cpp;h=d78e6276ec14bd98780badaef96ddece39ed2be7;hb=59115614ee941fe801416aca2f8f380065bdc2b7;hp=10227222daac98b38956c14975d0a334ad2cc3ab;hpb=dec294d40194a640e7b4bccf20dd1baa4a87038c;p=r2c2.git

diff --git a/source/libr2c2/train.cpp b/source/libr2c2/train.cpp
index 1022722..d78e627 100644
--- a/source/libr2c2/train.cpp
+++ b/source/libr2c2/train.cpp
@@ -1,6 +1,7 @@
 #include <algorithm>
 #include <cmath>
 #include <msp/core/maputils.h>
+#include <msp/core/raii.h>
 #include <msp/strings/format.h>
 #include <msp/time/units.h>
 #include <msp/time/utils.h>
@@ -23,19 +24,6 @@
 using namespace std;
 using namespace Msp;
 
-namespace {
-
-struct SetFlag
-{
-	bool &flag;
-
-	SetFlag(bool &f): flag(f) { flag = true; }
-	~SetFlag() { flag = false; }
-};
-
-}
-
-
 namespace R2C2 {
 
 Train::Train(Layout &l, const VehicleType &t, unsigned a, const string &p):
@@ -258,6 +246,8 @@ void Train::unplace()
 void Train::stop_at(Block *block)
 {
 	stop_at_block = block;
+	if(active && !stop_at_block)
+		reserve_more();
 }
 
 bool Train::free_block(Block &block)