35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
from __future__ import annotations
|
|
|
|
from backend.node_registry import register_node
|
|
from backend.data_types import DataField
|
|
from backend.nodes.spectral_common import preprocess_spectral_data, psdf_field_from_data
|
|
|
|
|
|
@register_node(display_name="PSDF")
|
|
class PSDF:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"field": ("DATA_FIELD",),
|
|
"windowing": (["hann", "hamming", "blackman", "none"], {"default": "hann"}),
|
|
"level": (["mean", "plane", "none"], {"default": "mean"}),
|
|
}
|
|
}
|
|
|
|
OUTPUTS = (
|
|
('DATA_FIELD', 'psdf'),
|
|
)
|
|
FUNCTION = "process"
|
|
|
|
DESCRIPTION = (
|
|
"Compute the two-dimensional power spectral density function with Gwyddion-style "
|
|
"window RMS compensation and centered zero frequency."
|
|
)
|
|
|
|
KEYWORDS = ("power spectrum", "fourier", "frequency", "roughness", "spectral density")
|
|
|
|
def process(self, field: DataField, windowing: str, level: str) -> tuple:
|
|
data = preprocess_spectral_data(field, level=level, windowing=windowing)
|
|
return (psdf_field_from_data(field, data),)
|