windows-nt/Source/XPSP1/NT/admin/activec/samples/sdksamples/complete/comp.cpp

422 lines
12 KiB
C++
Raw Normal View History

2020-09-26 03:20:57 -05:00
//==============================================================;
//
// 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 "Comp.h"
#include "DataObj.h"
#include <commctrl.h> // Needed for button styles...
#include <crtdbg.h>
#include "resource.h"
#include "DeleBase.h"
#include "CompData.h"
#include "globals.h"
#define ARRAYLEN(x) (sizeof(x) / sizeof((x)[0]))
CComponent::CComponent(CComponentData *parent)
: m_pComponentData(parent), m_cref(0), m_ipConsole(NULL)
{
OBJECT_CREATED
m_ipControlBar = NULL;
m_ipToolbar = NULL;
}
CComponent::~CComponent()
{
OBJECT_DESTROYED
}
STDMETHODIMP CComponent::QueryInterface(REFIID riid, LPVOID *ppv)
{
if (!ppv)
return E_FAIL;
*ppv = NULL;
if (IsEqualIID(riid, IID_IUnknown))
*ppv = static_cast<IComponent *>(this);
else if (IsEqualIID(riid, IID_IComponent))
*ppv = static_cast<IComponent *>(this);
// first things first, make sure that when MMC
// asks if we do property sheets, that we actually
// say "yes"
else if (IsEqualIID(riid, IID_IExtendPropertySheet))
*ppv = static_cast<IExtendPropertySheet2 *>(this);
else if (IsEqualIID(riid, IID_IExtendPropertySheet2))
*ppv = static_cast<IExtendPropertySheet2 *>(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) CComponent::AddRef()
{
return InterlockedIncrement((LONG *)&m_cref);
}
STDMETHODIMP_(ULONG) CComponent::Release()
{
if (InterlockedDecrement((LONG *)&m_cref) == 0)
{
delete this;
return 0;
}
return m_cref;
}
///////////////////////////////
// Interface IComponent
///////////////////////////////
STDMETHODIMP CComponent::Initialize(
/* [in] */ LPCONSOLE lpConsole)
{
HRESULT hr = S_OK;
// Save away all the interfaces we'll need.
// Fail if we can't QI the required interfaces.
m_ipConsole = lpConsole;
m_ipConsole->AddRef();
return hr;
}
STDMETHODIMP CComponent::Notify(
/* [in] */ LPDATAOBJECT lpDataObject,
/* [in] */ MMC_NOTIFY_TYPE event,
/* [in] */ LPARAM arg,
/* [in] */ LPARAM param)
{
MMCN_Crack(FALSE, lpDataObject, NULL, this, event, arg, param);
HRESULT hr = S_FALSE;
CDelegationBase *base = NULL;
// we need to watch for property change and delegate it
// a little differently, we're actually going to send
// the CDelegationBase object pointer in the property page
// PSN_APPLY handler via MMCPropPageNotify()
if (MMCN_PROPERTY_CHANGE != event && MMCN_VIEW_CHANGE != event) {
if (NULL == lpDataObject)
return S_FALSE;
base = GetOurDataObject(lpDataObject)->GetBaseNodeObject();
} else if (MMCN_PROPERTY_CHANGE == event) {
base = (CDelegationBase *)param;
}
// MMCN_VIEW_CHANGE
static CDelegationBase *pLastPasteQuery = NULL;
if (MMCN_VIEW_CHANGE == event) {
switch (param) {//arg holds the data. For a scope item, this is the
//item's myhscopeitem. For a result item, this is
//the item's nId value, but we don't use it
//param holds the hint passed to IConsole::UpdateAllViews.
//hint is a value of the UPDATE_VIEWS_HINT enumeration
case UPDATE_SCOPEITEM:
hr = m_ipConsole->SelectScopeItem( (HSCOPEITEM)arg );
_ASSERT( S_OK == hr);
break;
case UPDATE_RESULTITEM:
base = GetOurDataObject(lpDataObject)->GetBaseNodeObject();
hr = base->OnUpdateItem(m_ipConsole, (long)arg, RESULT);
break;
}
return S_OK;
}
//The remaining notifications
switch (event) {
case MMCN_SHOW:
hr = base->OnShow(m_ipConsole, (BOOL)arg, (HSCOPEITEM)param);
break;
case MMCN_ADD_IMAGES:
hr = base->OnAddImages((IImageList *)arg, (HSCOPEITEM)param);
break;
case MMCN_SELECT:
hr = base->OnSelect(this, m_ipConsole, (BOOL)LOWORD(arg), (BOOL)HIWORD(arg));
break;
case MMCN_RENAME:
hr = base->OnRename((LPOLESTR)param);
//Now call IConsole::UpdateAllViews to redraw the item in all views.
hr = m_pComponentData->m_ipConsole->UpdateAllViews(lpDataObject, 0, UPDATE_RESULTITEM);
_ASSERT( S_OK == hr);
break;
case MMCN_REFRESH:
//we pass CComponentData's stored IConsole pointer here,
//so that the IConsole::UpdateAllViews can be called in OnRefresh
hr = base->OnRefresh(m_pComponentData->m_ipConsole);
break;
case MMCN_DELETE: {
//first delete the selected result item
hr = base->OnDelete(m_ipConsole);
//Now call IConsole::UpdateAllViews to redraw all views
//owned by the parent scope item. OnRefresh already does
//this for us, so use it.
hr = base->OnRefresh(m_pComponentData->m_ipConsole);
break;
}
// handle the property change notification if we need to do anything
// special with it
case MMCN_PROPERTY_CHANGE:
//we pass CComponentData's stored IConsole pointer here,
//so that the IConsole::UpdateAllViews can be called in OnPropertyChange
hr = base->OnPropertyChange(m_pComponentData->m_ipConsole, this);
break;
}
return hr;
}
STDMETHODIMP CComponent::Destroy(
/* [in] */ MMC_COOKIE cookie)
{
if (m_ipConsole) {
m_ipConsole->Release();
m_ipConsole = NULL;
}
return S_OK;
}
STDMETHODIMP CComponent::QueryDataObject(
/* [in] */ MMC_COOKIE cookie,
/* [in] */ DATA_OBJECT_TYPES type,
/* [out] */ LPDATAOBJECT __RPC_FAR *ppDataObject)
{
CDataObject *pObj = NULL;
if (cookie == 0)
pObj = new CDataObject((MMC_COOKIE)m_pComponentData->m_pStaticNode, type);
else
pObj = new CDataObject(cookie, type);
if (!pObj)
return E_OUTOFMEMORY;
pObj->QueryInterface(IID_IDataObject, (void **)ppDataObject);
return S_OK;
}
STDMETHODIMP CComponent::GetResultViewType(
/* [in] */ MMC_COOKIE cookie,
/* [out] */ LPOLESTR __RPC_FAR *ppViewType,
/* [out] */ long __RPC_FAR *pViewOptions)
{
CDelegationBase *base = (CDelegationBase *)cookie;
//
// Ask for default listview.
//
if (base == NULL)
{
*pViewOptions = MMC_VIEW_OPTIONS_NONE;
*ppViewType = NULL;
}
else
return base->GetResultViewType(ppViewType, pViewOptions);
return S_OK;
}
STDMETHODIMP CComponent::GetDisplayInfo(
/* [out][in] */ RESULTDATAITEM __RPC_FAR *pResultDataItem)
{
HRESULT hr = S_OK;
CDelegationBase *base = NULL;
// if they are asking for the RDI_STR we have one of those to give
if (pResultDataItem->lParam) {
base = (CDelegationBase *)pResultDataItem->lParam;
if (pResultDataItem->mask & RDI_STR) {
LPCTSTR pszT = base->GetDisplayName(pResultDataItem->nCol);
MAKE_WIDEPTR_FROMTSTR_ALLOC(pszW, pszT);
pResultDataItem->str = pszW;
}
if (pResultDataItem->mask & RDI_IMAGE) {
pResultDataItem->nImage = base->GetBitmapIndex();
}
}
return hr;
}
STDMETHODIMP CComponent::CompareObjects(
/* [in] */ LPDATAOBJECT lpDataObjectA,
/* [in] */ LPDATAOBJECT lpDataObjectB)
{
CDelegationBase *baseA = GetOurDataObject(lpDataObjectA)->GetBaseNodeObject();
CDelegationBase *baseB = GetOurDataObject(lpDataObjectB)->GetBaseNodeObject();
// compare the object pointers
if (baseA->GetCookie() == baseB->GetCookie())
return S_OK;
return S_FALSE;
}
///////////////////////////////////
// Interface IExtendPropertySheet2
///////////////////////////////////
HRESULT CComponent::CreatePropertyPages(
/* [in] */ LPPROPERTYSHEETCALLBACK lpProvider,
/* [in] */ LONG_PTR handle,
/* [in] */ LPDATAOBJECT lpIDataObject)
{
CDelegationBase *base = GetOurDataObject(lpIDataObject)->GetBaseNodeObject();
return base->CreatePropertyPages(lpProvider, handle);
}
HRESULT CComponent::QueryPagesFor(
/* [in] */ LPDATAOBJECT lpDataObject)
{
CDelegationBase *base = GetOurDataObject(lpDataObject)->GetBaseNodeObject();
return base->HasPropertySheets();
}
HRESULT CComponent::GetWatermarks(
/* [in] */ LPDATAOBJECT lpIDataObject,
/* [out] */ HBITMAP __RPC_FAR *lphWatermark,
/* [out] */ HBITMAP __RPC_FAR *lphHeader,
/* [out] */ HPALETTE __RPC_FAR *lphPalette,
/* [out] */ BOOL __RPC_FAR *bStretch)
{
CDelegationBase *base = GetOurDataObject(lpIDataObject)->GetBaseNodeObject();
return base->GetWatermarks(lphWatermark, lphHeader, lphPalette, bStretch);
}
///////////////////////////////
// Interface IExtendControlBar
///////////////////////////////
static MMCBUTTON SnapinButtons1[] =
{
{ 0, ID_BUTTONSTART, TBSTATE_ENABLED, TBSTYLE_GROUP, L"Start Vehicle", L"Start Vehicle" },
{ 1, ID_BUTTONPAUSE, TBSTATE_ENABLED, TBSTYLE_GROUP, L"Pause Vehicle", L"Pause Vehicle"},
{ 2, ID_BUTTONSTOP, TBSTATE_ENABLED, TBSTYLE_GROUP, L"Stop Vehicle", L"Stop Vehicle" },
};
HRESULT CComponent::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));
// The IControlbar::Create AddRefs the toolbar object it created
// so no need to do any addref on the interface.
// 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 CComponent::ControlbarNotify(
/* [in] */ MMC_NOTIFY_TYPE event,
/* [in] */ LPARAM arg,
/* [in] */ LPARAM param)
{
HRESULT hr = S_OK;
if (event == MMCN_SELECT) {
BOOL bScope = (BOOL) LOWORD(arg);
BOOL bSelect = (BOOL) HIWORD(arg);
CDelegationBase *base = GetOurDataObject(reinterpret_cast<IDataObject *>(param))->GetBaseNodeObject();
hr = base->OnSetToolbar(m_ipControlBar, m_ipToolbar, bScope, bSelect);
}
else if (event == MMCN_BTN_CLICK) {
CDelegationBase *base = GetOurDataObject(reinterpret_cast<IDataObject *>(arg))->GetBaseNodeObject();
hr = base->OnToolbarCommand(m_pComponentData->m_ipConsole, (MMC_CONSOLE_VERB)param, reinterpret_cast<IDataObject *>(arg));
}
return hr;
}