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

diff --git a/blender/io_mspgl/material.py b/blender/io_mspgl/material.py
index f7f2c5eb..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,16 @@ 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"])
 
 				if alpha_socket:
 					alpha_from, _ = get_linked_node_and_socket(node_tree, alpha_socket)
@@ -40,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'
@@ -84,7 +122,7 @@ class Material:
 			base_color = self.create_property("base_color", (0.8, 0.8, 0.8, 1.0))
 			metalness = self.create_property("metalness", 0.0)
 			roughness = self.create_property("roughness", 0.5)
-			normal = self.create_property("normal")
+			normal = self.create_property("normal_map")
 			emission = self.create_property("emission", (0.0, 0.0, 0.0))
 
 			base_color.set_from_input(material.node_tree, surface_node.inputs["Base Color"], surface_node.inputs["Alpha"])