preliminary support for animation editors

Author: Roberto De Ioris

Source/UnrealEnginePython/Private/Slate/UEPySlate.cpp

@@ -4,6 +4,9 @@
 #if WITH_EDITOR
 #include "LevelEditor.h"
 #include "Editor/UnrealEd/Public/Toolkits/AssetEditorToolkit.h"
+#include "Editor/Persona/Public/PersonaModule.h"
+#include "Editor/AnimationEditor/Public/IAnimationEditorModule.h"
+#include "Editor/StaticMeshEditor/Public/StaticMeshEditorModule.h"
 #endif
 
 #include "Runtime/Slate/Public/Framework/Commands/UICommandList.h"
@@ -702,12 +705,37 @@ PyObject *py_unreal_engine_add_menu_extension(PyObject * self, PyObject * args)
 	PyObject *py_callable;
 
 	char *hook = (char *)"WindowLayout";
+	char *module = (char*)"LevelEditor";
 
-	if (!PyArg_ParseTuple(args, "sO|s:add_menu_extension", &command_name, &py_callable, &hook)) {
+	if (!PyArg_ParseTuple(args, "sO|ss:add_menu_extension", &command_name, &py_callable, &hook, &module)) {
 		return NULL;
 	}
 
-	FLevelEditorModule &ExtensibleModule = FModuleManager::LoadModuleChecked<FLevelEditorModule>("LevelEditor");
+	if (!FModuleManager::Get().ModuleExists(UTF8_TO_TCHAR(module)))
+		return PyErr_Format(PyExc_Exception, "module %s does not exist", module);
+
+	// unfortunately we need to manually check for module names :(
+	IHasMenuExtensibility *menu_extension_interface = nullptr;
+
+	if (!strcmp(module, (char *)"LevelEditor")) {
+		FLevelEditorModule &Module = FModuleManager::LoadModuleChecked<FLevelEditorModule>(module);
+		menu_extension_interface = (IHasMenuExtensibility *)&Module;
+	}
+	else if (!strcmp(module, (char *)"Persona")) {
+		FPersonaModule &Module = FModuleManager::LoadModuleChecked<FPersonaModule>(module);
+		menu_extension_interface = (IHasMenuExtensibility *)&Module;
+	}
+	else if (!strcmp(module, (char *)"AnimationEditor")) {
+		IAnimationEditorModule &Module = FModuleManager::LoadModuleChecked<IAnimationEditorModule>(module);
+		menu_extension_interface = (IHasMenuExtensibility *)&Module;
+	}
+	else if (!strcmp(module, (char *)"StaticMeshEditor")) {
+		IStaticMeshEditorModule &Module = FModuleManager::LoadModuleChecked<IStaticMeshEditorModule>(module);
+		menu_extension_interface = (IHasMenuExtensibility *)&Module;
+	}
+
+	if (!menu_extension_interface)
+		return PyErr_Format(PyExc_Exception, "module %s is not supported", module);
 
 	if (!PyCallable_Check(py_callable))
 		return PyErr_Format(PyExc_Exception, "argument is not callable");
@@ -721,10 +749,9 @@ PyObject *py_unreal_engine_add_menu_extension(PyObject * self, PyObject * args)
 	TSharedRef<FExtender> extender = MakeShareable(new FExtender());
 	extender->AddMenuExtension(hook, EExtensionHook::After, commands->Get().GetCommands(), FMenuExtensionDelegate::CreateRaw(&commands->Get(), &FPythonSlateCommands::MenuBuilder));
 
-	ExtensibleModule.GetMenuExtensibilityManager()->AddExtender(extender);
+	menu_extension_interface->GetMenuExtensibilityManager()->AddExtender(extender);
 
-	Py_INCREF(Py_None);
-	return Py_None;
+	Py_RETURN_NONE;
 }
 
 

Source/UnrealEnginePython/UnrealEnginePython.Build.cs

@@ -149,7 +149,8 @@ public UnrealEnginePython(TargetInfo Target)
                 "RawMesh",
                 "DesktopWidgets",
                 "EditorWidgets",
-                "FBX"
+                "FBX",
+                "Persona"
             });
         }