model/boblamp: add C representation of md5mesh

md5/generate.py

@@ -0,0 +1 @@
+../regs/gen/generate.py

md5/md5.h

@@ -0,0 +1,48 @@
+#pragma once
+
+struct md5_mesh_joint {
+  const char * bone_name;
+  const int parent_index;
+  const vec3 pos;
+  const vec4 orient;
+};
+
+struct md5_mesh_vert {
+  const int vert_index;
+  const vec2 tex;
+  const int weight_index;
+  const int weight_elem;
+};
+
+struct md5_mesh_tri {
+  const int tri_index;
+  struct {
+    const int a;
+    const int b;
+    const int c;
+  } vert_index;
+};
+
+struct md5_mesh_weight {
+  const int weight_index;
+  const int joint_index;
+  const float weight_value;
+  const vec3 pos;
+};
+
+struct md5_mesh_mesh {
+  const char * shader;
+  const int num_verts;
+  const md5_mesh_vert * verts;
+  const int num_tris;
+  const md5_mesh_tri * tris;
+  const int num_weights;
+  const md5_mesh_weight * weights;
+};
+
+struct md5_mesh {
+  const int num_joints;
+  const int num_meshes;
+  const md5_mesh_joint * joints;
+  const md5_mesh_mesh * meshes;
+};

md5/md5mesh.py

@@ -11,7 +11,7 @@ class MD5MeshJoint:
     z_pos: float = None
     x_orient: float = None
     y_orient: float = None
-    y_orient: float = None
+    z_orient: float = None
 
 @dataclass
 class MD5MeshVert:
@@ -215,9 +215,7 @@ def parse_ordered_list(l, ix, md5mesh):
     if string == "joints":
         ix = parse_joints(l, ix, md5mesh)
     elif string == "mesh":
-        print("parse-mesh1", ix)
         ix = parse_mesh(l, ix, md5mesh)
-        print("parse-mesh2", ix)
     else:
         assert False, string
 
@@ -233,12 +231,11 @@ def parse_file(l):
     while ix < len(l):
         ix = parse_ordered_list(l, ix, md5mesh)
 
-    pprint(md5mesh)
+    return md5mesh
 
 if __name__ == "__main__":
     with open(sys.argv[1], 'r') as f:
         buf = f.read()
-
     l = [i.strip() for i in buf.split('\n') if i.strip()]
-
-    parse_file(l)
+    md5mesh = parse_file(l)
+    pprint(md5mesh)

md5/md5mesh_gen.py

