Upload app.py

app.py

@@ -0,0 +1,134 @@
+import streamlit as st
+import matplotlib.pyplot as plt
+import numpy as np
+
+from loudspeaker_tmatrix.core import simulate_loudspeaker
+from loudspeaker_tmatrix.visualization import plot_loudspeaker_response
+from loudspeaker_tmatrix.config import (
+    load_config,
+    load_loudspeaker_config,
+    AcousticalConstantsConfig,
+)
+
+config_path = "configs/config.yaml"
+default_loudspeaker_path = "configs/default_loudspeaker.yaml"
+
+cfg = load_config(config_path)
+loudspeaker_cfg = load_loudspeaker_config(default_loudspeaker_path)
+
+freq_array = np.logspace(
+    np.log10(cfg.frequency.min), np.log10(cfg.frequency.max), num=cfg.frequency.n_bins
+)
+
+angular_freq_array = 2 * np.pi * freq_array
+
+st.sidebar.title("Thiele Small parameters")
+
+### Sliders
+slider_Re = st.sidebar.slider(
+    "Electrical Coil Resistance (Re) [Ohm]",
+    0.0,
+    10.0,
+    loudspeaker_cfg.electrical.coil_resistance,
+)
+slider_Le = st.sidebar.slider(
+    "Electrical Coil Inductance (Le) [mH]",
+    0.0,
+    10.0,
+    loudspeaker_cfg.electrical.coil_inductance * 1e3,
+)
+slider_Le /= 1e3
+slider_Bl = st.sidebar.slider(
+    "Electromechanical factor (Bl) [N/A]",
+    0.0,
+    10.0,
+    loudspeaker_cfg.electromechanical_factor,
+)
+slider_Mm = st.sidebar.slider(
+    "Mechanical Mass (Mm) [mg]",
+    0.0,
+    50.0,
+    loudspeaker_cfg.mechanical.mass * 1e3,
+)
+slider_Mm /= 1e3
+slider_Cm = st.sidebar.slider(
+    "Mechanical Compliance (Cm) [mm/N]",
+    0.0,
+    5.0,
+    loudspeaker_cfg.mechanical.compliance * 1e3,
+)
+slider_Cm /= 1e3
+slider_Rm = st.sidebar.slider(
+    "Mechanical Resistance (Rm) [kg/s]",
+    0.0,
+    10.0,
+    loudspeaker_cfg.mechanical.resistance,
+)
+slider_diam = st.sidebar.slider(
+    "Effective diameter of radiation [cm]",
+    0.0,
+    50.0,
+    loudspeaker_cfg.acoustical.effective_diameter * 1e2,
+)
+slider_diam /= 1e2
+
+default_params = st.sidebar.button("Set default parameters")
+
+if default_params:
+    st.rerun(scope="app")
+
+
+freq_array = np.logspace(
+    np.log10(cfg.frequency.min), np.log10(cfg.frequency.max), num=cfg.frequency.n_bins
+)
+
+angular_freq_array = 2 * np.pi * freq_array
+
+thiele_small_params = {
+    "Re": slider_Re,
+    "Le": slider_Le,
+    "Bl": slider_Bl,
+    "Mm": slider_Mm,
+    "Cm": slider_Cm,
+    "Rm": slider_Rm,
+    "effective_diameter": slider_diam,
+}
+
+loudspeaker_responses = simulate_loudspeaker(
+    thiele_small_params, angular_freq_array, cfg.acoustical_constants
+)
+
+
+# Electrical impedance
+electrical_impedance_plot = plot_loudspeaker_response(
+    response_array=loudspeaker_responses["electrical_impedance"],
+    freq_array=freq_array,
+    title="Electrical Impedance",
+    magnitude_in_db=False,
+    magnitude_units="Ohm",
+    shift_phase=False,
+)
+
+
+mechanical_velocity_plot = plot_loudspeaker_response(
+    response_array=loudspeaker_responses["mechanical_velocity"],
+    freq_array=freq_array,
+    title="Mechanical Velocity",
+    magnitude_in_db=False,
+    magnitude_units="m/s",
+    shift_phase=True,
+)
+
+
+acoustical_pressure = plot_loudspeaker_response(
+    response_array=loudspeaker_responses["acoustical_pressure"],
+    freq_array=freq_array,
+    title="Acoustical Pressure",
+    magnitude_in_db=True,
+    magnitude_units="dB",
+    shift_phase=False,
+)
+
+st.pyplot(electrical_impedance_plot)
+st.pyplot(mechanical_velocity_plot)
+st.pyplot(acoustical_pressure)