mantis/link_definitions.py

import bpy
from bpy.types import NodeTree, Node, NodeSocket
from .base_definitions import MantisNode, LinkNode, GraphError
from .utilities import (prRed, prGreen, prPurple, prWhite,
                              prOrange,
                              wrapRed, wrapGreen, wrapPurple, wrapWhite,
                              wrapOrange,)

from .base_definitions import get_signature_from_edited_tree

def TellClasses():
    return [ LinkInheritNode,
             LinkInverseKinematics,
             LinkCopyLocationNode,
             LinkCopyRotationNode,
             LinkCopyScaleNode,
             LinkInheritConstraintNode,
             LinkCopyTransformNode,
             LinkStretchToNode,
             LinkDampedTrackNode,
             LinkLockedTrackNode,
             LinkTrackToNode,
             LinkLimitLocationNode,
             LinkLimitScaleNode,
             LinkLimitRotationNode,
             LinkLimitDistanceNode,
             LinkDrivenParameterNode,
             LinkArmatureNode,
             LinkSplineIKNode,
             LinkTransformationNode,
           ]

def default_traverse(self, socket):
        if (socket == self.outputs["Output Relationship"]):
            return self.inputs["Input Relationship"]
        if (socket == self.inputs["Input Relationship"]):
            return self.outputs["Output Relationship"]
        return None

class LinkInheritNode(Node, LinkNode):
    '''A node representing inheritance'''
    # cuss, messed this up
    bl_idname = 'linkInherit' # l should be L
    # need to fix this
    bl_label = "Inherit"
    bl_icon = 'CONSTRAINT_BONE'
    initialized : bpy.props.BoolProperty(default = False)
    useTarget : bpy.props.BoolProperty(default=False)
    
    # bone_prev : bpy.props.BoolProperty(default=False)
    # bone_next : bpy.props.BoolProperty(default=False)

    def init(self, context):
        r = self.inputs.new('BooleanSocket', "Inherit Rotation")
        s = self.inputs.new('EnumInheritScale', "Inherit Scale")
        c = self.inputs.new('BooleanSocket', "Connected")
        i = self.outputs.new('RelationshipSocket', "Inheritance")
        p = self.inputs.new('xFormSocket', "Parent")
        # set default values...
        self.initialized = True

    def traverse(self, socket):
        if (socket == self.outputs["Inheritance"]):
            return self.inputs["Parent"]
        if (socket == self.inputs["Parent"]):
            return self.outputs["Inheritance"]
        return None
    
    def display_update(self, parsed_tree, context):
        node_tree = context.space_data.path[0].node_tree
        nc = parsed_tree.get(get_signature_from_edited_tree(self, context))
        if nc:
            bone_prev, bone_next = False, False
            if (inp := nc.inputs["Parent"]).is_connected:
                if  from_node := inp.links[0].from_node:
                    if from_node.__class__.__name__ in ["xFormBone"]:
                        bone_prev=True
            bone_next=True
            try:
                xForm = nc.GetxForm()
                if xForm.__class__.__name__ not in "xFormBone":
                    
                    bone_next=False
            except GraphError:

                bone_next=False
            # print(bone_prev, bone_next )
            if bone_next and bone_prev:
                self.inputs["Inherit Rotation"].hide = False
                self.inputs["Inherit Scale"].hide    = False
                self.inputs["Connected"].hide       = False
            else:
                self.inputs["Inherit Rotation"].hide = True
                self.inputs["Inherit Scale"].hide    = True
                self.inputs["Connected"].hide        = True
            # the node_groups on the way here ought to be active if there
            #  is no funny business going on.
    

