blender/io_mspgl/export_texture.py

   1 import os
   2 import base64
   3 import codecs
   4
   5 def encode_pixels(pixels):
   6         from .datafile import Token
   7         return Token(codecs.decode(b"="+base64.b64encode(bytes(pixels))+b"=", "ascii"))
   8
   9 def pixels_to_rgba(pixels):
  10         return (int(p*255) for p in pixels)
  11
  12 def pixels_to_rgb(pixels):
  13         for i in range(0, len(pixels), 4):
  14                 yield int(pixels[i]*255)
  15                 yield int(pixels[i+1]*255)
  16                 yield int(pixels[i+2]*255)
  17
  18 def pixels_to_gray(pixels):
  19         for i in range(0, len(pixels), 4):
  20                 yield int((pixels[i]+pixels[i+1]+pixels[i+2])*255/3)
  21
  22 class TextureExporter:
  23         def __init__(self):
  24                 self.inline_data = True
  25
  26         def export_texture(self, tex_node, usage='RGB'):
  27                 image = tex_node.image
  28                 from .datafile import Resource, Statement, Token
  29                 tex_res = Resource(image.name+".tex2d", "texture2d")
  30
  31                 if tex_node.use_mipmap:
  32                         tex_res.statements.append(Statement("generate_mipmap", True))
  33
  34                 colorspace = image.colorspace_settings.name
  35                 if usage=='GRAY' and colorspace=='sRGB':
  36                                 raise Exception("Grayscale textures with sRGB colorspace are not supported")
  37
  38                 from .util import basename
  39                 fn = basename(image.filepath)
  40                 if not self.inline_data and fn:
  41                         srgb = "_srgb" if colorspace=='sRGB' else ""
  42                         tex_res.statements.append(Statement("external_image"+srgb, fn))
  43                 else:
  44                         if usage=='RGBA':
  45                                 fmt = 'SRGB8_ALPHA8' if colorspace=='sRGB' else 'RGBA8'
  46                         elif usage=='GRAY':
  47                                 fmt = 'LUMINANCE8'
  48                         else:
  49                                 fmt = 'SRGB8' if colorspace=='sRGB' else 'RGB8'
  50
  51                         tex_res.statements.append(Statement("storage", Token(fmt), image.size[0], image.size[1]))
  52
  53                         pixels = tuple(image.pixels)
  54                         texdata = ""
  55                         if usage=='RGBA':
  56                                 texdata = encode_pixels(pixels_to_rgba(pixels))
  57                         elif usage=='GRAY':
  58                                 texdata = encode_pixels(pixels_to_gray(pixels))
  59                         else:
  60                                 texdata = encode_pixels(pixels_to_rgb(pixels))
  61                         tex_res.statements.append(Statement("raw_data", texdata))
  62
  63                 return tex_res
  64
  65 class SamplerExporter:
  66         def export_sampler(self, tex_node):
  67                 from .datafile import Resource, Statement, Token
  68                 samp_res = Resource(self.get_sampler_name(tex_node), "sampler")
  69
  70                 use_interpolation = tex_node.interpolation!='Closest'
  71                 if use_interpolation:
  72                         if tex_node.use_mipmap:
  73                                 samp_res.statements.append(Statement("filter", Token('LINEAR_MIPMAP_LINEAR')))
  74                         else:
  75                                 samp_res.statements.append(Statement("filter", Token('LINEAR')))
  76                         samp_res.statements.append(Statement("max_anisotropy", tex_node.max_anisotropy))
  77                 else:
  78                         if tex_node.use_mipmap:
  79                                 samp_res.statements.append(Statement("filter", Token('NEAREST_MIPMAP_NEAREST')))
  80                         else:
  81                                 samp_res.statements.append(Statement("filter", Token('NEAREST')))
  82
  83                 if tex_node.extension=="REPEAT":
  84                         samp_res.statements.append(Statement("wrap", Token('REPEAT')))
  85                 elif tex_node.extension=="EXTEND":
  86                         samp_res.statements.append(Statement("wrap", Token('CLAMP_TO_EDGE')))
  87                 elif tex_node.extension=="CLIP":
  88                         samp_res.statements.append(Statement("wrap", Token('CLAMP_TO_BORDER')))
  89                         samp_res.statements.append(Statement("border_color", 0.0, 0.0, 0.0, 0.0))
  90
  91                 return samp_res
  92
  93         def get_sampler_name(self, tex_node):
  94                 name_parts = []
  95
  96                 use_interpolation = tex_node.interpolation!='Closest'
  97                 name_parts.append("linear" if use_interpolation else "nearest")
  98                 if tex_node.use_mipmap:
  99                         name_parts.append("mip")
 100                 if use_interpolation and tex_node.max_anisotropy>1:
 101                         name_parts.append("aniso{:g}x".format(tex_node.max_anisotropy))
 102                 if tex_node.extension!="REPEAT":
 103                         name_parts.append("clip" if tex_node.extension=="CLIP" else "clamp")
 104
 105                 return "_".join(name_parts)+".samp"