added archetype management

Author: Roberto De Ioris

Source/UnrealEnginePython/Private/UEPyEditor.cpp

@@ -37,9 +37,22 @@
 #include "Editor/MainFrame/Public/Interfaces/IMainFrameModule.h"
 
 #include "Runtime/Core/Public/HAL/ThreadHeartBeat.h"
+#include "Runtime/Engine/Public/EditorSupportDelegates.h"
 
 #include "UEPyIPlugin.h"
 
+PyObject *py_unreal_engine_redraw_all_viewports(PyObject * self, PyObject * args)
+{
+	FEditorSupportDelegates::RedrawAllViewports.Broadcast();
+	Py_RETURN_NONE;
+}
+
+PyObject *py_unreal_engine_update_ui(PyObject * self, PyObject * args)
+{
+	FEditorSupportDelegates::UpdateUI.Broadcast();
+	Py_RETURN_NONE;
+}
+
 PyObject *py_unreal_engine_editor_play_in_viewport(PyObject * self, PyObject * args)
 {
 
@@ -273,7 +286,7 @@ PyObject *py_unreal_engine_editor_play(PyObject * self, PyObject * args)
 	Py_END_ALLOW_THREADS;
 
 	Py_RETURN_NONE;
-}
+	}
 
 PyObject *py_unreal_engine_editor_select_actor(PyObject * self, PyObject * args)
 {
@@ -449,7 +462,7 @@ PyObject *py_unreal_engine_import_asset(PyObject * self, PyObject * args)
 	}
 
 	Py_RETURN_NONE;
