jan/meanas
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Return list-of-vectors from waveguide mode solve

Browse Source
This commit is contained in:
Jan Petykiewicz 2025-01-07 00:04:53 -08:00
parent e459b5e61f
commit 47415a0beb
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
meanas/fdfd/waveguide_2d.py
View File

@ -179,6 +179,7 @@ to account for numerical dispersion if the result is introduced into a space wit
# TODO update module docs # TODO update module docs
from typing import Any from typing import Any
from collections.abc import Sequence
import numpy import numpy
from numpy.typing import NDArray, ArrayLike from numpy.typing import NDArray, ArrayLike
from numpy.linalg import norm from numpy.linalg import norm
@ -871,7 +872,7 @@ def solve_modes(
wavenumbers = numpy.sqrt(eigvals) wavenumbers = numpy.sqrt(eigvals)
wavenumbers *= numpy.sign(numpy.real(wavenumbers)) wavenumbers *= numpy.sign(numpy.real(wavenumbers))
return e_xys, wavenumbers return e_xys.T, wavenumbers
def solve_mode( def solve_mode(
@ -892,4 +893,4 @@ def solve_mode(
""" """
kwargs['mode_numbers'] = [mode_number] kwargs['mode_numbers'] = [mode_number]
e_xys, wavenumbers = solve_modes(*args, **kwargs) e_xys, wavenumbers = solve_modes(*args, **kwargs)
return e_xys[:, 0], wavenumbers[0] return e_xys[0], wavenumbers[0]