added create_widget()

Roberto De Ioris · Roberto De Ioris · commit 592ef4cde9b5 · 2017-08-11T09:13:08.000+02:00
diff --git a/Source/UnrealEnginePython/Private/UEPyModule.cpp b/Source/UnrealEnginePython/Private/UEPyModule.cpp
@@ -629,6 +629,7 @@ static PyMethodDef ue_PyUObject_methods[] = {
 
 	// Widget
 	{ "take_widget", (PyCFunction)py_ue_take_widget, METH_VARARGS, "" },
+	{ "create_widget", (PyCFunction)py_ue_create_widget, METH_VARARGS, "" },
 
 	// WidgetComponent
 	{ "set_slate_widget", (PyCFunction)py_ue_set_slate_widget, METH_VARARGS, "" },
diff --git a/Source/UnrealEnginePython/Private/UObject/UEPyWidget.cpp b/Source/UnrealEnginePython/Private/UObject/UEPyWidget.cpp
@@ -8,11 +8,41 @@ PyObject *py_ue_take_widget(ue_PyUObject * self, PyObject * args) {
 
 	ue_py_check(self);
 
-    UWidget *widget = ue_py_check_type<UWidget>(self);
+	UWidget *widget = ue_py_check_type<UWidget>(self);
 	if (!widget)
 		return PyErr_Format(PyExc_Exception, "uobject is not a UWidget");
 
 	ue_PySWidget *s_widget = ue_py_get_swidget(widget->TakeWidget());
 
-    return (PyObject *)s_widget;
+	return (PyObject *)s_widget;
+}
+
+PyObject *py_ue_create_widget(ue_PyUObject * self, PyObject * args) {
+
+	ue_py_check(self);
+
+	PyObject *py_class;
+	if (!PyArg_ParseTuple(args, "O", &py_class))
+		return nullptr;
+
+	APlayerController *player = ue_py_check_type<APlayerController>(self);
+	if (!player)
+		return PyErr_Format(PyExc_Exception, "uobject is not a APlayerController");
+
+	UClass *u_class = ue_py_check_type<UClass>(py_class);
+	if (!u_class)
+		return PyErr_Format(PyExc_Exception, "argument is not a UClass");
+
+	if (!u_class->IsChildOf<UUserWidget>())
+		return PyErr_Format(PyExc_Exception, "argument is not a child of UUserWidget");
+
+	UUserWidget *widget = CreateWidget<UUserWidget>(player, u_class);
+	if (!widget)
+		return PyErr_Format(PyExc_Exception, "unable to create new widget");
+
+	ue_PyUObject *ret = ue_get_python_wrapper(widget);
+	if (!ret)
+		return PyErr_Format(PyExc_Exception, "PyUObject is in invalid state");
+	Py_INCREF(ret);
+	return (PyObject *)ret;
 }
diff --git a/Source/UnrealEnginePython/Private/UObject/UEPyWidget.h b/Source/UnrealEnginePython/Private/UObject/UEPyWidget.h
@@ -4,4 +4,5 @@
 #include "UnrealEnginePython.h"
 
 
-PyObject *py_ue_take_widget(ue_PyUObject *, PyObject *);
+PyObject *py_ue_take_widget(ue_PyUObject *, PyObject *);
+PyObject *py_ue_create_widget(ue_PyUObject *, PyObject *);
