NLA: rewrite evaluation channel data structures.
[blender.git] / source / blender / datatoc / datatoc_icon.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 # <pep8 compliant>
22 _IS_BIG_ENDIAN = (__import__("sys").byteorder != 'little')
23
24
25 def write_png(buf, width, height):
26     import zlib
27     import struct
28     # reverse the vertical line order and add null bytes at the start
29     width_byte_4 = width * 4
30     raw_data = b"".join(b'\x00' + buf[span:span + width_byte_4] for span in range((height - 1) * width * 4, -1, - width_byte_4))
31
32     def png_pack(png_tag, data):
33         chunk_head = png_tag + data
34         return struct.pack("!I", len(data)) + chunk_head + struct.pack("!I", 0xFFFFFFFF & zlib.crc32(chunk_head))
35
36     return b"".join([
37         b'\x89PNG\r\n\x1a\n',
38         png_pack(b'IHDR', struct.pack("!2I5B", width, height, 8, 6, 0, 0, 0)),
39         png_pack(b'IDAT', zlib.compress(raw_data, 9)),
40         png_pack(b'IEND', b'')])
41
42
43 def icon_decode_head(f_src):
44     import struct
45
46     # 2 ints
47     temp_data = f_src.read(4 * 2)
48     icon_w, icon_h = struct.unpack('<2I', temp_data)
49
50     temp_data = f_src.read(4 * 2)
51     orig_x, orig_y = struct.unpack('<2I', temp_data)
52
53     temp_data = f_src.read(4 * 2)
54     canvas_w, canvas_h = struct.unpack('<2I', temp_data)
55
56     return (icon_w, icon_h,
57             orig_x, orig_y,
58             canvas_w, canvas_h)
59
60
61 def icon_decode(f_src):
62     head = icon_decode_head(f_src)
63
64     (icon_w, icon_h,
65      orig_x, orig_y,
66      canvas_w, canvas_h) = head
67
68     # pixels
69     import array
70
71     pixels = f_src.read(icon_w * icon_h * 4)
72     pixels = array.array('I', pixels)
73     if _IS_BIG_ENDIAN:
74         pixels.byteswap()
75
76     return head, pixels
77
78
79 def icon_read(file_src):
80     with open(file_src, 'rb') as f_src:
81         head, pixels = icon_decode(f_src)
82     return head, pixels
83
84
85 def icon_merge(file_src, pixels_canvas, canvas_w, canvas_h):
86     """ Takes an icon filepath and merges into a pixel array
87     """
88     head, pixels = icon_read(file_src)
89
90     (icon_w, icon_h,
91      orig_x, orig_y,
92      w_canvas_test, h_canvas_test) = head
93
94     assert(w_canvas_test == canvas_w)
95     assert(h_canvas_test == canvas_h)
96
97     for x in range(icon_w):
98         for y in range(icon_h):
99             # get pixel
100             pixel = pixels[(y * icon_w) + x]
101
102             # set pixel
103             dst_x = orig_x + x
104             dst_y = orig_y + y
105             pixels_canvas[(dst_y * canvas_w) + dst_x] = pixel
106
107
108 def icondir_to_png(path_src, file_dst):
109     """ Takes a path full of 'dat' files and writes out
110     """
111     import os
112     import array
113
114     files = [os.path.join(path_src, f) for f in os.listdir(path_src) if f.endswith(".dat")]
115
116     # First check if we need to bother.
117     if os.path.exists(file_dst):
118         dst_time = os.path.getmtime(file_dst)
119         has_newer = False
120         for f in files:
121             if os.path.getmtime(f) > dst_time:
122                 has_newer = True
123                 break
124         if not has_newer:
125             return
126
127     with open(files[0], 'rb') as f_src:
128         (icon_w, icon_h,
129          orig_x, orig_y,
130          canvas_w, canvas_h) = icon_decode_head(f_src)
131
132     # load in pixel data
133     pixels_canvas = array.array('I', [0]) * (canvas_w * canvas_h)
134     for f in files:
135         icon_merge(f, pixels_canvas, canvas_w, canvas_h)
136
137     # write pixels
138     with open(file_dst, 'wb') as f_dst:
139         import sys
140         # py2/3 compat
141         if sys.version.startswith("2"):
142             pixels_data = pixels_canvas.tostring()
143         else:
144             pixels_data = pixels_canvas.tobytes()
145
146         image_data = write_png(pixels_data, canvas_w, canvas_h)
147         f_dst.write(image_data)
148
149
150 def main_ex(argv):
151     import os
152
153     path_src = argv[-2].rstrip(os.sep)
154     file_dst = argv[-1]
155
156     icondir_to_png(path_src, file_dst)
157
158
159 def main():
160     import sys
161     main_ex(sys.argv)
162
163
164 if __name__ == "__main__":
165     main()