godot-cpp/include/core/Ref.hpp

183 lines
2.4 KiB
C++
Raw Normal View History

2017-06-19 00:03:59 +00:00
#ifndef REF_H
#define REF_H
#if defined(_WIN32)
# ifdef _GD_CPP_CORE_API_IMPL
# define GD_CPP_CORE_API __declspec(dllexport)
# else
# define GD_CPP_CORE_API __declspec(dllimport)
# endif
#else
# define GD_CPP_CORE_API
#endif
#include "Variant.hpp"
namespace godot {
template<class T>
class Ref {
T *reference;
void ref(const Ref &from)
{
if (from.reference == reference) return;
unref();
reference = from.reference;
if (reference) reference->reference();
}
void ref_pointer(T *r)
{
if (!r) return;
if (r->init_ref()) reference = r;
}
public:
inline bool operator==(const Ref<T> &r) const
{
return reference == r.reference;
}
inline bool operator!=(const Ref<T> &r) const
{
return reference != r.reference;
}
inline T *operator->()
{
return reference;
}
inline T *operator*()
{
return reference;
}
inline T *ptr()
{
return reference;
}
inline const T *operator->() const
{
return reference;
}
inline const T *operator*() const
{
return reference;
}
inline const T *ptr() const
{
return reference;
}
void operator=(const Ref &from)
{
ref(from);
}
template<class T_Other>
void operator=(const Ref<T_Other> &from)
{
Ref<T> n((T *) from.ptr());
ref(n);
}
2017-06-19 00:03:59 +00:00
void operator=(const Variant &variant)
{
T *r = (T *) (Object *) variant;
2017-06-19 00:03:59 +00:00
if (!r) {
unref();
return;
}
Ref re;
re.reference = r;
ref(re);
re.reference = nullptr;
}
operator Variant() const
{
if (reference) reference->reference();
return Variant((Object *) reference);
2017-06-19 00:03:59 +00:00
}
template<class T_Other>
Ref(const Ref<T_Other> &from)
{
if (from.ptr())
ref_pointer((T *) from.ptr());
else
reference = nullptr;
}
2017-06-19 00:03:59 +00:00
Ref(const Ref &from)
{
reference = nullptr;
ref(from);
}
2017-06-19 00:03:59 +00:00
Ref(T *r)
{
if (r)
ref_pointer(r);
else
reference = nullptr;
}
template<class T_Other>
Ref(T_Other *r) : Ref((T *) r) {}
2017-06-19 00:03:59 +00:00
Ref(const Variant &variant)
{
reference = nullptr;
T *r = (T *) (Object *) variant;
2017-06-19 00:03:59 +00:00
if (!r) {
unref();
return;
}
Ref re;
re.reference = r;
ref(re);
re.reference = nullptr;
}
inline bool is_valid() const { return reference != nullptr; }
inline bool is_null() const { return reference == nullptr; }
void unref()
{
if (reference && reference->unreference()) {
godot_object_destroy((godot_object *) reference);
}
reference = nullptr;
}
Ref()
{
reference = nullptr;
}
~Ref()
{
unref();
}
};
}
#endif