cIUnkAgg.h

Go to the documentation of this file.

//
//
 
#ifndef _INC_cIUnkAgg_H
#define _INC_cIUnkAgg_H
#ifndef NO_PRAGMA_ONCE
#pragma once
#endif
 
#include "cRefPtr.h"
 
namespace Gray
{
        template<class TYPE = IUnknown>
        class DECLSPEC_NOVTABLE cIUnkAggBase
        {
 
        protected:
                TYPE* m_pIAggOuter;     
 
        public:
                cIUnkAggBase(TYPE* pIAggOuter)
                        : m_pIAggOuter(pIAggOuter)
                {
                        ASSERT(pIAggOuter != nullptr);
                }
                virtual ~cIUnkAggBase()
                {
                }
                // Support IUnknown functions.
                HRESULT QueryInterface(const IID& riid, void **ppv)
                {
                        if (ppv == nullptr || m_pIAggOuter == nullptr)
                        {
                                *ppv = nullptr;
                                return E_POINTER;
                        }
                        return m_pIAggOuter->QueryInterface(riid, ppv);
                }
                ULONG AddRef(void)
                {
                        return m_pIAggOuter->AddRef();
                }
                ULONG Release(void)
                {
                        return m_pIAggOuter->Release();
                }
        };
 
        class DECLSPEC_NOVTABLE cIUnkAgg : public cRefBase, public cIUnkAggBase < IUnknown >
        {
        public:
                cIUnkAgg(IUnknown* pIAggOuter)
                        : cIUnkAggBase((pIAggOuter == nullptr) ? this : pIAggOuter)
                {
                }
                // Support IUknown functions.
                STDMETHODIMP_(ULONG) AddRef(void) override
                {
                        if (m_pIAggOuter == this)
                        {
                                IncRefCount();
                                return (ULONG) get_RefCount();
                        }
                        return m_pIAggOuter->AddRef();
                }
                STDMETHODIMP_(ULONG) Release(void) override
                {
                        if (m_pIAggOuter == this)
                        {
                                DecRefCount();
                                return (ULONG) get_RefCount();
                        }
                        return m_pIAggOuter->Release();
                }
                STDMETHODIMP QueryInterface(const IID& riid, void **ppv) override
                {
                        if (ppv == nullptr)
                        {
                                return E_POINTER;
                        }
                        if (m_pIAggOuter == nullptr)
                        {
                                *ppv = nullptr;
                                return E_POINTER;
                        }
                        if (m_pIAggOuter == this)
                        {
                                if (riid == __uuidof(IUnknown))
                                {
                                        *ppv = this;
                                        IncRefCount();
                                        return S_OK;
                                }
                                return E_NOINTERFACE;
                        }
                        return m_pIAggOuter->QueryInterface(riid, ppv);
                }
        };
}
#endif // _INC_cIUnkAgg_H