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

Added the ability for users to export "static meshes" for meshes that don't have armature modifiers

Browse Source
This commit is contained in:
Colin Basnett 2021-08-14 18:00:16 -07:00
parent e51013eec7
commit a76215569d
1 changed files with 72 additions and 60 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

132
io_export_psk_psa/psk/builder.py
View File

@ -30,22 +30,23 @@ class PskBuilder(object):
if len(obj.data.materials) == 0: if len(obj.data.materials) == 0:
raise RuntimeError(f'Mesh "{obj.name}" must have at least one material') raise RuntimeError(f'Mesh "{obj.name}" must have at least one material')
# ensure that there is exactly one armature modifier object shared between all selected meshes # Ensure that there are either no armature modifiers (static mesh)
# or that there is exactly one armature modifier object shared between
# all selected meshes
armature_modifier_objects = set() armature_modifier_objects = set()
for obj in input_objects.mesh_objects: for obj in input_objects.mesh_objects:
modifiers = [x for x in obj.modifiers if x.type == 'ARMATURE'] modifiers = [x for x in obj.modifiers if x.type == 'ARMATURE']
if len(modifiers) != 1: if len(modifiers) == 0:
raise RuntimeError(f'Mesh "{obj.name}" must have one armature modifier') continue
elif len(modifiers) == 2:
raise RuntimeError(f'Mesh "{obj.name}" must have only one armature modifier')
armature_modifier_objects.add(modifiers[0].object) armature_modifier_objects.add(modifiers[0].object)
if len(armature_modifier_objects) > 1: if len(armature_modifier_objects) > 1:
raise RuntimeError('All selected meshes must have the same armature modifier') raise RuntimeError('All selected meshes must have the same armature modifier')
elif len(armature_modifier_objects) == 1:
input_objects.armature_object = list(armature_modifier_objects)[0] input_objects.armature_object = list(armature_modifier_objects)[0]
if input_objects.armature_object is None:
raise RuntimeError('Armature modifier has no linked object')
return input_objects return input_objects
@ -61,42 +62,53 @@ class PskBuilder(object):
materials = OrderedDict() materials = OrderedDict()
bones = list(input_objects.armature_object.data.bones) if input_objects.armature_object is None:
for bone in bones: # Static mesh (no armature)
psk_bone = Psk.Bone() psk_bone = Psk.Bone()
psk_bone.name = bytes(bone.name, encoding='utf-8') psk_bone.name = bytes('static', encoding='utf-8')
psk_bone.flags = 0 psk_bone.flags = 0
psk_bone.children_count = len(bone.children) psk_bone.children_count = 0
psk_bone.parent_index = 0
try: psk_bone.location = Vector3(0, 0, 0)
psk_bone.parent_index = bones.index(bone.parent) psk_bone.rotation = Quaternion(0, 0, 0, 1)
except ValueError:
psk_bone.parent_index = 0
if bone.parent is not None:
rotation = bone.matrix.to_quaternion()
rotation.x = -rotation.x
rotation.y = -rotation.y
rotation.z = -rotation.z
quat_parent = bone.parent.matrix.to_quaternion().inverted()
parent_head = quat_parent @ bone.parent.head
parent_tail = quat_parent @ bone.parent.tail
location = (parent_tail - parent_head) + bone.head
else:
location = input_objects.armature_object.matrix_local @ bone.head
rot_matrix = bone.matrix @ input_objects.armature_object.matrix_local.to_3x3()
rotation = rot_matrix.to_quaternion()
psk_bone.location.x = location.x
psk_bone.location.y = location.y
psk_bone.location.z = location.z
psk_bone.rotation.x = rotation.x
psk_bone.rotation.y = rotation.y
psk_bone.rotation.z = rotation.z
psk_bone.rotation.w = rotation.w
psk.bones.append(psk_bone) psk.bones.append(psk_bone)
else:
bones = list(input_objects.armature_object.data.bones)
for bone in bones:
psk_bone = Psk.Bone()
psk_bone.name = bytes(bone.name, encoding='utf-8')
psk_bone.flags = 0
psk_bone.children_count = len(bone.children)
try:
psk_bone.parent_index = bones.index(bone.parent)
except ValueError:
psk_bone.parent_index = 0
if bone.parent is not None:
rotation = bone.matrix.to_quaternion()
rotation.x = -rotation.x
rotation.y = -rotation.y
rotation.z = -rotation.z
quat_parent = bone.parent.matrix.to_quaternion().inverted()
parent_head = quat_parent @ bone.parent.head
parent_tail = quat_parent @ bone.parent.tail
location = (parent_tail - parent_head) + bone.head
else:
location = input_objects.armature_object.matrix_local @ bone.head
rot_matrix = bone.matrix @ input_objects.armature_object.matrix_local.to_3x3()
rotation = rot_matrix.to_quaternion()
psk_bone.location.x = location.x
psk_bone.location.y = location.y
psk_bone.location.z = location.z
psk_bone.rotation.x = rotation.x
psk_bone.rotation.y = rotation.y
psk_bone.rotation.z = rotation.z
psk_bone.rotation.w = rotation.w
psk.bones.append(psk_bone)
vertex_offset = 0 vertex_offset = 0
wedge_offset = 0 wedge_offset = 0
@ -143,7 +155,6 @@ class PskBuilder(object):
material_indices.append(material_index) material_indices.append(material_index)
# FACES # FACES
# TODO: this is making the assumption that the mesh is triangulated
object.data.calc_loop_triangles() object.data.calc_loop_triangles()
poly_groups, groups = object.data.calc_smooth_groups(use_bitflags=True) poly_groups, groups = object.data.calc_smooth_groups(use_bitflags=True)
for f in object.data.loop_triangles: for f in object.data.loop_triangles:
@ -160,24 +171,25 @@ class PskBuilder(object):
# WEIGHTS # WEIGHTS
# TODO: bone ~> vg might not be 1:1, provide a nice error message if this is the case # TODO: bone ~> vg might not be 1:1, provide a nice error message if this is the case
armature = input_objects.armature_object.data if input_objects.armature_object is not None:
bone_names = [x.name for x in armature.bones] armature = input_objects.armature_object.data
vertex_group_names = [x.name for x in object.vertex_groups] bone_names = [x.name for x in armature.bones]
bone_indices = [bone_names.index(name) for name in vertex_group_names] vertex_group_names = [x.name for x in object.vertex_groups]
for vertex_group_index, vertex_group in enumerate(object.vertex_groups): bone_indices = [bone_names.index(name) for name in vertex_group_names]
bone_index = bone_indices[vertex_group_index] for vertex_group_index, vertex_group in enumerate(object.vertex_groups):
for vertex_index in range(len(object.data.vertices)): bone_index = bone_indices[vertex_group_index]
try: for vertex_index in range(len(object.data.vertices)):
weight = vertex_group.weight(vertex_index) try:
except RuntimeError: weight = vertex_group.weight(vertex_index)
continue except RuntimeError:
if weight == 0.0: continue
continue if weight == 0.0:
w = Psk.Weight() continue
w.bone_index = bone_index w = Psk.Weight()
w.point_index = vertex_offset + vertex_index w.bone_index = bone_index
w.weight = weight w.point_index = vertex_offset + vertex_index
psk.weights.append(w) w.weight = weight
psk.weights.append(w)
vertex_offset += len(psk.points) vertex_offset += len(psk.points)
wedge_offset += len(psk.wedges) wedge_offset += len(psk.wedges)