mirror of
https://github.com/DarklightGames/io_scene_psk_psa.git
synced 2024-11-13 18:00:52 +01:00
PSA exports now work!
This commit is contained in:
parent
f8ac443bb1
commit
1a96d0816e
3
.gitignore
vendored
3
.gitignore
vendored
@ -102,3 +102,6 @@ venv.bak/
|
|||||||
|
|
||||||
# mypy
|
# mypy
|
||||||
.mypy_cache/
|
.mypy_cache/
|
||||||
|
|
||||||
|
# PyCharm
|
||||||
|
.idea
|
30
src/data.py
Normal file
30
src/data.py
Normal file
@ -0,0 +1,30 @@
|
|||||||
|
from ctypes import *
|
||||||
|
|
||||||
|
class Vector3(Structure):
|
||||||
|
_fields_ = [
|
||||||
|
('x', c_float),
|
||||||
|
('y', c_float),
|
||||||
|
('z', c_float),
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
class Quaternion(Structure):
|
||||||
|
_fields_ = [
|
||||||
|
('x', c_float),
|
||||||
|
('y', c_float),
|
||||||
|
('z', c_float),
|
||||||
|
('w', c_float),
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
class Section(Structure):
|
||||||
|
_fields_ = [
|
||||||
|
('name', c_char * 20),
|
||||||
|
('type_flags', c_int32),
|
||||||
|
('data_size', c_int32),
|
||||||
|
('data_count', c_int32)
|
||||||
|
]
|
||||||
|
|
||||||
|
def __init__(self, *args, **kw):
|
||||||
|
super().__init__(*args, **kw)
|
||||||
|
self.type_flags = 1999801
|
@ -2,6 +2,8 @@ import bpy
|
|||||||
import mathutils
|
import mathutils
|
||||||
from .data import *
|
from .data import *
|
||||||
|
|
||||||
|
|
||||||
|
# https://git.cth451.me/cth451/blender-addons/blob/master/io_export_unreal_psk_psa.py
|
||||||
class PsaBuilder(object):
|
class PsaBuilder(object):
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
# TODO: add options in here (selected anims, eg.)
|
# TODO: add options in here (selected anims, eg.)
|
||||||
@ -22,6 +24,14 @@ class PsaBuilder(object):
|
|||||||
|
|
||||||
bones = list(armature.data.bones)
|
bones = list(armature.data.bones)
|
||||||
|
|
||||||
|
# The order of the armature bones and the pose bones is not guaranteed to be the same.
|
||||||
|
# As as a result, we need to reconstruct the list of pose bones in the same order as the
|
||||||
|
# armature bones.
|
||||||
|
bone_names = [x.name for x in bones]
|
||||||
|
pose_bones = [(bone_names.index(bone.name), bone) for bone in armature.pose.bones]
|
||||||
|
pose_bones.sort(key=lambda x: x[0])
|
||||||
|
pose_bones = [x[1] for x in pose_bones]
|
||||||
|
|
||||||
for bone in bones:
|
for bone in bones:
|
||||||
psa_bone = Psa.Bone()
|
psa_bone = Psa.Bone()
|
||||||
psa_bone.name = bytes(bone.name, encoding='utf-8')
|
psa_bone.name = bytes(bone.name, encoding='utf-8')
|
||||||
@ -58,8 +68,8 @@ class PsaBuilder(object):
|
|||||||
psa.bones.append(psa_bone)
|
psa.bones.append(psa_bone)
|
||||||
|
|
||||||
print('---- ACTIONS ----')
|
print('---- ACTIONS ----')
|
||||||
|
|
||||||
frame_start_index = 0
|
frame_start_index = 0
|
||||||
|
|
||||||
for action in bpy.data.actions:
|
for action in bpy.data.actions:
|
||||||
if len(action.fcurves) == 0:
|
if len(action.fcurves) == 0:
|
||||||
continue
|
continue
|
||||||
@ -73,15 +83,18 @@ class PsaBuilder(object):
|
|||||||
sequence.name = bytes(action.name, encoding='utf-8')
|
sequence.name = bytes(action.name, encoding='utf-8')
|
||||||
sequence.frame_count = frame_max - frame_min + 1
|
sequence.frame_count = frame_max - frame_min + 1
|
||||||
sequence.frame_start_index = frame_start_index
|
sequence.frame_start_index = frame_start_index
|
||||||
|
print(frame_start_index)
|
||||||
sequence.fps = 30 # TODO: fill in later with r
|
sequence.fps = 30 # TODO: fill in later with r
|
||||||
|
|
||||||
for frame in range(frame_min, frame_max + 1):
|
print(action.name, frame_min, frame_max)
|
||||||
|
|
||||||
|
frame_count = frame_max - frame_min + 1
|
||||||
|
|
||||||
|
for frame in range(frame_count):
|
||||||
context.scene.frame_set(frame)
|
context.scene.frame_set(frame)
|
||||||
|
|
||||||
print(frame)
|
for bone in pose_bones:
|
||||||
|
# TODO: is the cast-to-matrix necessary? (guessing no)
|
||||||
for bone_index, bone in enumerate(armature.pose.bones):
|
|
||||||
# TODO: is the cast-to-matrix necesssary? (guessing no)
|
|
||||||
key = Psa.Key()
|
key = Psa.Key()
|
||||||
pose_bone_matrix = bone.matrix
|
pose_bone_matrix = bone.matrix
|
||||||
|
|
||||||
@ -92,6 +105,9 @@ class PsaBuilder(object):
|
|||||||
location = pose_bone_matrix.to_translation()
|
location = pose_bone_matrix.to_translation()
|
||||||
rotation = pose_bone_matrix.to_quaternion().normalized()
|
rotation = pose_bone_matrix.to_quaternion().normalized()
|
||||||
|
|
||||||
|
if action.name == 'shoot_open' and bone.name == 'barrel':
|
||||||
|
print(location)
|
||||||
|
|
||||||
if bone.parent is not None:
|
if bone.parent is not None:
|
||||||
rotation.x = -rotation.x
|
rotation.x = -rotation.x
|
||||||
rotation.y = -rotation.y
|
rotation.y = -rotation.y
|
||||||
@ -110,6 +126,9 @@ class PsaBuilder(object):
|
|||||||
|
|
||||||
frame_start_index += 1
|
frame_start_index += 1
|
||||||
|
|
||||||
|
sequence.bone_count = len(pose_bones)
|
||||||
|
sequence.track_time = frame_count
|
||||||
|
|
||||||
psa.sequences.append(sequence)
|
psa.sequences.append(sequence)
|
||||||
|
|
||||||
return psa
|
return psa
|
||||||
|
Loading…
Reference in New Issue
Block a user