adding more nodes

backend/nodes/tilt.py

@@ -0,0 +1,49 @@
+"""Tilt — apply or remove a linear tilt."""
+
+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="Tilt")
+class Tilt:
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "field": ("DATA_FIELD",),
+                "slope_x": ("FLOAT", {"default": 0.0, "min": -1e6, "max": 1e6, "step": 0.001}),
+                "slope_y": ("FLOAT", {"default": 0.0, "min": -1e6, "max": 1e6, "step": 0.001}),
+                "mode": (["add", "subtract"], {"default": "add"}),
+            }
+        }
+
+    OUTPUTS = (
+        ('DATA_FIELD', 'tilted'),
+    )
+    FUNCTION = "process"
+
+    DESCRIPTION = (
+        "Apply or subtract a linear tilt (plane). slope_x and slope_y are "
+        "in data units per physical unit (e.g., m/m for height data). "
+        "Use 'subtract' mode to remove a known tilt. "
+    )
+
+    def process(self, field: DataField, slope_x: float, slope_y: float,
+                mode: str) -> tuple:
+        data = np.asarray(field.data, dtype=np.float64)
+        yres, xres = data.shape
+
+        x = np.arange(xres) * field.dx
+        y = np.arange(yres) * field.dy
+        X, Y = np.meshgrid(x, y)
+
+        tilt_plane = slope_x * X + slope_y * Y
+
+        if mode == "subtract":
+            return (field.replace(data=data - tilt_plane),)
+        else:
+            return (field.replace(data=data + tilt_plane),)