73 lines
2.3 KiB
Python
73 lines
2.3 KiB
Python
"""Multiple profiles — extract and compare profiles from multiple images."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import numpy as np
|
|
|
|
from backend.node_registry import register_node
|
|
from backend.data_types import DataField, LineData
|
|
|
|
|
|
@register_node(display_name="Multiple Profiles")
|
|
class MultipleProfiles:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"field_a": ("DATA_FIELD",),
|
|
"field_b": ("DATA_FIELD",),
|
|
"row": ("INT", {"default": -1, "min": -1, "max": 10000, "step": 1}),
|
|
"direction": (["horizontal", "vertical"], {"default": "horizontal"}),
|
|
"mode": (["overlay", "mean", "difference"], {"default": "overlay"}),
|
|
}
|
|
}
|
|
|
|
OUTPUTS = (
|
|
('LINE_DATA', 'profile'),
|
|
)
|
|
FUNCTION = "process"
|
|
|
|
DESCRIPTION = (
|
|
"Extract and compare line profiles from two fields. "
|
|
"Row=-1 uses the center row/column. Modes: overlay returns field_a's "
|
|
"profile, mean averages both, difference subtracts b from a. "
|
|
)
|
|
|
|
def process(self, field_a: DataField, field_b: DataField,
|
|
row: int, direction: str, mode: str) -> tuple:
|
|
a = np.asarray(field_a.data, dtype=np.float64)
|
|
b = np.asarray(field_b.data, dtype=np.float64)
|
|
|
|
if direction == "horizontal":
|
|
if row < 0:
|
|
row = a.shape[0] // 2
|
|
row = min(row, a.shape[0] - 1, b.shape[0] - 1)
|
|
pa = a[row, :]
|
|
pb = b[row, :min(a.shape[1], b.shape[1])]
|
|
pa = pa[:len(pb)]
|
|
dx = field_a.dx
|
|
x_unit = field_a.si_unit_xy
|
|
else:
|
|
if row < 0:
|
|
row = a.shape[1] // 2
|
|
row = min(row, a.shape[1] - 1, b.shape[1] - 1)
|
|
pa = a[:, row]
|
|
pb = b[:min(a.shape[0], b.shape[0]), row]
|
|
pa = pa[:len(pb)]
|
|
dx = field_a.dy
|
|
x_unit = field_a.si_unit_xy
|
|
|
|
x_axis = np.arange(len(pa)) * dx
|
|
|
|
if mode == "overlay":
|
|
result = pa
|
|
elif mode == "mean":
|
|
result = 0.5 * (pa + pb)
|
|
elif mode == "difference":
|
|
result = pa - pb
|
|
else:
|
|
result = pa
|
|
|
|
return (LineData(data=result, x_axis=x_axis, x_unit=x_unit,
|
|
y_unit=field_a.si_unit_z),)
|