sequencer outline, credits on a black background wasnt easy to see.
[blender.git] / release / freedesktop / blender-thumbnailer.py
1 #!/usr/bin/python
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 """
22 Thumbnailer runs with python 2.6 and 3.x.
23 To run automatically with nautilus:
24    gconftool --type boolean --set /desktop/gnome/thumbnailers/application@x-blender/enable true
25    gconftool --type string --set /desktop/gnome/thumbnailers/application@x-blender/command "blender-thumbnailer.py %i %o"
26 """
27
28 import os
29 import struct
30 import sys
31
32 def blend_extract_thumb(path):
33     # def MAKE_ID(tag): ord(tag[0])<<24 | ord(tag[1])<<16 | ord(tag[2])<<8 | ord(tag[3])
34     REND = 1145980242 # MAKE_ID(b'REND')
35     TEST = 1414743380 # MAKE_ID(b'TEST')
36
37     blendfile = open(path, 'rb')
38
39     head = blendfile.read(7)
40
41     if head[0:2] == b'\x1f\x8b': # gzip magic
42         import gzip
43         blendfile.close()
44         blendfile = gzip.open(path, 'rb')
45         head = blendfile.read(7)
46
47     if head != b'BLENDER':
48         blendfile.close()
49         return None, 0, 0
50
51     is_64_bit = (blendfile.read(1) == b'-')
52
53     # true for PPC, false for X86
54     is_big_endian = (blendfile.read(1) == b'V')
55
56     # Now read the bhead chunk!!!
57     blendfile.read(3) # skip the version
58
59     sizeof_pointer = 8 if is_64_bit else 4
60
61     sizeof_bhead = 24 if is_64_bit else 20
62     
63     int_endian = '>i' if is_big_endian else '<i'
64     int_endian_pair = '>ii' if is_big_endian else '<ii'
65     
66     while True:
67         try:
68             code, length = struct.unpack(int_endian_pair, blendfile.read(8)) # 8 == sizeof(int) * 2
69         except IOError:
70             return None, 0, 0
71         
72         # finally read the rest of the bhead struct, pointer and 2 ints
73         blendfile.seek(sizeof_bhead - 8, os.SEEK_CUR)
74
75         if code == REND:
76             blendfile.seek(length, os.SEEK_CUR)
77         else:
78             break
79     
80     if code != TEST:
81         return None, 0, 0
82
83     try:
84         x, y = struct.unpack(int_endian_pair, blendfile.read(8)) # 8 == sizeof(int) * 2
85     except struct.error:
86         return None, 0, 0
87
88     length -= 8 # sizeof(int) * 2
89     
90     if length != x * y * 4:
91         return None, 0, 0
92     
93     image_buffer = blendfile.read(length)
94     
95     if len(image_buffer) != length:
96         return None, 0, 0
97
98     return image_buffer, x, y
99
100
101 def write_png(buf, width, height):
102     import zlib
103
104     # reverse the vertical line order and add null bytes at the start
105     width_byte_4 = width * 4
106     raw_data = b"".join([b'\x00' + buf[span:span + width_byte_4] for span in range((height - 1) * width * 4, -1, - width_byte_4)])
107     
108     def png_pack(png_tag, data):
109         chunk_head = png_tag + data
110         return struct.pack("!I", len(data)) + chunk_head + struct.pack("!I", 0xFFFFFFFF & zlib.crc32(chunk_head))
111
112     return b"".join([
113         b'\x89PNG\r\n\x1a\n',
114         png_pack(b'IHDR', struct.pack("!2I5B", width, height, 8, 6, 0, 0, 0)),
115         png_pack(b'IDAT', zlib.compress(raw_data, 9)),
116         png_pack(b'IEND', b'')])
117
118
119 if __name__ == '__main__':
120     if len(sys.argv) < 2:
121         print("Expected 2 arguments <input.blend> <output.png>")
122     else:
123         file_in = sys.argv[-2]
124
125         buf, width, height = blend_extract_thumb(file_in)
126         
127         if buf:
128             file_out = sys.argv[-1]
129
130             f = open(file_out, "wb")
131             f.write(write_png(buf, width, height))
132             f.close()