1 files changed, 155 insertions, 0 deletions
diff --git a/source/blender/datatoc/datatoc_icon.py b/source/blender/datatoc/datatoc_icon.py
new file mode 100755
index 00000000000..4f995a3219a
--- /dev/null
+++ b/source/blender/datatoc/datatoc_icon.py
@@ -0,0 +1,155 @@
+#!/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()
+