253 lines
9.3 KiB
Python
253 lines
9.3 KiB
Python
import bpy.types
|
|
from bpy.props import *
|
|
from nodeitems_utils import NodeItem
|
|
import arm.utils
|
|
|
|
nodes = []
|
|
category_items = {}
|
|
|
|
object_sockets = dict()
|
|
array_nodes = dict()
|
|
|
|
class ArmLogicTreeNode:
|
|
@classmethod
|
|
def poll(cls, ntree):
|
|
return ntree.bl_idname == 'ArmLogicTreeType'
|
|
|
|
class ArmActionSocket(bpy.types.NodeSocket):
|
|
bl_idname = 'ArmNodeSocketAction'
|
|
bl_label = 'Action Socket'
|
|
|
|
def draw(self, context, layout, node, text):
|
|
layout.label(text=self.name)
|
|
|
|
def draw_color(self, context, node):
|
|
return (0.8, 0.3, 0.3, 1)
|
|
|
|
class ArmArraySocket(bpy.types.NodeSocket):
|
|
bl_idname = 'ArmNodeSocketArray'
|
|
bl_label = 'Array Socket'
|
|
|
|
def draw(self, context, layout, node, text):
|
|
layout.label(text=self.name)
|
|
|
|
def draw_color(self, context, node):
|
|
return (0.8, 0.4, 0.0, 1)
|
|
|
|
class ArmObjectSocket(bpy.types.NodeSocket):
|
|
bl_idname = 'ArmNodeSocketObject'
|
|
bl_label = 'Object Socket'
|
|
default_value_get: PointerProperty(name='Object', type=bpy.types.Object)
|
|
|
|
def get_default_value(self):
|
|
if self.default_value_get == None:
|
|
return ''
|
|
if self.default_value_get.name not in bpy.data.objects:
|
|
return self.default_value_get.name
|
|
return arm.utils.asset_name(bpy.data.objects[self.default_value_get.name])
|
|
|
|
def __init__(self):
|
|
global object_sockets
|
|
# Buckle up..
|
|
# Match id strings to socket dict to retrieve socket in eyedropper operator
|
|
object_sockets[str(id(self))] = self
|
|
|
|
def draw(self, context, layout, node, text):
|
|
if self.is_output:
|
|
layout.label(text=self.name)
|
|
elif self.is_linked:
|
|
layout.label(text=self.name)
|
|
else:
|
|
row = layout.row(align=True)
|
|
row.prop_search(self, 'default_value_get', bpy.context.scene, 'objects', icon='NONE', text=self.name)
|
|
op = row.operator('arm.node_eyedrop', text='', icon='EYEDROPPER', emboss=True)
|
|
op.socket_index = str(id(self))
|
|
|
|
def draw_color(self, context, node):
|
|
return (0.15, 0.55, 0.75, 1)
|
|
|
|
class ArmNodeEyedropButton(bpy.types.Operator):
|
|
'''Pick selected object'''
|
|
bl_idname = 'arm.node_eyedrop'
|
|
bl_label = 'Eyedrop'
|
|
socket_index: StringProperty(name='Socket Index', default='')
|
|
|
|
def execute(self, context):
|
|
global object_sockets
|
|
obj = bpy.context.active_object
|
|
if obj != None:
|
|
object_sockets[self.socket_index].default_value_get = obj
|
|
return{'FINISHED'}
|
|
|
|
class ArmAnimActionSocket(bpy.types.NodeSocket):
|
|
bl_idname = 'ArmNodeSocketAnimAction'
|
|
bl_label = 'Action Socket'
|
|
default_value_get: PointerProperty(name='Action', type=bpy.types.Action)
|
|
|
|
def get_default_value(self):
|
|
if self.default_value_get == None:
|
|
return ''
|
|
if self.default_value_get.name not in bpy.data.actions:
|
|
return self.default_value_get.name
|
|
name = arm.utils.asset_name(bpy.data.actions[self.default_value_get.name])
|
|
return arm.utils.safestr(name)
|
|
|
|
def draw(self, context, layout, node, text):
|
|
if self.is_output:
|
|
layout.label(text=self.name)
|
|
elif self.is_linked:
|
|
layout.label(text=self.name)
|
|
else:
|
|
layout.prop_search(self, 'default_value_get', bpy.data, 'actions', icon='NONE', text='')
|
|
|
|
def draw_color(self, context, node):
|
|
return (0.8, 0.8, 0.8, 1)
|
|
|
|
class ArmNodeAddInputButton(bpy.types.Operator):
|
|
'''Add new input'''
|
|
bl_idname = 'arm.node_add_input'
|
|
bl_label = 'Add Input'
|
|
node_index: StringProperty(name='Node Index', default='')
|
|
socket_type: StringProperty(name='Socket Type', default='NodeSocketShader')
|
|
name_format: StringProperty(name='Name Format', default='Input {0}')
|
|
index_name_offset: IntProperty(name='Index Name Offset', default=0)
|
|
|
|
def execute(self, context):
|
|
global array_nodes
|
|
inps = array_nodes[self.node_index].inputs
|
|
inps.new(self.socket_type, self.name_format.format(str(len(inps) + self.index_name_offset)))
|
|
return{'FINISHED'}
|
|
|
|
class ArmNodeAddInputValueButton(bpy.types.Operator):
|
|
'''Add new input'''
|
|
bl_idname = 'arm.node_add_input_value'
|
|
bl_label = 'Add Input'
|
|
node_index: StringProperty(name='Node Index', default='')
|
|
socket_type: StringProperty(name='Socket Type', default='NodeSocketShader')
|
|
|
|
def execute(self, context):
|
|
global array_nodes
|
|
inps = array_nodes[self.node_index].inputs
|
|
inps.new(self.socket_type, 'Value')
|
|
return{'FINISHED'}
|
|
|
|
class ArmNodeRemoveInputButton(bpy.types.Operator):
|
|
'''Remove last input'''
|
|
bl_idname = 'arm.node_remove_input'
|
|
bl_label = 'Remove Input'
|
|
node_index: StringProperty(name='Node Index', default='')
|
|
|
|
def execute(self, context):
|
|
global array_nodes
|
|
node = array_nodes[self.node_index]
|
|
inps = node.inputs
|
|
min_inps = 0 if not hasattr(node, 'min_inputs') else node.min_inputs
|
|
if len(inps) > min_inps:
|
|
inps.remove(inps.values()[-1])
|
|
return{'FINISHED'}
|
|
|
|
class ArmNodeRemoveInputValueButton(bpy.types.Operator):
|
|
'''Remove last input'''
|
|
bl_idname = 'arm.node_remove_input_value'
|
|
bl_label = 'Remove Input'
|
|
node_index: StringProperty(name='Node Index', default='')
|
|
|
|
def execute(self, context):
|
|
global array_nodes
|
|
node = array_nodes[self.node_index]
|
|
inps = node.inputs
|
|
min_inps = 0 if not hasattr(node, 'min_inputs') else node.min_inputs
|
|
if len(inps) > min_inps and inps[-1].name == 'Value':
|
|
inps.remove(inps.values()[-1])
|
|
return{'FINISHED'}
|
|
|
|
class ArmNodeAddOutputButton(bpy.types.Operator):
|
|
'''Add new output'''
|
|
bl_idname = 'arm.node_add_output'
|
|
bl_label = 'Add Output'
|
|
node_index: StringProperty(name='Node Index', default='')
|
|
socket_type: StringProperty(name='Socket Type', default='NodeSocketShader')
|
|
name_format: StringProperty(name='Name Format', default='Output {0}')
|
|
index_name_offset: IntProperty(name='Index Name Offset', default=0)
|
|
|
|
def execute(self, context):
|
|
global array_nodes
|
|
outs = array_nodes[self.node_index].outputs
|
|
outs.new(self.socket_type, self.name_format.format(str(len(outs) + self.index_name_offset)))
|
|
return{'FINISHED'}
|
|
|
|
class ArmNodeRemoveOutputButton(bpy.types.Operator):
|
|
'''Remove last output'''
|
|
bl_idname = 'arm.node_remove_output'
|
|
bl_label = 'Remove Output'
|
|
node_index: StringProperty(name='Node Index', default='')
|
|
|
|
def execute(self, context):
|
|
global array_nodes
|
|
node = array_nodes[self.node_index]
|
|
outs = node.outputs
|
|
min_outs = 0 if not hasattr(node, 'min_outputs') else node.min_outputs
|
|
if len(outs) > min_outs:
|
|
outs.remove(outs.values()[-1])
|
|
return{'FINISHED'}
|
|
|
|
class ArmNodeAddInputOutputButton(bpy.types.Operator):
|
|
'''Add new input and output'''
|
|
bl_idname = 'arm.node_add_input_output'
|
|
bl_label = 'Add Input Output'
|
|
node_index: StringProperty(name='Node Index', default='')
|
|
in_socket_type: StringProperty(name='In Socket Type', default='NodeSocketShader')
|
|
out_socket_type: StringProperty(name='Out Socket Type', default='NodeSocketShader')
|
|
in_name_format: StringProperty(name='In Name Format', default='Input {0}')
|
|
out_name_format: StringProperty(name='Out Name Format', default='Output {0}')
|
|
in_index_name_offset: IntProperty(name='Index Name Offset', default=0)
|
|
|
|
def execute(self, context):
|
|
global array_nodes
|
|
node = array_nodes[self.node_index]
|
|
inps = node.inputs
|
|
outs = node.outputs
|
|
inps.new(self.in_socket_type, self.in_name_format.format(str(len(inps) + self.in_index_name_offset)))
|
|
outs.new(self.out_socket_type, self.out_name_format.format(str(len(outs))))
|
|
return{'FINISHED'}
|
|
|
|
class ArmNodeRemoveInputOutputButton(bpy.types.Operator):
|
|
'''Remove last input and output'''
|
|
bl_idname = 'arm.node_remove_input_output'
|
|
bl_label = 'Remove Input Output'
|
|
node_index: StringProperty(name='Node Index', default='')
|
|
|
|
def execute(self, context):
|
|
global array_nodes
|
|
node = array_nodes[self.node_index]
|
|
inps = node.inputs
|
|
outs = node.outputs
|
|
min_inps = 0 if not hasattr(node, 'min_inputs') else node.min_inputs
|
|
min_outs = 0 if not hasattr(node, 'min_outputs') else node.min_outputs
|
|
if len(inps) > min_inps:
|
|
inps.remove(inps.values()[-1])
|
|
if len(outs) > min_outs:
|
|
outs.remove(outs.values()[-1])
|
|
return{'FINISHED'}
|
|
|
|
def add_node(node_class, category):
|
|
global nodes
|
|
nodes.append(node_class)
|
|
if category_items.get(category) == None:
|
|
category_items[category] = []
|
|
category_items[category].append(NodeItem(node_class.bl_idname))
|
|
|
|
bpy.utils.register_class(ArmActionSocket)
|
|
bpy.utils.register_class(ArmArraySocket)
|
|
bpy.utils.register_class(ArmObjectSocket)
|
|
bpy.utils.register_class(ArmNodeEyedropButton)
|
|
bpy.utils.register_class(ArmAnimActionSocket)
|
|
bpy.utils.register_class(ArmNodeAddInputButton)
|
|
bpy.utils.register_class(ArmNodeAddInputValueButton)
|
|
bpy.utils.register_class(ArmNodeRemoveInputButton)
|
|
bpy.utils.register_class(ArmNodeRemoveInputValueButton)
|
|
bpy.utils.register_class(ArmNodeAddOutputButton)
|
|
bpy.utils.register_class(ArmNodeRemoveOutputButton)
|
|
bpy.utils.register_class(ArmNodeAddInputOutputButton)
|
|
bpy.utils.register_class(ArmNodeRemoveInputOutputButton)
|