Yusarina
236 lines
9.1 KiB
Python
236 lines
9.1 KiB
Python
import bpy
|
|
import logging
|
|
import os
|
|
import pathlib
|
|
import typing
|
|
from bpy.types import Operator, Context
|
|
from bpy_extras.io_utils import ImportHelper
|
|
from typing import Optional, Callable, Dict, List, Union, Set
|
|
from ..common import clear_default_objects
|
|
from ..translations import t
|
|
from ..mmd.core.pmx.importer import PMXImporter
|
|
|
|
# Configure logging
|
|
logging.basicConfig(level=logging.INFO)
|
|
logger: logging.Logger = logging.getLogger(__name__)
|
|
|
|
import importlib.util
|
|
|
|
if importlib.util.find_spec("io_scene_valvesource") is not None:
|
|
from io_scene_valvesource.import_smd import SmdImporter
|
|
|
|
class ImportProgress:
|
|
"""Tracks and logs the progress of multi-file imports"""
|
|
def __init__(self, total_files: int):
|
|
self.total: int = total_files
|
|
self.current: int = 0
|
|
|
|
def update(self, filename: str) -> None:
|
|
"""Update import progress and log current file"""
|
|
self.current += 1
|
|
logger.info(f"Importing {filename} ({self.current}/{self.total})")
|
|
|
|
def validate_file(filepath: str) -> bool:
|
|
"""
|
|
Validate if a file exists and is accessible
|
|
Returns: True if file is valid, False otherwise
|
|
"""
|
|
if not os.path.exists(filepath):
|
|
logger.error(f"File not found: {filepath}")
|
|
return False
|
|
if not os.path.isfile(filepath):
|
|
logger.error(f"Not a file: {filepath}")
|
|
return False
|
|
return True
|
|
|
|
def import_multi_files(
|
|
method: Optional[Callable] = None,
|
|
directory: Optional[str] = None,
|
|
files: Optional[List[Dict[str, str]]] = None,
|
|
filepath: str = "",
|
|
progress_callback: Optional[Callable[[str], None]] = None
|
|
) -> None:
|
|
"""
|
|
Import multiple files using the specified import method
|
|
|
|
Args:
|
|
method: Import method to use
|
|
directory: Directory containing files
|
|
files: List of files to import
|
|
filepath: Single file path to import
|
|
progress_callback: Callback for progress updates
|
|
"""
|
|
try:
|
|
if not method:
|
|
raise ValueError("Import method not specified")
|
|
|
|
if not files:
|
|
if not validate_file(filepath):
|
|
return
|
|
method(directory, filepath)
|
|
if progress_callback:
|
|
progress_callback(filepath)
|
|
else:
|
|
progress = ImportProgress(len(files))
|
|
for file in files:
|
|
fullpath: str = os.path.join(directory, os.path.basename(file["name"]))
|
|
if not validate_file(fullpath):
|
|
continue
|
|
|
|
logger.info(f"Importing file: {fullpath}")
|
|
method(directory, fullpath)
|
|
|
|
if progress_callback:
|
|
progress_callback(fullpath)
|
|
progress.update(file["name"])
|
|
|
|
except Exception as e:
|
|
logger.error(f"Import failed: {str(e)}", exc_info=True)
|
|
raise
|
|
|
|
ImportMethod = Callable[[str, List[Dict[str, str]], str], None]
|
|
|
|
import_types: Dict[str, ImportMethod] = {
|
|
"fbx": lambda directory, files, filepath: bpy.ops.import_scene.fbx(
|
|
files=files, directory=directory, filepath=filepath,
|
|
automatic_bone_orientation=False, use_prepost_rot=False, use_anim=False
|
|
),
|
|
"pmx": lambda directory, files, filepath: import_multi_files(
|
|
directory=directory,
|
|
files=files,
|
|
filepath=filepath,
|
|
method=lambda directory, filepath: import_pmx_file(filepath)
|
|
),
|
|
"smd": lambda directory, files, filepath: eval("bpy."+SmdImporter.bl_idname+".(files=files, directory=directory, filepath=filepath)"),
|
|
"dmx": lambda directory, files, filepath: eval("bpy."+SmdImporter.bl_idname+".(files=files, directory=directory, filepath=filepath)"),
|
|
"gltf": lambda directory, files, filepath: bpy.ops.import_scene.gltf(files=files, filepath=filepath),
|
|
"glb": lambda directory, files, filepath: bpy.ops.import_scene.gltf(files=files, filepath=filepath),
|
|
"qc": lambda directory, files, filepath: eval("bpy."+SmdImporter.bl_idname+".(files=files, directory=directory, filepath=filepath)"),
|
|
"obj": lambda directory, files, filepath: bpy.ops.wm.obj_import(files=files, directory=directory, filepath=filepath),
|
|
"dae": lambda directory, files, filepath: import_multi_files(
|
|
directory=directory,
|
|
files=files,
|
|
filepath=filepath,
|
|
method=lambda directory, filepath: bpy.ops.wm.collada_import(
|
|
filepath=filepath, auto_connect=True, find_chains=True, fix_orientation=True
|
|
)
|
|
),
|
|
"3ds": lambda directory, files, filepath: bpy.ops.import_scene.max3ds(files=files, directory=directory, filepath=filepath),
|
|
"stl": lambda directory, files, filepath: bpy.ops.import_mesh.stl(files=files, directory=directory, filepath=filepath),
|
|
"mtl": lambda directory, files, filepath: bpy.ops.wm.obj_import(files=files, directory=directory, filepath=filepath),
|
|
"x3d": lambda directory, files, filepath: bpy.ops.import_scene.x3d(files=files, directory=directory, filepath=filepath),
|
|
"wrl": lambda directory, files, filepath: bpy.ops.import_scene.x3d(files=files, directory=directory, filepath=filepath),
|
|
"vmd": lambda directory, files, filepath: import_multi_files(
|
|
directory=directory,
|
|
files=files,
|
|
filepath=filepath,
|
|
method=lambda directory, filepath: bpy.ops.tuxedo.import_mmd_animation(directory=directory, filepath=filepath)
|
|
),
|
|
"vrm": lambda directory, files, filepath: bpy.ops.import_scene.vrm(filepath=filepath),
|
|
"animx": (lambda directory, files, filepath : bpy.ops.avatar_toolkit.animx_importer(directory=directory,files=files,filepath=filepath)),
|
|
}
|
|
|
|
def concat_imports_filter(imports: Dict[str, ImportMethod]) -> str:
|
|
"""Create a file filter string from import types"""
|
|
return "".join(f"*.{importer};" for importer in imports.keys())
|
|
|
|
imports: str = concat_imports_filter(import_types)
|
|
|
|
|
|
class AvatarToolKit_OT_Import(Operator, ImportHelper):
|
|
"""Import files into Blender with Avatar Toolkit settings"""
|
|
bl_idname: str = "avatar_toolkit.import"
|
|
bl_label: str = t("QuickAccess.import")
|
|
|
|
files: bpy.props.CollectionProperty(
|
|
type=bpy.types.OperatorFileListElement,
|
|
options={'HIDDEN', 'SKIP_SAVE'}
|
|
)
|
|
|
|
filter_glob: bpy.props.StringProperty(
|
|
default=imports,
|
|
options={'HIDDEN', 'SKIP_SAVE'}
|
|
)
|
|
|
|
directory: bpy.props.StringProperty(
|
|
maxlen=1024,
|
|
subtype='FILE_PATH',
|
|
options={'HIDDEN', 'SKIP_SAVE'}
|
|
)
|
|
|
|
def execute(self, context: Context) -> Set[str]:
|
|
clear_default_objects()
|
|
|
|
file_grouping_dict: Dict[str, List[Dict[str, str]]] = {}
|
|
is_multi = len(self.files) > 0
|
|
|
|
if is_multi:
|
|
for file in self.files:
|
|
fullpath = os.path.join(self.directory, os.path.basename(file.name))
|
|
ext = pathlib.Path(fullpath).suffix.replace(".", "")
|
|
|
|
if ext not in file_grouping_dict:
|
|
file_grouping_dict[ext] = []
|
|
file_grouping_dict[ext].append({"name": os.path.basename(file.name)})
|
|
else:
|
|
fullpath = os.path.join(os.path.dirname(self.filepath), os.path.basename(self.filepath))
|
|
ext = pathlib.Path(fullpath).suffix.replace(".", "")
|
|
|
|
if ext not in file_grouping_dict:
|
|
file_grouping_dict[ext] = []
|
|
file_grouping_dict[ext].append({"name": fullpath})
|
|
|
|
for file_group_name, files in file_grouping_dict.items():
|
|
try:
|
|
if file_group_name == "vrm" and not hasattr(bpy.ops.import_scene, "vrm"):
|
|
bpy.ops.wm.vrm_importer_popup('INVOKE_DEFAULT')
|
|
return {'CANCELLED'}
|
|
|
|
directory = self.directory if self.directory else ""
|
|
import_types[file_group_name](directory, files, self.filepath)
|
|
|
|
except AttributeError as e:
|
|
if file_group_name == "vrm":
|
|
bpy.ops.wm.vrm_importer_popup('INVOKE_DEFAULT')
|
|
else:
|
|
self.report({'ERROR'}, t('Importing.need_importer').format(extension=file_group_name))
|
|
logger.error(f"Importer error: {e}")
|
|
return {'CANCELLED'}
|
|
|
|
self.report({'INFO'}, t('Quick_Access.import_success'))
|
|
return {'FINISHED'}
|
|
|
|
def import_pmx_file(filepath: str) -> None:
|
|
"""
|
|
Import a PMX file using the MMD Tools PMXImporter
|
|
|
|
Args:
|
|
filepath: Path to the PMX file
|
|
"""
|
|
|
|
# Default import settings
|
|
import_settings = {
|
|
"filepath": filepath,
|
|
"scale": 0.08,
|
|
"types": {"MESH", "ARMATURE", "MORPHS", "DISPLAY"},
|
|
"clean_model": True,
|
|
"remove_doubles": False,
|
|
"fix_IK_links": True,
|
|
"ik_loop_factor": 3,
|
|
"use_mipmap": True,
|
|
"sph_blend_factor": 1.0,
|
|
"spa_blend_factor": 1.0,
|
|
"rename_LR_bones": False,
|
|
"use_underscore": False,
|
|
"apply_bone_fixed_axis": False,
|
|
}
|
|
|
|
# Create and execute the importer
|
|
importer = PMXImporter()
|
|
try:
|
|
importer.execute(**import_settings)
|
|
logger.info(f"Successfully imported PMX file: {filepath}")
|
|
except Exception as e:
|
|
logger.error(f"Failed to import PMX file: {str(e)}", exc_info=True)
|
|
raise
|