mirror of
https://github.com/Relintai/pandemonium_engine.git
synced 2024-11-21 16:37:20 +01:00
466 lines
11 KiB
C++
466 lines
11 KiB
C++
#ifndef REFERENCE_H
|
|
#define REFERENCE_H
|
|
|
|
/*************************************************************************/
|
|
/* reference.h */
|
|
/*************************************************************************/
|
|
/* This file is part of: */
|
|
/* PANDEMONIUM ENGINE */
|
|
/* https://github.com/Relintai/pandemonium_engine */
|
|
/*************************************************************************/
|
|
/* Copyright (c) 2022-present Péter Magyar. */
|
|
/* Copyright (c) 2014-2022 Godot Engine contributors (cf. AUTHORS.md). */
|
|
/* Copyright (c) 2007-2022 Juan Linietsky, Ariel Manzur. */
|
|
/* */
|
|
/* Permission is hereby granted, free of charge, to any person obtaining */
|
|
/* a copy of this software and associated documentation files (the */
|
|
/* "Software"), to deal in the Software without restriction, including */
|
|
/* without limitation the rights to use, copy, modify, merge, publish, */
|
|
/* distribute, sublicense, and/or sell copies of the Software, and to */
|
|
/* permit persons to whom the Software is furnished to do so, subject to */
|
|
/* the following conditions: */
|
|
/* */
|
|
/* The above copyright notice and this permission notice shall be */
|
|
/* included in all copies or substantial portions of the Software. */
|
|
/* */
|
|
/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
|
|
/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
|
|
/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
|
|
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
|
|
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
|
|
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
|
|
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
|
|
/*************************************************************************/
|
|
|
|
#include "core/object/class_db.h"
|
|
#include "core/object/object.h"
|
|
#include "core/object/ref_ptr.h"
|
|
#include "core/os/safe_refcount.h"
|
|
|
|
class Reference : public Object {
|
|
GDCLASS(Reference, Object);
|
|
friend class RefBase;
|
|
SafeRefCount refcount;
|
|
SafeRefCount refcount_init;
|
|
|
|
protected:
|
|
static void _bind_methods();
|
|
|
|
public:
|
|
_FORCE_INLINE_ bool is_referenced() const { return refcount_init.get() != 1; }
|
|
bool init_ref();
|
|
bool reference(); // returns false if refcount is at zero and didn't get increased
|
|
bool unreference();
|
|
int reference_get_count() const;
|
|
|
|
Reference();
|
|
~Reference();
|
|
};
|
|
|
|
template <class T>
|
|
class Ref {
|
|
T *reference;
|
|
|
|
void ref(const Ref &p_from) {
|
|
if (p_from.reference == reference) {
|
|
return;
|
|
}
|
|
|
|
unref();
|
|
|
|
reference = p_from.reference;
|
|
if (reference) {
|
|
reference->reference();
|
|
}
|
|
}
|
|
|
|
void ref_pointer(T *p_ref) {
|
|
ERR_FAIL_COND(!p_ref);
|
|
|
|
if (p_ref->init_ref()) {
|
|
reference = p_ref;
|
|
}
|
|
}
|
|
|
|
//virtual Reference * get_reference() const { return reference; }
|
|
public:
|
|
_FORCE_INLINE_ bool operator==(const T *p_ptr) const {
|
|
return reference == p_ptr;
|
|
}
|
|
_FORCE_INLINE_ bool operator!=(const T *p_ptr) const {
|
|
return reference != p_ptr;
|
|
}
|
|
|
|
_FORCE_INLINE_ bool operator<(const Ref<T> &p_r) const {
|
|
return reference < p_r.reference;
|
|
}
|
|
_FORCE_INLINE_ bool operator==(const Ref<T> &p_r) const {
|
|
return reference == p_r.reference;
|
|
}
|
|
_FORCE_INLINE_ bool operator!=(const Ref<T> &p_r) const {
|
|
return reference != p_r.reference;
|
|
}
|
|
|
|
_FORCE_INLINE_ T *operator->() {
|
|
return reference;
|
|
}
|
|
|
|
_FORCE_INLINE_ T *operator*() {
|
|
return reference;
|
|
}
|
|
|
|
_FORCE_INLINE_ const T *operator->() const {
|
|
return reference;
|
|
}
|
|
|
|
_FORCE_INLINE_ const T *ptr() const {
|
|
return reference;
|
|
}
|
|
_FORCE_INLINE_ T *ptr() {
|
|
return reference;
|
|
}
|
|
|
|
_FORCE_INLINE_ const T *operator*() const {
|
|
return reference;
|
|
}
|
|
|
|
RefPtr get_ref_ptr() const {
|
|
RefPtr refptr;
|
|
Ref<Reference> *irr = reinterpret_cast<Ref<Reference> *>(refptr.get_data());
|
|
*irr = *this;
|
|
return refptr;
|
|
};
|
|
|
|
operator Variant() const {
|
|
return Variant(get_ref_ptr());
|
|
}
|
|
|
|
void operator=(const Ref &p_from) {
|
|
ref(p_from);
|
|
}
|
|
|
|
template <class T_Other>
|
|
void operator=(const Ref<T_Other> &p_from) {
|
|
Reference *refb = const_cast<Reference *>(static_cast<const Reference *>(p_from.ptr()));
|
|
if (!refb) {
|
|
unref();
|
|
return;
|
|
}
|
|
Ref r;
|
|
r.reference = Object::cast_to<T>(refb);
|
|
ref(r);
|
|
r.reference = nullptr;
|
|
}
|
|
|
|
void operator=(const RefPtr &p_refptr) {
|
|
Ref<Reference> *irr = reinterpret_cast<Ref<Reference> *>(p_refptr.get_data());
|
|
Reference *refb = irr->ptr();
|
|
if (!refb) {
|
|
unref();
|
|
return;
|
|
}
|
|
Ref r;
|
|
r.reference = Object::cast_to<T>(refb);
|
|
ref(r);
|
|
r.reference = nullptr;
|
|
}
|
|
|
|
void operator=(const Variant &p_variant) {
|
|
RefPtr refptr = p_variant;
|
|
Ref<Reference> *irr = reinterpret_cast<Ref<Reference> *>(refptr.get_data());
|
|
Reference *refb = irr->ptr();
|
|
if (!refb) {
|
|
unref();
|
|
return;
|
|
}
|
|
Ref r;
|
|
r.reference = Object::cast_to<T>(refb);
|
|
ref(r);
|
|
r.reference = nullptr;
|
|
}
|
|
|
|
template <class T_Other>
|
|
void reference_ptr(T_Other *p_ptr) {
|
|
if (reference == p_ptr) {
|
|
return;
|
|
}
|
|
unref();
|
|
|
|
T *r = Object::cast_to<T>(p_ptr);
|
|
if (r) {
|
|
ref_pointer(r);
|
|
}
|
|
}
|
|
|
|
Ref(const Ref &p_from) {
|
|
reference = nullptr;
|
|
ref(p_from);
|
|
}
|
|
|
|
template <class T_Other>
|
|
Ref(const Ref<T_Other> &p_from) {
|
|
reference = nullptr;
|
|
Reference *refb = const_cast<Reference *>(static_cast<const Reference *>(p_from.ptr()));
|
|
if (!refb) {
|
|
unref();
|
|
return;
|
|
}
|
|
Ref r;
|
|
r.reference = Object::cast_to<T>(refb);
|
|
ref(r);
|
|
r.reference = nullptr;
|
|
}
|
|
|
|
Ref(T *p_reference) {
|
|
reference = nullptr;
|
|
if (p_reference) {
|
|
ref_pointer(p_reference);
|
|
}
|
|
}
|
|
|
|
Ref(const Variant &p_variant) {
|
|
RefPtr refptr = p_variant;
|
|
Ref<Reference> *irr = reinterpret_cast<Ref<Reference> *>(refptr.get_data());
|
|
reference = nullptr;
|
|
Reference *refb = irr->ptr();
|
|
if (!refb) {
|
|
unref();
|
|
return;
|
|
}
|
|
Ref r;
|
|
r.reference = Object::cast_to<T>(refb);
|
|
ref(r);
|
|
r.reference = nullptr;
|
|
}
|
|
|
|
Ref(const RefPtr &p_refptr) {
|
|
Ref<Reference> *irr = reinterpret_cast<Ref<Reference> *>(p_refptr.get_data());
|
|
reference = nullptr;
|
|
Reference *refb = irr->ptr();
|
|
if (!refb) {
|
|
unref();
|
|
return;
|
|
}
|
|
Ref r;
|
|
r.reference = Object::cast_to<T>(refb);
|
|
ref(r);
|
|
r.reference = nullptr;
|
|
}
|
|
|
|
inline bool is_valid() const { return reference != nullptr; }
|
|
inline bool is_null() const { return reference == nullptr; }
|
|
|
|
void unref() {
|
|
//TODO this should be moved to mutexes, since this engine does not really
|
|
// do a lot of referencing on references and stuff
|
|
// mutexes will avoid more crashes?
|
|
|
|
if (reference && reference->unreference()) {
|
|
memdelete(reference);
|
|
}
|
|
reference = nullptr;
|
|
}
|
|
|
|
void instance() {
|
|
ref(memnew(T));
|
|
}
|
|
|
|
Ref() {
|
|
reference = nullptr;
|
|
}
|
|
|
|
~Ref() {
|
|
unref();
|
|
}
|
|
};
|
|
|
|
typedef Ref<Reference> REF;
|
|
|
|
class WeakRef : public Reference {
|
|
GDCLASS(WeakRef, Reference);
|
|
|
|
ObjectID ref;
|
|
|
|
protected:
|
|
static void _bind_methods();
|
|
|
|
public:
|
|
Variant get_ref() const;
|
|
void set_obj(Object *p_object);
|
|
void set_ref(const REF &p_ref);
|
|
|
|
WeakRef();
|
|
};
|
|
|
|
template <class T>
|
|
class WRef {
|
|
public:
|
|
_FORCE_INLINE_ bool operator==(const T *p_ptr) const {
|
|
ObjectID oref;
|
|
|
|
if (p_ptr) {
|
|
oref = p_ptr->get_instance_id();
|
|
}
|
|
|
|
return ref == oref;
|
|
}
|
|
_FORCE_INLINE_ bool operator!=(const T *p_ptr) const {
|
|
ObjectID oref;
|
|
|
|
if (p_ptr) {
|
|
oref = p_ptr->get_instance_id();
|
|
}
|
|
|
|
return ref != oref;
|
|
}
|
|
|
|
_FORCE_INLINE_ bool operator==(const Ref<T> &p_r) const {
|
|
ObjectID oref;
|
|
|
|
if (p_r.is_valid()) {
|
|
oref = p_r->get_instance_id();
|
|
}
|
|
|
|
return ref == oref;
|
|
}
|
|
_FORCE_INLINE_ bool operator!=(const Ref<T> &p_r) const {
|
|
ObjectID oref;
|
|
|
|
if (p_r.is_valid()) {
|
|
oref = p_r->get_instance_id();
|
|
}
|
|
|
|
return ref != oref;
|
|
}
|
|
|
|
_FORCE_INLINE_ bool operator<(const WRef<T> &p_r) const {
|
|
return ref < p_r.ref;
|
|
}
|
|
_FORCE_INLINE_ bool operator==(const WRef<T> &p_r) const {
|
|
return ref == p_r.ref;
|
|
}
|
|
_FORCE_INLINE_ bool operator!=(const WRef<T> &p_r) const {
|
|
return ref != p_r.ref;
|
|
}
|
|
|
|
_FORCE_INLINE_ Ref<T> get_ref() const {
|
|
if (ref == 0) {
|
|
return Ref<T>();
|
|
}
|
|
|
|
Object *obj = ObjectDB::get_instance(ref);
|
|
if (!obj) {
|
|
return Ref<T>();
|
|
}
|
|
|
|
Reference *r = Object::cast_to<Reference>(obj);
|
|
|
|
if (r) {
|
|
return Ref<T>(r);
|
|
}
|
|
|
|
return Ref<T>();
|
|
}
|
|
|
|
_FORCE_INLINE_ void set_obj(T *p_object) {
|
|
ref = p_object ? p_object->get_instance_id() : 0;
|
|
}
|
|
|
|
_FORCE_INLINE_ void set_ref(const Ref<T> &p_ref) {
|
|
ref = p_ref.is_valid() ? p_ref->get_instance_id() : 0;
|
|
}
|
|
|
|
void operator=(const WRef<T> &p_from) {
|
|
ref = p_from.ref;
|
|
}
|
|
|
|
WRef(const WRef<T> &p_from) {
|
|
ref = p_from.ref;
|
|
}
|
|
|
|
WRef(T *p_reference) {
|
|
set_obj(p_reference);
|
|
}
|
|
|
|
WRef(const Ref<T> &p_reference) {
|
|
set_ref(p_reference);
|
|
}
|
|
|
|
WRef() {
|
|
}
|
|
|
|
protected:
|
|
static void _bind_methods();
|
|
|
|
ObjectID ref;
|
|
};
|
|
|
|
#ifdef PTRCALL_ENABLED
|
|
|
|
template <class T>
|
|
struct PtrToArg<Ref<T>> {
|
|
_FORCE_INLINE_ static Ref<T> convert(const void *p_ptr) {
|
|
return Ref<T>(const_cast<T *>(reinterpret_cast<const T *>(p_ptr)));
|
|
}
|
|
|
|
_FORCE_INLINE_ static void encode(Ref<T> p_val, const void *p_ptr) {
|
|
*(Ref<Reference> *)p_ptr = p_val;
|
|
}
|
|
};
|
|
|
|
template <class T>
|
|
struct PtrToArg<const Ref<T> &> {
|
|
_FORCE_INLINE_ static Ref<T> convert(const void *p_ptr) {
|
|
return Ref<T>((T *)p_ptr);
|
|
}
|
|
};
|
|
|
|
//this is for RefPtr
|
|
|
|
template <>
|
|
struct PtrToArg<RefPtr> {
|
|
_FORCE_INLINE_ static RefPtr convert(const void *p_ptr) {
|
|
return Ref<Reference>(const_cast<Reference *>(reinterpret_cast<const Reference *>(p_ptr))).get_ref_ptr();
|
|
}
|
|
|
|
_FORCE_INLINE_ static void encode(RefPtr p_val, const void *p_ptr) {
|
|
Ref<Reference> r = p_val;
|
|
*(Ref<Reference> *)p_ptr = r;
|
|
}
|
|
};
|
|
|
|
template <>
|
|
struct PtrToArg<const RefPtr &> {
|
|
_FORCE_INLINE_ static RefPtr convert(const void *p_ptr) {
|
|
return Ref<Reference>(const_cast<Reference *>(reinterpret_cast<const Reference *>(p_ptr))).get_ref_ptr();
|
|
}
|
|
};
|
|
|
|
#endif // PTRCALL_ENABLED
|
|
|
|
#ifdef DEBUG_METHODS_ENABLED
|
|
|
|
template <class T>
|
|
struct GetTypeInfo<Ref<T>> {
|
|
static const Variant::Type VARIANT_TYPE = Variant::OBJECT;
|
|
static const PandemoniumTypeInfo::Metadata METADATA = PandemoniumTypeInfo::METADATA_NONE;
|
|
|
|
static inline PropertyInfo get_class_info() {
|
|
return PropertyInfo(Variant::OBJECT, String(), PROPERTY_HINT_RESOURCE_TYPE, T::get_class_static());
|
|
}
|
|
};
|
|
|
|
template <class T>
|
|
struct GetTypeInfo<const Ref<T> &> {
|
|
static const Variant::Type VARIANT_TYPE = Variant::OBJECT;
|
|
static const PandemoniumTypeInfo::Metadata METADATA = PandemoniumTypeInfo::METADATA_NONE;
|
|
|
|
static inline PropertyInfo get_class_info() {
|
|
return PropertyInfo(Variant::OBJECT, String(), PROPERTY_HINT_RESOURCE_TYPE, T::get_class_static());
|
|
}
|
|
};
|
|
|
|
#endif // DEBUG_METHODS_ENABLED
|
|
|
|
#endif // REFERENCE_H
|