# DO: make another node for ITASC IK, eh?
class LinkInverseKinematics(Node, LinkNode):
    '''A node representing inverse kinematics'''
    bl_idname = 'LinkInverseKinematics'
    bl_label = "Inverse Kinematics"
    bl_icon = 'CON_KINEMATIC'
    initialized : bpy.props.BoolProperty(default = False)
    useTarget : bpy.props.BoolProperty(default=True)

    def init(self, context):
        self.inputs.new('RelationshipSocket', "Input Relationship")
        self.inputs.new ('xFormSocket', "Target")
        self.inputs.new ('xFormSocket', "Pole Target")
        self.inputs.new ('IKChainLengthSocket', "Chain Length")
        self.inputs.new ('BooleanSocket', "Use Tail")
        self.inputs.new ('BooleanSocket', "Stretch")
        self.inputs.new ('FloatFactorSocket', "Position")
        self.inputs.new ('FloatFactorSocket', "Rotation")
        self.inputs.new ('FloatFactorSocket', "Influence")
        self.inputs.new ('EnableSocket', "Enable")
        
        #Well, it turns out that this has to be a constraint like
        # everything else, because of course, there can be more than one.
        
        #self.outputs.new('RelationshipSocket', "Inheritance")
        self.outputs.new('RelationshipSocket', "Output Relationship")

        self.initialized = True


        
class LinkSplineIK(Node, LinkNode):
    '''A node representing Spline IK'''
    bl_idname = 'LinkSplineIK'
    bl_label = "LinkSplineIK"
    bl_icon = 'CON_SPLINEIK'
    useTarget : bpy.props.BoolProperty(default=True)
    def init(self, context):
        self.inputs.new('xFormSocket', "Parent")
        self.inputs.new ('xFormSocket', "Target")
        self.inputs.new ('xFormSocket', "Pole Target")
        self.inputs.new ('BooleanSocket', "Use Tail")
        self.inputs.new ('BooleanSocket', "Stretch")
        self.inputs.new ('FloatFactorSocket', "Position")
        self.inputs.new ('FloatFactorSocket', "Rotation")
        self.inputs.new ('FloatFactorSocket', "Influence")
        self.inputs.new ('EnableSocket', "Enable")
        self.outputs.new('RelationshipSocket', "Inheritance")

    def traverse(self, socket):
        if (socket == self.outputs["Inheritance"]):
            return self.inputs["Parent"]
        if (socket == self.inputs["Parent"]):
            return self.outputs["Inheritance"]
        return None

class LinkCopyLocationNode(Node, LinkNode):
    '''A node representing Copy Location'''
    bl_idname = 'LinkCopyLocation'
    bl_label = "Copy Location"
    bl_icon = 'CON_LOCLIKE'
    useTarget : bpy.props.BoolProperty(default=True)

    def init(self, context):
        self.inputs.new ('RelationshipSocket', "Input Relationship")
        self.inputs.new ('FloatFactorSocket', "Head/Tail")
        self.inputs.new ('BooleanSocket', "UseBBone")
        self.inputs.new ('BooleanThreeTupleSocket', "Axes")
        self.inputs.new ('BooleanThreeTupleSocket', "Invert")
        self.inputs.new ('TransformSpaceSocket', "Target Space")
        self.inputs.new ('TransformSpaceSocket', "Owner Space")
        self.inputs.new ('FloatFactorSocket', "Influence")
        self.inputs.new ('xFormSocket', "Target")
        self.inputs.new ('EnableSocket', "Enable")
        #
        self.outputs.new('RelationshipSocket', "Output Relationship")


        
    
    # def display_update(self, parsed_tree, context):
        # node_tree = context.space_data.path[0].node_tree
        # nc = parsed_tree.get(get_signature_from_edited_tree(self, context))
        # if nc:
            # bone_prev, bone_next = False, False
            # if (inp := nc.inputs["Input Relationship"]).is_connected:
                # if  from_node := inp.links[0].from_node:
                    # if from_node.__class__.__name__ in ["xFormBone"]:
                        # bone_prev=True
            # bone_next=True
            # try:
                # xForm = nc.GetxForm()
                # if xForm.__class__.__name__ not in "xFormBone":
                    # bone_next=False
            # except GraphError:
                # bone_next=False
            # if bone_next and bone_prev:
                # # this is where we will set whether you can pick
                # #  the other spaces or not.
        
