40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
from __future__ import annotations
|
|
|
|
import numpy as np
|
|
|
|
from backend.data_types import DataField
|
|
|
|
|
|
_LENGTH_UNITS = {"m", "km", "cm", "mm", "um", "µm", "nm", "pm", "fm"}
|
|
|
|
|
|
def unit_dimension_key(unit: str) -> str:
|
|
text = str(unit or "").strip().replace("µ", "u")
|
|
if not text:
|
|
return ""
|
|
if text in _LENGTH_UNITS:
|
|
return "length"
|
|
return text
|
|
|
|
|
|
def slope_unit(field: DataField) -> str:
|
|
"""Return the physical slope unit string (z_unit/xy_unit)."""
|
|
z = str(field.si_unit_z or "").strip()
|
|
xy = str(field.si_unit_xy or "").strip()
|
|
return f"{z}/{xy}" if z and xy else (z or xy)
|
|
|
|
|
|
def physical_sobel_gradient(field: DataField) -> tuple[np.ndarray, np.ndarray]:
|
|
"""Compute physical Sobel gradient (gx, gy) in z_unit/xy_unit."""
|
|
from scipy.ndimage import sobel
|
|
gx = sobel(field.data, axis=1) / (8.0 * field.dx)
|
|
gy = sobel(field.data, axis=0) / (8.0 * field.dy)
|
|
return gx, gy
|
|
|
|
|
|
def require_compatible_xy_z_units(field: DataField, node_name: str) -> None:
|
|
xy_key = unit_dimension_key(field.si_unit_xy)
|
|
z_key = unit_dimension_key(field.si_unit_z)
|
|
if xy_key and z_key and xy_key != z_key:
|
|
raise ValueError(f"{node_name} requires compatible XY and Z units, matching Gwyddion's topography-only behavior.")
|