TBTR: Use virtual train head in CmdReplaceTemplateVehicle

Check that train is virtual
This commit is contained in:
Jonathan G Rennison
2020-10-09 01:00:26 +01:00
parent 08eba56a28
commit 34a6445e77

View File

@@ -1150,8 +1150,12 @@ CommandCost CmdReplaceTemplateVehicle(TileIndex tile, DoCommandFlag flags, uint3
CommandCost ret = CheckOwnership(vehicle->owner);
if (ret.Failed()) return ret;
Train* train = Train::From(vehicle);
vehicle = vehicle->First();
Train* train = Train::From(vehicle);
if (!train->IsVirtual()) {
return CMD_ERROR;
}
if (!TemplateVehicle::CanAllocateItem(CountVehiclesInChain(train))) {
return CMD_ERROR;
}