feat: implement dea module

This commit is contained in:
ws50529
2026-05-14 09:41:00 +08:00
parent edec4e6db6
commit a9489dafe4
14 changed files with 873 additions and 679 deletions
Executable
+26
View File
@@ -0,0 +1,26 @@
import numpy as np
from scipy import signal
from .config import BODE_MAX_MULTIPLIER, BODE_POINTS
from .schemas import BodeParams
from .validation import require_finite, validate_coefficients
def calculate_bode_response(params: BodeParams):
fs_val = require_finite(params.fs, "fs")
b_vals = validate_coefficients(params.b, "b")
a_vals = validate_coefficients(params.a, "a")
f_min = params.fs / 50000.0
f_max = params.fs * BODE_MAX_MULTIPLIER
f_eval = np.logspace(np.log10(f_min), np.log10(f_max), BODE_POINTS)
_, h = signal.freqz(b_vals, a_vals, worN=f_eval, fs=fs_val)
mag = 20 * np.log10(np.abs(h) + 1e-12)
phase = np.angle(h, deg=True)
return {
"freq": f_eval.tolist(),
"mag": mag.tolist(),
"phase": phase.tolist(),
}