tKgdZgZddlmZmZmZmZddlmZm 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 e/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/scipy/sparse/linalg/_isolve/utils.pycoercers)   1 c|SNrs ridrs Hrc|}t|}|jd|jdk7rtd|jf|jd}t|}|j|dfk(s6|j|fk(s&td|jd|jd|jj dvr|j d}d }t|d r|jj }n%|j|jj }t||jj }t|| }|j}|t|d r |j}nt}t|d r |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)aMake 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, postprocess) A : LinearOperator matrix of the linear system M : LinearOperator preconditioner x : rank 1 ndarray initial guess b : rank 1 ndarray right hand side postprocess : function converts the solution vector to the appropriate type and dimensions (e.g. (N,1) matrix) rz&expected square matrix, but got shape=z shapes of A z and b z are incompatiblerr c|Srrrs r postprocessz make_system..postprocessQsrdtype)rpsolverpsolve)shaper)matvecrmatvecrz/matrix and preconditioner have different shapesMbz and x0 )rr" ValueErrorrrcharastypehasattrr#rrravelr rr!r rr isinstancestrcopyr) AMx0bA_Nrxtyper r!rbCopys r make_systemr6slB BAwwqzQWWQZA177*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 rrrr6rrrr;s% 43 3 # y 3 )# /8 3 )# /8 3 !*# 09 3  !*#  09  3   `#r