Recognize a certain configuration of nodes as inverting Y for normals

[libs/gl.git] / blender / io_mspgl / export_texture.py

diff --git a/blender/io_mspgl/export_texture.py b/blender/io_mspgl/export_texture.py
index 6ec13eadc583ff680495451609b25651bc173f1c..8e3872b5c8646f1084c83324b73f15499c9d9e9a 100644 (file)
--- a/blender/io_mspgl/export_texture.py
+++ b/blender/io_mspgl/export_texture.py
@@ -15,6 +15,12 @@ def pixels_to_rgb(pixels):
                yield int(pixels[i+1]*255)
                yield int(pixels[i+2]*255)
 
+def pixels_to_rgb_invert_green(pixels):
+       for i in range(0, len(pixels), 4):
+               yield int(pixels[i]*255)
+               yield 255-int(pixels[i+1]*255)
+               yield int(pixels[i+2]*255)
+
 def pixels_to_gray(pixels):
        for i in range(0, len(pixels), 4):
                yield int((pixels[i]+pixels[i+1]+pixels[i+2])*255/3)
@@ -23,7 +29,7 @@ class TextureExporter:
        def __init__(self):
                self.inline_data = True
 
-       def export_texture(self, tex_node, usage='RGB'):
+       def export_texture(self, tex_node, usage='RGB', *, invert_green=False):
                image = tex_node.image
                from .datafile import Resource, Statement, Token
                tex_res = Resource(image.name+".tex2d", "texture2d")
@@ -37,7 +43,7 @@ class TextureExporter:
 
                from .util import basename
                fn = basename(image.filepath)
-               if not self.inline_data and fn:
+               if not self.inline_data and not invert_green and fn:
                        srgb = "_srgb" if colorspace=='sRGB' else ""
                        tex_res.statements.append(Statement("external_image"+srgb, fn))
                else:
@@ -56,6 +62,8 @@ class TextureExporter:
                                texdata = encode_pixels(pixels_to_rgba(pixels))
                        elif usage=='GRAY':
                                texdata = encode_pixels(pixels_to_gray(pixels))
+                       elif invert_green:
+                               texdata = encode_pixels(pixels_to_rgb_invert_green(pixels))
                        else:
                                texdata = encode_pixels(pixels_to_rgb(pixels))
                        tex_res.statements.append(Statement("raw_data", texdata))