refactor: updating animation and armature handling

Author: sawich

io_soulworker/__init__.py

@@ -1,9 +1,14 @@
-import bpy
-
 from logging import DEBUG, INFO, basicConfig, debug
-from io_soulworker.file_import.runner import FileImportRunner
-from io_soulworker.file_import.object_panel_default_values import FileImportObjectPanelDefaultValues
-from io_soulworker.file_import.object_panel_features import FileImportObjectPanelFeatures
+
+try:
+    import bpy
+except ModuleNotFoundError:
+    bpy = None
+
+if bpy is not None:
+    from io_soulworker.file_import.runner import FileImportRunner
+    from io_soulworker.file_import.object_panel_default_values import FileImportObjectPanelDefaultValues
+    from io_soulworker.file_import.object_panel_features import FileImportObjectPanelFeatures
 
 basicConfig(
     level=DEBUG if __debug__ else INFO,
@@ -14,30 +19,37 @@
     "name": "SoulWorker",
     "author": "sawich",
     "version": (1, 0, 0),
-    "blender": (4, 4, 0),
+    "blender": (5, 1, 1),
     "location": "File > Import/Export",
     "description": "Import-Export SoulWorker content",
     "support": "COMMUNITY",
     "category": "Import-Export",
 }
 
 
-classes = {
-    FileImportObjectPanelDefaultValues,
-    FileImportObjectPanelFeatures,
-    FileImportRunner,
-}
+classes = set()
+
+if bpy is not None:
+    classes = {
+        FileImportObjectPanelDefaultValues,
+        FileImportObjectPanelFeatures,
+        FileImportRunner,
+    }
 
 
 def menu_func_import(self, context):
+    if bpy is None:
+        return
 
     self.layout.operator(
         FileImportRunner.bl_idname,
-        text="SoulWorker (.model, .vmesh)"
+        text="SoulWorker (.model, .vmesh, .anim)"
     )
 
 
 def register():
+    if bpy is None:
+        return
 
     for cls in classes:
         bpy.utils.register_class(cls)
@@ -46,6 +58,8 @@ def register():
 
 
 def unregister():
+    if bpy is None:
+        return
 
     bpy.types.TOPBAR_MT_file_import.remove(menu_func_import)
 

io_soulworker/file_import/animation/action_builder.py

@@ -0,0 +1,29 @@
+from collections.abc import Iterable, Sequence
+from typing import Any
+
+
+VISION_FRAMES_PER_SECOND = 33
+
+
+def vision_time_to_frame(time: float) -> int:
+    return round(time * VISION_FRAMES_PER_SECOND) + 1
+
+
+def group_keyframes_by_bone(
+    keyframes: Iterable[Any],
+    bone_names: Sequence[str],
+    values_attr: str,
+) -> dict[str, list[tuple[int, Any]]]:
+    grouped = {name: [] for name in bone_names}
+
+    for keyframe in keyframes:
+        frame = vision_time_to_frame(keyframe.time)
+        values = getattr(keyframe, values_attr)
+
+        for bone_index, bone_name in enumerate(bone_names):
+            if bone_index >= len(values):
+                break
+
+            grouped[bone_name].append((frame, values[bone_index]))
+
+    return grouped

io_soulworker/file_import/animation/chunk_reader.py

@@ -3,6 +3,7 @@
 from io_soulworker.chunks.brot_chunk import BrotChunk
 from io_soulworker.chunks.xbsv_chunk import XbsvChunk
 from io_soulworker.chunks.head_chunk import HeadChunk
+from io_soulworker.chunks.skel_chunk import SkelChunk
 from io_soulworker.core.binary_reader import BinaryReader
 from io_soulworker.core.vis_chunk_file import VisChunkFileReader
 from io_soulworker.core.vis_chunk_id import VisChunkId
@@ -26,6 +27,12 @@ def on_bone_rotation(self, chunk: BrotChunk) -> None:
     def on_animation(self, skeleton_index: int, name: str) -> None:
         debug('Not impl callback')
 
+    def on_animation_end(self) -> None:
+        debug('Not impl callback')
+
+    def on_skeleton(self, chunk: SkelChunk) -> None:
+        debug('Not impl callback')
+
     def on_chunk_start(self, scope: VisChunkReaderScope, reader: BinaryReader) -> None:
 
         if scope.chunk == VisChunkId.HEAD:
@@ -50,6 +57,11 @@ def on_chunk_start(self, scope: VisChunkReaderScope, reader: BinaryReader) -> No
 
             self.on_animation(skeleton_index, name)
             self.run_sub(reader, scope)
+            self.on_animation_end()
+
+        elif scope.chunk == VisChunkId.SKEL:
+
+            self.on_skeleton(SkelChunk(reader))
 
         elif scope.chunk == VisChunkId.VSBX: