New logic node to add custom physics constraints. Add documentation.

blender/arm/logicnode/physics/LN_physics_constraint.py

@@ -0,0 +1,87 @@
+from arm.logicnode.arm_nodes import *
+
+class PhysicsConstraintNode(ArmLogicTreeNode):
+    """
+    Custom physics constraint to add to `Add Physics Constarint` node.
+
+    @option Linear/Angualr: Select if constrint is applied along linear or angular axis.
+
+    @option Axis: Local axis of the pivot object along which the constraint is applied.
+
+    @option Spring: Constraint is a Spring along the selected axis.
+
+    @input Limit Lower: Lower limit of the consraint in that particular axis
+
+    @input Limit Upper: Upper limit of the constraint in that particular axis. (`lower limit` = `upper limit`) --> Fully constrained. (`lower limit` < `upper limit`) --> Partially constrained
+    (`lower limit` > `upper limit`) --> Full freedom.
+
+    @seeNode Add Physics Constraint
+    """
+
+    bl_idname = 'LNPhysicsConstraintNode'
+    bl_label = 'Physics Constraint'
+    arm_sction = 'add'
+    arm_version = 1
+
+    def update_spring(self, context):
+        self.update_sockets(context)
+
+    property0: EnumProperty(
+        items = [('Linear', 'Linear', 'Linear'),
+                 ('Angular', 'Angular', 'Angular')],
+        name='Type', default='Linear')
+    
+    @property
+    def property1(self):
+        if(self.property1_ == 'X'):
+            return 'X'
+        if(self.property1_ == 'Y'):
+            return 'Y'
+        if(self.property1_ == 'Z'):
+            return 'Z'
+
+    property1_: EnumProperty(
+        items = [('X', 'X', 'X'),
+                 ('Y', 'Y', 'Y'),
+                 ('Z', 'Z', 'Z')],
+        name='Axis', default='X')
+
+    @property
+    def property2(self):
+        if(self.property2_):
+            return 'true' if self.property2_ else 'false'
+
+    property2_: BoolProperty(
+        name="Spring",
+        description="Is a spring constraint",
+        default=False,
+        update=update_spring
+    )
+    
+    def __init__(self):
+        array_nodes[str(id(self))] = self
+
+    def init(self, context):
+        super(PhysicsConstraintNode, self).init(context)
+        self.add_input('NodeSocketFloat', 'Lower limit')
+        self.add_input('NodeSocketFloat', 'Upper limit')
+        self.add_output('NodeSocketShader', 'Constraint')
+    
+    def update_sockets(self, context):
+
+        while (len(self.inputs) > 0):
+            self.inputs.remove(self.inputs.values()[-1])
+
+        # Add dynamic input sockets
+        if self.property2_:
+            self.add_input('NodeSocketFloat', 'Stiffness', 10.0)
+            self.add_input('NodeSocketFloat', 'Damping', 0.5)
+        else:
+            self.add_input('NodeSocketFloat', 'Lower limit')
+            self.add_input('NodeSocketFloat', 'Upper limit')
+
+    def draw_buttons(self, context, layout):
+        layout.prop(self, 'property0')
+        layout.prop(self, 'property1_')
+        layout.prop(self, 'property2_')
+