owning api

Author: Roberto De Ioris

Source/UnrealEnginePython/Private/UEPyModule.cpp

@@ -656,6 +656,10 @@ static PyMethodDef ue_PyUObject_methods[] = {
 	{ "auto_root", (PyCFunction)py_ue_auto_root, METH_VARARGS, "" },
 	{ "remove_from_root", (PyCFunction)py_ue_remove_from_root, METH_VARARGS, "" },
 
+	{ "own", (PyCFunction)py_ue_own, METH_VARARGS, "" },
+	{ "disown", (PyCFunction)py_ue_disown, METH_VARARGS, "" },
+	{ "is_owned", (PyCFunction)py_ue_is_owned, METH_VARARGS, "" },
+
 	{ "find_function", (PyCFunction)py_ue_find_function, METH_VARARGS, "" },
 
 #pragma warning(suppress: 4191)

Source/UnrealEnginePython/Private/UObject/UEPyObject.cpp

@@ -1244,6 +1244,55 @@ PyObject *py_ue_add_to_root(ue_PyUObject *self, PyObject * args)
 	Py_RETURN_NONE;
 }
 
+PyObject *py_ue_own(ue_PyUObject *self, PyObject * args)
+{
+
+	ue_py_check(self);
+
+	if (self->owned)
+	{
+		return PyErr_Format(PyExc_Exception, "uobject already owned");
+	}
+
+	Py_DECREF(self);
+
+	self->owned = 1;
+	FUnrealEnginePythonHouseKeeper::Get()->TrackUObject(self->ue_object);
+
+	Py_RETURN_NONE;
+}
+
+PyObject *py_ue_disown(ue_PyUObject *self, PyObject * args)
+{
+
+	ue_py_check(self);
+
+	if (!self->owned)
+	{
+		return PyErr_Format(PyExc_Exception, "uobject not owned");
+	}
+
+	Py_INCREF(self);
+
+	self->owned = 0;
+	FUnrealEnginePythonHouseKeeper::Get()->UntrackUObject(self->ue_object);
+
+	Py_RETURN_NONE;
+}
+
+PyObject *py_ue_is_owned(ue_PyUObject *self, PyObject * args)
+{
+
+	ue_py_check(self);
+
+	if (!self->owned)
+	{
+		Py_RETURN_FALSE;
+	}
+
+	Py_RETURN_TRUE;
+}
+
 PyObject *py_ue_auto_root(ue_PyUObject *self, PyObject * args)
 {
 

Source/UnrealEnginePython/Private/UObject/UEPyObject.h

@@ -31,6 +31,9 @@ PyObject *py_ue_is_rooted(ue_PyUObject *, PyObject *);
 PyObject *py_ue_add_to_root(ue_PyUObject *, PyObject *);
 PyObject *py_ue_remove_from_root(ue_PyUObject *, PyObject *);
 PyObject *py_ue_auto_root(ue_PyUObject *, PyObject *);
+PyObject *py_ue_own(ue_PyUObject *, PyObject *);
+PyObject *py_ue_disown(ue_PyUObject *, PyObject *);
+PyObject *py_ue_is_owned(ue_PyUObject *, PyObject *);
 
 PyObject *py_ue_save_config(ue_PyUObject *, PyObject *);