class LinkCopyRotationNode(Node, LinkNode):
    '''A node representing Copy Rotation'''
    bl_idname = 'LinkCopyRotation'
    bl_label = "Copy Rotation"
    bl_icon = 'CON_ROTLIKE'
    useTarget : bpy.props.BoolProperty(default=True)

    def init(self, context):
        self.inputs.new ('RelationshipSocket', "Input Relationship")
        self.inputs.new ('RotationOrderSocket', "RotationOrder")
        self.inputs.new ('EnumRotationMix', "Rotation Mix")
        self.inputs.new ('BooleanThreeTupleSocket', "Axes")
        self.inputs.new ('BooleanThreeTupleSocket', "Invert")
        self.inputs.new ('TransformSpaceSocket', "Target Space")
        self.inputs.new ('TransformSpaceSocket', "Owner Space")
        self.inputs.new ('FloatFactorSocket', "Influence")
        self.inputs.new ('xFormSocket', "Target")
        self.inputs.new ('EnableSocket', "Enable")
        #
        self.outputs.new('RelationshipSocket', "Output Relationship")


        
class LinkCopyScaleNode(Node, LinkNode):
    '''A node representing Copy Scale'''
    bl_idname = 'LinkCopyScale'
    bl_label = "Copy Scale"
    bl_icon = 'CON_SIZELIKE'
    useTarget : bpy.props.BoolProperty(default=True)

    def init(self, context):
        self.inputs.new ('RelationshipSocket', "Input Relationship")
        self.inputs.new ('BooleanSocket', "Offset")
        self.inputs.new ('BooleanSocket', "Average")
        self.inputs.new ('BooleanThreeTupleSocket', "Axes")
        #self.inputs.new ('BooleanThreeTupleSocket', "Invert")
        # dingus, this one doesn't have inverts
        self.inputs.new ('TransformSpaceSocket', "Target Space")
        self.inputs.new ('TransformSpaceSocket', "Owner Space")
        self.inputs.new ('FloatFactorSocket', "Influence")
        self.inputs.new ('xFormSocket', "Target")
        self.inputs.new ('EnableSocket', "Enable")
        #
        self.outputs.new('RelationshipSocket', "Output Relationship")


        
class LinkInheritConstraintNode(Node, LinkNode):
    # === Basics ===
    # Description string
    '''A node representing a parent constraint'''
    bl_idname = 'LinkInheritConstraint'
    bl_label = "Inherit (constraint)"
    bl_icon = 'CON_CHILDOF'
    useTarget : bpy.props.BoolProperty(default=True)

    # === Optional Functions ===
    def init(self, context):
        self.inputs.new ('RelationshipSocket', "Input Relationship")
        self.inputs.new ('BooleanThreeTupleSocket', "Location")
        self.inputs.new ('BooleanThreeTupleSocket', "Rotation")
        self.inputs.new ('BooleanThreeTupleSocket', "Scale")
        self.inputs.new ('FloatFactorSocket', "Influence")
        self.inputs.new ('xFormSocket', "Target")
        self.inputs.new ('EnableSocket', "Enable")
        #
        self.outputs.new('RelationshipSocket', "Output Relationship")


        
class LinkCopyTransformNode(Node, LinkNode):
    # === Basics ===
    # Description string
    '''A node representing Copy Transform'''
    bl_idname = 'LinkCopyTransforms'
    bl_label = "Copy Transform"
    bl_icon = 'CON_TRANSLIKE'
    useTarget : bpy.props.BoolProperty(default=True)


    # === Optional Functions ===
    def init(self, context):
        self.inputs.new ('RelationshipSocket', "Input Relationship")
        self.inputs.new ('FloatFactorSocket', "Head/Tail")
        self.inputs.new ('BooleanSocket', "UseBBone")
        self.inputs.new ('EnumRotationMixCopyTransforms', "Mix")
        self.inputs.new ('TransformSpaceSocket', "Target Space")
        self.inputs.new ('TransformSpaceSocket', "Owner Space")
        self.inputs.new ('FloatFactorSocket', "Influence")
        self.inputs.new ('xFormSocket', "Target")
        self.inputs.new ('EnableSocket', "Enable")
        #
        self.outputs.new('RelationshipSocket', "Output Relationship")


        
