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