DarklightGames/io_scene_psk_psa
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Reformatting for brevity and reducing redundancy

Browse Source
This commit is contained in:
Colin Basnett
2022-01-25 15:36:48 -08:00
parent c672941663
commit fbf3ec599b
3 changed files with 134 additions and 136 deletions
Download Patch File Download Diff File Expand all files Collapse all files

67
io_scene_psk_psa/psa/exporter.py
View File

@@ -47,10 +47,10 @@ class PsaExportActionListItem(PropertyGroup):
def update_action_names(context): def update_action_names(context):
property_group = context.scene.psa_export pg = context.scene.psa_export
for item in property_group.action_list: for item in pg.action_list:
action = item.action action = item.action
item.action_name = get_psa_sequence_name(action, property_group.should_use_original_sequence_names) item.action_name = get_psa_sequence_name(action, pg.should_use_original_sequence_names)
def should_use_original_sequence_names_updated(property, context): def should_use_original_sequence_names_updated(property, context):
@@ -98,26 +98,25 @@ class PsaExportOperator(Operator, ExportHelper):
def draw(self, context): def draw(self, context):
layout = self.layout layout = self.layout
property_group = context.scene.psa_export pg = context.scene.psa_export
# ACTIONS # ACTIONS
layout.label(text='Actions', icon='ACTION') layout.label(text='Actions', icon='ACTION')
row = layout.row(align=True) row = layout.row(align=True)
row.label(text='Select') row.label(text='Select')
row.operator('psa_export.actions_select_all', text='All') row.operator(PsaExportSelectAll.bl_idname, text='All')
row.operator('psa_export.actions_deselect_all', text='None') row.operator(PsaExportDeselectAll.bl_idname, text='None')
row = layout.row() row = layout.row()
rows = max(3, min(len(property_group.action_list), 10)) rows = max(3, min(len(pg.action_list), 10))
row.template_list('PSA_UL_ExportActionList', '', property_group, 'action_list', property_group, row.template_list('PSA_UL_ExportActionList', '', pg, 'action_list', pg, 'action_list_index', rows=rows)
'action_list_index', rows=rows)
col = layout.column(heading="Options") col = layout.column(heading="Options")
col.use_property_split = True col.use_property_split = True
col.use_property_decorate = False col.use_property_decorate = False
col.prop(property_group, 'should_use_original_sequence_names') col.prop(pg, 'should_use_original_sequence_names')
# Determine if there is going to be a naming conflict and display an error, if so. # Determine if there is going to be a naming conflict and display an error, if so.
selected_actions = [x for x in property_group.action_list if x.is_selected] selected_actions = [x for x in pg.action_list if x.is_selected]
action_names = [x.action_name for x in selected_actions] action_names = [x.action_name for x in selected_actions]
action_name_counts = Counter(action_names) action_name_counts = Counter(action_names)
for action_name, count in action_name_counts.items(): for action_name, count in action_name_counts.items():
@@ -130,18 +129,18 @@ class PsaExportOperator(Operator, ExportHelper):
# BONES # BONES
box = layout.row() box = layout.row()
box.label(text='Bones', icon='BONE_DATA') box.label(text='Bones', icon='BONE_DATA')
bone_filter_mode_items = property_group.bl_rna.properties['bone_filter_mode'].enum_items_static bone_filter_mode_items = pg.bl_rna.properties['bone_filter_mode'].enum_items_static
row = box.row(align=True) row = box.row(align=True)
for item in bone_filter_mode_items: for item in bone_filter_mode_items:
identifier = item.identifier identifier = item.identifier
item_layout = row.row(align=True) item_layout = row.row(align=True)
item_layout.prop_enum(property_group, 'bone_filter_mode', item.identifier) item_layout.prop_enum(pg, 'bone_filter_mode', item.identifier)
item_layout.enabled = is_bone_filter_mode_item_available(context, identifier) item_layout.enabled = is_bone_filter_mode_item_available(context, identifier)
if property_group.bone_filter_mode == 'BONE_GROUPS': if pg.bone_filter_mode == 'BONE_GROUPS':
rows = max(3, min(len(property_group.bone_group_list), 10)) rows = max(3, min(len(pg.bone_group_list), 10))
layout.template_list('PSX_UL_BoneGroupList', '', property_group, 'bone_group_list', property_group, layout.template_list('PSX_UL_BoneGroupList', '', pg, 'bone_group_list', pg,
'bone_group_list_index', rows=rows) 'bone_group_list_index', rows=rows)
def is_action_for_armature(self, action): def is_action_for_armature(self, action):
@@ -158,7 +157,7 @@ class PsaExportOperator(Operator, ExportHelper):
return False return False
def invoke(self, context, event): def invoke(self, context, event):
property_group = context.scene.psa_export pg = context.scene.psa_export
if context.view_layer.objects.active is None: if context.view_layer.objects.active is None:
self.report({'ERROR_INVALID_CONTEXT'}, 'An armature must be selected') self.report({'ERROR_INVALID_CONTEXT'}, 'An armature must be selected')
@@ -171,32 +170,32 @@ class PsaExportOperator(Operator, ExportHelper):
self.armature = context.view_layer.objects.active self.armature = context.view_layer.objects.active
# Populate actions list. # Populate actions list.
property_group.action_list.clear() pg.action_list.clear()
for action in bpy.data.actions: for action in bpy.data.actions:
if not self.is_action_for_armature(action): if not self.is_action_for_armature(action):
continue continue
item = property_group.action_list.add() item = pg.action_list.add()
item.action = action item.action = action
item.action_name = action.name item.action_name = action.name
item.is_selected = True item.is_selected = True
update_action_names(context) update_action_names(context)
if len(property_group.action_list) == 0: if len(pg.action_list) == 0:
# If there are no actions at all, we have nothing to export, so just cancel the operation. # If there are no actions at all, we have nothing to export, so just cancel the operation.
self.report({'ERROR_INVALID_CONTEXT'}, 'There are no actions to export.') self.report({'ERROR_INVALID_CONTEXT'}, 'There are no actions to export.')
return {'CANCELLED'} return {'CANCELLED'}
# Populate bone groups list. # Populate bone groups list.
populate_bone_group_list(self.armature, property_group.bone_group_list) populate_bone_group_list(self.armature, pg.bone_group_list)
context.window_manager.fileselect_add(self) context.window_manager.fileselect_add(self)
return {'RUNNING_MODAL'} return {'RUNNING_MODAL'}
def execute(self, context): def execute(self, context):
property_group = context.scene.psa_export pg = context.scene.psa_export
actions = [x.action for x in property_group.action_list if x.is_selected] actions = [x.action for x in pg.action_list if x.is_selected]
if len(actions) == 0: if len(actions) == 0:
self.report({'ERROR_INVALID_CONTEXT'}, 'No actions were selected for export.') self.report({'ERROR_INVALID_CONTEXT'}, 'No actions were selected for export.')
@@ -204,9 +203,9 @@ class PsaExportOperator(Operator, ExportHelper):
options = PsaBuilderOptions() options = PsaBuilderOptions()
options.actions = actions options.actions = actions
options.bone_filter_mode = property_group.bone_filter_mode options.bone_filter_mode = pg.bone_filter_mode
options.bone_group_indices = [x.index for x in property_group.bone_group_list if x.is_selected] options.bone_group_indices = [x.index for x in pg.bone_group_list if x.is_selected]
options.should_use_original_sequence_names = property_group.should_use_original_sequence_names options.should_use_original_sequence_names = pg.should_use_original_sequence_names
builder = PsaBuilder() builder = PsaBuilder()
try: try:
psa = builder.build(context, options) psa = builder.build(context, options)
@@ -246,14 +245,14 @@ class PsaExportSelectAll(bpy.types.Operator):
@classmethod @classmethod
def poll(cls, context): def poll(cls, context):
property_group = context.scene.psa_export pg = context.scene.psa_export
action_list = property_group.action_list action_list = pg.action_list
has_unselected_actions = any(map(lambda action: not action.is_selected, action_list)) has_unselected_actions = any(map(lambda action: not action.is_selected, action_list))
return len(action_list) > 0 and has_unselected_actions return len(action_list) > 0 and has_unselected_actions
def execute(self, context): def execute(self, context):
property_group = context.scene.psa_export pg = context.scene.psa_export
for action in property_group.action_list: for action in pg.action_list:
action.is_selected = True action.is_selected = True
return {'FINISHED'} return {'FINISHED'}
@@ -265,14 +264,14 @@ class PsaExportDeselectAll(bpy.types.Operator):
@classmethod @classmethod
def poll(cls, context): def poll(cls, context):
property_group = context.scene.psa_export pg = context.scene.psa_export
action_list = property_group.action_list action_list = pg.action_list
has_selected_actions = any(map(lambda action: action.is_selected, action_list)) has_selected_actions = any(map(lambda action: action.is_selected, action_list))
return len(action_list) > 0 and has_selected_actions return len(action_list) > 0 and has_selected_actions
def execute(self, context): def execute(self, context):
property_group = context.scene.psa_export pg = context.scene.psa_export
for action in property_group.action_list: for action in pg.action_list:
action.is_selected = False action.is_selected = False
return {'FINISHED'} return {'FINISHED'}

