Le repo des sources pour le site web des JM2L
Nelze vybrat více než 25 témat Téma musí začínat písmenem nebo číslem, může obsahovat pomlčky („-“) a může být dlouhé až 35 znaků.

blenderthumbnailer.py 5.5 KiB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198
  1. #!/usr/bin/env python
  2. # ##### BEGIN GPL LICENSE BLOCK #####
  3. #
  4. # This program is free software; you can redistribute it and/or
  5. # modify it under the terms of the GNU General Public License
  6. # as published by the Free Software Foundation; either version 2
  7. # of the License, or (at your option) any later version.
  8. #
  9. # This program is distributed in the hope that it will be useful,
  10. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  12. # GNU General Public License for more details.
  13. #
  14. # You should have received a copy of the GNU General Public License
  15. # along with this program; if not, write to the Free Software Foundation,
  16. # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  17. #
  18. # ##### END GPL LICENSE BLOCK #####
  19. # <pep8 compliant>
  20. """
  21. Thumbnailer runs with python 2.7 and 3.x.
  22. To run automatically with a file manager such as Nautilus, save this file
  23. in a directory that is listed in PATH environment variable, and create
  24. blender.thumbnailer file in ${HOME}/.local/share/thumbnailers/ directory
  25. with the following contents:
  26. [Thumbnailer Entry]
  27. TryExec=blender-thumbnailer.py
  28. Exec=blender-thumbnailer.py %u %o
  29. MimeType=application/x-blender;
  30. """
  31. import struct
  32. def open_wrapper_get():
  33. """ wrap OS spesific read functionality here, fallback to 'open()'
  34. """
  35. class GFileWrapper:
  36. __slots__ = ("mode", "g_file")
  37. def __init__(self, url, mode='r'):
  38. self.mode = mode # used in gzip module
  39. self.g_file = Gio.File.parse_name(url).read(None)
  40. def read(self, size):
  41. return self.g_file.read_bytes(size, None).get_data()
  42. def seek(self, offset, whence=0):
  43. self.g_file.seek(offset, [1, 0, 2][whence], None)
  44. return self.g_file.tell()
  45. def tell(self):
  46. return self.g_file.tell()
  47. def close(self):
  48. self.g_file.close(None)
  49. def open_local_url(url, mode='r'):
  50. o = urlparse(url)
  51. if o.scheme == '':
  52. path = o.path
  53. elif o.scheme == 'file':
  54. path = unquote(o.path)
  55. else:
  56. raise(IOError('URL scheme "%s" needs gi.repository.Gio module' % o.scheme))
  57. return open(path, mode)
  58. try:
  59. from gi.repository import Gio
  60. return GFileWrapper
  61. except ImportError:
  62. try:
  63. # Python 3
  64. from urllib.parse import urlparse, unquote
  65. except ImportError:
  66. # Python 2
  67. from urlparse import urlparse
  68. from urllib import unquote
  69. return open_local_url
  70. def blend_extract_thumb(path):
  71. import os
  72. open_wrapper = open_wrapper_get()
  73. # def MAKE_ID(tag): ord(tag[0])<<24 | ord(tag[1])<<16 | ord(tag[2])<<8 | ord(tag[3])
  74. REND = 1145980242 # MAKE_ID(b'REND')
  75. TEST = 1414743380 # MAKE_ID(b'TEST')
  76. blendfile = open_wrapper(path, 'rb')
  77. head = blendfile.read(12)
  78. if head[0:2] == b'\x1f\x8b': # gzip magic
  79. import gzip
  80. blendfile.close()
  81. blendfile = gzip.GzipFile('', 'rb', 0, open_wrapper(path, 'rb'))
  82. head = blendfile.read(12)
  83. if not head.startswith(b'BLENDER'):
  84. blendfile.close()
  85. return None, 0, 0
  86. is_64_bit = (head[7] == b'-'[0])
  87. # true for PPC, false for X86
  88. is_big_endian = (head[8] == b'V'[0])
  89. # blender pre 2.5 had no thumbs
  90. if head[9:11] <= b'24':
  91. blendfile.close()
  92. return None, 0, 0
  93. sizeof_bhead = 24 if is_64_bit else 20
  94. int_endian_pair = '>ii' if is_big_endian else '<ii'
  95. while True:
  96. bhead = blendfile.read(sizeof_bhead)
  97. if len(bhead) < sizeof_bhead:
  98. return None, 0, 0
  99. code, length = struct.unpack(int_endian_pair, bhead[0:8]) # 8 == sizeof(int) * 2
  100. if code == REND:
  101. blendfile.seek(length, os.SEEK_CUR)
  102. else:
  103. break
  104. if code != TEST:
  105. blendfile.close()
  106. return None, 0, 0
  107. try:
  108. x, y = struct.unpack(int_endian_pair, blendfile.read(8)) # 8 == sizeof(int) * 2
  109. except struct.error:
  110. blendfile.close()
  111. return None, 0, 0
  112. length -= 8 # sizeof(int) * 2
  113. if length != x * y * 4:
  114. blendfile.close()
  115. return None, 0, 0
  116. image_buffer = blendfile.read(length)
  117. if len(image_buffer) != length:
  118. blendfile.close()
  119. return None, 0, 0
  120. blendfile.close()
  121. return image_buffer, x, y
  122. def write_png(buf, width, height):
  123. import zlib
  124. # reverse the vertical line order and add null bytes at the start
  125. width_byte_4 = width * 4
  126. raw_data = b"".join(b'\x00' + buf[span:span + width_byte_4] for span in range((height - 1) * width * 4, -1, - width_byte_4))
  127. def png_pack(png_tag, data):
  128. chunk_head = png_tag + data
  129. return struct.pack("!I", len(data)) + chunk_head + struct.pack("!I", 0xFFFFFFFF & zlib.crc32(chunk_head))
  130. return b"".join([
  131. b'\x89PNG\r\n\x1a\n',
  132. png_pack(b'IHDR', struct.pack("!2I5B", width, height, 8, 6, 0, 0, 0)),
  133. png_pack(b'IDAT', zlib.compress(raw_data, 9)),
  134. png_pack(b'IEND', b'')])
  135. def main():
  136. import sys
  137. if len(sys.argv) < 3:
  138. print("Expected 2 arguments <input.blend> <output.png>")
  139. else:
  140. file_in = sys.argv[-2]
  141. buf, width, height = blend_extract_thumb(file_in)
  142. if buf:
  143. file_out = sys.argv[-1]
  144. f = open(file_out, "wb")
  145. f.write(write_png(buf, width, height))
  146. f.close()
  147. if __name__ == '__main__':
  148. main()