TBTR: Fix virtual trains being created at load using wrong company.

Fix virtual trains showing up in company engine counts.
This commit is contained in:
Jonathan G Rennison
2016-03-21 18:22:29 +00:00
parent cf638e836a
commit 6741c98f6b
4 changed files with 14 additions and 1 deletions

View File

@@ -3,6 +3,7 @@
#include "../tbtr_template_vehicle.h"
#include "../tbtr_template_vehicle_func.h"
#include "../train.h"
#include "../core/backup_type.hpp"
#include "saveload.h"
@@ -102,6 +103,7 @@ void AfterLoadTemplateVehiclesUpdateImage()
FOR_ALL_TEMPLATES(tv) {
if (tv->Prev() == NULL) {
Backup<CompanyByte> cur_company(_current_company, tv->owner, FILE_LINE);
StringID err;
Train* t = VirtualTrainFromTemplateVehicle(tv, err);
if (t != NULL) {
@@ -123,6 +125,7 @@ void AfterLoadTemplateVehiclesUpdateImage()
}
delete t;
}
cur_company.Restore();
}
}
}