X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Flibr2c2%2Ftrainrouteplanner.cpp;h=770c0ea537897a01ee0f3a70ae5418f6aa78927d;hb=18c529dfdd34e07f4f6021151ddce29fa85437bc;hp=0de707b16d21c167a88b9f153ac4945c719c647a;hpb=54392d65e2053d1eacb4cfcc435f1013993f2973;p=r2c2.git

diff --git a/source/libr2c2/trainrouteplanner.cpp b/source/libr2c2/trainrouteplanner.cpp
index 0de707b..770c0ea 100644
--- a/source/libr2c2/trainrouteplanner.cpp
+++ b/source/libr2c2/trainrouteplanner.cpp
@@ -226,34 +226,18 @@ TrainRoutePlanner::TrainRoutingState::TrainRoutingState(TrainRoutingInfo &inf):
 	state(MOVING)
 {
 	const Vehicle *veh = &info->train->get_vehicle(0);
-	track = veh->get_track_iter();
 	// TODO margins
-	offset = veh->get_offset()+veh->get_type().get_length()/2;
+	TrackOffsetIter track_and_offs = veh->get_placement().get_position(VehiclePlacement::FRONT_BUFFER);
+	track = track_and_offs.track_iter();
+	offset = track_and_offs.offset();
 	path = track->get_active_path();
 
-	float path_length = track->get_type().get_path_length(path);
-	while(offset>path_length)
-	{
-		offset -= path_length;
-		track = track.next();
-		path = track->get_active_path();
-		path_length = track->get_type().get_path_length(path);
-	}
-
 	while(Vehicle *next = veh->get_link(1))
 		veh = next;
-	back_offset = veh->get_offset()-veh->get_type().get_length()/2;
-
-	TrackIter iter = veh->get_track_iter();
-	while(back_offset<0)
-	{
-		TrackIter prev = iter.flip().reverse();
-		if(!prev)
-			break;
-		iter = prev;
-		back_offset += iter->get_type().get_path_length(iter->get_active_path());
-	}
+	track_and_offs = veh->get_placement().get_position(VehiclePlacement::BACK_BUFFER);
+	back_offset = track_and_offs.offset();
 
+	TrackIter iter = track_and_offs.track_iter();
 	while(1)
 	{
 		occupied_tracks = new OccupiedTrack(*iter, iter->get_active_path(), occupied_tracks);