Repository for mbEditorPro 2.0
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

126 lines
4.9 KiB

# • ▌ ▄ ·. ▄▄▄▄· ▄▄▄ .·▄▄▄▄ ▪ ▄▄▄▄▄ ▄▄▄ ▄▄▄·▄▄▄
# ·██ ▐███▪▐█ ▀█▪ ▀▄.▀·██▪ ██ ██ •██ ▪ ▀▄ █· ▐█ ▄█▀▄ █·▪
# ▐█ ▌▐▌▐█·▐█▀▀█▄ ▐▀▀▪▄▐█· ▐█▌▐█· ▐█.▪ ▄█▀▄ ▐▀▀▄ ██▀·▐▀▀▄ ▄█▀▄
# ██ ██▌▐█▌██▄▪▐█ ▐█▄▄▌██. ██ ▐█▌ ▐█▌·▐█▌.▐▌▐█•█▌ ▐█▪·•▐█•█▌▐█▌.▐▌
# ▀▀ █▪▀▀▀·▀▀▀▀ ▀▀▀ ▀▀▀▀▀• ▀▀▀ ▀▀▀ ▀█▄▀▪.▀ ▀ .▀ .▀ ▀ ▀█▄▀▪
# Magicbane Emulator Project © 2013 - 2022
# www.magicbane.com
from collections import OrderedDict
from arcane.util import ResStream
class ArcBone:
def load_binary(self, stream: ResStream):
self.bone_id = stream.read_dword()
self.bone_name = stream.read_string()
self.bone_direction = stream.read_tuple()
self.bone_length = stream.read_float()
self.bone_axis = stream.read_tuple()
self.bone_dof = stream.read_string()
self.bone_order = stream.read_tuple()
self.bone_position = stream.read_tuple()
self.bone_orientation = stream.read_tuple()
self.bone_u0 = stream.read_bool()
self.bone_u1 = stream.read_bool()
num = stream.read_dword()
self.bone_hierarchy = [ArcBone() for _ in range(num)]
for bone in self.bone_hierarchy:
bone.load_binary(stream)
def save_binary(self, stream: ResStream):
stream.write_dword(self.bone_id)
stream.write_string(self.bone_name)
stream.write_tuple(self.bone_direction)
stream.write_float(self.bone_length)
stream.write_tuple(self.bone_axis)
stream.write_string(self.bone_dof)
stream.write_tuple(self.bone_order)
stream.write_tuple(self.bone_position)
stream.write_tuple(self.bone_orientation)
stream.write_bool(self.bone_u0)
stream.write_bool(self.bone_u1)
stream.write_dword(len(self.bone_hierarchy))
for bone in self.bone_hierarchy:
bone.save_binary(stream)
def load_json(self, data):
self.bone_id = data['bone_id']
self.bone_name = data['bone_name']
self.bone_direction = data['bone_direction']
self.bone_length = data['bone_length']
self.bone_axis = data['bone_axis']
self.bone_dof = data['bone_dof']
self.bone_order = data['bone_order']
self.bone_position = data['bone_position']
self.bone_orientation = data['bone_orientation']
self.bone_u0 = data['bone_u0']
self.bone_u1 = data['bone_u1']
self.bone_hierarchy = []
for bone_data in data['bone_hierarchy']:
bone = ArcBone()
bone.load_json(bone_data)
self.bone_hierarchy.append(bone)
def save_json(self):
data = OrderedDict()
data['bone_id'] = self.bone_id
data['bone_name'] = self.bone_name
data['bone_direction'] = self.bone_direction
data['bone_length'] = self.bone_length
data['bone_axis'] = self.bone_axis
data['bone_dof'] = self.bone_dof
data['bone_order'] = self.bone_order
data['bone_position'] = self.bone_position
data['bone_orientation'] = self.bone_orientation
data['bone_u0'] = self.bone_u0
data['bone_u1'] = self.bone_u1
data['bone_hierarchy'] = []
for bone in self.bone_hierarchy:
data['bone_hierarchy'].append(bone.save_json())
return data
class ArcSkeleton:
def load_binary(self, stream: ResStream):
self.skeleton_name = stream.read_string()
num = stream.read_dword()
self.skeleton_motion = [
[
stream.read_qword(),
stream.read_qword(),
] for _ in range(num)
]
self.skeleton_root = ArcBone()
self.skeleton_root.load_binary(stream)
def save_binary(self, stream: ResStream):
stream.write_string(self.skeleton_name)
stream.write_dword(len(self.skeleton_motion))
for i in range(len(self.skeleton_motion)):
stream.write_qword(self.skeleton_motion[i][0])
stream.write_qword(self.skeleton_motion[i][1])
self.skeleton_root.save_binary(stream)
def load_json(self, data):
self.skeleton_name = data['skeleton_name']
self.skeleton_motion = data['skeleton_motion']
self.skeleton_root = ArcBone()
self.skeleton_root.load_json(data['skeleton_root'])
def save_json(self):
data = OrderedDict()
data['skeleton_name'] = self.skeleton_name
data['skeleton_motion'] = self.skeleton_motion
data['skeleton_root'] = self.skeleton_root.save_json()
return data