155 lines
4.1 KiB
Python
Executable File
155 lines
4.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
# ##### BEGIN GPL LICENSE BLOCK #####
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU General Public License
|
|
# as published by the Free Software Foundation; either version 2
|
|
# of the License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software Foundation,
|
|
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
#
|
|
# ##### END GPL LICENSE BLOCK #####
|
|
|
|
# <pep8 compliant>
|
|
_IS_BIG_ENDIAN = (__import__("sys").byteorder != 'little')
|
|
|
|
|
|
def write_png(buf, width, height):
|
|
import zlib
|
|
import struct
|
|
# reverse the vertical line order and add null bytes at the start
|
|
width_byte_4 = width * 4
|
|
raw_data = b"".join(b'\x00' + buf[span:span + width_byte_4] for span in range((height - 1) * width * 4, -1, - width_byte_4))
|
|
|
|
def png_pack(png_tag, data):
|
|
chunk_head = png_tag + data
|
|
return struct.pack("!I", len(data)) + chunk_head + struct.pack("!I", 0xFFFFFFFF & zlib.crc32(chunk_head))
|
|
|
|
return b"".join([
|
|
b'\x89PNG\r\n\x1a\n',
|
|
png_pack(b'IHDR', struct.pack("!2I5B", width, height, 8, 6, 0, 0, 0)),
|
|
png_pack(b'IDAT', zlib.compress(raw_data, 9)),
|
|
png_pack(b'IEND', b'')])
|
|
|
|
|
|
def icon_decode_head(f_src):
|
|
import struct
|
|
|
|
# 2 ints
|
|
temp_data = f_src.read(4 * 2)
|
|
icon_w, icon_h = struct.unpack('<2I', temp_data)
|
|
|
|
temp_data = f_src.read(4 * 2)
|
|
orig_x, orig_y = struct.unpack('<2I', temp_data)
|
|
|
|
temp_data = f_src.read(4 * 2)
|
|
canvas_w, canvas_h = struct.unpack('<2I', temp_data)
|
|
|
|
return (icon_w, icon_h,
|
|
orig_x, orig_y,
|
|
canvas_w, canvas_h)
|
|
|
|
|
|
def icon_decode(f_src):
|
|
head = icon_decode_head(f_src)
|
|
|
|
(icon_w, icon_h,
|
|
orig_x, orig_y,
|
|
canvas_w, canvas_h) = head
|
|
|
|
# pixels
|
|
import array
|
|
|
|
pixels = f_src.read(icon_w * icon_h * 4)
|
|
pixels = array.array('I', pixels)
|
|
if _IS_BIG_ENDIAN:
|
|
pixels.byteswap()
|
|
|
|
return head, pixels
|
|
|
|
|
|
def icon_read(file_src):
|
|
with open(file_src, 'rb') as f_src:
|
|
head, pixels = icon_decode(f_src)
|
|
return head, pixels
|
|
|
|
|
|
def icon_merge(file_src, pixels_canvas, canvas_w, canvas_h):
|
|
""" Takes an icon filepath and merges into a pixel array
|
|
"""
|
|
head, pixels = icon_read(file_src)
|
|
|
|
(icon_w, icon_h,
|
|
orig_x, orig_y,
|
|
w_canvas_test, h_canvas_test) = head
|
|
|
|
assert(w_canvas_test == canvas_w)
|
|
assert(h_canvas_test == canvas_h)
|
|
|
|
for x in range(icon_w):
|
|
for y in range(icon_h):
|
|
# get pixel
|
|
pixel = pixels[(y * icon_w) + x]
|
|
|
|
# set pixel
|
|
dst_x = orig_x + x
|
|
dst_y = orig_y + y
|
|
pixels_canvas[(dst_y * canvas_w) + dst_x] = pixel
|
|
|
|
|
|
def icondir_to_png(path_src, file_dst):
|
|
""" Takes a path full of 'dat' files and writes out
|
|
"""
|
|
import os
|
|
import array
|
|
|
|
files = [os.path.join(path_src, f) for f in os.listdir(path_src) if f.endswith(".dat")]
|
|
|
|
with open(files[0], 'rb') as f_src:
|
|
(icon_w, icon_h,
|
|
orig_x, orig_y,
|
|
canvas_w, canvas_h) = icon_decode_head(f_src)
|
|
|
|
# load in pixel data
|
|
pixels_canvas = array.array('I', [0]) * (canvas_w * canvas_h)
|
|
for f in files:
|
|
icon_merge(f, pixels_canvas, canvas_w, canvas_h)
|
|
|
|
# write pixels
|
|
with open(file_dst, 'wb') as f_dst:
|
|
import sys
|
|
# py2/3 compat
|
|
if sys.version.startswith("2"):
|
|
pixels_data = pixels_canvas.tostring()
|
|
else:
|
|
pixels_data = pixels_canvas.tobytes()
|
|
|
|
image_data = write_png(pixels_data, canvas_w, canvas_h)
|
|
f_dst.write(image_data)
|
|
|
|
|
|
def main_ex(argv):
|
|
import os
|
|
|
|
path_src = argv[-2].rstrip(os.sep)
|
|
file_dst = argv[-1]
|
|
|
|
icondir_to_png(path_src, file_dst)
|
|
|
|
|
|
def main():
|
|
import sys
|
|
main_ex(sys.argv)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|