godot/core/reference.h

396 lines
9.0 KiB
C++
Raw Normal View History

2014-02-10 09:10:30 +08:00
/*************************************************************************/
/* reference.h */
/*************************************************************************/
/* This file is part of: */
/* GODOT ENGINE */
/* https://godotengine.org */
2014-02-10 09:10:30 +08:00
/*************************************************************************/
/* Copyright (c) 2007-2020 Juan Linietsky, Ariel Manzur. */
/* Copyright (c) 2014-2020 Godot Engine contributors (cf. AUTHORS.md). */
2014-02-10 09:10:30 +08:00
/* */
/* 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. */
/*************************************************************************/
2014-02-10 09:10:30 +08:00
#ifndef REFERENCE_H
#define REFERENCE_H
#include "core/class_db.h"
#include "core/object.h"
#include "core/ref_ptr.h"
#include "core/safe_refcount.h"
2014-02-10 09:10:30 +08:00
class Reference : public Object {
2014-02-10 09:10:30 +08:00
GDCLASS(Reference, Object);
friend class RefBase;
2014-02-10 09:10:30 +08:00
SafeRefCount refcount;
SafeRefCount refcount_init;
protected:
2014-02-10 09:10:30 +08:00
static void _bind_methods();
public:
2019-09-04 00:04:37 +08:00
_FORCE_INLINE_ bool is_referenced() const { return refcount_init.get() != 1; }
2014-02-10 09:10:30 +08:00
bool init_ref();
bool reference(); // returns false if refcount is at zero and didn't get increased
2014-02-10 09:10:30 +08:00
bool unreference();
int reference_get_count() const;
2014-02-10 09:10:30 +08:00
Reference();
~Reference();
};
template <class T>
2014-02-10 09:10:30 +08:00
class Ref {
T *reference;
2014-02-10 09:10:30 +08:00
void ref(const Ref &p_from) {
2014-02-10 09:10:30 +08:00
if (p_from.reference == reference)
2014-02-10 09:10:30 +08:00
return;
unref();
reference = p_from.reference;
2014-02-10 09:10:30 +08:00
if (reference)
reference->reference();
}
void ref_pointer(T *p_ref) {
2014-02-10 09:10:30 +08:00
ERR_FAIL_COND(!p_ref);
if (p_ref->init_ref())
reference = p_ref;
2014-02-10 09:10:30 +08:00
}
//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 {
2014-02-10 09:10:30 +08:00
return reference < p_r.reference;
2014-02-10 09:10:30 +08:00
}
_FORCE_INLINE_ bool operator==(const Ref<T> &p_r) const {
2014-02-10 09:10:30 +08:00
return reference == p_r.reference;
2014-02-10 09:10:30 +08:00
}
_FORCE_INLINE_ bool operator!=(const Ref<T> &p_r) const {
2014-02-10 09:10:30 +08:00
return reference != p_r.reference;
2014-02-10 09:10:30 +08:00
}
_FORCE_INLINE_ T *operator->() {
2014-02-10 09:10:30 +08:00
return reference;
}
_FORCE_INLINE_ T *operator*() {
2014-02-10 09:10:30 +08:00
return reference;
}
_FORCE_INLINE_ const T *operator->() const {
2014-02-10 09:10:30 +08:00
return reference;
}
_FORCE_INLINE_ const T *ptr() const {
2014-02-10 09:10:30 +08:00
return reference;
}
_FORCE_INLINE_ T *ptr() {
2014-02-10 09:10:30 +08:00
return reference;
}
_FORCE_INLINE_ const T *operator*() const {
2014-02-10 09:10:30 +08:00
return reference;
}
RefPtr get_ref_ptr() const {
RefPtr refptr;
Ref<Reference> *irr = reinterpret_cast<Ref<Reference> *>(refptr.get_data());
2014-02-10 09:10:30 +08:00
*irr = *this;
return refptr;
};
operator Variant() const {
return Variant(get_ref_ptr());
2014-02-10 09:10:30 +08:00
}
void operator=(const Ref &p_from) {
2014-02-10 09:10:30 +08:00
ref(p_from);
}
template <class T_Other>
void operator=(const Ref<T_Other> &p_from) {
2014-02-10 09:10:30 +08:00
Reference *refb = const_cast<Reference *>(static_cast<const Reference *>(p_from.ptr()));
2014-02-10 09:10:30 +08:00
if (!refb) {
unref();
return;
}
Ref r;
r.reference = Object::cast_to<T>(refb);
2014-02-10 09:10:30 +08:00
ref(r);
r.reference = NULL;
2014-02-10 09:10:30 +08:00
}
void operator=(const RefPtr &p_refptr) {
2014-02-10 09:10:30 +08:00
Ref<Reference> *irr = reinterpret_cast<Ref<Reference> *>(p_refptr.get_data());
2014-02-10 09:10:30 +08:00
Reference *refb = irr->ptr();
if (!refb) {
unref();
return;
}
Ref r;
r.reference = Object::cast_to<T>(refb);
2014-02-10 09:10:30 +08:00
ref(r);
r.reference = NULL;
2014-02-10 09:10:30 +08:00
}
void operator=(const Variant &p_variant) {
2014-02-10 09:10:30 +08:00
RefPtr refptr = p_variant;
Ref<Reference> *irr = reinterpret_cast<Ref<Reference> *>(refptr.get_data());
2014-02-10 09:10:30 +08:00
Reference *refb = irr->ptr();
if (!refb) {
unref();
return;
}
Ref r;
r.reference = Object::cast_to<T>(refb);
2014-02-10 09:10:30 +08:00
ref(r);
r.reference = NULL;
2014-02-10 09:10:30 +08:00
}
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) {
2014-02-10 09:10:30 +08:00
reference = NULL;
2014-02-10 09:10:30 +08:00
ref(p_from);
}
template <class T_Other>
Ref(const Ref<T_Other> &p_from) {
2014-02-10 09:10:30 +08:00
reference = NULL;
Reference *refb = const_cast<Reference *>(static_cast<const Reference *>(p_from.ptr()));
2014-02-10 09:10:30 +08:00
if (!refb) {
unref();
return;
}
Ref r;
r.reference = Object::cast_to<T>(refb);
2014-02-10 09:10:30 +08:00
ref(r);
r.reference = NULL;
2014-02-10 09:10:30 +08:00
}
Ref(T *p_reference) {
2014-02-10 09:10:30 +08:00
reference = NULL;
2014-02-10 09:10:30 +08:00
if (p_reference)
ref_pointer(p_reference);
}
Ref(const Variant &p_variant) {
2014-02-10 09:10:30 +08:00
RefPtr refptr = p_variant;
Ref<Reference> *irr = reinterpret_cast<Ref<Reference> *>(refptr.get_data());
reference = NULL;
2014-02-10 09:10:30 +08:00
Reference *refb = irr->ptr();
if (!refb) {
unref();
return;
}
Ref r;
r.reference = Object::cast_to<T>(refb);
2014-02-10 09:10:30 +08:00
ref(r);
r.reference = NULL;
2014-02-10 09:10:30 +08:00
}
Ref(const RefPtr &p_refptr) {
2014-02-10 09:10:30 +08:00
Ref<Reference> *irr = reinterpret_cast<Ref<Reference> *>(p_refptr.get_data());
reference = NULL;
2014-02-10 09:10:30 +08:00
Reference *refb = irr->ptr();
if (!refb) {
unref();
return;
}
Ref r;
r.reference = Object::cast_to<T>(refb);
2014-02-10 09:10:30 +08:00
ref(r);
r.reference = NULL;
2014-02-10 09:10:30 +08:00
}
inline bool is_valid() const { return reference != NULL; }
inline bool is_null() const { return reference == NULL; }
2014-02-10 09:10:30 +08:00
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 = NULL;
2014-02-10 09:10:30 +08:00
}
2015-06-14 13:13:47 +08:00
void instance() {
ref(memnew(T));
2015-06-14 13:13:47 +08:00
}
2014-02-10 09:10:30 +08:00
Ref() {
reference = NULL;
2014-02-10 09:10:30 +08:00
}
~Ref() {
unref();
}
};
typedef Ref<Reference> REF;
class WeakRef : public Reference {
GDCLASS(WeakRef, Reference);
2014-02-10 09:10:30 +08:00
ObjectID ref;
2014-02-10 09:10:30 +08:00
protected:
static void _bind_methods();
public:
Variant get_ref() const;
void set_obj(Object *p_object);
void set_ref(const REF &p_ref);
2014-02-10 09:10:30 +08:00
WeakRef();
};
#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);
}
};
2014-02-10 09:10:30 +08:00
//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 GodotTypeInfo::Metadata METADATA = GodotTypeInfo::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 GodotTypeInfo::Metadata METADATA = GodotTypeInfo::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
2014-02-10 09:10:30 +08:00
#endif // REFERENCE_H