83 lines
2.1 KiB
C++
83 lines
2.1 KiB
C++
#pragma once
|
|
|
|
#include "fl/namespace.h"
|
|
|
|
namespace fl {
|
|
|
|
template <typename T> class Ptr; // Forward declaration
|
|
template <typename T> class WeakPtr; // Forward declaration
|
|
|
|
class Referent; // Forward declaration
|
|
|
|
// Don't inherit from this, this is an internal object.
|
|
class WeakReferent {
|
|
public:
|
|
WeakReferent() : mRefCount(0), mReferent(nullptr) {}
|
|
~WeakReferent() {}
|
|
|
|
void ref() { mRefCount++; }
|
|
int ref_count() const { return mRefCount; }
|
|
void unref() {
|
|
if (--mRefCount == 0) {
|
|
destroy();
|
|
}
|
|
}
|
|
void destroy() { delete this; }
|
|
void setReferent(Referent *referent) { mReferent = referent; }
|
|
Referent *getReferent() const { return mReferent; }
|
|
|
|
protected:
|
|
WeakReferent(const WeakReferent &) = default;
|
|
WeakReferent &operator=(const WeakReferent &) = default;
|
|
|
|
private:
|
|
int mRefCount;
|
|
Referent *mReferent;
|
|
};
|
|
|
|
// Base class for reference counted objects.
|
|
// NOTE: This is legacy - new code should use regular classes with fl::shared_ptr<T>
|
|
class Referent {
|
|
public:
|
|
// NOTE: Start with ref count 0, TakeOwnership() will call ref() to make it 1
|
|
Referent() : mRefCount(0), mWeakReferent(nullptr) {}
|
|
|
|
// NOTE: Copy constructor also starts with ref count 0 for new object
|
|
Referent(const Referent &) : mRefCount(0), mWeakReferent(nullptr) {}
|
|
|
|
// Assignment does not change reference count
|
|
Referent &operator=(const Referent &) { return *this; }
|
|
|
|
virtual ~Referent() {
|
|
if (mWeakReferent) {
|
|
mWeakReferent->setReferent(nullptr);
|
|
mWeakReferent->unref();
|
|
}
|
|
}
|
|
|
|
void ref() { mRefCount++; }
|
|
|
|
int ref_count() const { return mRefCount; }
|
|
|
|
void unref() {
|
|
if (--mRefCount == 0) {
|
|
delete this;
|
|
}
|
|
}
|
|
|
|
WeakReferent *getWeakReferent() {
|
|
if (!mWeakReferent) {
|
|
mWeakReferent = new WeakReferent();
|
|
mWeakReferent->setReferent(this);
|
|
mWeakReferent->ref();
|
|
}
|
|
return mWeakReferent;
|
|
}
|
|
|
|
protected:
|
|
int mRefCount;
|
|
WeakReferent *mWeakReferent;
|
|
};
|
|
|
|
} // namespace fl
|