1d0a6d7955
Also fixed a ton of bugs in the python part. NOTE: this requires a yet-to-be-done commit to Iron to work.
47 lines
1.8 KiB
Python
47 lines
1.8 KiB
Python
from arm.logicnode.arm_nodes import *
|
|
|
|
class TransformNode(ArmLogicTreeNode):
|
|
"""Stores the location, rotation and scale values as a transform."""
|
|
bl_idname = 'LNTransformNode'
|
|
bl_label = 'Transform'
|
|
arm_version = 2
|
|
|
|
def init(self, context):
|
|
super(TransformNode, self).init(context)
|
|
self.add_input('NodeSocketVector', 'Location')
|
|
self.add_input('ArmNodeSocketRotation', 'Rotation')
|
|
self.add_input('NodeSocketVector', 'Scale', default_value=[1.0, 1.0, 1.0])
|
|
|
|
self.add_output('NodeSocketShader', 'Transform', is_var=True)
|
|
|
|
|
|
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: make rotations their own sockets
|
|
# this transition is a mess, I know.
|
|
newself = self.id_data.nodes.new('LNTransformNode')
|
|
ret = [newself]
|
|
|
|
for link in self.inputs[0].links:
|
|
self.id_data.links.new(link.from_socket, newself.inputs[0])
|
|
for link in self.inputs[2].links:
|
|
self.id_data.links.new(link.from_socket, newself.inputs[2])
|
|
for link in self.outputs[0].links:
|
|
self.id_data.links.new(newself.outputs[0], link.to_socket)
|
|
|
|
links_rot = self.inputs[1].links
|
|
if len(links_rot) > 0:
|
|
converter = self.id_data.nodes.new('LNRotationNode')
|
|
self.id_data.links.new(converter.outputs[0], newself.inputs[1])
|
|
converter.property0 = 'EulerAngles'
|
|
converter.property1 = 'Rad'
|
|
converter.property2 = 'XZY'
|
|
ret.append(converter)
|
|
for link in links_rot:
|
|
self.id_data.links.new(link.from_socket, converter.inputs[0])
|
|
|
|
return ret
|