Repository navigation
Expand file tree
/
Copy pathUEPySPythonTreeView.cpp
More file actions
131 lines (108 loc) · 4.7 KB
/
Copy pathUEPySPythonTreeView.cpp
File metadata and controls
131 lines (108 loc) · 4.7 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
#include "UnrealEnginePythonPrivatePCH.h"
#include "Runtime/Slate/Public/Widgets/Views/STreeView.h"
#include "UEPySPythonTreeView.h"
#define sw_python_tree_view StaticCastSharedRef<SPythonTreeView>(self->s_tree_view.s_list_view.s_table_view_base.s_compound_widget.s_widget.s_widget)
static PyObject *py_ue_spython_tree_view_set_item_expansion(ue_PySPythonTreeView *self, PyObject * args)
{
PyObject *py_item;
PyObject *py_bool;
if (!PyArg_ParseTuple(args, "OO:set_item_expansion", &py_item, &py_bool))
{
return nullptr;
}
sw_python_tree_view->SetPythonItemExpansion(py_item, PyObject_IsTrue(py_bool) ? true : false);
Py_RETURN_NONE;
}
void SPythonTreeView::SetPythonItemExpansion(PyObject *item, bool InShouldExpandItem)
{
for (TSharedPtr<struct FPythonItem> PythonItem : *ItemsSource)
{
if (PythonItem->py_object == item)
{
SetItemExpansion(PythonItem, InShouldExpandItem);
}
}
}
static PyMethodDef ue_PySPythonTreeView_methods[] = {
{ "set_item_expansion", (PyCFunction)py_ue_spython_tree_view_set_item_expansion, METH_VARARGS, "" },
{ NULL } /* Sentinel */
};
PyTypeObject ue_PySPythonTreeViewType = {
PyVarObject_HEAD_INIT(NULL, 0)
"unreal_engine.SPythonTreeView", /* tp_name */
sizeof(ue_PySPythonTreeView), /* tp_basicsize */
0, /* tp_itemsize */
0, /* tp_dealloc */
0, /* tp_print */
0, /* tp_getattr */
0, /* tp_setattr */
0, /* tp_reserved */
0, /* tp_repr */
0, /* tp_as_number */
0, /* tp_as_sequence */
0, /* tp_as_mapping */
0, /* tp_hash */
0, /* tp_call */
0, /* tp_str */
0, /* tp_getattro */
0, /* tp_setattro */
0, /* tp_as_buffer */
Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
"Unreal Engine SPythonTreeView", /* tp_doc */
0, /* tp_traverse */
0, /* tp_clear */
0, /* tp_richcompare */
0, /* tp_weaklistoffset */
0, /* tp_iter */
0, /* tp_iternext */
ue_PySPythonTreeView_methods, /* tp_methods */
};
static int ue_py_spython_tree_view_init(ue_PySPythonTreeView *self, PyObject *args, PyObject *kwargs)
{
ue_py_slate_setup_farguments(SPythonTreeView);
// first of all check for values
PyObject *values = ue_py_dict_get_item(kwargs, "tree_items_source");
if (!values)
{
PyErr_SetString(PyExc_Exception, "you must specify tree items");
return -1;
}
values = PyObject_GetIter(values);
if (!values)
{
PyErr_SetString(PyExc_Exception, "values field is not an iterable");
return -1;
}
TArray<TSharedPtr<FPythonItem>> *items = new TArray<TSharedPtr<FPythonItem>>();
while (PyObject *item = PyIter_Next(values))
{
Py_INCREF(item);
// keep track of items
self->s_tree_view.s_list_view.s_table_view_base.s_compound_widget.s_widget.py_refs.Add(item);
items->Add(TSharedPtr<FPythonItem>(new FPythonItem(item)));
}
Py_DECREF(values);
arguments.TreeItemsSource(items);
ue_py_slate_farguments_optional_enum("allow_overscroll", AllowOverscroll, EAllowOverscroll);
ue_py_slate_farguments_optional_bool("clear_selection_on_click", ClearSelectionOnClick);
ue_py_slate_farguments_optional_enum("consume_mouse_wheel", ConsumeMouseWheel, EConsumeMouseWheel);
ue_py_slate_farguments_float("item_height", ItemHeight);
ue_py_slate_farguments_event("on_generate_row", OnGenerateRow, TSlateDelegates<TSharedPtr<FPythonItem>>::FOnGenerateRow, GenerateRow);
ue_py_slate_farguments_event("on_selection_changed", OnSelectionChanged, TSlateDelegates<TSharedPtr<FPythonItem>>::FOnSelectionChanged, OnSelectionChanged);
ue_py_slate_farguments_enum("selection_mode", SelectionMode, ESelectionMode::Type);
#if ENGINE_MINOR_VERSION > 12
ue_py_slate_farguments_optional_float("wheel_scroll_multiplier", WheelScrollMultiplier);
#endif
ue_py_slate_farguments_event("on_get_children", OnGetChildren, TSlateDelegates<TSharedPtr<FPythonItem>>::FOnGetChildren, GetChildren);
ue_py_snew(SPythonTreeView, s_tree_view.s_list_view.s_table_view_base.s_compound_widget.s_widget);
return 0;
}
void ue_python_init_spython_tree_view(PyObject *ue_module)
{
ue_PySPythonTreeViewType.tp_init = (initproc)ue_py_spython_tree_view_init;
ue_PySPythonTreeViewType.tp_base = &ue_PySTreeViewType;
if (PyType_Ready(&ue_PySPythonTreeViewType) < 0)
return;
Py_INCREF(&ue_PySPythonTreeViewType);
PyModule_AddObject(ue_module, "SPythonTreeView", (PyObject *)&ue_PySPythonTreeViewType);
}