maxmustermann/armory
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Merge pull request #1868 from MoritzBrueckner/addnode-poll

Browse source 
Add poll() function to AddNodeOverride operator
This commit is contained in:
Lubos Lenco 2020-09-15 12:49:10 +02:00 committed by GitHub
parent 43aa2eca24 73d8fbd1d6
commit 495da3c4f2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
blender/arm/nodes_logic.py
View file

@ -11,6 +11,7 @@ import nodeitems_utils
from arm.logicnode import arm_nodes
from arm.logicnode.arm_nodes import ArmNodeCategory
from arm.logicnode import arm_sockets
# Do not remove this line as it runs all node modules for registering!
from arm.logicnode import *
registered_nodes = []
@ -80,6 +81,10 @@ class ARM_OT_AddNodeOverride(bpy.types.Operator):
return nodetype.__doc__.split('.')[0]
@classmethod
def poll(cls, context):
return context.space_data.tree_type == 'ArmLogicTreeType' and context.space_data.edit_tree
def get_category_draw_func(category: ArmNodeCategory):
def draw_category_menu(self, context):
@ -287,7 +292,7 @@ def replace(tree: bpy.types.NodeTree, node: bpy.types.Node):
# the node can either return a NodeReplacement object (for simple replacements)
# or a brand new node, for more complex stuff.
response = node.get_replacement_node(tree)
if isinstance(response, bpy.types.Node):
newnode = response
# some misc. properties