armory/blender/arm/logicnode/math/LN_mix_vector.py
2020-11-17 12:01:15 -03:00

50 lines
1.7 KiB
Python

from arm.logicnode.arm_nodes import *
class VectorMixNode(ArmLogicTreeNode):
"""Interpolates between the two given vectors."""
bl_idname = 'LNVectorMixNode'
bl_label = 'Mix Vector'
arm_section = 'vector'
arm_version = 1
property0: EnumProperty(
items = [('Linear', 'Linear', 'Linear'),
('Sine', 'Sine', 'Sine'),
('Quad', 'Quad', 'Quad'),
('Cubic', 'Cubic', 'Cubic'),
('Quart', 'Quart', 'Quart'),
('Quint', 'Quint', 'Quint'),
('Expo', 'Expo', 'Expo'),
('Circ', 'Circ', 'Circ'),
('Back', 'Back', 'Back'),
('Bounce', 'Bounce', 'Bounce'),
('Elastic', 'Elastic', 'Elastic'),
],
name='', default='Linear')
property1: EnumProperty(
items = [('In', 'In', 'In'),
('Out', 'Out', 'Out'),
('InOut', 'InOut', 'InOut'),
],
name='', default='Out')
@property
def property2(self):
return 'true' if self.property2_ else 'false'
property2_: BoolProperty(name='Clamp', default=False)
def init(self, context):
super(VectorMixNode, self).init(context)
self.add_input('NodeSocketFloat', 'Factor', default_value=0.0)
self.add_input('NodeSocketVector', 'Vector 1', default_value=[0.0, 0.0, 0.0])
self.add_input('NodeSocketVector', 'Vector 2', default_value=[1.0, 1.0, 1.0])
self.add_output('NodeSocketVector', 'Result')
def draw_buttons(self, context, layout):
layout.prop(self, 'property2_')
layout.prop(self, 'property0')
if self.property0 != 'Linear':
layout.prop(self, 'property1')