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
from typing import Any
from collections.abc import Sequence
import numpy
from numpy.typing import NDArray, ArrayLike
from numpy.linalg import norm
@ -871,7 +872,7 @@ def solve_modes(
wavenumbers = numpy.sqrt(eigvals)
wavenumbers *= numpy.sign(numpy.real(wavenumbers))
return e_xys, wavenumbers
return e_xys.T, wavenumbers
def solve_mode(
@ -892,4 +893,4 @@ def solve_mode(
"""
kwargs['mode_numbers'] = [mode_number]
e_xys, wavenumbers = solve_modes(*args, **kwargs)
return e_xys[:, 0], wavenumbers[0]
return e_xys[0], wavenumbers[0]