added (broken) graph_add_node_variable_get, graph_add_node_variable_set

Author: Roberto De Ioris

Source/UnrealEnginePython/Private/UEPyEdGraph.cpp

@@ -6,6 +6,9 @@
 #include "Editor/BlueprintGraph/Classes/K2Node_CallFunction.h"
 #include "Editor/BlueprintGraph/Classes/EdGraphSchema_K2.h"
 #include "Editor/BlueprintGraph/Classes/K2Node_CustomEvent.h"
+#include "Editor/BlueprintGraph/Classes/K2Node_VariableGet.h"
+#include "Editor/BlueprintGraph/Classes/K2Node_VariableSet.h"
+#include "Editor/UnrealEd/Public/Kismet2/BlueprintEditorUtils.h"
 
 PyObject *py_ue_graph_add_node_call_function(ue_PyUObject * self, PyObject * args) {
 
@@ -98,5 +101,105 @@ PyObject *py_ue_graph_add_node_custom_event(ue_PyUObject * self, PyObject * args
 	Py_INCREF(ret);
 	return ret;
 
+}
+
+PyObject *py_ue_graph_add_node_variable_get(ue_PyUObject * self, PyObject * args) {
+
+	ue_py_check(self);
+
+	char *name = nullptr;
+	PyObject *py_struct = nullptr;
+	int x = 0;
+	int y = 0;
+	if (!PyArg_ParseTuple(args, "s|Oii:graph_add_node_variable_get", &name, &py_struct, &x, &y)) {
+		return NULL;
+	}
+
+	if (!self->ue_object->IsA<UEdGraph>()) {
+		return PyErr_Format(PyExc_Exception, "uobject is not a UEdGraph");
+	}
+
+	UEdGraph *graph = (UEdGraph *)self->ue_object;
+	UStruct *u_struct = nullptr;
+
+	if (py_struct && py_struct != Py_None) {
+		if (!ue_is_pyuobject(py_struct)) {
+			return PyErr_Format(PyExc_Exception, "argument is not a UObject");
+		}
+		ue_PyUObject *ue_py_struct = (ue_PyUObject *)py_struct;
+		if (!ue_py_struct->ue_object->IsA<UStruct>()) {
+			return PyErr_Format(PyExc_Exception, "argument is not a UStruct");
+		}
+		u_struct = (UStruct *)ue_py_struct->ue_object;
+	}
+
+	UK2Node_VariableGet *node = NewObject<UK2Node_VariableGet>(graph);
+
+	node->CreateNewGuid();
+	node->PostPlacedNewNode();
+	node->SetFlags(RF_Transactional);
+	node->AllocateDefaultPins();
+	node->NodePosX = x;
+	node->NodePosY = y;
+	UEdGraphSchema_K2::ConfigureVarNode(node, FName(UTF8_TO_TCHAR(name)), u_struct, FBlueprintEditorUtils::FindBlueprintForGraph(graph));
+	UEdGraphSchema_K2::SetNodeMetaData(node, FNodeMetadata::DefaultGraphNode);
+	graph->AddNode(node);
+
+	PyObject *ret = (PyObject *)ue_get_python_wrapper(node);
+	if (!ret)
+		return PyErr_Format(PyExc_Exception, "uobject is in invalid state");
+	Py_INCREF(ret);
+	return ret;
+
+}
+
+PyObject *py_ue_graph_add_node_variable_set(ue_PyUObject * self, PyObject * args) {
+
+	ue_py_check(self);
+
+	char *name = nullptr;
+	PyObject *py_struct = nullptr;
+	int x = 0;
+	int y = 0;
+	if (!PyArg_ParseTuple(args, "s|Oii:graph_add_node_variable_set", &name, &py_struct, &x, &y)) {
+		return NULL;
+	}
+
+	if (!self->ue_object->IsA<UEdGraph>()) {
+		return PyErr_Format(PyExc_Exception, "uobject is not a UEdGraph");
+	}
+
+	UEdGraph *graph = (UEdGraph *)self->ue_object;
+	UStruct *u_struct = nullptr;
+
+	if (py_struct && py_struct != Py_None) {
+		if (!ue_is_pyuobject(py_struct)) {
+			return PyErr_Format(PyExc_Exception, "argument is not a UObject");
+		}
+		ue_PyUObject *ue_py_struct = (ue_PyUObject *)py_struct;
+		if (!ue_py_struct->ue_object->IsA<UStruct>()) {
+			return PyErr_Format(PyExc_Exception, "argument is not a UStruct");
+		}
+		u_struct = (UStruct *)ue_py_struct->ue_object;
+	}
+
+	UK2Node_VariableSet *node = NewObject<UK2Node_VariableSet>(graph);
+
+	node->CreateNewGuid();
+	node->PostPlacedNewNode();
+	node->SetFlags(RF_Transactional);
+	node->AllocateDefaultPins();
+	node->NodePosX = x;
+	node->NodePosY = y;
+	UEdGraphSchema_K2::ConfigureVarNode(node, FName(UTF8_TO_TCHAR(name)), u_struct, FBlueprintEditorUtils::FindBlueprintForGraph(graph));
+	UEdGraphSchema_K2::SetNodeMetaData(node, FNodeMetadata::DefaultGraphNode);
+	graph->AddNode(node);
+
+	PyObject *ret = (PyObject *)ue_get_python_wrapper(node);
+	if (!ret)
+		return PyErr_Format(PyExc_Exception, "uobject is in invalid state");
+	Py_INCREF(ret);
+	return ret;
+
 }
 #endif

Source/UnrealEnginePython/Private/UEPyEdGraph.h

@@ -7,4 +7,6 @@
 #if WITH_EDITOR
 PyObject *py_ue_graph_add_node_call_function(ue_PyUObject *, PyObject *);
 PyObject *py_ue_graph_add_node_custom_event(ue_PyUObject *, PyObject *);
+PyObject *py_ue_graph_add_node_variable_get(ue_PyUObject *, PyObject *);
+PyObject *py_ue_graph_add_node_variable_set(ue_PyUObject *, PyObject *);
 #endif

Source/UnrealEnginePython/Private/UEPyModule.cpp

@@ -300,6 +300,8 @@ static PyMethodDef ue_PyUObject_methods[] = {
 
 	{ "graph_add_node_call_function", (PyCFunction)py_ue_graph_add_node_call_function, METH_VARARGS, "" },
 	{ "graph_add_node_custom_event", (PyCFunction)py_ue_graph_add_node_custom_event, METH_VARARGS, "" },
+	{ "graph_add_node_variable_get", (PyCFunction)py_ue_graph_add_node_variable_get, METH_VARARGS, "" },
+	{ "graph_add_node_variable_set", (PyCFunction)py_ue_graph_add_node_variable_set, METH_VARARGS, "" },
 #endif
 
 	{ "is_rooted", (PyCFunction)py_ue_is_rooted, METH_VARARGS, "" },
@@ -528,7 +530,7 @@ void ue_pydelegates_cleanup(ue_PyUObject *self) {
 			UE_LOG(LogPython, Warning, TEXT("Removing UPythonDelegate %p from ue_PyUObject %p mapped to UObject %p"), py_delegate, self, self->ue_object);
 #endif
 			py_delegate->RemoveFromRoot();
-		}
+}
 }
 	self->python_delegates_gc->clear();
 	delete self->python_delegates_gc;