diff options
Diffstat (limited to 'WebKit/win/COMEnumVariant.h')
| -rw-r--r-- | WebKit/win/COMEnumVariant.h | 221 |
1 files changed, 221 insertions, 0 deletions
diff --git a/WebKit/win/COMEnumVariant.h b/WebKit/win/COMEnumVariant.h new file mode 100644 index 0000000..be0ac8c --- /dev/null +++ b/WebKit/win/COMEnumVariant.h @@ -0,0 +1,221 @@ +/* + * Copyright (C) 2007 Apple Inc. All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * 3. Neither the name of Apple Computer, Inc. ("Apple") nor the names of + * its contributors may be used to endorse or promote products derived + * from this software without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY + * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED + * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE + * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY + * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES + * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; + * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND + * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF + * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ + +#ifndef COMEnumVariant_h +#define COMEnumVariant_h + +#define NOMINMAX +#include <unknwn.h> + +#include <WebCore/BString.h> +#include <wtf/Assertions.h> +#include <wtf/Noncopyable.h> + +namespace WebCore { + class String; +} + +template<typename T> struct COMVariantSetter {}; + +template<> struct COMVariantSetter<WebCore::String> +{ + static void setVariant(VARIANT* variant, const WebCore::String& value) + { + ASSERT(V_VT(variant) == VT_EMPTY); + + V_VT(variant) = VT_BSTR; + V_BSTR(variant) = WebCore::BString(value).release(); + } +}; + +template<> struct COMVariantSetter<unsigned long long> +{ + static void setVariant(VARIANT* variant, unsigned long long value) + { + ASSERT(V_VT(variant) == VT_EMPTY); + + V_VT(variant) = VT_UI8; + V_UI8(variant) = value; + } +}; + +template<typename COMType, typename UnderlyingType> +struct COMIUnknownVariantSetter +{ + static void setVariant(VARIANT* variant, const UnderlyingType& value) + { + ASSERT(V_VT(variant) == VT_EMPTY); + + V_VT(variant) = VT_UNKNOWN; + V_UNKNOWN(variant) = COMType::createInstance(value); + } +}; + +template<typename ContainerType> +class COMEnumVariant : public IEnumVARIANT, Noncopyable { +public: + static COMEnumVariant* adopt(ContainerType&); + static COMEnumVariant* createInstance(const ContainerType&); + + // IUnknown + virtual HRESULT STDMETHODCALLTYPE QueryInterface(REFIID riid, void** ppvObject); + virtual ULONG STDMETHODCALLTYPE AddRef(); + virtual ULONG STDMETHODCALLTYPE Release(); + + // IEnumVARIANT + virtual HRESULT STDMETHODCALLTYPE Next(ULONG celt, VARIANT* rgVar, ULONG* pCeltFetched); + virtual HRESULT STDMETHODCALLTYPE Skip(ULONG celt); + virtual HRESULT STDMETHODCALLTYPE Reset(); + virtual HRESULT STDMETHODCALLTYPE Clone(IEnumVARIANT** ppEnum); + +private: + COMEnumVariant() + : m_refCount(0) + { + } + + COMEnumVariant(const ContainerType& container) + : m_refCount(0) + , m_container(container) + , m_currentPos(m_container.begin()) + { + } + + ~COMEnumVariant() {} + + ULONG m_refCount; + + ContainerType m_container; + typename ContainerType::const_iterator m_currentPos; +}; + +// COMEnumVariant ------------------------------------------------------------------ +template<typename ContainerType> +COMEnumVariant<typename ContainerType>* COMEnumVariant<ContainerType>::adopt(ContainerType& container) +{ + COMEnumVariant* instance = new COMEnumVariant; + instance->m_container.swap(container); + instance->m_currentPos = instance->m_container.begin(); + instance->AddRef(); + return instance; +} + +template<typename ContainerType> +COMEnumVariant<typename ContainerType>* COMEnumVariant<ContainerType>::createInstance(const ContainerType& container) +{ + COMEnumVariant* instance = new COMEnumVariant(container); + instance->AddRef(); + return instance; +} + +// IUnknown ------------------------------------------------------------------------ +template<typename ContainerType> +HRESULT STDMETHODCALLTYPE COMEnumVariant<ContainerType>::QueryInterface(REFIID riid, void** ppvObject) +{ + *ppvObject = 0; + if (IsEqualGUID(riid, IID_IUnknown)) + *ppvObject = static_cast<COMEnumVariant*>(this); + else if (IsEqualGUID(riid, IID_IEnumVARIANT)) + *ppvObject = static_cast<COMEnumVariant*>(this); + else + return E_NOINTERFACE; + + AddRef(); + return S_OK; +} + +template<typename ContainerType> +ULONG STDMETHODCALLTYPE COMEnumVariant<ContainerType>::AddRef() +{ + return ++m_refCount; +} + +template<typename ContainerType> +ULONG STDMETHODCALLTYPE COMEnumVariant<ContainerType>::Release() +{ + ULONG newRef = --m_refCount; + if (!newRef) + delete this; + + return newRef; +} + +// IEnumVARIANT -------------------------------------------------------------------- +template<typename ContainerType> +HRESULT STDMETHODCALLTYPE COMEnumVariant<ContainerType>::Next(ULONG celt, VARIANT* rgVar, ULONG* pCeltFetched) +{ + if (pCeltFetched) + *pCeltFetched = 0; + if (!rgVar) + return E_POINTER; + for (unsigned i = 0 ; i < celt; i++) + VariantInit(&rgVar[i]); + + for (unsigned i = 0; i < celt; i++) { + if (m_currentPos == m_container.end()) + return S_FALSE; + + COMVariantSetter<ContainerType::ValueType>::setVariant(&rgVar[i], *m_currentPos); + ++m_currentPos; + if (pCeltFetched) + (*pCeltFetched)++; + } + + return S_OK; +} + +template<typename ContainerType> +HRESULT STDMETHODCALLTYPE COMEnumVariant<ContainerType>::Skip(ULONG celt) +{ + for (unsigned i = 0; i < celt; i++) { + if (m_currentPos == m_container.end()) + return S_FALSE; + + ++m_currentPos; + } + return S_OK; +} + +template<typename ContainerType> +HRESULT STDMETHODCALLTYPE COMEnumVariant<ContainerType>::Reset() +{ + m_currentPos = m_container.begin(); + return S_OK; +} + +template<typename ContainerType> +HRESULT STDMETHODCALLTYPE COMEnumVariant<ContainerType>::Clone(IEnumVARIANT** ppEnum) +{ + if (!ppEnum) + return E_POINTER; + + *ppEnum = 0; + return E_NOTIMPL; +} + +#endif |