added get_world_type()

Roberto De Ioris · Roberto De Ioris · commit d7773548188f · 2017-10-10T11:02:01.000+02:00
diff --git a/Source/UnrealEnginePython/Private/UEPyModule.cpp b/Source/UnrealEnginePython/Private/UEPyModule.cpp
@@ -634,6 +634,8 @@ static PyMethodDef ue_PyUObject_methods[] = {
 	{ "quit_game", (PyCFunction)py_ue_quit_game, METH_VARARGS, "" },
 	{ "play", (PyCFunction)py_ue_play, METH_VARARGS, "" },
 
+	{ "get_world_type", (PyCFunction)py_ue_get_world_type, METH_VARARGS, "" },
+
 	{ "world_exec", (PyCFunction)py_ue_world_exec, METH_VARARGS, "" },
 
 	{ "simple_move_to_location", (PyCFunction)py_ue_simple_move_to_location, METH_VARARGS, "" },
diff --git a/Source/UnrealEnginePython/Private/UObject/UEPyWorld.cpp b/Source/UnrealEnginePython/Private/UObject/UEPyWorld.cpp
@@ -42,6 +42,18 @@ PyObject *py_ue_quit_game(ue_PyUObject *self, PyObject * args)
 	Py_RETURN_NONE;
 }
 
+PyObject *py_ue_get_world_type(ue_PyUObject *self, PyObject * args)
+{
+
+	ue_py_check(self);
+
+	UWorld *world = ue_get_uworld(self);
+	if (!world)
+		return PyErr_Format(PyExc_Exception, "unable to retrieve UWorld from uobject");
+
+	return PyLong_FromUnsignedLong(world->WorldType);
+}
+
 PyObject *py_ue_play(ue_PyUObject *self, PyObject * args)
 {
 
diff --git a/Source/UnrealEnginePython/Private/UObject/UEPyWorld.h b/Source/UnrealEnginePython/Private/UObject/UEPyWorld.h
@@ -26,4 +26,6 @@ PyObject *py_ue_get_levels(ue_PyUObject *, PyObject *);
 PyObject *py_ue_get_game_viewport(ue_PyUObject *, PyObject *);
 
 PyObject *py_ue_get_current_level(ue_PyUObject *, PyObject *);
-PyObject *py_ue_set_current_level(ue_PyUObject *, PyObject *);
+PyObject *py_ue_set_current_level(ue_PyUObject *, PyObject *);
+
+PyObject *py_ue_get_world_type(ue_PyUObject *, PyObject *);