-}
+	}
 
 PyObject *py_unreal_engine_editor_tick(PyObject * self, PyObject * args)
 {
@@ -742,7 +755,7 @@ PyObject *py_unreal_engine_rename_asset(PyObject * self, PyObject * args)
 #endif
 
 	Py_RETURN_NONE;
-}
+	}
 
 PyObject *py_unreal_engine_duplicate_asset(PyObject * self, PyObject * args)
 {
@@ -1849,10 +1862,11 @@ PyObject *py_unreal_engine_on_main_frame_creation_finished(PyObject * self, PyOb
 	py_delegate->SetPyCallable(py_callable);
 	IMainFrameModule& MainFrameModule = FModuleManager::LoadModuleChecked<IMainFrameModule>(TEXT("MainFrame"));
 	MainFrameModule.OnMainFrameCreationFinished().AddRaw(py_delegate, &FPythonSmartDelegate::PyFOnMainFrameCreationFinished);
-	
+
 	Py_RETURN_NONE;
 }
 
+
 PyObject *py_unreal_engine_create_material_instance(PyObject * self, PyObject * args)
 {
 
@@ -2060,7 +2074,7 @@ PyObject *py_unreal_engine_add_level_to_world(PyObject *self, PyObject * args)
 #endif
 
 	Py_RETURN_UOBJECT(level_streaming);
-}
+	}
 
 PyObject *py_unreal_engine_move_selected_actors_to_level(PyObject *self, PyObject * args)
 {

Source/UnrealEnginePython/Private/UEPyEditor.h

@@ -24,6 +24,9 @@ PyObject *py_unreal_engine_get_assets_by_class(PyObject *, PyObject *);
 PyObject *py_unreal_engine_get_assets_by_filter(PyObject *, PyObject *, PyObject *);
 PyObject *py_unreal_engine_set_fbx_import_option(PyObject *, PyObject *);
 
+PyObject *py_unreal_engine_redraw_all_viewports(PyObject *, PyObject *);
+PyObject *py_unreal_engine_update_ui(PyObject *, PyObject *);
+
 PyObject *py_unreal_engine_create_modal_save_asset_dialog(PyObject *, PyObject *);
 
 PyObject *py_unreal_engine_console_exec(PyObject *, PyObject * args);

Source/UnrealEnginePython/Private/UEPyModule.cpp

@@ -304,6 +304,9 @@ static PyMethodDef unreal_engine_methods[] = {
 
 	{ "add_asset_view_context_menu_extension", py_unreal_engine_add_asset_view_context_menu_extension, METH_VARARGS, "" },
 
+	{ "redraw_all_viewports", py_unreal_engine_redraw_all_viewports, METH_VARARGS, "" },
+	{ "update_ui", py_unreal_engine_update_ui, METH_VARARGS, "" },
+
 #pragma warning(suppress: 4191)
 	{ "create_detail_view", (PyCFunction)py_unreal_engine_create_detail_view, METH_VARARGS | METH_KEYWORDS, "" },
 #pragma warning(suppress: 4191)
@@ -463,7 +466,7 @@ static PyMethodDef unreal_engine_methods[] = {
 	{ "get_viewport_size", py_unreal_engine_get_viewport_size, METH_VARARGS, "" },
 	{ "get_resolution", py_unreal_engine_get_resolution, METH_VARARGS, "" },
 	{ "get_game_viewport_size", py_unreal_engine_get_game_viewport_size, METH_VARARGS, "" },
-		
+
 	{ "get_game_viewport_client", py_unreal_engine_get_game_viewport_client, METH_VARARGS, "" },
 #pragma warning(suppress: 4191)
 	{ "open_color_picker", (PyCFunction)py_unreal_engine_open_color_picker, METH_VARARGS | METH_KEYWORDS, "" },
@@ -1138,6 +1141,8 @@ static PyMethodDef ue_PyUObject_methods[] = {
 	{ "as_dict", (PyCFunction)py_ue_as_dict, METH_VARARGS, "" },
 
 	{ "get_cdo", (PyCFunction)py_ue_get_cdo, METH_VARARGS, "" },
+	{ "get_archetype", (PyCFunction)py_ue_get_archetype, METH_VARARGS, "" },
+	{ "get_archetype_instances", (PyCFunction)py_ue_get_archetype_instances, METH_VARARGS, "" },
 	{ "enum_values", (PyCFunction)py_ue_enum_values, METH_VARARGS, "" },
 	{ "enum_names", (PyCFunction)py_ue_enum_names, METH_VARARGS, "" },
 #if ENGINE_MINOR_VERSION >= 15
@@ -1295,8 +1300,35 @@ static int ue_PyUObject_setattro(ue_PyUObject *self, PyObject *attr_name, PyObje
 		UProperty *u_property = u_struct->FindPropertyByName(FName(UTF8_TO_TCHAR(attr)));
 		if (u_property)
 		{
+#if WITH_EDITOR
+			self->ue_object->PreEditChange(u_property);
+#endif
 			if (ue_py_convert_pyobject(value, u_property, (uint8*)self->ue_object, 0))
 			{
+#if WITH_EDITOR
+				FPropertyChangedEvent PropertyEvent(u_property, EPropertyChangeType::ValueSet);
+				self->ue_object->PostEditChangeProperty(PropertyEvent);
+
+				if (self->ue_object->HasAnyFlags(RF_ArchetypeObject | RF_ClassDefaultObject))
+				{
+					TArray<UObject *> Instances;
+					self->ue_object->GetArchetypeInstances(Instances);
+					for (UObject *Instance : Instances)
+					{
+						Instance->PreEditChange(u_property);
+						if (ue_py_convert_pyobject(value, u_property, (uint8*)Instance, 0))
+						{
+							FPropertyChangedEvent PropertyEvent(u_property, EPropertyChangeType::ValueSet);
+							Instance->PostEditChangeProperty(PropertyEvent);
+						}
+						else
+						{
+							PyErr_SetString(PyExc_ValueError, "invalid value for UProperty");
+							return -1;
+						}
+					}
+				}
+#endif
 				return 0;
 			}
 			PyErr_SetString(PyExc_ValueError, "invalid value for UProperty");
@@ -1726,7 +1758,9 @@ void unreal_engine_init_py_module()
 	PyDict_SetItemString(unreal_engine_dict, "RF_MARKASNATIVE", PyLong_FromUnsignedLongLong((uint64)RF_MarkAsNative));
 	PyDict_SetItemString(unreal_engine_dict, "RF_TRANSACTIONAL", PyLong_FromUnsignedLongLong((uint64)RF_Transactional));
 	PyDict_SetItemString(unreal_engine_dict, "RF_CLASSDEFAULTOBJECT", PyLong_FromUnsignedLongLong((uint64)RF_ClassDefaultObject));
+	PyDict_SetItemString(unreal_engine_dict, "RF_CLASS_DEFAULT_OBJECT", PyLong_FromUnsignedLongLong((uint64)RF_ClassDefaultObject));
 	PyDict_SetItemString(unreal_engine_dict, "RF_ARCHETYPEOBJECT", PyLong_FromUnsignedLongLong((uint64)RF_ArchetypeObject));
+	PyDict_SetItemString(unreal_engine_dict, "RF_ARCHETYPE_OBJECT", PyLong_FromUnsignedLongLong((uint64)RF_ArchetypeObject));
 	PyDict_SetItemString(unreal_engine_dict, "RF_TRANSIENT", PyLong_FromUnsignedLongLong((uint64)RF_Transient));
 	PyDict_SetItemString(unreal_engine_dict, "RF_MARKASROOTSET", PyLong_FromUnsignedLongLong((uint64)RF_MarkAsRootSet));
 	PyDict_SetItemString(unreal_engine_dict, "RF_TAGGARBAGETEMP", PyLong_FromUnsignedLongLong((uint64)RF_TagGarbageTemp));
@@ -1814,10 +1848,10 @@ ue_PyUObject *ue_get_python_uobject(UObject *ue_obj)
 		UE_LOG(LogPython, Warning, TEXT("CREATED UPyObject at %p for %p %s"), ue_py_object, ue_obj, *ue_obj->GetName());
 #endif
 		return ue_py_object;
-	}
+		}
 	return ret;
 
-}
+	}
 
 ue_PyUObject *ue_get_python_uobject_inc(UObject *ue_obj)
 {
@@ -1905,7 +1939,7 @@ void unreal_engine_py_log_error()
 	}
 
 	PyErr_Clear();
-}
+	}
 
 // retrieve a UWorld from a generic UObject (if possible)
 UWorld *ue_get_uworld(ue_PyUObject *py_obj)
