diff options
| -rw-r--r-- | src/company_cmd.cpp | 2 | ||||
| -rw-r--r-- | src/misc_gui.cpp | 6 |
2 files changed, 8 insertions, 0 deletions
diff --git a/src/company_cmd.cpp b/src/company_cmd.cpp index 5e7d4ed41..29343a09b 100644 --- a/src/company_cmd.cpp +++ b/src/company_cmd.cpp @@ -78,6 +78,8 @@ void Company::PostDestructor(size_t index) InvalidateWindowData(WC_GRAPH_LEGEND, 0, (int)index); InvalidateWindowData(WC_PERFORMANCE_DETAIL, 0, (int)index); InvalidateWindowData(WC_COMPANY_LEAGUE, 0, 0); + /* If the currently shown error message has this company in it, the close it. */ + InvalidateWindowData(WC_ERRMSG, 0); } /** diff --git a/src/misc_gui.cpp b/src/misc_gui.cpp index 42776ca5e..797273af0 100644 --- a/src/misc_gui.cpp +++ b/src/misc_gui.cpp @@ -620,6 +620,12 @@ public: this->DrawWidgets(); } + virtual void OnInvalidateData(int data) + { + /* If company gets shut down, while displaying an error about it, remove the error message. */ + if (this->face != INVALID_COMPANY && !Company::IsValidID(this->face)) delete this; + } + virtual void SetStringParameters(int widget) const { if (widget == EMW_CAPTION) CopyInDParam(0, this->decode_params, lengthof(this->decode_params)); |