43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
from arm.logicnode.arm_nodes import *
|
|
|
|
class SetMaterialRgbParamNode(ArmLogicTreeNode):
|
|
"""Set a color or vector value material parameter to the specified object.
|
|
|
|
@seeNode Get Scene Root
|
|
|
|
@input Object: Object whose material parameter should change. Use `Get Scene Root` node to set parameter globally.
|
|
|
|
@input Per Object:
|
|
- `Enabled`: Set material parameter specific to this object. Global parameter will be ignored.
|
|
- `Disabled`: Set parameter globally, including this object.
|
|
|
|
@input Material: Material whose parameter to be set.
|
|
|
|
@input Node: Name of the parameter.
|
|
|
|
@input Color: Color or vector input.
|
|
"""
|
|
bl_idname = 'LNSetMaterialRgbParamNode'
|
|
bl_label = 'Set Material RGB Param'
|
|
arm_section = 'params'
|
|
arm_version = 2
|
|
|
|
def arm_init(self, context):
|
|
self.add_input('ArmNodeSocketAction', 'In')
|
|
self.add_input('ArmNodeSocketObject', 'Object')
|
|
self.add_input('ArmBoolSocket', 'Per Object')
|
|
self.add_input('ArmDynamicSocket', 'Material')
|
|
self.add_input('ArmStringSocket', 'Node')
|
|
self.add_input('ArmColorSocket', 'Color')
|
|
|
|
self.add_output('ArmNodeSocketAction', 'Out')
|
|
|
|
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:3, 2:4, 3:5}, out_socket_mapping={0:0}
|
|
)
|