989onan
139 lines
5.3 KiB
Python
139 lines
5.3 KiB
Python
import traceback
|
|
import bpy
|
|
import re
|
|
from typing import Any, Set, Dict, List, Optional, Tuple
|
|
from bpy.types import (
|
|
Operator,
|
|
Context,
|
|
Object,
|
|
Material,
|
|
NodeTree,
|
|
ShaderNodeTexImage
|
|
)
|
|
import mathutils
|
|
import bmesh
|
|
from ...core.logging_setup import logger
|
|
from ...core.translations import t
|
|
from ...core.common import (
|
|
get_active_armature,
|
|
get_all_meshes,
|
|
ProgressTracker,
|
|
calculate_bone_orientation,
|
|
add_armature_modifier,
|
|
get_modifiers,
|
|
has_shapekeys
|
|
)
|
|
from ...core.armature_validation import validate_armature
|
|
|
|
class AvatarToolkit_OT_ApplyModifierForShapkeyObj(bpy.types.Operator):
|
|
"""Operator for forcing the application of a modifier. A shortened way of saying \"Apply modifier for object with shapekeys\""""
|
|
bl_idname: str = 'avatar_toolkit.merge_armatures'
|
|
bl_label: str = t('Tools.apply_modifier_on_shapekey_obj')
|
|
bl_description: str = t('Tools.apply_modifier_on_shapekey_obj_desc')
|
|
bl_options: Set[str] = {'REGISTER', 'UNDO'}
|
|
|
|
modifier: bpy.props.EnumProperty(items=get_modifiers,name="Modifier To Apply")
|
|
|
|
|
|
def draw(self, context: Context) -> None:
|
|
"""Draw the operator's UI"""
|
|
layout = self.layout
|
|
layout.prop(self, "modifier")
|
|
|
|
def invoke(self, context: Context, event: bpy.types.Event) -> set[str]:
|
|
"""Initialize the operator"""
|
|
return context.window_manager.invoke_props_dialog(self)
|
|
|
|
@classmethod
|
|
def poll(cls, context: Context) -> bool:
|
|
if context.active_object != None:
|
|
return context.active_object.type == "MESH"
|
|
return False
|
|
|
|
def execute(self, context: Context) -> Set[str]:
|
|
|
|
obj: bpy.types.Object = context.active_object
|
|
mesh: bpy.types.Mesh = obj.data
|
|
|
|
shapes: list[bpy.types.Object] = []
|
|
|
|
bpy.ops.object.mode_set(mode="OBJECT")
|
|
|
|
if has_shapekeys(obj):
|
|
#reset shapekeys
|
|
for idx,key in enumerate(mesh.shape_keys.key_blocks):
|
|
obj.active_shape_key_index = idx
|
|
obj.active_shape_key.value = 0
|
|
|
|
for idx,key in enumerate(mesh.shape_keys.key_blocks):
|
|
# duplicate object for shapekey
|
|
bpy.ops.object.select_all(action="DESELECT")
|
|
context.view_layer.objects.active = obj
|
|
obj.select_set(True)
|
|
bpy.ops.object.duplicate()
|
|
|
|
# name new object after shapekey
|
|
new_obj = context.view_layer.objects.active
|
|
new_obj.select_set(True)
|
|
new_obj.active_shape_key_index = idx
|
|
new_obj.name = new_obj.active_shape_key.name
|
|
|
|
#add to cleanup list
|
|
shapes.append(new_obj)
|
|
|
|
#make basis the same shape as shapekey
|
|
for idx,point in enumerate(new_obj.active_shape_key.points):
|
|
new_obj.data.vertices[idx].co.xyz = point.co.xyz
|
|
|
|
#remove all shaoekeys on new object and then apply modifier
|
|
bpy.ops.object.shape_key_remove(all=True,apply_mix=False)
|
|
try:
|
|
bpy.ops.object.modifier_apply(modifier=self.modifier)
|
|
except Exception as e:
|
|
self.report({'ERROR'}, f"Shapekey modifier apply for shapekey \"{new_obj.name}\" failed!!")
|
|
print(f"Shapekey modifier apply for shapekey \"{new_obj.name}\" failed!!")
|
|
print(traceback.format_exc(e))
|
|
#clean up after critical failure
|
|
for shape in shapes:
|
|
bpy.data.objects.remove(shape)#faster than ops delete
|
|
bpy.ops.object.select_all(action="DESELECT")
|
|
|
|
|
|
|
|
try:
|
|
#remove shapekeys on original object
|
|
bpy.ops.object.select_all(action="DESELECT")
|
|
obj.select_set(True)
|
|
context.view_layer.objects.active = obj
|
|
bpy.ops.object.shape_key_remove(all=True,apply_mix=False)
|
|
bpy.ops.object.modifier_apply(modifier=self.modifier)
|
|
bpy.ops.object.select_all(action="DESELECT")
|
|
#delete first shapekey object aka basis
|
|
bpy.data.objects.remove(shapes.pop(0))
|
|
|
|
#join all objects with applied modifiers back together as shapes
|
|
for shape in shapes:
|
|
shape.select_set(True)
|
|
obj.select_set(True)
|
|
context.view_layer.objects.active = obj
|
|
bpy.ops.object.join_shapes()
|
|
except Exception as e:
|
|
|
|
self.report({'ERROR'}, f"Shapekey joining failed!!")
|
|
print(f"Shapekey joining failed!!")
|
|
print(traceback.format_exc(e))
|
|
#clean up after critical failure
|
|
for shape in shapes:
|
|
bpy.data.objects.remove(shape)#faster than ops delete
|
|
|
|
#final clean up
|
|
for shape in shapes:
|
|
bpy.data.objects.remove(shape)#faster than ops delete
|
|
|
|
else:
|
|
#mesh has no shapekeys, just apply normally.
|
|
bpy.ops.object.modifier_apply(modifier=self.modifier)
|
|
|
|
|
|
|
|
return {'FINISHED'} |