@@ -2841,8 +2875,8 @@ PyObject *py_ue_ufunction_call(UFunction *u_function, UObject *u_obj, PyObject *
 #endif
 			}
 #endif
+			}
 		}
-	}
 
 
 	Py_ssize_t tuple_len = PyTuple_Size(args);
@@ -2955,7 +2989,7 @@ PyObject *py_ue_ufunction_call(UFunction *u_function, UObject *u_obj, PyObject *
 		return ret;
 
 	Py_RETURN_NONE;
-}
+	}
 
 PyObject *ue_bind_pyevent(ue_PyUObject *u_obj, FString event_name, PyObject *py_callable, bool fail_on_wrong_property)
 {
@@ -3015,8 +3049,8 @@ UFunction *unreal_engine_add_function(UClass *u_class, char *name, PyObject *py_
 		{
 			UE_LOG(LogPython, Error, TEXT("function %s is already registered"), UTF8_TO_TCHAR(name));
 			return nullptr;
-		}
 	}
+}
 
 	UPythonFunction *function = NewObject<UPythonFunction>(u_class, UTF8_TO_TCHAR(name), RF_Public | RF_Transient | RF_MarkAsNative);
 	function->SetPyCallable(py_callable);
@@ -3337,7 +3371,7 @@ UFunction *unreal_engine_add_function(UClass *u_class, char *name, PyObject *py_
 #endif
 
 	return function;
-}
+	}
 
 FGuid *ue_py_check_fguid(PyObject *py_obj)
 {

Source/UnrealEnginePython/Private/UObject/UEPyObject.cpp

@@ -1894,16 +1894,47 @@ PyObject *py_ue_get_cdo(ue_PyUObject * self, PyObject * args)
 
 	ue_py_check(self);
 
-	if (!self->ue_object->IsA<UClass>())
+	UClass *u_class = ue_py_check_type<UClass>(self);
+	if (!u_class)
 	{
 		return PyErr_Format(PyExc_Exception, "uobject is not a UClass");
 	}
 
-	UClass *u_class = (UClass *)self->ue_object;
-
 	Py_RETURN_UOBJECT(u_class->GetDefaultObject());
 }
 
+PyObject *py_ue_get_archetype(ue_PyUObject * self, PyObject * args)
+{
+
+	ue_py_check(self);
+
+	UObject *Archetype = self->ue_object->GetArchetype();
+
+	if (!Archetype)
+		return PyErr_Format(PyExc_Exception, "uobject has no archetype");
+
+	Py_RETURN_UOBJECT(Archetype);
+}
+
+PyObject *py_ue_get_archetype_instances(ue_PyUObject * self, PyObject * args)
+{
+
+	ue_py_check(self);
+
+	TArray<UObject *> Instances;
+
+	self->ue_object->GetArchetypeInstances(Instances);
+
+	PyObject *py_list = PyList_New(0);
+
+	for (UObject *Instance : Instances)
+	{
+		PyList_Append(py_list, (PyObject *)ue_get_python_uobject(Instance));
+	}
+
+	return py_list;
+}
+
 
 #if WITH_EDITOR
 PyObject *py_ue_save_package(ue_PyUObject * self, PyObject * args)

Source/UnrealEnginePython/Private/UObject/UEPyObject.h

@@ -41,6 +41,8 @@ PyObject *py_ue_is_owned(ue_PyUObject *, PyObject *);
 PyObject *py_ue_save_config(ue_PyUObject *, PyObject *);
 
 PyObject *py_ue_get_cdo(ue_PyUObject *, PyObject *);
+PyObject *py_ue_get_archetype(ue_PyUObject *, PyObject *);
+PyObject *py_ue_get_archetype_instances(ue_PyUObject *, PyObject *);
 PyObject *py_ue_enum_values(ue_PyUObject *, PyObject *);
 PyObject *py_ue_enum_names(ue_PyUObject *, PyObject *);
 #if ENGINE_MINOR_VERSION >= 15