Export script

#!BPY

"""
Name: 'MX Simulator Model (.jm)...'
Blender: 232
Group: 'Export'
Tooltip: 'Export selected mesh for MX Simulator'
"""

__author__ = "Josh Vanderhoof"
__version__ = "0.1"

__bpydoc__ = """\
This script exports meshes for use in MX Simulator.

Usage:

Select meshes to be exported and run this script from "File->Export" menu.
"""


import Blender
import sys
from Blender import *

def reduce_verts(v, t):
    s = []  # sorted with original indices

    for i in xrange(0, len(v)):
        s.append((v[i], i))

    s.sort()
    r = []  # new vertex array
    tr = [] # new triangle array
    m = [0] * len(s)    # maps original vertex indices to new indices

    r.append(s[0][0])

    for i in xrange(0, len(s)):
        if r[len(r) - 1] != s[i][0]:
            r.append(s[i][0])
        m[s[i][1]] = len(r) - 1

    for i in xrange(0, len(t)):
        tr.append(m[t[i]])

    return (r, tr)

def bone_index(bonelist, bonename):
    for i in xrange(len(bonelist)):
        if bonename == bonelist[i]:
            return i
    return 0

def vert_to_string(v, bones):
    s = "%f %f %f %f %f %f %f %f" % (v[0], v[1], v[2], v[3], v[4], v[5], v[6], v[7])

    for i in xrange(len(bones)):
        w = " 0"

        for b in v[8]:
            if bones[i] == b[0]:
                w = " %f" % (b[1])

        s += w

    s += "\n"

    return s

def add_face(face, m, key, mesh):
    if not key in m:
        m[key] = ([],[])

    verts, tris = m[key]

    vbase = len(verts)
    for i in xrange(len(face.v)):
        v = face.v[i]
        try:
            tc = (face.uv[i][0], face.uv[i][1])
        except:
            tc = (0.0, 0.0)
        verts.append((v.co[0], v.co[1], v.co[2], v.no[0], v.no[1], v.no[2]) + tc + (mesh.getVertexInfluences(v.index),))

    for i in range(3):
        tris.append(vbase + i)

    if len(face.v) == 4:
        # output other half of quad
        for i in (2, 3, 0):
            tris.append(vbase + i)

def add_object_by_material(obj, m):
    mesh = obj.getData(False, True)

    for face in mesh.faces:
        try:
            mat = mesh.materials[face.mat]
            mirror_alpha = mat.rayMirr
            mirror_rgb = mat.mirCol
            spec_alpha = mat.spec
            spec_rgb = mat.specCol
            spec_exp = mat.hard
        except:
            mirror_alpha = 0.0
            mirror_rgb = [0.0, 0.0, 0.0]
            spec_alpha = 0.0
            spec_rgb = [0.0, 0.0, 0.0]
            spec_exp = 0


        if mirror_alpha == 0.0:
            key = (spec_exp,) + (spec_alpha,) + tuple(spec_rgb)
            add_face(face, m, key, mesh)
        elif mirror_alpha == 1.0:
            key = (-1.0,) + (mirror_alpha,) + tuple(mirror_rgb)
            add_face(face, m, key, mesh)
        else:
            key = (spec_exp,) + (spec_alpha,) + tuple(spec_rgb)
            add_face(face, m, key, mesh)
            key = (-1.0,) + (mirror_alpha,) + tuple(mirror_rgb)
            add_face(face, m, key, mesh)

def findarmature(obj):
    mseq = obj.modifiers

    for i in xrange(mseq.__len__()):
        m = mseq.__getitem__(i)
        if m.type == Modifier.Types.ARMATURE:
            return m.__getitem__(Modifier.Settings.OBJECT)
    return None

def getbones(obj):
    armobj = findarmature(obj)

    if armobj == None:
        return []

    arm = armobj.getData()

    return arm.bones.keys()

def write(filename):
    file = open(filename, "wb")

    objects = Blender.Object.GetSelected()

    mesh_by_material = {}

    bones = []

    for obj in objects:
        add_object_by_material(obj, mesh_by_material)
        bones = bones + getbones(obj)


    keys = mesh_by_material.keys()

    keys.sort()
    keys.reverse()

    bones.sort()

    if len(bones) > 0:
        file.write("JM2\n")
        file.write("%d\n" % (len(bones)))
        file.write(",".join(bones))
        file.write("\n")
    else:
        file.write("JM1\n")

    for key in keys:

        verts, tris = mesh_by_material[key]

        verts, tris = reduce_verts(verts, tris)

        pad = ""
        for val in key:
            file.write("%s%f" % (pad, val))
            pad = " "

        file.write("\n%d %d\n" % (len(verts), len(tris) / 3))

        for v in verts:
            file.write(vert_to_string(v, bones))

        for t in tris:
            file.write("%d\n" % t)

    file.close()

def fs_callback(filename):
    if filename.find('.jm', -3) <= 0: filename += '.jm'
    write(filename)

Blender.Window.FileSelector(fs_callback, "Export jm", Blender.Object.GetSelected()[0].name + ".jm")