allow initial value

master
Jan Petykiewicz 1 year ago
parent 44465f1bc9
commit c7e823b0b3

@ -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

Loading…
Cancel
Save