from lnx.logicnode.lnx_nodes import *

class GetRigidBodyDataNode(LnxLogicTreeNode):
    """Returns the data of the given rigid body."""
    bl_idname = 'LNGetRigidBodyDataNode'
    bl_label = 'Get RB Data'
    lnx_section = 'props'
    lnx_version = 1

    def lnx_init(self, context):
        self.add_input('LnxNodeSocketObject', 'RB')

        self.add_output('LnxBoolSocket', 'Is RB')
        self.add_output('LnxIntSocket', 'Collision Group')
        self.add_output('LnxIntSocket', 'Collision Mask')
        self.add_output('LnxBoolSocket', 'Is Animated')
        self.add_output('LnxBoolSocket', 'Is Static')
        self.add_output('LnxFloatSocket', 'Linear Damping')
        self.add_output('LnxFloatSocket', 'Angular Damping')
        self.add_output('LnxFloatSocket', 'Friction')
        self.add_output('LnxFloatSocket', 'Angular Friction')
        self.add_output('LnxFloatSocket', 'Mass')
        self.add_output('LnxBoolSocket', 'Is Trigger')
        self.add_output('LnxFloatSocket', 'Bounciness')
        self.add_output('LnxBoolSocket', 'Gravity Enabled')
        self.add_output('LnxVectorSocket', 'Gravity')
        self.add_output('LnxVectorSocket', 'Linear Factor')
        self.add_output('LnxVectorSocket', 'Angular Factor')