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

allow initial value

Browse Source
This commit is contained in:
Jan Petykiewicz 2023-05-23 12:49:55 -07:00
parent 44465f1bc9
commit c7e823b0b3
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
meanas/fdfd/bloch.py
View File

@ -443,6 +443,7 @@ def find_k(
k_guess: float | None = None, k_guess: float | None = None,
solve_callback: Callable[..., None] | None = None, solve_callback: Callable[..., None] | None = None,
iter_callback: Callable[..., None] | None = None, iter_callback: Callable[..., None] | None = None,
v0: NDArray[numpy.complex128] | None = None,
) -> tuple[float, float, NDArray[numpy.complex128], NDArray[numpy.complex128]]: ) -> 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.
@ -475,7 +476,7 @@ def find_k(
k_guess = sum(k_bounds) / 2 k_guess = sum(k_bounds) / 2
n = None n = None
v = None v = v0
def get_f(k0_mag: float, band: int = 0) -> float: def get_f(k0_mag: float, band: int = 0) -> float:
nonlocal n, v nonlocal n, v