fixed blueprint component management, add blueprntfunctionlibrary infrastructure

Author: Roberto De Ioris

Source/UnrealEnginePython/Private/UEPyEditor.cpp

@@ -701,30 +701,34 @@ PyObject *py_unreal_engine_add_component_to_blueprint(PyObject * self, PyObject
 	}
 
 	USCS_Node *parentNode = nullptr;
-	if (parentName)
-	{
-		FString strParentName = UTF8_TO_TCHAR(parentName);
-		if (strParentName.IsEmpty())
+
+	if (component_class->IsChildOf<USceneComponent>()) {
+
+		if (parentName)
 		{
-			parentNode = bp->SimpleConstructionScript->GetDefaultSceneRootNode();
+			FString strParentName = UTF8_TO_TCHAR(parentName);
+			if (strParentName.IsEmpty())
+			{
+				parentNode = bp->SimpleConstructionScript->GetDefaultSceneRootNode();
+			}
+			else
+			{
+				parentNode = bp->SimpleConstructionScript->FindSCSNode(UTF8_TO_TCHAR(parentName));
+			}
 		}
 		else
 		{
-			parentNode = bp->SimpleConstructionScript->FindSCSNode(UTF8_TO_TCHAR(parentName));
+			parentNode = bp->SimpleConstructionScript->GetDefaultSceneRootNode();
 		}
 	}
-	else
-	{
-		parentNode = bp->SimpleConstructionScript->GetDefaultSceneRootNode();
-	}
-
 
 	if (parentNode) {
 		parentNode->AddChildNode(node);
 	}
 	else {
 		bp->SimpleConstructionScript->AddNode(node);
 	}
+
 	FBlueprintEditorUtils::MarkBlueprintAsStructurallyModified(bp);
 
 	ue_PyUObject *ret = ue_get_python_wrapper(node->ComponentTemplate);
@@ -769,6 +773,35 @@ PyObject *py_unreal_engine_blueprint_add_member_variable(PyObject * self, PyObje
 	return Py_False;
 }
 
+PyObject *py_unreal_engine_blueprint_set_variable_visibility(PyObject * self, PyObject * args) {
+
+	PyObject *py_blueprint;
+	char *name;
+	PyObject *visibility;
+	if (!PyArg_ParseTuple(args, "OsO:blueprint_set_variable_visibility", &py_blueprint, &name, &visibility)) {
+		return NULL;
+	}
+
+	if (!ue_is_pyuobject(py_blueprint)) {
+		return PyErr_Format(PyExc_Exception, "argument is not a UObject");
+	}
+
+	ue_PyUObject *py_obj = (ue_PyUObject *)py_blueprint;
+	if (!py_obj->ue_object->IsA<UBlueprint>())
+		return PyErr_Format(PyExc_Exception, "uobject is not a UBlueprint");
+	UBlueprint *bp = (UBlueprint *)py_obj->ue_object;
+
+	bool visible = false;
+	if (PyObject_IsTrue(visibility)) {
+		visible = true;
+	}
+
+	FBlueprintEditorUtils::SetBlueprintOnlyEditableFlag(bp, FName(UTF8_TO_TCHAR(name)), visible);
+
+	Py_INCREF(Py_None);
+	return Py_None;
+}
+
 PyObject *py_unreal_engine_blueprint_add_new_timeline(PyObject * self, PyObject * args) {
 
 	PyObject *py_blueprint;
@@ -963,7 +996,7 @@ PyObject *py_unreal_engine_move_selected_actors_to_level(PyObject *self, PyObjec
 	ULevel *level = (ULevel *)py_obj_level->ue_object;
 
 	GEditor->MoveSelectedActorsToLevel(level);
-	
+
 	Py_INCREF(Py_None);
 	return Py_None;
 }

Source/UnrealEnginePython/Private/UEPyEditor.h

@@ -28,6 +28,7 @@ PyObject *py_unreal_engine_add_component_to_blueprint(PyObject *, PyObject *);
 
 PyObject *py_unreal_engine_blueprint_add_member_variable(PyObject *, PyObject *);
 PyObject *py_unreal_engine_blueprint_add_new_timeline(PyObject *, PyObject *);
+PyObject *py_unreal_engine_blueprint_set_variable_visibility(PyObject *, PyObject *);
 
 PyObject *py_unreal_engine_editor_play(PyObject *, PyObject *);
 PyObject *py_unreal_engine_editor_on_asset_post_import(PyObject *, PyObject *);

Source/UnrealEnginePython/Private/UEPyModule.cpp

@@ -155,6 +155,7 @@ static PyMethodDef unreal_engine_methods[] = {
 	{ "compile_blueprint", py_unreal_engine_compile_blueprint, METH_VARARGS, "" },
 	{ "blueprint_add_member_variable", py_unreal_engine_blueprint_add_member_variable, METH_VARARGS, "" },
 	{ "blueprint_add_new_timeline", py_unreal_engine_blueprint_add_new_timeline, METH_VARARGS, "" },
+	{ "blueprint_set_variable_visibility", py_unreal_engine_blueprint_set_variable_visibility, METH_VARARGS, "" },
 	{ "add_component_to_blueprint", py_unreal_engine_add_component_to_blueprint, METH_VARARGS, "" },
     { "create_material_instance", py_unreal_engine_create_material_instance, METH_VARARGS, "" },
 	{ "message_dialog_open", py_unreal_engine_message_dialog_open, METH_VARARGS, "" },

Source/UnrealEnginePython/Public/PythonBlueprintFunctionLibrary.cpp

@@ -0,0 +1,8 @@
+#include "UnrealEnginePythonPrivatePCH.h"
+
+#include "PythonBlueprintFunctionLibrary.h"
+
+void UPythonBlueprintFunctionLibrary::ExecutePythonScript(FString script) {
+	FUnrealEnginePythonModule &PythonModule = FModuleManager::GetModuleChecked<FUnrealEnginePythonModule>("UnrealEnginePython");
+	PythonModule.RunFile(TCHAR_TO_UTF8(*script));
+}

Source/UnrealEnginePython/Public/PythonBlueprintFunctionLibrary.h

@@ -0,0 +1,17 @@
+#pragma once
+
+
+#include "PythonBlueprintFunctionLibrary.generated.h"
+
+
+
+UCLASS()
+class UPythonBlueprintFunctionLibrary : public UBlueprintFunctionLibrary
+{
+	GENERATED_BODY()
+
+public:
+	UFUNCTION(BlueprintCallable, Category = "Python")
+	static void ExecutePythonScript(FString script);
+};
+