ftjR6ddlmcmZddlmZddlmZddlmZddlm Z GddZ d e iZ d Z ed ddZ ed dZed ddZed dZe d d dS)N)xp_capabilities)_basic_backend)_realtransforms_backend)_fftlog_backendc,eZdZdZdZedZdS) _ScipyBackenda-The default backend for fft calculations Notes ----- We use the domain ``numpy.scipy`` rather than ``scipy`` because ``uarray`` treats the domain as a hierarchy. This means the user can install a single backend for ``numpy`` and have it implement ``numpy.scipy.fft`` as well. numpy.scipy.fftctt|jd}|tt|jd}|tt|jd}|t S||i|S)N)getattrr__name__rrNotImplemented)methodargskwargsfns P/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/scipy/fft/_backend.py__ua_function__z_ScipyBackend.__ua_function__sh^V_d ; ; :0&/4HHB :&/4@@B :! !r4"6"""N)r __module__ __qualname____doc__ __ua_domain__ staticmethodrrrr r s>&M # #\ # # #rr scipyct|tr4 t|}n%#t$r}t d||d}~wwxYw|jdkrt d|S)z8Maps strings to known backends and validates the backendzUnknown backend Nr z,Backend does not implement "numpy.scipy.fft") isinstancestr_named_backendsKeyError ValueErrorr)backendes r_backend_from_argr%%s'3B B%g.GG B B B999:: A B 111GHHH Ns % AAAT) out_of_scopeFcTt|}tj||||dS)a,Sets the global fft backend This utility method replaces the default backend for permanent use. It will be tried in the list of backends automatically, unless the ``only`` flag is set on a backend. This will be the first tried backend outside the :obj:`set_backend` context manager. Parameters ---------- backend : {object, 'scipy'} The backend to use. Can either be a ``str`` containing the name of a known backend {'scipy'} or an object that implements the uarray protocol. coerce : bool Whether to coerce input types when trying this backend. only : bool If ``True``, no more backends will be tried if this fails. Implied by ``coerce=True``. try_last : bool If ``True``, the global backend is tried after registered backends. Raises ------ ValueError: If the backend does not implement ``numpy.scipy.fft``. Notes ----- This will overwrite the previously set global backend, which, by default, is the SciPy implementation. Examples -------- We can set the global fft backend: >>> from scipy.fft import fft, set_global_backend >>> set_global_backend("scipy") # Sets global backend (default is "scipy"). >>> fft([1]) # Calls the global backend array([1.+0.j]) )coerceonlytry_lastN)r%uaset_global_backend)r#r(r)r*s rr,r,4s2R ((G'&thOOOOOOrcLt|}tj|dS)a Register a backend for permanent use. Registered backends have the lowest priority and will be tried after the global backend. Parameters ---------- backend : {object, 'scipy'} The backend to use. Can either be a ``str`` containing the name of a known backend {'scipy'} or an object that implements the uarray protocol. Raises ------ ValueError: If the backend does not implement ``numpy.scipy.fft``. Examples -------- We can register a new fft backend: >>> from scipy.fft import fft, register_backend, set_global_backend >>> class NoopBackend: # Define an invalid Backend ... __ua_domain__ = "numpy.scipy.fft" ... def __ua_function__(self, func, args, kwargs): ... return NotImplemented >>> set_global_backend(NoopBackend()) # Set the invalid backend as global >>> register_backend("scipy") # Register a new backend # The registered backend is called because # the global backend returns `NotImplemented` >>> fft([1]) array([1.+0.j]) >>> set_global_backend("scipy") # Restore global backend to default N)r%r+register_backendr#s rr.r.as)J ((G     rcNt|}tj|||S)aOContext manager to set the backend within a fixed scope. Upon entering the ``with`` statement, the given backend will be added to the list of available backends with the highest priority. Upon exit, the backend is reset to the state before entering the scope. Parameters ---------- backend : {object, 'scipy'} The backend to use. Can either be a ``str`` containing the name of a known backend {'scipy'} or an object that implements the uarray protocol. coerce : bool, optional Whether to allow expensive conversions for the ``x`` parameter. e.g., copying a NumPy array to the GPU for a CuPy backend. Implies ``only``. only : bool, optional If only is ``True`` and this backend returns ``NotImplemented``, then a BackendNotImplemented error will be raised immediately. Ignoring any lower priority backends. Examples -------- >>> import scipy.fft as fft >>> with fft.set_backend('scipy', only=True): ... fft.fft([1]) # Always calls the scipy implementation array([1.+0.j]) )r(r))r%r+ set_backend)r#r(r)s rr1r1s(: ((G >'&t < < <>> import scipy.fft as fft >>> fft.fft([1]) # Calls default SciPy backend array([1.+0.j]) >>> with fft.skip_backend('scipy'): # We explicitly skip the SciPy backend ... fft.fft([1]) # leaving no implementation available Traceback (most recent call last): ... BackendNotImplementedError: No selected backends had an implementation ... )r%r+ skip_backendr/s rr3r3s!4 ((G ?7 # ##r)r*)FFF)FF)scipy._lib.uarray_libuarrayr+scipy._lib._array_apirrrrr r r%r,r.r1r3rrrr9s111111%%%%%%########2 ]   d###)P)P)P$#)PXd###%!%!$#%!Pd###===$#=@d###$$$#$:7T******r