|
|
import os |
|
|
from glob import glob |
|
|
|
|
|
import bpy |
|
|
from tqdm import tqdm |
|
|
|
|
|
from utils import HiddenPrints, get_all_armature_obj, get_all_mesh_obj, load_file, remove_all, select_objs, update |
|
|
|
|
|
|
|
|
def rename_mixamo_bone(armature_obj): |
|
|
"""Replace name like 'mixamorig10:xxx' to 'mixamorig:xxx, so that character can be correctly animated.'""" |
|
|
import re |
|
|
|
|
|
pattern = re.compile(r"mixamorig[0-9]+:") |
|
|
for bone in armature_obj.data.bones: |
|
|
bone.name = re.sub(pattern, "mixamorig:", bone.name) |
|
|
return armature_obj |
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
input_dir = "character_rabit" |
|
|
output_dir = "character_rabit_refined" |
|
|
os.makedirs(output_dir, exist_ok=True) |
|
|
character_list = sorted(glob(os.path.join(input_dir, "*.fbx"))) |
|
|
|
|
|
for file in tqdm(character_list, dynamic_ncols=True): |
|
|
with HiddenPrints(): |
|
|
remove_all() |
|
|
obj_list = load_file(file) |
|
|
assert len(get_all_armature_obj(obj_list)) == 1, "Armature number is not 1" |
|
|
bpy.ops.wm.save_as_mainfile(filepath="test.blend") |
|
|
|
|
|
select_objs(get_all_mesh_obj(obj_list), deselect_first=True) |
|
|
bpy.ops.object.transform_apply(location=True, rotation=True, scale=True) |
|
|
|
|
|
update() |
|
|
select_objs(obj_list, deselect_first=True) |
|
|
bpy.ops.export_scene.fbx( |
|
|
filepath=os.path.join(output_dir, os.path.basename(file)), |
|
|
check_existing=False, |
|
|
use_selection=True, |
|
|
use_triangles=True, |
|
|
add_leaf_bones=False, |
|
|
bake_anim=False, |
|
|
|
|
|
|
|
|
) |
|
|
|
