source/blender/datatoc/datatoc_icon_split_to_png.py

   1 #!/usr/bin/env python3
   2
   3 # ##### BEGIN GPL LICENSE BLOCK #####
   4 #
   5 #  This program is free software; you can redistribute it and/or
   6 #  modify it under the terms of the GNU General Public License
   7 #  as published by the Free Software Foundation; either version 2
   8 #  of the License, or (at your option) any later version.
   9 #
  10 #  This program is distributed in the hope that it will be useful,
  11 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 #  GNU General Public License for more details.
  14 #
  15 #  You should have received a copy of the GNU General Public License
  16 #  along with this program; if not, write to the Free Software Foundation,
  17 #  Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  18 #
  19 # ##### END GPL LICENSE BLOCK #####
  20
  21 # This script is just to view the icons
  22
  23
  24 def write_png(buf, width, height):
  25     import zlib
  26     import struct
  27     # reverse the vertical line order and add null bytes at the start
  28     width_byte_4 = width * 4
  29     raw_data = b"".join(b'\x00' + buf[span:span + width_byte_4] for span in range((height - 1) * width * 4, -1, - width_byte_4))
  30
  31     def png_pack(png_tag, data):
  32         chunk_head = png_tag + data
  33         return struct.pack("!I", len(data)) + chunk_head + struct.pack("!I", 0xFFFFFFFF & zlib.crc32(chunk_head))
  34
  35     return b"".join([
  36         b'\x89PNG\r\n\x1a\n',
  37         png_pack(b'IHDR', struct.pack("!2I5B", width, height, 8, 6, 0, 0, 0)),
  38         png_pack(b'IDAT', zlib.compress(raw_data, 9)),
  39         png_pack(b'IEND', b'')])
  40
  41
  42 def icondata_to_png(file_src, file_dst):
  43     import struct
  44
  45     with open(file_src, 'rb') as f_src:
  46         # 2 ints
  47         temp_data = f_src.read(4 * 2)
  48         w, h = struct.unpack('<2I', temp_data)
  49         temp_data = f_src.read(4 * 2)  # (x, y)         - ignored
  50         temp_data = f_src.read(4 * 2)  # (xfrom, yfrom) - ignored
  51         # pixels
  52         temp_data = f_src.read(w * h * 4)
  53
  54     buf = write_png(temp_data, w, h)
  55
  56     with open(file_dst, 'wb') as f_dst:
  57         f_dst.write(buf)
  58
  59
  60 def main():
  61     import sys
  62     import os
  63
  64     for arg in sys.argv[1:]:
  65         file_src = arg
  66         file_dst = os.path.splitext(arg)[0] + ".png"
  67
  68         icondata_to_png(file_src, file_dst)
  69
  70
  71 if __name__ == "__main__":
  72     main()