ó
    Eñi  ã                   óF   • S SK rSSKJr  SSKJr  S/rSSSSSSSS	.S
 jjrg)é    Né   )Ú_get_atol_rtol)Úmake_systemÚtfqmrgñhãˆµøä>ç        F)ÚrtolÚatolÚmaxiterÚMÚcallbackÚshowc                ó6  • U R                   n	[        R                  " U	[        R                  5      (       a  [        n	U R                  U	5      n [        R                  " UR                   [        R                  5      (       a  UR                  U	5      n[        XX!5      u  pp¡[        R                  R                  U5      S:X  a  UR                  5       n
U
S4$ U R                  S   nUc  [        SUS-  5      nUc  UR                  5       nOXR                  U
5      -
  nUnUR                  5       nUnUR                  U R                  U5      5      nUnS=n=nn[        R                  " UR                  5       U5      R                  nUn[        R                   " U5      nUnUS:X  a  U
S4$ [#        SUXC5      u  nn[%        U5       GH°  nUS-  S:H  nU(       a>  [        R                  " UR                  5       U5      nUS:X  a  U
S4s  $ UU-  nUUU-  -
  nUWU-  -  nUUS-  U-  U-  U-  -   n[        R                  R                  U5      U-  n[        R                   " SS	US-  -   -  5      nUUU-  -  nUS-  U-  nUR                  U5      n U
UU -  -  n
Ub  U" U
5        U[        R                   " US	-   5      -  U:  a  U(       a  ['        S
US	-    35        U
S4s  $ U(       dh  [        R                  " UR                  5       U5      nUU-  n!UU!U-  -   nU!U-  U!S-  U-  -   nUR                  U R                  U5      5      nUU-  nGMŒ  UR                  U R                  W5      5      nUnUnGM³     U(       a  ['        SWS	-    35        X¥4$ )a)  
Solve ``Ax = b`` with the Transpose-Free Quasi-Minimal Residual method.

Parameters
----------
A : {sparse array, ndarray, LinearOperator}
    The real or complex N-by-N matrix of the linear system.
    Alternatively, `A` can be a linear operator which can
    produce ``Ax`` using, e.g.,
    `scipy.sparse.linalg.LinearOperator`.
b : {ndarray}
    Right hand side of the linear system. Has shape (N,) or (N,1).
x0 : {ndarray}
    Starting guess for the solution.
rtol, atol : float, optional
    Parameters for the convergence test. For convergence,
    ``norm(b - A @ x) <= max(rtol*norm(b), atol)`` should be satisfied.
    The default is ``rtol=1e-5``, the default for ``atol`` is ``0.0``.
maxiter : int, optional
    Maximum number of iterations.  Iteration will stop after maxiter
    steps even if the specified tolerance has not been achieved.
    Default is ``min(10000, ndofs * 10)``, where ``ndofs = A.shape[0]``.
M : {sparse array, ndarray, LinearOperator}
    Inverse of the preconditioner of A.  M should approximate the
    inverse of A and be easy to solve for (see Notes).  Effective
    preconditioning dramatically improves the rate of convergence,
    which implies that fewer iterations are needed to reach a given
    error tolerance.  By default, no preconditioner is used.
callback : function, optional
    User-supplied function to call after each iteration.  It is called
    as ``callback(xk)``, where ``xk`` is the current solution vector.
show : bool, optional
    Specify ``show = True`` to show the convergence, ``show = False`` is
    to close the output of the convergence.
    Default is `False`.

Returns
-------
x : ndarray
    The converged solution.
info : int
    Provides convergence information:

        - 0  : successful exit
        - >0 : convergence to tolerance not achieved, number of iterations
        - <0 : illegal input or breakdown

Notes
-----
The Transpose-Free QMR algorithm is derived from the CGS algorithm.
However, unlike CGS, the convergence curves for the TFQMR method is
smoothed by computing a quasi minimization of the residual norm. The
implementation supports left preconditioner, and the "residual norm"
to compute in convergence criterion is actually an upper bound on the
actual residual norm ``||b - Axk||``.

References
----------
.. [1] R. W. Freund, A Transpose-Free Quasi-Minimal Residual Algorithm for
       Non-Hermitian Linear Systems, SIAM J. Sci. Comput., 14(2), 470-482,
       1993.
.. [2] Y. Saad, Iterative Methods for Sparse Linear Systems, 2nd edition,
       SIAM, Philadelphia, 2003.
.. [3] C. T. Kelley, Iterative Methods for Linear and Nonlinear Equations,
       number 16 in Frontiers in Applied Mathematics, SIAM, Philadelphia,
       1995.

Examples
--------
>>> import numpy as np
>>> from scipy.sparse import csc_array
>>> from scipy.sparse.linalg import tfqmr
>>> A = csc_array([[3, 2, 0], [1, -1, 0], [0, 5, 1]], dtype=float)
>>> b = np.array([2, 4, -1], dtype=float)
>>> x, exitCode = tfqmr(A, b, atol=0.0)
>>> print(exitCode)            # 0 indicates successful convergence
0
>>> np.allclose(A.dot(x), b)
True
r   r   i'  é
   r   é   éÿÿÿÿg      ð?r   z:TFQMR: Linear solve converged due to reach TOL iterations z@TFQMR: Linear solve not converged due to reach MAXIT iterations )ÚdtypeÚnpÚ
issubdtypeÚint64ÚfloatÚastyper   ÚlinalgÚnormÚcopyÚshapeÚminÚmatvecÚinnerÚ	conjugateÚrealÚsqrtr   ÚrangeÚprint)"ÚAÚbÚx0r   r	   r
   r   r   r   r   ÚxÚndofsÚrÚuÚwÚrstarÚvÚuhatÚdÚthetaÚetaÚrhoÚrhoLastÚr0normÚtauÚ_ÚiterÚevenÚvtrstarÚalphaÚuNextÚcÚzÚbetas"                                     Ú^/home/mande/repo/quber/.venv/lib/python3.13/site-packages/scipy/sparse/linalg/_isolve/tfqmr.pyr   r   	   sY  € ðh �G‰G€EÜ	‡}‚}�UœBŸH™H×%Ñ%ÜˆØ�H‰H�U‹OˆÜ	‡}‚}�Q—W‘WœbŸh™h×'Ñ'Ø�H‰H�U‹Oˆä˜Q 2Ó)�J€Aˆ!ô 
‡y�y‡~�~�aÓ˜BÓØ�F‰F‹HˆØ�1ˆvˆà�G‰G�A‰J€EØ�Ü�e˜U R™ZÓ(ˆà	�zØ�F‰F‹H‰à—‘˜“‰OˆØ	€AØ	�‰‹€Aà€EØ	�‰�—‘˜!“Ó€AØ€DØÐ€AÐˆ�ä
�(Š(�5—?‘?Ó$ aÓ
(×
-Ñ
-€CØ€GÜ�WŠW�S‹\€FØ
€CØ�ƒ{Ø�1ˆvˆô ˜W f¨dÓ9�G€Dˆ!ä�g—ˆØ�a‰x˜1‰}ˆÞÜ—h’h˜uŸ™Ó0°!Ó4ˆGà˜"‹}Ø˜2�w’Ø˜'‘MˆEØ˜ ™	‘MˆEØ	ˆU�T‰\ÑˆØ�˜‘˜EÑ! SÑ(¨1Ñ,Ñ,ˆä—	‘	—‘˜qÓ! CÑ'ˆÜ�GŠG�B˜!˜e Q™h™,Ñ'Ó(ˆØˆu�q‰yÑˆà�!‰t�u‰nˆØ�H‰H�Q‹KˆØ	ˆS�1‰W‰ˆàÑÙ�QŒKð ”—’˜˜a™“Ñ  4Ó'ÞÜð $Ø$(¨¡F 8ð-ô .à�q�6ŠMæä—(’(˜5Ÿ?™?Ó,¨aÓ0ˆCØ˜‘=ˆDØ�D˜1‘H‘ˆAØ�t‘˜t Q™w¨!™mÑ+ˆAØ—8‘8˜AŸH™H Q›KÓ(ˆDØ�‰I‹Aà—8‘8˜AŸH™H U›OÓ,ˆDØˆAØ‹GñS öV 	Üð Ø  ™F˜8ð%ô 	&àˆ<Ðó    )N)Únumpyr   Ú	iterativer   Úutilsr   Ú__all__r   © r@   r?   Ú<module>rF      s0   ðÛ Ý %Ý ð ˆ)€ðj ¨B¸ÀØ˜e÷jr@   