kannonier8/blender
1
0
Fork 0
forked from bartvdbraak/blender
Code Pull Requests Activity

Minor code simplification in previous commit.

Browse Source
This commit is contained in:
Brecht Van Lommel 2016-04-27 22:58:00 +02:00
parent cc692c0924
commit 2b485e21f4
1 changed files with 10 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
release/scripts/startup/nodeitems_builtins.py
View File

@ -24,45 +24,35 @@ from nodeitems_utils import NodeCategory, NodeItem, NodeItemCustom
# Subclasses for standard node types
def alphabetical(items):
# for builtin nodes the convention is to sort by name
if isinstance(items, list):
return sorted(items, key=lambda item: item.label().lower)
return items
class CompositorNodeCategory(NodeCategory):
class SortedNodeCategory(NodeCategory):
def __init__(self, identifier, name, description="", items=None):
super().__init__(identifier, name, description, alphabetical(items))
# for builtin nodes the convention is to sort by name
if isinstance(items, list):
items = sorted(items, key=lambda item: item.label.lower())
super().__init__(identifier, name, description, items)
class CompositorNodeCategory(SortedNodeCategory):
@classmethod
def poll(cls, context):
return (context.space_data.tree_type == 'CompositorNodeTree')
class ShaderNewNodeCategory(NodeCategory):
def __init__(self, identifier, name, description="", items=None):
super().__init__(identifier, name, description, alphabetical(items))
class ShaderNewNodeCategory(SortedNodeCategory):
@classmethod
def poll(cls, context):
return (context.space_data.tree_type == 'ShaderNodeTree' and
context.scene.render.use_shading_nodes)
class ShaderOldNodeCategory(NodeCategory):
def __init__(self, identifier, name, description="", items=None):
super().__init__(identifier, name, description, alphabetical(items))
class ShaderOldNodeCategory(SortedNodeCategory):
@classmethod
def poll(cls, context):
return (context.space_data.tree_type == 'ShaderNodeTree' and
not context.scene.render.use_shading_nodes)
class TextureNodeCategory(NodeCategory):
def __init__(self, identifier, name, description="", items=None):
super().__init__(identifier, name, description, alphabetical(items))
class TextureNodeCategory(SortedNodeCategory):
@classmethod
def poll(cls, context):
return context.space_data.tree_type == 'TextureNodeTree'