Signed-off-by: myperworld <myperworld@126.com>
This commit is contained in:
1 parent
0986832fd1
commit
aacd052c06
3403 files changed
+597376
-12
No files matched your search
@@ -0,0 +1,82 @@
|
||||
#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
|
||||
Reference in new issue
Block a user