Implemented Comparison Nodes and Low Health Alerts

Nodes/comparison_node.py

@@ -0,0 +1,128 @@
+#!/usr/bin/env python3
+"""
+Comparison Node:
+  - Inputs: Two input ports ("A" and "B").
+  - Output: One output port ("Result").
+  - Operation: A dropdown (combo menu) to select:
+      Equal (==), Not Equal (!=), Greater Than (>), Less Than (<),
+      Greater Than or Equal (>=), Less Than or Equal (<=).
+  - Displays the computed result in a read-only text box labeled "Result".
+"""
+
+from OdenGraphQt import BaseNode
+
+class ComparisonNode(BaseNode):
+    __identifier__ = 'bunny-lab.io.comparison_node'
+    NODE_NAME = 'Comparison Node'
+
+    def __init__(self):
+        super(ComparisonNode, self).__init__()
+
+        # ----------------------------------------------------------------------
+        # Initialization Section:
+        #   - Create two input ports: A, B
+        #   - Create one output port: Result
+        #   - Add a combo box for logical operator selection
+        #   - Add a text input for displaying the computed result
+        # ----------------------------------------------------------------------
+        self.add_input('A')
+        self.add_input('B')
+        self.add_output('Result')
+
+        # Operator combo box (==, !=, >, <, >=, <=)
+        self.add_combo_menu('operator', 'Operator', items=[
+            'Equal (==)', 
+            'Not Equal (!=)', 
+            'Greater Than (>)', 
+            'Less Than (<)', 
+            'Greater Than or Equal (>=)', 
+            'Less Than or Equal (<=)'
+        ])
+
+        # Text input for displaying the computed result.
+        # We'll make it read-only by accessing the underlying QLineEdit.
+        self.add_text_input('calc_result', 'Result', text='0')
+        result_widget = self.get_widget('calc_result')  # This is a NodeLineEdit wrapper
+        if result_widget:
+            # Get the underlying QLineEdit
+            line_edit = result_widget.get_custom_widget()
+            # Make the QLineEdit read-only
+            line_edit.setReadOnly(True)
+
+        self.value = 0
+        self.set_name("Comparison Node")
+        self.process_input()
+
+    def process_input(self, event=None):
+        """
+        Compute Section:
+          - For each input port (A, B), if connected, grab the 'value' from
+            the upstream node; otherwise default to 0.0.
+          - Convert to float when possible, apply the selected comparison operator,
+            update the "Result" text box, node title, and output port.
+        """
+        # Gather input A
+        input_a = self.input(0)
+        if input_a and input_a.connected_ports():
+            a_raw = input_a.connected_ports()[0].node().get_property('value')
+        else:
+            a_raw = 0.0
+
+        # Gather input B
+        input_b = self.input(1)
+        if input_b and input_b.connected_ports():
+            b_raw = input_b.connected_ports()[0].node().get_property('value')
+        else:
+            b_raw = 0.0
+
+        # Convert raw inputs to float if possible, otherwise keep as-is for string comparison.
+        try:
+            a_val = float(a_raw)
+            b_val = float(b_raw)
+        except (ValueError, TypeError):
+            a_val = a_raw
+            b_val = b_raw
+
+        # Retrieve the selected operator from the combo box.
+        operator = self.get_property('operator')
+        result = False
+
+        if operator == 'Equal (==)':
+            result = a_val == b_val
+        elif operator == 'Not Equal (!=)':
+            result = a_val != b_val
+        elif operator == 'Greater Than (>)':
+            result = a_val > b_val
+        elif operator == 'Less Than (<)':
+            result = a_val < b_val
+        elif operator == 'Greater Than or Equal (>=)':
+            result = a_val >= b_val
+        elif operator == 'Less Than or Equal (<=)':
+            result = a_val <= b_val
+
+        # Convert boolean result to integer (1 for True, 0 for False)
+        self.value = 1 if result else 0
+
+        # Update the read-only text input and node title.
+        self.set_property('calc_result', str(self.value))
+
+        # Transmit the numeric result to any connected output nodes.
+        output_port = self.output(0)
+        if output_port and output_port.connected_ports():
+            for cp in output_port.connected_ports():
+                connected_node = cp.node()
+                if hasattr(connected_node, 'receive_data'):
+                    connected_node.receive_data(self.value, source_port_name='Result')
+
+    def on_input_connected(self, input_port, output_port):
+        self.process_input()
+
+    def on_input_disconnected(self, input_port, output_port):
+        self.process_input()
+
+    def property_changed(self, property_name):
+        if property_name in ['operator']:
+            self.process_input()
+
+    def receive_data(self, data, source_port_name=None):
+        self.process_input()