Yusarina
2107c11e66
- Organised some of the code better. - Fixed sort order loop. - Added typing in some places there wasn't. This is a very basic start to Viesme creation, I still need to add translations for some stuff and improve it. This is very much inspired from the Cats Version.
89 lines
3.6 KiB
Python
89 lines
3.6 KiB
Python
import bpy
|
|
from ..core import common
|
|
from ..core.register import register_wrap
|
|
from ..functions.translations import t
|
|
from typing import List, Tuple
|
|
|
|
@register_wrap
|
|
class AutoVisemeButton(bpy.types.Operator):
|
|
bl_idname = 'avatar_toolkit.create_visemes'
|
|
bl_label = t('AutoVisemeButton.label')
|
|
bl_description = t('AutoVisemeButton.desc')
|
|
bl_options = {'REGISTER', 'UNDO'}
|
|
|
|
@classmethod
|
|
def poll(cls, context: bpy.types.Context) -> bool:
|
|
return context.active_object and context.active_object.type == 'MESH'
|
|
|
|
def execute(self, context: bpy.types.Context) -> set:
|
|
print("Starting viseme creation...")
|
|
mesh = context.active_object
|
|
if not mesh or not common.has_shapekeys(mesh):
|
|
self.report({'ERROR'}, t('AutoVisemeButton.error.noShapekeys'))
|
|
return {'CANCELLED'}
|
|
|
|
shape_a = context.scene.mouth_a
|
|
shape_o = context.scene.mouth_o
|
|
shape_ch = context.scene.mouth_ch
|
|
|
|
print(f"Selected shapes: A={shape_a}, O={shape_o}, CH={shape_ch}")
|
|
|
|
if shape_a == "Basis" or shape_o == "Basis" or shape_ch == "Basis":
|
|
self.report({'ERROR'}, t('AutoVisemeButton.error.selectShapekeys'))
|
|
return {'CANCELLED'}
|
|
|
|
# Create visemes
|
|
visemes: List[Tuple[str, List[Tuple[str, float]]]] = [
|
|
('vrc.v_aa', [(shape_a, 0.9998)]),
|
|
('vrc.v_ch', [(shape_ch, 0.9996)]),
|
|
('vrc.v_dd', [(shape_a, 0.3), (shape_ch, 0.7)]),
|
|
('vrc.v_e', [(shape_a, 0.5), (shape_ch, 0.2)]),
|
|
('vrc.v_ff', [(shape_a, 0.2), (shape_ch, 0.4)]),
|
|
('vrc.v_ih', [(shape_ch, 0.7), (shape_o, 0.3)]),
|
|
('vrc.v_kk', [(shape_a, 0.7), (shape_ch, 0.4)]),
|
|
('vrc.v_nn', [(shape_a, 0.2), (shape_ch, 0.7)]),
|
|
('vrc.v_oh', [(shape_a, 0.2), (shape_o, 0.8)]),
|
|
('vrc.v_ou', [(shape_o, 0.9994)]),
|
|
('vrc.v_pp', [(shape_a, 0.0004), (shape_o, 0.0004)]),
|
|
('vrc.v_rr', [(shape_ch, 0.5), (shape_o, 0.3)]),
|
|
('vrc.v_sil', [(shape_a, 0.0002), (shape_ch, 0.0002)]),
|
|
('vrc.v_ss', [(shape_ch, 0.8)]),
|
|
('vrc.v_th', [(shape_a, 0.4), (shape_o, 0.15)])
|
|
]
|
|
|
|
for viseme_name, shape_mix in visemes:
|
|
print(f"Creating viseme: {viseme_name}")
|
|
self.create_viseme(mesh, viseme_name, shape_mix, context.scene.shape_intensity)
|
|
|
|
print("Sorting shape keys...")
|
|
common.sort_shape_keys(mesh)
|
|
|
|
print("Viseme creation completed.")
|
|
self.report({'INFO'}, t('AutoVisemeButton.success'))
|
|
return {'FINISHED'}
|
|
|
|
def create_viseme(self, mesh: bpy.types.Object, viseme_name: str, shape_mix: List[Tuple[str, float]], intensity: float) -> None:
|
|
print(f" Creating viseme: {viseme_name}")
|
|
shape_keys = mesh.data.shape_keys.key_blocks
|
|
|
|
# Remove existing viseme if it exists
|
|
if viseme_name in shape_keys:
|
|
print(f" Removing existing viseme: {viseme_name}")
|
|
mesh.shape_key_remove(shape_keys[viseme_name])
|
|
|
|
# Create new viseme
|
|
new_key = mesh.shape_key_add(name=viseme_name, from_mix=False)
|
|
new_key.value = 1.0
|
|
|
|
# Mix shapes
|
|
for shape_name, value in shape_mix:
|
|
if shape_name in shape_keys:
|
|
source_shape = shape_keys[shape_name]
|
|
print(f" Mixing shape: {shape_name} with value: {value * intensity}")
|
|
for i, vert in enumerate(new_key.data):
|
|
vert.co += (source_shape.data[i].co - shape_keys['Basis'].data[i].co) * value * intensity
|
|
|
|
print(f" Viseme {viseme_name} created successfully.")
|
|
|
|
|