#!/usr/bin/env python
# -*- Mode: Python; py-indent-offset: 8 -*-

# (C) Copyright Zack Rusin 2005
# All Rights Reserved.
# 
# Permission is hereby granted, free of charge, to any person obtaining a
# copy of this software and associated documentation files (the "Software"),
# to deal in the Software without restriction, including without limitation
# on the rights to use, copy, modify, merge, publish, distribute, sub
# license, and/or sell copies of the Software, and to permit persons to whom
# the Software is furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice (including the next
# paragraph) shall be included in all copies or substantial portions of the
# Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.  IN NO EVENT SHALL
# IBM AND/OR ITS SUPPLIERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
# IN THE SOFTWARE.
#
# Authors:
#    Zack Rusin <zack@kde.org>

import gl_XML, license
import sys, getopt

class PrintGenericStubs(gl_XML.gl_print_base):

	def __init__(self):
		gl_XML.gl_print_base.__init__(self)

		self.name = "gl_ppc_asm.py (from Mesa)"
		self.license = license.bsd_license_template % ( \
"""Copyright (C) 1999-2001  Brian Paul   All Rights Reserved.
(C) Copyright Zack Rusin 2005""", "BRIAN PAUL, ZACK RUSIN")
		return


	def get_stack_size(self, f):
		size = 0
		for p in f.parameterIterator():
			size += p.get_stack_size()

		return size


	def printRealHeader(self):
		print '#include "glapioffsets.h"'
		print ''
		print '#define CONCAT(x, y)          x ## y'
		print '#define PTHREADS'
		print ''
		print '#if defined(STDCALL_API)'
		print '# if defined(USE_MGL_NAMESPACE)'
		print '#  define GL_PREFIX(n,n2) CONCAT(mgl,n2)'
		print '# else'
		print '#  define GL_PREFIX(n,n2) CONCAT(gl,n2)'
		print '# endif'
		print '#else'
		print '# if defined(USE_MGL_NAMESPACE)'
		print '#  define GL_PREFIX(n,n2) CONCAT(mgl,n)'
		print '# else'
		print '#  define GL_PREFIX(n,n2) CONCAT(gl,n)'
		print '# endif'
		print '#endif'
		print ''
		print '#define BRANCH_OFFSET(offset) \t\t\\'
		print '\t\tlwz\t0,4*offset(29) ;\\'
		print '\t\tmtctr\t0 ;\t\t\\'
		print '\t\tbctr\t'
		print ''
		print '#define GLOBL_FN(x) .globl x ; .type x, function'
		print ''
		print '#if defined(PTHREADS) || defined(USE_XTHREADS) || defined(BEOS_THREADS)'
		print '#  define THREADS'
		print '#endif'
		print ''
		print '#ifdef GLX_USE_TLS'
		print ''
		print '#  define GL_STUB(fn,off,fn_alt)\t\t\t\\'
		print '\t.balign 16 ;\t\t\t\t\t\\'
		print '\tGLOBL_FN(GL_PREFIX(fn, fn_alt));\t\t\\'
		print '\tGL_PREFIX(fn, fn_alt):\t\t\t\t\\'
		print '\t\tbl\t_ppc_get_dispatch ;\t\t\\'
		print '\t\tBRANCH_OFFSET(off) ;'
		print ''
		print '#elif defined(PTHREADS)'
		print '#  define GL_STUB(fn,off,fn_alt)\t\t\t\\'
		print '\t.balign 16;\t\t\t\t\t\\'
		print '\tGLOBL_FN(GL_PREFIX(fn, fn_alt));\t\t\\'
		print '\tGL_PREFIX(fn, fn_alt):\t\t\t\t\\'
		print '\tlis 29,_glapi_Dispatch@ha ;\t\t\t\\'
		print '\tla 29,_glapi_Dispatch@l(29) ;\t\t\t\\'
		print '\tbe 1f ;\t\t\t\t\t\t\\'
		print '\t\tBRANCH_OFFSET(off) ;\t\t\t\\'
		print '\t1:\tbl\t_ppc_get_dispatch ;\t\t\\'
		print '\t\tBRANCH_OFFSET(off) ;'
		print '#elif defined(THREADS)'
		print '#  define GL_STUB(fn,off,fn_alt)\t\t\t\\'
		print '\t.balign 16;\t\t\t\t\t\\'
		print '\tGLOBL_FN(GL_PREFIX(fn, fn_alt));\t\t\\'
		print '\tGL_PREFIX(fn, fn_alt):\t\t\t\t\\'
		print '\tlis 29,_glapi_Dispatch@ha ;\t\t\t\\'
		print '\tla 29,_glapi_Dispatch@l(29) ;\t\t\t\\'
		print '\tbe 1f ;\t\t\t\t\t\t\\'
		print '\t\tBRANCH_OFFSET(off) ;\t\t\t\\'
		print '\t1:\tbl\t_ppc_get_dispatch ;\t\t\\'
		print '\t\tBRANCH_OFFSET(off) ;'
		print '#else /* Non-threaded version. */'
		print '#  define GL_STUB(fn,off,fn_alt)\t\t\t\\'
		print '\t.balign 16;\t\t\t\t\t\\'
		print '\tGLOBL_FN(GL_PREFIX(fn, fn_alt));\t\t\\'
		print '\tGL_PREFIX(fn, fn_alt):\t\t\t\t\\'
		print '\tlis 29,_glapi_Dispatch@ha ;\t\t\t\\'
		print '\tla 29,_glapi_Dispatch@l(29) ;\t\t\t\\'
		print '\t\tBRANCH_OFFSET(off) ;'
		print '#endif'
		print ''
		print '#ifdef HAVE_ALIAS'
		print '#  define GL_STUB_ALIAS(fn,off,fn_alt,alias,alias_alt)\t\t\\'
		print '\t.globl\tGL_PREFIX(fn, fn_alt) ;\t\t\t\t\\'
		print '\t.set\tGL_PREFIX(fn, fn_alt), GL_PREFIX(alias, alias_alt)'
		print '#else'
		print '#  define GL_STUB_ALIAS(fn,off,fn_alt,alias,alias_alt)\t\\'
		print '    GL_STUB(fn, off, fn_alt)'
		print '#endif'
		print ''
		print '.section ".text"'
		print ''
		print '#ifdef GLX_USE_TLS'
		print ''
		print '\t.globl\t_ppc_get_dispatch'
		print '\t.hidden\t_ppc_get_dispatch'
		print '\t.balign 16'
		print '_ppc_get_dispatch:'
		print '\taddis\t29,2,_glapi_tls_Dispatch@tprel@ha'
		print '\taddi\t29,9,_glapi_tls_Dispatch@tprel@l'
		print '\tblr'
		print ''
		print '#elif defined(PTHREADS)'
		print '.globl _glapi_Dispatch'
		print '.globl _gl_DispatchTSD'
		print '.globl pthread_getspecific'
		print ''
		print '.balign 16'
		print '_ppc_get_dispatch:'
		print '\tstwu\t1,-16(1)'
		print '\tmflr\t0'
		print '\tlis\t29,_gl_DispatchTSD@ha'
		print '\tlwz\t3,_gl_DispatchTSD@l(29)'
		print '\tstw\t0,20(1)'
		print '\tbl\tpthread_getspecific'
		print '\tlwz\t0,20(1)'
		print '\tmr\t29, 0'
		print '\tmtlr\t0'
		print '\taddi\t1,1,16'
		print '\tblr'
		print '#elif defined(THREADS)'
		print '.globl _glapi_get_dispatch'
		print '#endif'
		print ''

		print '#if defined( GLX_USE_TLS )'
		print '\t\t.section\twtext, "awx", @progbits'
		print '#endif /* defined( GLX_USE_TLS ) */'

		print ''
		print '\t.balign 16'
		print '\t.globl gl_dispatch_functions_start'
		print '\t.hidden\tgl_dispatch_functions_start'
		print 'gl_dispatch_functions_start:'
		print ''
		return


	def printRealFooter(self):
		print ''
		print '\t.globl\tgl_dispatch_functions_end'
		print '\t.hidden\tgl_dispatch_functions_end'
		print '\t.balign 16'
		print 'gl_dispatch_functions_end:'
		print ''
		print '#if defined(GLX_USE_TLS) && defined(__linux__)'
		print '	.section ".note.ABI-tag", "a"'
		print '	.p2align 2'
		print '	.long	1f - 0f   /* name length */'
		print '	.long	3f - 2f   /* data length */'
		print '	.long	1         /* note length */'
		print '0:	.asciz "GNU"      /* vendor name */'
		print '1:	.p2align 2'
		print '2:	.long	0         /* note data: the ABI tag */'
		print '	.long	2,4,20    /* Minimum kernel version w/TLS */'
		print '3:	.p2align 2        /* pad out section */'
		print '#endif /* GLX_USE_TLS */'
		return


	def printBody(self, api):
		for f in api.functionIterateByOffset():
			stack = self.get_stack_size(f)

			alt = "%s@%u" % (f.name, stack)
			print '\tGL_STUB(%s, _gloffset_%s, %s)' % (f.name, f.name, alt)

		for f in api.functionIterateByOffset():
			stack = self.get_stack_size(f)

			alt = "%s@%u" % (f.name, stack)

			for n in f.entry_points:
				if n != f.name:
					alt2 = "%s@%u" % (n, stack)
					print '\tGL_STUB_ALIAS(%s, _gloffset_%s, %s, %s, %s)' % (n, f.name, alt2, f.name, alt)

		return

def show_usage():
	print "Usage: %s [-f input_file_name] [-m output_mode]" % sys.argv[0]
	sys.exit(1)

if __name__ == '__main__':
	file_name = "gl_API.xml"
	mode = "generic"

	try:
		(args, trail) = getopt.getopt(sys.argv[1:], "m:f:")
	except Exception,e:
		show_usage()

	for (arg,val) in args:
		if arg == '-m':
			mode = val
		elif arg == "-f":
			file_name = val

	if mode == "generic":
		printer = PrintGenericStubs()
	else:
		print "ERROR: Invalid mode \"%s\" specified." % mode
		show_usage()

	api = gl_XML.parse_GL_API( file_name )

	printer.Print( api )