#!BPY
"""
Name: 'xfig export (.fig)'
Blender: 244
Group: 'Export'
Tooltip: 'Export selected mesh to xfig Format (.fig)'
"""

__author__ = "Dino Ghilardi  "
__url__ = ("blender", "elysiun")
__version__ = "1.1"

__bpydoc__ = """\
		This script exports the selected mesh to xfig (www.xfig.org) file format (i.e.: .fig)

		The starting point of this script was Anthony D'Agostino's raw triangle format export.
		(some code is still here and there, cut'n pasted from his script)

		Usage:<br>
			Select the mesh to be exported and run this script from "File->Export" menu.
			The toggle button 'export 3 files' enables the generation of 4 files: one global
			and three with the three different views of the object.
			This script is licensed under the GPL license. (c) Dino Ghilardi, 2005
			
"""

# .fig export, mostly brutally cut-n pasted from the 
# 'Raw triangle export' (Anthony D'Agostino, http://www.redrival.com/scorpius)|

import Blender
from Blender import Draw, BGL
from Blender.Window import DrawProgressBar
#, meshtools
import sys
#import time

# =================================
# === Write xfig Format.===
# =================================

#globals definition and init
mystring = ''
mymsg = ''
toggle=0
sel3files=0
maxX=-1000000000
maxY=-1000000000
maxZ=-1000000000
minX=minY=minZ=10000000000
boolmode=0 #0= export in inches, 1= export in cm
dimscale=float(1200) #scale due to the cm/inches select. default: inches
hidden_flag=0

space = float(2) #space between figures, in blender units.
scale= float(1200) #conversion scale to xfig units.
guiscale=float(1) #scale shown on the ruler in the GUI
#return values from gui items, just to deallocate them on exit.
guiret1=guiret2=guiret3=guiret4=guiret5=guiret6=guiret7=0

ScalePopup=0
DistancePopup=0
SpacePopup=0
#end of globals definition


def getmaxmin(ob, mesh):
	"""Gets the max-min coordinates of the mesh"""
	
	global maxX,maxY,maxZ,minX,minY,minZ
	"""Getting the extremes of the mesh to be exported"""
	
	maxX=maxY=maxZ = -1000000000
	minX=minY=minZ =  1000000000
	
	for face in mesh.faces:
		for v in face:
			x,y,z = v.co
			maxX = max(maxX, x)
			maxY = max(maxY, y)
			maxZ = max(maxZ, z)
			minX = min(minX, x)
			minY = min(minY, y)
			minZ = min(minZ, z)

def xfigheader():
	global export_type
	print '#FIG 3.2  Produced by xfig version 3.2.5-alpha5'
	print 'Landscape'
	print 'Center'
	if boolmode==0:
		print 'Inches'
	else:
		print 'Metric'
	#print export_type
	print 'Letter'
	print '100.00'
	print 'Single'
	print '-2'
	print '1200 2'

def xytransform(face):
	"""gives the face vertexes coordinates in the xfig format/translation (view xy)"""
	v4=None
	x4=y4=z4=None
	if len(face)==3:
		v1,v2,v3=face.v
	else:
		v1,v2,v3,v4=face.v	
	
	x1,y1,z1 = v1.co
	x2,y2,z2 = v2.co
	x3,y3,z3 = v3.co
	y1=-y1
	y2=-y2
	y3=-y3
	if v4:
		x4,y4,z4 = v4.co
		y4=-y4
	return x1,y1,z1,x2,y2,z2,x3,y3,z3,x4,y4,z4 

def xztransform(face):
	"""gives the face vertexes coordinates in the xfig format/translation (view xz)"""
	v4=None
	x4=y4=z4=None
	if len(face)==3:
		v1,v2,v3=face.v
	else:
		v1,v2,v3,v4=face.v	

	#Order vertexes
	x1,y1,z1 = v1.co
	x2,y2,z2 = v2.co
	x3,y3,z3 = v3.co
	y1=-y1
	y2=-y2
	y3=-y3
	
	z1=-z1+maxZ-minY +space
	z2=-z2+maxZ-minY +space
	z3=-z3+maxZ-minY +space

	if v4:
		x4,y4,z4 = v4.co
		y4=-y4
		z4=-z4+maxZ-minY +space 
	return x1,y1,z1,x2,y2,z2,x3,y3,z3,x4,y4,z4 

def yztransform(face):
	"""gives the face vertexes coordinates in the xfig format/translation (view xz)"""
	v4=None
	x4=y4=z4=None
	if len(face)==3:
		v1,v2,v3=face.v
	else:
		v1,v2,v3,v4=face.v	

	#Order vertexes
	x1,y1,z1 = v1.co
	x2,y2,z2 = v2.co
	x3,y3,z3 = v3.co
	y1=-y1; y2=-y2; y3=-y3
	z1=-(z1-maxZ-maxX-space)
	z2=-(z2-maxZ-maxX-space)
	z3=-(z3-maxZ-maxX-space)
	
	if v4:
		x4,y4,z4 = v4.co
		y4=-y4
		z4=-(z4-maxZ-maxX-space)
	return x1,y1,z1,x2,y2,z2,x3,y3,z3,x4,y4,z4 

