BogdanNagirniak/blender-addons
0
1
Fork 2
forked from blender/blender-addons
Code Pull Requests Activity

Adjust generating node classes and searching of mtlx files #7

Merged
Bogdan Nagirniak merged 2 commits from BLEN-363 into materialx-addon 2023-04-11 17:50:44 +02:00
Conversation 0 Commits 2 Files Changed 2 +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 e919466c2d - Show all commits

57
materialx/bl_nodes/shader.py
View File

@ -5,6 +5,7 @@ import math
import MaterialX as mx import MaterialX as mx
from .node_parser import NodeParser from .node_parser import NodeParser
from ..utils import get_mx_node_input_types
from . import log from . import log
@ -227,18 +228,31 @@ class ShaderNodeMixShader(NodeParser):
shader2 = self.get_input_link(2) shader2 = self.get_input_link(2)
mix = None mix = None
input_types = get_mx_node_input_types('mix', 'PBR')
if shader1 is None and shader2 is None: if shader1 is None and shader2 is None:
return None return None
if shader2 is 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, 'fg': shader1,
'mix': factor 'mix': factor
}) })
if shader1 is None: 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, 'bg': shader2,
'mix': factor 'mix': factor
}) })
@ -247,10 +261,15 @@ class ShaderNodeMixShader(NodeParser):
shader1_type = get_node_type(shader1) shader1_type = get_node_type(shader1)
shader2_type = get_node_type(shader2) shader2_type = get_node_type(shader2)
if shader1_type != shader2_type: if shader1_type != shader2_type:
log.warn(f'Types of input shaders must be the same.' log.warn(f'Types of input shaders must be the same. '
f' First shader type: {shader1_type}, second shader type: {shader2_type}') 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={ mix = self.create_node('mix', shader1_type.lower(), prefix='PBR', inputs={
'fg': shader1, 'fg': shader1,
@ -277,17 +296,30 @@ class ShaderNodeAddShader(NodeParser):
shader2 = self.get_input_link(1) shader2 = self.get_input_link(1)
add = None add = None
input_types = get_mx_node_input_types('add', 'PBR')
if shader1 is None and shader2 is None: if shader1 is None and shader2 is None:
return None return None
if shader2 is 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 'in1': shader1
}) })
if shader1 is None: 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 'in2': shader2
}) })
@ -295,10 +327,15 @@ class ShaderNodeAddShader(NodeParser):
shader1_type = get_node_type(shader1) shader1_type = get_node_type(shader1)
shader2_type = get_node_type(shader2) shader2_type = get_node_type(shader2)
if shader1_type != shader2_type: if shader1_type != shader2_type:
log.warn(f'Types of input shaders must be the same.' log.warn(f'Types of input shaders must be the same. '
f' First shader type: {shader1_type}, second shader type: {shader2_type}') 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={ add = self.create_node('add', shader1_type.lower(), prefix='PBR', inputs={
'in1': shader1, 'in1': shader1,

16
materialx/utils.py
View File

@ -665,3 +665,19 @@ def get_output_node(material):
node.inputs['surfaceshader'].links), None) node.inputs['surfaceshader'].links), None)
return mx_output_node 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