f53df7da64
Code submissions have continually suffered from formatting inconsistencies that constantly have to be addressed. Using clang-format simplifies this by making code formatting more consistent, and allows automation of the code formatting so that maintainers can focus more on the code itself instead of code formatting.
161 lines
2.8 KiB
C++
161 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
|