S_f4XddlZddlmZmZdZdZdZdZdZ dZ d dZ dd Z d Z d Zd Zy)N) _RichResult_call_callback_maybe_haltc t|}tjg||}tj|Dcgc]}|jc}}tj |tj rtjn|}|d|||d}}|D cgc]} | j|dd}} |D cgc]} tj|| g| } } |dj} | dj|re| |dfd }tj| } |D cgc]} tj| | }} |D cgc]} tj| | }} d} |d | j} | Dcgc]}|j| k(}}tj|s t| tj| Dcgc]}|jc}|gz}|s/tj |tj s td |D cgc]} | j|d d}} | Dcgc]}|j|d d} }|D cgc]} | j#}} | Dcgc]}|j#} }|D cgc]} | j%}} ||| || |fScc}wcc} wcc} wcc} wcc} wcc}wcc}wcc} wcc}wcc} wcc}wcc} w) aInitialize abscissa, function, and args arrays for elementwise function Parameters ---------- func : callable An elementwise function with signature func(x: ndarray, *args) -> ndarray where each element of ``x`` is a finite real and ``args`` is a tuple, which may contain an arbitrary number of arrays that are broadcastable with ``x``. xs : tuple of arrays Finite real abscissa arrays. Must be broadcastable. args : tuple, optional Additional positional arguments to be passed to `func`. preserve_shape : bool, default:False When ``preserve_shape=False`` (default), `func` may be passed arguments of any shape; `_scalar_optimization_loop` is permitted to reshape and compress arguments at will. When ``preserve_shape=False``, arguments passed to `func` must have shape `shape` or ``shape + (n,)``, where ``n`` is any integer. Returns ------- xs, fs, args : tuple of arrays Broadcasted, writeable, 1D abscissa and function value arrays (or NumPy floats, if appropriate). The dtypes of the `xs` and `fs` are `xfat`; the dtype of the `args` are unchanged. shape : tuple of ints Original shape of broadcasted arrays. xfat : NumPy dtype Result dtype of abscissae, function values, and args determined using `np.result_type`, except integer types are promoted to `np.float64`. Raises ------ ValueError If the result dtype is not that of a real scalar Notes ----- Useful for initializing the input of SciPy functions that accept an elementwise callable, abscissae, and arguments; e.g. `scipy.optimize._chandrupatla`. NF)copyr)shapefunccVdtt|z z}|||g|i|S)N)r)len)xr rargskwargsifshapes Hlib/python3.12/site-packages/scipy/_lib/_elementwise_iterative_method.pyrz_initialize..funcYs3c&kCJ./A!.t.v. .zpThe shape of the array returned by `func` must be the same as the broadcasted shape of `x` and all other `args`.zWhen `preserve_shape=False`, z3Abscissae and function output must be real numbers.T)rnpbroadcast_arrays result_typedtype issubdtypeintegerfloat64astypeasarrayr broadcast_shapes broadcast_tolowerall ValueErrorfloatingravelflatten)rxsr complex_okpreserve_shapenxxasxaxatrfsr argmessagef shapes_equalxfatrs @r _initializer6s^ RB   )r )D )C ..c22882 3C c2::6"**CC3BxRSB13 4A!((3U( #B ' 4B 4.0 1"**T!^d^ $ 1B 1 qEKKE U[[F!&T /##FE213 4Abooa' 4 47;<U+<<DG!1'--/1BC.01AGGu$1L1 66, !! >>b1QWW1SE9 ;D bmmD"++>NOO13 4A!((4d( #B ' 4B 413 4A!((4d( #B ' 4B 4 !'') B  !'') B %) *cCKKM *D * Rud **O3 5 15< 22 5 4 ! *sGJ6J;5#KKK :KKK8KK#8K(K-c d}ttj|}tj|}| Dcic]\}}|tj||}}}tj|t |d<tj |t|d<tj|t|d<tj|t|d<t|}||_ t||| || | }|t||| ||| | }t||rd}|j|kr|jr|s|r||}|jr|jd j|jk7rxtj|jtj }|jDcgc].}tj"|t%||jd0c}|_ |j&}| r|j)|d z}||g|j}tj*||}| r"|j)|}|j)|}|xj,|jd k(rd n|j&d z c_|||||xjd z c_ t||| || | }| t||| ||| | }t||rd}n9|jd k(rn)| ||j|kr|jr|s|r|rt.nt0|j2ddt||| ||| | Scc}}wcc}w) a Main loop of a vectorized scalar optimization algorithm Parameters ---------- work : _RichResult All variables that need to be retained between iterations. Must contain attributes `nit`, `nfev`, and `success` callback : callable User-specified callback function shape : tuple of ints The shape of all output arrays maxiter : Maximum number of iterations of the algorithm func : callable The user-specified callable that is being optimized or solved args : tuple Additional positional arguments to be passed to `func`. dtype : NumPy dtype The common dtype of all abscissae and function values pre_func_eval : callable A function that accepts `work` and returns `x`, the active elements of `x` at which `func` will be evaluated. May modify attributes of `work` with any algorithmic steps that need to happen at the beginning of an iteration, before `func` is evaluated, post_func_eval : callable A function that accepts `x`, `func(x)`, and `work`. May modify attributes of `work` with any algorithmic steps that need to happen in the middle of an iteration, after `func` is evaluated but before the termination check. check_termination : callable A function that accepts `work` and returns `stop`, a boolean array indicating which of the active elements have met a termination condition. post_termination_check : callable A function that accepts `work`. May modify `work` with any algorithmic steps that need to happen after the termination check and before the end of the iteration. customize_result : callable A function that accepts `res` and `shape` and returns `shape`. May modify `res` (in-place) according to preferences (e.g. rearrange elements between attributes) and modify `shape` if needed. res_work_pairs : list of (str, str) Identifies correspondence between attributes of `res` and attributes of `work`; i.e., attributes of active elements of `work` will be copied to the appropriate indices of `res` when appropriate. The order determines the order in which _RichResult attributes will be pretty-printed. Returns ------- res : _RichResult The final result object Notes ----- Besides providing structure, this framework provides several important services for a vectorized optimization algorithm. - It handles common tasks involving iteration count, function evaluation count, a user-specified callback, and associated termination conditions. - It compresses the attributes of `work` to eliminate unnecessary computation on elements that have already converged. F)rsuccessstatusnitnfevNTr)rrr)intrprodarangezerosboolfull _EINPROGRESSrr_check_termination_prepare_resultrr:sizendimint64 expand_dimstupler reshaper r; _ECALLBACK _ECONVERRr9)workcallbackr maxiterrrr pre_func_evalpost_func_evalcheck_terminationpost_termination_checkcustomize_resultres_work_pairsr+ cb_terminate n_elementsactiverjres_dictrestemprdimsr1x_shaper3s r_loopr_ysFLRWWU^$J YYz "FAOPA288Je44PHP((:T:HY\:HXhhz5HUOxx #6HV h CDI c>6 1>CFtS.&%/A $Xt 4L ((W \j $  991**aff499QVV2884D$(II/ U4 ?-CD/DI''  %%-(A  TYY  JJq &  '"A '"A !&&A+Q1772;6 q!T" A #D#~v$5~G  "4nfe#3^ED(48#  ;;!  t$I ((W \jL$0ZYDKKN 4nfe+^ ==qQ2/s "M43McH||}tj|rt|||||||r||}|}||}|s^|jD])\}} t | tj r| |n| ||<+|j D cgc]} | | c} |_|Scc} w)N)rany_update_activeitems isinstancendarrayr) rMr[rUrXrRr+stopproceedkeyvalr1s rrCrCs T "D vvd| tS.&$O rysN9       ^+F