@@ -0,0 +1,129 @@
+from generate import renderer
+from math import sqrt
+import sys
+import md5mesh
+
+def vec2(a, b):
+    return f"{{{a:.6f}, {b:.6f}}}"
+
+def vec3(a, b, c):
+    return f"{{{a:.6f}, {b:.6f}, {c:.6f}}}"
+
+def vec4(a, b, c, d):
+    return f"{{{a:.6f}, {b:.6f}, {c:.6f}, {d:.6f}}}"
+
+def unit_quaternion_w(x, y, z):
+    t = 1.0 - (x * x) - (y * y) - (z * z);
+    if t < 0.0:
+        return 0.0
+    else:
+        return -sqrt(t)
+
+def render_md5_mesh_joint(joint):
+    yield f'.bone_name = "{joint.bone_name}",'
+    yield f".parent_index = {joint.parent_index},"
+    pos = vec3(joint.x_pos, joint.y_pos, joint.z_pos)
+    yield f".pos = {pos},"
+    w = unit_quaternion_w(joint.x_orient, joint.y_orient, joint.z_orient)
+    orient = vec4(joint.x_orient, joint.y_orient, joint.z_orient, w)
+    yield f".orient = {orient},"
+
+def render_md5_mesh_vert(vert):
+    yield f".vert_index = {vert.vert_index},"
+    tex = vec2(vert.tex_u, vert.tex_v)
+    yield f".tex = {tex},"
+    yield f".weight_index = {vert.weight_index},"
+    yield f".weight_elem = {vert.weight_elem},"
+
+def render_md5_mesh_tri(tri):
+    yield f".tri_index = {tri.tri_index},"
+    vi = f"{{{tri.vert_index2}, {tri.vert_index2}, {tri.vert_index3}}}"
+    yield f".vert_index = {vi},"
+
+def render_md5_mesh_weight(weight):
+    yield f".weight_index = {weight.weight_index},"
+    yield f".joint_index = {weight.joint_index},"
+    yield f".weight_value = {weight.weight_value},"
+    pos = vec3(weight.x_pos, weight.y_pos, weight.z_pos)
+    yield f".pos = {pos},"
+
+def render_md5_mesh_mesh(prefix, i, mesh):
+    yield f'.shader = "{mesh.shader}",'
+    yield f".num_verts = {mesh.num_verts},"
+    yield f".verts = {prefix}_{i}_verts,"
+    yield f".num_tris = {mesh.num_tris},"
+    yield f".tris = {prefix}_{i}_tris,"
+    yield f".num_weights = {mesh.num_weights},"
+    yield f".weights = {prefix}_{i}_weights,"
+
+def render_md5_mesh(prefix, m):
+    yield f".num_joints = {m.num_joints},"
+    yield f".num_meshes = {m.num_meshes},"
+    yield f".joints = {prefix}_joints,"
+    yield f".meshes = {prefix}_meshes,"
+
+def render_md5_mesh_verts(prefix, i, verts):
+    yield f"struct md5_mesh_vert {prefix}_{i}_verts[] = {{"
+    for vert in verts:
+        yield "{"
+        yield from render_md5_mesh_vert(vert)
+        yield "},"
+    yield "};"
+
+def render_md5_mesh_tris(prefix, i, tris):
+    yield f"struct md5_mesh_tri {prefix}_{i}_tris[] = {{"
+    for tri in tris:
+        yield "{"
+        yield from render_md5_mesh_tri(tri)
+        yield "},"
+    yield "};"
+
+def render_md5_mesh_weights(prefix, i, weights):
+    yield f"struct md5_mesh_weight {prefix}_{i}_weights[] = {{"
+    for weight in weights:
+        yield "{"
+        yield from render_md5_mesh_weight(weight)
+        yield "},"
+    yield "};"
+
+def render_md5_mesh_joints(prefix, joints):
+    yield f"struct md5_mesh_joint {prefix}_joints[] = {{"
+    for joint in joints:
+        yield "{"
+        yield from render_md5_mesh_joint(joint)
+        yield "},"
+    yield "};"
+
+def render_md5_mesh_meshes(prefix, meshes):
+    for i, mesh in enumerate(meshes):
+        yield from render_md5_mesh_verts(prefix, i, mesh.verts)
+        yield from render_md5_mesh_tris(prefix, i, mesh.tris)
+        yield from render_md5_mesh_weights(prefix, i, mesh.weights)
+
+    yield f"struct md5_mesh_mesh {prefix}_meshes[] = {{"
+    for i, mesh in enumerate(meshes):
+        yield "{"
+        yield from render_md5_mesh_mesh(prefix, i, mesh)
+        yield "},"
+    yield "};"
+
+def render_mesh(prefix, m):
+    yield from render_md5_mesh_joints(prefix, m.joints)
+    yield from render_md5_mesh_meshes(prefix, m.meshes)
+
+    yield f"struct md5_mesh {prefix}_mesh = {{"
+    yield from render_md5_mesh(prefix, m)
+    yield "};"
+
+def render_all(prefix, m):
+    yield from render_mesh(prefix, m)
+
+if __name__ == "__main__":
+    with open(sys.argv[1], 'r') as f:
+        buf = f.read()
+    l = [i.strip() for i in buf.split('\n') if i.strip()]
+    m = md5mesh.parse_file(l)
+
+    render, out = renderer()
+    render(render_all(sys.argv[2], m))
+    sys.stdout.write(out.getvalue())