mirror of
https://github.com/DarklightGames/io_scene_psk_psa.git
synced 2024-12-02 18:27:15 +01:00
128 lines
3.5 KiB
Python
128 lines
3.5 KiB
Python
from typing import List
|
|
|
|
from ..shared.data import *
|
|
|
|
|
|
class Psk(object):
|
|
class Wedge(object):
|
|
def __init__(self, point_index: int, u: float, v: float, material_index: int = 0):
|
|
self.point_index: int = point_index
|
|
self.u: float = u
|
|
self.v: float = v
|
|
self.material_index = material_index
|
|
|
|
def __hash__(self):
|
|
return hash(f'{self.point_index}-{self.u}-{self.v}-{self.material_index}')
|
|
|
|
class Wedge16(Structure):
|
|
_fields_ = [
|
|
('point_index', c_uint32),
|
|
('u', c_float),
|
|
('v', c_float),
|
|
('material_index', c_uint8),
|
|
('reserved', c_int8),
|
|
('padding2', c_int16)
|
|
]
|
|
|
|
class Wedge32(Structure):
|
|
_fields_ = [
|
|
('point_index', c_uint32),
|
|
('u', c_float),
|
|
('v', c_float),
|
|
('material_index', c_uint32)
|
|
]
|
|
|
|
class Face(Structure):
|
|
_fields_ = [
|
|
('wedge_indices', c_uint16 * 3),
|
|
('material_index', c_uint8),
|
|
('aux_material_index', c_uint8),
|
|
('smoothing_groups', c_int32)
|
|
]
|
|
|
|
class Face32(Structure):
|
|
_pack_ = 1
|
|
_fields_ = [
|
|
('wedge_indices', c_uint32 * 3),
|
|
('material_index', c_uint8),
|
|
('aux_material_index', c_uint8),
|
|
('smoothing_groups', c_int32)
|
|
]
|
|
|
|
class Material(Structure):
|
|
_fields_ = [
|
|
('name', c_char * 64),
|
|
('texture_index', c_int32),
|
|
('poly_flags', c_int32),
|
|
('aux_material', c_int32),
|
|
('aux_flags', c_int32),
|
|
('lod_bias', c_int32),
|
|
('lod_style', c_int32)
|
|
]
|
|
|
|
class Bone(Structure):
|
|
_fields_ = [
|
|
('name', c_char * 64),
|
|
('flags', c_int32),
|
|
('children_count', c_int32),
|
|
('parent_index', c_int32),
|
|
('rotation', Quaternion),
|
|
('location', Vector3),
|
|
('length', c_float),
|
|
('size', Vector3)
|
|
]
|
|
|
|
class Weight(Structure):
|
|
_fields_ = [
|
|
('weight', c_float),
|
|
('point_index', c_int32),
|
|
('bone_index', c_int32),
|
|
]
|
|
|
|
class MorphInfo(Structure):
|
|
_fields_ = [
|
|
('name', c_char * 64),
|
|
('vertex_count', c_int32)
|
|
]
|
|
|
|
class MorphData(Structure):
|
|
_fields_ = [
|
|
('position_delta', Vector3),
|
|
('tangent_z_delta', Vector3),
|
|
('point_index', c_int32)
|
|
]
|
|
|
|
@property
|
|
def has_extra_uvs(self):
|
|
return len(self.extra_uvs) > 0
|
|
|
|
@property
|
|
def has_vertex_colors(self):
|
|
return len(self.vertex_colors) > 0
|
|
|
|
@property
|
|
def has_vertex_normals(self):
|
|
return len(self.vertex_normals) > 0
|
|
|
|
@property
|
|
def has_material_references(self):
|
|
return len(self.material_references) > 0
|
|
|
|
@property
|
|
def has_morph_data(self):
|
|
return len(self.morph_infos) > 0
|
|
|
|
def __init__(self):
|
|
self.points: List[Vector3] = []
|
|
self.wedges: List[Psk.Wedge] = []
|
|
self.faces: List[Psk.Face] = []
|
|
self.materials: List[Psk.Material] = []
|
|
self.weights: List[Psk.Weight] = []
|
|
self.bones: List[Psk.Bone] = []
|
|
self.extra_uvs: List[Vector2] = []
|
|
self.vertex_colors: List[Color] = []
|
|
self.vertex_normals: List[Vector3] = []
|
|
self.morph_infos: List[Psk.MorphInfo] = []
|
|
self.morph_data: List[Psk.MorphData] = []
|
|
self.material_references: List[str] = []
|