Fix: change all Company planes on paint, not only the first that needs changing

Otherwise it can take a few OnPaint() calls before all planes are
set correctly when switching companies.
This commit is contained in:
Patric Stout 2021-01-05 16:40:37 +01:00 committed by Patric Stout
parent dff7ee8e1c
commit fc474b02fa
1 changed files with 3 additions and 5 deletions

View File

@ -2294,7 +2294,7 @@ struct CompanyWindow : Window
if (plane != wi->shown_plane) {
wi->SetDisplayedPlane(plane);
this->InvalidateData();
return;
reinit = true;
}
/* Build HQ button handling. */
@ -2302,8 +2302,7 @@ struct CompanyWindow : Window
wi = this->GetWidget<NWidgetStacked>(WID_C_SELECT_VIEW_BUILD_HQ);
if (plane != wi->shown_plane) {
wi->SetDisplayedPlane(plane);
this->SetDirty();
return;
reinit = true;
}
this->SetWidgetDisabledState(WID_C_VIEW_HQ, c->location_of_HQ == INVALID_TILE);
@ -2313,8 +2312,7 @@ struct CompanyWindow : Window
wi = this->GetWidget<NWidgetStacked>(WID_C_SELECT_RELOCATE);
if (plane != wi->shown_plane) {
wi->SetDisplayedPlane(plane);
this->SetDirty();
return;
reinit = true;
}
/* Owners of company */