57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
"""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),)
|