from lnx.logicnode.lnx_nodes import *

class SetRigidBodyFrictionNode (LnxLogicTreeNode):
    """Sets the friction of the given rigid body."""
    bl_idname = 'LNSetRigidBodyFrictionNode'
    bl_label = 'Set RB Friction'
    lnx_section = 'props'
    lnx_version = 1

    def lnx_init(self, context):
        self.add_input('LnxNodeSocketAction', 'In')
        self.add_input('LnxNodeSocketObject', 'RB')
        self.add_input('LnxFloatSocket', 'Friction', default_value=0.5)
        self.add_input('LnxFloatSocket', 'Angular Friction', default_value=0.1)

        self.add_output('LnxNodeSocketAction', 'Out')