3
���h� � @ s� d Z ddlZddlmZ ddlmZ ddl m
Z ddl m
Z ddlmZ dd lmZ ee eefZejefe Zd
d � Zd d
� Zdj� ZxeD ]Zee�e � e< q�W dS )ar xarray 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� )�Variable)�Dataset)� DataArray)�GroupBy)�dask_array_type)�_dask_or_eager_funcc C s( x"t t�D ]\}}t| |�r
|S q
W dS )Nr ���)� enumerate�_dispatch_order�
isinstance)�obj�priority�cls� r �-/tmp/pip-build-5_djhm0z/xray/xarray/ufuncs.py�_dispatch_priority s
r c s2 � �fdd�� �� _ tt��j}d�|f � _� S )Nc s� | }t ��}t| �dks$t| �dkr:tdt| ��f ��n�t| �dkrdt| d t�r�| d j� �}n`tt| �\}}||kr�t| d t�r�| d j� �}n,t| d t�r�| d j� dd�}t t
| ��}|||�}|t kr�td�t | d �t | d �f ��|S )N� r z!cannot handle %s arguments for %rr T)Z reflexivez%%r not implemented for types (%r, %r))
r �len� TypeErrorr �
_xarray_typesZ _unary_op�mapr Z
_binary_op�tuple�reversed�NotImplemented�type)�args�kwargsZnew_args�f�p1�p2�res)�func�namer r r" ( s(
z_create_op.<locals>.funcz�xarray specific variant of numpy.%s. Handles xarray.Dataset, xarray.DataArray, xarray.Variable, numpy.ndarray and dask.array.Array objects with automatic dispatching.
Documentation from numpy:
%s)�__name__�getattr�_np�__doc__)r# �docr )r" r# r �
_create_op& s r) a� logaddexp logaddexp2 conj exp log log2 log10 log1p expm1 sqrt
square sin cos tan arcsin arccos arctan arctan2 hypot sinh cosh
tanh arcsinh arccosh arctanh deg2rad rad2deg logical_and
logical_or logical_xor logical_not maximum minimum fmax fmin
isreal iscomplex isfinite isinf isnan signbit copysign nextafter
ldexp fmod floor ceil trunc degrees radians rint fix angle real
imag fabs sign frexp fmod
) r' Znumpyr& Z
core.variabler Z _VariableZ core.datasetr Z_DatasetZcore.dataarrayr Z
_DataArrayZ core.groupbyr Z_GroupByZ
core.pycompatr Z_dask_array_typeZcore.opsr r Zndarrayr r r) �split�__all__r# �globalsr r r r �<module> s +
|