finalized tagging api

Author: Roberto De Ioris

Source/UnrealEnginePython/Private/UEPyModule.cpp

@@ -769,6 +769,7 @@ static PyMethodDef ue_PyUObject_methods[] = {
 #pragma warning(suppress: 4191)
 	{ "actor_spawn", (PyCFunction)py_ue_actor_spawn, METH_VARARGS | METH_KEYWORDS, "" },
 	{ "actor_has_tag", (PyCFunction)py_ue_actor_has_tag, METH_VARARGS, "" },
+	{ "component_has_tag", (PyCFunction)py_ue_component_has_tag, METH_VARARGS, "" },
 	{ "get_actor_bounds", (PyCFunction)py_ue_get_actor_bounds, METH_VARARGS, "" },
 
 	{ "line_trace_single_by_channel", (PyCFunction)py_ue_line_trace_single_by_channel, METH_VARARGS, "" },
@@ -799,11 +800,17 @@ static PyMethodDef ue_PyUObject_methods[] = {
 	{ "get_actor_root_component", (PyCFunction)py_ue_get_actor_root_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, "" },
+	{ "get_actor_component_by_class", (PyCFunction)py_ue_get_actor_component_by_type, METH_VARARGS, "" },
 	{ "get_component_by_type", (PyCFunction)py_ue_get_actor_component_by_type, METH_VARARGS, "" },
+	{ "get_component_by_class", (PyCFunction)py_ue_get_actor_component_by_type, METH_VARARGS, "" },
 	{ "get_component", (PyCFunction)py_ue_get_actor_component, METH_VARARGS, "" },
 	{ "get_actor_component", (PyCFunction)py_ue_get_actor_component, METH_VARARGS, "" },
 	{ "get_actor_components_by_type", (PyCFunction)py_ue_get_actor_components_by_type, METH_VARARGS, "" },
 	{ "get_components_by_type", (PyCFunction)py_ue_get_actor_components_by_type, METH_VARARGS, "" },
+	{ "get_actor_components_by_class", (PyCFunction)py_ue_get_actor_components_by_type, METH_VARARGS, "" },
+	{ "get_components_by_class", (PyCFunction)py_ue_get_actor_components_by_type, METH_VARARGS, "" },
+	{ "get_actor_components_by_tag", (PyCFunction)py_ue_get_actor_components_by_tag, METH_VARARGS, "" },
+	{ "get_components_by_tag", (PyCFunction)py_ue_get_actor_components_by_tag, METH_VARARGS, "" },
 
 	{ "add_python_component", (PyCFunction)py_ue_add_python_component, METH_VARARGS, "" },
 

Source/UnrealEnginePython/Private/UObject/UEPyActor.cpp

@@ -27,6 +27,29 @@ PyObject *py_ue_actor_has_tag(ue_PyUObject * self, PyObject * args)
 	Py_RETURN_FALSE;
 }
 
+PyObject *py_ue_component_has_tag(ue_PyUObject * self, PyObject * args)
+{
+
+	ue_py_check(self);
+
+	char *tag;
+	if (!PyArg_ParseTuple(args, "s:component_has_tag", &tag))
+	{
+		return nullptr;
+	}
+
+	UActorComponent *component = ue_py_check_type<UActorComponent>(self);
+	if (!component)
+		return PyErr_Format(PyExc_Exception, "uobject is not an UActorComponent");
+
+	if (component->ComponentHasTag(FName(UTF8_TO_TCHAR(tag))))
+	{
+		Py_RETURN_TRUE;
+	}
+
+	Py_RETURN_FALSE;
+}
+
 PyObject *py_ue_actor_begin_play(ue_PyUObject * self, PyObject * args)
 {
 
@@ -692,6 +715,51 @@ PyObject *py_ue_get_actor_components_by_type(ue_PyUObject * self, PyObject * arg
 
 }
 
+PyObject *py_ue_get_actor_components_by_tag(ue_PyUObject * self, PyObject * args)
+{
+
+	ue_py_check(self);
+
+	char *tag;
+	PyObject *py_uclass = nullptr;
+	if (!PyArg_ParseTuple(args, "s|O:get_actor_components_by_tag", &tag, &py_uclass))
+	{
+		return nullptr;
+	}
+
+	AActor *actor = ue_get_actor(self);
+	if (!actor)
+		return PyErr_Format(PyExc_Exception, "uobject is not an AActor");
+
+	PyObject *components = PyList_New(0);
+
+	UClass *u_class = UActorComponent::StaticClass();
+
+	if (py_uclass)
+	{
+		u_class = ue_py_check_type<UClass>(py_uclass);
+		if (!u_class)
+		{
+			return PyErr_Format(PyExc_Exception, "argument is not a UClass");
+		}
+
+		if (!u_class->IsChildOf<UActorComponent>())
+		{
+			return PyErr_Format(PyExc_Exception, "argument is not a UClass inheriting from UActorComponent");
+		}
+	}
+
+	for (UActorComponent *component : actor->GetComponentsByTag(u_class, FName(UTF8_TO_TCHAR(tag))))
+	{
+		ue_PyUObject *item = ue_get_python_uobject(component);
+		if (item)
+			PyList_Append(components, (PyObject *)item);
+	}
+
+	return components;
+
+}
+
 
 PyObject *py_ue_actor_spawn(ue_PyUObject * self, PyObject * args, PyObject *kwargs)
 {
@@ -790,7 +858,7 @@ PyObject *py_ue_get_overlapping_actors(ue_PyUObject * self, PyObject * args)
 
 	ue_py_check(self);
 
-	
+
 	PyObject *class_filter = nullptr;
 	if (!PyArg_ParseTuple(args, "|O:get_overlapping_actors", &class_filter))
 	{

Source/UnrealEnginePython/Private/UObject/UEPyActor.h

@@ -16,6 +16,7 @@
 #include "UObject/UObjectThreadContext.h"
 
 PyObject *py_ue_actor_has_tag(ue_PyUObject *, PyObject *);
+PyObject *py_ue_component_has_tag(ue_PyUObject *, PyObject *);
 PyObject *py_ue_actor_begin_play(ue_PyUObject *, PyObject *);
 PyObject *py_ue_get_actor_bounds(ue_PyUObject *, PyObject *);
 PyObject *py_ue_get_actor_component(ue_PyUObject *, PyObject *);
@@ -37,6 +38,7 @@ PyObject *py_ue_add_actor_root_component(ue_PyUObject *, PyObject *);
 PyObject *py_ue_actor_has_component_of_type(ue_PyUObject *, PyObject *);
 PyObject *py_ue_get_actor_component_by_type(ue_PyUObject *, PyObject *);
 PyObject *py_ue_get_actor_components_by_type(ue_PyUObject *, PyObject *);
+PyObject *py_ue_get_actor_components_by_tag(ue_PyUObject *, PyObject *);
 PyObject *py_ue_actor_spawn(ue_PyUObject * self, PyObject *, PyObject *);
 PyObject *py_ue_get_overlapping_actors(ue_PyUObject *, PyObject *);