
#include "UnrealEnginePythonPrivatePCH.h"

#include "Runtime/Slate/Public/Widgets/Views/SListView.h"

#include "UEPySPythonListView.h"


#define sw_python_list_view StaticCastSharedRef<SPythonListView>(self->s_list_view.s_table_view_base.s_compound_widget.s_widget.s_widget)

static PyObject *py_ue_spython_list_view_get_selected_items(ue_PySPythonListView *self, PyObject * args) {

	PyObject *py_list = PyList_New(0);

	TArray<TSharedPtr<FPythonItem>> items = sw_python_list_view->GetSelectedItems();

	for (auto item : items) {
		PyList_Append(py_list, item->py_object);
	}

	return py_list;
}

static PyObject *py_ue_spython_list_view_clear_selection(ue_PySPythonListView *self, PyObject * args) {


	sw_python_list_view->ClearSelection();

	Py_INCREF(Py_None);
	return Py_None;

}

static PyObject *py_ue_spython_list_view_get_num_items_selected(ue_PySPythonListView *self, PyObject * args) {


	return PyLong_FromLong(sw_python_list_view->GetNumItemsSelected());
}


static PyMethodDef ue_PySPythonListView_methods[] = {
	{ "get_selected_items", (PyCFunction)py_ue_spython_list_view_get_selected_items, METH_VARARGS, "" },
	{ "get_num_items_selected", (PyCFunction)py_ue_spython_list_view_get_num_items_selected, METH_VARARGS, "" },
	{ "clear_selection", (PyCFunction)py_ue_spython_list_view_clear_selection, METH_VARARGS, "" },
	{ NULL }  /* Sentinel */
};

PyTypeObject ue_PySPythonListViewType = {
	PyVarObject_HEAD_INIT(NULL, 0)
	"unreal_engine.SPythonListView", /* tp_name */
	sizeof(ue_PySPythonListView), /* 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 SPythonListView",           /* tp_doc */
	0,                         /* tp_traverse */
	0,                         /* tp_clear */
	0,                         /* tp_richcompare */
	0,                         /* tp_weaklistoffset */
	0,                         /* tp_iter */
	0,                         /* tp_iternext */
	ue_PySPythonListView_methods,             /* tp_methods */
};

static int ue_py_spython_list_view_init(ue_PySPythonListView *self, PyObject *args, PyObject *kwargs) {

	ue_py_slate_setup_farguments(SPythonListView);

	// first of all check for values
	PyObject *values = ue_py_dict_get_item(kwargs, "list_items_source");
	if (!values) {
		PyErr_SetString(PyExc_Exception, "you must specify list 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_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.ListItemsSource(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_optional_bool("handle_gamepad_events", HandleGamepadEvents);
	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);
	ue_py_slate_farguments_optional_float("wheel_scroll_multiplier", WheelScrollMultiplier);

	ue_py_snew(SPythonListView, s_list_view.s_table_view_base.s_compound_widget.s_widget);
	return 0;
}

void ue_python_init_spython_list_view(PyObject *ue_module) {

	ue_PySPythonListViewType.tp_init = (initproc)ue_py_spython_list_view_init;

	ue_PySPythonListViewType.tp_base = &ue_PySListViewType;

	if (PyType_Ready(&ue_PySPythonListViewType) < 0)
		return;

	Py_INCREF(&ue_PySListViewType);
	PyModule_AddObject(ue_module, "SPythonListView", (PyObject *)&ue_PySPythonListViewType);
}
