77 lines
3.2 KiB
Python
77 lines
3.2 KiB
Python
from arm.logicnode.arm_nodes import *
|
|
from mathutils import Vector
|
|
|
|
@deprecated(message='Do not use quaternion sockets')
|
|
class QuaternionNode(ArmLogicTreeNode):
|
|
"""TO DO."""
|
|
bl_idname = 'LNQuaternionNode'
|
|
bl_label = 'Quaternion'
|
|
bl_label = 'Create a quaternion variable (transported through a vector socket)'
|
|
arm_section = 'quaternions'
|
|
arm_version = 2 # deprecate
|
|
|
|
def init(self, context):
|
|
super(QuaternionNode, self).init(context)
|
|
self.add_input('NodeSocketFloat', 'X')
|
|
self.add_input('NodeSocketFloat', 'Y')
|
|
self.add_input('NodeSocketFloat', 'Z')
|
|
self.add_input('NodeSocketFloat', 'W', default_value=1.0)
|
|
|
|
self.add_output('NodeSocketVector', 'Quaternion')
|
|
self.add_output('NodeSocketVector', 'XYZ')
|
|
self.add_output('NodeSocketFloat', 'W')
|
|
|
|
|
|
def get_replacement_node(self, node_tree: bpy.types.NodeTree):
|
|
if self.arm_version not in (0, 1):
|
|
raise LookupError()
|
|
|
|
# transition from version 1 to version 2[deprecated]
|
|
|
|
|
|
newnodes = []
|
|
|
|
rawlinks = self.outputs[0].links
|
|
xyzlinks = self.outputs[1].links
|
|
wlinks = self.outputs[2].links
|
|
if len(rawlinks)>0 or len(xyzlinks)>0:
|
|
xyzcomb = node_tree.nodes.new('LNVectorNode')
|
|
newnodes.append(xyzcomb)
|
|
|
|
xyzcomb.inputs[0].default_value = self.inputs[0].default_value
|
|
xyzcomb.inputs[1].default_value = self.inputs[1].default_value
|
|
xyzcomb.inputs[2].default_value = self.inputs[2].default_value
|
|
for link in self.inputs[0].links:
|
|
node_tree.links.new(link.from_socket, xyzcomb.inputs[0])
|
|
for link in self.inputs[1].links:
|
|
node_tree.links.new(link.from_socket, xyzcomb.inputs[1])
|
|
for link in self.inputs[2].links:
|
|
node_tree.links.new(link.from_socket, xyzcomb.inputs[2])
|
|
|
|
for link in xyzlinks:
|
|
node_tree.links.new(xyzcomb.outputs[0], link.to_socket)
|
|
if len(rawlinks)>0:
|
|
rotnode = node_tree.nodes.new('LNRotationNode')
|
|
newnodes.append(rotnode)
|
|
rotnode.property0 = 'Quaternion'
|
|
rotnode.inputs[0].default_value = Vector(
|
|
(self.inputs[0].default_value,
|
|
self.inputs[1].default_value,
|
|
self.inputs[2].default_value))
|
|
rotnode.inputs[1].default_value = self.inputs[3].default_value
|
|
node_tree.links.new(xyzcomb.outputs[0], rotnode.inputs[0])
|
|
for link in self.inputs[3].links: # 0 or 1
|
|
node_tree.links.new(link.from_socket, rotnode.inputs[1])
|
|
for link in rawlinks:
|
|
node_tree.links.new(rotnode.outputs[0], link.to_socket)
|
|
|
|
if len(self.inputs[3].links)>0:
|
|
fromval = self.inputs[3].links[0].from_socket
|
|
for link in self.outputs[2].links:
|
|
node_tree.links.new(fromval, link.to_socket)
|
|
else:
|
|
for link in self.outputs[2].links:
|
|
link.to_socket.default_value = self.inputs[3].default_value
|
|
|
|
return newnodes
|