myatlctrl.h

来自「VC.net 的一些有用的技巧」· C头文件 代码 · 共 166 行

// MyATLCtrl.h : CMyATLCtrl 的声明
#pragma once
#include <math.h>
#include "resource.h"       // 主符号
#include <atlctl.h>
#include "ATLDemo.h"
#include "_IMyATLCtrlEvents_CP.h"

// CMyATLCtrl
class ATL_NO_VTABLE CMyATLCtrl : 
	public CComObjectRootEx<CComSingleThreadModel>,
	public CStockPropImpl<CMyATLCtrl, IMyATLCtrl>,
	public IPersistStreamInitImpl<CMyATLCtrl>,
	public IOleControlImpl<CMyATLCtrl>,
	public IOleObjectImpl<CMyATLCtrl>,
	public IOleInPlaceActiveObjectImpl<CMyATLCtrl>,
	public IViewObjectExImpl<CMyATLCtrl>,
	public IOleInPlaceObjectWindowlessImpl<CMyATLCtrl>,
	public ISupportErrorInfo,
	public IConnectionPointContainerImpl<CMyATLCtrl>,
	public CProxy_IMyATLCtrlEvents<CMyATLCtrl>, 
	public IPersistStorageImpl<CMyATLCtrl>,
	public ISpecifyPropertyPagesImpl<CMyATLCtrl>,
	public IQuickActivateImpl<CMyATLCtrl>,
	public IDataObjectImpl<CMyATLCtrl>,
	public IProvideClassInfo2Impl<&CLSID_MyATLCtrl, &__uuidof(_IMyATLCtrlEvents), &LIBID_ATLDemoLib>,
	public CComCoClass<CMyATLCtrl, &CLSID_MyATLCtrl>,
	public CComControl<CMyATLCtrl>,
	public IObjectSafetyImpl<CMyATLCtrl, INTERFACESAFE_FOR_UNTRUSTED_CALLER>
{
public:

	CMyATLCtrl()
	{
		m_nSides = 3;
		m_clrFillColor = RGB(0, 0xFF, 0);
	}

DECLARE_OLEMISC_STATUS(OLEMISC_RECOMPOSEONRESIZE | 
	OLEMISC_CANTLINKINSIDE | 
	OLEMISC_INSIDEOUT | 
	OLEMISC_ACTIVATEWHENVISIBLE | 
	OLEMISC_SETCLIENTSITEFIRST
)

DECLARE_REGISTRY_RESOURCEID(IDR_MYATLCTRL)

BEGIN_COM_MAP(CMyATLCtrl)
	COM_INTERFACE_ENTRY(IMyATLCtrl)
	COM_INTERFACE_ENTRY(IDispatch)
	COM_INTERFACE_ENTRY(IViewObjectEx)
	COM_INTERFACE_ENTRY(IViewObject2)
	COM_INTERFACE_ENTRY(IViewObject)
	COM_INTERFACE_ENTRY(IOleInPlaceObjectWindowless)
	COM_INTERFACE_ENTRY(IOleInPlaceObject)
	COM_INTERFACE_ENTRY2(IOleWindow, IOleInPlaceObjectWindowless)
	COM_INTERFACE_ENTRY(IOleInPlaceActiveObject)
	COM_INTERFACE_ENTRY(IOleControl)
	COM_INTERFACE_ENTRY(IOleObject)
	COM_INTERFACE_ENTRY(IPersistStreamInit)
	COM_INTERFACE_ENTRY2(IPersist, IPersistStreamInit)
	COM_INTERFACE_ENTRY(ISupportErrorInfo)
	COM_INTERFACE_ENTRY(IConnectionPointContainer)
	COM_INTERFACE_ENTRY(ISpecifyPropertyPages)
	COM_INTERFACE_ENTRY(IQuickActivate)
	COM_INTERFACE_ENTRY(IPersistStorage)
	COM_INTERFACE_ENTRY(IDataObject)
	COM_INTERFACE_ENTRY(IProvideClassInfo)
	COM_INTERFACE_ENTRY(IProvideClassInfo2)
	COM_INTERFACE_ENTRY(IObjectSafety)
END_COM_MAP()

BEGIN_PROP_MAP(CMyATLCtrl)
	PROP_DATA_ENTRY("_cx", m_sizeExtent.cx, VT_UI4)
	PROP_DATA_ENTRY("_cy", m_sizeExtent.cy, VT_UI4)
	PROP_ENTRY("FillColor", DISPID_FILLCOLOR, CLSID_StockColorPage)
	PROP_ENTRY("Sides", 1, CLSID_ATLDemoProp)
	// 示例项
	// PROP_ENTRY("Property Description", dispid, clsid)
	// PROP_PAGE(CLSID_StockColorPage)
END_PROP_MAP()

BEGIN_CONNECTION_POINT_MAP(CMyATLCtrl)
	CONNECTION_POINT_ENTRY(__uuidof(_IMyATLCtrlEvents))
END_CONNECTION_POINT_MAP()

BEGIN_MSG_MAP(CMyATLCtrl)
	MESSAGE_HANDLER(WM_LBUTTONDOWN, OnLButtonDown)
	CHAIN_MSG_MAP(CComControl<CMyATLCtrl>)
	DEFAULT_REFLECTION_HANDLER()
END_MSG_MAP()
// 处理程序原型：
//  LRESULT MessageHandler(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled);
//  LRESULT CommandHandler(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled);
//  LRESULT NotifyHandler(int idCtrl, LPNMHDR pnmh, BOOL& bHandled);

// ISupportsErrorInfo
	STDMETHOD(InterfaceSupportsErrorInfo)(REFIID riid)
	{
		static const IID* arr[] = 
		{
			&IID_IMyATLCtrl,
		};

		for (int i=0; i<sizeof(arr)/sizeof(arr[0]); i++)
		{
			if (InlineIsEqualGUID(*arr[i], riid))
				return S_OK;
		}
		return S_FALSE;
	}

// IViewObjectEx
	DECLARE_VIEW_STATUS(VIEWSTATUS_SOLIDBKGND | VIEWSTATUS_OPAQUE)

// IMyATLCtrl
public:
		HRESULT OnDraw(ATL_DRAWINFO& di)
		{
			RECT& rc = *(RECT*)di.prcBounds;
			HDC hdc  = di.hdcDraw;
			COLORREF    colFore;
			HBRUSH      hOldBrush, hBrush;
			HPEN        hOldPen, hPen;
            
			OleTranslateColor(m_clrFillColor, NULL, &colFore);
			hPen = (HPEN)GetStockObject(BLACK_PEN);
			hOldPen = (HPEN)SelectObject(hdc, hPen);
			hBrush = (HBRUSH)GetStockObject(WHITE_BRUSH);
			hOldBrush = (HBRUSH)SelectObject(hdc, hBrush);
			Ellipse(hdc, rc.left, rc.top, rc.right, rc.bottom);
			hBrush = CreateSolidBrush(colFore);
			SelectObject(hdc, hBrush);
            CalcPoints(rc);
			Polygon(hdc, &m_arrPoint[0], m_nSides);            
			SelectObject(hdc, hOldPen);
			SelectObject(hdc, hOldBrush);
			DeleteObject(hBrush);
            return S_OK;
		}

	OLE_COLOR m_clrFillColor;
	short m_nSides;
	POINT m_arrPoint[100];
	void OnFillColorChanged()
	{
		ATLTRACE(_T("OnFillColorChanged\n"));
	}

	DECLARE_PROTECT_FINAL_CONSTRUCT()

	HRESULT FinalConstruct()
	{
		return S_OK;
	}
	
	void FinalRelease() 
	{
	}
	STDMETHOD(get_Sides)(SHORT* pVal);
	STDMETHOD(put_Sides)(SHORT newVal);
	void CalcPoints(const RECT& rc);
	LRESULT OnLButtonDown(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& /*bHandled*/);
};

OBJECT_ENTRY_AUTO(__uuidof(MyATLCtrl), CMyATLCtrl)