XComUtils.h

// XComUtils.h: interface for the file related class.
//
//////////////////////////////////////////////////////////////////////
 
#if !defined(CCOMUTILSINCLUDE_H)
#define CCOMUTILSINCLUDE_H
 
#ifdef _MSC_VER
#pragma once
#endif // _MSC_VER
 
#include <atlbase.h>
#include <shlwapi.h>
 
BEGIN_MOOTOOLS_NAMESPACE
 
template <class T>
class CXComPtr
{
protected:
    T* p;
 
public:
    CXComPtr() throw()
    {
        p = NULL;
    }
 
    CXComPtr(T* lp) throw()
    {
        p = lp;
        if (p != NULL)
            p->AddRef();
    }
 
    ~CXComPtr() throw()
    {
        if (p)
            p->Release();
    }
 
    T* operator->() const throw()
    {
        XASSERT(p!=NULL);
        return p;
    }
 
    operator T*() const throw()
    {
        return p;
    }
 
    T& operator*() const
    {
        XASSERT(p != NULL);
        return *p;
    }
 
    T** operator&() throw()
    {
        XASSERT(p==NULL);
        return &p;
    }
 
    T* operator=(const CXComPtr<T>& lp) throw()
    {
        if (*this != lp)
        {
            if (p != NULL)
            {
                p->Release();
                p = NULL;
            }
            
            if (lp.p != NULL)
            {
                p = lp.p;
                p->AddRef();
            }
        }
        return *this;
    }   
 
    bool operator==(T* pT) const throw()
    {
        return p == pT;
    }
 
    // Compare two objects for equivalence
    bool IsEqualObject(IUnknown* pOther) throw()
    {
        if (p == NULL && pOther == NULL)
            return true;    // They are both NULL objects
 
        if (p == NULL || pOther == NULL)
            return false;   // One is NULL the other is not
 
        CComPtr<IUnknown> punk1;
        CComPtr<IUnknown> punk2;
        p->QueryInterface(__uuidof(IUnknown), (void**)&punk1);
        pOther->QueryInterface(__uuidof(IUnknown), (void**)&punk2);
        return punk1 == punk2;
    }
 
};
 
END_MOOTOOLS_NAMESPACE
 
#endif // !defined(CCOMUTILSINCLUDE_H)