Fix: Recompute road/railtype availability after disabling the engine

This commit is contained in:
dP 2021-03-17 22:22:11 +03:00 committed by Charles Pigott
parent b43cc21214
commit d8dfa1613f
1 changed files with 6 additions and 0 deletions

View File

@ -747,8 +747,14 @@ static void EnableEngineForCompany(EngineID eid, CompanyID company)
static void DisableEngineForCompany(EngineID eid, CompanyID company)
{
Engine *e = Engine::Get(eid);
Company *c = Company::Get(company);
ClrBit(e->company_avail, company);
if (e->type == VEH_TRAIN) {
c->avail_railtypes = GetCompanyRailtypes(c->index);
} else if (e->type == VEH_ROAD) {
c->avail_roadtypes = GetCompanyRoadTypes(c->index);
}
if (company == _local_company) {
AddRemoveEngineFromAutoreplaceAndBuildWindows(e->type);