from ctypes import *
import sys
import os
import xml.etree.ElementTree as ET

binaries = '../../../binaries'

dll_filename = {
	'posix': './libCollada_dbg.so',
	'nt': 'Collada_dbg.dll',
}[os.name]

library = cdll.LoadLibrary('%s/system/%s' % (binaries, dll_filename))

def log(severity, message):
	print '[%s] %s' % (('INFO', 'WARNING', 'ERROR')[severity], message)

clog = CFUNCTYPE(None, c_int, c_char_p)(log)
	# (the CFUNCTYPE must not be GC'd, so try to keep a reference)
library.set_logger(clog)

def convert_dae_to_pmd(filename):
	output = []
	def cb(cbdata, str, len):
		output.append(string_at(str, len))

	cbtype = CFUNCTYPE(None, POINTER(None), POINTER(c_char), c_uint)
	status = library.convert_dae_to_pmd(filename, cbtype(cb), None)
	assert(status == 0)
	return ''.join(output)

def clean_dir(path):
	# Remove all files first
	try:
		for f in os.listdir(path):
			os.remove(path+'/'+f)
		os.rmdir(path)
	except OSError:
		pass # (ignore errors if files are in use)
	# Make sure the directory exists
	try:
		os.makedirs(path)
	except OSError:
		pass # (ignore errors if it already exists)

def create_actor(mesh, texture, idleanim, corpseanim, gatheranim):
	actor = ET.Element('actor', version='1')
	ET.SubElement(actor, 'castshadow')
	group = ET.SubElement(actor, 'group')
	variant = ET.SubElement(group, 'variant', frequency='100', name='Base')
	ET.SubElement(variant, 'mesh').text = mesh+'.pmd'
	ET.SubElement(variant, 'texture').text = texture+'.dds'
	animations = ET.SubElement(variant, 'animations')
	ET.SubElement(animations, 'animation', file=idleanim+'.psa', name='Idle', speed='100')
	ET.SubElement(animations, 'animation', file=corpseanim+'.psa', name='Corpse', speed='100')
	ET.SubElement(animations, 'animation', file=gatheranim+'.psa', name='Build', speed='100')
	return ET.tostring(actor)
	
################################

test_data = binaries + '/data/tests/collada'
test_mod = binaries + '/data/mods/_test.collada'

clean_dir(test_mod + '/art/meshes')
clean_dir(test_mod + '/art/actors')

#for test_file in ['cube', 'jav2', 'teapot_basic', 'teapot_skin', 'plane_skin', 'dude_skin']:
for test_file in ['sphere']:
	input_filename = '%s/%s.dae' % (test_data, test_file)
	output_filename = '%s/art/meshes/%s.pmd' % (test_mod, test_file)
	
	input = open(input_filename).read()
	output = convert_dae_to_pmd(input)
	open(output_filename, 'wb').write(output)

	xml = create_actor(test_file, 'male', 'dudeidle', 'dudecorpse', 'dudechop')
	open('%s/art/actors/%s.xml' % (test_mod, test_file), 'w').write(xml)