#!BPY
 
"""
Name: 'Warzone model (.pie)...'
Blender: 244
Group: 'Export'
Tooltip: 'Save a Warzone model file'
"""

__author__ = "Gerard Krol, Kevin Gillette"
__version__ = "0.2"
__bpydoc__ = """\
This script exports Warzone 2100 PIE files.
"""

#
# --------------------------------------------------------------------------
# PIE Export v0.1 by Gerard Krol (gerard_)
#            v0.2 by Kevin Gillette (kage)
# --------------------------------------------------------------------------
# ***** BEGIN GPL LICENSE BLOCK *****
#
# 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.
#
# 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
#
# ***** END GPL LICENCE BLOCK *****
# --------------------------------------------------------------------------

import Blender
import os
import math

def fs_callback(filepath):
		# TODO: export all meshes instead of only the selected one
		ob = Blender.Object.GetSelected()
		mesh = ob[0].getData(mesh=1)
		
		out = file(filepath, 'w')
		out.write( "PIE %i\n" % pie_version )
		out.write( 'TYPE 200\n' )
		#print mesh.faces[0].image.getFilename()
		texturename = os.path.basename(mesh.faces[0].image.getFilename())
		print 'texture:', texturename
		out.write( 'TEXTURE 0 ' + texturename + ' 256 256\n')
		out.write( 'LEVELS 1\n')
		out.write( 'LEVEL 1\n')
		out.write('POINTS ' + str(len(mesh.verts)) + '\n')

		if precision > 0:
			format_str = "\t" + (" %%.%if" % precision * 3)[1:] + "\n"
		else:
			format_str = '\t%i %i %i\n'

		for vert in mesh.verts:
			out.write( format_str % (vert.co.x*128, vert.co.z*128, -vert.co.y*128) )
		
		out.write('POLYGONS ' + str(len(mesh.faces)) + '\n')
		uvLayers = mesh.getUVLayerNames()
		for face in mesh.faces:
			flags = 0x200
			if len(uvLayers) > 1:
				# team colored
				mesh.activeUVLayer = uvLayers[0]
				orgUVCoords = []
				for n in xrange(0, len(face.v)):
					orgUVCoords.append((face.uv[n][0], 1.0-face.uv[n][1]) )
				mesh.activeUVLayer = uvLayers[1]
				if face.mode & Blender.Mesh.FaceModes['TEX']:
					uDiff = 0.0
					vDiff = 0.0
					for n in xrange(0, len(face.v)):
						uDiff += face.uv[n][0] - orgUVCoords[n][0]
						vDiff += 1.0-face.uv[n][1] - orgUVCoords[n][1]
					uDiff /= len(face.v)
					vDiff /= len(face.v)
					print uDiff, vDiff
					if abs(uDiff) > abs(vDiff):
						width = uDiff*256/7
						height = 256
					else:
						width = 256
						height = vDiff*256/7
					flags |= 0x4000
			if face.mode & Blender.Mesh.FaceModes['TWOSIDE']:
				flags |= 0x2000
			if face.transp == Blender.Mesh.FaceTranspModes['ALPHA']:
				flags |= 0x800
				
			mesh.activeUVLayer = uvLayers[0]
			out.write('\t' + hex(flags).split('x')[-1] + ' ' + str(len(face.v)))
			
			for vert in face.v:
					out.write( ' %i' % vert.index )
			if flags & 0x4000:
				# team colors
				out.write( ' 8 1 %i %i' % (width, height) )
			for n, vert in enumerate(face.v):
					#print face.uv[n]
					out.write( ' %i %i' % (face.uv[n][0]*256, (1.0-face.uv[n][1])*256) )
			out.write('\n')
		# export the connectors
		connectors = []
		for object in Blender.Scene.GetCurrent().getChildren():
			if object.getData() == None:
				x,y,z = object.getLocation()
				connectors.append((object.getName(), (x*128, -y*128 ,z*128)))
		connectors.sort()
		if connectors:
			out.write('CONNECTORS %i\n' % len(connectors))
			for c, xyz in connectors:
				out.write('\t%i %i %i\n' % xyz)
		out.close()
fname = Blender.sys.makename(ext=".pie")

precision, pie_version = Blender.Draw.Create(0), 2

def opts_draw():
	global precision, pie_version
	if precision.val > 0: pie_version = 5
	else: pie_version = 2
	Blender.Draw.Label("Version: PIE %i" % pie_version, 10, 80, 130, 30)
	precision = Blender.Draw.Number("precision", 2, 10, 50, 130, 30, precision.val, 0, 5)
	Blender.Draw.Button("Export", 0, 10, 10, 60, 30)
	Blender.Draw.Button("Cancel", 1, 80, 10, 60, 30)

def opts_evt(val):
	if 0 == val:
		global precision
		precision = precision.val
		Blender.Draw.Exit()
		print "starting export at floating-point precision of", precision
		Blender.Window.FileSelector(fs_callback, "Export Warzone Model", fname)
	elif 1 == val:
		Blender.Draw.Exit()
		print "export aborted"
	else:
		Blender.Draw.Redraw()

Blender.Draw.Register(opts_draw, None, opts_evt)