def figdata(ob, expview):
	"""Prints all the xfig data (no header)"""
	mesh = ob.getData(mesh=1)
	facenumber = len(mesh.faces)
	for face in mesh.faces:
			if len(face) == 3:		# triangle
				print '2 3 0 1 0 7 50 -1 -1 0.000 0 0 -1 0 0 4'
				if expview=='xy':
					x1,y1,z1,x2,y2,z2,x3,y3,z3,x4,y4,z4=xytransform(face)
					faceverts = int(x1*scale),int(y1*scale),int(x2*scale),int(y2*scale),int(x3*scale),int(y3*scale), int(x1*scale),int(y1*scale)
				elif expview=='xz':
					x1,y1,z1,x2,y2,z2,x3,y3,z3,x4,y4,z4=xztransform(face)
					faceverts = int(x1*scale),int(z1*scale),int(x2*scale),int(z2*scale),int(x3*scale),int(z3*scale), int(x1*scale),int(z1*scale)
				elif expview=='yz':
					x1,y1,z1,x2,y2,z2,x3,y3,z3,x4,y4,z4=yztransform(face)
					faceverts = int(z1*scale),int(y1*scale),int(z2*scale),int(y2*scale),int(z3*scale),int(y3*scale),int(z1*scale),int(y1*scale)
				print '\t% i % i % i % i % i % i % i % i' % faceverts
			else: # Quad
				print '2 3 0 1 0 7 50 -1 -1 0.000 0 0 -1 0 0 5'
				if expview=='xy':
					x1,y1,z1,x2,y2,z2,x3,y3,z3,x4,y4,z4=xytransform(face)
					faceverts = int(x1*scale),int(y1*scale),int(x2*scale),int(y2*scale),int(x3*scale),int(y3*scale),int(x4*scale),int(y4*scale), int(x1*scale),int(y1*scale)

				elif expview=='xz':
					x1,y1,z1,x2,y2,z2,x3,y3,z3,x4,y4,z4=xztransform(face)
					faceverts = int(x1*scale),int(z1*scale),int(x2*scale),int(z2*scale),int(x3*scale),int(z3*scale),int(x4*scale),int(z4*scale), int(x1*scale),int(z1*scale)

				elif expview=='yz':
					x1,y1,z1,x2,y2,z2,x3,y3,z3,x4,y4,z4=yztransform(face)
					faceverts = int(z1*scale),int(y1*scale),int(z2*scale),int(y2*scale),int(z3*scale),int(y3*scale),int(z4*scale),int(y4*scale), int(z1*scale),int(y1*scale)
				print '\t% i % i % i % i % i % i % i % i % i % i' % faceverts



def writexy(ob, filename):
	"""writes the x-y view file exported"""
	global maxX, maxY, maxZ
	global minX, minY, minZ
	global space
	global scale
	#start = time.clock()
	file = open(filename, 'wb')
	std=sys.stdout
	sys.stdout=file
	xfigheader()
	figdata(ob, 'xy')# xydata()
	sys.stdout=std
	Blender.Window.DrawProgressBar(1.0, '')  # clear progressbar
	file.close()
	#end = time.clock()
	#seconds = " in %.2f %s" % (end-start, "seconds")
	print 'Successfully exported ', Blender.sys.basename(filename)# + seconds

def writexz(filename):
	"""writes the x-z view file exported"""
	global space,maxX,maxY,maxZ, scale
	#start = time.clock()
	file = open(filename, 'wb')
	std=sys.stdout
	sys.stdout=file
	xfigheader()
	figdata(ob, 'xz')#xzdata()
	sys.stdout=std
	Blender.Window.DrawProgressBar(1.0, '')  # clear progressbar
	file.close()
	#end = time.clock()
	#seconds = " in %.2f %s" % (end-start, "seconds")
	print 'Successfully exported ', Blender.sys.basename(filename)# + seconds

def writeyz(filename):
	"""writes the y-z view file exported"""
	global maxX, maxY, maxZ, minX, minY, minZ,scale
	#start = time.clock()
	file = open(filename, 'wb')

	std=sys.stdout
	sys.stdout=file

	xfigheader()
	figdata(ob, 'yz')#yzdata()	
	sys.stdout=std
	Blender.Window.DrawProgressBar(1.0, '')  # clear progressbar
	file.close()
	#end = time.clock()
	#seconds = " in %.2f %s" % (end-start, "seconds")
	print 'Successfully exported ', Blender.sys.basename(filename)# + seconds

