blender/blender-addons
19
43
Fork 96
Code Issues 630 Pull Requests 27 Projects 1 Releases Wiki Activity

WIP: MaterialX addon #104594

Closed
Bogdan Nagirniak wants to merge 34 commits from BogdanNagirniak/blender-addons:materialx-addon into main
pull from: BogdanNagirniak/blender-addons:materialx-addon
merge into: blender:main

When changing the target branch, be careful to rebase the branch in your fork to match. See documentation.
Conversation 3 Commits 34 Files Changed 18 +63 -10
2 changed files with 63 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit bcc93b0452 - Show all commits

49
materialx/bl_nodes/shader.py
View File

@ -5,6 +5,7 @@ import math
import MaterialX as mx
from .node_parser import NodeParser
from ..utils import get_mx_node_input_types
from . import log
@ -227,18 +228,31 @@ class ShaderNodeMixShader(NodeParser):
shader2 = self.get_input_link(2)
mix = None
input_types = get_mx_node_input_types('mix', 'PBR')
if shader1 is None and shader2 is None:
return None
if shader2 is None:
mix = self.create_node('mix', get_node_type(shader1).lower(), prefix='PBR', inputs={
shader1_type = get_node_type(shader1)
if shader1_type not in input_types:
log.warn(f'Node {self.node.bl_idname} ({self.material.name_full}) ignored. '
f'Input type must be of types {input_types}, actual: {shader1_type}')
return shader1
mix = self.create_node('mix', shader1_type.lower(), prefix='PBR', inputs={
'fg': shader1,
'mix': factor
})
if shader1 is None:
mix = self.create_node('mix', get_node_type(shader2).lower(), prefix='PBR', inputs={
shader2_type = get_node_type(shader2)
if shader2_type not in input_types:
log.warn(f'Node {self.node.bl_idname} ({self.material.name_full}) ignored. '
f'Input type must be of types {input_types}, actual: {shader2_type}')
return shader2
mix = self.create_node('mix', shader2_type.lower(), prefix='PBR', inputs={
'bg': shader2,
'mix': factor
})
@ -250,7 +264,12 @@ class ShaderNodeMixShader(NodeParser):
log.warn(f'Types of input shaders must be the same. '
f'First shader type: {shader1_type}, second shader type: {shader2_type}')
return None
return shader1
if shader1_type not in input_types:
log.warn(f'Node {self.node.bl_idname} ({self.material.name_full}) ignored. '
f'Input type must be of types {input_types}, actual: {shader1_type}, {shader2_type}')
return shader1
mix = self.create_node('mix', shader1_type.lower(), prefix='PBR', inputs={
'fg': shader1,
@ -277,17 +296,30 @@ class ShaderNodeAddShader(NodeParser):
shader2 = self.get_input_link(1)
add = None
input_types = get_mx_node_input_types('add', 'PBR')
if shader1 is None and shader2 is None:
return None
if shader2 is None:
add = self.create_node('add', get_node_type(shader1).lower(), prefix='PBR', inputs={
shader1_type = get_node_type(shader1)
if shader1_type not in input_types:
log.warn(f'Node {self.node.bl_idname} ({self.material.name_full}) ignored. '
f'Input type must be of types {input_types}, actual: {shader1_type}')
return shader1
add = self.create_node('add', shader1_type.lower(), prefix='PBR', inputs={
'in1': shader1
})
if shader1 is None:
add = self.create_node('add', get_node_type(shader2).lower(), prefix='PBR', inputs={
shader2_type = get_node_type(shader2)
if shader2_type not in input_types:
log.warn(f'Node {self.node.bl_idname} ({self.material.name_full}) ignored. '
f'Input type must be of types {input_types}, actual: {shader2_type}')
return shader2
add = self.create_node('add', shader2_type.lower(), prefix='PBR', inputs={
'in2': shader2
})
@ -298,7 +330,12 @@ class ShaderNodeAddShader(NodeParser):
log.warn(f'Types of input shaders must be the same. '
f'First shader type: {shader1_type}, second shader type: {shader2_type}')
return None
return shader1
if shader1_type not in input_types:
log.warn(f'Node {self.node.bl_idname} ({self.material.name_full}) ignored. '
f'Input type must be of types {input_types}, actual: {shader1_type}, {shader2_type}')
return shader1
add = self.create_node('add', shader1_type.lower(), prefix='PBR', inputs={
'in1': shader1,

16
materialx/utils.py
View File

@ -662,3 +662,19 @@ def get_output_node(material):
node.inputs['surfaceshader'].links), None)
return mx_output_node
def get_mx_node_input_types(node_name, prefix=''):
from .nodes import mx_node_classes
suffix = f'_{node_name}'
if prefix:
suffix = prefix + suffix
input_types = set()
classes = tuple(cls for cls in mx_node_classes if cls.__name__.endswith(suffix))
for cls in classes:
for nodedef, data_type in cls.get_nodedefs():
input_types |= {p.getType() for p in nodedef.getActiveInputs()}
return input_types