o tBh@sdZgZddlmZmZmZmZmZddlm Z ddl m Z m Z m Z iddddd d d d d ddddd dd dd dd dd dd dd dd dd dd ZddZddZddZdS) zrestructuredtext en) asanyarrayasarrayarraymatrixzeros)asmatrix)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)rr )rr )rr )rrcCs$|dvrd}|dvr d}t||fS)NfdFDr ) _coerce_rules)xyrx/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/scipy/sparse/linalg/_isolve/utils.pycoerces  rcCs|SNrrrrridsrc s|}t|}|jd|jdkrtd|jf|jd}tj|dfks8j|fks8td|jjjjdvrCdfdd}t|d rS|jj}n| jj}t |jj}t |d  |d urt|d rx|j }nt}t|d r|j} nt} |tur| turt|j|jd}nt|j|| |jd}nt|}|j|jkrtd|d urt||d } n7t|tr|dkr̈} | | } n$t||d } | j|dfks| j|fkstd|jd| jd| } ||| |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=%sz(shapes of A {} and b {} are incompatiblerr csttr t|}|jSr) isinstancerrreshapeshaperbrr postprocessRs  z make_system..postprocessdtype)r Npsolverpsolve)rr )matvecrmatvecr z/matrix and preconditioner have different shapesMbz shapes of A z and x0 z are incompatible)rr ValueErrorrformatr charastypehasattrr#rrravelr!rr"r r rrstrcopyr) AMx0rA_Nrxtyper!r"rbCopyrrr make_system s`!                 r5N) __docformat____all__numpyrrrrrscipy.sparse._sputilsrscipy.sparse.linalg._interfacerr r rrrr5rrrrsF