160 lines
2.8 KiB
C++
160 lines
2.8 KiB
C++
/*
|
|
* Copyright (c) 2013 Hugh Bailey <obs.jim@gmail.com>
|
|
*
|
|
* Permission to use, copy, modify, and distribute this software for any
|
|
* purpose with or without fee is hereby granted, provided that the above
|
|
* copyright notice and this permission notice appear in all copies.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
|
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
|
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
/* Oh no I have my own com pointer class, the world is ending, how dare you
|
|
* write your own! */
|
|
|
|
template<class T> class ComPtr {
|
|
|
|
protected:
|
|
T *ptr;
|
|
|
|
inline void Kill()
|
|
{
|
|
if (ptr)
|
|
ptr->Release();
|
|
}
|
|
|
|
inline void Replace(T *p)
|
|
{
|
|
if (ptr != p) {
|
|
if (p)
|
|
p->AddRef();
|
|
if (ptr)
|
|
ptr->Release();
|
|
ptr = p;
|
|
}
|
|
}
|
|
|
|
public:
|
|
inline ComPtr() : ptr(nullptr) {}
|
|
inline ComPtr(T *p) : ptr(p)
|
|
{
|
|
if (ptr)
|
|
ptr->AddRef();
|
|
}
|
|
inline ComPtr(const ComPtr<T> &c) : ptr(c.ptr)
|
|
{
|
|
if (ptr)
|
|
ptr->AddRef();
|
|
}
|
|
inline ComPtr(ComPtr<T> &&c) : ptr(c.ptr) { c.ptr = nullptr; }
|
|
inline ~ComPtr() { Kill(); }
|
|
|
|
inline void Clear()
|
|
{
|
|
if (ptr) {
|
|
ptr->Release();
|
|
ptr = nullptr;
|
|
}
|
|
}
|
|
|
|
inline ComPtr<T> &operator=(T *p)
|
|
{
|
|
Replace(p);
|
|
return *this;
|
|
}
|
|
|
|
inline ComPtr<T> &operator=(const ComPtr<T> &c)
|
|
{
|
|
Replace(c.ptr);
|
|
return *this;
|
|
}
|
|
|
|
inline ComPtr<T> &operator=(ComPtr<T> &&c)
|
|
{
|
|
if (&ptr != &c.ptr) {
|
|
Kill();
|
|
ptr = c.ptr;
|
|
c.ptr = nullptr;
|
|
}
|
|
|
|
return *this;
|
|
}
|
|
|
|
inline T *Detach()
|
|
{
|
|
T *out = ptr;
|
|
ptr = nullptr;
|
|
return out;
|
|
}
|
|
|
|
inline void CopyTo(T **out)
|
|
{
|
|
if (out) {
|
|
if (ptr)
|
|
ptr->AddRef();
|
|
*out = ptr;
|
|
}
|
|
}
|
|
|
|
inline ULONG Release()
|
|
{
|
|
ULONG ref;
|
|
|
|
if (!ptr)
|
|
return 0;
|
|
ref = ptr->Release();
|
|
ptr = nullptr;
|
|
return ref;
|
|
}
|
|
|
|
inline T **Assign()
|
|
{
|
|
Clear();
|
|
return &ptr;
|
|
}
|
|
inline void Set(T *p)
|
|
{
|
|
Kill();
|
|
ptr = p;
|
|
}
|
|
|
|
inline T *Get() const { return ptr; }
|
|
|
|
inline T **operator&() { return Assign(); }
|
|
|
|
inline operator T *() const { return ptr; }
|
|
inline T *operator->() const { return ptr; }
|
|
|
|
inline bool operator==(T *p) const { return ptr == p; }
|
|
inline bool operator!=(T *p) const { return ptr != p; }
|
|
|
|
inline bool operator!() const { return !ptr; }
|
|
};
|
|
|
|
#ifdef _WIN32
|
|
|
|
template<class T> class ComQIPtr : public ComPtr<T> {
|
|
|
|
public:
|
|
inline ComQIPtr(IUnknown *unk)
|
|
{
|
|
this->ptr = nullptr;
|
|
unk->QueryInterface(__uuidof(T), (void **)&this->ptr);
|
|
}
|
|
|
|
inline ComPtr<T> &operator=(IUnknown *unk)
|
|
{
|
|
ComPtr<T>::Clear();
|
|
unk->QueryInterface(__uuidof(T), (void **)&this->ptr);
|
|
return *this;
|
|
}
|
|
};
|
|
|
|
#endif
|