Embedding PNG Images-Bin2c In Python

From WxWiki
Jump to navigation Jump to search
#!/usr/bin/python
import sys
import os
import os.path
import re
import array

USAGE = """png2c - Embed a PNG in a C header file (like XPM)
Usage: png2c [file ..] Convert all the file <file.png> to <file.png.h>"""

if len(sys.argv) < 2:
	print USAGE
	sys.exit(1)

r = re.compile("^([a-zA-Z._][a-zA-Z._0-9]*)[.][pP][nN][gG]$")

for path in sys.argv[1:]:
	filename = os.path.basename(path)
	m = r.match(filename)
	# Allow only filenames that make sense
	# as C variable names
	if not(m):
		print "Skipped file (unsuitable filename): " + filename
		continue

	# Read PNG file as character array
	bytes = array.array('B', open(path, "rb").read())
	count = len(bytes)

	# Create the C header
	text = "/* %s - %d bytes */\n" \
		"static const unsigned char %s_png[] = {\n" % (filename, count, m.group(1))

	# Iterate the characters, we want
	# lines like:
	#   0x01, 0x02, .... (8 values per line maximum)
	i = 0
	count = len(bytes)
	for byte in bytes:
		# Every new line starts with two whitespaces
		if (i % 8) == 0:
			text += "  "
		# Then the hex data (up to 8 values per line)
		text += "0x%02x" % (byte)
		# Separate all but the last values 
		if (i + 1) < count:
			text += ", "
		if (i % 8) == 7:
			text += '\n'
		i += 1

	# Now conclude the C source
	text += "};\n/* End Of File */\n"

	open(path + ".h", 'w').write(text)