added pawn and controller optimized classes

Author: Roberto De Ioris

Source/UnrealEnginePython/Private/UEPyActor.cpp

@@ -1,6 +1,7 @@
 #include "UnrealEnginePythonPrivatePCH.h"
 
 
+
 PyObject *py_ue_actor_has_tag(ue_PyUObject * self, PyObject * args) {
 
 	ue_py_check(self);
@@ -536,8 +537,8 @@ PyObject *py_ue_actor_spawn(ue_PyUObject * self, PyObject * args) {
 		return PyErr_Format(PyExc_Exception, "argument is not a UClass derived from AActor");
 	}
 
-	FVector location = FVector(0,0,0);
-	FRotator rotation = FRotator(0,0,0);
+	FVector location = FVector(0, 0, 0);
+	FRotator rotation = FRotator(0, 0, 0);
 
 	if (py_obj_location) {
 		ue_PyFVector *py_location = py_ue_is_fvector(py_obj_location);

Source/UnrealEnginePython/Private/UEPyController.cpp

@@ -0,0 +1,50 @@
+#include "UnrealEnginePythonPrivatePCH.h"
+
+
+PyObject *py_ue_controller_posses(ue_PyUObject * self, PyObject * args) {
+
+	ue_py_check(self);
+
+	PyObject *obj;
+	if (!PyArg_ParseTuple(args, "O:posses", &obj)) {
+		return NULL;
+	}
+
+	if (!self->ue_object->IsA<AController>()) {
+		return PyErr_Format(PyExc_Exception, "uobject is not an APawn");
+	}
+
+	if (!ue_is_pyuobject(obj)) {
+		return PyErr_Format(PyExc_Exception, "argument is not a UObject");
+	}
+
+	ue_PyUObject *py_obj = (ue_PyUObject *)obj;
+
+	if (!py_obj->ue_object->IsA<APawn>()) {
+		return PyErr_Format(PyExc_Exception, "argument is not a APAwn");
+	}
+
+	APawn *pawn = (APawn *)py_obj->ue_object;
+	AController *controller = (AController *)self->ue_object;
+
+	controller->Possess(pawn);
+
+	Py_INCREF(Py_None);
+	return Py_None;
+}
+
+PyObject *py_ue_controller_unposses(ue_PyUObject * self, PyObject * args) {
+
+	ue_py_check(self);
+
+	if (!self->ue_object->IsA<AController>()) {
+		return PyErr_Format(PyExc_Exception, "uobject is not an APawn");
+	}
+
+	AController *controller = (AController *)self->ue_object;
+
+	controller->UnPossess();
+
+	Py_INCREF(Py_None);
+	return Py_None;
+}

Source/UnrealEnginePython/Private/UEPyController.h

@@ -0,0 +1,8 @@
+#pragma once
+
+
+
+#include "UnrealEnginePython.h"
+
+PyObject *py_ue_controller_posses(ue_PyUObject *, PyObject *);
+PyObject *py_ue_controller_unposses(ue_PyUObject *, PyObject *);

Source/UnrealEnginePython/Private/UEPyModule.cpp

@@ -25,6 +25,8 @@
 #include "UEPyAssetUserData.h"
 #include "UEPyTexture.h"
 #include "UEPyMaterial.h"
+#include "UEPyPawn.h"
+#include "UEPyController.h"
 #if WITH_EDITOR
 #include "UEPyEditor.h"
 #include "UEPyEdGraph.h"
@@ -510,6 +512,13 @@ static PyMethodDef ue_PyUObject_methods[] = {
 
 	{ "get_overlapping_actors", (PyCFunction)py_ue_get_overlapping_actors, METH_VARARGS, "" },
 
+	// Pawn
+	{ "get_controller", (PyCFunction)py_ue_pawn_get_controller, METH_VARARGS, "" },
+
+	// Controller
+	{ "posses", (PyCFunction)py_ue_controller_posses, METH_VARARGS, "" },
+	{ "unposses", (PyCFunction)py_ue_controller_unposses, METH_VARARGS, "" },
+
 
 	// Attaching
 
@@ -587,8 +596,8 @@ void ue_pydelegates_cleanup(ue_PyUObject *self) {
 			UE_LOG(LogPython, Warning, TEXT("Removing UPythonDelegate %p from ue_PyUObject %p mapped to UObject %p"), py_delegate, self, self->ue_object);
 #endif
 			py_delegate->RemoveFromRoot();
+		}
 	}
-}
 	self->python_delegates_gc->clear();
 	delete self->python_delegates_gc;
 	self->python_delegates_gc = nullptr;
@@ -1263,7 +1272,7 @@ void unreal_engine_py_log_error() {
 	}
 
 	PyErr_Clear();
-	}
+}
 
 // retrieve a UWorld from a generic UObject (if possible)
 UWorld *ue_get_uworld(ue_PyUObject *py_obj) {

Source/UnrealEnginePython/Private/UEPyPawn.cpp

@@ -0,0 +1,22 @@
+#include "UnrealEnginePythonPrivatePCH.h"
+
+
+PyObject *py_ue_pawn_get_controller(ue_PyUObject * self, PyObject * args) {
+
+	ue_py_check(self);
+
+	if (!self->ue_object->IsA<APAwn>()) {
+		return PyErr_Format(PyExc_Exception, "uobject is not an APawn");
+	}
+
+	APawn *pawn = (APAwn *)self->ue_object;
+
+	ue_PyUObject *ret = ue_get_python_wrapper(pawn->GetController());
+	if (!ret)
+		return PyErr_Format(PyExc_Exception, "uobject is in invalid state");
+	Py_INCREF(ret);
+	return (PyObject *)ret;
+}
+
+
+

Source/UnrealEnginePython/Private/UEPyPawn.h

@@ -0,0 +1,7 @@
+#pragma once
+
+
+
+#include "UnrealEnginePython.h"
+
+PyObject *py_ue_pawn_get_controller(ue_PyUObject *, PyObject *);