a İa@sdZddlZddlZddlZddlmZddl m Z ddl m ZddlmZddlmZeee efZejefeZeZd d ZGd d d Zd dZddZddZddZdZ e D]Z!ee!e"e!<qdS)arxarray specific universal functions Handles unary and binary operations for the following types, in ascending priority order: - scalars - numpy.ndarray - dask.array.Array - xarray.Variable - xarray.DataArray - xarray.Dataset - xarray.core.groupby.GroupBy Once NumPy 1.10 comes out with support for overriding ufuncs, this module will hopefully no longer be necessary. N) DataArray)Dataset)GroupBy)dask_array_type)VariablecCs(ttD]\}}t||r|SqdS)N) enumerate_dispatch_order isinstance)objpriorityclsr,lib/python3.9/site-packages/xarray/ufuncs.py_dispatch_priority s  rc@s eZdZdZddZddZdS)_UFuncDispatcherzWrapper for dispatching ufuncs.cCs ||_dSN)_name)selfnamerrr__init__*sz_UFuncDispatcher.__init__cOsF|jdvrtjdtdd|}t}t|dks:t|dkrRtdt||jnt|dkr|t|dt r|d |}nlt t |\}}||krt|dt r|d |d|}n2t|dt r|dj |d|dd }tt|}|tur tt|j}||i|}|turBt|jd t|dd t|dd |S) N)angle iscomplexz@xarray.ufuncs is deprecated. Instead, use numpy ufuncs directly.) stacklevelrz#cannot handle {} arguments for {!r}rT)Z reflexivez not implemented for types (z, ))r _warningswarn FutureWarning _UNDEFINEDlen TypeErrorformatr _xarray_typesZ _unary_opmaprZ _binary_optuplereversedgetattr_npNotImplementedtype)rargskwargsnew_argsresZp1Zp2frrr__call__-s<      &z_UFuncDispatcher.__call__N)__name__ __module__ __qualname____doc__rr1rrrrr'srcCs6t|ts|S||r2|d}||dd}|S)Nz r)r str startswithfind)docrZ signature_endrrr_skip_signatureRs    r:cCsdt|ts|Sd}t|D]D}d|d}d|d}|d}||vs||vrRq|||}q|S)Nz.. [][z]_z. )r r6rangereplace)r9Zmax_referencesZnumZlabelZ referenceindexrrr_remove_unused_reference_labels]s     rAcCst|ts|St|Sr)r r6textwrapdedent)r9rrr_dedentos rDcCsBt|}||_tt|j}ttt||}d|d||_|S)Nz!xarray specific variant of numpy.z. Handles xarray.Dataset, xarray.DataArray, xarray.Variable, numpy.ndarray and dask.array.Array objects with automatic dispatching. Documentation from numpy: )rr2r(r)r5rAr:rD)rfuncr9rrr _create_opvs rF);rZarccosZarccoshZarcsinZarcsinhZarctanZarctan2ZarctanhZceilZconjZcopysignZcosZcoshZdeg2radZdegreesZexpZexpm1ZfabsZfixZfloorZfmaxZfminfmodrGZfrexpZhypotimagrZisfiniteZisinfZisnanZisrealZldexplogZlog10Zlog1pZlog2Z logaddexpZ logaddexp2Z logical_andZ logical_notZ logical_orZ logical_xorZmaximumZminimumZ nextafterZrad2degZradiansrealZrintsignZsignbitZsinZsinhZsqrtZsquareZtanZtanhZtrunc)#r5rBwarningsrZnumpyr)Zcore.dataarrayrZ _DataArrayZ core.datasetrZ_DatasetZ core.groupbyrZ_GroupByZ core.pycompatrZ_dask_array_typeZ core.variablerZ _Variabler$Zndarrayr objectr rrr:rArDrF__all__rglobalsrrrrs(      + ?