326 lines
9.3 KiB
C++
326 lines
9.3 KiB
C++
|
|
||
|
|
//==============================================================;
|
||
|
|
//
|
||
|
|
// This source code is only intended as a supplement to existing Microsoft documentation.
|
||
|
|
//
|
||
|
|
//
|
||
|
|
//
|
||
|
|
//
|
||
|
|
// THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF ANY
|
||
|
|
// KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
|
||
|
|
// IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A PARTICULAR
|
||
|
|
// PURPOSE.
|
||
|
|
//
|
||
|
|
// Copyright (C) 1999 Microsoft Corporation. All Rights Reserved.
|
||
|
|
//
|
||
|
|
//
|
||
|
|
//
|
||
|
|
//==============================================================;
|
||
|
|
|
||
|
|
#include "tbarExt.h"
|
||
|
|
#include "globals.h"
|
||
|
|
#include "resource.h"
|
||
|
|
#include <commctrl.h> // Needed for button styles...
|
||
|
|
#include <crtdbg.h>
|
||
|
|
#include "resource.h"
|
||
|
|
#include <stdio.h>
|
||
|
|
|
||
|
|
#define ARRAYLEN(x) (sizeof(x) / sizeof((x)[0]))
|
||
|
|
|
||
|
|
|
||
|
|
// we need to do this to get around MMC.IDL - it explicitly defines
|
||
|
|
// the clipboard formats as WCHAR types...
|
||
|
|
#define _T_CCF_DISPLAY_NAME _T("CCF_DISPLAY_NAME")
|
||
|
|
#define _T_CCF_NODETYPE _T("CCF_NODETYPE")
|
||
|
|
#define _T_CCF_SNAPIN_CLASSID _T("CCF_SNAPIN_CLASSID")
|
||
|
|
|
||
|
|
// These are the clipboard formats that we must supply at a minimum.
|
||
|
|
// mmc.h actually defined these. We can make up our own to use for
|
||
|
|
// other reasons. We don't need any others at this time.
|
||
|
|
UINT CToolBarExtension::s_cfDisplayName = RegisterClipboardFormat(_T_CCF_DISPLAY_NAME);
|
||
|
|
UINT CToolBarExtension::s_cfNodeType = RegisterClipboardFormat(_T_CCF_NODETYPE);
|
||
|
|
UINT CToolBarExtension::s_cfSnapInCLSID = RegisterClipboardFormat(_T_CCF_SNAPIN_CLASSID);
|
||
|
|
|
||
|
|
|
||
|
|
CToolBarExtension::CToolBarExtension() : m_cref(0), m_ipControlBar(NULL),
|
||
|
|
m_ipToolbar(NULL)
|
||
|
|
{
|
||
|
|
OBJECT_CREATED
|
||
|
|
}
|
||
|
|
|
||
|
|
CToolBarExtension::~CToolBarExtension()
|
||
|
|
{
|
||
|
|
OBJECT_DESTROYED
|
||
|
|
}
|
||
|
|
|
||
|
|
///////////////////////
|
||
|
|
// IUnknown implementation
|
||
|
|
///////////////////////
|
||
|
|
|
||
|
|
STDMETHODIMP CToolBarExtension::QueryInterface(REFIID riid, LPVOID *ppv)
|
||
|
|
{
|
||
|
|
if (!ppv)
|
||
|
|
return E_FAIL;
|
||
|
|
|
||
|
|
*ppv = NULL;
|
||
|
|
|
||
|
|
if (IsEqualIID(riid, IID_IUnknown))
|
||
|
|
*ppv = static_cast<IExtendControlbar *>(this);
|
||
|
|
else if (IsEqualIID(riid, IID_IExtendControlbar))
|
||
|
|
*ppv = static_cast<IExtendControlbar *>(this);
|
||
|
|
|
||
|
|
if (*ppv)
|
||
|
|
{
|
||
|
|
reinterpret_cast<IUnknown *>(*ppv)->AddRef();
|
||
|
|
return S_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
return E_NOINTERFACE;
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP_(ULONG) CToolBarExtension::AddRef()
|
||
|
|
{
|
||
|
|
return InterlockedIncrement((LONG *)&m_cref);
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP_(ULONG) CToolBarExtension::Release()
|
||
|
|
{
|
||
|
|
if (InterlockedDecrement((LONG *)&m_cref) == 0)
|
||
|
|
{
|
||
|
|
// we need to decrement our object count in the DLL
|
||
|
|
delete this;
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
return m_cref;
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT CToolBarExtension::ExtractData( IDataObject* piDataObject,
|
||
|
|
CLIPFORMAT cfClipFormat,
|
||
|
|
BYTE* pbData,
|
||
|
|
DWORD cbData )
|
||
|
|
{
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
|
||
|
|
FORMATETC formatetc = {cfClipFormat, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL};
|
||
|
|
STGMEDIUM stgmedium = {TYMED_HGLOBAL, NULL};
|
||
|
|
|
||
|
|
stgmedium.hGlobal = ::GlobalAlloc(GPTR, cbData);
|
||
|
|
do // false loop
|
||
|
|
{
|
||
|
|
if (NULL == stgmedium.hGlobal)
|
||
|
|
{
|
||
|
|
hr = E_OUTOFMEMORY;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
hr = piDataObject->GetDataHere( &formatetc, &stgmedium );
|
||
|
|
if ( FAILED(hr) )
|
||
|
|
{
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
BYTE* pbNewData = reinterpret_cast<BYTE*>(stgmedium.hGlobal);
|
||
|
|
if (NULL == pbNewData)
|
||
|
|
{
|
||
|
|
hr = E_UNEXPECTED;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
::memcpy( pbData, pbNewData, cbData );
|
||
|
|
} while (FALSE); // false loop
|
||
|
|
|
||
|
|
if (NULL != stgmedium.hGlobal)
|
||
|
|
{
|
||
|
|
::GlobalFree(stgmedium.hGlobal);
|
||
|
|
}
|
||
|
|
return hr;
|
||
|
|
} // ExtractData()
|
||
|
|
|
||
|
|
HRESULT CToolBarExtension::ExtractString( IDataObject *piDataObject,
|
||
|
|
CLIPFORMAT cfClipFormat,
|
||
|
|
WCHAR *pstr,
|
||
|
|
DWORD cchMaxLength)
|
||
|
|
{
|
||
|
|
return ExtractData( piDataObject, cfClipFormat, (PBYTE)pstr, cchMaxLength );
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT CToolBarExtension::ExtractSnapInCLSID( IDataObject* piDataObject, CLSID* pclsidSnapin )
|
||
|
|
{
|
||
|
|
return ExtractData( piDataObject, s_cfSnapInCLSID, (PBYTE)pclsidSnapin, sizeof(CLSID) );
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT CToolBarExtension::ExtractObjectTypeGUID( IDataObject* piDataObject, GUID* pguidObjectType )
|
||
|
|
{
|
||
|
|
return ExtractData( piDataObject, s_cfNodeType, (PBYTE)pguidObjectType, sizeof(GUID) );
|
||
|
|
}
|
||
|
|
|
||
|
|
///////////////////////////////
|
||
|
|
// Interface IExtendControlBar
|
||
|
|
///////////////////////////////
|
||
|
|
static MMCBUTTON SnapinButtons1[] =
|
||
|
|
{
|
||
|
|
{ 0, ID_BUTTONSTART, TBSTATE_ENABLED, TBSTYLE_GROUP, L"Extension - Start Vehicle", L"Extension - Start Vehicle" },
|
||
|
|
{ 1, ID_BUTTONPAUSE, TBSTATE_ENABLED, TBSTYLE_GROUP, L"Extension - Pause Vehicle", L"Extension - Pause Vehicle"},
|
||
|
|
{ 2, ID_BUTTONSTOP, TBSTATE_ENABLED, TBSTYLE_GROUP, L"Extension - Stop Vehicle", L"Extension - Stop Vehicle" },
|
||
|
|
};
|
||
|
|
|
||
|
|
HRESULT CToolBarExtension::SetControlbar(
|
||
|
|
/* [in] */ LPCONTROLBAR pControlbar)
|
||
|
|
{
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
|
||
|
|
//
|
||
|
|
// Clean up
|
||
|
|
//
|
||
|
|
|
||
|
|
// if we've got a cached toolbar, release it
|
||
|
|
if (m_ipToolbar) {
|
||
|
|
m_ipToolbar->Release();
|
||
|
|
m_ipToolbar = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
// if we've got a cached control bar, release it
|
||
|
|
if (m_ipControlBar) {
|
||
|
|
m_ipControlBar->Release();
|
||
|
|
m_ipControlBar = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Install new pieces if necessary
|
||
|
|
//
|
||
|
|
|
||
|
|
// if a new one came in, cache and AddRef
|
||
|
|
if (pControlbar) {
|
||
|
|
m_ipControlBar = pControlbar;
|
||
|
|
m_ipControlBar->AddRef();
|
||
|
|
|
||
|
|
hr = m_ipControlBar->Create(TOOLBAR, // type of control to be created
|
||
|
|
dynamic_cast<IExtendControlbar *>(this),
|
||
|
|
reinterpret_cast<IUnknown **>(&m_ipToolbar));
|
||
|
|
_ASSERT(SUCCEEDED(hr));
|
||
|
|
|
||
|
|
// add the bitmap to the toolbar
|
||
|
|
HBITMAP hbmp = LoadBitmap(g_hinst, MAKEINTRESOURCE(IDR_TOOLBAR1));
|
||
|
|
hr = m_ipToolbar->AddBitmap(3, hbmp, 16, 16, RGB(0, 128, 128)); // NOTE, hardcoded value 3
|
||
|
|
_ASSERT(SUCCEEDED(hr));
|
||
|
|
|
||
|
|
// Add the buttons to the toolbar
|
||
|
|
hr = m_ipToolbar->AddButtons(ARRAYLEN(SnapinButtons1), SnapinButtons1);
|
||
|
|
_ASSERT(SUCCEEDED(hr));
|
||
|
|
}
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT CToolBarExtension::ControlbarNotify(
|
||
|
|
/* [in] */ MMC_NOTIFY_TYPE event,
|
||
|
|
/* [in] */ LPARAM arg,
|
||
|
|
/* [in] */ LPARAM param)
|
||
|
|
{
|
||
|
|
|
||
|
|
_TCHAR pszMsg[255];
|
||
|
|
|
||
|
|
BYTE *pbVehicleStatus = NULL;
|
||
|
|
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
|
||
|
|
if (event == MMCN_SELECT) {
|
||
|
|
|
||
|
|
BOOL bScope = (BOOL) LOWORD(arg);
|
||
|
|
BOOL bSelect = (BOOL) HIWORD(arg);
|
||
|
|
|
||
|
|
if (bSelect) {
|
||
|
|
|
||
|
|
// Always make sure the toolbar is attached
|
||
|
|
hr = m_ipControlBar->Attach(TOOLBAR, m_ipToolbar);
|
||
|
|
|
||
|
|
// Set button states
|
||
|
|
|
||
|
|
//fake value to set toolbar button states
|
||
|
|
iStatus = RUNNING;
|
||
|
|
SetToolbarButtons(iStatus);
|
||
|
|
|
||
|
|
} else {
|
||
|
|
// Always make sure the toolbar is detached
|
||
|
|
hr = m_ipControlBar->Detach(m_ipToolbar);
|
||
|
|
}
|
||
|
|
|
||
|
|
} else if (event == MMCN_BTN_CLICK) {
|
||
|
|
//the arg parameter contains the data object from the primary
|
||
|
|
//snap-in. Use it to get the display name of the currently selected item
|
||
|
|
WCHAR pszName[255];
|
||
|
|
HRESULT hr = ExtractString(reinterpret_cast<IDataObject *>(arg), s_cfDisplayName, pszName, sizeof(pszName));
|
||
|
|
MAKE_TSTRPTR_FROMWIDE(ptrname, pszName);
|
||
|
|
|
||
|
|
switch ((int)param)
|
||
|
|
{
|
||
|
|
case ID_BUTTONSTART:
|
||
|
|
iStatus = RUNNING;
|
||
|
|
break;
|
||
|
|
|
||
|
|
case ID_BUTTONPAUSE:
|
||
|
|
iStatus = PAUSED;
|
||
|
|
break;
|
||
|
|
|
||
|
|
case ID_BUTTONSTOP:
|
||
|
|
iStatus = STOPPED;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
_stprintf(pszMsg, _T("%s selected and extension button %s pressed"), ptrname,
|
||
|
|
(long)param == ID_BUTTONSTART ? _T("1") :
|
||
|
|
(long)param == ID_BUTTONPAUSE ? _T("2") :
|
||
|
|
(long)param == ID_BUTTONSTOP ? _T("3") : _T("!!!unknown command!!!"));
|
||
|
|
|
||
|
|
|
||
|
|
::MessageBox(NULL, pszMsg, _T("Messagebox from Toolbar Extension"), MB_OK|MB_ICONEXCLAMATION);
|
||
|
|
|
||
|
|
//Reset toolbar button states
|
||
|
|
SetToolbarButtons(iStatus);
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT CToolBarExtension::SetToolbarButtons(STATUS iVehicleStatus)
|
||
|
|
{
|
||
|
|
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
|
||
|
|
switch (iVehicleStatus)
|
||
|
|
{
|
||
|
|
case RUNNING:
|
||
|
|
m_ipToolbar->SetButtonState(ID_BUTTONSTART, BUTTONPRESSED, TRUE);
|
||
|
|
m_ipToolbar->SetButtonState(ID_BUTTONSTART, ENABLED, FALSE);
|
||
|
|
m_ipToolbar->SetButtonState(ID_BUTTONPAUSE, BUTTONPRESSED, FALSE);
|
||
|
|
m_ipToolbar->SetButtonState(ID_BUTTONPAUSE, ENABLED, TRUE);
|
||
|
|
m_ipToolbar->SetButtonState(ID_BUTTONSTOP, BUTTONPRESSED, FALSE);
|
||
|
|
m_ipToolbar->SetButtonState(ID_BUTTONSTOP, ENABLED, TRUE);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case PAUSED:
|
||
|
|
m_ipToolbar->SetButtonState(ID_BUTTONSTART, BUTTONPRESSED, FALSE);
|
||
|
|
m_ipToolbar->SetButtonState(ID_BUTTONSTART, ENABLED, TRUE);
|
||
|
|
m_ipToolbar->SetButtonState(ID_BUTTONPAUSE, BUTTONPRESSED, TRUE);
|
||
|
|
m_ipToolbar->SetButtonState(ID_BUTTONPAUSE, ENABLED, FALSE);
|
||
|
|
m_ipToolbar->SetButtonState(ID_BUTTONSTOP, BUTTONPRESSED, FALSE);
|
||
|
|
m_ipToolbar->SetButtonState(ID_BUTTONSTOP, ENABLED, TRUE);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case STOPPED:
|
||
|
|
m_ipToolbar->SetButtonState(ID_BUTTONSTART, BUTTONPRESSED, FALSE);
|
||
|
|
m_ipToolbar->SetButtonState(ID_BUTTONSTART, ENABLED, TRUE);
|
||
|
|
m_ipToolbar->SetButtonState(ID_BUTTONPAUSE, BUTTONPRESSED, FALSE);
|
||
|
|
m_ipToolbar->SetButtonState(ID_BUTTONPAUSE, ENABLED, TRUE);
|
||
|
|
m_ipToolbar->SetButtonState(ID_BUTTONSTOP, BUTTONPRESSED, TRUE);
|
||
|
|
m_ipToolbar->SetButtonState(ID_BUTTONSTOP, ENABLED, FALSE);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|