From 91d99046b83b905cca79406cd219bb500da8613b Mon Sep 17 00:00:00 2001 From: Treeki Date: Sun, 13 Mar 2011 01:31:59 +0100 Subject: a bit more cleaning up --- oldStuff/3dlib/obj2tmdl - Copy.py | 347 -------------------------------------- 1 file changed, 347 deletions(-) delete mode 100755 oldStuff/3dlib/obj2tmdl - Copy.py (limited to 'oldStuff/3dlib/obj2tmdl - Copy.py') diff --git a/oldStuff/3dlib/obj2tmdl - Copy.py b/oldStuff/3dlib/obj2tmdl - Copy.py deleted file mode 100755 index e265cb5..0000000 --- a/oldStuff/3dlib/obj2tmdl - Copy.py +++ /dev/null @@ -1,347 +0,0 @@ -#!/usr/bin/env python -# Obj2tmdl -# Converter for Wavefront OBJ => Treeki Model - -import os -import struct -import sys -u32 = struct.Struct('>I') -u16 = struct.Struct('>H') -u8 = struct.Struct('>B') -f32 = struct.Struct('>f') - -sys.path.append('/home/Treeki/Wii.py/Wii.py') -import Wii - -class MaterialLib: - class Material: - pass - - - def __init__(self, filename): - self.current_mat = None - self.materials = {} - - for line in open(filename, 'r'): - self.parse_line(line) - - - def parse_line(self, line): - line = line.strip() - if line == '' or line[0] == '#': - return - - line = line.split() - cmd = line[0] - - if cmd == 'newmtl': - self.current_mat = MaterialLib.Material() - self.materials[line[1]] = self.current_mat - - elif cmd == 'map_Ka': - self.current_mat.texture = ' '.join(line[1:]) - - - def prepare_textures(self): - # get a list of every texture file used - self.texture_id = {} - textures = [] - id = 0 - - for mat in self.materials.values(): - if hasattr(mat, 'texture'): - if mat.texture not in textures: - textures.append(mat.texture) - self.texture_id[mat.texture] = id - id += 1 - - - if len(textures) == 0: - self.tpl = None - return - - # create a gxtexconv script file and an ID mapping - #script = [] - #self.texture_id = {} - # - #for tex, id in zip(textures, range(len(textures))): - # script.append('' % (tex.replace('\\','/'),id)) - # self.texture_id[id] = tex - # - #print script - # - #open('texture_script_temp_0991.scf', 'w').write('\n'.join(script)) - # - #os.system('gxtexconv -s texture_script_temp_0991.scf -o generated_texture_0991.tpl') - - # fix this later - texture_files = [tex.replace('\\','/').replace(' ','\\ ').replace('.jpg','.png') for tex in textures] - print 'zetsubou wpng RGB5A3 %s generated_texture_0991.tpl' % ' '.join(texture_files) - os.system('zetsubou wpng RGB5A3 %s generated_texture_0991.tpl' % ' '.join(texture_files)) - self.tpl = open('generated_texture_0991.tpl', 'rb').read() - - #os.remove('texture_script_temp_0991.scf') - os.remove('generated_texture_0991.tpl') - - - -class ObjReader: - class ObjShape: - def __init__(self, name): - self.faces = [] - self.shape_name = name - self.material_name = None - - - class ObjFace: - __slots__ = ('vertices', 'texcoords', 'normals') - - def __init__(self, vertices, texcoords, normals): - self.vertices = vertices - self.texcoords = texcoords - self.normals = normals - - - def __init__(self): - self.material_lib = None - - self.vertex_lists = [] - self.texcoord_lists = [] - self.normal_lists = [] - self.shapes = [] - - self.current_shape = None - - - def parse_line(self, line): - line = line.strip() - if line == '' or line[0] == '#': - return - - line = line.split() - cmd = line[0] - - if cmd == 'v': - self.vertex_lists.append(map(float, line[1:])) - - elif cmd == 'vt': - self.texcoord_lists.append(map(float, line[1:])[:2]) # chop off third coord - - elif cmd == 'vn': - self.normal_lists.append(map(float, line[1:])) - - elif cmd == 'f': - v, t, n = [], [], [] - - for entry in line[1:]: - entry_split = map(lambda x:-1 if x == '' else int(x), entry.split('/')) - - v.append(self.vertex_lists[entry_split[0] - 1]) - - if entry_split[1] == -1: - t.append([0.0, 0.0]) - else: - t.append(self.texcoord_lists[entry_split[1] - 1]) - - if entry_split[2] == -1: - n.append([0.0, 0.0, 0.0]) - else: - n.append(self.normal_lists[entry_split[2] - 1]) - - self.current_shape.faces.append(ObjReader.ObjFace(v, t, n)) - - #elif cmd == 'g': - # self.current_shape = ObjReader.ObjShape(line[1]) - # self.shapes.append(self.current_shape) - - elif cmd == 'mtllib': - self.material_lib = MaterialLib(' '.join(line[1:])) - - elif cmd == 'usemtl': - self.current_shape = ObjReader.ObjShape(line[1]) - self.shapes.append(self.current_shape) - self.current_shape.material_name = line[1] - - - -# === Treeki Model Format === -# Header: -# -# struct tmdl_header { -# u32 magic; // always TMDL -# u32 version; // currently 2 -# u32 shape_count; -# }; -# [[Followed by an array of u32s for each shape offset]] -# -# struct tmdl_shape { -# u32 dl_offset; // must be aligned to 0x20 -# u32 dl_size; // must be aligned to 0x20 -# u32 mat_offset; -# }; -# -# struct tmdl_material { -# u32 texture_id; // once the model is bound, points to id within tpl -# }; -# -# dl_offset points to a GX display list which is used for -# rendering the model in question. - - -class TmdlWriter: - def __init__(self): - self.shapes = [] - - - def build_with_obj(self, obj): - self.shapes = obj.shapes - self.material_lib = obj.material_lib - self.materials = obj.material_lib.materials - - self.material_lib.prepare_textures() - - - def pack(self): - offset = 0 - - header = struct.pack('>4sI I', 'TMDL', 2, len(self.shapes)) - offset += len(header) - - # reserve space for the shape offsets, we'll add them later - offset += (4 * len(self.shapes)) - - # create the materials - material_offs = {} - material_data = '' - - for name,material in self.materials.iteritems(): - material_offs[name] = offset - material_struct = self.build_material_struct(material) - material_data += material_struct - offset += len(material_struct) - - - # create the shapes -- first, we'll build the DLs - shape_display_lists = map(self.build_display_list_with_shape, self.shapes) - - # calculate offsets to every display list - # shape struct is currently 12 bytes, change this if the size changes - # also, make sure they're aligned to an offset of 0x20 - temp_offset = offset + (len(self.shapes) * 12) - - if temp_offset & 0x1f != 0: - aligned_offset = (temp_offset + 0x20) & ~0x1f - dl_start_padding = '\0' * (aligned_offset - temp_offset) - temp_offset = aligned_offset - else: - dl_start_padding = '' - - shape_display_list_offsets = [] - for dl in shape_display_lists: - shape_display_list_offsets.append(temp_offset) - temp_offset += len(dl) - - - # and now, create the shape structs themselves - shape_offsets = [] - shape_data = '' - - for shape, dl, dl_offset in zip(self.shapes, shape_display_lists, shape_display_list_offsets): - # first off, store the offset - shape_offsets.append(offset) - - # now build the struct - shape_struct = struct.pack('>III', dl_offset, len(dl), material_offs[shape.material_name]) - shape_data += shape_struct - offset += len(shape_struct) - - - # almost there! - tmdl_bits = [header] - tmdl_add = tmdl_bits.append - - for offs in shape_offsets: - tmdl_add(u32.pack(offs)) - - tmdl_add(material_data) - tmdl_add(shape_data) - tmdl_add(dl_start_padding) - tmdl_bits += shape_display_lists - - return ''.join(tmdl_bits) - - - def build_material_struct(self, material): - texture_id = 0xFFFFFFFF - if hasattr(material, 'texture'): - texture_id = self.material_lib.texture_id[material.texture] - - return u32.pack(texture_id) - - - def build_display_list_with_shape(self, shape): - # http://hitmen.c02.at/files/yagcd/yagcd/chap5.html - # assume vertex descriptor: [data is specified in this order] - # GXClearVtxDesc() - # GXSetVtxDesc(GX_VA_POS, GX_DIRECT) - # GXSetVtxDesc(GX_VA_NRM, GX_DIRECT) - # GXSetVtxDesc(GX_VA_TEX0, GX_DIRECT) - - vtx_num = reduce(lambda x,y:x+len(y.vertices), shape.faces, 0) - - dl_bits = [] - dl_add = dl_bits.append - - # commands: GX_QUADS = 0x80, GX_TRIANGLES = 0x90 - dl_add('\x90') # opcode - dl_add(u16.pack(vtx_num)) - - # now send over the data - for face in shape.faces: - for v, t, n in zip(face.vertices, face.texcoords, face.normals): - for f in v: - dl_add(f32.pack(f)) - - for f in n: - dl_add(f32.pack(f)) - - for f in t: - dl_add(f32.pack(f)) - - # done! - return self.build_display_list(dl_bits) - - - def build_display_list(self, dl_bits): - # assemble it - dl = ''.join(dl_bits) - - # pad the display list to 0x20 bytes with GX_NOP (0x00) - if len(dl) % 0x20 != 0: - pad_count = ((len(dl) + 0x20) & ~0x1F) - len(dl) - dl += '\x00' * pad_count - - return dl - - -#file = r'H:\ISOs\NSMBWii\testmush3' -#file = 'testmdl2' -#file = 'simple2' -file = sys.argv[1] -shortfn = file[file.rfind('/')+1:] - -obj = ObjReader() -for line in open(file+'.obj'): - obj.parse_line(line) - -tmdl = TmdlWriter() -tmdl.build_with_obj(obj) - -tmdl_file = tmdl.pack() -tpl_file = tmdl.material_lib.tpl - -arc = Wii.U8() -arc['t3d'] = None -arc['t3d/mdl_%s.tmdl' % shortfn] = tmdl_file -if tpl_file: arc['t3d/tex_%s.tpl' % shortfn] = tpl_file -arc.dumpFile('%s.arc' % file) -- cgit v1.2.3