update similar nodes

This commit is contained in:
QuantumCoderQC 2021-06-09 19:09:02 +02:00
parent 1eb4339d74
commit 6c3a4d12a6
3 changed files with 43 additions and 5 deletions

View file

@ -1,17 +1,36 @@
from arm.logicnode.arm_nodes import *
class SetMaterialImageParamNode(ArmLogicTreeNode):
"""TO DO."""
"""Set a image or texture value material parameter to the specified object. If `per object` is disabled, value will be set to all objects with this material"""
bl_idname = 'LNSetMaterialImageParamNode'
bl_label = 'Set Material Image Param'
arm_section = 'params'
arm_version = 1
arm_version = 2
property0: BoolProperty(
name="Per Object",
description="Set property per object",
default=False
)
def init(self, context):
super(SetMaterialImageParamNode, self).init(context)
self.add_input('ArmNodeSocketAction', 'In')
self.add_input('ArmNodeSocketObject', 'Object')
self.add_input('NodeSocketShader', 'Material')
self.add_input('NodeSocketString', 'Node')
self.add_input('NodeSocketString', 'Image')
self.add_output('ArmNodeSocketAction', 'Out')
def draw_buttons(self, context, layout):
layout.prop(self, 'property0')
def get_replacement_node(self, node_tree: bpy.types.NodeTree):
if self.arm_version not in (0, 1):
raise LookupError()
return NodeReplacement(
'LNSetMaterialImageParamNode', self.arm_version, 'LNSetMaterialImageParamNode', 2,
in_socket_mapping={0:0, 1:2, 2:3, 3:4}, out_socket_mapping={0:0}
)

View file

@ -1,17 +1,36 @@
from arm.logicnode.arm_nodes import *
class SetMaterialRgbParamNode(ArmLogicTreeNode):
"""TO DO."""
"""Set a color or vector value material parameter to the specified object. If `per object` is disabled, value will be set to all objects with this material"""
bl_idname = 'LNSetMaterialRgbParamNode'
bl_label = 'Set Material RGB Param'
arm_section = 'params'
arm_version = 1
arm_version = 2
property0: BoolProperty(
name="Per Object",
description="Set property per object",
default=False
)
def init(self, context):
super(SetMaterialRgbParamNode, self).init(context)
self.add_input('ArmNodeSocketAction', 'In')
self.add_input('ArmNodeSocketObject', 'Object')
self.add_input('NodeSocketShader', 'Material')
self.add_input('NodeSocketString', 'Node')
self.add_input('NodeSocketColor', 'Color')
self.add_output('ArmNodeSocketAction', 'Out')
def draw_buttons(self, context, layout):
layout.prop(self, 'property0')
def get_replacement_node(self, node_tree: bpy.types.NodeTree):
if self.arm_version not in (0, 1):
raise LookupError()
return NodeReplacement(
'LNSetMaterialRgbParamNode', self.arm_version, 'LNSetMaterialRgbParamNode', 2,
in_socket_mapping={0:0, 1:2, 2:3, 3:4}, out_socket_mapping={0:0}
)

View file

@ -1,7 +1,7 @@
from arm.logicnode.arm_nodes import *
class SetMaterialValueParamNode(ArmLogicTreeNode):
"""Set a float value material parameter to the specified object"""
"""Set a float value material parameter to the specified object. If `per object` is disabled, value will be set to all objects with this material"""
bl_idname = 'LNSetMaterialValueParamNode'
bl_label = 'Set Material Value Param'
arm_section = 'params'