[RSHELL]

[reactos.git] / base / shell / rshell / CMenuDeskBar.h

diff --git a/base/shell/rshell/CMenuDeskBar.h b/base/shell/rshell/CMenuDeskBar.h
index a0ccdf9..7c63a18 100644 (file)
--- a/base/shell/rshell/CMenuDeskBar.h
+++ b/base/shell/rshell/CMenuDeskBar.h
@@ -50,9 +50,12 @@ private:
     BOOL  m_Shown;
     DWORD m_ShowFlags;
 
+    BOOL m_didAddRef;
+
+    virtual void OnFinalMessage(HWND hWnd);
 public:
     CMenuDeskBar();
-    ~CMenuDeskBar();
+    virtual ~CMenuDeskBar();
 
     DECLARE_NOT_AGGREGATABLE(CMenuDeskBar)
     DECLARE_PROTECT_FINAL_CONSTRUCT()
@@ -60,6 +63,7 @@ public:
     DECLARE_WND_CLASS_EX(_T("BaseBar"), CS_SAVEBITS | CS_DROPSHADOW, COLOR_3DFACE)
 
     BEGIN_MSG_MAP(CMenuDeskBar)
+        MESSAGE_HANDLER(WM_CREATE, _OnCreate)
         MESSAGE_HANDLER(WM_SIZE, _OnSize)
         MESSAGE_HANDLER(WM_NOTIFY, _OnNotify)
         MESSAGE_HANDLER(WM_PAINT, _OnPaint)
@@ -125,6 +129,7 @@ public:
 
 private:
     // message handlers
+    LRESULT _OnCreate(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL &bHandled);
     LRESULT _OnSize(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL &bHandled);
     LRESULT _OnNotify(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL &bHandled);
     LRESULT _OnPaint(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL &bHandled);