_-Phm <dZddlZddlmZddlmZddlmZdZ dS)zRelaxation-centric utilities.N)LinearOperator)MultilevelSolver) relaxationc d}gd}tjj||\}}tj}|_||vrt d| ttj d|z}n%#t $r} t d|| d} ~ wwxYw||fi|  fd} tj | jS) aCreate a linear operator that applies a relaxation method to a right-hand-side. Parameters ---------- methods : {tuple or string} Relaxation descriptor: Each tuple must be of the form ('method','opts') where 'method' is the name of a supported smoother, e.g., gauss_seidel, and 'opts' a dict of keyword arguments to the smoother, e.g., opts = {'sweep':symmetric}. If string, must be that of a supported smoother, e.g., gauss_seidel. Returns ------- linear operator that applies the relaxation method to a vector for a fixed right-hand-side, b. Notes ----- This method is primarily used to improve B during the aggregation setup phase. Here b = 0, and each relaxation call can improve the quality of B, especially near the boundaries. Examples -------- >>> from pyamg.gallery import poisson >>> from pyamg.relaxation.utils import relaxation_as_linear_operator >>> import numpy as np >>> A = poisson((100,100), format='csr') # matrix >>> B = np.ones((A.shape[0],1)) # Candidate vector >>> b = np.zeros((A.shape[0])) # RHS >>> relax = relaxation_as_linear_operator('gauss_seidel', A, b) >>> B = relax*B cTt|tr|d|dfS|ifS)Nr) isinstancetuple)vs V/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pyamg/relaxation/utils.py unpack_argz1relaxation_as_linear_operator..unpack_arg-s/ a   Q41: "u ) gauss_seidelblock_gauss_seidelsorgauss_seidel_negauss_seidel_nrjacobi block_jacobi richardsonschwarzstrength_based_schwarz jacobi_ne)dtypezinvalid relaxation method: setup_zinvalid presmoother method: NcJ|}||S)N)copy)xxcopyAbrelaxs r matvecz-relaxation_as_linear_operator..matvecIs) a r) nparrayrrLevelr! NameErrorgetattrr smoothingrshape) methodr!r"raccepted_methodsfnkwargslvlsetup_smootherer$r#s `` @r relaxation_as_linear_operatorr3 s=H ??? !'"""AF##JB  " "C CE !!!:b::;;;D !5x"}EE DDD;r;;<r:sk##......))))))E:E:E:E:E:r