B ` @sdZgZddlmZmZmZmZmZddlm Z ddl m Z m Z m Z ddddddddddddddddd Zd d Zd d ZddZdS)zrestructuredtext en) asanyarrayasarrayarraymatrixzeros)asmatrix)aslinearoperatorLinearOperatorIdentityOperatorfdFD))r r )r r )r r )r r)r r )r r )r r )r r)r r )r r )r r )r r)rr )rr )rr )rrcCs$|dkr d}|dkrd}t||fS)NfdFDr ) _coerce_rules)xyrO/home/dcms/DCMS/lib/python3.7/site-packages/scipy/sparse/linalg/isolve/utils.pycoerces rcCs|S)Nr)rrrridsrc s|}t|}|jd|jdkr0td|jf|jd}tj|dfksdj|fksdtdjjdkrzdfdd}t|d r|jj}n|jj}t |jj}t |d  |d krt ||d }n:t ||d }|j|dfks|j|fkstd | }|d krt|d r8|j} nt} t|drP|j} nt} | tkrz| tkrzt|j|jd}nt|j| | |jd}nt|}|j|jkrtd||||fS)a8Make 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, None} initial guess to iterative method 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$A and b have incompatible dimensionsrr csttrt|}|jS)N) isinstancerrZreshapeshape)r)brr postprocessOs z make_system..postprocessdtype)rNz$A and x have incompatible dimensionspsolverpsolve)rr)matvecZrmatvecrz/matrix and preconditioner have different shapes)rr ValueErrorrrcharZastypehasattrrrrZravelrrrrrr r ) AMZx0rZA_NrZxtyperrrr)rr make_system sL             r&N)Z __docformat____all__numpyrrrrrZscipy.sparse.sputilsrZscipy.sparse.linalg.interfacerr r rrrr&rrrrs