Repository navigation
Expand file tree
/
Copy pathUEPyFoliage.cpp
More file actions
134 lines (96 loc) · 3.5 KB
/
Copy pathUEPyFoliage.cpp
File metadata and controls
134 lines (96 loc) · 3.5 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
#include "UEPyFoliage.h"
#include "Runtime/Foliage/Public/FoliageType.h"
#include "Runtime/Foliage/Public/InstancedFoliageActor.h"
#include "Wrappers/UEPyFFoliageInstance.h"
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));
}
PyObject *py_ue_get_instanced_foliage_actor_for_level(ue_PyUObject *self, PyObject * args)
{
ue_py_check(self);
ULevel *level = ue_py_check_type<ULevel>(self);
if (!level)
return PyErr_Format(PyExc_Exception, "uobject is not a ULevel");
Py_RETURN_UOBJECT(AInstancedFoliageActor::GetInstancedFoliageActorForLevel(level, true));
}
PyObject *py_ue_get_foliage_types(ue_PyUObject *self, PyObject * args)
{
ue_py_check(self);
AInstancedFoliageActor *foliage_actor = ue_py_check_type<AInstancedFoliageActor>(self);
if (!foliage_actor)
return PyErr_Format(PyExc_Exception, "uobject is not a AInstancedFoliageActor");
PyObject *py_list = PyList_New(0);
TArray<UFoliageType *> FoliageTypes;
foliage_actor->FoliageMeshes.GetKeys(FoliageTypes);
for (UFoliageType *FoliageType : FoliageTypes)
{
PyList_Append(py_list, (PyObject*)ue_get_python_uobject(FoliageType));
}
return py_list;
}
#if WITH_EDITOR
PyObject *py_ue_get_foliage_instances(ue_PyUObject *self, PyObject * args)
{
ue_py_check(self);
PyObject *py_foliage_type;
if (!PyArg_ParseTuple(args, "O:get_foliage_instances", &py_foliage_type))
{
return nullptr;
}
AInstancedFoliageActor *foliage_actor = ue_py_check_type<AInstancedFoliageActor>(self);
if (!foliage_actor)
return PyErr_Format(PyExc_Exception, "uobject is not a AInstancedFoliageActor");
UFoliageType *foliage_type = ue_py_check_type<UFoliageType>(py_foliage_type);
if (!foliage_type)
return PyErr_Format(PyExc_Exception, "argument is not a UFoliageType");
if (!foliage_actor->FoliageMeshes.Contains(foliage_type))
{
return PyErr_Format(PyExc_Exception, "specified UFoliageType not found in AInstancedFoliageActor");
}
FFoliageMeshInfo& info = foliage_actor->FoliageMeshes[foliage_type].Get();
PyObject *py_list = PyList_New(0);
for (FFoliageInstance& instance : info.Instances)
{
PyList_Append(py_list, py_ue_new_ffoliage_instance(instance));
}
return py_list;
}
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>())
{
foliage_type = (UFoliageType *)u_object;
ifa->AddFoliageType(foliage_type);
}
if (!foliage_type)
return PyErr_Format(PyExc_Exception, "unable to add foliage asset");
Py_RETURN_UOBJECT(foliage_type);
}
#endif