Compare commits

..

No commits in common. "a7d0f4d3b84d179e903d4aa763dc74857a8b0e60" and "9ffe57b4d06ecacdd93e89aec9998dd93b622950" have entirely different histories.

2 changed files with 19 additions and 24 deletions

View File

@ -3,7 +3,7 @@ import numpy
from numpy.linalg import norm
from meanas.fdmath import vec, unvec
from meanas.fdfd import waveguide_cyl, functional, scpml
from meanas.fdfd import waveguide_mode, functional, scpml
from meanas.fdfd.solvers import generic as generic_solver
import gridlock
@ -44,27 +44,22 @@ def test1(solver=generic_solver):
# #### Create the grid and draw the device ####
grid = gridlock.Grid(edge_coords)
epsilon = grid.allocate(n_air**2, dtype=numpy.float32)
grid.draw_cuboid(epsilon, center=center, dimensions=[8e3, w, th], foreground=n_wg**2)
grid.draw_cuboid(epsilon, center=center, dimensions=[8e3, w, th], eps=n_wg**2)
dxes = [grid.dxyz, grid.autoshifted_dxyz()]
for a in (1, 2):
for p in (-1, 1):
dxes = scpml.stretch_with_scpml(
dxes,
omega=omega,
axis=a,
polarity=p,
thickness=pml_thickness,
)
dxes = scmpl.stretch_with_scpml(dxes, omega=omega, axis=a, polarity=p,
thickness=pml_thickness)
wg_args = {
'omega': omega,
'dxes': [(d[1], d[2]) for d in dxes],
'epsilon': vec(epsilon.transpose([0, 2, 3, 1])),
'epsilon': vec(g.transpose([1, 2, 0]) for g in epsilon),
'r0': r0,
}
wg_results = waveguide_cyl.solve_mode(mode_number=0, **wg_args)
wg_results = waveguide_mode.solve_waveguide_mode_cylindrical(mode_number=0, **wg_args)
E = wg_results['E']
@ -75,17 +70,20 @@ def test1(solver=generic_solver):
'''
Plot results
'''
def pcolor(fig, ax, v, title):
def pcolor(v):
vmax = numpy.max(numpy.abs(v))
mappable = ax.pcolormesh(v.T, cmap='seismic', vmin=-vmax, vmax=vmax)
ax.set_aspect('equal', adjustable='box')
ax.set_title(title)
ax.figure.colorbar(mappable)
pyplot.pcolor(v.T, cmap='seismic', vmin=-vmax, vmax=vmax)
pyplot.axis('equal')
pyplot.colorbar()
fig, axes = pyplot.subplots(2, 2)
pcolor(fig, axes[0][0], numpy.real(E[0]), 'Ex')
pcolor(fig, axes[0][1], numpy.real(E[1]), 'Ey')
pcolor(fig, axes[1][0], numpy.real(E[2]), 'Ez')
pyplot.figure()
pyplot.subplot(2, 2, 1)
pcolor(numpy.real(E[0][:, :]))
pyplot.subplot(2, 2, 2)
pcolor(numpy.real(E[1][:, :]))
pyplot.subplot(2, 2, 3)
pcolor(numpy.real(E[2][:, :]))
pyplot.subplot(2, 2, 4)
pyplot.show()

View File

@ -25,9 +25,6 @@ def cylindrical_operator(
"""
Cylindrical coordinate waveguide operator of the form
(NOTE: See 10.1364/OL.33.001848)
TODO: consider 10.1364/OE.20.021583
TODO
for use with a field vector of the form `[E_r, E_y]`.