added thumbnail api to FAssetData, #254

Author: Roberto De Ioris

Source/UnrealEnginePython/Private/Wrappers/UEPyFAssetData.cpp

@@ -2,58 +2,117 @@
 
 #if WITH_EDITOR
 
-static PyObject *py_ue_fassetdata_get_asset(ue_PyFAssetData *self, PyObject * args) {
+#include "ObjectTools.h"
+
+static PyObject *py_ue_fassetdata_get_asset(ue_PyFAssetData *self, PyObject * args)
+{
 	PyObject *ret = (PyObject *)ue_get_python_wrapper(self->asset_data.GetAsset());
-	if (!ret) {
+	if (!ret)
+	{
 		return PyErr_Format(PyExc_Exception, "unable to get UObject from asset");
 	}
 	Py_INCREF(ret);
 	return ret;
 }
 
-static PyObject *py_ue_fassetdata_is_asset_loaded(ue_PyFAssetData *self, PyObject * args) {
+static PyObject *py_ue_fassetdata_is_asset_loaded(ue_PyFAssetData *self, PyObject * args)
+{
 	if (self->asset_data.IsAssetLoaded())
 		Py_RETURN_TRUE;
 	Py_RETURN_FALSE;
 }
 
+static PyObject *py_ue_fassetdata_get_thumbnail(ue_PyFAssetData *self, PyObject * args)
+{
+	TArray<FName> names;
+	FName name = FName(*self->asset_data.GetFullName());
+	names.Add(name);
+	FThumbnailMap map;
+
+	if (!ThumbnailTools::ConditionallyLoadThumbnailsForObjects(names, map))
+	{
+		return PyErr_Format(PyExc_Exception, "Unable to retrieve thumbnail from FAssetData");
+	}
+
+	FObjectThumbnail *thumbnail = map.Find(name);
+	if (!thumbnail)
+	{
+		return PyErr_Format(PyExc_Exception, "Unable to retrieve thumbnail from FAssetData");
+	}
+
+	return py_ue_new_fobject_thumbnail(*thumbnail);
+}
+
+static PyObject *py_ue_fassetdata_has_custom_thumbnail(ue_PyFAssetData *self, PyObject * args)
+{
+
+	if (!ThumbnailTools::AssetHasCustomThumbnail(self->asset_data))
+	{
+		Py_RETURN_FALSE;
+	}
+
+	Py_RETURN_TRUE;
+}
+
+static PyObject *py_ue_fassetdata_has_cached_thumbnail(ue_PyFAssetData *self, PyObject * args)
+{
+
+	if (!ThumbnailTools::FindCachedThumbnail(self->asset_data.GetFullName()))
+	{
+		Py_RETURN_FALSE;
+	}
+
+	Py_RETURN_TRUE;
+}
+
 static PyMethodDef ue_PyFAssetData_methods[] = {
 	{ "get_asset", (PyCFunction)py_ue_fassetdata_get_asset, METH_VARARGS, "" },
 	{ "is_asset_loaded", (PyCFunction)py_ue_fassetdata_is_asset_loaded, METH_VARARGS, "" },
+	{ "get_thumbnail", (PyCFunction)py_ue_fassetdata_get_thumbnail, METH_VARARGS, "" },
+	{ "has_custom_thumbnail", (PyCFunction)py_ue_fassetdata_has_custom_thumbnail, METH_VARARGS, "" },
+	{ "has_cached_thumbnail", (PyCFunction)py_ue_fassetdata_has_cached_thumbnail, METH_VARARGS, "" },
 	{ NULL }  /* Sentinel */
 };
 
-static PyObject *py_ue_fassetdata_get_asset_class(ue_PyFAssetData *self, void *closure) {
+static PyObject *py_ue_fassetdata_get_asset_class(ue_PyFAssetData *self, void *closure)
+{
 	return PyUnicode_FromString(TCHAR_TO_UTF8(*self->asset_data.AssetClass.ToString()));
 }
 
-static PyObject *py_ue_fassetdata_get_asset_name(ue_PyFAssetData *self, void *closure) {
+static PyObject *py_ue_fassetdata_get_asset_name(ue_PyFAssetData *self, void *closure)
+{
 	return PyUnicode_FromString(TCHAR_TO_UTF8(*self->asset_data.AssetName.ToString()));
 }
 
 #if ENGINE_MINOR_VERSION < 17
-static PyObject *py_ue_fassetdata_get_group_names(ue_PyFAssetData *self, void *closure) {
+static PyObject *py_ue_fassetdata_get_group_names(ue_PyFAssetData *self, void *closure)
+{
 	return PyUnicode_FromString(TCHAR_TO_UTF8(*self->asset_data.GroupNames.ToString()));
 }
 #endif
 
-static PyObject *py_ue_fassetdata_get_object_path(ue_PyFAssetData *self, void *closure) {
+static PyObject *py_ue_fassetdata_get_object_path(ue_PyFAssetData *self, void *closure)
+{
 	return PyUnicode_FromString(TCHAR_TO_UTF8(*self->asset_data.ObjectPath.ToString()));
 }
 
-static PyObject *py_ue_fassetdata_get_package_flags(ue_PyFAssetData *self, void *closure) {
+static PyObject *py_ue_fassetdata_get_package_flags(ue_PyFAssetData *self, void *closure)
+{
 	return PyLong_FromUnsignedLong(self->asset_data.PackageFlags);
 }
 
-static PyObject *py_ue_fassetdata_get_package_name(ue_PyFAssetData *self, void *closure) {
+static PyObject *py_ue_fassetdata_get_package_name(ue_PyFAssetData *self, void *closure)
+{
 	return PyUnicode_FromString(TCHAR_TO_UTF8(*self->asset_data.PackageName.ToString()));
 }
 
-static PyObject *py_ue_fassetdata_get_package_path(ue_PyFAssetData *self, void *closure) {
+static PyObject *py_ue_fassetdata_get_package_path(ue_PyFAssetData *self, void *closure)
+{
 	return PyUnicode_FromString(TCHAR_TO_UTF8(*self->asset_data.PackagePath.ToString()));
 }
 
-static PyObject *py_ue_fassetdata_get_tags_and_values(ue_PyFAssetData *self, void *closure) {
+static PyObject *py_ue_fassetdata_get_tags_and_values(ue_PyFAssetData *self, void *closure)
+{
 	PyObject *ret = PyDict_New();
 	for (auto It = self->asset_data.TagsAndValues.CreateConstIterator(); It; ++It)
 	{
@@ -78,7 +137,8 @@ static PyGetSetDef ue_PyFAssetData_getseters[] = {
 	{ NULL }  /* Sentinel */
 };
 
-static int ue_py_fassetdata_init(ue_PyFAssetData *self, PyObject *args, PyObject *kwargs) {
+static int ue_py_fassetdata_init(ue_PyFAssetData *self, PyObject *args, PyObject *kwargs)
+{
 	// avoid FAssetData manual creation
 	return -1;
 }
@@ -122,7 +182,8 @@ static PyTypeObject ue_PyFAssetDataType = {
 	ue_PyFAssetData_getseters,                         /* tp_getset */
 };
 
-void ue_python_init_fassetdata(PyObject *ue_module) {
+void ue_python_init_fassetdata(PyObject *ue_module)
+{
 	ue_PyFAssetDataType.tp_new = PyType_GenericNew;;
 	ue_PyFAssetDataType.tp_init = (initproc)ue_py_fassetdata_init;
 	if (PyType_Ready(&ue_PyFAssetDataType) < 0)
@@ -132,14 +193,16 @@ void ue_python_init_fassetdata(PyObject *ue_module) {
 	PyModule_AddObject(ue_module, "FAssetData", (PyObject *)&ue_PyFAssetDataType);
 }
 
-PyObject *py_ue_new_fassetdata(FAssetData asset_data) {
+PyObject *py_ue_new_fassetdata(FAssetData asset_data)
+{
 	ue_PyFAssetData *ret = (ue_PyFAssetData *)PyObject_New(ue_PyFAssetData, &ue_PyFAssetDataType);
 
 	new(&ret->asset_data) FAssetData(asset_data);
 	return (PyObject *)ret;
 }
 
-ue_PyFAssetData *py_ue_is_fassetdata(PyObject *obj) {
+ue_PyFAssetData *py_ue_is_fassetdata(PyObject *obj)
+{
 	if (!PyObject_IsInstance(obj, (PyObject *)&ue_PyFAssetDataType))
 		return nullptr;
 	return (ue_PyFAssetData *)obj;