#!BPY

"""
Name: 'FCollada (Curves only) (.dae)...'
Blender: 243
Group: 'Export'
Tooltip: 'Save Selected Curves to COLLADA .dae file'
"""

__author__ = "Matt Ebb / ProMotion Studios"
__url__ = ['mke3.net']
__version__ = "1.1"

__bpydoc__ = """\
This script exports 3D bezier curves to the COLLADA file format.

It is designed for compatibility with the open source FCollada library by 
Feeling Software, which their ColladaMax and ColladaMaya I/O plugins
are based on. FCollada's method of interpreting bezier handles and 
segment interpolation seems to differ from what's in the COLLADA spec,
so I don't know how this will work when imported into other applications.

See http://www.feelingsoftware.com for more information on this library.

Usage:

Select the objects you wish to export and run this script from "File->Export" menu.
Due to limitations in the COLLADA file format, all individual curves (even inside 
a singlecurve datablock) will be exported as separate objects.
"""

import Blender, BPySys
from Blender import *


def save_curve_fcollada(filename):

	scn = Scene.GetCurrent()
		
	file = open(filename, "w")
	
	header = '''<?xml version="1.0" encoding="utf-8"?>
<COLLADA version="1.4.1" xmlns="http://www.collada.org/2005/COLLADASchema">
\t<asset>
\t\t<contributor>
\t\t\t<authoring_tool>Blender 2.43 FCollada Curve Exporter</authoring_tool>
\t\t</contributor>
\t\t<up_axis>Z_UP</up_axis>
\t\t<unit meter="0.01" name="centimeter"/>
\t</asset>
'''
	file.write(header)
	
	file.write('\t<library_geometries>\n')
	
	geomlist = []
	
	for ob in scn.objects.selected:

		if ob.type != 'Curve': continue

		curvedata = ob.data
		curvedataname = BPySys.cleanName(curvedata.name)

		for cni, curnurb in enumerate(curvedata):
			cname = "%s-%d" % (curvedataname, cni)

			geominfo = {}
			
			if len(curnurb) > 1: geominfo['obname'] = BPySys.cleanName(ob.name) + '-' + cname
			else: geominfo['obname'] = BPySys.cleanName(ob.name)
			
			# Store geometry info for each sub-curve, since they will be split
			# into individual geometry nodes
			geominfo['dataname'] = cname
			geominfo['translate'] = '%.6f %.6f %.6f' % (ob.LocX, ob.LocY, ob.LocZ)
			geominfo['rotateZ'] = '%.6f' % ob.RotZ
			geominfo['rotateY'] = '%.6f' % ob.RotY
			geominfo['rotateX'] = '%.6f' % ob.RotX
			geominfo['scale'] = '%.6f %.6f %.6f' % (ob.SizeX, ob.SizeY, ob.SizeZ)
			
			geomlist.append(geominfo)
			
			file.write('\t\t<geometry id="%s-spline">\n' % cname)
			file.write('\t\t\t<spline closed="%d">\n' % curnurb.isCyclic())
			file.write('\t\t\t\t<source id="%s-cvs">\n' % cname)
			
			
			# This way below of representing curve CVs and tangents seems to conflict with the actual spec, 
			# but it's the way that the Fcollada/colladaMax importer seems to expect the data, so 
			# practicality wins over correctness here.
			
			if curnurb.isCyclic(): numpoints = 3*len(curnurb)	
			else: numpoints = 3*(len(curnurb))-2					# minus start h2 and end h2
			file.write('\t\t\t\t\t<float_array id="%s-cvs-array" count="%d">\n' % (cname, numpoints*3))		# *3 --> X,Y,Z
			
			cvs = ''
			
			for pni, point in enumerate(curnurb):
				cvs += '						'
				
				# only include h1 for first point of closed (cyclic) curves, ignore for open curves
				if pni or curnurb.isCyclic():
					cvs += '%.6f %.6f %.6f ' % (point.vec[0][0], point.vec[0][1], point.vec[0][2])
				
				cvs += '%.6f %.6f %.6f ' % (point.vec[1][0], point.vec[1][1], point.vec[1][2])
				
				# only include h2 for last point of closed (cyclic) curves, ignore for open curves
				if pni!=len(curnurb)-1 or curnurb.isCyclic():
					cvs += '%.6f %.6f %.6f ' % (point.vec[2][0], point.vec[2][1], point.vec[2][2])
				
				cvs += '\n'

			file.write(cvs)
			file.write('\t\t\t\t\t</float_array>\n')
			file.write('\t\t\t\t\t<technique_common>\n')
			
			file.write('\t\t\t\t\t\t<accessor source="#%s-cvs-array" count="%d" stride="3">\n' % (cname, numpoints))

			paramxyz = '''\t\t\t\t\t\t\t<param name="X" type="float"/>
\t\t\t\t\t\t\t<param name="Y" type="float"/>
\t\t\t\t\t\t\t<param name="Z" type="float"/>
'''
			file.write(paramxyz)
			
			file.write('\t\t\t\t\t\t</accessor>\n')
			file.write('\t\t\t\t\t</technique_common>\n')
			file.write('\t\t\t\t</source>\n')
			
			file.write('\t\t\t\t<control_vertices>\n')
			file.write('\t\t\t\t\t<input semantic="POSITION" source="#%s-cvs"/>\n' % cname)
			file.write('\t\t\t\t</control_vertices>\n')

			extra = '''\t\t\t<extra>
\t\t\t\t<technique profile="FCOLLADA">
\t\t\t\t\t<type>BEZIER</type>
\t\t\t\t</technique>
\t\t\t</extra>
'''
			file.write(extra)

			file.write('\t\t\t</spline>\n')
			file.write('\t\t</geometry>\n')
			
	file.write('\t</library_geometries>\n')
	
	file.write('\t<library_visual_scenes>\n')
	
	scnname = BPySys.cleanName(scn.name)
	file.write('\t\t<visual_scene id="%s" name="%s">\n' % (scnname, scnname))

	# Separate node for each connected curve segment.
	# COLLADA doesn't support multiple curve per node.
	for g in geomlist:
		file.write('\t\t\t<node id="%s" name="%s" type="NODE">\n' % (g['obname'], g['obname']))
		file.write('\t\t\t\t<translate>%s</translate>\n' % g['translate'])
		file.write('\t\t\t\t<rotate sid="rotateZ">0 0 1 %s</rotate>\n' % g['rotateZ'])
		file.write('\t\t\t\t<rotate sid="rotateY">0 1 0 %s</rotate>\n' % g['rotateY'])
		file.write('\t\t\t\t<rotate sid="rotateX">1 0 0 %s</rotate>\n' % g['rotateX'])
		file.write('\t\t\t\t<scale>%s</scale>\n' % g['scale'])
		file.write('\t\t\t\t<instance_geometry url="#%s-spline"/>\n' % g['dataname'])
		file.write('\t\t\t</node>\n')
	
	file.write('\t\t</visual_scene>\n')
	file.write('\t</library_visual_scenes>\n')
	
	file.write('\t<scene>\n')
	file.write('\t\t<instance_visual_scene url="#%s"/>\n' % scnname)
	file.write('\t</scene>\n')
	
	file.write('</COLLADA>\n')
	
	file.close()


Window.FileSelector(save_curve_fcollada, 'Export Curve to FCollada')