Update Files

leenkx/blender/lnx/logicnode/math/LN_math_expression.py

@@ -0,0 +1,80 @@
+from lnx.logicnode.lnx_nodes import *
+import re
+
+class MathExpressionNode(LnxLogicTreeNode):
+    """Mathematical operations on values."""
+    bl_idname = 'LNMathExpressionNode'
+    bl_label = 'Math Expression'
+    lnx_version = 2
+
+    num_params: IntProperty(default=2, min=0)
+
+    @staticmethod
+    def get_variable_name(index):
+        return chr( range(ord('a'), ord('z')+1)[index] )
+
+    def set_exp(self, value):
+        self['property2'] = value
+        # TODO: Check expression for errors
+        self['exp_error'] = False
+
+    def get_exp(self):
+        return self.get('property2', 'a + b')
+
+    property0: HaxeBoolProperty('property0', name='Clamp Result', default=False)
+    property1: HaxeIntProperty('property1', name='Number of Params', default=2)
+    property2: HaxeStringProperty('property2', name='', description='Math Expression: +, -, *, /, ^, %, (, ), log(a, b), ln(a), abs(a), max(a,b), min(a,b), sin(a), cos(a), tan(a), cot(a), asin(a), acos(a), atan(a), atan2(a,b), pi(), e()', set=set_exp, get=get_exp)
+
+    def __init__(self):
+        super(MathExpressionNode, self).__init__()
+        self.register_id()
+
+
+    def lnx_init(self, context):
+        
+        # OUTPUTS:
+        self.add_output('LnxFloatSocket', 'Result')
+        
+        # two default parameters at start
+        self.add_input('LnxFloatSocket', self.get_variable_name(0), default_value=0.0)
+        self.add_input('LnxFloatSocket', self.get_variable_name(1), default_value=0.0)
+
+    def add_sockets(self):
+        if self.num_params < 26:
+            self.add_input('LnxFloatSocket', self.get_variable_name(self.num_params), default_value=0.0)
+            self.num_params += 1
+            self['property1'] = self.num_params
+
+    def remove_sockets(self):
+        if self.num_params > 0:
+            self.inputs.remove(self.inputs.values()[-1])
+            self.num_params -= 1
+            self['property1'] = self.num_params
+
+    def draw_buttons(self, context, layout):
+        # Clamp Property
+        layout.prop(self, 'property0')
+
+        # Expression Property
+        row = layout.row(align=True)
+        column = row.column(align=True)
+        # TODO:
+        #column.alert = self['exp_error']
+        column.prop(self, 'property2', icon='FORCE_HARMONIC')
+
+        # Button ADD parameter
+        row = layout.row(align=True)
+        column = row.column(align=True)
+        op = column.operator('lnx.node_call_func', text='Add Param', icon='PLUS', emboss=True)
+        op.node_index = self.get_id_str()
+        op.callback_name = 'add_sockets'
+        if self.num_params == 26:
+            column.enabled = False
+
+        # Button REMOVE parameter
+        column = row.column(align=True)
+        op = column.operator('lnx.node_call_func', text='', icon='X', emboss=True)
+        op.node_index = self.get_id_str()
+        op.callback_name = 'remove_sockets'
+        if self.num_params == 0:
+            column.enabled = False