added api for building animations from python

Author: Roberto De Ioris

Source/UnrealEnginePython/Private/UEPyModule.cpp

@@ -462,6 +462,8 @@ static PyMethodDef ue_PyUObject_methods[] = {
 	{ "anim_sequence_get_skeleton", (PyCFunction)py_ue_anim_sequence_get_skeleton, METH_VARARGS, "" },
 	{ "get_raw_animation_data", (PyCFunction)py_ue_anim_sequence_get_raw_animation_data, METH_VARARGS, "" },
 	{ "get_raw_animation_track", (PyCFunction)py_ue_anim_sequence_get_raw_animation_track, METH_VARARGS, "" },
+	{ "set_skeleton", (PyCFunction)py_ue_anim_sequence_set_skeleton, METH_VARARGS, "" },
+	{ "add_new_raw_track", (PyCFunction)py_ue_anim_sequence_add_new_raw_track, METH_VARARGS, "" },
 
 
 	// StaticMesh

Source/UnrealEnginePython/Private/UObject/UEPyAnimSequence.cpp

@@ -57,3 +57,59 @@ PyObject *py_ue_anim_sequence_get_raw_animation_track(ue_PyUObject * self, PyObj
 
 	return py_ue_new_fraw_anim_sequence_track(anim_seq->GetRawAnimationTrack(index));
 }
+
+PyObject *py_ue_anim_sequence_add_new_raw_track(ue_PyUObject * self, PyObject * args) {
+	ue_py_check(self);
+
+	char *name;
+	PyObject *py_rast = nullptr;
+	if (!PyArg_ParseTuple(args, "s|O:add_new_raw_track", &name, &py_rast))
+		return nullptr;
+
+	UAnimSequence *anim_seq = ue_py_check_type<UAnimSequence>(self);
+	if (!anim_seq)
+		return PyErr_Format(PyExc_Exception, "UObject is not a UAnimSequence.");
+
+	FRawAnimSequenceTrack *rast = nullptr;
+
+	if (py_rast) {
+		ue_PyFRawAnimSequenceTrack *py_f_rast = py_ue_is_fraw_anim_sequence_track(py_rast);
+		if (py_f_rast) {
+			rast = &py_f_rast->raw_anim_sequence_track;
+		}
+		else {
+			return PyErr_Format(PyExc_Exception, "argument is not a FRawAnimSequenceTrack.");
+		}
+	}
+
+	anim_seq->Modify();
+
+	int32 index = anim_seq->AddNewRawTrack(FName(UTF8_TO_TCHAR(name)), rast);
+
+	if (anim_seq->DoesNeedRebake())
+		anim_seq->BakeTrackCurvesToRawAnimation();
+
+	anim_seq->MarkRawDataAsModified();
+
+	return PyLong_FromLong(index);
+}
+
+PyObject *py_ue_anim_sequence_set_skeleton(ue_PyUObject * self, PyObject * args) {
+	ue_py_check(self);
+
+	PyObject *py_skeleton;
+	if (!PyArg_ParseTuple(args, "O:set_skeleton", &py_skeleton))
+		return nullptr;
+
+	UAnimSequence *anim_seq = ue_py_check_type<UAnimSequence>(self);
+	if (!anim_seq)
+		return PyErr_Format(PyExc_Exception, "UObject is not a UAnimSequence.");
+
+	USkeleton *skeleton = ue_py_check_type<USkeleton>(py_skeleton);
+	if (!skeleton)
+		return PyErr_Format(PyExc_Exception, "argument is not a USkeleton.");
+
+	anim_seq->SetSkeleton(skeleton);
+	
+	Py_RETURN_NONE;
+}

Source/UnrealEnginePython/Private/UObject/UEPyAnimSequence.h

@@ -6,4 +6,6 @@
 
 PyObject *py_ue_anim_sequence_get_skeleton(ue_PyUObject *, PyObject *);
 PyObject *py_ue_anim_sequence_get_raw_animation_data(ue_PyUObject *, PyObject *);
-PyObject *py_ue_anim_sequence_get_raw_animation_track(ue_PyUObject *, PyObject *);
+PyObject *py_ue_anim_sequence_get_raw_animation_track(ue_PyUObject *, PyObject *);
+PyObject *py_ue_anim_sequence_add_new_raw_track(ue_PyUObject *, PyObject *);
+PyObject *py_ue_anim_sequence_set_skeleton(ue_PyUObject *, PyObject *);

Source/UnrealEnginePython/Private/Wrappers/UEPyFRawAnimSequenceTrack.cpp

@@ -168,6 +168,12 @@ void ue_python_init_fraw_anim_sequence_track(PyObject *ue_module) {
 	PyModule_AddObject(ue_module, "FRawAnimSequenceTrack", (PyObject *)&ue_PyFRawAnimSequenceTrackType);
 }
 
+ue_PyFRawAnimSequenceTrack *py_ue_is_fraw_anim_sequence_track(PyObject *obj) {
+	if (!PyObject_IsInstance(obj, (PyObject *)&ue_PyFRawAnimSequenceTrackType))
+		return nullptr;
+	return (ue_PyFRawAnimSequenceTrack *)obj;
+}
+
 PyObject *py_ue_new_fraw_anim_sequence_track(FRawAnimSequenceTrack raw_anim_sequence_track) {
 	ue_PyFRawAnimSequenceTrack *ret = (ue_PyFRawAnimSequenceTrack *)PyObject_New(ue_PyFRawAnimSequenceTrack, &ue_PyFRawAnimSequenceTrackType);
 	new(&ret->raw_anim_sequence_track) FRawAnimSequenceTrack(raw_anim_sequence_track);

Source/UnrealEnginePython/Private/Wrappers/UEPyFRawAnimSequenceTrack.h

@@ -12,4 +12,6 @@ typedef struct {
 
 PyObject *py_ue_new_fraw_anim_sequence_track(FRawAnimSequenceTrack);
 
-void ue_python_init_fraw_anim_sequence_track(PyObject *);
+void ue_python_init_fraw_anim_sequence_track(PyObject *);
+
+ue_PyFRawAnimSequenceTrack *py_ue_is_fraw_anim_sequence_track(PyObject *);

Source/UnrealEnginePython/Private/Wrappers/UEPyFRotator.cpp

@@ -20,12 +20,17 @@ static PyObject *py_ue_frotator_normalized(ue_PyFRotator *self, PyObject * args)
 	return py_ue_new_frotator(rot);
 }
 
+static PyObject *py_ue_frotator_quaternion(ue_PyFRotator *self, PyObject * args) {
+	FQuat quat = self->rot.Quaternion();
+	return py_ue_new_fquat(quat);
+}
 
 static PyMethodDef ue_PyFRotator_methods[] = {
 	{ "get_vector", (PyCFunction)py_ue_frotator_get_vector, METH_VARARGS, "" },
 	{ "get_euler", (PyCFunction)py_ue_frotator_get_euler, METH_VARARGS, "" },
 	{ "normalized", (PyCFunction)py_ue_frotator_normalized, METH_VARARGS, "" },
 	{ "inversed", (PyCFunction)py_ue_frotator_normalized, METH_VARARGS, "" },
+	{ "quaternion", (PyCFunction)py_ue_frotator_quaternion, METH_VARARGS, "" },
 	{ NULL }  /* Sentinel */
 };