matei/tono
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update FFT naming

Browse Source
This commit is contained in:
matei jordache
2026-03-28 21:08:05 -07:00
parent b8133c064a
commit 6e113725d5
4 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
backend/node_menu.py
View File

@@ -14,7 +14,7 @@ from typing import Any
MENU_LAYOUT: dict[str, list[str]] = { MENU_LAYOUT: dict[str, list[str]] = {
"Input": [ "Input": [
"Image", "Image",
"IBWNote", "Note",
"ImageDemo", "ImageDemo",
"Folder", "Folder",
"Number", "Number",
@@ -53,7 +53,7 @@ MENU_LAYOUT: dict[str, list[str]] = {
], ],
"Spectral": [ "Spectral": [
"FFT2D", "FFT2D",
"InverseFFT2D", "FFT2DInverse",
"FFTFilter1D", "FFTFilter1D",
"FFTFilter2D", "FFTFilter2D",
"ACF2D", "ACF2D",

2
backend/nodes/fft_2d_invert.py → backend/nodes/fft_2d_inverse.py
View File

@@ -5,7 +5,7 @@ from backend.data_types import DataField
@register_node(display_name="Inverse 2D FFT") @register_node(display_name="Inverse 2D FFT")
class InverseFFT2D: class FFT2DInverse:
@classmethod @classmethod
def INPUT_TYPES(cls): def INPUT_TYPES(cls):
return { return {

4
backend/nodes/note.py
View File

@@ -30,8 +30,8 @@ def _parse_ibw_note(note_bytes: bytes) -> list[dict]:
return rows return rows
@register_node(display_name="IBW Note") @register_node(display_name="Note")
class IBWNote: class Note:
@classmethod @classmethod
def INPUT_TYPES(cls): def INPUT_TYPES(cls):
return { return {

10
tests/test_fft.py
View File

@@ -10,7 +10,7 @@ import numpy as np
sys.path.insert(0, ".") sys.path.insert(0, ".")
from backend.data_types import DataField from backend.data_types import DataField
from backend.nodes.fft_2d import FFT2D from backend.nodes.fft_2d import FFT2D
from backend.nodes.fft_2d_invert import InverseFFT2D from backend.nodes.fft_2d_inverse import FFT2DInverse
def make_field(data, xreal=1e-6, yreal=1e-6): def make_field(data, xreal=1e-6, yreal=1e-6):
@@ -255,7 +255,7 @@ def test_inverse_fft_reconstructs_from_magnitude_and_phase():
field = make_field(data, xreal=2.4e-6, yreal=1.6e-6) field = make_field(data, xreal=2.4e-6, yreal=1.6e-6)
fft_node = FFT2D() fft_node = FFT2D()
ifft_node = InverseFFT2D() ifft_node = FFT2DInverse()
_, magnitude, phase, _ = fft_node.process(field, windowing="none", level="none") _, magnitude, phase, _ = fft_node.process(field, windowing="none", level="none")
reconstructed, = ifft_node.process(magnitude, representation="magnitude", phase=phase) reconstructed, = ifft_node.process(magnitude, representation="magnitude", phase=phase)
@@ -282,7 +282,7 @@ def test_inverse_fft_reconstructs_from_log_magnitude_and_phase():
field = make_field(data, xreal=1.6e-6, yreal=1.44e-6) field = make_field(data, xreal=1.6e-6, yreal=1.44e-6)
fft_node = FFT2D() fft_node = FFT2D()
ifft_node = InverseFFT2D() ifft_node = FFT2DInverse()
log_magnitude, _, phase, _ = fft_node.process(field, windowing="none", level="none") log_magnitude, _, phase, _ = fft_node.process(field, windowing="none", level="none")
reconstructed, = ifft_node.process(log_magnitude, representation="log_magnitude", phase=phase) reconstructed, = ifft_node.process(log_magnitude, representation="log_magnitude", phase=phase)
@@ -301,7 +301,7 @@ def test_inverse_fft_reconstructs_from_psdf_and_phase():
field = make_field(data, xreal=3.2e-6, yreal=2.4e-6) field = make_field(data, xreal=3.2e-6, yreal=2.4e-6)
fft_node = FFT2D() fft_node = FFT2D()
ifft_node = InverseFFT2D() ifft_node = FFT2DInverse()
_, _, phase, psdf = fft_node.process(field, windowing="none", level="none") _, _, phase, psdf = fft_node.process(field, windowing="none", level="none")
reconstructed, = ifft_node.process(psdf, representation="psdf", phase=phase) reconstructed, = ifft_node.process(psdf, representation="psdf", phase=phase)
@@ -320,7 +320,7 @@ def test_inverse_fft_zero_phase_mode_returns_valid_image():
field = make_field(data, xreal=1e-6, yreal=1e-6) field = make_field(data, xreal=1e-6, yreal=1e-6)
fft_node = FFT2D() fft_node = FFT2D()
ifft_node = InverseFFT2D() ifft_node = FFT2DInverse()
_, magnitude, _, _ = fft_node.process(field, windowing="none", level="none") _, magnitude, _, _ = fft_node.process(field, windowing="none", level="none")
reconstructed, = ifft_node.process(magnitude, representation="magnitude") reconstructed, = ifft_node.process(magnitude, representation="magnitude")