jan/fdfd_tools
2
2
Fork 1
Code Issues Pull Requests Releases Wiki Activity

In-place normalization during eigensolve

Browse Source
This commit is contained in:
jan 2017-09-24 19:13:37 -07:00
parent 7342c8efd7
commit 17fa2aa3d3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
fdfd_tools/waveguide_mode.py
View File

@ -67,8 +67,8 @@ def solve_waveguide_mode_2d(mode_number: int,
eigval = v.conj() @ A @ v eigval = v.conj() @ A @ v
if numpy.linalg.norm(A @ v - eigval * v) < 1e-13: if numpy.linalg.norm(A @ v - eigval * v) < 1e-13:
break break
w = spalg.spsolve(A - eigval * sparse.eye(A.shape[0]), v) v = spalg.spsolve(A - eigval * sparse.eye(A.shape[0]), v)
v = w / numpy.linalg.norm(w) v /= numpy.linalg.norm(v)
# Calculate the wave-vector (force the real part to be positive) # Calculate the wave-vector (force the real part to be positive)
wavenumber = numpy.sqrt(eigval) wavenumber = numpy.sqrt(eigval)