adding more nodes

backend/nodes/wrap_value.py

@@ -0,0 +1,56 @@
+"""Value wrapping — rewrap periodic values to different ranges."""
+
+from __future__ import annotations
+
+import numpy as np
+
+from backend.node_registry import register_node
+from backend.data_types import DataField
+
+
+@register_node(display_name="Wrap Value")
+class WrapValue:
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "field": ("DATA_FIELD",),
+                "range": (["0_to_360", "neg180_to_180", "0_to_2pi", "neg_pi_to_pi", "custom"],
+                          {"default": "0_to_360"}),
+                "custom_min": ("FLOAT", {"default": 0.0, "min": -1e6, "max": 1e6, "step": 0.1}),
+                "custom_max": ("FLOAT", {"default": 360.0, "min": -1e6, "max": 1e6, "step": 0.1}),
+            }
+        }
+
+    OUTPUTS = (
+        ('DATA_FIELD', 'wrapped'),
+    )
+    FUNCTION = "process"
+
+    DESCRIPTION = (
+        "Rewrap periodic values (phase, angle) to a specified range. "
+        "Preset ranges for degrees and radians, or specify a custom range. "
+    )
+
+    def process(self, field: DataField, range: str,
+                custom_min: float, custom_max: float) -> tuple:
+        data = np.asarray(field.data, dtype=np.float64)
+
+        ranges = {
+            "0_to_360": (0.0, 360.0),
+            "neg180_to_180": (-180.0, 180.0),
+            "0_to_2pi": (0.0, 2 * np.pi),
+            "neg_pi_to_pi": (-np.pi, np.pi),
+        }
+
+        if range == "custom":
+            lo, hi = custom_min, custom_max
+        else:
+            lo, hi = ranges[range]
+
+        period = hi - lo
+        if period <= 0:
+            return (field.replace(data=data),)
+
+        wrapped = lo + (data - lo) % period
+        return (field.replace(data=wrapped),)