File size: 1,805 Bytes
fa34304 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
import numpy as np
import matplotlib.pyplot as plt
import plotly.tools as tls
import plotly.io as pio
import warnings
from bpm.mode_solver import slab_mode_source
def test_slab_mode_plot():
# Define transverse coordinate and waveguide parameters
x = np.linspace(-10, 10, 1000) # Adjust range/resolution as needed
w = 5.0 # waveguide width in microns
n_WG = 1.1 # core refractive index
n0 = 1.0 # cladding refractive index
wavelength = 0.532 # in microns
# Try to compute up to 5 modes
num_modes = 5
mode_fields = []
mode_indices = []
for m in range(num_modes):
try:
E_mode = slab_mode_source(x, w, n_WG, n0, wavelength, ind_m=m)
mode_fields.append(E_mode)
mode_indices.append(m)
except Exception as err:
warnings.warn(f"Mode {m} not found: {err}. Stopping mode search.")
break
# Plot the real part of each mode with an offset for clarity
plt.rcParams.update({'font.size': 14})
fig, ax = plt.subplots(figsize=(8, 6))
offset = 1.0 # Vertical offset between mode plots
for i, E_mode in enumerate(mode_fields):
ax.plot(x, np.real(E_mode) + i * offset, label=f"Mode {mode_indices[i]}")
ax.set_xlabel("x (µm)")
ax.set_ylabel("Field amplitude (Real part)")
ax.set_title("Slab Waveguide TE Modes")
ax.legend()
plt.grid(True)
plt.tight_layout()
# Save the plot as a PNG file
plt.savefig("test_mode_solver.png")
# plt.show()
# Optionally convert the matplotlib figure to a Plotly figure for interactive viewing
# plotly_fig = tls.mpl_to_plotly(fig)
# plotly_fig.update_layout(legend=dict(font=dict(size=12)))
# pio.show(plotly_fig)
if __name__ == "__main__":
test_slab_mode_plot()
|