blender-thumbnailer.py: Fix GVFS support not working with Python 3
[blender.git] / release / bin / blender-thumbnailer.py
1 #!/usr/bin/env 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 # <pep8 compliant>
22
23 """
24 Thumbnailer runs with python 2.7 and 3.x.
25 To run automatically with a file manager such as Nautilus, save this file
26 in a directory that is listed in PATH environment variable, and create
27 blender.thumbnailer file in ${HOME}/.local/share/thumbnailers/ directory
28 with the following contents:
29
30 [Thumbnailer Entry]
31 TryExec=blender-thumbnailer.py
32 Exec=blender-thumbnailer.py %u %o
33 MimeType=application/x-blender;
34 """
35
36 import struct
37
38
39 def open_wrapper_get():
40     """ wrap OS spesific read functionality here, fallback to 'open()'
41     """
42
43     class GFileWrapper:
44         __slots__ = ("mode", "g_file")
45         def __init__(self, url, mode='r'):
46             self.mode = mode # used in gzip module
47             self.g_file = Gio.File.parse_name(url).read(None)
48
49         def read(self, size):
50             return self.g_file.read_bytes(size, None).get_data()
51
52         def seek(self, offset, whence=0):
53             self.g_file.seek(offset, [1, 0, 2][whence], None)
54             return self.g_file.tell()
55
56         def tell(self):
57             return self.g_file.tell()
58
59         def close(self):
60             self.g_file.close(None)
61
62     def open_local_url(url, mode='r'):
63         o = urlparse(url)
64         if o.scheme == '':
65             path = o.path
66         elif o.scheme == 'file':
67             path = unquote(o.path)
68         else:
69             raise(IOError('URL scheme "%s" needs gi.repository.Gio module' % o.scheme))
70         return open(path, mode)
71
72     try:
73         from gi.repository import Gio
74         return GFileWrapper
75     except ImportError:
76         try:
77             # Python 3
78             from urllib.parse import urlparse, unquote
79         except ImportError:
80             # Python 2
81             from urlparse import urlparse
82             from urllib import unquote
83         return open_local_url
84
85
86 def blend_extract_thumb(path):
87     import os
88     open_wrapper = open_wrapper_get()
89
90     # def MAKE_ID(tag): ord(tag[0])<<24 | ord(tag[1])<<16 | ord(tag[2])<<8 | ord(tag[3])
91     REND = 1145980242  # MAKE_ID(b'REND')
92     TEST = 1414743380  # MAKE_ID(b'TEST')
93
94     blendfile = open_wrapper(path, 'rb')
95
96     head = blendfile.read(12)
97
98     if head[0:2] == b'\x1f\x8b':  # gzip magic
99         import gzip
100         blendfile.close()
101         blendfile = gzip.GzipFile('', 'rb', 0, open_wrapper(path, 'rb'))
102         head = blendfile.read(12)
103
104     if not head.startswith(b'BLENDER'):
105         blendfile.close()
106         return None, 0, 0
107
108     is_64_bit = (head[7] == b'-'[0])
109
110     # true for PPC, false for X86
111     is_big_endian = (head[8] == b'V'[0])
112
113     # blender pre 2.5 had no thumbs
114     if head[9:11] <= b'24':
115         return None, 0, 0
116
117     sizeof_bhead = 24 if is_64_bit else 20
118     int_endian_pair = '>ii' if is_big_endian else '<ii'
119
120     while True:
121         bhead = blendfile.read(sizeof_bhead)
122
123         if len(bhead) < sizeof_bhead:
124             return None, 0, 0
125
126         code, length = struct.unpack(int_endian_pair, bhead[0:8])  # 8 == sizeof(int) * 2
127
128         if code == REND:
129             blendfile.seek(length, os.SEEK_CUR)
130         else:
131             break
132
133     if code != TEST:
134         return None, 0, 0
135
136     try:
137         x, y = struct.unpack(int_endian_pair, blendfile.read(8))  # 8 == sizeof(int) * 2
138     except struct.error:
139         return None, 0, 0
140
141     length -= 8  # sizeof(int) * 2
142
143     if length != x * y * 4:
144         return None, 0, 0
145
146     image_buffer = blendfile.read(length)
147
148     if len(image_buffer) != length:
149         return None, 0, 0
150
151     return image_buffer, x, y
152
153
154 def write_png(buf, width, height):
155     import zlib
156
157     # reverse the vertical line order and add null bytes at the start
158     width_byte_4 = width * 4
159     raw_data = b"".join(b'\x00' + buf[span:span + width_byte_4] for span in range((height - 1) * width * 4, -1, - width_byte_4))
160
161     def png_pack(png_tag, data):
162         chunk_head = png_tag + data
163         return struct.pack("!I", len(data)) + chunk_head + struct.pack("!I", 0xFFFFFFFF & zlib.crc32(chunk_head))
164
165     return b"".join([
166         b'\x89PNG\r\n\x1a\n',
167         png_pack(b'IHDR', struct.pack("!2I5B", width, height, 8, 6, 0, 0, 0)),
168         png_pack(b'IDAT', zlib.compress(raw_data, 9)),
169         png_pack(b'IEND', b'')])
170
171
172 if __name__ == '__main__':
173     import sys
174
175     if len(sys.argv) < 3:
176         print("Expected 2 arguments <input.blend> <output.png>")
177     else:
178         file_in = sys.argv[-2]
179
180         buf, width, height = blend_extract_thumb(file_in)
181
182         if buf:
183             file_out = sys.argv[-1]
184
185             f = open(file_out, "wb")
186             f.write(write_png(buf, width, height))
187             f.close()