(svn r23331) -Fix [FS#4851] (rCS, r148): abort building/moving HQ when clicking on the button again, just like when building rail, stations, etc (sbr)

1 files changed, 10 insertions, 0 deletions

diff --git a/src/company_gui.cpp b/src/company_gui.cpp
index 5541993a5..d33a831ce 100644
--- a/src/company_gui.cpp
+++ b/src/company_gui.cpp
@@ -2001,6 +2001,11 @@ struct CompanyWindow : Window
 
 			case CW_WIDGET_BUILD_HQ:
 				if ((byte)this->window_number != _local_company) return;
+				if (this->IsWidgetLowered(CW_WIDGET_BUILD_HQ)) {
+					ResetObjectToPlace();
+					this->RaiseButtons();
+					break;
+				}
 				SetObjectToPlaceWnd(SPR_CURSOR_HQ, PAL_NONE, HT_RECT, this);
 				SetTileSelectSize(2, 2);
 				this->LowerWidget(CW_WIDGET_BUILD_HQ);
@@ -2008,6 +2013,11 @@ struct CompanyWindow : Window
 				break;
 
 			case CW_WIDGET_RELOCATE_HQ:
+				if (this->IsWidgetLowered(CW_WIDGET_RELOCATE_HQ)) {
+					ResetObjectToPlace();
+					this->RaiseButtons();
+					break;
+				}
 				SetObjectToPlaceWnd(SPR_CURSOR_HQ, PAL_NONE, HT_RECT, this);
 				SetTileSelectSize(2, 2);
 				this->LowerWidget(CW_WIDGET_RELOCATE_HQ);