183
io_scene_psk_psa/psa/importer.py
View File

@@ -188,7 +188,7 @@ class PsaImporter(object):
actions.append(action) actions.append(action)
# If the user specifies, store the new animations as strips on a non-contributing NLA stack. # If the user specifies, store the new animations as strips on a non-contributing NLA track.
if options.should_stash: if options.should_stash:
if armature_object.animation_data is None: if armature_object.animation_data is None:
armature_object.animation_data_create() armature_object.animation_data_create()
@@ -199,43 +199,29 @@ class PsaImporter(object):
nla_track.strips.new(name=action.name, start=0, action=action) nla_track.strips.new(name=action.name, start=0, action=action)
class PsaImportPsaBoneItem(PropertyGroup):
bone_name: StringProperty()
@property
def name(self):
return self.bone_name
class PsaImportActionListItem(PropertyGroup): class PsaImportActionListItem(PropertyGroup):
action_name: StringProperty() action_name: StringProperty()
frame_count: IntProperty()
is_selected: BoolProperty(default=False) is_selected: BoolProperty(default=False)
@property
def name(self):
return self.action_name
def load_psa_file(context): def load_psa_file(context):
property_group = context.scene.psa_import pg = context.scene.psa_import
property_group.sequence_list.clear() pg.sequence_list.clear()
property_group.psa_bones.clear() pg.psa.bones.clear()
property_group.action_list.clear() pg.action_list.clear()
property_group.psa_error = '' pg.psa_error = ''
try: try:
# Read the file and populate the action list. # Read the file and populate the action list.
p = os.path.abspath(property_group.psa_file_path) p = os.path.abspath(pg.psa_file_path)
psa_reader = PsaReader(p) psa_reader = PsaReader(p)
for sequence in psa_reader.sequences.values(): for sequence in psa_reader.sequences.values():
item = property_group.sequence_list.add() item = pg.sequence_list.add()
item.action_name = sequence.name.decode('windows-1252') item.action_name = sequence.name.decode('windows-1252')
item.frame_count = sequence.frame_count
for psa_bone in psa_reader.bones: for psa_bone in psa_reader.bones:
item = property_group.psa_bones.add() item = pg.psa.bones.add()
item.bone_name = psa_bone.name.decode('windows-1252') item.bone_name = psa_bone.name.decode('windows-1252')
except Exception as e: except Exception as e:
property_group.psa_error = str(e) pg.psa_error = str(e)
def on_psa_file_path_updated(property, context): def on_psa_file_path_updated(property, context):
@@ -247,7 +233,6 @@ class PsaBonePropertyGroup(PropertyGroup):
class PsaDataPropertyGroup(PropertyGroup): class PsaDataPropertyGroup(PropertyGroup):
bone_count: IntProperty(default=0)
bones: CollectionProperty(type=PsaBonePropertyGroup) bones: CollectionProperty(type=PsaBonePropertyGroup)
sequence_count: IntProperty(default=0) sequence_count: IntProperty(default=0)
@@ -255,7 +240,7 @@ class PsaDataPropertyGroup(PropertyGroup):
class PsaImportPropertyGroup(PropertyGroup): class PsaImportPropertyGroup(PropertyGroup):
psa_file_path: StringProperty(default='', update=on_psa_file_path_updated, name='PSA File Path') psa_file_path: StringProperty(default='', update=on_psa_file_path_updated, name='PSA File Path')
psa_error: StringProperty(default='') psa_error: StringProperty(default='')
psa_bones: CollectionProperty(type=PsaImportPsaBoneItem) psa: PointerProperty(type=PsaDataPropertyGroup)
sequence_list: CollectionProperty(type=PsaImportActionListItem) sequence_list: CollectionProperty(type=PsaImportActionListItem)
sequence_list_index: IntProperty(name='', default=0) sequence_list_index: IntProperty(name='', default=0)
action_list: CollectionProperty(type=PsaImportActionListItem) action_list: CollectionProperty(type=PsaImportActionListItem)
@@ -288,7 +273,6 @@ class PSA_UL_SequenceList(UIList):
def filter_items(self, context, data, property): def filter_items(self, context, data, property):
actions = getattr(data, property) actions = getattr(data, property)
flt_flags = [] flt_flags = []
flt_neworder = []
if self.filter_name: if self.filter_name:
flt_flags = bpy.types.UI_UL_list.filter_items_by_name( flt_flags = bpy.types.UI_UL_list.filter_items_by_name(
self.filter_name, self.filter_name,
@@ -297,6 +281,7 @@ class PSA_UL_SequenceList(UIList):
'action_name', 'action_name',
reverse=self.use_filter_invert reverse=self.use_filter_invert
) )
flt_neworder = bpy.types.UI_UL_list.sort_items_by_name(actions, 'action_name')
return flt_flags, flt_neworder return flt_flags, flt_neworder
@@ -315,14 +300,14 @@ class PsaImportSequencesSelectAll(bpy.types.Operator):
@classmethod @classmethod
def poll(cls, context): def poll(cls, context):
property_group = context.scene.psa_import pg = context.scene.psa_import
sequence_list = property_group.sequence_list sequence_list = pg.sequence_list
has_unselected_actions = any(map(lambda action: not action.is_selected, sequence_list)) has_unselected_actions = any(map(lambda action: not action.is_selected, sequence_list))
return len(sequence_list) > 0 and has_unselected_actions return len(sequence_list) > 0 and has_unselected_actions
def execute(self, context): def execute(self, context):
property_group = context.scene.psa_import pg = context.scene.psa_import
for action in property_group.sequence_list: for action in pg.sequence_list:
action.is_selected = True action.is_selected = True
return {'FINISHED'} return {'FINISHED'}
@@ -334,14 +319,14 @@ class PsaImportActionsSelectAll(bpy.types.Operator):
@classmethod @classmethod
def poll(cls, context): def poll(cls, context):
property_group = context.scene.psa_import pg = context.scene.psa_import
action_list = property_group.action_list action_list = pg.action_list
has_unselected_actions = any(map(lambda action: not action.is_selected, action_list)) has_unselected_actions = any(map(lambda action: not action.is_selected, action_list))
return len(action_list) > 0 and has_unselected_actions return len(action_list) > 0 and has_unselected_actions
def execute(self, context): def execute(self, context):
property_group = context.scene.psa_import pg = context.scene.psa_import
for action in property_group.action_list: for action in pg.action_list:
action.is_selected = True action.is_selected = True
return {'FINISHED'} return {'FINISHED'}
@@ -353,14 +338,14 @@ class PsaImportSequencesDeselectAll(bpy.types.Operator):
@classmethod @classmethod
def poll(cls, context): def poll(cls, context):
property_group = context.scene.psa_import pg = context.scene.psa_import
sequence_list = property_group.sequence_list sequence_list = pg.sequence_list
has_selected_sequences = any(map(lambda action: action.is_selected, sequence_list)) has_selected_sequences = any(map(lambda action: action.is_selected, sequence_list))
return len(sequence_list) > 0 and has_selected_sequences return len(sequence_list) > 0 and has_selected_sequences
def execute(self, context): def execute(self, context):
property_group = context.scene.psa_import pg = context.scene.psa_import
for action in property_group.sequence_list: for action in pg.sequence_list:
action.is_selected = False action.is_selected = False
return {'FINISHED'} return {'FINISHED'}
@@ -372,14 +357,14 @@ class PsaImportActionsDeselectAll(bpy.types.Operator):
@classmethod @classmethod
def poll(cls, context): def poll(cls, context):
property_group = context.scene.psa_import pg = context.scene.psa_import
action_list = property_group.action_list action_list = pg.action_list
has_selected_actions = any(map(lambda action: action.is_selected, action_list)) has_selected_actions = any(map(lambda action: action.is_selected, action_list))
return len(action_list) > 0 and has_selected_actions return len(action_list) > 0 and has_selected_actions
def execute(self, context): def execute(self, context):
property_group = context.scene.psa_import pg = context.scene.psa_import
for action in property_group.action_list: for action in pg.action_list:
action.is_selected = False action.is_selected = False
return {'FINISHED'} return {'FINISHED'}
@@ -393,20 +378,35 @@ class PSA_PT_ImportPanel_Advanced(Panel):
def draw(self, context): def draw(self, context):
layout = self.layout layout = self.layout
property_group = context.scene.psa_import pg = context.scene.psa_import
col = layout.column(heading="Options") col = layout.column(heading="Options")
col.use_property_split = True col.use_property_split = True
col.use_property_decorate = False col.use_property_decorate = False
col.prop(property_group, 'should_clean_keys') col.prop(pg, 'should_clean_keys')
col.separator() col.separator()
col.prop(property_group, 'should_use_fake_user') col.prop(pg, 'should_use_fake_user')
col.prop(property_group, 'should_stash') col.prop(pg, 'should_stash')
col.separator() col.separator()
col.prop(property_group, 'should_use_action_name_prefix') col.prop(pg, 'should_use_action_name_prefix')
if property_group.should_use_action_name_prefix: if pg.should_use_action_name_prefix:
col.prop(property_group, 'action_name_prefix') col.prop(pg, 'action_name_prefix')
class PSA_PT_ImportPanel_PsaData(Panel):
bl_space_type = 'PROPERTIES'
bl_region_type = 'WINDOW'
bl_label = 'PSA Info'
bl_options = {'DEFAULT_CLOSED'}
bl_parent_id = 'PSA_PT_ImportPanel'
def draw(self, context):
layout = self.layout
pg = context.scene.psa_import.psa
layout.label(text=f'{len(pg.bones)} Bones', icon='BONE_DATA')
layout.label(text=f'{len(pg.sequence_count)} Sequences', icon='SEQUENCE')
class PSA_PT_ImportPanel(Panel): class PSA_PT_ImportPanel(Panel):
@@ -423,14 +423,14 @@ class PSA_PT_ImportPanel(Panel):
def draw(self, context): def draw(self, context):
layout = self.layout layout = self.layout
property_group = context.scene.psa_import pg = context.scene.psa_import
row = layout.row(align=True) row = layout.row(align=True)
row.operator('psa_import.select_file', text='', icon='FILEBROWSER') row.operator(PsaImportSelectFile.bl_idname, text='', icon='FILEBROWSER')
row.prop(property_group, 'psa_file_path', text='') row.prop(pg, 'psa_file_path', text='')
row.operator('psa_import.file_reload', text='', icon='FILE_REFRESH') row.operator(PsaImportFileReload.bl_idname, text='', icon='FILE_REFRESH')
if property_group.psa_error != '': if pg.psa_error != '':
row = layout.row() row = layout.row()
row.label(text='File could not be read', icon='ERROR') row.label(text='File could not be read', icon='ERROR')
@@ -439,33 +439,31 @@ class PSA_PT_ImportPanel(Panel):
box.label(text=f'Sequences', icon='ARMATURE_DATA') box.label(text=f'Sequences', icon='ARMATURE_DATA')
# select # select
rows = max(3, min(len(property_group.sequence_list) + len(property_group.action_list), 10)) rows = max(3, min(len(pg.sequence_list) + len(pg.action_list), 10))
row = box.row() row = box.row()
col = row.column() col = row.column()
row2 = col.row(align=True) row2 = col.row(align=True)
row2.label(text='Select') row2.label(text='Select')
row2.operator('psa_import.sequences_select_all', text='All') row2.operator(PsaImportSequencesSelectAll.bl_idname, text='All')
row2.operator('psa_import.sequences_deselect_all', text='None') row2.operator(PsaImportSequencesDeselectAll.bl_idname, text='None')
col = col.row() col = col.row()
col.template_list('PSA_UL_ImportSequenceList', '', property_group, 'sequence_list', property_group, col.template_list('PSA_UL_ImportSequenceList', '', pg, 'sequence_list', pg, 'sequence_list_index', rows=rows)
'sequence_list_index', rows=rows)
col = row.column(align=True) col = row.column(align=True)
col.operator('psa_import.push_to_actions', icon='TRIA_RIGHT', text='') col.operator(PsaImportPushToActions.bl_idname, icon='TRIA_RIGHT', text='')
col.operator('psa_import.pop_from_actions', icon='TRIA_LEFT', text='') col.operator(PsaImportPopFromActions.bl_idname, icon='TRIA_LEFT', text='')
col = row.column() col = row.column()
row2 = col.row(align=True) row2 = col.row(align=True)
row2.label(text='Select') row2.label(text='Select')
row2.operator('psa_import.actions_select_all', text='All') row2.operator(PsaImportActionsSelectAll.bl_idname, text='All')
row2.operator('psa_import.actions_deselect_all', text='None') row2.operator(PsaImportActionsDeselectAll.bl_idname, text='None')
col.template_list('PSA_UL_ImportActionList', '', property_group, 'action_list', property_group, col.template_list('PSA_UL_ImportActionList', '', pg, 'action_list', pg, 'action_list_index', rows=rows)
'action_list_index', rows=rows)
col.separator() col.separator()
col.operator('psa_import.import', text=f'Import') col.operator(PsaImportOperator.bl_idname, text=f'Import')
class PsaImportFileReload(Operator): class PsaImportFileReload(Operator):
@@ -503,21 +501,21 @@ class PsaImportOperator(Operator):
@classmethod @classmethod
def poll(cls, context): def poll(cls, context):
property_group = context.scene.psa_import pg = context.scene.psa_import
active_object = context.view_layer.objects.active active_object = context.view_layer.objects.active
action_list = property_group.action_list action_list = pg.action_list
return len(action_list) and active_object is not None and active_object.type == 'ARMATURE' return len(action_list) and active_object is not None and active_object.type == 'ARMATURE'
def execute(self, context): def execute(self, context):
property_group = context.scene.psa_import pg = context.scene.psa_import
psa_reader = PsaReader(property_group.psa_file_path) psa_reader = PsaReader(pg.psa_file_path)
sequence_names = [x.action_name for x in property_group.action_list] sequence_names = [x.action_name for x in pg.action_list]
options = PsaImportOptions() options = PsaImportOptions()
options.sequence_names = sequence_names options.sequence_names = sequence_names
options.should_clean_keys = property_group.should_clean_keys options.should_clean_keys = pg.should_clean_keys
options.should_use_fake_user = property_group.should_use_fake_user options.should_use_fake_user = pg.should_use_fake_user
options.should_stash = property_group.should_stash options.should_stash = pg.should_stash
options.action_name_prefix = property_group.action_name_prefix options.action_name_prefix = pg.action_name_prefix
PsaImporter().import_psa(psa_reader, context.view_layer.objects.active, options) PsaImporter().import_psa(psa_reader, context.view_layer.objects.active, options)
self.report({'INFO'}, f'Imported {len(sequence_names)} action(s)') self.report({'INFO'}, f'Imported {len(sequence_names)} action(s)')
return {'FINISHED'} return {'FINISHED'}
@@ -529,20 +527,20 @@ class PsaImportPushToActions(Operator):
@classmethod @classmethod
def poll(cls, context): def poll(cls, context):
property_group = context.scene.psa_import pg = context.scene.psa_import
has_sequences_selected = any(map(lambda x: x.is_selected, property_group.sequence_list)) has_sequences_selected = any(map(lambda x: x.is_selected, pg.sequence_list))
return has_sequences_selected return has_sequences_selected
def execute(self, context): def execute(self, context):
property_group = context.scene.psa_import pg = context.scene.psa_import
indices_to_remove = [] indices_to_remove = []
for sequence_index, item in enumerate(property_group.sequence_list): for sequence_index, item in enumerate(pg.sequence_list):
if item.is_selected: if item.is_selected:
indices_to_remove.append(sequence_index) indices_to_remove.append(sequence_index)
action = property_group.action_list.add() action = pg.action_list.add()
action.action_name = item.action_name action.action_name = item.action_name
for index in reversed(indices_to_remove): for index in reversed(indices_to_remove):
property_group.sequence_list.remove(index) pg.sequence_list.remove(index)
return {'FINISHED'} return {'FINISHED'}
@@ -552,20 +550,20 @@ class PsaImportPopFromActions(Operator):
@classmethod @classmethod
def poll(cls, context): def poll(cls, context):
property_group = context.scene.psa_import pg = context.scene.psa_import
has_actions_selected = any(map(lambda x: x.is_selected, property_group.action_list)) has_actions_selected = any(map(lambda x: x.is_selected, pg.action_list))
return has_actions_selected return has_actions_selected
def execute(self, context): def execute(self, context):
property_group = context.scene.psa_import pg = context.scene.psa_import
indices_to_remove = [] indices_to_remove = []
for action_index, item in enumerate(property_group.action_list): for action_index, item in enumerate(pg.action_list):
if item.is_selected: if item.is_selected:
indices_to_remove.append(action_index) indices_to_remove.append(action_index)
sequence = property_group.sequence_list.add() sequence = pg.sequence_list.add()
sequence.action_name = item.action_name sequence.action_name = item.action_name
for index in reversed(indices_to_remove): for index in reversed(indices_to_remove):
property_group.action_list.remove(index) pg.action_list.remove(index)
return {'FINISHED'} return {'FINISHED'}
@@ -585,15 +583,15 @@ class PsaImportFileSelectOperator(Operator, ImportHelper):
return {'RUNNING_MODAL'} return {'RUNNING_MODAL'}
def execute(self, context): def execute(self, context):
property_group = context.scene.psa_import pg = context.scene.psa_import
property_group.psa_file_path = self.filepath pg.psa_file_path = self.filepath
# Load the sequence names from the selected file
return {'FINISHED'} return {'FINISHED'}
classes = ( classes = (
PsaImportPsaBoneItem,
PsaImportActionListItem, PsaImportActionListItem,
PsaBonePropertyGroup,
PsaDataPropertyGroup,
PsaImportPropertyGroup, PsaImportPropertyGroup,
PSA_UL_SequenceList, PSA_UL_SequenceList,
PSA_UL_ImportSequenceList, PSA_UL_ImportSequenceList,
@@ -605,6 +603,7 @@ classes = (
PsaImportFileReload, PsaImportFileReload,
PSA_PT_ImportPanel, PSA_PT_ImportPanel,
PSA_PT_ImportPanel_Advanced, PSA_PT_ImportPanel_Advanced,
PSA_PT_ImportPanel_PsaData,
PsaImportOperator, PsaImportOperator,
PsaImportFileSelectOperator, PsaImportFileSelectOperator,
PsaImportSelectFile, PsaImportSelectFile,

20
io_scene_psk_psa/psk/exporter.py
View File

@@ -90,10 +90,10 @@ class PskExportOperator(Operator, ExportHelper):
self.report({'ERROR_INVALID_CONTEXT'}, str(e)) self.report({'ERROR_INVALID_CONTEXT'}, str(e))
return {'CANCELLED'} return {'CANCELLED'}
property_group = context.scene.psk_export pg = context.scene.psk_export
# Populate bone groups list. # Populate bone groups list.
populate_bone_group_list(input_objects.armature_object, property_group.bone_group_list) populate_bone_group_list(input_objects.armature_object, pg.bone_group_list)
context.window_manager.fileselect_add(self) context.window_manager.fileselect_add(self)
@@ -102,29 +102,29 @@ class PskExportOperator(Operator, ExportHelper):
def draw(self, context): def draw(self, context):
layout = self.layout layout = self.layout
scene = context.scene scene = context.scene
property_group = scene.psk_export pg = scene.psk_export
# BONES # BONES
box = layout.box() box = layout.box()
box.label(text='Bones', icon='BONE_DATA') box.label(text='Bones', icon='BONE_DATA')
bone_filter_mode_items = property_group.bl_rna.properties['bone_filter_mode'].enum_items_static bone_filter_mode_items = pg.bl_rna.properties['bone_filter_mode'].enum_items_static
row = box.row(align=True) row = box.row(align=True)
for item in bone_filter_mode_items: for item in bone_filter_mode_items:
identifier = item.identifier identifier = item.identifier
item_layout = row.row(align=True) item_layout = row.row(align=True)
item_layout.prop_enum(property_group, 'bone_filter_mode', item.identifier) item_layout.prop_enum(pg, 'bone_filter_mode', item.identifier)
item_layout.enabled = is_bone_filter_mode_item_available(context, identifier) item_layout.enabled = is_bone_filter_mode_item_available(context, identifier)
if property_group.bone_filter_mode == 'BONE_GROUPS': if pg.bone_filter_mode == 'BONE_GROUPS':
row = box.row() row = box.row()
rows = max(3, min(len(property_group.bone_group_list), 10)) rows = max(3, min(len(pg.bone_group_list), 10))
row.template_list('PSX_UL_BoneGroupList', '', property_group, 'bone_group_list', property_group, 'bone_group_list_index', rows=rows) row.template_list('PSX_UL_BoneGroupList', '', pg, 'bone_group_list', pg, 'bone_group_list_index', rows=rows)
def execute(self, context): def execute(self, context):
property_group = context.scene.psk_export pg = context.scene.psk_export
builder = PskBuilder() builder = PskBuilder()
options = PskBuilderOptions() options = PskBuilderOptions()
options.bone_group_indices = [x.index for x in property_group.bone_group_list if x.is_selected] options.bone_group_indices = [x.index for x in pg.bone_group_list if x.is_selected]
try: try:
psk = builder.build(context, options) psk = builder.build(context, options)
exporter = PskExporter(psk) exporter = PskExporter(psk)