class LinkStretchToNode(Node, LinkNode):
    '''A node representing Stretch-To'''
    bl_idname = 'LinkStretchTo'
    bl_label = "Stretch To"
    bl_icon = 'CON_STRETCHTO'
    useTarget : bpy.props.BoolProperty(default=True)
    initialized : bpy.props.BoolProperty(default = False)
    def init(self, context):
        self.inputs.new ('RelationshipSocket', "Input Relationship")
        self.inputs.new ('FloatFactorSocket', "Head/Tail")
        self.inputs.new ('BooleanSocket', "UseBBone")
        self.inputs.new ('FloatSocket', "Original Length")
        self.inputs.new ('FloatSocket', "Volume Variation")
        self.inputs.new ('BoolUpdateParentNode', "Use Volume Min")
        self.inputs.new ('FloatSocket', "Volume Min")
        self.inputs.new ('BoolUpdateParentNode', "Use Volume Max")
        self.inputs.new ('FloatSocket', "Volume Max")
        self.inputs.new ('FloatFactorSocket', "Smooth")
        self.inputs.new ('EnumMaintainVolumeStretchToSocket', "Maintain Volume")
        self.inputs.new ('EnumRotationStretchTo', "Rotation")
        self.inputs.new ('FloatFactorSocket', "Influence")
        self.inputs.new ('xFormSocket', "Target")
        self.inputs.new ('EnableSocket', "Enable")
        #
        self.outputs.new('RelationshipSocket', "Output Relationship")

        self.initialized = True


        
class LinkDampedTrackNode(Node, LinkNode):
    '''A node representing Stretch-To'''
    bl_idname = 'LinkDampedTrack'
    bl_label = "Damped Track"
    bl_icon = 'CON_TRACKTO'
    useTarget : bpy.props.BoolProperty(default=True)
    initialized : bpy.props.BoolProperty(default = False)
    def init(self, context):
        self.inputs.new ('RelationshipSocket', "Input Relationship")
        self.inputs.new ('FloatFactorSocket', "Head/Tail")
        self.inputs.new ('BooleanSocket', "UseBBone")
        self.inputs.new ('EnumTrackAxis', "Track Axis")
        self.inputs.new ('FloatFactorSocket', "Influence")
        self.inputs.new ('xFormSocket', "Target")
        self.inputs.new ('EnableSocket', "Enable")
        #
        self.outputs.new('RelationshipSocket', "Output Relationship")

        self.initialized = True


        
class LinkLockedTrackNode(Node, LinkNode):
    '''A node representing Stretch-To'''
    bl_idname = 'LinkLockedTrack'
    bl_label = "Locked Track"
    bl_icon = 'CON_LOCKTRACK'
    useTarget : bpy.props.BoolProperty(default=True)
    initialized : bpy.props.BoolProperty(default = False)

    def init(self, context):
        self.inputs.new ('RelationshipSocket', "Input Relationship")
        self.inputs.new ('FloatFactorSocket', "Head/Tail")
        self.inputs.new ('BooleanSocket', "UseBBone")
        self.inputs.new ('EnumTrackAxis', "Track Axis")
        self.inputs.new ('EnumLockAxis', "Lock Axis")
        self.inputs.new ('FloatFactorSocket', "Influence")
        self.inputs.new ('xFormSocket', "Target")
        self.inputs.new ('EnableSocket', "Enable")
        #
        self.outputs.new('RelationshipSocket', "Output Relationship")

        self.initialized = True


        
class LinkTrackToNode(Node, LinkNode):
    '''A node representing Stretch-To'''
    bl_idname = 'LinkTrackTo'
    bl_label = "Track To"
    bl_icon = 'CON_TRACKTO'
    useTarget : bpy.props.BoolProperty(default=True)
    initialized : bpy.props.BoolProperty(default = False)

    def init(self, context):
        self.inputs.new ('RelationshipSocket', "Input Relationship")
        self.inputs.new ('FloatFactorSocket', "Head/Tail")
        self.inputs.new ('BooleanSocket', "UseBBone")
        self.inputs.new ('EnumTrackAxis', "Track Axis")
        self.inputs.new ('EnumUpAxis', "Up Axis")
        self.inputs.new ('BooleanSocket', "Use Target Z")
        self.inputs.new ('TransformSpaceSocket', "Target Space")
        self.inputs.new ('TransformSpaceSocket', "Owner Space")
        self.inputs.new ('FloatFactorSocket', "Influence")
        self.inputs.new ('xFormSocket', "Target")
        self.inputs.new ('EnableSocket', "Enable")
        #
        self.outputs.new('RelationshipSocket', "Output Relationship")

        self.initialized = True


        
