2014-08-16 00:50:36 -07:00
|
|
|
#!/usr/bin/env python3
|
|
|
|
import zlib
|
|
|
|
import struct
|
|
|
|
import sys
|
|
|
|
import time
|
2014-08-16 09:34:24 -07:00
|
|
|
import getopt
|
2014-08-16 11:50:08 -07:00
|
|
|
import re
|
2014-08-16 00:50:36 -07:00
|
|
|
|
|
|
|
# mt2obj - MTS schematic to OBJ converter
|
|
|
|
# Copyright (C) 2014 sfan5
|
|
|
|
#
|
|
|
|
# This program is free software; you can redistribute it and/or
|
|
|
|
# modify it under the terms of the GNU General Public License
|
|
|
|
# as published by the Free Software Foundation; either version 2
|
|
|
|
# of the License, or (at your option) any later version.
|
|
|
|
#
|
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
# GNU General Public License for more details.
|
2014-08-16 09:48:55 -07:00
|
|
|
#
|
2014-08-16 09:22:27 -07:00
|
|
|
# You should have received a copy of the GNU General Public License
|
|
|
|
# along with this program; if not, write to the Free Software
|
|
|
|
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
2014-08-16 00:50:36 -07:00
|
|
|
|
2014-08-16 11:50:08 -07:00
|
|
|
def convert(desc, vals):
|
|
|
|
out = tuple()
|
|
|
|
i = 0
|
|
|
|
for val in vals:
|
|
|
|
if val is None:
|
|
|
|
i += 1
|
|
|
|
continue
|
|
|
|
c = desc[i]
|
|
|
|
if c == 'x': # copy
|
|
|
|
out += val,
|
|
|
|
elif c == 's': # string
|
|
|
|
out += str(val),
|
|
|
|
elif c == 'i': # int
|
|
|
|
out += int(val),
|
|
|
|
elif c == 'f': # float
|
|
|
|
out += float(val),
|
|
|
|
elif c == 'h': # hexadecimal int
|
|
|
|
out += int(val, 16),
|
|
|
|
elif c == 'b': # bool
|
|
|
|
out += val.strip().lower() in ['1', 'true', 'yes'],
|
|
|
|
i += 1
|
|
|
|
return out
|
|
|
|
|
|
|
|
|
|
|
|
nodetbl = {}
|
|
|
|
|
|
|
|
r_entry = re.compile(r'^(\S+) (\S+) (\d+) (\d+) (\d+)(?: (\d+))?$')
|
|
|
|
|
|
|
|
f = open("nodes.txt", "r")
|
|
|
|
for l in f:
|
|
|
|
m = r_entry.match(l)
|
|
|
|
nodetbl[m.group(1)] = convert('siiii', m.groups()[1:])
|
|
|
|
f.close()
|
2014-08-16 00:50:36 -07:00
|
|
|
|
2014-08-16 09:34:24 -07:00
|
|
|
optargs, args = getopt.getopt(sys.argv[1:], '')
|
|
|
|
|
2014-08-16 09:48:55 -07:00
|
|
|
if len(args) < 1:
|
2014-08-16 09:34:24 -07:00
|
|
|
print("Usage: %s <.mts schematic>" % sys.argv[0])
|
|
|
|
print("Converts .mts schematics to Wavefront .obj geometry files")
|
2014-08-16 00:50:36 -07:00
|
|
|
print("")
|
2014-08-16 11:50:08 -07:00
|
|
|
print("Output files are written into directory of source file.")
|
2014-08-16 00:50:36 -07:00
|
|
|
exit(1)
|
|
|
|
else:
|
2014-08-16 09:48:55 -07:00
|
|
|
sch = open(args[0], "rb")
|
2014-08-16 00:50:36 -07:00
|
|
|
if sch.read(4) != b"MTSM":
|
|
|
|
print("This file does not look like an MTS schematic..")
|
|
|
|
exit(1)
|
2014-08-16 09:34:24 -07:00
|
|
|
v = struct.unpack("!H", sch.read(2))[0]
|
|
|
|
if v != 3:
|
|
|
|
print("Wrong file version: got %d, expected %d" % (v, 3))
|
2014-08-16 00:50:36 -07:00
|
|
|
exit(1)
|
|
|
|
width, height, depth = struct.unpack("!HHH", sch.read(6))
|
|
|
|
sch.seek(height, 1)
|
|
|
|
nodecount = struct.unpack("!H", sch.read(2))[0]
|
|
|
|
nodemap = {}
|
|
|
|
for i in range(nodecount):
|
|
|
|
l = struct.unpack("!H", sch.read(2))[0]
|
|
|
|
name = sch.read(l).decode('ascii')
|
|
|
|
nodemap[i] = name
|
|
|
|
# TODO use zlib.compressobj
|
|
|
|
cdata = sch.read()
|
|
|
|
sch.close()
|
|
|
|
data = zlib.decompress(cdata)
|
|
|
|
del cdata
|
2014-08-16 09:48:55 -07:00
|
|
|
filepart = args[0][:args[0].find(".")]
|
2014-08-16 00:50:36 -07:00
|
|
|
obj = open(filepart + ".obj", "w")
|
|
|
|
obj.write("# Exported by mt2obj\nmtllib %s\n\n\n" % (filepart + ".mtl", ))
|
|
|
|
i = 0
|
|
|
|
foundnodes = []
|
|
|
|
unknownnodes = []
|
|
|
|
for x in range(width):
|
|
|
|
for y in range(height):
|
|
|
|
for z in range(depth):
|
|
|
|
off = (x + y*width + z*width*height) * 2
|
|
|
|
nid = struct.unpack("!H", data[off:off + 2])[0]
|
|
|
|
nname = nodemap[nid]
|
|
|
|
if nname == "air":
|
|
|
|
continue
|
2014-08-16 11:50:08 -07:00
|
|
|
if not nname in nodetbl.keys():
|
2014-08-16 00:50:36 -07:00
|
|
|
if not nname in unknownnodes:
|
|
|
|
unknownnodes.append(nname)
|
|
|
|
continue
|
|
|
|
else:
|
|
|
|
if not nname in foundnodes:
|
|
|
|
foundnodes.append(nname)
|
|
|
|
obj.write("o node%d\n" % i)
|
|
|
|
obj.write("usemtl %s\n" % nname.replace(":", "__"))
|
2014-08-16 11:50:08 -07:00
|
|
|
objd = open("models/" + nodetbl[nname][0] + ".obj", 'r')
|
2014-08-16 11:19:11 -07:00
|
|
|
for line in objd:
|
|
|
|
if line.strip() == "":
|
|
|
|
pass
|
|
|
|
elif line.startswith("#"):
|
|
|
|
pass # comment
|
|
|
|
elif line.startswith("v "):
|
|
|
|
tmp = line.split(" ")
|
|
|
|
vx, vy, vz = float(tmp[1]), float(tmp[2]), float(tmp[3])
|
|
|
|
vx += x
|
|
|
|
vy += y
|
|
|
|
vz += z
|
|
|
|
obj.write("v %f %f %f\n" % (vx, vy, vz))
|
|
|
|
else:
|
|
|
|
obj.write(line)
|
|
|
|
objd.close()
|
|
|
|
obj.write("\n")
|
2014-08-16 00:50:36 -07:00
|
|
|
i += 1
|
|
|
|
obj.close()
|
|
|
|
mtl = open(filepart + ".mtl", "w")
|
|
|
|
mtl.write("# Generated by mt2obj\n\n\n")
|
|
|
|
for node in foundnodes:
|
|
|
|
mtl.write("newmtl %s\n" % node.replace(":", "__"))
|
2014-08-16 11:50:08 -07:00
|
|
|
c = nodetbl[node]
|
|
|
|
mtld = open("models/" + nodetbl[node][0] + ".mtl", 'r')
|
2014-08-16 11:19:11 -07:00
|
|
|
for line in mtld:
|
|
|
|
if line.strip() == "":
|
|
|
|
pass
|
|
|
|
elif line.startswith("#"):
|
|
|
|
pass # comment
|
|
|
|
else:
|
|
|
|
if len(c) > 4: # if there is transparency
|
|
|
|
tmp1 = c[4]/255
|
|
|
|
else:
|
|
|
|
tmp1 = 1.0
|
|
|
|
tmp2 = line.replace("{r}", str(c[1]/255)).replace("{g}", str(c[2]/255)).replace("{b}", str(c[3]/255)).replace("{a}", str(tmp1))
|
|
|
|
mtl.write(tmp2)
|
2014-08-16 00:50:36 -07:00
|
|
|
mtl.write("\n")
|
2014-08-16 11:19:11 -07:00
|
|
|
mtld.close()
|
2014-08-16 00:50:36 -07:00
|
|
|
mtl.close()
|
|
|
|
if len(unknownnodes) > 0:
|
|
|
|
print("There were some unknown nodes that were ignored during the conversion:")
|
|
|
|
for e in unknownnodes:
|
|
|
|
print(e)
|
|
|
|
|