from lnx.logicnode.lnx_nodes import *

class BoneFKNode(LnxLogicTreeNode):
    """Applies forward kinematics in the given object bone."""
    bl_idname = 'LNBoneFKNode'
    bl_label = 'Bone FK'
    lnx_version = 2
    lnx_section = 'armature'

    def lnx_init(self, context):
        self.add_input('LnxNodeSocketObject', 'Object')
        self.add_input('LnxNodeSocketAnimTree', 'Action Tree')
        self.add_input('LnxStringSocket', 'Bone')
        self.add_input('LnxDynamicSocket', 'Transform')
        self.add_output('LnxNodeSocketAnimTree', 'Action Tree')

    def get_replacement_node(self, node_tree: bpy.types.NodeTree):
        if self.lnx_version not in (0, 1):
            raise LookupError()

        return NodeReplacement(
            'LNBoneFKNode', self.lnx_version, 'LNBoneFKNode', 2,
            in_socket_mapping={}, out_socket_mapping={}
        )