1610 lines
58 KiB
C++
1610 lines
58 KiB
C++
#ifndef __ATL_SNAPIN_H__
|
|
#define __ATL_SNAPIN_H__
|
|
|
|
#ifndef UNICODE
|
|
#error "Only Unicode builds supported"
|
|
#endif
|
|
|
|
#include <mmc.h>
|
|
#include <commctrl.h>
|
|
#pragma comment(lib, "mmc.lib")
|
|
|
|
|
|
// Wrappers for propertypage and HBITMAP
|
|
#ifndef CPropertyPageImpl
|
|
#pragma comment(lib, "comctl32.lib")
|
|
|
|
template <class T>
|
|
class ATL_NO_VTABLE CPropertyPageImpl : public CDialogImplBase
|
|
{
|
|
public:
|
|
PROPSHEETPAGE m_psp;
|
|
|
|
operator PROPSHEETPAGE*() { return &m_psp; }
|
|
|
|
// Construction
|
|
CPropertyPageImpl(LPCTSTR lpszTitle = NULL)
|
|
{
|
|
// initialize PROPSHEETPAGE struct
|
|
memset(&m_psp, 0, sizeof(PROPSHEETPAGE));
|
|
m_psp.dwSize = sizeof(PROPSHEETPAGE);
|
|
m_psp.dwFlags = PSP_USECALLBACK;
|
|
m_psp.hInstance = _Module.GetResourceInstance();
|
|
m_psp.pszTemplate = MAKEINTRESOURCE(T::IDD);
|
|
m_psp.pfnDlgProc = (DLGPROC)T::StartDialogProc;
|
|
m_psp.pfnCallback = T::PropPageCallback;
|
|
m_psp.lParam = (LPARAM)this;
|
|
|
|
if(lpszTitle != NULL)
|
|
{
|
|
m_psp.pszTitle = lpszTitle;
|
|
m_psp.dwFlags |= PSP_USETITLE;
|
|
}
|
|
}
|
|
|
|
static UINT CALLBACK PropPageCallback(HWND hWnd, UINT uMsg, LPPROPSHEETPAGE ppsp)
|
|
{
|
|
if(uMsg == PSPCB_CREATE)
|
|
{
|
|
_ASSERTE(hWnd == NULL);
|
|
CDialogImplBase* pPage = (CDialogImplBase*)ppsp->lParam;
|
|
_Module.AddCreateWndData(&pPage->m_thunk.cd, pPage);
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
HPROPSHEETPAGE Create()
|
|
{
|
|
return ::CreatePropertySheetPage(&m_psp);
|
|
}
|
|
|
|
BOOL EndDialog(int)
|
|
{
|
|
// do nothing here, calling ::EndDialog will close the whole sheet
|
|
_ASSERTE(FALSE);
|
|
return FALSE;
|
|
}
|
|
|
|
// Operations
|
|
void CancelToClose()
|
|
{
|
|
_ASSERTE(::IsWindow(m_hWnd));
|
|
_ASSERTE(GetParent() != NULL);
|
|
|
|
::SendMessage(GetParent(), PSM_CANCELTOCLOSE, 0, 0L);
|
|
}
|
|
void SetModified(BOOL bChanged = TRUE)
|
|
{
|
|
_ASSERTE(::IsWindow(m_hWnd));
|
|
_ASSERTE(GetParent() != NULL);
|
|
|
|
if(bChanged)
|
|
::SendMessage(GetParent(), PSM_CHANGED, (WPARAM)m_hWnd, 0L);
|
|
else
|
|
::SendMessage(GetParent(), PSM_UNCHANGED, (WPARAM)m_hWnd, 0L);
|
|
}
|
|
LRESULT QuerySiblings(WPARAM wParam, LPARAM lParam)
|
|
{
|
|
_ASSERTE(::IsWindow(m_hWnd));
|
|
_ASSERTE(GetParent() != NULL);
|
|
|
|
return ::SendMessage(GetParent(), PSM_QUERYSIBLINGS, wParam, lParam);
|
|
}
|
|
|
|
BEGIN_MSG_MAP(CPropertyPageImpl< T >)
|
|
MESSAGE_HANDLER(WM_NOTIFY, OnNotify)
|
|
END_MSG_MAP()
|
|
|
|
// Message handler
|
|
LRESULT OnNotify(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
|
|
{
|
|
_ASSERTE(::IsWindow(m_hWnd));
|
|
NMHDR* pNMHDR = (NMHDR*)lParam;
|
|
|
|
// don't handle messages not from the page/sheet itself
|
|
if(pNMHDR->hwndFrom != m_hWnd && pNMHDR->hwndFrom != ::GetParent(m_hWnd))
|
|
{
|
|
bHandled = FALSE;
|
|
return 1;
|
|
}
|
|
|
|
T* pT = (T*)this;
|
|
LRESULT lResult = 0;
|
|
// handle default
|
|
switch(pNMHDR->code)
|
|
{
|
|
case PSN_SETACTIVE:
|
|
lResult = pT->OnSetActive() ? 0 : -1;
|
|
break;
|
|
case PSN_KILLACTIVE:
|
|
lResult = !pT->OnKillActive();
|
|
break;
|
|
case PSN_APPLY:
|
|
lResult = pT->OnApply() ? PSNRET_NOERROR : PSNRET_INVALID_NOCHANGEPAGE;
|
|
break;
|
|
case PSN_RESET:
|
|
pT->OnReset();
|
|
break;
|
|
case PSN_QUERYCANCEL:
|
|
lResult = !pT->OnQueryCancel();
|
|
break;
|
|
case PSN_WIZNEXT:
|
|
lResult = !pT->OnWizardNext();
|
|
break;
|
|
case PSN_WIZBACK:
|
|
lResult = !pT->OnWizardBack();
|
|
break;
|
|
case PSN_WIZFINISH:
|
|
lResult = !pT->OnWizardFinish();
|
|
break;
|
|
case PSN_HELP:
|
|
lResult = pT->OnHelp();
|
|
break;
|
|
default:
|
|
bHandled = FALSE; // not handled
|
|
}
|
|
|
|
return lResult;
|
|
}
|
|
|
|
// Overridables
|
|
BOOL OnSetActive()
|
|
{
|
|
return TRUE;
|
|
}
|
|
BOOL OnKillActive()
|
|
{
|
|
return TRUE;
|
|
}
|
|
BOOL OnApply()
|
|
{
|
|
return TRUE;
|
|
}
|
|
void OnReset()
|
|
{
|
|
}
|
|
BOOL OnQueryCancel()
|
|
{
|
|
return TRUE; // ok to cancel
|
|
}
|
|
BOOL OnWizardBack()
|
|
{
|
|
return TRUE;
|
|
}
|
|
BOOL OnWizardNext()
|
|
{
|
|
return TRUE;
|
|
}
|
|
BOOL OnWizardFinish()
|
|
{
|
|
return TRUE;
|
|
}
|
|
BOOL OnHelp()
|
|
{
|
|
return TRUE;
|
|
}
|
|
};
|
|
#endif
|
|
|
|
#ifndef CBitmap
|
|
class CBitmap
|
|
{
|
|
public:
|
|
HBITMAP m_hBitmap;
|
|
|
|
CBitmap(HBITMAP hBitmap = NULL) : m_hBitmap(hBitmap)
|
|
{ }
|
|
~CBitmap()
|
|
{
|
|
if(m_hBitmap != NULL)
|
|
DeleteObject();
|
|
}
|
|
|
|
CBitmap& operator=(HBITMAP hBitmap)
|
|
{
|
|
m_hBitmap = hBitmap;
|
|
return *this;
|
|
}
|
|
|
|
void Attach(HBITMAP hBitmap)
|
|
{
|
|
m_hBitmap = hBitmap;
|
|
}
|
|
HBITMAP Detach()
|
|
{
|
|
HBITMAP hBitmap = m_hBitmap;
|
|
m_hBitmap = NULL;
|
|
return hBitmap;
|
|
}
|
|
|
|
operator HBITMAP() const { return m_hBitmap; }
|
|
|
|
HBITMAP LoadBitmap(LPCTSTR lpszResourceName)
|
|
{
|
|
_ASSERTE(m_hBitmap == NULL);
|
|
m_hBitmap = ::LoadBitmap(_Module.GetResourceInstance(), lpszResourceName);
|
|
return m_hBitmap;
|
|
}
|
|
HBITMAP LoadBitmap(UINT nIDResource)
|
|
{
|
|
_ASSERTE(m_hBitmap == NULL);
|
|
m_hBitmap = ::LoadBitmap(_Module.GetResourceInstance(), MAKEINTRESOURCE(nIDResource));
|
|
return m_hBitmap;
|
|
}
|
|
HBITMAP LoadOEMBitmap(UINT nIDBitmap) // for OBM_/OCR_/OIC_
|
|
{
|
|
_ASSERTE(m_hBitmap == NULL);
|
|
m_hBitmap = ::LoadBitmap(NULL, MAKEINTRESOURCE(nIDBitmap));
|
|
return m_hBitmap;
|
|
}
|
|
HBITMAP LoadMappedBitmap(UINT nIDBitmap, UINT nFlags = 0, LPCOLORMAP lpColorMap = NULL, int nMapSize = 0)
|
|
{
|
|
_ASSERTE(m_hBitmap == NULL);
|
|
m_hBitmap = ::CreateMappedBitmap(_Module.GetResourceInstance(), nIDBitmap, (WORD)nFlags, lpColorMap, nMapSize);
|
|
return m_hBitmap;
|
|
}
|
|
HBITMAP CreateBitmap(int nWidth, int nHeight, UINT nPlanes, UINT nBitcount, const void* lpBits)
|
|
{
|
|
_ASSERTE(m_hBitmap == NULL);
|
|
m_hBitmap = ::CreateBitmap(nWidth, nHeight, nPlanes, nBitcount, lpBits);
|
|
return m_hBitmap;
|
|
}
|
|
HBITMAP CreateBitmapIndirect(LPBITMAP lpBitmap)
|
|
{
|
|
_ASSERTE(m_hBitmap == NULL);
|
|
m_hBitmap = ::CreateBitmapIndirect(lpBitmap);
|
|
return m_hBitmap;
|
|
}
|
|
HBITMAP CreateCompatibleBitmap(HDC hDC, int nWidth, int nHeight)
|
|
{
|
|
_ASSERTE(m_hBitmap == NULL);
|
|
m_hBitmap = ::CreateCompatibleBitmap(hDC, nWidth, nHeight);
|
|
return m_hBitmap;
|
|
}
|
|
HBITMAP CreateDiscardableBitmap(HDC hDC, int nWidth, int nHeight)
|
|
{
|
|
_ASSERTE(m_hBitmap == NULL);
|
|
m_hBitmap = ::CreateDiscardableBitmap(hDC, nWidth, nHeight);
|
|
return m_hBitmap;
|
|
}
|
|
|
|
BOOL DeleteObject()
|
|
{
|
|
_ASSERTE(m_hBitmap != NULL);
|
|
BOOL bRet = ::DeleteObject(m_hBitmap);
|
|
if(bRet)
|
|
m_hBitmap = NULL;
|
|
return bRet;
|
|
}
|
|
|
|
// Attributes
|
|
int GetBitmap(BITMAP* pBitMap)
|
|
{
|
|
_ASSERTE(m_hBitmap != NULL);
|
|
return ::GetObject(m_hBitmap, sizeof(BITMAP), pBitMap);
|
|
}
|
|
// Operations
|
|
DWORD SetBitmapBits(DWORD dwCount, const void* lpBits)
|
|
{
|
|
_ASSERTE(m_hBitmap != NULL);
|
|
return ::SetBitmapBits(m_hBitmap, dwCount, lpBits);
|
|
}
|
|
DWORD GetBitmapBits(DWORD dwCount, LPVOID lpBits) const
|
|
{
|
|
_ASSERTE(m_hBitmap != NULL);
|
|
return ::GetBitmapBits(m_hBitmap, dwCount, lpBits);
|
|
}
|
|
BOOL SetBitmapDimension(int nWidth, int nHeight, LPSIZE lpSize = NULL)
|
|
{
|
|
_ASSERTE(m_hBitmap != NULL);
|
|
return ::SetBitmapDimensionEx(m_hBitmap, nWidth, nHeight, lpSize);
|
|
}
|
|
BOOL GetBitmapDimension(LPSIZE lpSize) const
|
|
{
|
|
_ASSERTE(m_hBitmap != NULL);
|
|
return ::GetBitmapDimensionEx(m_hBitmap, lpSize);
|
|
}
|
|
};
|
|
|
|
#endif
|
|
|
|
class ATL_NO_VTABLE ISnapInDataInterface
|
|
{
|
|
public:
|
|
STDMETHOD(Notify)(MMC_NOTIFY_TYPE event,
|
|
long arg,
|
|
long param,
|
|
BOOL bComponentData,
|
|
IConsole *pConsole,
|
|
IHeaderCtrl *pHeader,
|
|
IToolbar *pToolbar) = 0;
|
|
|
|
STDMETHOD(GetDispInfo)(SCOPEDATAITEM *pScopeDataItem) = 0;
|
|
|
|
STDMETHOD(GetResultViewType)(LPOLESTR *ppVIewType,
|
|
long *pViewOptions) = 0;
|
|
|
|
STDMETHOD(GetDisplayInfo)(RESULTDATAITEM *pResultDataItem) = 0;
|
|
|
|
STDMETHOD(AddMenuItems)(LPCONTEXTMENUCALLBACK piCallback,
|
|
long *pInsertionAllowed) = 0;
|
|
|
|
STDMETHOD(Command)(long lCommandID) = 0;
|
|
|
|
STDMETHOD(CreatePropertyPages)(LPPROPERTYSHEETCALLBACK lpProvider,
|
|
LONG_PTR handle,
|
|
IUnknown* pUnk) = 0;
|
|
|
|
STDMETHOD(QueryPagesFor)() = 0;
|
|
|
|
STDMETHOD(SetControlbar)(IControlbar *pControlbar,
|
|
IExtendControlbar *pExtendControlbar) = 0;
|
|
|
|
STDMETHOD(ControlbarNotify)(IControlbar *pControlbar,
|
|
IExtendControlbar *pExtendControlbar,
|
|
MMC_NOTIFY_TYPE event,
|
|
long arg,
|
|
long param) = 0;
|
|
|
|
STDMETHOD(GetScopeData)(SCOPEDATAITEM * *pScopeDataItem) = 0;
|
|
|
|
STDMETHOD(GetResultData)(RESULTDATAITEM * *pResultDataItem) = 0;
|
|
|
|
STDMETHOD(FillData)(CLIPFORMAT cf,
|
|
LPSTREAM pStream) = 0;
|
|
|
|
static ISnapInDataInterface* GetDataClass(IDataObject* pDataObj)
|
|
{
|
|
STGMEDIUM stgmedium = { TYMED_HGLOBAL, NULL };
|
|
FORMATETC formatetc = { m_CCF_SNAPIN_GETCOOKIE,
|
|
NULL,
|
|
DVASPECT_CONTENT,
|
|
-1,
|
|
TYMED_HGLOBAL
|
|
};
|
|
|
|
stgmedium.hGlobal = GlobalAlloc(0, sizeof(ISnapInDataInterface*));
|
|
|
|
if (FAILED(pDataObj->GetDataHere(&formatetc, &stgmedium)))
|
|
return NULL;
|
|
|
|
ISnapInDataInterface* pTemp = *(ISnapInDataInterface**)stgmedium.hGlobal;
|
|
|
|
GlobalFree(stgmedium.hGlobal);
|
|
|
|
return pTemp;
|
|
}
|
|
virtual HRESULT STDMETHODCALLTYPE GetDataObject(IDataObject** pDataObj) = 0;
|
|
|
|
static void Init()
|
|
{
|
|
m_CCF_NODETYPE = (CLIPFORMAT) RegisterClipboardFormat(CCF_NODETYPE);;
|
|
m_CCF_SZNODETYPE = (CLIPFORMAT) RegisterClipboardFormat(CCF_SZNODETYPE);
|
|
m_CCF_DISPLAY_NAME = (CLIPFORMAT) RegisterClipboardFormat(CCF_DISPLAY_NAME);
|
|
m_CCF_SNAPIN_CLASSID = (CLIPFORMAT) RegisterClipboardFormat(CCF_SNAPIN_CLASSID);
|
|
m_CCF_SNAPIN_GETCOOKIE = (CLIPFORMAT) RegisterClipboardFormat(_T("CCF_GETCOOKIE"));
|
|
}
|
|
public:
|
|
static CLIPFORMAT m_CCF_NODETYPE;
|
|
static CLIPFORMAT m_CCF_SZNODETYPE;
|
|
static CLIPFORMAT m_CCF_DISPLAY_NAME;
|
|
static CLIPFORMAT m_CCF_SNAPIN_CLASSID;
|
|
static CLIPFORMAT m_CCF_SNAPIN_GETCOOKIE;
|
|
};
|
|
|
|
template <class T>
|
|
class CSnapinObjectRootEx : public CComObjectRootEx< T >
|
|
{
|
|
public:
|
|
ISnapInDataInterface* GetDataClass(IDataObject* pDataObject)
|
|
{
|
|
return ISnapInDataInterface::GetDataClass(pDataObject);
|
|
}
|
|
};
|
|
|
|
#define EXTENSION_SNAPIN_DATACLASS(dataClass) dataClass m_##dataClass;
|
|
|
|
#define BEGIN_EXTENSION_SNAPIN_NODEINFO_MAP(classname) \
|
|
ISnapInDataInterface* GetDataClass(IDataObject* pDataObject) \
|
|
{ \
|
|
STGMEDIUM stgmedium = { TYMED_HGLOBAL, NULL }; \
|
|
FORMATETC formatetc = { ISnapInDataInterface::m_CCF_NODETYPE, \
|
|
NULL, \
|
|
DVASPECT_CONTENT, \
|
|
-1, \
|
|
TYMED_HGLOBAL \
|
|
}; \
|
|
\
|
|
stgmedium.hGlobal = GlobalAlloc(0, sizeof(GUID)); \
|
|
\
|
|
if (FAILED(pDataObject->GetDataHere(&formatetc, &stgmedium))) \
|
|
return NULL; \
|
|
\
|
|
GUID guid; \
|
|
memcpy(&guid, stgmedium.hGlobal, sizeof(GUID)); \
|
|
\
|
|
GlobalFree(stgmedium.hGlobal);
|
|
|
|
|
|
|
|
#define EXTENSION_SNAPIN_NODEINFO_ENTRY(dataClass) \
|
|
if (IsEqualGUID(guid, *(GUID*)m_##dataClass.GetNodeType())) \
|
|
{ \
|
|
m_##dataClass.InitDataClass(pDataObject); \
|
|
return &m_##dataClass; \
|
|
}
|
|
|
|
#define END_EXTENSION_SNAPIN_NODEINFO_MAP() \
|
|
return ISnapInDataInterface::GetDataClass(pDataObject); \
|
|
};
|
|
|
|
class ATL_NO_VTABLE CSnapInDataObjectImpl : public IDataObject,
|
|
public CComObjectRoot
|
|
{
|
|
public:
|
|
BEGIN_COM_MAP(CSnapInDataObjectImpl)
|
|
COM_INTERFACE_ENTRY(IDataObject)
|
|
END_COM_MAP()
|
|
STDMETHOD(GetData)(FORMATETC *pformatetcIn, STGMEDIUM *pmedium)
|
|
{
|
|
ATLTRACENOTIMPL(_T("SnapInDataObjectImpl::GetData\n"));
|
|
}
|
|
|
|
STDMETHOD(GetDataHere)(FORMATETC* pformatetc, STGMEDIUM* pmedium)
|
|
{
|
|
ATLTRACE(_T("SnapInDataObjectImpl::GetDataHere\n"));
|
|
if (pmedium == NULL)
|
|
return E_POINTER;
|
|
|
|
HRESULT hr = DV_E_TYMED;
|
|
// Make sure the type medium is HGLOBAL
|
|
if (pmedium->tymed == TYMED_HGLOBAL)
|
|
{
|
|
// Create the stream on the hGlobal passed in
|
|
LPSTREAM pStream;
|
|
hr = CreateStreamOnHGlobal(pmedium->hGlobal, FALSE, &pStream);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = m_pData->FillData(pformatetc->cfFormat, pStream);
|
|
pStream->Release();
|
|
}
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
STDMETHOD(QueryGetData)(FORMATETC* /* pformatetc */)
|
|
{
|
|
ATLTRACENOTIMPL(_T("SnapInDataObjectImpl::QueryGetData\n"));
|
|
}
|
|
STDMETHOD(GetCanonicalFormatEtc)(FORMATETC* /* pformatectIn */,FORMATETC* /* pformatetcOut */)
|
|
{
|
|
ATLTRACENOTIMPL(_T("SnapInDataObjectImpl::GetCanonicalFormatEtc\n"));
|
|
}
|
|
STDMETHOD(SetData)(FORMATETC* /* pformatetc */, STGMEDIUM* /* pmedium */, BOOL /* fRelease */)
|
|
{
|
|
ATLTRACENOTIMPL(_T("SnapInDataObjectImpl::SetData\n"));
|
|
}
|
|
STDMETHOD(EnumFormatEtc)(DWORD /* dwDirection */, IEnumFORMATETC** /* ppenumFormatEtc */)
|
|
{
|
|
ATLTRACENOTIMPL(_T("SnapInDataObjectImpl::EnumFormatEtc\n"));
|
|
}
|
|
STDMETHOD(DAdvise)(FORMATETC *pformatetc, DWORD advf, IAdviseSink *pAdvSink,
|
|
DWORD *pdwConnection)
|
|
{
|
|
ATLTRACENOTIMPL(_T("SnapInDataObjectImpl::SetData\n"));
|
|
}
|
|
STDMETHOD(DUnadvise)(DWORD dwConnection)
|
|
{
|
|
ATLTRACENOTIMPL(_T("SnapInDataObjectImpl::SetDatan\n"));
|
|
}
|
|
STDMETHOD(EnumDAdvise)(IEnumSTATDATA **ppenumAdvise)
|
|
{
|
|
ATLTRACENOTIMPL(_T("SnapInDataObjectImpl::SetData\n"));
|
|
}
|
|
|
|
ISnapInDataInterface* m_pData;
|
|
};
|
|
|
|
|
|
template <class T, class C>
|
|
class ATL_NO_VTABLE IComponentDataImpl : public IComponentData
|
|
{
|
|
public :
|
|
CComQIPtr<IConsoleNameSpace, &IID_IConsoleNameSpace> m_spConsoleNameSpace;
|
|
CComQIPtr<IConsole, &IID_IConsole> m_spConsole;
|
|
|
|
IComponentDataImpl()
|
|
{
|
|
m_pNode = NULL;
|
|
}
|
|
|
|
STDMETHOD(Initialize)(LPUNKNOWN pUnknown)
|
|
{
|
|
ATLTRACE(_T("IComponentDataImpl::Initialize\n"));
|
|
|
|
if (pUnknown == NULL)
|
|
{
|
|
ATLTRACE(_T("IComponentData::Initialize called with pUnknown == NULL\n"));
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
m_spConsoleNameSpace = pUnknown;
|
|
if (m_spConsoleNameSpace == NULL)
|
|
{
|
|
ATLTRACE(_T("QI for IConsoleNameSpace failed\n"));
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
m_spConsole = pUnknown;
|
|
if (m_spConsole == NULL)
|
|
{
|
|
ATLTRACE(_T("QI for IConsole failed\n"));
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHOD(CreateComponent)(LPCOMPONENT *ppComponent)
|
|
{
|
|
ATLTRACE(_T("IComponentDataImpl::CreateComponent\n"));
|
|
if (ppComponent == NULL)
|
|
{
|
|
ATLTRACE(_T("IComponentData::CreateComponent called with ppComponent == NULL\n"));
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
*ppComponent = NULL;
|
|
|
|
CComObject< C >* pComponent;
|
|
HRESULT hr = CComObject< C >::CreateInstance(&pComponent);
|
|
if (FAILED(hr))
|
|
{
|
|
ATLTRACE(_T("IComponentData::CreateComponent : Could not create IComponent object\n"));
|
|
return hr;
|
|
}
|
|
|
|
return pComponent->QueryInterface(IID_IComponent, (void**)ppComponent);
|
|
}
|
|
|
|
|
|
STDMETHOD(Notify)(
|
|
LPDATAOBJECT lpDataObject,
|
|
MMC_NOTIFY_TYPE event,
|
|
long arg,
|
|
long param)
|
|
{
|
|
ATLTRACE(_T("IComponentDataImpl::Notify\n"));
|
|
if (lpDataObject == NULL)
|
|
{
|
|
ATLTRACE(_T("IComponentData::Notify called with lpDataObject == NULL\n"));
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
ISnapInDataInterface* pData = ISnapInDataInterface::GetDataClass(lpDataObject);
|
|
if (pData == NULL)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
return pData->Notify(event, arg, param, TRUE, m_spConsole, NULL, NULL);
|
|
}
|
|
|
|
STDMETHOD(Destroy)(void)
|
|
{
|
|
ATLTRACE(_T("IComponentDataImpl::Destroy\n"));
|
|
|
|
m_spConsole.Release();
|
|
m_spConsoleNameSpace.Release();
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHOD(QueryDataObject)(long cookie,
|
|
DATA_OBJECT_TYPES type,
|
|
LPDATAOBJECT *ppDataObject)
|
|
{
|
|
ATLTRACE(_T("IComponentDataImpl::QueryDataObject\n"));
|
|
_ASSERTE(m_pNode != NULL);
|
|
|
|
if (ppDataObject == NULL)
|
|
{
|
|
ATLTRACE(_T("IComponentData::QueryDataObject called with ppDataObject == NULL\n"));
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
*ppDataObject = NULL;
|
|
|
|
if (cookie == NULL)
|
|
return m_pNode->GetDataObject(ppDataObject);
|
|
|
|
ISnapInDataInterface* pData = (ISnapInDataInterface*) cookie;
|
|
return pData->GetDataObject(ppDataObject);
|
|
}
|
|
|
|
STDMETHOD(GetDisplayInfo)(SCOPEDATAITEM *pScopeDataItem)
|
|
{
|
|
ATLTRACE(_T("IComponentDataImpl::GetDisplayInfo\n"));
|
|
|
|
if (pScopeDataItem == NULL)
|
|
{
|
|
ATLTRACE(_T("IComponentData::GetDisplayInfo called with pScopeDataItem == NULL\n"));
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
ISnapInDataInterface* pData= (ISnapInDataInterface*) pScopeDataItem->lParam;
|
|
if (pData == NULL)
|
|
pData = m_pNode;
|
|
|
|
if (pData == NULL)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
return pData->GetDispInfo(pScopeDataItem);
|
|
}
|
|
|
|
STDMETHOD(CompareObjects)(LPDATAOBJECT lpDataObjectA,
|
|
LPDATAOBJECT lpDataObjectB)
|
|
{
|
|
ATLTRACENOTIMPL(_T("IComponentDataImpl::CompareObjects\n"));
|
|
return S_FALSE;
|
|
}
|
|
|
|
protected:
|
|
ISnapInDataInterface* m_pNode;
|
|
};
|
|
|
|
|
|
template <class T>
|
|
class ATL_NO_VTABLE IComponentImpl : public IComponent
|
|
{
|
|
public:
|
|
//Review all of these may not be required
|
|
CComPtr<IConsole> m_spConsole;
|
|
CComQIPtr<IHeaderCtrl, &IID_IHeaderCtrl> m_spHeaderCtrl;
|
|
CComPtr<IImageList> m_spImageList;
|
|
CComPtr<IConsoleVerb> m_spConsoleVerb;
|
|
|
|
STDMETHOD(Initialize)(LPCONSOLE lpConsole)
|
|
{
|
|
ATLTRACE(_T("IComponentImpl::Initialize\n"));
|
|
|
|
if (lpConsole == NULL)
|
|
{
|
|
ATLTRACE(_T("lpConsole is NULL\n"));
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
m_spConsole = lpConsole;
|
|
|
|
m_spHeaderCtrl = lpConsole;
|
|
if (m_spHeaderCtrl == NULL)
|
|
{
|
|
ATLTRACE(_T("QI for IHeaderCtrl failed\n"));
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
HRESULT hr = m_spConsole->SetHeader(m_spHeaderCtrl);
|
|
if (FAILED(hr))
|
|
{
|
|
ATLTRACE(_T("IConsole::SetHeader failed (HRESULT = %x)\n"), hr);
|
|
return hr;
|
|
}
|
|
|
|
hr = lpConsole->QueryResultImageList(&m_spImageList);
|
|
if (FAILED(hr))
|
|
{
|
|
ATLTRACE(_T("IConsole::QueryResultImageList failed (HRESULT = %x)\n"), hr);
|
|
return hr;
|
|
}
|
|
|
|
lpConsole->QueryConsoleVerb(&m_spConsoleVerb) ;
|
|
if (FAILED(hr))
|
|
{
|
|
ATLTRACE(_T("IConsole::QueryConsoleVerb failed (HRESULT = %x)\n"), hr);
|
|
return hr;
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHOD(Notify)(LPDATAOBJECT lpDataObject,
|
|
MMC_NOTIFY_TYPE event,
|
|
long arg,
|
|
long param)
|
|
{
|
|
ATLTRACE(_T("IComponentImpl::Notify\n"));
|
|
if (lpDataObject == NULL)
|
|
{
|
|
ATLTRACE(_T("IComponent::Notify called with lpDataObject==NULL \n"));
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
ISnapInDataInterface* pData = ISnapInDataInterface::GetDataClass(lpDataObject);
|
|
if (pData == NULL)
|
|
{
|
|
ATLTRACE(_T("Invalid Data Object\n"));
|
|
return E_UNEXPECTED;
|
|
}
|
|
return pData->Notify(event, arg, param, FALSE, m_spConsole, m_spHeaderCtrl, NULL);
|
|
}
|
|
|
|
STDMETHOD(Destroy)(long cookie)
|
|
{
|
|
ATLTRACE(_T("IComponentImpl::Destroy\n"));
|
|
|
|
m_spConsoleVerb = NULL;
|
|
m_spImageList = NULL;
|
|
m_spHeaderCtrl.Release();
|
|
m_spConsole.Release();
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHOD(QueryDataObject)(long cookie,
|
|
DATA_OBJECT_TYPES type,
|
|
LPDATAOBJECT *ppDataObject)
|
|
{
|
|
ATLTRACE(_T("IComponentImpl::QueryDataObject\n"));
|
|
|
|
if (ppDataObject == NULL)
|
|
{
|
|
ATLTRACE(_T("IComponent::QueryDataObject called with ppDataObject==NULL \n"));
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
if (cookie == NULL)
|
|
{
|
|
ATLTRACE(_T("IComponent::QueryDataObject called with cookie==NULL \n"));
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
*ppDataObject = NULL;
|
|
|
|
ISnapInDataInterface* pData = (ISnapInDataInterface*) cookie;
|
|
return pData->GetDataObject(ppDataObject);
|
|
}
|
|
|
|
STDMETHOD(GetResultViewType)(long cookie,
|
|
LPOLESTR *ppViewType,
|
|
long *pViewOptions)
|
|
{
|
|
ATLTRACE(_T("IComponentImpl::GetResultViewType\n"));
|
|
|
|
if (cookie == NULL)
|
|
{
|
|
*ppViewType = NULL;
|
|
*pViewOptions = MMC_VIEW_OPTIONS_NONE;
|
|
return S_OK;
|
|
}
|
|
|
|
ISnapInDataInterface* pData = (ISnapInDataInterface*)cookie;
|
|
return pData->GetResultViewType(ppViewType, pViewOptions);
|
|
}
|
|
|
|
STDMETHOD(GetDisplayInfo)(RESULTDATAITEM *pResultDataItem)
|
|
{
|
|
ATLTRACE(_T("IComponentImpl::GetDisplayInfo\n"));
|
|
if (pResultDataItem == NULL)
|
|
{
|
|
ATLTRACE(_T("IComponent::GetDisplayInfo called with pResultDataItem==NULL\n"));
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
ISnapInDataInterface* pData = (ISnapInDataInterface*) pResultDataItem->lParam;
|
|
|
|
if (pData == NULL)
|
|
{
|
|
ATLTRACE(_T("Invalid Item\n"));
|
|
return E_UNEXPECTED;
|
|
}
|
|
return pData->GetDisplayInfo(pResultDataItem);
|
|
}
|
|
|
|
STDMETHOD(CompareObjects)( LPDATAOBJECT lpDataObjectA,
|
|
LPDATAOBJECT lpDataObjectB)
|
|
{
|
|
ATLTRACENOTIMPL(_T("IComponentImpl::CompareObjects\n"));
|
|
}
|
|
};
|
|
|
|
template <class T, class D>
|
|
class ATL_NO_VTABLE IResultDataCompareImpl : public IResultDataCompare
|
|
{
|
|
public:
|
|
STDMETHOD(Compare)(long lUserParam,
|
|
long cookieA,
|
|
long cookieB,
|
|
int *pnResult)
|
|
{
|
|
ATLTRACENOTIMPL(_T("IResultDataCompareImpl::Compare"));
|
|
}
|
|
};
|
|
|
|
|
|
template <class T>
|
|
class ATL_NO_VTABLE IExtendContextMenuImpl : public IExtendContextMenu
|
|
{
|
|
public:
|
|
STDMETHOD(AddMenuItems)(LPDATAOBJECT pDataObject,
|
|
LPCONTEXTMENUCALLBACK piCallback,
|
|
long *pInsertionAllowed)
|
|
{
|
|
ATLTRACE(_T("IExtendContextMenuImpl::AddMenuItems\n"));
|
|
if (pDataObject == NULL)
|
|
{
|
|
ATLTRACE(_T("IExtendContextMenu::AddMenuItems called with pDataObject==NULL\n"));
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
T* pT = static_cast<T*>(this);
|
|
ISnapInDataInterface* pData = pT->GetDataClass(pDataObject);
|
|
|
|
if (pData == NULL)
|
|
{
|
|
ATLTRACE(_T("Invalid Data Object\n"));
|
|
return E_UNEXPECTED;
|
|
}
|
|
return pData->AddMenuItems(piCallback, pInsertionAllowed);
|
|
}
|
|
|
|
STDMETHOD(Command)(long lCommandID,
|
|
LPDATAOBJECT pDataObject)
|
|
{
|
|
ATLTRACE(_T("IExtendContextMenuImpl::Command\n"));
|
|
if (pDataObject == NULL)
|
|
{
|
|
ATLTRACE(_T("IExtendContextMenu::Command called with pDataObject==NULL\n"));
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
T* pT = static_cast<T*>(this);
|
|
ISnapInDataInterface* pData = pT->GetDataClass(pDataObject);
|
|
|
|
if (pData == NULL)
|
|
{
|
|
ATLTRACE(_T("Invalid Data Object\n"));
|
|
return E_UNEXPECTED;
|
|
}
|
|
return pData->Command(lCommandID);
|
|
}
|
|
};
|
|
|
|
template<class T>
|
|
class ATL_NO_VTABLE IExtendPropertySheetImpl : public IExtendPropertySheet
|
|
{
|
|
public:
|
|
STDMETHOD(CreatePropertyPages)(LPPROPERTYSHEETCALLBACK lpProvider,
|
|
LONG_PTR handle,
|
|
LPDATAOBJECT pDataObject)
|
|
{
|
|
ATLTRACE(_T("IExtendPropertySheetImpl::CreatePropertyPages\n"));
|
|
if (pDataObject == NULL)
|
|
{
|
|
ATLTRACE(_T("IExtendPropertySheetImpl::CreatePropertyPages called with pDataObject==NULL\n"));
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
T* pT = static_cast<T*>(this);
|
|
ISnapInDataInterface* pData = pT->GetDataClass(pDataObject);
|
|
|
|
if (pData == NULL)
|
|
{
|
|
ATLTRACE(_T("Invalid Data Object\n"));
|
|
return E_UNEXPECTED;
|
|
}
|
|
return pData->CreatePropertyPages(lpProvider, handle, this);
|
|
}
|
|
|
|
STDMETHOD(QueryPagesFor)(LPDATAOBJECT pDataObject)
|
|
{
|
|
ATLTRACE(_T("IExtendPropertySheetImpl::QueryPagesFor\n"));
|
|
if (pDataObject == NULL)
|
|
{
|
|
ATLTRACE(_T("IExtendPropertySheetImpl::QueryPagesFor called with pDataObject==NULL\n"));
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
T* pT = static_cast<T*>(this);
|
|
ISnapInDataInterface* pData = pT->GetDataClass(pDataObject);
|
|
|
|
if (pData == NULL)
|
|
{
|
|
ATLTRACE(_T("Invalid Data Object\n"));
|
|
return E_UNEXPECTED;
|
|
}
|
|
return pData->QueryPagesFor();
|
|
}
|
|
};
|
|
|
|
template <class T>
|
|
class ATL_NO_VTABLE IExtendControlbarImpl : public IExtendControlbar
|
|
{
|
|
CComPtr <IControlbar> m_spControlbar;
|
|
public:
|
|
STDMETHOD(SetControlbar)(LPCONTROLBAR pControlbar)
|
|
{
|
|
ATLTRACE(_T("IExtendControlbarImpl::SetControlbar\n"));
|
|
if (pControlbar == NULL)
|
|
{
|
|
ATLTRACE(_T("IExtendControlbar::SetControlbar called with pControlbar==NULL\n"));
|
|
return E_UNEXPECTED;
|
|
}
|
|
m_spControlbar = pControlbar;
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHOD(ControlbarNotify)(MMC_NOTIFY_TYPE event,
|
|
long arg,
|
|
long param)
|
|
{
|
|
ATLTRACE(_T("IExtendControlbarImpl::ControlbarNotify\n"));
|
|
|
|
ISnapInDataInterface* pData = NULL;
|
|
T* pT = static_cast<T*>(this);
|
|
if (event == MMCN_BTN_CLICK)
|
|
{
|
|
pData = pT->GetDataClass((IDataObject*) arg);
|
|
}
|
|
else if (event == MMCN_SELECT)
|
|
{
|
|
BOOL bScope = (BOOL) LOWORD(arg);
|
|
if (bScope)
|
|
{
|
|
LPDATAOBJECT* ppDataobject = (LPDATAOBJECT*) param;
|
|
if (ppDataobject[0])
|
|
{
|
|
pData = pT->GetDataClass(ppDataobject[0]);
|
|
if (pData != NULL)
|
|
pData->ControlbarNotify(m_spControlbar, this, event, arg, param);
|
|
}
|
|
|
|
if (ppDataobject[1])
|
|
{
|
|
pData = pT->GetDataClass(ppDataobject[1]);
|
|
}
|
|
}
|
|
else
|
|
pData = pT->GetDataClass((IDataObject*) param);
|
|
}
|
|
if (pData == NULL)
|
|
{
|
|
ATLTRACE(_T("Invalid Data Object\n"));
|
|
return E_UNEXPECTED;
|
|
}
|
|
HRESULT hr = pData->ControlbarNotify(m_spControlbar, this, event, arg, param);
|
|
ATLTRACE(_T("Exiting : IExtendControlbarImpl::ControlbarNotify\n"));
|
|
return hr;
|
|
}
|
|
};
|
|
|
|
#define SNAPINMENUID(id) \
|
|
public: \
|
|
static const UINT GetMenuID() \
|
|
{ \
|
|
static const UINT IDMENU = id; \
|
|
return id; \
|
|
}
|
|
|
|
#define EXT_SNAPINMENUID(id) \
|
|
public: \
|
|
static const UINT GetMenuID() \
|
|
{ \
|
|
static const UINT IDMENU = id; \
|
|
return id; \
|
|
}
|
|
|
|
|
|
#define BEGIN_SNAPINCOMMAND_MAP(theClass, bIsExtension) \
|
|
typedef CSnapInDataInterface< theClass, bIsExtension > baseClass; \
|
|
HRESULT ProcessCommand(UINT nID, IDataObject* pDataObject) \
|
|
{ \
|
|
BOOL _bIsExtension = bIsExtension;
|
|
|
|
#define SNAPINCOMMAND_ENTRY(id, func) \
|
|
if (id == nID) \
|
|
return func();
|
|
|
|
#define SNAPINCOMMAND_RANGE_ENTRY(id1, id2, func) \
|
|
if (id1 >= nID && nID <= id2) \
|
|
return func(nID);
|
|
|
|
#define END_SNAPINCOMMAND_MAP() \
|
|
return baseClass::ProcessCommand(nID, pDataObject); \
|
|
}
|
|
|
|
struct CSnapInToolBarData
|
|
{
|
|
WORD wVersion;
|
|
WORD wWidth;
|
|
WORD wHeight;
|
|
WORD wItemCount;
|
|
//WORD aItems[wItemCount]
|
|
|
|
WORD* items()
|
|
{ return (WORD*)(this+1); }
|
|
};
|
|
|
|
#define RT_TOOLBAR MAKEINTRESOURCE(241)
|
|
|
|
struct CSnapInToolbarInfo
|
|
{
|
|
public:
|
|
TCHAR** m_pStrToolTip;
|
|
TCHAR** m_pStrButtonText;
|
|
UINT* m_pnButtonID;
|
|
UINT m_idToolbar;
|
|
UINT m_nButtonCount;
|
|
IToolbar* m_pToolbar;
|
|
|
|
~CSnapInToolbarInfo()
|
|
{
|
|
if (m_pStrToolTip)
|
|
{
|
|
for (UINT i = 0; i < m_nButtonCount; i++)
|
|
{
|
|
delete m_pStrToolTip[i];
|
|
m_pStrToolTip[i] = NULL;
|
|
}
|
|
delete [] m_pStrToolTip;
|
|
m_pStrToolTip = NULL;
|
|
}
|
|
|
|
if (m_pStrButtonText)
|
|
{
|
|
for (UINT i = 0; i < m_nButtonCount; i++)
|
|
{
|
|
delete m_pStrButtonText[i];
|
|
m_pStrButtonText[i] = NULL;
|
|
}
|
|
|
|
delete [] m_pStrButtonText;
|
|
m_pStrButtonText = NULL;
|
|
}
|
|
if (m_pnButtonID)
|
|
{
|
|
delete m_pnButtonID;
|
|
m_pnButtonID = NULL;
|
|
}
|
|
|
|
m_nButtonCount = 0;
|
|
if (m_pToolbar)
|
|
m_pToolbar->Release();
|
|
m_pToolbar = NULL;
|
|
}
|
|
};
|
|
|
|
#define BEGIN_SNAPINTOOLBARID_MAP(theClass) \
|
|
public: \
|
|
static CSnapInToolbarInfo* GetToolbarInfo() \
|
|
{ \
|
|
static CSnapInToolbarInfo m_toolbarInfo[] = \
|
|
{
|
|
|
|
#define SNAPINTOOLBARID_ENTRY(id) \
|
|
{ NULL, NULL, NULL, id, 0, NULL},
|
|
|
|
#define END_SNAPINTOOLBARID_MAP() \
|
|
{ NULL, NULL, NULL, 0, 0, NULL} \
|
|
}; \
|
|
return m_toolbarInfo; \
|
|
}
|
|
|
|
template <class T, BOOL bIsExtension = FALSE>
|
|
class ATL_NO_VTABLE CSnapInDataInterface : public ISnapInDataInterface
|
|
{
|
|
public:
|
|
OLECHAR* m_pszDisplayName;
|
|
|
|
SCOPEDATAITEM m_scopeDataItem;
|
|
RESULTDATAITEM m_resultDataItem;
|
|
CSnapInDataInterface()
|
|
{
|
|
m_pszDisplayName = NULL;
|
|
|
|
memset(&m_scopeDataItem, 0, sizeof(SCOPEDATAITEM));
|
|
|
|
m_scopeDataItem.mask = SDI_STR | SDI_IMAGE | SDI_OPENIMAGE | SDI_PARAM | SDI_PARENT;
|
|
m_scopeDataItem.displayname = MMC_CALLBACK;
|
|
m_scopeDataItem.nImage = 0;
|
|
m_scopeDataItem.nOpenImage = 1;
|
|
|
|
memset(&m_resultDataItem, 0, sizeof(RESULTDATAITEM));
|
|
|
|
m_resultDataItem.mask = RDI_STR | RDI_IMAGE | RDI_PARAM;
|
|
m_resultDataItem.str = MMC_CALLBACK;
|
|
m_resultDataItem.nImage = 2;
|
|
|
|
}
|
|
|
|
virtual ~CSnapInDataInterface()
|
|
{
|
|
delete [] m_pszDisplayName;
|
|
m_pszDisplayName = NULL;
|
|
}
|
|
|
|
public:
|
|
|
|
STDMETHOD(Notify)( MMC_NOTIFY_TYPE event,
|
|
long arg,
|
|
long param,
|
|
BOOL bComponentData,
|
|
IConsole* pConsole,
|
|
IHeaderCtrl* pHeader,
|
|
IToolbar* pToolbar)
|
|
{
|
|
ATLTRACENOTIMPL(_T("ISnapInDataInterfaceImpl::Notify"));
|
|
}
|
|
|
|
STDMETHOD(GetDispInfo)(SCOPEDATAITEM *pScopeDataItem)
|
|
{
|
|
ATLTRACENOTIMPL(_T("ISnapInDataInterfaceImpl::GetDispInfo"));
|
|
}
|
|
|
|
STDMETHOD(GetResultViewType)(LPOLESTR *ppVIewType,
|
|
long *pViewOptions)
|
|
{
|
|
ATLTRACENOTIMPL(_T("ISnapInDataInterfaceImpl::GetResultViewType"));
|
|
}
|
|
|
|
STDMETHOD(GetDisplayInfo)(RESULTDATAITEM *pResultDataItem)
|
|
{
|
|
ATLTRACE(_T("ISnapInDataInterfaceImpl::GetDisplayInfo"));
|
|
T* pT = static_cast<T*> (this);
|
|
|
|
if (pResultDataItem->bScopeItem)
|
|
{
|
|
if (pResultDataItem->mask & RDI_STR)
|
|
{
|
|
pResultDataItem->str = pT->GetResultPaneInfo(pResultDataItem->nCol);
|
|
}
|
|
if (pResultDataItem->mask & RDI_IMAGE)
|
|
{
|
|
pResultDataItem->nImage = m_scopeDataItem.nImage;
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
if (pResultDataItem->mask & RDI_STR)
|
|
{
|
|
pResultDataItem->str = pT->GetResultPaneInfo(pResultDataItem->nCol);
|
|
}
|
|
if (pResultDataItem->mask & RDI_IMAGE)
|
|
{
|
|
pResultDataItem->nImage = m_resultDataItem.nImage;
|
|
}
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHOD(AddMenuItems)(LPCONTEXTMENUCALLBACK piCallback,
|
|
long *pInsertionAllowed)
|
|
{
|
|
ATLTRACE(_T("ISnapInDataInterfaceImpl::AddMenuItems"));
|
|
T* pT = static_cast<T*> (this);
|
|
UINT menuID = pT->GetMenuID();
|
|
if (menuID == 0)
|
|
return S_OK;
|
|
// return SnapInMenuHelper<T> (pT, menuID, piCallback, pInsertionAllowed);
|
|
|
|
HMENU hMenu = LoadMenu(_Module.GetResourceInstance(), MAKEINTRESOURCE(menuID));
|
|
long insertionID;
|
|
if (hMenu)
|
|
{
|
|
for (int i = 0; 1; i++)
|
|
{
|
|
HMENU hSubMenu = GetSubMenu(hMenu, i);
|
|
if (hSubMenu == NULL)
|
|
break;
|
|
|
|
MENUITEMINFO menuItemInfo;
|
|
memset(&menuItemInfo, 0, sizeof(menuItemInfo));
|
|
menuItemInfo.cbSize = sizeof(menuItemInfo);
|
|
|
|
switch (i)
|
|
{
|
|
case 0:
|
|
if (! (*pInsertionAllowed & CCM_INSERTIONALLOWED_TOP) )
|
|
continue;
|
|
insertionID = CCM_INSERTIONPOINTID_PRIMARY_TOP;
|
|
break;
|
|
|
|
case 1:
|
|
if (! (*pInsertionAllowed & CCM_INSERTIONALLOWED_NEW) )
|
|
continue;
|
|
if (bIsExtension)
|
|
insertionID = CCM_INSERTIONPOINTID_3RDPARTY_NEW;
|
|
else
|
|
insertionID = CCM_INSERTIONPOINTID_PRIMARY_NEW;
|
|
break;
|
|
|
|
case 2:;
|
|
if (! (*pInsertionAllowed & CCM_INSERTIONALLOWED_TASK) )
|
|
continue;
|
|
if (bIsExtension)
|
|
insertionID = CCM_INSERTIONPOINTID_3RDPARTY_TASK;
|
|
else
|
|
insertionID = CCM_INSERTIONPOINTID_PRIMARY_TASK;
|
|
break;
|
|
case 3:;
|
|
if (! (*pInsertionAllowed & CCM_INSERTIONALLOWED_VIEW) )
|
|
continue;
|
|
insertionID = CCM_INSERTIONPOINTID_PRIMARY_VIEW;
|
|
break;
|
|
default:
|
|
{
|
|
insertionID = 0;
|
|
continue;
|
|
// Review
|
|
// Determine what to do here.
|
|
menuItemInfo.fMask = MIIM_TYPE ;
|
|
menuItemInfo.fType = MFT_STRING;
|
|
TCHAR buf[128];
|
|
menuItemInfo.dwTypeData = buf;
|
|
menuItemInfo.cch = 128;
|
|
if (!GetMenuItemInfo(hMenu, i, TRUE, &menuItemInfo))
|
|
continue;
|
|
// insertionID = _ttol(buf);
|
|
}
|
|
break;
|
|
}
|
|
|
|
menuItemInfo.fMask = MIIM_TYPE | MIIM_STATE | MIIM_ID;
|
|
menuItemInfo.fType = MFT_STRING;
|
|
TCHAR buf[128];
|
|
menuItemInfo.dwTypeData = buf;
|
|
|
|
for (int j = 0; 1; j++)
|
|
{
|
|
menuItemInfo.fMask = MIIM_TYPE | MIIM_STATE | MIIM_ID;
|
|
menuItemInfo.fType = MFT_STRING;
|
|
menuItemInfo.cch = 128;
|
|
TCHAR strStatusBar[257];
|
|
|
|
if (!GetMenuItemInfo(hSubMenu, j, TRUE, &menuItemInfo))
|
|
break;
|
|
|
|
pT->UpdateMenuState(menuItemInfo.wID, buf, &menuItemInfo.fState);
|
|
LoadString(_Module.GetResourceInstance(), menuItemInfo.wID, strStatusBar, 257);
|
|
|
|
CONTEXTMENUITEM contextMenuItem;
|
|
memset(&contextMenuItem, 0, sizeof(contextMenuItem));
|
|
contextMenuItem.strName = buf;
|
|
contextMenuItem.strStatusBarText = strStatusBar;
|
|
contextMenuItem.lCommandID = menuItemInfo.wID;
|
|
contextMenuItem.lInsertionPointID = insertionID;
|
|
contextMenuItem.fFlags = menuItemInfo.fState;
|
|
|
|
HRESULT hr = piCallback->AddItem(&contextMenuItem);
|
|
_ASSERTE(SUCCEEDED(hr));
|
|
}
|
|
}
|
|
DestroyMenu(hMenu);
|
|
}
|
|
if (!bIsExtension)
|
|
*pInsertionAllowed = CCM_INSERTIONALLOWED_TOP | CCM_INSERTIONALLOWED_NEW |
|
|
CCM_INSERTIONALLOWED_TASK | CCM_INSERTIONALLOWED_VIEW;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHOD(Command)(long lCommandID)
|
|
{
|
|
ATLTRACE(_T("ISnapInDataInterfaceImpl::Command\n"));
|
|
T* pT = static_cast<T*>(this);
|
|
return pT->ProcessCommand(lCommandID, NULL);
|
|
}
|
|
|
|
STDMETHOD(CreatePropertyPages)(LPPROPERTYSHEETCALLBACK lpProvider,
|
|
LONG_PTR handle, IUnknown* pUnk)
|
|
{
|
|
ATLTRACENOTIMPL(_T("ISnapInDataInterfaceImpl::CreatePropertyPages"));
|
|
}
|
|
|
|
STDMETHOD(QueryPagesFor)(void)
|
|
{
|
|
ATLTRACENOTIMPL(_T("ISnapInDataInterfaceImpl::QueryPagesFor"));
|
|
}
|
|
|
|
STDMETHOD(SetControlbar)(IControlbar *pControlbar, IExtendControlbar* pExtendControlBar)
|
|
{
|
|
ATLTRACE(_T("ISnapInDataInterfaceImpl::SetControlbar\n"));
|
|
T* pT = static_cast<T*>(this);
|
|
|
|
CSnapInToolbarInfo* pInfo = pT->GetToolbarInfo();
|
|
if (pInfo == NULL)
|
|
return S_OK;
|
|
|
|
for( ; pInfo->m_idToolbar; pInfo++)
|
|
{
|
|
if (pInfo->m_pToolbar)
|
|
continue;
|
|
|
|
HBITMAP hBitmap = LoadBitmap(_Module.GetResourceInstance(), MAKEINTRESOURCE(pInfo->m_idToolbar));
|
|
if (hBitmap == NULL)
|
|
return S_OK;
|
|
|
|
HRSRC hRsrc = ::FindResource(_Module.GetResourceInstance(), MAKEINTRESOURCE(pInfo->m_idToolbar), RT_TOOLBAR);
|
|
if (hRsrc == NULL)
|
|
return S_OK;
|
|
|
|
HGLOBAL hGlobal = LoadResource(_Module.GetResourceInstance(), hRsrc);
|
|
if (hGlobal == NULL)
|
|
return S_OK;
|
|
|
|
CSnapInToolBarData* pData = (CSnapInToolBarData*)LockResource(hGlobal);
|
|
if (pData == NULL)
|
|
return S_OK;
|
|
_ASSERTE(pData->wVersion == 1);
|
|
_ASSERTE(pData->wWidth == 16);
|
|
_ASSERTE(pData->wHeight == 16);
|
|
|
|
pInfo->m_nButtonCount = pData->wItemCount;
|
|
pInfo->m_pnButtonID = new UINT[pInfo->m_nButtonCount];
|
|
MMCBUTTON *pButtons = new MMCBUTTON[pData->wItemCount];
|
|
|
|
pInfo->m_pStrToolTip = new TCHAR* [pData->wItemCount];
|
|
if (pInfo->m_pStrToolTip == NULL)
|
|
continue;
|
|
|
|
for (int i = 0, j = 0; i < pData->wItemCount; i++)
|
|
{
|
|
pInfo->m_pStrToolTip[i] = NULL;
|
|
memset(&pButtons[i], 0, sizeof(MMCBUTTON));
|
|
pInfo->m_pnButtonID[i] = pButtons[i].idCommand = pData->items()[i];
|
|
if (pButtons[i].idCommand)
|
|
{
|
|
pButtons[i].nBitmap = j++;
|
|
// get the statusbar string and allow modification of the button state
|
|
TCHAR strStatusBar[512];
|
|
LoadString(_Module.GetResourceInstance(), pButtons[i].idCommand, strStatusBar, 512);
|
|
|
|
pInfo->m_pStrToolTip[i] = new TCHAR[lstrlen(strStatusBar) + 1];
|
|
if (pInfo->m_pStrToolTip[i] == NULL)
|
|
continue;
|
|
lstrcpy(pInfo->m_pStrToolTip[i], strStatusBar);
|
|
pButtons[i].lpTooltipText = pInfo->m_pStrToolTip[i];
|
|
pButtons[i].lpButtonText = _T("");
|
|
pT->SetToolbarButtonInfo(pButtons[i].idCommand, &pButtons[i].fsState, &pButtons[i].fsType);
|
|
}
|
|
else
|
|
{
|
|
pButtons[i].lpTooltipText = _T("");
|
|
pButtons[i].lpButtonText = _T("");
|
|
pButtons[i].fsType = TBSTYLE_SEP;
|
|
}
|
|
}
|
|
|
|
HRESULT hr = pControlbar->Create(TOOLBAR, pExtendControlBar, reinterpret_cast<LPUNKNOWN*>(&pInfo->m_pToolbar));
|
|
if (FAILED(hr))
|
|
continue;
|
|
|
|
hr = pInfo->m_pToolbar->AddBitmap(pData->wItemCount, hBitmap, pData->wWidth, pData->wHeight, RGB(192, 192, 192));
|
|
if (FAILED(hr))
|
|
{
|
|
pInfo->m_pToolbar->Release();
|
|
pInfo->m_pToolbar = NULL;
|
|
continue;
|
|
}
|
|
|
|
hr = pInfo->m_pToolbar->AddButtons(pData->wItemCount, pButtons);
|
|
if (FAILED(hr))
|
|
{
|
|
pInfo->m_pToolbar->Release();
|
|
pInfo->m_pToolbar = NULL;
|
|
}
|
|
|
|
delete [] pButtons;
|
|
}
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHOD(ControlbarNotify)(IControlbar *pControlbar,
|
|
IExtendControlbar *pExtendControlbar,
|
|
MMC_NOTIFY_TYPE event,
|
|
long arg, long param)
|
|
{
|
|
ATLTRACE(_T("ISnapInDataInterfaceImpl::ControlbarNotify\n"));
|
|
T* pT = static_cast<T*>(this);
|
|
|
|
SetControlbar(pControlbar, pExtendControlbar);
|
|
|
|
if(event == MMCN_SELECT)
|
|
{
|
|
BOOL bScope = (BOOL) LOWORD(arg);
|
|
BOOL bSelect = (BOOL) HIWORD (arg);
|
|
|
|
if (!bScope)
|
|
{
|
|
CSnapInToolbarInfo* pInfo = pT->GetToolbarInfo();
|
|
if (pInfo == NULL)
|
|
return S_OK;
|
|
|
|
if (!bSelect)
|
|
return S_OK;
|
|
|
|
for(; pInfo->m_idToolbar; pInfo++)
|
|
{
|
|
for (UINT i = 0; i < pInfo->m_nButtonCount; i++)
|
|
{
|
|
if (pInfo->m_pnButtonID[i])
|
|
{
|
|
for (int j = ENABLED; j <= BUTTONPRESSED; j++)
|
|
{
|
|
pInfo->m_pToolbar->SetButtonState(pInfo->m_pnButtonID[i],
|
|
(MMC_BUTTON_STATE)j,
|
|
pT->UpdateToolbarButton(pInfo->m_pnButtonID[i], (MMC_BUTTON_STATE)j));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
LPDATAOBJECT* pData = (LPDATAOBJECT*) param;
|
|
if (pData[0] == NULL)
|
|
return S_OK;
|
|
if (pData[0] == pData[1])
|
|
return S_OK;
|
|
ISnapInDataInterface* pCookie = ISnapInDataInterface::GetDataClass(pData[0]);
|
|
if (pCookie == (ISnapInDataInterface*)this)
|
|
{
|
|
CSnapInToolbarInfo* pInfo = pT->GetToolbarInfo();
|
|
if (pInfo == NULL)
|
|
return S_OK;
|
|
|
|
for(; pInfo->m_idToolbar; pInfo++)
|
|
{
|
|
pControlbar->Detach(pInfo->m_pToolbar);
|
|
}
|
|
return S_OK;
|
|
}
|
|
|
|
CSnapInToolbarInfo* pInfo = pT->GetToolbarInfo();
|
|
if (pInfo == NULL)
|
|
return S_OK;
|
|
|
|
for(; pInfo->m_idToolbar; pInfo++)
|
|
{
|
|
pControlbar->Attach(TOOLBAR, pInfo->m_pToolbar);
|
|
for (UINT i = 0; i < pInfo->m_nButtonCount; i++)
|
|
{
|
|
if (pInfo->m_pnButtonID[i])
|
|
{
|
|
for (int j = ENABLED; j <= BUTTONPRESSED; j++)
|
|
{
|
|
pInfo->m_pToolbar->SetButtonState(pInfo->m_pnButtonID[i],
|
|
(MMC_BUTTON_STATE)j,
|
|
pT->UpdateToolbarButton(pInfo->m_pnButtonID[i], (MMC_BUTTON_STATE)j));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return S_OK;
|
|
}
|
|
}
|
|
return pT->ProcessCommand((UINT) param, NULL);
|
|
}
|
|
|
|
STDMETHOD(GetScopeData)(SCOPEDATAITEM **pScopeDataItem)
|
|
{
|
|
if (pScopeDataItem == NULL)
|
|
return E_FAIL;
|
|
|
|
*pScopeDataItem = &m_scopeDataItem;
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHOD(GetResultData)(RESULTDATAITEM **pResultDataItem)
|
|
{
|
|
if (pResultDataItem == NULL)
|
|
return E_FAIL;
|
|
|
|
*pResultDataItem = &m_resultDataItem;
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHOD(GetDataObject)(IDataObject** pDataObj)
|
|
{
|
|
CComObject<CSnapInDataObjectImpl>* pData;
|
|
HRESULT hr = CComObject<CSnapInDataObjectImpl>::CreateInstance(&pData);
|
|
if (FAILED(hr))
|
|
return hr;
|
|
|
|
pData->m_pData = this;
|
|
|
|
hr = pData->QueryInterface(IID_IDataObject, (void**)(pDataObj));
|
|
|
|
return hr;
|
|
}
|
|
|
|
void UpdateMenuState(UINT id, LPTSTR pBuf, UINT *flags)
|
|
{
|
|
return;
|
|
}
|
|
|
|
void SetToolbarButtonInfo(UINT id, BYTE *fsState, BYTE *fsType)
|
|
{
|
|
*fsState = TBSTATE_ENABLED;
|
|
*fsType = TBSTYLE_BUTTON;
|
|
}
|
|
|
|
BOOL UpdateToolbarButton(UINT id, BYTE fsState)
|
|
{
|
|
if (fsState == ENABLED)
|
|
return TRUE;
|
|
return FALSE;
|
|
}
|
|
|
|
HRESULT ProcessCommand(UINT nID, IDataObject* pDataObject)
|
|
{
|
|
ATLTRACE(_T("No handler for item with ID %d\n"), nID);
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHOD (FillData)(CLIPFORMAT cf, LPSTREAM pStream)
|
|
{
|
|
HRESULT hr = DV_E_CLIPFORMAT;
|
|
ULONG uWritten;
|
|
|
|
T* pT = static_cast<T*> (this);
|
|
|
|
if (cf == m_CCF_NODETYPE)
|
|
{
|
|
hr = pStream->Write(pT->GetNodeType(), sizeof(GUID), &uWritten);
|
|
return hr;
|
|
}
|
|
|
|
if (cf == m_CCF_SZNODETYPE)
|
|
{
|
|
hr = pStream->Write(pT->GetSZNodeType(), (lstrlen((LPCTSTR)pT->GetSZNodeType()) + 1 )* sizeof(TCHAR), &uWritten);
|
|
return hr;
|
|
}
|
|
|
|
if (cf == m_CCF_DISPLAY_NAME)
|
|
{
|
|
hr = pStream->Write(pT->GetDisplayName(), (lstrlen((LPCTSTR)pT->GetDisplayName()) + 1) * sizeof(TCHAR), &uWritten);
|
|
return hr;
|
|
}
|
|
|
|
if (cf == m_CCF_SNAPIN_CLASSID)
|
|
{
|
|
hr = pStream->Write(pT->GetSnapInCLSID(), sizeof(CLSID), &uWritten);
|
|
return hr;
|
|
}
|
|
if (cf == m_CCF_SNAPIN_GETCOOKIE)
|
|
{
|
|
hr = pStream->Write(&pT, sizeof(T*), &uWritten);
|
|
return hr;
|
|
}
|
|
|
|
|
|
return hr;
|
|
}
|
|
OLECHAR* GetResultPaneInfo(int nCol)
|
|
{
|
|
if (nCol == 0 && m_pszDisplayName)
|
|
return m_pszDisplayName;
|
|
|
|
return L"Override GetResultPaneInfo in your derived class";
|
|
}
|
|
|
|
static CSnapInToolbarInfo* GetToolbarInfo()
|
|
{
|
|
return NULL;
|
|
}
|
|
|
|
static const UINT GetMenuID()
|
|
{
|
|
return 0;
|
|
}
|
|
};
|
|
|
|
|
|
_declspec( selectany ) CLIPFORMAT ISnapInDataInterface::m_CCF_NODETYPE = 0;
|
|
_declspec( selectany ) CLIPFORMAT ISnapInDataInterface::m_CCF_SZNODETYPE = 0;
|
|
_declspec( selectany ) CLIPFORMAT ISnapInDataInterface::m_CCF_DISPLAY_NAME = 0;
|
|
_declspec( selectany ) CLIPFORMAT ISnapInDataInterface::m_CCF_SNAPIN_CLASSID = 0;
|
|
_declspec( selectany ) CLIPFORMAT ISnapInDataInterface::m_CCF_SNAPIN_GETCOOKIE = 0;
|
|
|
|
#endif //__ATL_SNAPIN_H__
|