Add force location & get socket local location

Author: kobraksq

Source/UnrealEnginePython/Private/UEPyModule.cpp

@@ -945,6 +945,7 @@ static PyMethodDef ue_PyUObject_methods[] = {
 	{ "add_impulse", (PyCFunction)py_ue_add_impulse, METH_VARARGS, "" },
 	{ "add_angular_impulse", (PyCFunction)py_ue_add_angular_impulse, METH_VARARGS, "" },
 	{ "add_force", (PyCFunction)py_ue_add_force, METH_VARARGS, "" },
+	{ "add_force_location", (PyCFunction)py_ue_add_force_location, METH_VARARGS, "" },
 	{ "add_force_location_local", (PyCFunction)py_ue_add_force_location_local, METH_VARARGS, "" },
 	{ "add_torque", (PyCFunction)py_ue_add_torque, METH_VARARGS, "" },
 	{ "set_physics_linear_velocity", (PyCFunction)py_ue_set_physics_linear_velocity, METH_VARARGS, "" },
@@ -1035,6 +1036,7 @@ static PyMethodDef ue_PyUObject_methods[] = {
 	// Attaching
 
 	{ "get_socket_location", (PyCFunction)py_ue_get_socket_location, METH_VARARGS, "" },
+	{ "get_socket_location_local", (PyCFunction)py_ue_get_socket_location_local, METH_VARARGS, "" },
 	{ "get_socket_rotation", (PyCFunction)py_ue_get_socket_rotation, METH_VARARGS, "" },
 	{ "get_socket_transform", (PyCFunction)py_ue_get_socket_transform, METH_VARARGS, "" },
 	{ "get_socket_world_transform", (PyCFunction)py_ue_get_socket_world_transform, METH_VARARGS, "" },

Source/UnrealEnginePython/Private/UObject/UEPyAttaching.cpp

@@ -23,6 +23,27 @@ PyObject *py_ue_get_socket_location(ue_PyUObject *self, PyObject * args)
 	return py_ue_new_fvector(vec);
 }
 
+PyObject* py_ue_get_socket_location_local(ue_PyUObject* self, PyObject* args)
+{
+
+	ue_py_check(self);
+
+	char* socket_name;
+	if (!PyArg_ParseTuple(args, "s:get_socket_location_local", &socket_name))
+	{
+		return NULL;
+	}
+
+	if (!self->ue_object->IsA<USceneComponent>())
+		return PyErr_Format(PyExc_Exception, "uobject is not a USceneComponent");
+
+	USceneComponent* component = (USceneComponent*)self->ue_object;
+
+	FTransform transform = component->GetSocketTransform(UTF8_TO_TCHAR(socket_name), ERelativeTransformSpace::RTS_Component);
+	return py_ue_new_fvector(transform.GetTranslation());
+}
+
+
 PyObject *py_ue_get_socket_rotation(ue_PyUObject *self, PyObject * args)
 {
 

Source/UnrealEnginePython/Private/UObject/UEPyAttaching.h

@@ -5,6 +5,7 @@
 #include "UEPyModule.h"
 
 PyObject *py_ue_get_socket_location(ue_PyUObject *, PyObject *);
+PyObject* py_ue_get_socket_location_local(ue_PyUObject*, PyObject*);
 PyObject *py_ue_get_socket_rotation(ue_PyUObject *, PyObject *);
 PyObject *py_ue_get_socket_transform(ue_PyUObject *, PyObject *);
 PyObject *py_ue_get_socket_world_transform(ue_PyUObject *, PyObject *);

Source/UnrealEnginePython/Private/UObject/UEPyPhysics.cpp

@@ -200,6 +200,62 @@ PyObject *py_ue_add_force(ue_PyUObject * self, PyObject * args)
 	return Py_None;
 }
 
+PyObject* py_ue_add_force_location(ue_PyUObject* self, PyObject* args)
+{
+
+	ue_py_check(self);
+
+	PyObject* py_obj_force = nullptr;
+	PyObject* py_obj_location = nullptr;
+	char* bone_name = nullptr;
+
+	if (!PyArg_ParseTuple(args, "O|Os:add_force_location_local", &py_obj_force, &py_obj_location, &bone_name))
+	{
+		return nullptr;
+	}
+
+	UPrimitiveComponent* primitive = nullptr;
+
+	if (self->ue_object->IsA<UPrimitiveComponent>())
+	{
+		primitive = (UPrimitiveComponent*)self->ue_object;
+	}
+	else
+	{
+		return PyErr_Format(PyExc_Exception, "uobject is not an UPrimitiveComponent");
+	}
+
+	FVector force = FVector(0, 0, 0);
+	if (py_obj_force)
+	{
+		ue_PyFVector* py_force = py_ue_is_fvector(py_obj_force);
+		if (!py_force)
+			return PyErr_Format(PyExc_Exception, "force must be a FVector");
+		force = py_force->vec;
+	}
+
+	FVector location = FVector(0, 0, 0);
+	if (py_obj_location)
+	{
+		ue_PyFVector* py_location = py_ue_is_fvector(py_obj_location);
+		if (!py_location)
+			return PyErr_Format(PyExc_Exception, "location must be a FVector");
+		location = py_location->vec;
+	}
+
+	FName f_bone_name = NAME_None;
+	if (bone_name)
+	{
+		f_bone_name = FName(UTF8_TO_TCHAR(bone_name));
+	}
+
+	primitive->AddForceAtLocation(force, location, f_bone_name);
+
+
+	Py_INCREF(Py_None);
+	return Py_None;
+}
+
 PyObject* py_ue_add_force_location_local(ue_PyUObject* self, PyObject* args)
 {
 

Source/UnrealEnginePython/Private/UObject/UEPyPhysics.h

@@ -8,6 +8,7 @@ PyObject *py_ue_set_simulate_physics(ue_PyUObject *, PyObject *);
 PyObject *py_ue_add_impulse(ue_PyUObject *, PyObject *);
 PyObject *py_ue_add_angular_impulse(ue_PyUObject *, PyObject *);
 PyObject *py_ue_add_force(ue_PyUObject *, PyObject *);
+PyObject* py_ue_add_force_location(ue_PyUObject*, PyObject*);
 PyObject* py_ue_add_force_location_local(ue_PyUObject*, PyObject*);
 PyObject *py_ue_add_torque(ue_PyUObject *, PyObject *);
 PyObject *py_ue_set_physics_linear_velocity(ue_PyUObject *, PyObject *);