def write(self, stream, resources=None):
s = StreamWriter(stream)
s.u32(self.version)
tgi = resources
use_tgi = not isinstance(tgi, TGIList)
if use_tgi:
tgi = TGIList(order='IGT',
use_length=False,
package=resources,
count_size=8)
tgi.begin_write(self, stream)
if self.version >= 7:
s.i8(len(self.fabrics))
for fabric in self.fabrics:
s.i8(tgi.get_resource_index(fabric))
fabric_len = 0
with BytesIO() as fabric_stream:
fabric.write(fabric_stream, tgi)
s.u32(fabric_stream.tell())
fabric_stream.seek(0, SEEK_SET)
stream.write(fabric_stream.read(-1))
s.u32(self.target_size)
s.u32(self.part_type)
s.u8(0)
s.i32(len(self.steps))
if self.version >= 0x08:
s.u8(0)
for step in self.steps:
self.write_step(stream, tgi, step)
if use_tgi:
tgi.end_read(stream)
else:
s.u32(0)
def read(self, stream, resource=None):
s = StreamReader(stream)
self.version = s.u32()
tgi = TGIList(order='igt', count_size=8, package=resource, use_length=False)
tgi.begin_read(stream)
self.presets = [self.CasPreset(stream, tgi) for i in range(s.u32())]
self.part_name = s.s7(16, '>')
self.display_index = s.f32()
self.has_unique_texture_space = s.u8()
self.body_type = s.u32()
self.part_flags = s.u32()
self.age_gender_flags = s.u32()
self.clothing_category = s.u32()
self.naked_cas_part = tgi.get_resource(s.i8())
self.base_cas_part = tgi.get_resource(s.i8())
self.blend_fat = tgi.get_resource(s.i8())
self.blend_fit = tgi.get_resource(s.i8())
self.blend_thin = tgi.get_resource(s.i8())
self.blend_special = tgi.get_resource(s.i8())
self.draw_layer = s.u32()
self.sources = [tgi.get_resource(s.i8()) for i in range(s.u8())]
self.lod_infos = [CASLodInfo(stream) for i in range(s.u8())]
self.diffuse_refs = [tgi.get_resource(s.i8()) for i in range(s.u8())]
self.specular_refs = [tgi.get_resource(s.i8()) for i in range(s.u8())]
self.secondary_diffuse_refs = [tgi.get_resource(s.i8()) for i in range(s.u8())]
self.secondary_specular_refs = [tgi.get_resource(s.i8()) for i in range(s.u8())]
self.slot_poses = [tgi.get_resource(s.i8()) for i in range(s.u8())]
self.shoe_material = s.s7(16, '>')
tgi.end_read(stream)
def write(self, stream, resources=None):
s = StreamWriter(stream)
s.u32(self.version)
tgi = TGIList()
tgi.begin_write(stream)
for shader_key in self.shader_keys: shader_key.write(stream, tgi)
s.u32(tgi.get_resource_index(self.sub_skin_ramp))
s.u32(tgi.get_resource_index(self.tone_ramp))
for texture_key in self.texture_keys: texture_key.write(stream, tgi)
s.u8(self.is_dominant)
def read(self, stream, resources=None):
s = StreamReader(stream)
self.version = s.u32()
tgi = TGIList()
tgi.begin_read(stream)
self.shader_keys = [self.ShaderKey(self, stream, tgi) for i in range(s.i32())]
self.sub_skin_ramp = tgi.get_resource(s.u32())
self.tone_ramp = tgi.get_resource(s.u32())
self.texture_keys = [self.TextureKey(self, stream, tgi) for i in range(s.i32())]
self.dominant = bool(s.u8())
def read(self, stream, resources=None):
s = StreamReader(stream)
self.version = s.u32()
tgi = TGIList(package=resources)
tgi.begin_read(stream)
self.locale_key = s.u64()
self.indexers = [tgi.get_resource(s.i32()) for i in range(s.i32())]
self.is_bi_directional = bool(s.u32())
self.cas_panel_group = s.u32()
self.sort_index = s.i32()
assert s.u32() == 0
tgi.end_read(stream)
def write(self, stream, resources=None):
s = StreamWriter(stream)
s.u32(self.version)
tgi = TGIList(package=resources)
tgi.begin_write(stream)
s.u64(self.locale_key)
for indexer in self.indexers:
s.u32(tgi.get_resource_index(indexer))
s.u32(0 if not self.is_bi_directional else 1)
s.u32(self.cas_panel_group)
s.i32(self.sort_index)
s.u32(0)
tgi.end_read(stream)
def write(self, stream, resource=None):
s = StreamWriter(stream)
s.u32(self.version)
tgi = TGIList(use_length=self.version >= 7, add_eight=True)
tgi.begin_write(stream)
s.s7(self.part_name, 16, '>')
s.u32(self.blend_type)
s.i32(len(self.entries))
for entry in self.entries:
s.u32(entry.region_flags)
s.i32(len(entry.geom_entries))
for geom_entry in entry.geom_entries: geom_entry.write_rcol(stream, tgi)
s.i32(len(entry.bone_entries))
for bone_entry in entry.bone_entries: bone_entry.write_rcol(stream, tgi)
tgi.end_write(stream)
def read_rcol(self, stream, rcol):
s = StreamReader(stream)
self.read_tag(stream)
self.version = s.u32()
tgi = TGIList()
tgi.begin_read(stream)
self.shader = s.u32()
if self.shader:
end_material = s.u32() + stream.tell()
self.material = MaterialBlock()
self.material.read(stream, tgi)
assert stream.tell() == end_material
self.merge_group = s.u32()
self.sort_order = s.u32()
cVertices = s.u32()
self.vertex_format.read(stream)
for vertex_index in range(cVertices):
vertex = Vertex()
for declaration in self.vertex_format.declarations:
if declaration.usage == self.VertexFormat.Declaration.USAGE.POSITION:
vertex.position = [s.f32(), s.f32(), s.f32()]
elif declaration.usage == self.VertexFormat.Declaration.USAGE.NORMAL:
vertex.normal = [s.f32(), s.f32(), s.f32()]
elif declaration.usage == self.VertexFormat.Declaration.USAGE.UV:
uv = [s.f32(), s.f32()]
if vertex.uv == None:
vertex.uv = []
vertex.uv.append(uv)
elif declaration.usage == self.VertexFormat.Declaration.USAGE.BLEND_INDEX:
vertex.blend_indices = [s.i8(), s.i8(), s.i8(), s.i8()]
elif declaration.usage == self.VertexFormat.Declaration.USAGE.BLEND_WEIGHT:
vertex.blend_weights = [s.f32(), s.f32(), s.f32(), s.f32()]
elif declaration.usage == self.VertexFormat.Declaration.USAGE.TANGENT:
vertex.tangent = [s.f32(), s.f32(), s.f32()]
elif declaration.usage == self.VertexFormat.Declaration.USAGE.COLOUR:
vertex.colour = [s.u8(), s.u8(), s.u8(), s.u8()]
elif declaration.usage == self.VertexFormat.Declaration.USAGE.ID:
vertex.id = s.u32()
self.vertices.append(vertex)
assert s.u32() == 1
bytes_per_index = s.u8()
assert bytes_per_index == 2
self.indices = [[s.u16() for i in range(3)]
for i in range(int(s.u32() / 3))]
self.skin_controller = tgi.get_resource(s.u32())
self.bones = [s.u32() for i in range(s.u32())]
tgi.end_read(stream)
def write_rcol(self, stream, rcol):
s = StreamWriter(stream)
self.write_tag(stream)
s.u32(self.version)
tgi = TGIList(32, 'TGI')
tgi.begin_write(stream)
s.hash(self.shader)
if self.shader:
self.material.write(stream, tgi)
s.u32(self.merge_group)
s.u32(self.sort_order)
s.u32(len(self.vertices))
self.vertex_format.from_vertex(self.vertices[0])
self.vertex_format.write(stream)
for vertex in self.vertices:
uv_index = 0
for declaration in self.vertex_format.declarations:
if declaration.usage == self.VertexFormat.Declaration.USAGE.POSITION:
for i in range(3):
s.f32(vertex.position)
elif declaration.usage == self.VertexFormat.Declaration.USAGE.NORMAL:
for i in range(3):
s.f32(vertex.normal)
elif declaration.usage == self.VertexFormat.Declaration.USAGE.UV:
for i in range(2):
s.f32(vertex.uv[uv_index])
uv_index += 1
elif declaration.usage == self.VertexFormat.Declaration.USAGE.BLEND_INDEX:
for i in range(4):
s.i8(vertex.blend_indices)
elif declaration.usage == self.VertexFormat.Declaration.USAGE.BLEND_WEIGHT:
for i in range(4):
s.f32(vertex.blend_weights)
elif declaration.usage == self.VertexFormat.Declaration.USAGE.TANGENT:
for i in range(3):
s.f32(vertex.tangents)
elif declaration.usage == self.VertexFormat.Declaration.USAGE.COLOUR:
for i in range(4):
s.i8(vertex.colour)
elif declaration.usage == self.VertexFormat.Declaration.USAGE.ID:
s.u32(vertex.id)
s.u32(tgi.get_resource_index(self.skin_controller))
s.u32(len(self.bones))
for bone in self.bones:
s.u32(bone)
def write_rcol(self, stream, rcol):
self.write_tag(stream)
s = StreamWriter(stream)
s.u32(self.version)
tgi = TGIList()
tgi.begin_write(stream)
s.u8(len(self.entries))
for entry in self.entries:
s.u8(entry.TYPE)
entry.write(stream, tgi)
s.u8(2)
self.bounds.write(stream)
s.u32(self.flags)
if self.routing_footprint.key != ResourceKey():
s.u8(1)
s.tgi(self.routing_footprint.key, 'TGI')
else: s.u8(0)
tgi.end_write(stream)
def read_rcol(self, stream, rcol):
self.read_tag(stream)
s = StreamReader(stream)
tgi = TGIList()
self.version = s.u32()
tgi.begin_read(stream)
cEntries = s.u8()
for entry_index in range(cEntries):
type = s.u8()
entry = self.Entry.create_instance(type)
entry.read(stream, tgi)
self.entries.append(entry)
assert s.u8() == 0x02
self.bounds.read(stream)
self.flags = s.u32()
if s.u8():
self.routing_footprint = tgi.get_resource(s.u32())
tgi.end_read(stream)
def write(self, stream, resource=None):
s = StreamWriter(stream)
s.u32(self.version)
tgi = TGIList('igt', 8, False)
tgi.begin_write(stream)
for preset in self.presets:
preset.write(stream)
s.s7(self.part_name, 16, '>')
s.f32(self.display_index)
s.u8(self.has_unique_texture_space)
s.u32(self.body_type)
s.u32(self.part_flags)
s.u32(self.age_gender_flags)
s.u32(self.clothing_category)
s.i8(tgi.get_resource_index(self.naked_cas_part))
s.i8(tgi.get_resource_index(self.base_cas_part))
s.i8(tgi.get_resource_index(self.blend_fat))
s.i8(tgi.get_resource_index(self.blend_fit))
s.i8(tgi.get_resource_index(self.blend_thin))
s.i8(tgi.get_resource_index(self.blend_special))
s.u32(self.draw_layer)
s.i8(len(self.sources))
for source in self.sources:
s.i8(tgi.get_resource_index(source))
s.i8(len(self.lod_infos))
for lod_info in self.lod_infos:
lod_info.write(stream)
s.i8(len(self.diffuse_refs))
for diffuse_ref in self.diffuse_refs:
s.i8(tgi.get_resource_index(diffuse_ref))
s.i8(len(self.specular_refs))
for specular_ref in self.specular_refs:
s.i8(tgi.get_resource_index(specular_ref))
s.u8(len(self.secondary_diffuse_refs))
for block in self.secondary_diffuse_refs:
s.u8(tgi.get_resource_index(block))
s.u8(len(self.secondary_specular_refs))
for block in self.secondary_specular_refs:
s.u8(tgi.get_resource_index(block))
s.u8(len(self.slot_poses))
for block in self.slot_poses:
s.u8(tgi.get_resource_index(block))
s.s7(self.shoe_material, 16, '>')
tgi.end_write(stream)
def read(self, stream, resource=None):
s = StreamReader(stream)
self.version = s.u32()
tgi = TGIList(use_length=self.version >= 7, add_eight=True)
tgi.begin_read(stream)
self.part_name = s.s7(16, '>')
self.blend_type = s.u32()
if self.version >= 8:
key = s.tgi()
bgeo = ExternalResource(key=key)
self.blend_geometry = bgeo
cEntries = s.i32()
for i in range(cEntries):
entry = self.RegionEntry()
entry.region_flags = s.u32()
entry.geom_entries = [self.GeomEntry(stream, tgi) for i in range(s.i32())]
entry.bone_entries = [self.BoneEntry(stream, tgi) for i in range(s.i32())]
self.entries.append(entry)
tgi.end_read(stream)
def read_xml(self, xml_string, resources):
def parse_value(value_string, resources):
items = []
strings = value_string.split(',')
for string in strings:
if string[:4] == 'key:':
keyVals = string[4:].split(':')
key = ResourceKey(int(keyVals[0], 16), int(keyVals[1], 16),
int(keyVals[2], 16))
items.append(key)
elif string in ('true', 'false'):
items.append(string == 'true')
else:
items.append(string)
return items if len(items) > 1 else items[0]
doc = parseString(xml_string)
presetElement = doc.documentElement
complateElement = presetElement.getElementsByTagName('complate')[0]
if not resources:
resources = TGIList()
self.complate = self.Element()
self.complate.name = complateElement.nodeName
self.complate.resource = parse_value(
complateElement.attributes['reskey'].nodeValue, resources)
for keyElement in complateElement.getElementsByTagName('value'):
key = keyElement.attributes['key'].nodeValue
value = keyElement.attributes['value'].nodeValue
self.complate.values[key] = parse_value(value, resources)
for patternElement in complateElement.getElementsByTagName('pattern'):
pattern = self.Element()
pattern.name = patternElement.nodeName
pattern.resource = parse_value(
patternElement.attributes['reskey'].nodeValue, resources)
pattern.variable = parse_value(
patternElement.attributes['variable'].nodeValue, resources)
for keyElement in patternElement.getElementsByTagName('value'):
key = keyElement.attributes['key'].nodeValue
value = keyElement.attributes['value'].nodeValue
pattern.values[key] = parse_value(value, resources)
self.complate.patterns.append(pattern)
def deserialize(self, stream, parent_tgi):
def read_element(s, tgi_list):
def read_complate_string(s):
a = s.i8()
if not a: return None
if a & 0x80: return s.chars(s.i8() if a & 0x40 else a & 0x3F)
if a & 0x40: a = (a & 0x3F) + s.i8()
return self.complate_string_lookup[a]
def read_typecode(s, tgi_list):
tc = s.u8()
if tc == 1: return read_complate_string(s)
elif tc == 0x02: return [s.u8() for i in range(4)]
elif tc == 0x03: return tgi_list.get_resource(s.i8())
elif tc == 0x04: return s.f32()
elif tc == 0x05: return [s.f32() for i in range(2)]
elif tc == 0x06: return [s.f32() for i in range(3)]
elif tc == 0x07: return bool(s.i8())
else: raise Exception("Unknown typecode %02X" % tc)
element = Preset.Element()
element.resource = tgi_list.get_resource(s.u8())
element.name = read_complate_string(s)
element.variable = read_complate_string(s)
for i in range(s.i32()):
name = read_complate_string(s)
value = read_typecode(s, tgi_list)
element.values[name] = value
element.patterns = [
read_element(s, tgi_list) for i in range(s.i32())
]
return element
s = StreamReader(stream)
unk = s.i16()
preset_tgi = TGIList(use_length=True)
preset_tgi.begin_read(stream)
element = read_element(s, preset_tgi)
preset_tgi.end_read(stream)
return element
def read(self, stream, resources=None):
s = StreamReader(stream)
self.version = s.u32()
use_tgi = not isinstance(resources, TGIList)
tgi = resources
if use_tgi:
tgi = TGIList(order='IGT',
use_length=False,
package=resources,
count_size=8)
tgi.begin_read(stream)
if self.version >= 7:
cFabrics = s.i8()
for fabric_index in range(cFabrics):
key = tgi.get_resource(s.u8())
fabric = TextureCompositor(key)
fabric_len = s.u32()
with BytesIO() as fabric_stream:
fabric_stream.write(stream.read(fabric_len))
fabric_stream.seek(0, SEEK_SET)
fabric.read(fabric_stream, tgi)
self.fabrics.append(fabric)
self.target_size = s.u32()
self.part_type = s.u32()
assert s.u8() == 0
cSteps = s.i32()
if self.version >= 0x08:
assert s.u8() == 0
self.steps = []
for step_index in range(cSteps):
self.steps.append(self.read_step(stream, tgi))
if use_tgi:
tgi.end_read(stream)
else:
assert s.u32() == 0
