added bytes serialization

Author: Roberto De Ioris

Source/UnrealEnginePython/Private/UEPyEditor.cpp

@@ -70,6 +70,26 @@ PyObject *py_unreal_engine_editor_play_in_viewport(PyObject * self, PyObject * a
 
 }
 
+PyObject *py_unreal_engine_request_play_session(PyObject * self, PyObject * args)
+{
+
+	PyObject *py_at_player_start = nullptr;
+	PyObject *py_simulate_in_editor = nullptr;
+
+	if (!PyArg_ParseTuple(args, "|OO:request_play_session", &py_at_player_start, &py_simulate_in_editor))
+	{
+		return nullptr;
+	}
+
+	bool bAtPlayerStart = py_at_player_start && PyObject_IsTrue(py_at_player_start);
+	bool bSimulate = py_simulate_in_editor && PyObject_IsTrue(py_simulate_in_editor);
+
+	GEditor->RequestPlaySession(bAtPlayerStart, nullptr, bSimulate);
+
+	Py_RETURN_NONE;
+
+}
+
 
 PyObject *py_unreal_engine_get_editor_world(PyObject * self, PyObject * args)
 {
@@ -396,7 +416,7 @@ PyObject *py_unreal_engine_import_asset(PyObject * self, PyObject * args)
 		char * filename = PyString_AsString(PyObject_Str(assetsObject));
 #endif
 		files.Add(UTF8_TO_TCHAR(filename));
-	}
+}
 	else
 	{
 		return PyErr_Format(PyExc_Exception, "Not a string nor valid list of string");

Source/UnrealEnginePython/Private/UEPyEditor.h

@@ -118,4 +118,6 @@ PyObject *py_unreal_engine_play_preview_sound(PyObject *, PyObject *);
 PyObject *py_unreal_engine_register_settings(PyObject *, PyObject *);
 PyObject *py_unreal_engine_show_viewer(PyObject *, PyObject *);
 PyObject *py_unreal_engine_unregister_settings(PyObject *, PyObject *);
+
+PyObject *py_unreal_engine_request_play_session(PyObject *, PyObject *);
 #endif

Source/UnrealEnginePython/Private/UEPyModule.cpp

@@ -360,6 +360,7 @@ static PyMethodDef unreal_engine_methods[] = {
 	{ "play_sound", py_unreal_engine_play_sound, METH_VARARGS, "" },
 #if WITH_EDITOR
 	{ "editor_play_in_viewport", py_unreal_engine_editor_play_in_viewport, METH_VARARGS, "" },
+	{ "request_play_session", py_unreal_engine_request_play_session, METH_VARARGS, "" },
 	{ "get_editor_pie_game_viewport_client", py_unreal_engine_get_editor_pie_game_viewport_client, METH_VARARGS, "" },
 	{ "editor_get_active_viewport_screenshot", py_unreal_engine_editor_get_active_viewport_screenshot, METH_VARARGS, "" },
 	{ "editor_get_pie_viewport_screenshot", py_unreal_engine_editor_get_pie_viewport_screenshot, METH_VARARGS, "" },
@@ -799,6 +800,7 @@ static PyMethodDef ue_PyUObject_methods[] = {
 	{ "capture_initialize", (PyCFunction)py_ue_capture_initialize, METH_VARARGS, "" },
 	{ "capture_start", (PyCFunction)py_ue_capture_start, METH_VARARGS, "" },
 	{ "capture_stop", (PyCFunction)py_ue_capture_stop, METH_VARARGS, "" },
+	{ "capture_load_from_config", (PyCFunction)py_ue_capture_load_from_config, METH_VARARGS, "" },
 
 	// Pawn
 	{ "get_controller", (PyCFunction)py_ue_pawn_get_controller, METH_VARARGS, "" },
@@ -953,6 +955,11 @@ static PyMethodDef ue_PyUObject_methods[] = {
 #if ENGINE_MINOR_VERSION >= 15
 	{ "enum_user_defined_names", (PyCFunction)py_ue_enum_user_defined_names, METH_VARARGS, "" },
 #endif
+
+	// serialization
+	{ "to_bytes", (PyCFunction)py_ue_to_bytes, METH_VARARGS, "" },
+	{ "to_bytearray", (PyCFunction)py_ue_to_bytearray, METH_VARARGS, "" },
+	{ "from_bytes", (PyCFunction)py_ue_from_bytes, METH_VARARGS, "" },
 	{ NULL }  /* Sentinel */
 };
 

Source/UnrealEnginePython/Private/UObject/UEPyCapture.cpp

@@ -3,53 +3,92 @@
 #include "Runtime/MovieSceneCapture/Public/MovieSceneCapture.h"
 
 
-PyObject *py_ue_capture_initialize(ue_PyUObject * self, PyObject * args) {
+PyObject *py_ue_capture_initialize(ue_PyUObject * self, PyObject * args)
+{
 
 	ue_py_check(self);
 
-	PyObject *py_widget;
+	PyObject *py_widget = nullptr;
 
-	if (!PyArg_ParseTuple(args, "O", &py_widget)) {
+	if (!PyArg_ParseTuple(args, "|O:capture_initialize", &py_widget))
+	{
 		return nullptr;
 	}
 
 	UMovieSceneCapture *capture = ue_py_check_type<UMovieSceneCapture>(self);
 	if (!capture)
 		return PyErr_Format(PyExc_Exception, "uobject is not a UMovieSceneCapture");
 
-	ue_PySWidget *s_widget = py_ue_is_swidget(py_widget);
-	if (!s_widget)
-		return PyErr_Format(PyExc_Exception, "argument is not a SWidget");
-
 #if WITH_EDITOR
-	if (s_widget->s_widget->GetType().Compare(FName("SPythonEditorViewport")) == 0) {
-		TSharedRef<SPythonEditorViewport> s_viewport = StaticCastSharedRef<SPythonEditorViewport>(s_widget->s_widget);
-		capture->Initialize(s_viewport->GetSceneViewport());
-		capture->StartWarmup();
+	if (py_widget)
+	{
+		ue_PySWidget *s_widget = py_ue_is_swidget(py_widget);
+		if (!s_widget)
+			return PyErr_Format(PyExc_Exception, "argument is not a SWidget");
+
+
+		if (s_widget->s_widget->GetType().Compare(FName("SPythonEditorViewport")) == 0)
+		{
+			TSharedRef<SPythonEditorViewport> s_viewport = StaticCastSharedRef<SPythonEditorViewport>(s_widget->s_widget);
+			capture->Initialize(s_viewport->GetSceneViewport());
+			capture->StartWarmup();
+		}
+		else
+		{
+			return PyErr_Format(PyExc_Exception, "argument is not a supported Viewport-based SWidget");
+		}
+
 	}
-	else {
-		return PyErr_Format(PyExc_Exception, "argument is not a supported Viewport-based SWidget");
+	else
+	{
+		for (const FWorldContext &Context : GEngine->GetWorldContexts())
+		{
+			if (Context.WorldType == EWorldType::PIE)
+			{
+				if (Context.GameViewport)
+				{
+					UE_LOG(LogPython, Error, TEXT("Found Viewport at %p"), Context.GameViewport);
+					FSlatePlayInEditorInfo *SlatePIEInfo = GEditor->SlatePlayInEditorMap.Find(Context.ContextHandle);
+					 
+					capture->Initialize(SlatePIEInfo->SlatePlayInEditorWindowViewport);
+				}
+			}
+		}
 	}
 #endif
+	Py_RETURN_NONE;
+}
 
+PyObject *py_ue_capture_start(ue_PyUObject * self, PyObject * args)
+{
+
+	ue_py_check(self);
+
+	UMovieSceneCapture *capture = ue_py_check_type<UMovieSceneCapture>(self);
+	if (!capture)
+		return PyErr_Format(PyExc_Exception, "uobject is not a UMovieSceneCapture");
+
+	capture->StartCapture();
 
 	Py_RETURN_NONE;
 }
 
-PyObject *py_ue_capture_start(ue_PyUObject * self, PyObject * args) {
+PyObject *py_ue_capture_load_from_config(ue_PyUObject * self, PyObject * args)
+{
 
 	ue_py_check(self);
 
 	UMovieSceneCapture *capture = ue_py_check_type<UMovieSceneCapture>(self);
 	if (!capture)
 		return PyErr_Format(PyExc_Exception, "uobject is not a UMovieSceneCapture");
 
-	capture->StartCapture();
+	capture->LoadFromConfig();
 
 	Py_RETURN_NONE;
 }
 
-PyObject *py_ue_capture_stop(ue_PyUObject * self, PyObject * args) {
+PyObject *py_ue_capture_stop(ue_PyUObject * self, PyObject * args)
+{
 
 	ue_py_check(self);
 

Source/UnrealEnginePython/Private/UObject/UEPyCapture.h

@@ -6,4 +6,5 @@
 
 PyObject *py_ue_capture_initialize(ue_PyUObject *, PyObject *);
 PyObject *py_ue_capture_start(ue_PyUObject *, PyObject *);
+PyObject *py_ue_capture_load_from_config(ue_PyUObject *, PyObject *);
 PyObject *py_ue_capture_stop(ue_PyUObject *, PyObject *);

Source/UnrealEnginePython/Private/UObject/UEPyObject.cpp

@@ -1834,3 +1834,44 @@ PyObject *py_ue_duplicate(ue_PyUObject * self, PyObject * args)
 }
 #endif
 
+
+PyObject *py_ue_to_bytes(ue_PyUObject * self, PyObject * args)
+{
+
+	ue_py_check(self);
+
+	TArray<uint8> Bytes;
+
+	FObjectWriter(self->ue_object, Bytes);
+
+	return PyBytes_FromStringAndSize((const char *)Bytes.GetData(), Bytes.Num());
+}
+
+PyObject *py_ue_to_bytearray(ue_PyUObject * self, PyObject * args)
+{
+
+	ue_py_check(self);
+
+	TArray<uint8> Bytes;
+
+	FObjectWriter(self->ue_object, Bytes);
+
+	return PyByteArray_FromStringAndSize((const char *)Bytes.GetData(), Bytes.Num());
+}
+
+PyObject *py_ue_from_bytes(ue_PyUObject * self, PyObject * args)
+{
+
+	Py_buffer py_buf;
+
+	if (!PyArg_ParseTuple(args, "z*:from_bytes", &py_buf))
+		return nullptr;
+
+	ue_py_check(self);
+
+	TArray<uint8> Bytes((const uint8 *)py_buf.buf, py_buf.len);
+
+	FObjectReader(self->ue_object, Bytes);
+
+	Py_RETURN_NONE;
+}

Source/UnrealEnginePython/Private/UObject/UEPyObject.h

@@ -89,4 +89,8 @@ PyObject *py_ue_import_custom_properties(ue_PyUObject *, PyObject *);
 #endif
 
 PyObject *py_ue_get_thumbnail(ue_PyUObject *, PyObject *);
-PyObject *py_ue_render_thumbnail(ue_PyUObject *, PyObject *);
+PyObject *py_ue_render_thumbnail(ue_PyUObject *, PyObject *);
+
+PyObject *py_ue_to_bytes(ue_PyUObject *, PyObject *);
+PyObject *py_ue_to_bytearray(ue_PyUObject *, PyObject *);
+PyObject *py_ue_from_bytes(ue_PyUObject *, PyObject *);