improved EditorViewportClient

Author: Roberto De Ioris

Source/UnrealEnginePython/Private/Slate/UEPySPythonEditorViewport.cpp

@@ -4,13 +4,20 @@
 #if WITH_EDITOR
 
 #include "Components/DirectionalLightComponent.h"
+#include "Wrappers/UEPyFEditorViewportClient.h"
 
 static PyObject *py_ue_spython_editor_viewport_get_world(ue_PySPythonEditorViewport *self, PyObject * args)
 {
 	ue_py_slate_cast(SPythonEditorViewport);
 	Py_RETURN_UOBJECT(py_SPythonEditorViewport->GetPythonWorld());
 }
 
+static PyObject *py_ue_spython_editor_viewport_get_editor_viewport_client(ue_PySPythonEditorViewport *self, PyObject * args)
+{
+	ue_py_slate_cast(SPythonEditorViewport);
+	return py_ue_new_feditor_viewport_client(py_SPythonEditorViewport->GetViewportClient().ToSharedRef());
+}
+
 static PyObject *py_ue_spython_editor_viewport_set_show_bounds(ue_PySPythonEditorViewport *self, PyObject * args)
 {
 	ue_py_slate_cast(SPythonEditorViewport);
@@ -241,6 +248,7 @@ static PyObject *py_ue_spython_editor_viewport_get_light(ue_PySPythonEditorViewp
 static PyMethodDef ue_PySPythonEditorViewport_methods[] = {
 	{ "simulate", (PyCFunction)py_ue_spython_editor_viewport_simulate, METH_VARARGS, "" },
 	{ "get_world", (PyCFunction)py_ue_spython_editor_viewport_get_world, METH_VARARGS, "" },
+	{ "get_editor_viewport_client", (PyCFunction)py_ue_spython_editor_viewport_get_editor_viewport_client, METH_VARARGS, "" },
 	{ "set_show_bounds", (PyCFunction)py_ue_spython_editor_viewport_set_show_bounds, METH_VARARGS, "" },
 	{ "set_show_stats", (PyCFunction)py_ue_spython_editor_viewport_set_show_stats, METH_VARARGS, "" },
 	{ "set_view_mode", (PyCFunction)py_ue_spython_editor_viewport_set_view_mode, METH_VARARGS, "" },

Source/UnrealEnginePython/Private/Wrappers/UEPyFEditorViewportClient.cpp

@@ -37,6 +37,11 @@ static PyObject *py_ue_feditor_viewport_client_get_view_location(ue_PyFEditorVie
 	return py_ue_new_fvector(self->editor_viewport_client->GetViewLocation());
 }
 
+static PyObject *py_ue_feditor_viewport_client_get_view_rotation(ue_PyFEditorViewportClient *self, PyObject * args)
+{
+	return py_ue_new_frotator(self->editor_viewport_client->GetViewRotation());
+}
+
 static PyObject *py_ue_feditor_viewport_client_get_camera_speed(ue_PyFEditorViewportClient *self, PyObject * args)
 {
 	return PyFloat_FromDouble(self->editor_viewport_client->GetCameraSpeed());
@@ -84,6 +89,15 @@ static PyObject *py_ue_feditor_viewport_client_set_view_location(ue_PyFEditorVie
 	Py_RETURN_NONE;
 }
 
+static PyObject *py_ue_feditor_viewport_client_set_view_rotation(ue_PyFEditorViewportClient *self, PyObject * args)
+{
+	FRotator rot;
+	if (!py_ue_rotator_arg(args, rot))
+		return PyErr_Format(PyExc_Exception, "argument is not a FRotator");
+	self->editor_viewport_client->SetViewRotation(rot);
+	Py_RETURN_NONE;
+}
+
 static PyObject *py_ue_feditor_viewport_client_set_realtime(ue_PyFEditorViewportClient *self, PyObject * args)
 {
 	PyObject* bInRealtime;
@@ -101,12 +115,14 @@ static PyMethodDef ue_PyFEditorViewportClient_methods[] = {
 	{ "tick", (PyCFunction)py_ue_feditor_viewport_client_tick, METH_VARARGS, "" },
 	{ "get_look_at_location", (PyCFunction)py_ue_feditor_viewport_client_get_look_at_location, METH_VARARGS, "" },
 	{ "get_view_location", (PyCFunction)py_ue_feditor_viewport_client_get_view_location, METH_VARARGS, "" },
+	{ "get_view_rotation", (PyCFunction)py_ue_feditor_viewport_client_get_view_location, METH_VARARGS, "" },
 	{ "get_camera_speed", (PyCFunction)py_ue_feditor_viewport_client_get_camera_speed, METH_VARARGS, "" },
 	{ "get_viewport_dimensions", (PyCFunction)py_ue_feditor_viewport_client_get_viewport_dimensions, METH_VARARGS, "" },
 	{ "is_visible", (PyCFunction)py_ue_feditor_viewport_client_is_visible, METH_VARARGS, "" },
 	{ "get_scene_depth_at_location", (PyCFunction)py_ue_feditor_viewport_client_get_scene_depth_at_location, METH_VARARGS, "" },
 	{ "set_look_at_location", (PyCFunction)py_ue_feditor_viewport_client_set_look_at_location, METH_VARARGS, "" },
 	{ "set_view_location", (PyCFunction)py_ue_feditor_viewport_client_set_view_location, METH_VARARGS, "" },
+	{ "set_view_rotation", (PyCFunction)py_ue_feditor_viewport_client_set_view_rotation, METH_VARARGS, "" },
 	{ "set_realtime", (PyCFunction)py_ue_feditor_viewport_client_set_realtime, METH_VARARGS, "" },
 	{ nullptr }  /* Sentinel */
 };