Armature Selection Improvements.

- Added a check to make sure Armature is valid.
- Added a helper to select the current armature  selected in armature selection.
- Added a helper to get all meshes.
- Updated all current functions to work with the system.
This commit is contained in:
Yusarina
2024-07-24 00:27:14 +01:00
parent 06c7cff4b7
commit 76046f7c6d
6 changed files with 61 additions and 30 deletions
+6 -5
View File
@@ -5,7 +5,7 @@ import re
from typing import List, Tuple, Optional, TypedDict
from bpy.types import Material, Operator, Context, Object
from ..core.register import register_wrap
from ..core.common import get_selected_armature
from ..core.common import get_selected_armature, is_valid_armature, select_current_armature, get_all_meshes
class meshEntry(TypedDict):
@@ -23,17 +23,18 @@ class RemoveDoublesSafely(Operator):
@classmethod
def poll(cls, context: Context) -> bool:
return context.mode == 'OBJECT' and get_selected_armature(context) is not None
armature = get_selected_armature(context)
return armature is not None and is_valid_armature(armature)
def execute(self, context: Context) -> set:
armature = get_selected_armature(context)
if not armature:
if not select_current_armature(context):
self.report({'WARNING'}, "No armature selected")
return {'CANCELLED'}
armature = get_selected_armature(context)
bpy.ops.object.mode_set(mode='OBJECT')
bpy.ops.object.select_all(action='DESELECT')
objects: List[Object] = [obj for obj in armature.children if obj.type == 'MESH']
objects: List[Object] = get_all_meshes(context)
for mesh in objects:
if mesh.data.name not in [stored_object["mesh"].data.name for stored_object in self.objects_to_do]: