added foliage api, #278

Author: Roberto De Ioris

Source/UnrealEnginePython/Private/UEPyModule.cpp

@@ -713,6 +713,12 @@ static PyMethodDef ue_PyUObject_methods[] = {
 	{ "get_current_level", (PyCFunction)py_ue_get_current_level, METH_VARARGS, "" },
 	{ "set_current_level", (PyCFunction)py_ue_set_current_level, METH_VARARGS, "" },
 
+#if WITH_EDITOR
+	{ "add_foliage_asset", (PyCFunction)py_ue_add_foliage_asset, METH_VARARGS, "" },
+#endif
+	{ "get_instanced_foliage_actor_for_current_level", (PyCFunction)py_ue_get_instanced_foliage_actor_for_current_level, METH_VARARGS, "" },
+
+
 	{ "add_actor_component", (PyCFunction)py_ue_add_actor_component, METH_VARARGS, "" },
 	{ "add_actor_root_component", (PyCFunction)py_ue_add_actor_root_component, METH_VARARGS, "" },
 	{ "get_actor_component_by_type", (PyCFunction)py_ue_get_actor_component_by_type, METH_VARARGS, "" },

Source/UnrealEnginePython/Private/UObject/UEPyWorld.cpp

@@ -1,6 +1,8 @@
 #include "UnrealEnginePythonPrivatePCH.h"
 
 #include "Runtime/Engine/Classes/Kismet/KismetSystemLibrary.h"
+#include "Runtime/Foliage/Public/FoliageType.h"
+#include "Runtime/Foliage/Public/InstancedFoliageActor.h"
 
 PyObject *py_ue_world_exec(ue_PyUObject *self, PyObject * args)
 {
@@ -322,3 +324,59 @@ PyObject *py_ue_set_current_level(ue_PyUObject *self, PyObject * args)
 	Py_RETURN_FALSE;
 }
 
+PyObject *py_ue_get_instanced_foliage_actor_for_current_level(ue_PyUObject *self, PyObject * args)
+{
+	ue_py_check(self);
+
+	UWorld *world = ue_get_uworld(self);
+	if (!world)
+		return PyErr_Format(PyExc_Exception, "unable to retrieve UWorld from uobject");
+
+	Py_RETURN_UOBJECT(AInstancedFoliageActor::GetInstancedFoliageActorForCurrentLevel(world, true));
+}
+
+#if WITH_EDITOR
+PyObject *py_ue_add_foliage_asset(ue_PyUObject *self, PyObject * args)
+{
+
+	ue_py_check(self);
+
+	PyObject *py_uobject;
+
+	if (!PyArg_ParseTuple(args, "O:add_foliage_asset", &py_uobject))
+	{
+		return nullptr;
+	}
+
+	UWorld *world = ue_get_uworld(self);
+	if (!world)
+		return PyErr_Format(PyExc_Exception, "unable to retrieve UWorld from uobject");
+
+	UObject *u_object = ue_py_check_type<UObject>(py_uobject);
+	if (!u_object)
+		return PyErr_Format(PyExc_Exception, "argument is not a UObject");
+
+	UFoliageType *foliage_type = nullptr;
+
+	AInstancedFoliageActor *ifa = AInstancedFoliageActor::GetInstancedFoliageActorForCurrentLevel(world, true);
+	if (u_object->IsA<UStaticMesh>())
+	{
+		foliage_type = ifa->GetLocalFoliageTypeForMesh((UStaticMesh *)u_object);
+		if (!foliage_type)
+		{
+			ifa->AddMesh((UStaticMesh *)u_object, &foliage_type);
+		}
+	}
+	else if (u_object->IsA<UFoliageType>())
+	{
+		ifa->AddFoliageType((UFoliageType *)u_object);
+	}
+
+	if (!foliage_type)
+		return PyErr_Format(PyExc_Exception, "unable to add foliage asset");
+
+	Py_RETURN_UOBJECT(foliage_type);
+
+}
+#endif
+

Source/UnrealEnginePython/Private/UObject/UEPyWorld.h

@@ -28,4 +28,9 @@ PyObject *py_ue_get_game_viewport(ue_PyUObject *, PyObject *);
 PyObject *py_ue_get_current_level(ue_PyUObject *, PyObject *);
 PyObject *py_ue_set_current_level(ue_PyUObject *, PyObject *);
 
-PyObject *py_ue_get_world_type(ue_PyUObject *, PyObject *);
+PyObject *py_ue_get_world_type(ue_PyUObject *, PyObject *);
+
+PyObject *py_ue_get_instanced_foliage_actor_for_current_level(ue_PyUObject *, PyObject *);
+#if WITH_EDITOR
+PyObject *py_ue_add_foliage_asset(ue_PyUObject *, PyObject *);
+#endif

Source/UnrealEnginePython/UnrealEnginePython.Build.cs

@@ -124,7 +124,8 @@ public UnrealEnginePython(TargetInfo Target)
                 "Voice",
                 "RenderCore",
                 "MovieSceneCapture",
-                "Landscape"
+                "Landscape",
+                "Foliage"
 				// ... add private dependencies that you statically link with here ...
 			}
             );