train_cmd.c
changeset 2844 7a55ad0a6a36
parent 2840 d68a605b2087
child 2854 6b85a5f4b65f
--- a/train_cmd.c	Sun Jan 08 12:44:11 2006 +0000
+++ b/train_cmd.c	Sun Jan 08 16:56:41 2006 +0000
@@ -807,7 +807,10 @@
 
 	count = 0;
 	for (; v != NULL; v = v->next) {
-		count++;
+		/* This count is used by the depot code to determine the number of engines
+		 * in the consist. Exclude articulated parts so that autoreplacing to
+		 * engines with more articulated parts that before works correctly. */
+		if (!IsArticulatedPart(v)) count++;
 		if (v->u.rail.track != 0x80 || v->tile != tile ||
 				(IsFrontEngine(v) && !(v->vehstatus & VS_STOPPED))) {
 			_error_message = STR_881A_TRAINS_CAN_ONLY_BE_ALTERED;
@@ -998,7 +1001,19 @@
 		/* check if all vehicles in the source train are stopped inside a depot */
 		if (r < 0) return CMD_ERROR;
 
-		num += r;
+		if (HASBIT(p2, 0)) {
+			/* If moving the rest of the train, exclude wagons before the
+			 * selected one. */
+
+			Vehicle *u;
+			for (u = src_head; u != src && u != NULL; u = GetNextVehicle(u))
+				r--;
+
+			num += r;
+		} else {
+			// If moving only one vehicle, just count that.
+			num++;
+		}
 
 		/* check if all the vehicles in the dest train are stopped */
 		if (dst_head != NULL) {