1 files changed, 113 insertions, 0 deletions
diff --git a/source/RcPtr.hpp b/source/RcPtr.hpp
new file mode 100644
index 0000000..90097e1
--- /dev/null
+++ b/source/RcPtr.hpp
@@ -0,0 +1,113 @@
+#pragma once
+
+#include "Macros.hpp"
+#include "TypeTraits.hpp"
+
+#include <cstddef>
+#include <optional>
+#include <type_traits>
+
+class RefCounted {
+public:
+	// DO NOT MODIFY this field, unless explicitly documented the use
+	size_t refCount = 0;
+};
+
+template <class T, class TDeleter = DefaultDeleter<T>>
+class RcPtr : TDeleter {
+private:
+	static_assert(std::is_base_of_v<RefCounted, T>);
+	T* mPtr;
+
+public:
+	RcPtr()
+		: mPtr{ nullptr } {
+	}
+
+	explicit RcPtr(T* ptr)
+		: mPtr{ ptr } {
+		if (ptr) {
+			++ptr->refCount;
+		}
+	}
+
+	~RcPtr() {
+		CleanUp();
+	}
+
+	void Attach(T* ptr) {
+		CleanUp();
+		mPtr = ptr;
+		if (ptr) {
+			++ptr->refCount;
+		}
+	}
+
+	void Detatch() {
+		CleanUp();
+		mPtr = nullptr;
+	}
+
+	RcPtr(const RcPtr& that)
+		: mPtr{ that.mPtr } {
+		if (mPtr) {
+			++mPtr->refCount;
+		}
+	}
+
+	RcPtr& operator=(const RcPtr& that) {
+		CleanUp();
+		mPtr = that.mPtr;
+		if (mPtr) {
+			++mPtr->refCount;
+		}
+		return *this;
+	}
+
+	RcPtr(RcPtr&& that)
+		: mPtr{ that.mPtr } {
+		that.mPtr = nullptr;
+	}
+
+	RcPtr& operator=(RcPtr&& that) {
+		CleanUp();
+		mPtr = that.mPtr;
+		that.mPtr = nullptr;
+		return *this;
+	}
+
+	template <class TBase>
+	requires std::is_base_of_v<TBase, T>
+	operator RcPtr<TBase>() const {
+		return RcPtr<TBase>(mPtr);
+	}
+
+	bool operator==(std::nullptr_t ptr) const {
+		return mPtr == nullptr;
+	}
+
+	bool operator==(const T* ptr) const {
+		return mPtr == ptr;
+	}
+
+	bool operator==(T* ptr) const {
+		return mPtr == ptr;
+	}
+
+	T* Get() const {
+		return mPtr;
+	}
+
+	T& operator*() const { return *mPtr; }
+	T* operator->() const { return mPtr; }
+
+private:
+	void CleanUp() {
+		if (mPtr) {
+			--mPtr->refCount;
+			if (mPtr->refCount == 0) {
+				TDeleter::operator()(mPtr);
+			}
+		}
+	}
+};