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

Make find-k also return eigenvalues

Browse Source
This commit is contained in:
Jan Petykiewicz 2022-10-04 17:27:58 -07:00
parent 4e240988c9
commit 640b4d1ef7
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
meanas/fdfd/bloch.py
View File

@ -414,7 +414,7 @@ def find_k(
k_min: float = 0, k_min: float = 0,
k_max: float = 0.5, k_max: float = 0.5,
solve_callback: Optional[Callable] = None solve_callback: Optional[Callable] = None
) -> Tuple[NDArray[numpy.float64], float]: ) -> Tuple[float, float, NDArray[numpy.complex128], NDArray[numpy.complex128]]:
""" """
Search for a bloch vector that has a given frequency. Search for a bloch vector that has a given frequency.
@ -430,12 +430,16 @@ def find_k(
band: Which band to search in. Default 0 (lowest frequency). band: Which band to search in. Default 0 (lowest frequency).
Returns: Returns:
`(k, actual_frequency)` `(k, actual_frequency, eigenvalues, eigenvectors)`
The found k-vector and its frequency. The found k-vector and its frequency, along with all eigenvalues and eigenvectors.
""" """
direction = numpy.array(direction) / norm(direction) direction = numpy.array(direction) / norm(direction)
n = None
v = None
def get_f(k0_mag: float, band: int = 0) -> float: def get_f(k0_mag: float, band: int = 0) -> float:
nonlocal n, v
k0 = direction * k0_mag # type: ignore k0 = direction * k0_mag # type: ignore
n, v = eigsolve(band + 1, k0, G_matrix=G_matrix, epsilon=epsilon, mu=mu) n, v = eigsolve(band + 1, k0, G_matrix=G_matrix, epsilon=epsilon, mu=mu)
f = numpy.sqrt(numpy.abs(numpy.real(n[band]))) f = numpy.sqrt(numpy.abs(numpy.real(n[band])))
@ -450,7 +454,10 @@ def find_k(
bounds=(k_min, k_max), bounds=(k_min, k_max),
options={'xatol': abs(tolerance)}, options={'xatol': abs(tolerance)},
) )
return res.x * direction, res.fun + frequency
assert n is not None
assert v is not None
return float(res.x * direction), float(res.fun + frequency), n, v
def eigsolve( def eigsolve(