1 files changed, 5 insertions, 1 deletions
diff --git a/include/m_gui.h b/include/m_gui.h
index 0f3f161b60..eabf1040a6 100644
--- a/include/m_gui.h
+++ b/include/m_gui.h
@@ -440,7 +440,8 @@ private:
 	LIST<CCtrlBase> m_controls;
 
 	void NotifyControls(void (CCtrlBase::*fn)());
-	CCtrlBase *FindControl(int idCtrl);
+	CCtrlBase* FindControl(int idCtrl);
+	CCtrlBase* FindControl(HWND hwnd);
 
 	CTimer* FindTimer(int idEvent);
 
@@ -479,6 +480,8 @@ class MIR_CORE_EXPORT CCtrlBase
 {
 	friend class CDlgBase;
 
+	__forceinline CCtrlBase(HWND hwnd) : m_hwnd(hwnd) {}
+
 public:
 	CCtrlBase(CDlgBase *wnd, int idCtrl);
 	virtual ~CCtrlBase() { }
@@ -530,6 +533,7 @@ protected:
 
 public:
 	CCallback<CCtrlBase> OnChange;
+	CCallback<CCtrlBase> OnBuildMenu;
 
 protected:
 	virtual LRESULT CustomWndProc(UINT msg, WPARAM wParam, LPARAM lParam);