^j dZgZddlmZmZmZmZddlmZm Z m Z idddddd d d d dd ddd dd dd dd dd dd dd dd dd dd Z dZ dZ dZy)zrestructuredtext en) asanyarrayasarrayarrayzeros)aslinearoperatorLinearOperatorIdentityOperator)fr r )r dr )r Fr )r Dr )r r )r r )r r )r r )r r )r r )r r )r r )r r )r r )r r )r r c0|dvrd}|dvrd}t||fS)NfdFDr ) _coerce_rules)xys Z/opt/ringagent/.cad-venv/lib/python3.12/site-packages/scipy/sparse/linalg/_isolve/utils.pycoercers)   1 c|S)N)rs ridrs Hrc|}t|}|jdkDrt|jd|jd|jdk7rtd|jf|jd}t |}|j|dfk(s6|j|fk(s&td|jd|jd|j j d vr|jd }t|d r|j j }n%|j|j j }t||j j }t|| }|j}|t|d r |j}nt}t|dr |j}nt}|tur*|tur"t!|j|j }nSt#|j|||j }n/t|}|j|jk7r td|t%|| } nt'|t(r'|dk(r|j+} |j| } ndt-|| } | j|dfk(s6| j|fk(s&td|jd| jd| j} || |fS)a.Make a linear system Ax=b Parameters ---------- A : LinearOperator sparse or dense matrix (or any valid input to aslinearoperator) M : {LinearOperator, Nones} preconditioner sparse or dense matrix (or any valid input to aslinearoperator) x0 : {array_like, str, None} initial guess to iterative method. ``x0 = 'Mb'`` means using the nonzero initial guess ``M @ b``. Default is `None`, which means using the zero initial guess. b : array_like right hand side Returns ------- (A, M, x, b) A : LinearOperator matrix of the linear system M : LinearOperator preconditioner x : rank 1 ndarray initial guess b : rank 1 ndarray right hand side z.-dimensional `A` is unsupported, expected 2-D.rz&expected square matrix, but got shape=z shapes of A z and b z are incompatiblerr dtype)rpsolverpsolve)shaper)matvecrmatvecrz/matrix and preconditioner have different shapesMbz and x0 )rndim ValueErrorrrrcharastypehasattrr rrravelrrrr rr isinstancestrcopyr) AMx0bA_NxtyperrrbCopys r make_systemr4s< BAvvzAFF8#QRSSwwqzQWWQZA177*NOO  A1 A GG!u A4<yyA(() ) ww||6! HHSMq  !!&& 5!'',, 'EA  A y 2h YYFF 2i jjGG RAqwwvw%&WW.A Q  77agg NO O z !5 ! B  :FFHEA "E "Aq6!QWW_|AGG95##$77)+<>? ? GGI aA:rN) __docformat____all__numpyrrrrscipy.sparse.linalg._interfacerrr rrrr4rrrr9s% 43 3 # y 3 )# /8 3 )# /8 3 !*# 09 3  !*#  09  3   \r