class LinkLimitLocationNode(Node, LinkNode):
    '''A node representing Limit Location'''
    bl_idname = 'LinkLimitLocation'
    bl_label = "Limit Location"
    bl_icon = 'CON_LOCLIMIT'
    useTarget : bpy.props.BoolProperty(default=False)
    initialized : bpy.props.BoolProperty(default = False)

    def init(self, context):
        self.inputs.new ('RelationshipSocket', "Input Relationship")
        self.inputs.new ('BoolUpdateParentNode', "Use Max X")
        self.inputs.new ('FloatSocket', "Max X")
        self.inputs.new ('BoolUpdateParentNode', "Use Min X")
        self.inputs.new ('FloatSocket', "Min X")
        self.inputs.new ('BoolUpdateParentNode', "Use Max Y")
        self.inputs.new ('FloatSocket', "Max Y")
        self.inputs.new ('BoolUpdateParentNode', "Use Min Y")
        self.inputs.new ('FloatSocket', "Min Y")
        self.inputs.new ('BoolUpdateParentNode', "Use Max Z")
        self.inputs.new ('FloatSocket', "Max Z")
        self.inputs.new ('BoolUpdateParentNode', "Use Min Z")
        self.inputs.new ('FloatSocket', "Min Z")
        self.inputs.new ('BooleanSocket', "Affect Transform")
        self.inputs.new ('TransformSpaceSocket', "Owner Space")
        self.inputs.new ('FloatFactorSocket', "Influence")
        self.inputs.new ('EnableSocket', "Enable")
        #
        self.outputs.new('RelationshipSocket', "Output Relationship")
        self.initialized = True


            
class LinkLimitScaleNode(Node, LinkNode):
    '''A node representing Limit Scale'''
    bl_idname = 'LinkLimitScale'
    bl_label = "Limit Scale"
    bl_icon = 'CON_SIZELIMIT'
    useTarget : bpy.props.BoolProperty(default=False)
    initialized : bpy.props.BoolProperty(default = False)

    def init(self, context):
        self.inputs.new ('RelationshipSocket', "Input Relationship")
        self.inputs.new ('BoolUpdateParentNode', "Use Max X")
        self.inputs.new ('FloatSocket', "Max X")
        self.inputs.new ('BoolUpdateParentNode', "Use Min X")
        self.inputs.new ('FloatSocket', "Min X")
        self.inputs.new ('BoolUpdateParentNode', "Use Max Y")
        self.inputs.new ('FloatSocket', "Max Y")
        self.inputs.new ('BoolUpdateParentNode', "Use Min Y")
        self.inputs.new ('FloatSocket', "Min Y")
        self.inputs.new ('BoolUpdateParentNode', "Use Max Z")
        self.inputs.new ('FloatSocket', "Max Z")
        self.inputs.new ('BoolUpdateParentNode', "Use Min Z")
        self.inputs.new ('FloatSocket', "Min Z")
        self.inputs.new ('BooleanSocket', "Affect Transform")
        self.inputs.new ('TransformSpaceSocket', "Owner Space")
        self.inputs.new ('FloatFactorSocket', "Influence")
        self.inputs.new ('EnableSocket', "Enable")
        #
        self.outputs.new('RelationshipSocket', "Output Relationship")
        self.initialized = True


            
class LinkLimitRotationNode(Node, LinkNode):
    # === Basics ===
    # Description string
    '''A node representing Limit Rotation'''
    bl_idname = 'LinkLimitRotation'
    bl_label = "Limit Rotation"
    bl_icon = 'CON_ROTLIMIT'
    useTarget : bpy.props.BoolProperty(default=False)
    initialized : bpy.props.BoolProperty(default = False)

    # === Optional Functions ===
    def init(self, context):
        self.inputs.new ('RelationshipSocket', "Input Relationship")
        self.inputs.new ('BoolUpdateParentNode', "Use X")
        self.inputs.new ('FloatAngleSocket', "Min X")
        self.inputs.new ('FloatAngleSocket', "Max X")
        self.inputs.new ('BoolUpdateParentNode', "Use Y")
        self.inputs.new ('FloatAngleSocket', "Min Y")
        self.inputs.new ('FloatAngleSocket', "Max Y")
        self.inputs.new ('BoolUpdateParentNode', "Use Z")
        self.inputs.new ('FloatAngleSocket', "Min Z")
        self.inputs.new ('FloatAngleSocket', "Max Z")
        self.inputs.new ('BooleanSocket', "Affect Transform")
        self.inputs.new ('TransformSpaceSocket', "Owner Space")
        self.inputs.new ('FloatFactorSocket', "Influence")
        self.inputs.new ('EnableSocket', "Enable")
        #
        self.outputs.new('RelationshipSocket', "Output Relationship")
        self.initialized = True


        
