X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fengineer%2Ftrainpanel.cpp;h=e6a91a8f6dc1c2b272fa38adb88c4a3adc62c165;hb=54392d65e2053d1eacb4cfcc435f1013993f2973;hp=1076fedaedbbb17152084bb337d6f0e69d89c97f;hpb=ca9d278f9472206ad9a01190dcef9f0eb1bcc274;p=r2c2.git

diff --git a/source/engineer/trainpanel.cpp b/source/engineer/trainpanel.cpp
index 1076fed..e6a91a8 100644
--- a/source/engineer/trainpanel.cpp
+++ b/source/engineer/trainpanel.cpp
@@ -253,25 +253,9 @@ void TrainPanel::place(Track *track, unsigned ep)
 {
 	pick_conn.disconnect();
 
-	Block &block = track->get_block();
-	TrackIter iter(track, ep);
-
-	while(block.has_track(*iter))
-	{
-		const vector<Block::Endpoint> &eps = block.get_endpoints();
-		bool ok = false;
-		for(unsigned i=0; (!ok && i<eps.size()); ++i)
-			if(eps[i].track==iter.track() && eps[i].track_ep==iter.entry())
-			{
-				train.place(block, i);
-				ok = true;
-			}
-
-		if(ok)
-			break;
-
-		iter = iter.flip().reverse();
-	}
+	BlockIter block = TrackIter(track, ep).block_iter();
+	if(block)
+		train.place(block);
 }
 
 void TrainPanel::go_to(Track *track, unsigned)