Cool_Tools/io_scene_psk_psa
1
0
Fork 0
mirror of https://github.com/DarklightGames/io_scene_psk_psa.git synced 2025-03-04 07:38:14 +01:00
Code

[PSA Export] Remember selections for bone groups

Browse Source 
* Selections for bone groups are now preserved between exporter
  invocations.

* Added typing hints to the `populate_bone_group_list` function.
This commit is contained in:
Yurii Ti 2022-05-05 15:02:31 +03:00
parent 19ff47cc83
commit 1ac0870b31
1 changed files with 15 additions and 5 deletions

20
io_scene_psk_psa/helpers.py

@ -1,8 +1,9 @@
import datetime import datetime
from collections import Counter from collections import Counter
from typing import List from typing import List, Iterable
from bpy.types import NlaStrip from bpy.types import NlaStrip, Object
from .types import BoneGroupListItem
class Timer: class Timer:
@ -56,7 +57,16 @@ def get_nla_strips_in_timeframe(object, frame_min, frame_max) -> List[NlaStrip]:
return strips return strips
def populate_bone_group_list(armature_object, bone_group_list): def populate_bone_group_list(armature_object: Object, bone_group_list: Iterable[BoneGroupListItem]) -> None:
# Preserve group selections before clearing the list.
# We handle selections for the unassigned group separately to cover the edge
# case where there might be an actual group with 'Unassigned' as its name.
unassigned_group_idx, unassigned_group_is_selected = next(iter([
(i, g.is_selected) for i, g in enumerate(bone_group_list) if g.index == -1]), (-1, False))
selected_assigned_group_names = [
g.name for i, g in enumerate(bone_group_list) if i != unassigned_group_idx and g.is_selected]
bone_group_list.clear() bone_group_list.clear()
if armature_object and armature_object.pose: if armature_object and armature_object.pose:
@ -66,14 +76,14 @@ def populate_bone_group_list(armature_object, bone_group_list):
item.name = 'Unassigned' item.name = 'Unassigned'
item.index = -1 item.index = -1
item.count = 0 if None not in bone_group_counts else bone_group_counts[None] item.count = 0 if None not in bone_group_counts else bone_group_counts[None]
item.is_selected = True item.is_selected = unassigned_group_is_selected
for bone_group_index, bone_group in enumerate(armature_object.pose.bone_groups): for bone_group_index, bone_group in enumerate(armature_object.pose.bone_groups):
item = bone_group_list.add() item = bone_group_list.add()
item.name = bone_group.name item.name = bone_group.name
item.index = bone_group_index item.index = bone_group_index
item.count = 0 if bone_group not in bone_group_counts else bone_group_counts[bone_group] item.count = 0 if bone_group not in bone_group_counts else bone_group_counts[bone_group]
item.is_selected = True item.is_selected = bone_group.name in selected_assigned_group_names
def get_psa_sequence_name(action, should_use_original_sequence_name): def get_psa_sequence_name(action, should_use_original_sequence_name):