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.
578 lines
19 KiB
Python
578 lines
19 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.
|
|
|
|
"""Properties for MMD model root object"""
|
|
|
|
import bpy
|
|
|
|
from .. import utils
|
|
from ..bpyutils import FnContext
|
|
from ..core.material import FnMaterial
|
|
from ..core.model import FnModel
|
|
from ..core.sdef import FnSDEF
|
|
from . import patch_library_overridable
|
|
from .morph import BoneMorph, GroupMorph, MaterialMorph, UVMorph, VertexMorph
|
|
from .translations import MMDTranslation
|
|
|
|
|
|
def __driver_variables(constraint: bpy.types.Constraint, path: str, index=-1):
|
|
d = constraint.driver_add(path, index)
|
|
variables = d.driver.variables
|
|
for x in variables:
|
|
variables.remove(x)
|
|
return d.driver, variables
|
|
|
|
|
|
def __add_single_prop(variables, id_obj, data_path, prefix):
|
|
var = variables.new()
|
|
var.name = prefix + str(len(variables))
|
|
var.type = "SINGLE_PROP"
|
|
target = var.targets[0]
|
|
target.id_type = "OBJECT"
|
|
target.id = id_obj
|
|
target.data_path = data_path
|
|
return var
|
|
|
|
|
|
def _toggleUsePropertyDriver(self: "MMDRoot", _context):
|
|
root_object: bpy.types.Object = self.id_data
|
|
armature_object = FnModel.find_armature_object(root_object)
|
|
|
|
if armature_object is None:
|
|
ik_map = {}
|
|
else:
|
|
bones = armature_object.pose.bones
|
|
ik_map = {bones[c.subtarget]: (b, c) for b in bones for c in b.constraints if c.type == "IK" and c.is_valid and c.subtarget in bones}
|
|
|
|
if self.use_property_driver:
|
|
for ik, (b, c) in ik_map.items():
|
|
driver, variables = __driver_variables(c, "influence")
|
|
driver.expression = "%s" % __add_single_prop(variables, ik.id_data, ik.path_from_id("mmd_ik_toggle"), "use_ik").name
|
|
b = b if c.use_tail else b.parent
|
|
for b in ([b] + b.parent_recursive)[: c.chain_count]:
|
|
c = next((c for c in b.constraints if c.type == "LIMIT_ROTATION" and not c.mute), None)
|
|
if c:
|
|
driver, variables = __driver_variables(c, "influence")
|
|
driver.expression = "%s" % __add_single_prop(variables, ik.id_data, ik.path_from_id("mmd_ik_toggle"), "use_ik").name
|
|
for i in FnModel.iterate_mesh_objects(root_object):
|
|
for prop_hide in ("hide_viewport", "hide_render"):
|
|
driver, variables = __driver_variables(i, prop_hide)
|
|
driver.expression = "not %s" % __add_single_prop(variables, root_object, "mmd_root.show_meshes", "show").name
|
|
else:
|
|
for ik, (b, c) in ik_map.items():
|
|
c.driver_remove("influence")
|
|
b = b if c.use_tail else b.parent
|
|
for b in ([b] + b.parent_recursive)[: c.chain_count]:
|
|
c = next((c for c in b.constraints if c.type == "LIMIT_ROTATION" and not c.mute), None)
|
|
if c:
|
|
c.driver_remove("influence")
|
|
for i in FnModel.iterate_mesh_objects(root_object):
|
|
for prop_hide in ("hide_viewport", "hide_render"):
|
|
i.driver_remove(prop_hide)
|
|
|
|
|
|
# ===========================================
|
|
# Callback functions
|
|
# ===========================================
|
|
|
|
|
|
def _toggleUseToonTexture(self: "MMDRoot", _context):
|
|
use_toon = self.use_toon_texture
|
|
for i in FnModel.iterate_mesh_objects(self.id_data):
|
|
for m in i.data.materials:
|
|
if m:
|
|
FnMaterial(m).use_toon_texture(use_toon)
|
|
|
|
|
|
def _toggleUseSphereTexture(self: "MMDRoot", _context):
|
|
use_sphere = self.use_sphere_texture
|
|
for i in FnModel.iterate_mesh_objects(self.id_data):
|
|
for m in i.data.materials:
|
|
if m:
|
|
FnMaterial(m).use_sphere_texture(use_sphere, i)
|
|
|
|
|
|
def _toggleUseSDEF(self: "MMDRoot", _context):
|
|
mute_sdef = not self.use_sdef
|
|
for i in FnModel.iterate_mesh_objects(self.id_data):
|
|
FnSDEF.mute_sdef_set(i, mute_sdef)
|
|
|
|
|
|
def _toggleVisibilityOfMeshes(self: "MMDRoot", context: bpy.types.Context):
|
|
root = self.id_data
|
|
hide = not self.show_meshes
|
|
for i in FnModel.iterate_mesh_objects(self.id_data):
|
|
i.hide_set(hide)
|
|
i.hide_render = hide
|
|
if hide and context.active_object is None:
|
|
FnContext.set_active_object(context, root)
|
|
|
|
|
|
def _toggleVisibilityOfRigidBodies(self: "MMDRoot", context: bpy.types.Context):
|
|
root = self.id_data
|
|
hide = not self.show_rigid_bodies
|
|
for i in FnModel.iterate_rigid_body_objects(root):
|
|
i.hide_set(hide)
|
|
if hide and context.active_object is None:
|
|
FnContext.set_active_object(context, root)
|
|
|
|
|
|
def _toggleVisibilityOfJoints(self: "MMDRoot", context):
|
|
root_object = self.id_data
|
|
hide = not self.show_joints
|
|
for i in FnModel.iterate_joint_objects(root_object):
|
|
i.hide_set(hide)
|
|
if hide and context.active_object is None:
|
|
FnContext.set_active_object(context, root_object)
|
|
|
|
|
|
def _toggleVisibilityOfTemporaryObjects(self: "MMDRoot", context: bpy.types.Context):
|
|
root_object: bpy.types.Object = self.id_data
|
|
hide = not self.show_temporary_objects
|
|
with FnContext.temp_override_active_layer_collection(context, root_object):
|
|
for i in FnModel.iterate_temporary_objects(root_object):
|
|
i.hide_set(hide)
|
|
if hide and context.active_object is None:
|
|
FnContext.set_active_object(context, root_object)
|
|
|
|
|
|
def _toggleShowNamesOfRigidBodies(self: "MMDRoot", _context):
|
|
root = self.id_data
|
|
show_names = root.mmd_root.show_names_of_rigid_bodies
|
|
for i in FnModel.iterate_rigid_body_objects(root):
|
|
i.show_name = show_names
|
|
|
|
|
|
def _toggleShowNamesOfJoints(self: "MMDRoot", _context):
|
|
root = self.id_data
|
|
show_names = root.mmd_root.show_names_of_joints
|
|
for i in FnModel.iterate_joint_objects(root):
|
|
i.show_name = show_names
|
|
|
|
|
|
def _setVisibilityOfMMDRigArmature(prop: "MMDRoot", v: bool):
|
|
root = prop.id_data
|
|
arm = FnModel.find_armature_object(root)
|
|
if arm is None:
|
|
return
|
|
if not v and bpy.context.active_object == arm:
|
|
FnContext.set_active_object(bpy.context, root)
|
|
arm.hide_set(not v)
|
|
|
|
|
|
def _getVisibilityOfMMDRigArmature(prop: "MMDRoot"):
|
|
if prop.id_data.mmd_type != "ROOT":
|
|
return False
|
|
arm = FnModel.find_armature_object(prop.id_data)
|
|
return arm and not arm.hide_get()
|
|
|
|
|
|
def _setActiveRigidbodyObject(prop: "MMDRoot", v: int):
|
|
obj = FnContext.get_scene_objects(bpy.context)[v]
|
|
if FnModel.is_rigid_body_object(obj):
|
|
FnContext.set_active_and_select_single_object(bpy.context, obj)
|
|
prop["active_rigidbody_object_index"] = v
|
|
|
|
|
|
def _getActiveRigidbodyObject(prop: "MMDRoot"):
|
|
context = bpy.context
|
|
active_obj = FnContext.get_active_object(context)
|
|
if FnModel.is_rigid_body_object(active_obj):
|
|
prop["active_rigidbody_object_index"] = FnContext.get_scene_objects(context).find(active_obj.name)
|
|
return prop.get("active_rigidbody_object_index", 0)
|
|
|
|
|
|
def _setActiveJointObject(prop: "MMDRoot", v: int):
|
|
obj = FnContext.get_scene_objects(bpy.context)[v]
|
|
if FnModel.is_joint_object(obj):
|
|
FnContext.set_active_and_select_single_object(bpy.context, obj)
|
|
prop["active_joint_object_index"] = v
|
|
|
|
|
|
def _getActiveJointObject(prop: "MMDRoot"):
|
|
context = bpy.context
|
|
active_obj = FnContext.get_active_object(context)
|
|
if FnModel.is_joint_object(active_obj):
|
|
prop["active_joint_object_index"] = FnContext.get_scene_objects(context).find(active_obj.name)
|
|
return prop.get("active_joint_object_index", 0)
|
|
|
|
|
|
def _setActiveMorph(prop: "MMDRoot", v: bool):
|
|
if "active_morph_indices" not in prop:
|
|
prop["active_morph_indices"] = [0] * 5
|
|
prop["active_morph_indices"][prop.get("active_morph_type", 3)] = v
|
|
|
|
|
|
def _getActiveMorph(prop: "MMDRoot"):
|
|
if "active_morph_indices" in prop:
|
|
return prop["active_morph_indices"][prop.get("active_morph_type", 3)]
|
|
return 0
|
|
|
|
|
|
def _setActiveMeshObject(prop: "MMDRoot", v: int):
|
|
obj = FnContext.get_scene_objects(bpy.context)[v]
|
|
if FnModel.is_mesh_object(obj):
|
|
FnContext.set_active_and_select_single_object(bpy.context, obj)
|
|
prop["active_mesh_index"] = v
|
|
|
|
|
|
def _getActiveMeshObject(prop: "MMDRoot"):
|
|
context = bpy.context
|
|
active_obj = FnContext.get_active_object(context)
|
|
if FnModel.is_mesh_object(active_obj):
|
|
prop["active_mesh_index"] = FnContext.get_scene_objects(context).find(active_obj.name)
|
|
return prop.get("active_mesh_index", -1)
|
|
|
|
|
|
# ===========================================
|
|
# Property classes
|
|
# ===========================================
|
|
|
|
|
|
class MMDDisplayItem(bpy.types.PropertyGroup):
|
|
"""PMX 表示項目(表示枠内の1項目)"""
|
|
|
|
type: bpy.props.EnumProperty(
|
|
name="Type",
|
|
description="Select item type",
|
|
items=[
|
|
("BONE", "Bone", "", 1),
|
|
("MORPH", "Morph", "", 2),
|
|
],
|
|
)
|
|
|
|
morph_type: bpy.props.EnumProperty(
|
|
name="Morph Type",
|
|
description="Select morph type",
|
|
items=[
|
|
("material_morphs", "Material", "Material Morphs", 0),
|
|
("uv_morphs", "UV", "UV Morphs", 1),
|
|
("bone_morphs", "Bone", "Bone Morphs", 2),
|
|
("vertex_morphs", "Vertex", "Vertex Morphs", 3),
|
|
("group_morphs", "Group", "Group Morphs", 4),
|
|
],
|
|
default="vertex_morphs",
|
|
)
|
|
|
|
|
|
class MMDDisplayItemFrame(bpy.types.PropertyGroup):
|
|
"""PMX 表示枠
|
|
|
|
PMXファイル内では表示枠がリストで格納されています。
|
|
"""
|
|
|
|
name_e: bpy.props.StringProperty(
|
|
name="Name(Eng)",
|
|
description="English Name",
|
|
default="",
|
|
)
|
|
|
|
# 特殊枠フラグ
|
|
# 特殊枠はファイル仕様上の固定枠(削除、リネーム不可)
|
|
is_special: bpy.props.BoolProperty(
|
|
name="Special",
|
|
description="Is special",
|
|
default=False,
|
|
)
|
|
|
|
# 表示項目のリスト
|
|
data: bpy.props.CollectionProperty(
|
|
name="Display Items",
|
|
type=MMDDisplayItem,
|
|
)
|
|
|
|
# 現在アクティブな項目のインデックス
|
|
active_item: bpy.props.IntProperty(
|
|
name="Active Display Item",
|
|
min=0,
|
|
default=0,
|
|
)
|
|
|
|
|
|
class MMDRoot(bpy.types.PropertyGroup):
|
|
"""MMDモデルデータ
|
|
|
|
モデルルート用に作成されたEmtpyオブジェクトで使用します
|
|
"""
|
|
|
|
name: bpy.props.StringProperty(
|
|
name="Name",
|
|
description="The name of the MMD model",
|
|
default="",
|
|
)
|
|
|
|
name_e: bpy.props.StringProperty(
|
|
name="Name (English)",
|
|
description="The english name of the MMD model",
|
|
default="",
|
|
)
|
|
|
|
comment_text: bpy.props.StringProperty(
|
|
name="Comment",
|
|
description="The text datablock of the comment",
|
|
default="",
|
|
)
|
|
|
|
comment_e_text: bpy.props.StringProperty(
|
|
name="Comment (English)",
|
|
description="The text datablock of the english comment",
|
|
default="",
|
|
)
|
|
|
|
ik_loop_factor: bpy.props.IntProperty(
|
|
name="MMD IK Loop Factor",
|
|
description="Scaling factor of MMD IK loop",
|
|
min=1,
|
|
soft_max=10,
|
|
max=100,
|
|
default=1,
|
|
)
|
|
|
|
# TODO: Replace to driver for NLA
|
|
show_meshes: bpy.props.BoolProperty(
|
|
name="Show Meshes",
|
|
description="Show all meshes of the MMD model",
|
|
# get=_show_meshes_get,
|
|
# set=_show_meshes_set,
|
|
update=_toggleVisibilityOfMeshes,
|
|
default=True,
|
|
)
|
|
|
|
show_rigid_bodies: bpy.props.BoolProperty(
|
|
name="Show Rigid Bodies",
|
|
description="Show all rigid bodies of the MMD model",
|
|
update=_toggleVisibilityOfRigidBodies,
|
|
)
|
|
|
|
show_joints: bpy.props.BoolProperty(
|
|
name="Show Joints",
|
|
description="Show all joints of the MMD model",
|
|
update=_toggleVisibilityOfJoints,
|
|
)
|
|
|
|
show_temporary_objects: bpy.props.BoolProperty(
|
|
name="Show Temps",
|
|
description="Show all temporary objects of the MMD model",
|
|
update=_toggleVisibilityOfTemporaryObjects,
|
|
)
|
|
|
|
show_armature: bpy.props.BoolProperty(
|
|
name="Show Armature",
|
|
description="Show the armature object of the MMD model",
|
|
get=_getVisibilityOfMMDRigArmature,
|
|
set=_setVisibilityOfMMDRigArmature,
|
|
)
|
|
|
|
show_names_of_rigid_bodies: bpy.props.BoolProperty(
|
|
name="Show Rigid Body Names",
|
|
description="Show rigid body names",
|
|
update=_toggleShowNamesOfRigidBodies,
|
|
)
|
|
|
|
show_names_of_joints: bpy.props.BoolProperty(
|
|
name="Show Joint Names",
|
|
description="Show joint names",
|
|
update=_toggleShowNamesOfJoints,
|
|
)
|
|
|
|
use_toon_texture: bpy.props.BoolProperty(
|
|
name="Use Toon Texture",
|
|
description="Use toon texture",
|
|
update=_toggleUseToonTexture,
|
|
default=True,
|
|
)
|
|
|
|
use_sphere_texture: bpy.props.BoolProperty(
|
|
name="Use Sphere Texture",
|
|
description="Use sphere texture",
|
|
update=_toggleUseSphereTexture,
|
|
default=True,
|
|
)
|
|
|
|
use_sdef: bpy.props.BoolProperty(
|
|
name="Use SDEF",
|
|
description="Use SDEF",
|
|
update=_toggleUseSDEF,
|
|
default=True,
|
|
)
|
|
|
|
use_property_driver: bpy.props.BoolProperty(
|
|
name="Use Property Driver",
|
|
description="Setup drivers for MMD property animation (Visibility and IK toggles)",
|
|
update=_toggleUsePropertyDriver,
|
|
default=False,
|
|
)
|
|
|
|
is_built: bpy.props.BoolProperty(
|
|
name="Is Built",
|
|
)
|
|
|
|
active_rigidbody_index: bpy.props.IntProperty(
|
|
name="Active Rigidbody Index",
|
|
min=0,
|
|
get=_getActiveRigidbodyObject,
|
|
set=_setActiveRigidbodyObject,
|
|
)
|
|
|
|
active_joint_index: bpy.props.IntProperty(
|
|
name="Active Joint Index",
|
|
min=0,
|
|
get=_getActiveJointObject,
|
|
set=_setActiveJointObject,
|
|
)
|
|
|
|
# *************************
|
|
# Display Items
|
|
# *************************
|
|
display_item_frames: bpy.props.CollectionProperty(
|
|
name="Display Frames",
|
|
type=MMDDisplayItemFrame,
|
|
)
|
|
|
|
active_display_item_frame: bpy.props.IntProperty(
|
|
name="Active Display Item Frame",
|
|
min=0,
|
|
default=0,
|
|
)
|
|
|
|
# *************************
|
|
# Morph
|
|
# *************************
|
|
material_morphs: bpy.props.CollectionProperty(
|
|
name="Material Morphs",
|
|
type=MaterialMorph,
|
|
)
|
|
uv_morphs: bpy.props.CollectionProperty(
|
|
name="UV Morphs",
|
|
type=UVMorph,
|
|
)
|
|
bone_morphs: bpy.props.CollectionProperty(
|
|
name="Bone Morphs",
|
|
type=BoneMorph,
|
|
)
|
|
vertex_morphs: bpy.props.CollectionProperty(name="Vertex Morphs", type=VertexMorph)
|
|
group_morphs: bpy.props.CollectionProperty(
|
|
name="Group Morphs",
|
|
type=GroupMorph,
|
|
)
|
|
active_morph_type: bpy.props.EnumProperty(
|
|
name="Active Morph Type",
|
|
description="Select current morph type",
|
|
items=[
|
|
("material_morphs", "Material", "Material Morphs", 0),
|
|
("uv_morphs", "UV", "UV Morphs", 1),
|
|
("bone_morphs", "Bone", "Bone Morphs", 2),
|
|
("vertex_morphs", "Vertex", "Vertex Morphs", 3),
|
|
("group_morphs", "Group", "Group Morphs", 4),
|
|
],
|
|
default="vertex_morphs",
|
|
)
|
|
active_morph: bpy.props.IntProperty(
|
|
name="Active Morph",
|
|
min=0,
|
|
set=_setActiveMorph,
|
|
get=_getActiveMorph,
|
|
)
|
|
morph_panel_show_settings: bpy.props.BoolProperty(
|
|
name="Morph Panel Show Settings",
|
|
description="Show Morph Settings",
|
|
default=True,
|
|
)
|
|
active_mesh_index: bpy.props.IntProperty(
|
|
name="Active Mesh",
|
|
min=0,
|
|
set=_setActiveMeshObject,
|
|
get=_getActiveMeshObject,
|
|
)
|
|
|
|
# *************************
|
|
# Translation
|
|
# *************************
|
|
translation: bpy.props.PointerProperty(
|
|
name="Translation",
|
|
type=MMDTranslation,
|
|
)
|
|
|
|
@staticmethod
|
|
def __get_select(prop: bpy.types.Object) -> bool:
|
|
utils.warn_deprecation("Object.select", "v4.0.0", "Use Object.select_get() method instead")
|
|
return prop.select_get()
|
|
|
|
@staticmethod
|
|
def __set_select(prop: bpy.types.Object, value: bool) -> None:
|
|
utils.warn_deprecation("Object.select", "v4.0.0", "Use Object.select_set() method instead")
|
|
prop.select_set(value)
|
|
|
|
@staticmethod
|
|
def __get_hide(prop: bpy.types.Object) -> bool:
|
|
utils.warn_deprecation("Object.hide", "v4.0.0", "Use Object.hide_get() method instead")
|
|
return prop.hide_get()
|
|
|
|
@staticmethod
|
|
def __set_hide(prop: bpy.types.Object, value: bool) -> None:
|
|
utils.warn_deprecation("Object.hide", "v4.0.0", "Use Object.hide_set() method instead")
|
|
prop.hide_set(value)
|
|
if prop.hide_viewport != value:
|
|
prop.hide_viewport = value
|
|
|
|
@staticmethod
|
|
def register():
|
|
bpy.types.Object.mmd_type = patch_library_overridable(
|
|
bpy.props.EnumProperty(
|
|
name="Type",
|
|
description="Internal MMD type of this object (DO NOT CHANGE IT DIRECTLY)",
|
|
default="NONE",
|
|
items=[
|
|
("NONE", "None", "", 1),
|
|
("ROOT", "Root", "", 2),
|
|
("RIGID_GRP_OBJ", "Rigid Body Grp Empty", "", 3),
|
|
("JOINT_GRP_OBJ", "Joint Grp Empty", "", 4),
|
|
("TEMPORARY_GRP_OBJ", "Temporary Grp Empty", "", 5),
|
|
("PLACEHOLDER", "Place Holder", "", 6),
|
|
("CAMERA", "Camera", "", 21),
|
|
("JOINT", "Joint", "", 22),
|
|
("RIGID_BODY", "Rigid body", "", 23),
|
|
("LIGHT", "Light", "", 24),
|
|
("TRACK_TARGET", "Track Target", "", 51),
|
|
("NON_COLLISION_CONSTRAINT", "Non Collision Constraint", "", 52),
|
|
("SPRING_CONSTRAINT", "Spring Constraint", "", 53),
|
|
("SPRING_GOAL", "Spring Goal", "", 54),
|
|
],
|
|
)
|
|
)
|
|
bpy.types.Object.mmd_root = patch_library_overridable(bpy.props.PointerProperty(type=MMDRoot))
|
|
|
|
bpy.types.Object.select = patch_library_overridable(
|
|
bpy.props.BoolProperty(
|
|
get=MMDRoot.__get_select,
|
|
set=MMDRoot.__set_select,
|
|
options={
|
|
"SKIP_SAVE",
|
|
"ANIMATABLE",
|
|
"LIBRARY_EDITABLE",
|
|
},
|
|
)
|
|
)
|
|
bpy.types.Object.hide = patch_library_overridable(
|
|
bpy.props.BoolProperty(
|
|
get=MMDRoot.__get_hide,
|
|
set=MMDRoot.__set_hide,
|
|
options={
|
|
"SKIP_SAVE",
|
|
"ANIMATABLE",
|
|
"LIBRARY_EDITABLE",
|
|
},
|
|
)
|
|
)
|
|
|
|
@staticmethod
|
|
def unregister():
|
|
del bpy.types.Object.hide
|
|
del bpy.types.Object.select
|
|
del bpy.types.Object.mmd_root
|
|
del bpy.types.Object.mmd_type
|