Browse Source

Support for quaternion rotation animations

Put this together, through trial and error
There are probably better way to do this
Was able to export quaternion animation with it
pull/254/head
Hazado 4 years ago committed by GitHub
parent
commit
4c621f6457
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 25
      korman/exporter/animation.py

25
korman/exporter/animation.py

@ -387,10 +387,15 @@ class AnimationConverter:
def convert_transform_controller(self, fcurves, xform, allow_empty=False) -> Union[None, plCompoundController]: def convert_transform_controller(self, fcurves, xform, allow_empty=False) -> Union[None, plCompoundController]:
if not fcurves and not allow_empty: if not fcurves and not allow_empty:
return None return None
rotation_quaternion = (i for i in fcurves if i.data_path == "rotation_quaternion")
pos = self.make_pos_controller(fcurves, "location", xform.to_translation()) pos = self.make_pos_controller(fcurves, "location", xform.to_translation())
# TODO: support rotation_quaternion # TODO: support rotation_quaternion
rot = self.make_rot_controller(fcurves, "rotation_euler", xform.to_euler()) if rotation_quaternion:
rot = self.make_rot_controller(fcurves, "rotation_quaternion", xform.to_quaternion(), 4)
else:
rot = self.make_rot_controller(fcurves, "rotation_euler", xform.to_euler(), 3)
scale = self.make_scale_controller(fcurves, "scale", xform.to_scale()) scale = self.make_scale_controller(fcurves, "scale", xform.to_scale())
if pos is None and rot is None and scale is None: if pos is None and rot is None and scale is None:
if not allow_empty: if not allow_empty:
@ -459,9 +464,9 @@ class AnimationConverter:
ctrl = self._make_point3_controller(keyframes, bez_chans) ctrl = self._make_point3_controller(keyframes, bez_chans)
return ctrl return ctrl
def make_rot_controller(self, fcurves, data_path : str, default_xform, convert=None) -> Union[None, plCompoundController, plLeafController]: def make_rot_controller(self, fcurves, data_path : str, default_xform, num_channels, convert=None) -> Union[None, plCompoundController, plLeafController]:
rot_curves = [i for i in fcurves if i.data_path == data_path and i.keyframe_points] rot_curves = [i for i in fcurves if i.data_path == data_path and i.keyframe_points]
keyframes, bez_chans = self._process_keyframes(rot_curves, 3, default_xform, convert=None) keyframes, bez_chans = self._process_keyframes(rot_curves, num_channels, default_xform, convert=None)
if not keyframes: if not keyframes:
return None return None
@ -470,7 +475,7 @@ class AnimationConverter:
if bez_chans: if bez_chans:
ctrl = self._make_scalar_compound_controller(keyframes, bez_chans) ctrl = self._make_scalar_compound_controller(keyframes, bez_chans)
else: else:
ctrl = self._make_quat_controller( keyframes) ctrl = self._make_quat_controller( keyframes, num_channels)
return ctrl return ctrl
def make_scale_controller(self, fcurves, data_path : str, default_xform, convert=None) -> plLeafController: def make_scale_controller(self, fcurves, data_path : str, default_xform, convert=None) -> plLeafController:
@ -523,7 +528,7 @@ class AnimationConverter:
ctrl.keys = (exported_frames, keyframe_type) ctrl.keys = (exported_frames, keyframe_type)
return ctrl return ctrl
def _make_quat_controller(self, keyframes) -> plLeafController: def _make_quat_controller(self, keyframes, num_channels) -> plLeafController:
ctrl = plLeafController() ctrl = plLeafController()
keyframe_type = hsKeyFrame.kQuatKeyFrame keyframe_type = hsKeyFrame.kQuatKeyFrame
exported_frames = [] exported_frames = []
@ -535,8 +540,12 @@ class AnimationConverter:
exported.type = keyframe_type exported.type = keyframe_type
# NOTE: quat keyframes don't do bezier nonsense # NOTE: quat keyframes don't do bezier nonsense
value = mathutils.Euler(keyframe.values) if num_channels == 3:
exported.value = utils.quaternion(value.to_quaternion()) value = mathutils.Euler(keyframe.values)
exported.value = utils.quaternion(value.to_quaternion())
else:
value = mathutils.Quaternion(keyframe.values)
exported.value = utils.quaternion(value)
exported_frames.append(exported) exported_frames.append(exported)
ctrl.keys = (exported_frames, keyframe_type) ctrl.keys = (exported_frames, keyframe_type)
return ctrl return ctrl

Loading…
Cancel
Save