class LinkLimitDistanceNode(Node, LinkNode):
    '''A node representing Limit Distance'''
    bl_idname = 'LinkLimitDistance'
    bl_label = "Limit Distance"
    bl_icon = 'CON_DISTLIMIT'
    useTarget : bpy.props.BoolProperty(default=True)

    def init(self, context):
        self.inputs.new ('RelationshipSocket', "Input Relationship")
        self.inputs.new ('FloatFactorSocket', "Head/Tail")
        self.inputs.new ('BooleanSocket', "UseBBone")
        self.inputs.new ('FloatSocket', "Distance")
        self.inputs.new ('EnumLimitMode', "Clamp Region")
        self.inputs.new ('BooleanSocket', "Affect Transform")
        self.inputs.new ('TransformSpaceSocket', "Owner Space")
        self.inputs.new ('TransformSpaceSocket', "Target Space")
        self.inputs.new ('FloatFactorSocket', "Influence")
        self.inputs.new ('xFormSocket', "Target")
        self.inputs.new ('EnableSocket', "Enable")
        #
        self.outputs.new('RelationshipSocket', "Output Relationship")

        
        
class LinkTransformationNode(Node, LinkNode):
    '''A node representing Transformation (Constraint)'''
    bl_idname = 'LinkTransformation'
    bl_label = "Transformation"
    bl_icon = 'CON_TRANSFORM'
    useTarget : bpy.props.BoolProperty(default=True)

    def init(self, context):
        hide_me = []
        self.inputs.new ('RelationshipSocket', "Input Relationship")
        self.inputs.new ('xFormSocket', "Target")
        self.inputs.new ('TransformSpaceSocket', "Owner Space")
        self.inputs.new ('TransformSpaceSocket', "Target Space")
        self.inputs.new ('BooleanSocket', "Extrapolate")
        self.inputs.new ('EnumTransformationMap', "Map From")
        hide_me.append( self.inputs.new ('EnumTransformationRotationMode', "Rotation Mode"))
        self.inputs.new ('FloatSocket', "X Min From")
        self.inputs.new ('FloatSocket', "X Max From")
        self.inputs.new ('FloatSocket', "Y Min From")
        self.inputs.new ('FloatSocket', "Y Max From")
        self.inputs.new ('FloatSocket', "Z Min From")
        self.inputs.new ('FloatSocket', "Z Max From")
        self.inputs.new ('EnumTransformationMap', "Map To")
        hide_me.append( self.inputs.new ('EnumTransformationRotationOrder', "Rotation Order"))
        self.inputs.new ('EnumTransformationAxes', "X Source Axis")
        self.inputs.new ('FloatSocket', "X Min To")
        self.inputs.new ('FloatSocket', "X Max To")
        self.inputs.new ('EnumTransformationAxes', "Y Source Axis")
        self.inputs.new ('FloatSocket', "Y Min To")
        self.inputs.new ('FloatSocket', "Y Max To")
        self.inputs.new ('EnumTransformationAxes', "Z Source Axis")
        self.inputs.new ('FloatSocket', "Z Min To")
        self.inputs.new ('FloatSocket', "Z Max To")
        self.inputs.new ('EnumTransformationTranslationMixMode', "Mix Mode (Translation)")
        hide_me.append( self.inputs.new ('EnumTransformationRotationMixMode', "Mix Mode (Rotation)"))
        hide_me.append( self.inputs.new ('EnumTransformationScaleMixMode', "Mix Mode (Scale)"))
        self.inputs.new ('FloatFactorSocket', "Influence")
        self.inputs.new ('EnableSocket', "Enable")
        #
        self.outputs.new('RelationshipSocket', "Output Relationship")
        
        for s in hide_me:
            s.hide = True

    
    
    def display_update(self, parsed_tree, context):
        node_tree = context.space_data.path[0].node_tree
        nc = parsed_tree.get(get_signature_from_edited_tree(self, context))
        if nc:
            if nc.evaluate_input("Map From") == "ROTATION":
                self.inputs["Rotation Mode"].hide=False
            else:
                self.inputs["Rotation Mode"].hide=True
            if   nc.evaluate_input("Map To") == "TRANSLATION":
                    self.inputs["Rotation Order"].hide=True
                    self.inputs["Mix Mode (Translation)"].hide=False
                    self.inputs["Mix Mode (Rotation)"].hide=True
                    self.inputs["Mix Mode (Scale)"].hide=True
            elif nc.evaluate_input("Map To") == "ROTATION":
                    self.inputs["Rotation Order"].hide=False
                    self.inputs["Mix Mode (Translation)"].hide=True
                    self.inputs["Mix Mode (Rotation)"].hide=False
                    self.inputs["Mix Mode (Scale)"].hide=True
            elif nc.evaluate_input("Map To") == "SCALE":
                    self.inputs["Rotation Order"].hide=True
                    self.inputs["Mix Mode (Translation)"].hide=True
                    self.inputs["Mix Mode (Rotation)"].hide=True
                    self.inputs["Mix Mode (Scale)"].hide=False



