X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=blender%2Fio_mspgl%2Fmaterial.py;h=cadfe75e09b5ef4589da97a5e99c8caec34e3538;hb=d962add24cb7e55fa30b63763b4dbf7f37af0079;hp=93f177908b4b6ede25de2a24b2e1f7d8fc1d5246;hpb=c3ebbcb56c1ca9bb3022a7f49aab1da5e09150ba;p=libs%2Fgl.git

diff --git a/blender/io_mspgl/material.py b/blender/io_mspgl/material.py
index 93f17790..cadfe75e 100644
--- a/blender/io_mspgl/material.py
+++ b/blender/io_mspgl/material.py
@@ -8,6 +8,33 @@ def get_linked_node_and_socket(node_tree, socket):
 			return (l.to_node, l.to_socket)
 	return (None, None)
 
+def check_group(node_tree, group, func):
+	output = group.node_tree.nodes["Group Output"]
+	from_node, _ = get_linked_node_and_socket(group.node_tree, output.inputs[0])
+	if from_node:
+		from_node, _ = func(group.node_tree, from_node)
+		if from_node and from_node.type=='GROUP_INPUT':
+			return get_linked_node_and_socket(node_tree, group.inputs[0])
+	return (None, None)
+
+def check_invert_green(node_tree, node):
+	if node.type=='GROUP':
+		return check_group(node_tree, node, check_invert_green)
+	elif node.type!='COMBRGB':
+		return (None, None)
+
+	green, g_sock = get_linked_node_and_socket(node_tree, node.inputs["G"])
+	if not green or green.type!='MATH' or green.operation!='SUBTRACT':
+		return (None, None)
+	green, g_sock = get_linked_node_and_socket(node_tree, green.inputs[1])
+
+	red, r_sock = get_linked_node_and_socket(node_tree, node.inputs["R"])
+	blue, b_sock = get_linked_node_and_socket(node_tree, node.inputs["B"])
+	if not red or red.type!='SEPRGB' or blue!=red or green!=red:
+		return (None, None)
+
+	return get_linked_node_and_socket(node_tree, red.inputs["Image"])
+
 class MaterialProperty:
 	def __init__(self, keyword, tex_keyword, value):
 		self.keyword = keyword
@@ -15,6 +42,7 @@ class MaterialProperty:
 		self.value = value
 		self.texture = None
 		self.tex_usage = None
+		self.invert_green = False
 
 	def set_from_input(self, node_tree, input_socket, alpha_socket=None):
 		if self.keyword:
@@ -30,8 +58,14 @@ class MaterialProperty:
 			from_node, _ = get_linked_node_and_socket(node_tree, input_socket)
 			alpha_from = None
 			if from_node:
+				usage = None
 				if from_node.type=='NORMAL_MAP':
 					from_node, _ = get_linked_node_and_socket(node_tree, from_node.inputs["Color"])
+					invert, _ = check_invert_green(node_tree, from_node)
+					if invert:
+						from_node = invert
+						self.invert_green = True
+					usage = 'RGB'
 				elif from_node.type=='RGBTOBW':
 					from_node, _ = get_linked_node_and_socket(node_tree, from_node.inputs["Color"])
 
@@ -42,7 +76,9 @@ class MaterialProperty:
 
 				if from_node.type=='TEX_IMAGE':
 					self.texture = from_node
-					if alpha_from:
+					if usage:
+						self.tex_usage = usage
+					elif alpha_from:
 						self.tex_usage = 'RGBA'
 					elif type(self.value)==tuple:
 						self.tex_usage = 'RGB'