added update_raw_track() and apply_raw_anim_changes()

Author: unbit

Source/UnrealEnginePython/Private/UEPyModule.cpp

@@ -639,12 +639,17 @@ static PyMethodDef ue_PyUObject_methods[] = {
 	{ "get_blend_parameter", (PyCFunction)py_ue_get_blend_parameter, METH_VARARGS, "" },
 	{ "set_blend_parameter", (PyCFunction)py_ue_set_blend_parameter, METH_VARARGS, "" },
 
+	{ "get_bone_transform", (PyCFunction)py_ue_anim_get_bone_transform, METH_VARARGS, "" },
+	{ "extract_root_motion", (PyCFunction)py_ue_anim_extract_root_motion, METH_VARARGS, "" },
+
 #if WITH_EDITOR
 #if ENGINE_MINOR_VERSION > 13
 	{ "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, "" },
 	{ "add_new_raw_track", (PyCFunction)py_ue_anim_sequence_add_new_raw_track, METH_VARARGS, "" },
 	{ "update_compressed_track_map_from_raw", (PyCFunction)py_ue_anim_sequence_update_compressed_track_map_from_raw, METH_VARARGS, "" },
+	{ "update_raw_track", (PyCFunction)py_ue_anim_sequence_update_raw_track, METH_VARARGS, "" },
+	{ "apply_raw_anim_changes", (PyCFunction)py_ue_anim_sequence_apply_raw_anim_changes, METH_VARARGS, "" },
 #endif
 	{ "add_anim_composite_section", (PyCFunction)py_ue_add_anim_composite_section, METH_VARARGS, "" },
 #endif

Source/UnrealEnginePython/Private/UObject/UEPyAnimSequence.cpp

@@ -18,6 +18,61 @@ PyObject *py_ue_anim_get_skeleton(ue_PyUObject * self, PyObject * args)
 	Py_RETURN_UOBJECT((UObject *)skeleton);
 }
 
+PyObject *py_ue_anim_get_bone_transform(ue_PyUObject * self, PyObject * args)
+{
+	ue_py_check(self);
+
+	int track_index;
+        float frame_time;
+	PyObject *py_b_use_raw_data = nullptr;
+
+	if (!PyArg_ParseTuple(args, "if|O:get_bone_transform", &track_index, &frame_time, &py_b_use_raw_data))
+	{
+		return nullptr;
+	}
+
+	UAnimSequence *anim = ue_py_check_type<UAnimSequence>(self);
+	if (!anim)
+		return PyErr_Format(PyExc_Exception, "UObject is not a UAnimSequence.");
+
+	bool bUseRawData = false;
+	if (py_b_use_raw_data && PyObject_IsTrue(py_b_use_raw_data))
+		bUseRawData = true;
+
+	FTransform OutAtom;
+	anim->GetBoneTransform(OutAtom, track_index, frame_time, bUseRawData);
+
+	return py_ue_new_ftransform(OutAtom);
+}
+
+PyObject *py_ue_anim_extract_root_motion(ue_PyUObject * self, PyObject * args)
+{
+	ue_py_check(self);
+
+	float start_time;
+	float delta_time;
+	PyObject *py_b_allow_looping = nullptr;
+
+	if (!PyArg_ParseTuple(args, "ff|O:extract_root_motion", &start_time, &delta_time, &py_b_allow_looping))
+	{
+		return nullptr;
+	}
+
+	UAnimSequence *anim = ue_py_check_type<UAnimSequence>(self);
+	if (!anim)
+		return PyErr_Format(PyExc_Exception, "UObject is not a UAnimSequence.");
+
+	bool bAllowLooping = false;
+	if (py_b_allow_looping && PyObject_IsTrue(py_b_allow_looping))
+		bAllowLooping = true;
+
+	return py_ue_new_ftransform(anim->ExtractRootMotion(start_time, delta_time, bAllowLooping));
+}
+
+
+
+
+
 
 #if WITH_EDITOR
 #if ENGINE_MINOR_VERSION > 13
@@ -73,6 +128,32 @@ 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_apply_raw_anim_changes(ue_PyUObject * self, PyObject * args)
+{
+	ue_py_check(self);
+
+	UAnimSequence *anim_seq = ue_py_check_type<UAnimSequence>(self);
+	if (!anim_seq)
+		return PyErr_Format(PyExc_Exception, "UObject is not a UAnimSequence.");
+
+
+	if (anim_seq->DoesNeedRebake())
+        {
+        	anim_seq->Modify(true);
+                anim_seq->BakeTrackCurvesToRawAnimation();
+        }
+
+        if (anim_seq->DoesNeedRecompress())
+        {
+        	anim_seq->Modify(true);
+                anim_seq->RequestSyncAnimRecompression(false);
+        }
+
+	Py_RETURN_NONE;
+}
+
+
+
 PyObject *py_ue_anim_sequence_add_new_raw_track(ue_PyUObject * self, PyObject * args)
 {
 	ue_py_check(self);
@@ -105,14 +186,47 @@ PyObject *py_ue_anim_sequence_add_new_raw_track(ue_PyUObject * self, PyObject *
 
 	int32 index = anim_seq->AddNewRawTrack(FName(UTF8_TO_TCHAR(name)), rast);
 
-	if (anim_seq->DoesNeedRebake())
-		anim_seq->BakeTrackCurvesToRawAnimation();
-
 	anim_seq->MarkRawDataAsModified();
+	anim_seq->MarkPackageDirty();
 
 	return PyLong_FromLong(index);
 }
 
+PyObject *py_ue_anim_sequence_update_raw_track(ue_PyUObject * self, PyObject * args)
+{
+	ue_py_check(self);
+
+	int track_index;
+	PyObject *py_rast;
+	if (!PyArg_ParseTuple(args, "iO:update_raw_track", &track_index, &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.");
+
+	ue_PyFRawAnimSequenceTrack *py_f_rast = py_ue_is_fraw_anim_sequence_track(py_rast);
+	if (!py_f_rast)
+	{
+		return PyErr_Format(PyExc_Exception, "argument is not a FRawAnimSequenceTrack.");
+	}
+
+	anim_seq->Modify();
+
+	FRawAnimSequenceTrack& RawRef = anim_seq->GetRawAnimationTrack(track_index);
+	
+	RawRef.PosKeys = py_f_rast->raw_anim_sequence_track.PosKeys;
+	RawRef.RotKeys = py_f_rast->raw_anim_sequence_track.RotKeys;
+	RawRef.ScaleKeys = py_f_rast->raw_anim_sequence_track.ScaleKeys;
+
+	anim_seq->MarkRawDataAsModified();
+	anim_seq->MarkPackageDirty();
+
+	Py_RETURN_NONE;
+}
+
+
+
 PyObject *py_ue_add_anim_composite_section(ue_PyUObject * self, PyObject * args)
 {
 	ue_py_check(self);

Source/UnrealEnginePython/Private/UObject/UEPyAnimSequence.h

@@ -13,10 +13,14 @@ PyObject *py_ue_anim_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_add_new_raw_track(ue_PyUObject *, PyObject *);
+PyObject *py_ue_anim_sequence_update_raw_track(ue_PyUObject *, PyObject *);
 PyObject *py_ue_add_anim_composite_section(ue_PyUObject *, PyObject *);
 PyObject *py_ue_anim_sequence_update_compressed_track_map_from_raw(ue_PyUObject *, PyObject *);
+PyObject *py_ue_anim_sequence_apply_raw_anim_changes(ue_PyUObject *, PyObject *);
 #endif
 PyObject *py_ue_anim_set_skeleton(ue_PyUObject *, PyObject *);
+PyObject *py_ue_anim_get_bone_transform(ue_PyUObject *, PyObject *);
+PyObject *py_ue_anim_extract_root_motion(ue_PyUObject *, PyObject *);
 
 PyObject *py_ue_get_blend_parameter(ue_PyUObject *, PyObject *);
 PyObject *py_ue_set_blend_parameter(ue_PyUObject *, PyObject *);