2017-04-08 20:05:35 +02:00
|
|
|
from arm.logicnode.arm_nodes import *
|
|
|
|
|
2020-09-08 21:16:20 +02:00
|
|
|
class VectorFromTransformNode(ArmLogicTreeNode):
|
2020-10-05 19:55:56 +02:00
|
|
|
"""Returns vector from the given transform."""
|
2017-04-08 20:05:35 +02:00
|
|
|
bl_idname = 'LNVectorFromTransformNode'
|
|
|
|
bl_label = 'Vector From Transform'
|
2020-09-13 11:27:58 +02:00
|
|
|
arm_version = 1
|
2017-04-08 20:05:35 +02:00
|
|
|
|
|
|
|
def init(self, context):
|
2020-09-13 11:27:58 +02:00
|
|
|
super(VectorFromTransformNode, self).init(context)
|
2020-09-08 21:49:02 +02:00
|
|
|
self.add_input('NodeSocketShader', 'Transform')
|
|
|
|
self.add_output('NodeSocketVector', 'Vector')
|
|
|
|
self.add_output('NodeSocketVector', 'Quaternion XYZ')
|
|
|
|
self.add_output('NodeSocketFloat', 'Quaternion W')
|
2020-09-08 17:03:24 +02:00
|
|
|
|
2020-08-30 11:45:07 +02:00
|
|
|
def on_property_update(self, context):
|
|
|
|
"""called by the EnumProperty, used to update the node socket labels"""
|
|
|
|
# note: the conditions on len(self.outputs) are take in account "old version" (pre-2020.9) nodes, which only have one output
|
|
|
|
if self.property0 == "Quaternion":
|
|
|
|
self.outputs[0].name = "Quaternion"
|
|
|
|
if len(self.outputs) > 1:
|
|
|
|
self.outputs[1].name = "Quaternion XYZ"
|
|
|
|
self.outputs[2].name = "Quaternion W"
|
|
|
|
else:
|
|
|
|
self.outputs[0].name = "Vector"
|
|
|
|
if len(self.outputs) > 1:
|
|
|
|
self.outputs[1].name = "[quaternion only]"
|
|
|
|
self.outputs[2].name = "[quaternion only]"
|
2017-04-08 20:05:35 +02:00
|
|
|
|
|
|
|
def draw_buttons(self, context, layout):
|
|
|
|
layout.prop(self, 'property0')
|
2020-09-08 17:03:24 +02:00
|
|
|
|
2020-08-30 11:45:07 +02:00
|
|
|
property0: EnumProperty(
|
|
|
|
items = [('Up', 'Up', 'Up'),
|
|
|
|
('Right', 'Right', 'Right'),
|
|
|
|
('Look', 'Look', 'Look'),
|
|
|
|
('Quaternion', 'Quaternion', 'Quaternion')],
|
|
|
|
name='', default='Look',
|
|
|
|
update=on_property_update)
|
2017-04-08 20:05:35 +02:00
|
|
|
|
2020-09-09 20:48:35 +02:00
|
|
|
add_node(VectorFromTransformNode, category=PKG_AS_CATEGORY)
|