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

[fdfd.solvers.generic] add option to pass a guess solution

Browse Source
This commit is contained in:
Jan Petykiewicz 2025-02-05 00:13:46 -08:00
parent c4f8749941
commit 777ecbc024
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
meanas/fdfd/solvers.py
View File

@ -69,11 +69,13 @@ def generic(
J: vcfdfield_t, J: vcfdfield_t,
epsilon: vfdfield_t, epsilon: vfdfield_t,
mu: vfdfield_t | None = None, mu: vfdfield_t | None = None,
*,
pec: vfdfield_t | None = None, pec: vfdfield_t | None = None,
pmc: vfdfield_t | None = None, pmc: vfdfield_t | None = None,
adjoint: bool = False, adjoint: bool = False,
matrix_solver: Callable[..., ArrayLike] = _scipy_qmr, matrix_solver: Callable[..., ArrayLike] = _scipy_qmr,
matrix_solver_opts: dict[str, Any] | None = None, matrix_solver_opts: dict[str, Any] | None = None,
E_guess: vcfdfield_t | None = None,
) -> vcfdfield_t: ) -> vcfdfield_t:
""" """
Conjugate gradient FDFD solver using CSR sparse matrices. Conjugate gradient FDFD solver using CSR sparse matrices.
@ -100,6 +102,8 @@ def generic(
which doesn't return convergence info and logs the residual which doesn't return convergence info and logs the residual
every 100 iterations. every 100 iterations.
matrix_solver_opts: Passed as kwargs to `matrix_solver(...)` matrix_solver_opts: Passed as kwargs to `matrix_solver(...)`
E_guess: Guess at the solution E-field. `matrix_solver` must accept an
`x0` argument with the same purpose.
Returns: Returns:
E-field which solves the system. E-field which solves the system.
@ -120,6 +124,13 @@ def generic(
A = Pl @ A0 @ Pr A = Pl @ A0 @ Pr
b = Pl @ b0 b = Pl @ b0
if E_guess is not None:
if adjoint:
x0 = Pr.H @ E_guess
else:
x0 = Pl @ E_guess
matrix_solver_opts['x0'] = x0
x = matrix_solver(A.tocsr(), b, **matrix_solver_opts) x = matrix_solver(A.tocsr(), b, **matrix_solver_opts)
if adjoint: if adjoint: