Bpy Extras: Port cycles node functions to new node_utils.py
This commit is contained in:
parent
470d66397d
commit
7b379313de
@ -17,6 +17,7 @@
|
||||
# <pep8 compliant>
|
||||
|
||||
import bpy
|
||||
from bpy_extras.node_utils import find_node_input, find_output_node
|
||||
|
||||
from bpy.types import (
|
||||
Panel,
|
||||
@ -898,30 +899,6 @@ class CYCLES_OT_use_shading_nodes(Operator):
|
||||
return {'FINISHED'}
|
||||
|
||||
|
||||
def find_node(material, nodetype):
|
||||
if material and material.node_tree:
|
||||
ntree = material.node_tree
|
||||
|
||||
active_output_node = None
|
||||
for node in ntree.nodes:
|
||||
if getattr(node, "type", None) == nodetype:
|
||||
if getattr(node, "is_active_output", True):
|
||||
return node
|
||||
if not active_output_node:
|
||||
active_output_node = node
|
||||
return active_output_node
|
||||
|
||||
return None
|
||||
|
||||
|
||||
def find_node_input(node, name):
|
||||
for input in node.inputs:
|
||||
if input.name == name:
|
||||
return input
|
||||
|
||||
return None
|
||||
|
||||
|
||||
def panel_node_draw(layout, id_data, output_type, input_name):
|
||||
if not id_data.use_nodes:
|
||||
layout.operator("cycles.use_shading_nodes", icon='NODETREE')
|
||||
@ -929,7 +906,7 @@ def panel_node_draw(layout, id_data, output_type, input_name):
|
||||
|
||||
ntree = id_data.node_tree
|
||||
|
||||
node = find_node(id_data, output_type)
|
||||
node = find_output_node(ntree, output_type)
|
||||
if not node:
|
||||
layout.label(text="No output node")
|
||||
else:
|
||||
|
@ -29,5 +29,6 @@ __all__ = (
|
||||
"image_utils",
|
||||
"keyconfig_utils",
|
||||
"mesh_utils",
|
||||
"node_utils",
|
||||
"view3d_utils",
|
||||
)
|
||||
|
47
release/scripts/modules/bpy_extras/node_utils.py
Normal file
47
release/scripts/modules/bpy_extras/node_utils.py
Normal file
@ -0,0 +1,47 @@
|
||||
# ##### BEGIN GPL LICENSE BLOCK #####
|
||||
#
|
||||
# This program is free software; you can redistribute it and/or
|
||||
# modify it under the terms of the GNU General Public License
|
||||
# as published by the Free Software Foundation; either version 2
|
||||
# of the License, or (at your option) any later version.
|
||||
#
|
||||
# This program is distributed in the hope that it will be useful,
|
||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
# GNU General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU General Public License
|
||||
# along with this program; if not, write to the Free Software Foundation,
|
||||
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
||||
#
|
||||
# ##### END GPL LICENSE BLOCK #####
|
||||
|
||||
# <pep8-80 compliant>
|
||||
|
||||
__all__ = (
|
||||
"find_node_input",
|
||||
"find_output_node",
|
||||
)
|
||||
|
||||
|
||||
# XXX Names are not unique. Returns the first match.
|
||||
def find_node_input(node, name):
|
||||
for input in node.inputs:
|
||||
if input.name == name:
|
||||
return input
|
||||
|
||||
return None
|
||||
|
||||
# Return the output node to display in the UI
|
||||
def find_output_node(ntree, nodetype):
|
||||
if ntree:
|
||||
active_output_node = None
|
||||
for node in ntree.nodes:
|
||||
if getattr(node, "type", None) == nodetype:
|
||||
if getattr(node, "is_active_output", True):
|
||||
return node
|
||||
if not active_output_node:
|
||||
active_output_node = node
|
||||
return active_output_node
|
||||
|
||||
return None
|
Loading…
Reference in New Issue
Block a user