X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Flibr2c2%2Ftrainrouter.cpp;h=b5a5bef11f674f3301a36c1304558bc63b2bc8fa;hb=59b3cebecfd8f1462c95bab447be70dfb5f38e92;hp=a5e0d56543e13fab7082c5220c2b11f23ed96cce;hpb=9f4f169affcedcf249e935f93e0a15ed60232f6a;p=r2c2.git

diff --git a/source/libr2c2/trainrouter.cpp b/source/libr2c2/trainrouter.cpp
index a5e0d56..b5a5bef 100644
--- a/source/libr2c2/trainrouter.cpp
+++ b/source/libr2c2/trainrouter.cpp
@@ -14,7 +14,7 @@ namespace R2C2 {
 TrainRouter::TrainRouter(Train &t):
 	TrainAI(t),
 	priority(0),
-	arriving(false),
+	arriving(0),
 	dest_zone(0),
 	dest_block(0),
 	update_pending(false)
@@ -30,14 +30,14 @@ void TrainRouter::set_priority(int p)
 
 bool TrainRouter::set_route(const Route *r)
 {
-	train.free_noncritical_blocks();
+	BlockIter fncb = train.get_first_noncritical_block();
 
 	Route *lead = 0;
 	if(r && train.is_placed())
 	{
 		const BlockAllocator &allocator = train.get_block_allocator();
 		TrackIter first = allocator.first().track_iter();
-		TrackIter next = allocator.last().next().track_iter();
+		TrackIter next = fncb.track_iter();
 		if(!r->has_track(*next))
 		{
 			lead = Route::find(next, *r);
@@ -55,9 +55,9 @@ bool TrainRouter::set_route(const Route *r)
 	if(r)
 		routes.push_back(r);
 	train.stop_at(0);
-	arriving = false;
+	arriving = 0;
 
-	train.reserve_more();
+	train.refresh_blocks_from(*fncb);
 
 	const Route *route = get_route();
 	signal_route_changed.emit(route);
@@ -142,13 +142,14 @@ void TrainRouter::tick(const Time::TimeDelta &)
 	if(update_pending)
 		create_plans(train.get_layout());
 
-	if(arriving && !train.get_speed())
+	if(arriving==1 && !train.get_speed())
 	{
-		train.set_active(false);
 		signal_arrived.emit();
 		signal_event.emit(Message("arrived"));
-		set_route(0);
+		arriving = 2;
 	}
+	else if(arriving==2 && !train.get_block_allocator().is_active())
+		set_route(0);
 }
 
 void TrainRouter::save(list<DataFile::Statement> &st) const
@@ -182,9 +183,9 @@ void TrainRouter::block_reserved(Block &block, Train *t)
 	if(advance_route(route, block))
 	{
 		// Check if the block is a turnout and set it to proper path
-		if(unsigned tid = block.get_turnout_id())
+		if(unsigned taddr = block.get_turnout_address())
 		{
-			int path = (*route)->get_turnout(tid);
+			int path = (*route)->get_turnout(taddr);
 			if(path>=0)
 				b_iter.track_iter()->set_active_path(path);
 		}
@@ -224,7 +225,7 @@ void TrainRouter::train_advanced(Block &block)
 	{
 		b_iter = b_iter.next();
 		if(b_iter && !is_on_route(*b_iter))
-			arriving = true;
+			arriving = 1;
 	}
 }