U |e @s~dZddlZddlZddlmZddlmZddlm Z m Z Gddde Z dddZ dddZdddZddZddZdS)z, Spectral Algorithm for Nonlinear Equations N)OptimizeResult)_check_unknown_options)_nonmonotone_line_search_cruz_nonmonotone_line_search_chengc@s eZdZdS)_NoConvergenceN)__name__ __module__ __qualname__r r U/var/www/website-v5/atlas_env/lib/python3.8/site-packages/scipy/optimize/_spectral.pyr srr :0yE>YnF 绽|=?cruzc* sBt|| dkrtd| fd| dkr6fdd} |dkrLfdd}fd d }d g}t||||||\}}}}}d }| }||}tg| }d }}d }d}||}|rtd|||f|dk r|||||||krd}d}qt|d| kr d| t|}nt|| kr2| }| |}| |||} zV| dkrpt ||||| d\}!}"}#}$n,| dkrt |||||| d\}!}"}#}$}}Wnt k rYqYnX|"|}%|$|}&t |%|%t |%|&}|"}|$}|#| dkr| |#|d7}qt|||d}'t||}(t|'|||(|d |d})|)S)a Solve nonlinear equation with the DF-SANE method Options ------- ftol : float, optional Relative norm tolerance. fatol : float, optional Absolute norm tolerance. Algorithm terminates when ``||func(x)|| < fatol + ftol ||func(x_0)||``. fnorm : callable, optional Norm to use in the convergence check. If None, 2-norm is used. maxfev : int, optional Maximum number of function evaluations. disp : bool, optional Whether to print convergence process to stdout. eta_strategy : callable, optional Choice of the ``eta_k`` parameter, which gives slack for growth of ``||F||**2``. Called as ``eta_k = eta_strategy(k, x, F)`` with `k` the iteration number, `x` the current iterate and `F` the current residual. Should satisfy ``eta_k > 0`` and ``sum(eta, k=0..inf) < inf``. Default: ``||F||**2 / (1 + k)**2``. sigma_eps : float, optional The spectral coefficient is constrained to ``sigma_eps < sigma < 1/sigma_eps``. Default: 1e-10 sigma_0 : float, optional Initial spectral coefficient. Default: 1.0 M : int, optional Number of iterates to include in the nonmonotonic line search. Default: 10 line_search : {'cruz', 'cheng'} Type of line search to employ. 'cruz' is the original one defined in [Martinez & Raydan. Math. Comp. 75, 1429 (2006)], 'cheng' is a modified search defined in [Cheng & Li. IMA J. Numer. Anal. 29, 814 (2009)]. Default: 'cruz' References ---------- .. [1] "Spectral residual method without gradient information for solving large-scale nonlinear systems of equations." W. La Cruz, J.M. Martinez, M. Raydan. Math. Comp. **75**, 1429 (2006). .. [2] W. La Cruz, Opt. Meth. Software, 29, 24 (2014). .. [3] W. Cheng, D.-H. Li. IMA J. Numer. Anal. **29**, 814 (2009). )chengrz"Invalid value %r for 'line_search'Ncsd|dS)Nrrr )kxF)f_0r r eta_strategyJsz#_root_df_sane..eta_strategycs dS)Nrr r)f_knexpr r fnormOsz_root_df_sane..fnormcstj|S)N)nplinalgnormr)rr r fmeritSsz_root_df_sane..fmeritrrFz&too many function evaluations requiredziter %d: ||F|| = %g, sigma = %gzsuccessful convergenceTrr)etar)shape)rsuccessmessagefunnfevnit)r ValueError _wrap_func collectionsdequeprintabsrsignrrrvdotappend _wrap_resultr)*funcx0argsftolfatolmaxfevrcallbackdispMrZ sigma_epsZsigma_0 line_searchunknown_optionsr"r(fx_kx_shapeZF_k is_complexrZsigma_kZF_0_normprev_fsQC convergedr&ZF_k_normdr#alphaxpfpFpZs_kZy_krrresultr )rrrr _root_df_sanesx1       &     rMc st|}|jt|f}t|p:t|}|}dd<|rxfdd}t|}t|}nfdd}|||||fS)a Wrap a function and an initial value so that (i) complex values are wrapped to reals, and (ii) value for a merit function fmerit(x, f) is computed at the same time, (iii) iteration count is maintained and an exception is raised if it is exceeded. Parameters ---------- func : callable Function to wrap x0 : ndarray Initial value fmerit : callable Merit function fmerit(f) for computing merit value from residual. nfev_list : list List to store number of evaluations in. Should be [0] in the beginning. maxfev : int Maximum number of evaluations before _NoConvergence is raised. args : tuple Extra arguments to func Returns ------- wrap_func : callable Wrapped function, to be called as ``F, fp = wrap_func(x0)`` x0_wrap : ndarray of float Wrapped initial value; raveled to 1-D and complex values mapped to reals. x0_shape : tuple Shape of the initial value array f : float Merit function at F F : ndarray of float Residual at x0_wrap is_complex : bool Whether complex values were mapped to reals rrcs`dkrtdd7<t|}t|f}t|}|}||fSNrr)r _real2complexreshaperasarrayravel _complex2real)rzvrr?r6r"r4r9 nfev_listZx0_shaper r wrap_funcs z_wrap_func..wrap_funccsTdkrtdd7<|}t|f}|}||fSrN)rrPrrQrR)rrr?rVr r rXs  )rrQr$rR iscomplexobjrS) r4r5r"rWr9r6rrBrXr rVr r+s(    r+cCs(|rt|}n|}|dk r$||}|S)zA Convert from real to complex and reshape result arrays. N)rOrP)rLrBr$rTr r r r3s   r3cCstj|tdtjSN)dtype)rascontiguousarrayfloatview complex128)rr r r rOsrOcCstj|tdtjSrZ)rr\complexr^float64)rTr r r rSsrS) r r rrNNFrNrrr)r )N)__doc__r,numpyrscipy.optimizerZscipy.optimize._optimizer _linesearchrr ExceptionrrMr+r3rOrSr r r r s*    J