#!/usr/bin/env python
#
# Copyright (c) 2011 Martin Decky
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
#
# - Redistributions of source code must retain the above copyright
#   notice, this list of conditions and the following disclaimer.
# - Redistributions in binary form must reproduce the above copyright
#   notice, this list of conditions and the following disclaimer in the
#   documentation and/or other materials provided with the distribution.
# - The name of the author may not be used to endorse or promote products
#   derived from this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
# OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
# IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
# NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
# THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#
"""
C structure creator
"""

import sys
import os
import struct

def usage(prname):
	"Print usage syntax"
	print("%s <DESTINATION> <LABEL> <AS_PROLOG> [SOURCE ...]" % prname)

def main():
	if (len(sys.argv) < 4):
		usage(sys.argv[0])
		return
	
	dest = sys.argv[1]
	label = sys.argv[2]
	as_prolog = sys.argv[3]
	
	header_ctx = []
	data_ctx = []
	
	for src in sys.argv[4:]:
		basename = os.path.basename(src)
		symbol = basename.replace(".", "_")
		
		print("%s -> %s" % (src, symbol))
		
		src_in = open(src, "rb")
		src_data = src_in.read()
		src_in.close()
		
		data_rec = "\t.byte "
		
		fmt = 'B'
		item_size = struct.calcsize(fmt)
		offset = 0
		cnt = 0
		
		while (len(src_data[offset:]) >= item_size):
			byte = struct.unpack_from(fmt, src_data, offset)
			
			if (offset > 0):
				if ((cnt % 15) == 0):
					data_rec += "\n\t.byte "
				else:
					data_rec += ", "
			
			data_rec += "0x%x" % byte
			offset += item_size
			cnt += 1
		
		data_rec += "\n"
		
		header_ctx.append("extern uint64_t %s_size;" % symbol)
		header_ctx.append("extern uint8_t %s[];" % symbol)
		
		data_ctx.append(".globl %s_size" % symbol)
		data_ctx.append(".balign 8");
		data_ctx.append(".size %s_size, 8" % symbol)
		data_ctx.append("%s_size:" % symbol)
		data_ctx.append("\t.quad %u\n" % offset)
		
		data_ctx.append(".globl %s" % symbol)
		data_ctx.append(".balign 8")
		data_ctx.append(".size %s, %u" % (symbol, offset))
		data_ctx.append("%s:" % symbol)
		data_ctx.append(data_rec)
	
	header = open("%s.h" % dest, "w")
	
	header.write('/***************************************\n')
	header.write(' * AUTO-GENERATED FILE, DO NOT EDIT!!! *\n')
	header.write(' * Generated by: tools/mkarray.py      *\n')
	header.write(' ***************************************/\n\n')
	header.write("#ifndef %s_H_\n" % label)
	header.write("#define %s_H_\n\n" % label)
	header.write("#include <stddef.h>\n")
	header.write("#include <stdint.h>\n\n")
	header.write("\n".join(header_ctx))
	header.write("\n\n")
	header.write("#endif\n")
	
	header.close()
	
	data = open("%s.s" % dest, "w")
	
	data.write('/***************************************\n')
	data.write(' * AUTO-GENERATED FILE, DO NOT EDIT!!! *\n')
	data.write(' * Generated by: tools/mkarray.py      *\n')
	data.write(' ***************************************/\n\n')
	data.write(as_prolog)
	data.write('.data\n\n')
	data.write("\n".join(data_ctx))
	
	data.close()

if __name__ == '__main__':
	main()