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())