class LinkArmatureNode(Node, LinkNode):
    """A node representing Blender's Armature Constraint"""
    bl_idname = "LinkArmature"
    bl_label = "Armature (Constraint)"
    bl_icon = "CON_ARMATURE"
    
    def init(self, context):
        self.inputs.new ("RelationshipSocket", "Input Relationship")
        self.inputs.new("BooleanSocket", "Preserve Volume")
        self.inputs.new("BooleanSocket", "Use Envelopes")
        self.inputs.new("BooleanSocket", "Use Current Location")
        self.inputs.new("FloatSocket", "Influence")
        self.inputs.new ('EnableSocket', "Enable")
        self.outputs.new("RelationshipSocket", "Output Relationship")
    def traverse(self, socket):
        return default_traverse(self,socket)
    
    def draw_buttons(self, context, layout):
        layout.operator( 'mantis.link_armature_node_add_target' )
        if (len(self.inputs) > 6):
            layout.operator( 'mantis.link_armature_node_remove_target' )
        else:
            layout.label(text="")

class LinkSplineIKNode(Node, LinkNode):
    """"A node representing Spline IK"""
    bl_idname = "LinkSplineIK"
    bl_label = "Spline IK"
    bl_icon = "CON_SPLINEIK"
    
    def init(self, context):
        self.inputs.new ("RelationshipSocket", "Input Relationship")
        self.inputs.new("xFormSocket", "Target")
        self.inputs.new("IntSocket", "Chain Length")
        self.inputs.new("BooleanSocket", "Even Divisions")
        self.inputs.new("BooleanSocket", "Chain Offset")
        self.inputs.new("BooleanSocket", "Use Curve Radius")
        self.inputs.new("EnumYScaleMode", "Y Scale Mode")
        self.inputs.new("EnumXZScaleMode", "XZ Scale Mode")
        self.inputs.new("BooleanSocket", "Use Original Scale")
        self.inputs.new("FloatSocket", "Influence")
        self.outputs.new("RelationshipSocket", "Output Relationship")
    def traverse(self, socket):
        return default_traverse(self,socket)
        
        
# DRIVERS!!

class LinkDrivenParameterNode(Node, LinkNode):
    """Represents a driven parameter in the downstream xForm node."""
    bl_idname = "LinkDrivenParameter"
    bl_label = "Driven Parameter"
    bl_icon = "CONSTRAINT_BONE"
    
    def init(self, context):
        self.inputs.new ( "RelationshipSocket", "Input Relationship" )
        self.inputs.new ( "DriverSocket", "Driver" )
        self.inputs.new ( "ParameterStringSocket", "Parameter" )
        self.inputs.new ( "IntSocket", "Index" )
        self.inputs.new ('EnableSocket', "Enable")
        #
        self.outputs.new( "RelationshipSocket", "Output Relationship" )
    def traverse(self, socket):
        return default_traverse(self,socket)