mirror of
https://github.com/AquariaOSE/Aquaria.git
synced 2024-12-25 14:15:46 +00:00
144 lines
3.5 KiB
C++
144 lines
3.5 KiB
C++
#ifndef VFS_REFCOUNTED_H
|
|
#define VFS_REFCOUNTED_H
|
|
|
|
#include <algorithm>
|
|
#include <cassert>
|
|
|
|
VFS_NAMESPACE_START
|
|
|
|
|
|
template <typename T> class RefcountedT
|
|
{
|
|
public:
|
|
// Static methods for refcounting, could overload with atomics if there is the need
|
|
inline static void s_incRef(T& ref)
|
|
{
|
|
++ref;
|
|
}
|
|
inline static bool s_decRef(T& ref) // returns true if refcount is zero afterwards
|
|
{
|
|
return (--ref) == 0;
|
|
}
|
|
inline static void s_setRef(T& ref, int val)
|
|
{
|
|
ref = val;
|
|
}
|
|
inline static int s_getRef(const T& ref)
|
|
{
|
|
return ref;
|
|
}
|
|
|
|
RefcountedT()
|
|
{
|
|
s_setRef(_refcount, 0);
|
|
}
|
|
|
|
virtual ~RefcountedT()
|
|
{
|
|
int val = s_getRef(_refcount);
|
|
assert(val == 0 && "Object was deleted with refcount != 0");
|
|
}
|
|
|
|
inline void incref()
|
|
{
|
|
s_incRef(_refcount);
|
|
}
|
|
inline void decref()
|
|
{
|
|
if (s_decRef(_refcount))
|
|
{
|
|
// if the refcount is now zero, it will stay zero forever as nobody has a reference anymore
|
|
delete this;
|
|
}
|
|
}
|
|
inline int getRefCount() const
|
|
{
|
|
return s_getRef(_refcount);
|
|
}
|
|
|
|
private:
|
|
T _refcount;
|
|
};
|
|
|
|
// This is the typedef used for VFSBase
|
|
typedef RefcountedT<int> Refcounted;
|
|
|
|
|
|
template<typename T> class CountedPtr
|
|
{
|
|
public:
|
|
inline ~CountedPtr()
|
|
{
|
|
if(_p)
|
|
_p->decref();
|
|
}
|
|
inline CountedPtr() : _p(NULL)
|
|
{}
|
|
inline CountedPtr(T* p) : _p(p)
|
|
{
|
|
if(p)
|
|
p->incref();
|
|
}
|
|
inline CountedPtr(const CountedPtr& ref) : _p(ref._p)
|
|
{
|
|
if (_p)
|
|
_p->incref();
|
|
}
|
|
|
|
#ifdef HAVE_CXX_11
|
|
// C++11 move constructor
|
|
CountedPtr(CountedPtr&& ref) : CountedPtr() // initialize via default constructor
|
|
{
|
|
CountedPtr::swap(*this, ref);
|
|
}
|
|
#endif
|
|
|
|
// intentionally not a reference -- see http://stackoverflow.com/questions/3279543/what-is-the-copy-and-swap-idiom
|
|
CountedPtr& operator=(CountedPtr ref)
|
|
{
|
|
CountedPtr::swap(*this, ref);
|
|
return *this;
|
|
}
|
|
|
|
const T* operator->() const { return _p; }
|
|
T* operator->() { return _p; }
|
|
|
|
const T* operator*() const { return *_p; }
|
|
T* operator*() { return *_p; }
|
|
|
|
bool operator!() const { return !_p; }
|
|
|
|
// if you use these, make sure you also keep a counted reference to the object!
|
|
inline operator T* () { return _p; }
|
|
inline operator const T* () const { return _p; }
|
|
|
|
inline T* content() { return _p; }
|
|
inline const T* content() const { return _p; }
|
|
|
|
bool operator<(const CountedPtr& ref) const { return _p < ref._p; }
|
|
bool operator<=(const CountedPtr& ref) const { return _p <= ref._p; }
|
|
bool operator==(const CountedPtr& ref) const { return _p == ref._p; }
|
|
bool operator!=(const CountedPtr& ref) const { return _p != ref._p; }
|
|
bool operator>=(const CountedPtr& ref) const { return _p >= ref._p; }
|
|
bool operator>(const CountedPtr& ref) const { return _p > ref._p; }
|
|
|
|
bool operator<(const T* ptr) const { return _p < ptr; }
|
|
bool operator<=(const T* ptr) const { return _p <= ptr; }
|
|
bool operator==(const T* ptr) const { return _p == ptr; }
|
|
bool operator!=(const T* ptr) const { return _p != ptr; }
|
|
bool operator>=(const T* ptr) const { return _p >= ptr; }
|
|
bool operator>(const T* ptr) const { return _p > ptr; }
|
|
|
|
inline static void swap(CountedPtr& a, CountedPtr& b)
|
|
{
|
|
std::swap(a._p, b._p);
|
|
}
|
|
|
|
private:
|
|
|
|
T *_p;
|
|
};
|
|
|
|
VFS_NAMESPACE_END
|
|
|
|
#endif
|