# HG changeset patch
# User rubidium <rubidium@openttd.org>
# Date 2009-11-25 22:58:28
# Node ID bf4fbd1c24d248469d301d923bf8b3b1f2396652
# Parent  55872e4820c50da3674d17a69b6ddf813f35072b

(svn r18290) -Codechange: be more strict about what vehicles may use timetables

diff --git a/src/timetable_cmd.cpp b/src/timetable_cmd.cpp
--- a/src/timetable_cmd.cpp
+++ b/src/timetable_cmd.cpp
@@ -67,7 +67,7 @@ CommandCost CmdChangeTimetable(TileIndex
 	VehicleID veh = GB(p1, 0, 16);
 
 	Vehicle *v = Vehicle::GetIfValid(veh);
-	if (v == NULL || !CheckOwnership(v->owner)) return CMD_ERROR;
+	if (v == NULL || !CheckOwnership(v->owner) || !v->IsPrimaryVehicle()) return CMD_ERROR;
 
 	VehicleOrderID order_number = GB(p1, 16, 8);
 	Order *order = v->GetOrder(order_number);
@@ -127,10 +127,11 @@ CommandCost CmdSetVehicleOnTime(TileInde
 	VehicleID veh = GB(p1, 0, 16);
 
 	Vehicle *v = Vehicle::GetIfValid(veh);
-	if (v == NULL || !CheckOwnership(v->owner)) return CMD_ERROR;
+	if (v == NULL || !CheckOwnership(v->owner) || !v->IsPrimaryVehicle()) return CMD_ERROR;
 
 	if (flags & DC_EXEC) {
 		v->lateness_counter = 0;
+		SetWindowDirty(WC_VEHICLE_TIMETABLE, v->index);
 	}
 
 	return CommandCost();
@@ -156,7 +157,7 @@ CommandCost CmdAutofillTimetable(TileInd
 	VehicleID veh = GB(p1, 0, 16);
 
 	Vehicle *v = Vehicle::GetIfValid(veh);
-	if (v == NULL || !CheckOwnership(v->owner)) return CMD_ERROR;
+	if (v == NULL || !CheckOwnership(v->owner) || !v->IsPrimaryVehicle()) return CMD_ERROR;
 
 	if (flags & DC_EXEC) {
 		if (HasBit(p2, 0)) {