IE Drop-Down Button

#ifndef TOOLBARPARENTWND_H_INCLUDED
#define TOOLBARPARENTWND_H_INCLUDED

#pragma once
#include <shobjidl.h>
#include<atlwin.h>


#define BUTTON_NAME TEXT("Dropdown Arrow Button")   
#define WM_NOTIFY_SETBUTTONINFO WM_USER+1012 

// �������ݴ�����Ϣ�ṹ
typedef struct tagBrowserWndInfo
{
	IWebBrowser2 *pWebBrowser; // �����ָ��
	int nButtonState;  //��ǰ���ڵİ�ť״̬
	HWND hBrowserWnd;  //��ʾ���ݵ����������
} BrowserWndInfo;

class CToolBarParentWnd : public CWindowImpl<CToolBarParentWnd>
{
public:
    CToolBarParentWnd(void);

public:
    ~CToolBarParentWnd(void);

BEGIN_MSG_MAP(CToolBarParentWnd)
    MESSAGE_HANDLER(WM_COMMAND, OnCommand)
    NOTIFY_CODE_HANDLER(TBN_DROPDOWN, OnNotifyDropDown)
    MESSAGE_HANDLER(WM_NOTIFY_SETBUTTONINFO, OnNotifySetButtonInfo)
END_MSG_MAP()

public:
    LRESULT OnNotifySetButtonInfo(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& /*bHandled*/);
	LRESULT OnNotifyDropDown(WPARAM wParam, LPNMHDR lpnmhdr, BOOL& /*bHandled*/); 

    LRESULT OnCommand(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& /*bHandled*/);

	void SetBrowserWndInfo(BrowserWndInfo *pBrowserWndInfo);
	BrowserWndInfo *GetBrowserWndInfo();

public:
    void SetToolbarWnd(HWND hToolbarWnd);
    void SetButtonID(int nID);
    int GetButtonID();
    void NotifySetButtonInfo(int buttonID);
    void OnDropDown(LPNMTOOLBAR lpnmtb);
private:
    void SetButtonCheck(BOOL bChecked);
    void LoadMenu();
    void ReleaseMenu();

private:
    HMENU m_hMenu;
    HMENU m_hMemuTrackPopup;
    HWND m_ToolbarWnd;
    int  m_nButtonID;
    int  m_nButtonState;

	BrowserWndInfo *m_pBrowserWndInfo;
};

#endif // #ifndef TOOLBARPARENTWND_H_INCLUDED