Update core.py

core.py

@@ -73,6 +73,11 @@ def simulate_loudspeaker(
         ),
     )[:, 0]
 
+    mechanical_displacement = mechanical_velocity / (
+        1j * angular_freq_array
+    )  # [(m/s) / (rad/s)] = [m/rad]
+    mechanical_displacement = mechanical_displacement * 2 * np.pi * 1000  # [mm]
+
     # Acoustical response
     air_impedance = acoustical_constants.air_density * acoustical_constants.sound_speed
     wave_number_array = angular_freq_array / acoustical_constants.sound_speed
@@ -88,7 +93,7 @@ def simulate_loudspeaker(
     ZM_rad_imag_array = (
         Sd_value * air_impedance * (1j * (struve(2 * ka_array) / ka_array))
     )
-    ZM_rad_array = ZM_rad_real_array + ZM_rad_imag_array
+    ZM_rad_array = (ZM_rad_real_array + 1j * ZM_rad_imag_array).astype(np.complex128)
     ZM_rad_tmatrix = np.array(
         [[np.ones(n_bins), ZM_rad_array], [np.zeros(n_bins), np.ones(n_bins)]]
     )
@@ -139,6 +144,7 @@ def simulate_loudspeaker(
         "electrical_impedance": electrical_impedance_shorted_output,
         "mechanical_force": mechanical_force,
         "mechanical_velocity": mechanical_velocity,
+        "mechanical_displacement": mechanical_displacement,
         "acoustical_pressure": acoustical_pressure,
         "selectivity_params": {
             "fs": np.round(mechanical_fs, 2),