TBTR: Allow cloning virtual/template trains with unavailable engines

(cherry picked from commit 9d861fb8a0)
This commit is contained in:
Jonathan G Rennison
2019-04-27 21:25:56 +01:00
parent 21faae910b
commit 324ce7ae96
3 changed files with 15 additions and 22 deletions

View File

@@ -946,13 +946,13 @@ Train* VirtualTrainFromTemplateVehicle(TemplateVehicle* tv, StringID &err)
assert(tv->owner == _current_company);
head = CmdBuildVirtualRailVehicle(tv->engine_type, true, err);
if (!head) return NULL;
head = CmdBuildVirtualRailVehicle(tv->engine_type, err);
if (!head) return nullptr;
tail = head;
tv = tv->GetNextUnit();
while (tv) {
tmp = CmdBuildVirtualRailVehicle(tv->engine_type, true, err);
tmp = CmdBuildVirtualRailVehicle(tv->engine_type, err);
if (!tmp) {
CmdDeleteVirtualTrain(INVALID_TILE, DC_EXEC, head->index, 0, NULL);
return NULL;
@@ -998,13 +998,13 @@ CommandCost CmdVirtualTrainFromTrain(TileIndex tile, DoCommandFlag flags, uint32
Train *tmp, *head, *tail;
StringID err = INVALID_STRING_ID;
head = CmdBuildVirtualRailVehicle(train->engine_type, false, err);
head = CmdBuildVirtualRailVehicle(train->engine_type, err);
if (!head) return_cmd_error(err);
tail = head;
train = train->GetNextUnit();
while (train) {
tmp = CmdBuildVirtualRailVehicle(train->engine_type, false, err);
tmp = CmdBuildVirtualRailVehicle(train->engine_type, err);
if (!tmp) {
CmdDeleteVirtualTrain(tile, flags, head->index, 0, NULL);
return_cmd_error(err);
@@ -1136,8 +1136,9 @@ CommandCost CmdTemplateVehicleFromTrain(TileIndex tile, DoCommandFlag flags, uin
return CMD_ERROR;
}
for (Train *v = clicked; v != NULL; v = v->GetNextUnit()) {
if (!IsEngineBuildable(v->engine_type, VEH_TRAIN, _current_company)) {
for (Train *v = clicked; v != nullptr; v = v->GetNextUnit()) {
const Engine *e = Engine::GetIfValid(v->engine_type);
if (e == nullptr || e->type != VEH_TRAIN) {
return_cmd_error(STR_ERROR_RAIL_VEHICLE_NOT_AVAILABLE + VEH_TRAIN);
}
}