25 lines
799 B
Python
25 lines
799 B
Python
|
from bpy.types import Node
|
||
|
|
from arm.logicnode.arm_nodes import *
|
||
|
|
import arm.nodes_logic
|
||
|
|
|
||
|
|
class EnhancedTransformNode(Node, ArmLogicTreeNode):
|
||
|
|
'''EnhancedTransformNode'''
|
||
|
|
bl_idname = 'LNEnhancedTransformNode'
|
||
|
|
bl_label = 'Enhanced Transform'
|
||
|
|
bl_icon = 'GAME'
|
||
|
|
|
||
|
|
def init(self, context):
|
||
|
|
self.inputs.new('ArmNodeSocketAction', 'In')
|
||
|
|
self.inputs.new('ArmNodeSocketObject', 'Object')
|
||
|
|
self.inputs.new('NodeSocketVector', 'Location')
|
||
|
|
self.inputs.new('NodeSocketVector', 'Rotation')
|
||
|
|
self.inputs.new('NodeSocketVector', 'Scale')
|
||
|
|
self.outputs.new('ArmNodeSocketAction', 'Out')
|
||
|
|
|
||
|
|
def register():
|
||
|
|
# Add custom nodes
|
||
|
|
add_node(EnhancedTransformNode, category='Action')
|
||
|
|
|
||
|
|
# Register newly added nodes
|
||
|
|
arm.nodes_logic.register_nodes()
|