def writeall(ob, filename):
	"""writes all 3 views
	
	Every view is a combined object in the resulting xfig. file."""
	global maxX, maxY, maxZ, minX, minY, minZ,scale
	#start = time.clock()
	file = open(filename, 'wb')

	std=sys.stdout
	sys.stdout=file

	xfigheader()
	print '#upper view (7)'
	print '6 % i % i % i % i ',  minX, minY, maxX, maxY
	figdata(ob, 'xy') #xydata()
	print '-6'
	print '#bottom view (1)'
	print '6 %i %i %i %i', minX, -minZ+maxZ-minY +space, maxX,-maxZ+maxZ-minY +space
	figdata(ob, 'xz') #xzdata()
	print '-6'
	
	print '#right view (3)'
	print '6 %i %i %i %i', minX, minZ-maxZ-maxX-space, maxX,maxZ-maxZ-maxX-space
	figdata(ob, 'yz') #yzdata()	
	print '-6'
	
	sys.stdout=std
	Blender.Window.DrawProgressBar(1.0, '')  # clear progressbar
	file.close()
	#end = time.clock()
	#seconds = " in %.2f %s" % (end-start, "seconds")
	print 'Successfully exported ', Blender.sys.basename(filename)# + seconds


#********************************************************USER INTERFACE*****************************************************
#********************************************************USER INTERFACE*****************************************************
#********************************************************USER INTERFACE*****************************************************
def gui():              # the function to draw the screen
	global mystring, mymsg, toggle, sel3files, scale 
	global guiret1, guiret2, guiret3, guiret4, guiret5, guiret6, guiret7
	global ScalePopup, SpacePopup, boolmode, guiscale,hidden_flag
	if len(mystring) > 90: mystring = ''
	# BGL.glClearColor(0,0,1,1)
	# BGL.glClear(BGL.GL_COLOR_BUFFER_BIT)
	BGL.glColor3f(1,1,1)
	guiret2=Draw.PushButton('Cancel', 2, 10, 10, 55, 20,'Cancel')
	guiret3=Draw.Toggle('1 file per view',    3, 10, 40, 110,20, sel3files, 'Export a file for each view')
	guiret4=Draw.PushButton('Export', 4, 70, 10, 70, 20, 'Select filename and export') 

	ScalePopup=Draw.Number('Scale', 5, 10,70, 110,20, guiscale, 0.0001, 1000.1, 'Scaling factor')
	SpacePopup=Draw.Number('Space', 6, 10,90, 110,20, space, 0, 10000, 'Space between projections')

	guiret5=Draw.Toggle('cm', 7, 120,70, 40,20, boolmode, 'set scale to 1 blender unit = 1 cm in xfig')
	guiret6=Draw.Toggle('in', 8, 162,70, 40,20, not boolmode, 'set scale to 1 blender unit = 1 in in xfig')

	BGL.glRasterPos2i(72, 16)
	if toggle: toggle_state = 'down'
	else: toggle_state = 'up'
	#Draw.Text('The toggle button is %s.' % toggle_state, 'small')
	BGL.glRasterPos2i(10, 230)
	#Draw.Text('Type letters from a to z, ESC to leave.')
	BGL.glRasterPos2i(20, 200)
	Draw.Text(mystring)
	BGL.glColor3f(1,0.4,0.3)
	BGL.glRasterPos2i(340, 70)
	Draw.Text(mymsg, 'tiny')


def event(evt, val):    # the function to handle input events
	Draw.Redraw(1)

import bpy

def button_event(evt):  # the function to handle Draw Button events
	global toggle, guiret5,scale, space, SpacePopup, boolmode, dimscale, guiscale
	global hidden_flag, sel3files
	if evt==1:
		toggle = 1 - toggle
		Draw.Redraw(1)
	if evt==2:
		Draw.Exit()
		return
	if evt==3:
		sel3files = 1-sel3files
		Draw.Redraw(1)	
	if evt==4:
		try:	ob = bpy.data.scenes.active.objects.active
		except:	ob = None
		if not ob or ob.type != 'Mesh':
			BPyMessages.Error_NoMeshActive()
			return
		
		Blender.Window.FileSelector(fs_callback, 'Export fig', Blender.sys.makename(ext='.fig'))
		Draw.Exit()
		return 
	if evt==5:
		guiscale = ScalePopup.val
		scale=dimscale*guiscale
		Draw.Redraw(1)
	if evt==6:
		space =SpacePopup.val
	if evt==7: 
		boolmode=1
		dimscale=450 #converting to cm
		scale = dimscale*guiscale
		Draw.Redraw(1)
	if evt==8:
		boolmode=0
		dimscale = 1200
		scale = dimscale*guiscale
		Draw.Redraw(1)
	if evt==9:
		hidden_flag=1-hidden_flag
		Draw.Redraw(1)

Draw.Register(gui, event, button_event)  # registering the 3 callbacks
	
import BPyMessages
def fs_callback(filename):
	if filename.lower().endswith('.fig'): filename = filename[:-4]
	
	try:	ob = bpy.data.scenes.active.objects.active
	except:	ob = None
	if not ob or ob.type != 'Mesh':
		BPyMessages.Error_NoMeshActive()
		return
	
	mesh = ob.getData(mesh=1)
	getmaxmin(ob, mesh)
	
	if sel3files:
		
		writexy(ob, filename + '_XY.fig')
		writexz(ob, filename + '_XZ.fig')
		writeyz(ob, filename + '_YZ.fig')
	writeall(ob, filename + '.fig')
	print  scale
	Draw.Exit()