✨ feat: load from pre-animated files

.gitmodules

@@ -0,0 +1,3 @@
+[submodule "auto_rig_pro"]
+	path = auto_rig_pro
+	url = https://github.com/jasongzy/auto_rig_pro

apply_animation.py

@@ -0,0 +1,38 @@
+import os
+from glob import glob
+
+from tqdm import tqdm
+
+from utils import HiddenPrints, bpy, load_mixamo_anim, remove_all, reset, select_objs, update
+
+if __name__ == "__main__":
+    character_dir = "./character_vroid_refined"
+    animation_dir = "./animation"
+    output_dir = "./animated_vroid"
+    os.makedirs(output_dir, exist_ok=True)
+
+    reset()
+    character_list = sorted(glob(os.path.join(character_dir, "*.fbx")))
+    assert character_list
+    animation_list = sorted(glob(os.path.join(animation_dir, "*.fbx")))
+    assert animation_list
+    for char_file in tqdm(character_list, dynamic_ncols=True, desc="Character"):
+        for anim_file in tqdm(animation_list, dynamic_ncols=True, leave=False, desc="Animation"):
+            with HiddenPrints():
+                remove_all()
+                objs = load_mixamo_anim(char_file, anim_file, do_retarget=True, inplace=False)
+                update()
+                get_base_name = lambda s: os.path.splitext(os.path.basename(s))[0]
+                output_filename = f"{get_base_name(char_file)}-{get_base_name(anim_file)}"
+                # bpy.ops.wm.save_as_mainfile(filepath=os.path.join(output_dir, f"{output_filename}.blend"))
+                select_objs(objs, deselect_first=True)
+                bpy.ops.export_scene.fbx(
+                    filepath=os.path.join(output_dir, f"{output_filename}.fbx"),
+                    check_existing=False,
+                    use_selection=True,
+                    use_triangles=True,
+                    add_leaf_bones=False,
+                    bake_anim=True,
+                    # path_mode="COPY",
+                    # embed_textures=True,
+                )

auto_rig_pro

@@ -0,0 +1 @@
+Subproject commit 614be17f80cda54f58f378b7bd28f4b4c1ba4ec5

utils.py

@@ -242,6 +242,13 @@ def remove_empty_vgroups(mesh_obj_list: "list[Object]"):
             vertex_groups.remove(vertex_groups[i])
 
 
+def set_action(armature_obj: Object, action: Action):
+    if not armature_obj.animation_data:
+        armature_obj.animation_data_create()
+    armature_obj.animation_data.action = action
+    return armature_obj
+
+
 def mesh_quads2tris(obj_list: "list[Object]" = None):
     if not obj_list:
         obj_list = bpy.context.scene.objects
@@ -249,3 +256,61 @@ def mesh_quads2tris(obj_list: "list[Object]" = None):
         if obj.type == "MESH":
             with Mode("EDIT", obj):
                 bpy.ops.mesh.quads_convert_to_tris(quad_method="BEAUTY", ngon_method="BEAUTY")
+
+
+def get_enabled_addons() -> "list[str]":
+    return [x.module for x in bpy.context.preferences.addons]
+
+
+def enable_arp(armature_obj: Object, addon_path=os.path.join(os.path.dirname(__file__), "auto_rig_pro")):
+    import sys
+
+    assert os.path.isfile(os.path.join(addon_path, "__init__.py")), "Auto-Rig Pro not found"
+    dirname, addon_name = os.path.split(addon_path)
+    if addon_name in get_enabled_addons():
+        return
+    sys.path.insert(0, dirname)
+    with Mode("POSE", armature_obj):
+        # import addon_utils
+        # addon_utils.enable(addon_name)
+        bpy.ops.preferences.addon_enable(module=addon_name)
+
+
+def retarget(source_armature: Object, target_armature: Object, inplace=False):
+    enable_arp(target_armature)
+    scn = bpy.context.scene
+    scn.source_rig = source_armature.name
+    if inplace:
+        scn.arp_retarget_in_place = True
+    scn.target_rig = target_armature.name
+    bpy.ops.arp.auto_scale()
+    bpy.ops.arp.build_bones_list()
+    hips = scn.bones_map_v2["mixamorig:Hips"]
+    scn.bones_map_index = list(scn.bones_map_v2).index(hips)
+    hips.set_as_root = True
+    bpy.ops.arp.retarget()
+    return target_armature
+
+
+def load_mixamo_anim(char_file: str, anim_file: str, do_retarget=False, inplace=False, to_tris=False):
+    char_objs = load_file(char_file) if isinstance(char_file, str) else char_file
+    char_armature = get_armature_obj(char_objs)
+
+    anim_objs = load_file(anim_file)
+    anim_armature = get_armature_obj(anim_objs)
+    print(anim_armature)
+    print(anim_armature.animation_data)
+    assert anim_armature.animation_data is not None and len(bpy.data.actions) > 0, f"Animation not found in {anim_file}"
+
+    set_action(char_armature, anim_armature.animation_data.action)
+    if do_retarget:
+        retarget(anim_armature, char_armature, inplace=inplace)
+        for action in bpy.data.actions:
+            if action is not char_armature.animation_data.action:
+                bpy.data.actions.remove(action, do_unlink=True)
+    for obj in anim_objs:
+        bpy.data.objects.remove(obj, do_unlink=True)
+
+    if to_tris:
+        mesh_quads2tris(char_objs)
+    return char_objs