diff options
Diffstat (limited to 'oldStuff/3dlib/obj2tmdl - Copy.py')
| -rwxr-xr-x | oldStuff/3dlib/obj2tmdl - Copy.py | 347 |
1 files changed, 347 insertions, 0 deletions
diff --git a/oldStuff/3dlib/obj2tmdl - Copy.py b/oldStuff/3dlib/obj2tmdl - Copy.py new file mode 100755 index 0000000..e265cb5 --- /dev/null +++ b/oldStuff/3dlib/obj2tmdl - Copy.py @@ -0,0 +1,347 @@ +#!/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('<filepath=\"%s\" id=\"tex%d\" colfmt=5 />' % (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)
|