Yusarina
c31d25dd01
You can choose between errors, warning, info or full debug, errors will always log to ensure we don't have silent failures with debug on or off.
111 lines
3.5 KiB
Python
111 lines
3.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright 2014 MMD Tools authors
|
|
# This file was originally part of the MMD Tools add-on for Blender
|
|
# You can find MMD Tools here: https://github.com/MMD-Blender/blender_mmd_tools
|
|
# Neoneko has modified this file to work with Avatar Toolkit and may of made changes or improvements.
|
|
# MMD Tools is licensed under the terms of the GNU General Public License version 3 (GPLv3) same as Avatar Toolkit.
|
|
|
|
from typing import Set
|
|
|
|
import bpy
|
|
from bpy.types import Operator
|
|
|
|
from ..core.model import FnModel
|
|
from ..core.sdef import FnSDEF
|
|
|
|
|
|
def _get_target_objects(context):
|
|
root_objects: Set[bpy.types.Object] = set()
|
|
selected_objects: Set[bpy.types.Object] = set()
|
|
for i in context.selected_objects:
|
|
if i.type == "MESH":
|
|
selected_objects.add(i)
|
|
continue
|
|
|
|
root_object = FnModel.find_root_object(i)
|
|
if root_object is None:
|
|
continue
|
|
if root_object in root_objects:
|
|
continue
|
|
|
|
root_objects.add(root_object)
|
|
|
|
selected_objects |= set(FnModel.iterate_mesh_objects(root_object))
|
|
return selected_objects, root_objects
|
|
|
|
|
|
class ResetSDEFCache(Operator):
|
|
bl_idname = "mmd_tools.sdef_cache_reset"
|
|
bl_label = "Reset MMD SDEF cache"
|
|
bl_description = "Reset MMD SDEF cache of selected objects and clean unused cache"
|
|
bl_options = {"REGISTER", "UNDO", "INTERNAL"}
|
|
|
|
def execute(self, context):
|
|
target_meshes, _ = _get_target_objects(context)
|
|
for i in target_meshes:
|
|
FnSDEF.clear_cache(i)
|
|
FnSDEF.clear_cache(unused_only=True)
|
|
return {"FINISHED"}
|
|
|
|
|
|
class BindSDEF(Operator):
|
|
bl_idname = "mmd_tools.sdef_bind"
|
|
bl_label = "Bind SDEF Driver"
|
|
bl_description = "Bind MMD SDEF data of selected objects"
|
|
bl_options = {"REGISTER", "UNDO", "INTERNAL"}
|
|
|
|
mode: bpy.props.EnumProperty(
|
|
name="Mode",
|
|
description="Select mode",
|
|
items=[
|
|
("2", "Bulk", "Speed up with numpy (may be slower in some cases)", 2),
|
|
("1", "Normal", "Normal mode", 1),
|
|
("0", "- Auto -", "Select best mode by benchmark result", 0),
|
|
],
|
|
default="0",
|
|
)
|
|
use_skip: bpy.props.BoolProperty(
|
|
name="Skip",
|
|
description="Skip when the bones are not moving",
|
|
default=True,
|
|
)
|
|
use_scale: bpy.props.BoolProperty(
|
|
name="Scale",
|
|
description="Support bone scaling (slow)",
|
|
default=False,
|
|
)
|
|
|
|
def invoke(self, context, event):
|
|
vm = context.window_manager
|
|
return vm.invoke_props_dialog(self)
|
|
|
|
# TODO: Utility Functionalize
|
|
def execute(self, context):
|
|
target_meshes, root_objects = _get_target_objects(context)
|
|
|
|
for r in root_objects:
|
|
r.mmd_root.use_sdef = True
|
|
|
|
param = ((None, False, True)[int(self.mode)], self.use_skip, self.use_scale)
|
|
count = sum(FnSDEF.bind(i, *param) for i in target_meshes)
|
|
self.report({"INFO"}, f"Binded {count} of {len(target_meshes)} selected mesh(es)")
|
|
return {"FINISHED"}
|
|
|
|
|
|
class UnbindSDEF(Operator):
|
|
bl_idname = "mmd_tools.sdef_unbind"
|
|
bl_label = "Unbind SDEF Driver"
|
|
bl_description = "Unbind MMD SDEF data of selected objects"
|
|
bl_options = {"REGISTER", "UNDO", "INTERNAL"}
|
|
|
|
# TODO: Utility Functionalize
|
|
def execute(self, context):
|
|
target_meshes, root_objects = _get_target_objects(context)
|
|
for i in target_meshes:
|
|
FnSDEF.unbind(i)
|
|
|
|
for r in root_objects:
|
|
r.mmd_root.use_sdef = False
|
|
|
|
return {"FINISHED"}
|