3
���h
� @ sL d Z ddlZddlmZ ddlmZmZmZ e ddg�Z
dd � Z d
d � Z dS ) z4Functions for converting to and from xarray objects
� N� )� DataArray)�maybe_encode_timedelta�maybe_encode_datetime� decode_cf�nameZ tileIndexc sX dd� � t j| �}| j}� fdd�| j� D �}� | �}t||||d�}t|j� �|j S )z/Convert a cdms2 variable into an DataArray
c S s t dd� | jj� D ��S )Nc s s" | ]\}}|t kr||fV qd S )N)�
ignored_attrs)�.0�k�v� r �./tmp/pip-build-5_djhm0z/xray/xarray/convert.py� <genexpr> s z6from_cdms2.<locals>.get_cdms2_attrs.<locals>.<genexpr>)�dict�
attributes�items)�varr r r
�get_cdms2_attrs s z#from_cdms2.<locals>.get_cdms2_attrsc s" g | ]}|j tj|�� |�f�qS r )�id�np�asarray)r r )r r r
�
<listcomp> s z from_cdms2.<locals>.<listcomp>)�coordsr �attrs)r r r Z getAxisListr r Z
to_datasetr )�variable�valuesr r r � dataarrayr )r r
�
from_cdms2 s
r c
C s� ddl }dd� }dd� }g }xB| jD ]8}|| j| �}|j|j|d�}|||j� |j|� q$W || �}|j|j|| jd�} || |j� | S ) z.Convert a DataArray into a cdms2 variable
r Nc S s t t| j��S )N)r r r )r r r r
�encode" s zto_cdms2.<locals>.encodec S s& x |j � D ]\}}t| ||� q
W d S )N)r �setattr)r r r
r r r r
�set_cdms2_attrs% s z!to_cdms2.<locals>.set_cdms2_attrs)r )�axesr ) �cdms2Zdimsr Z
createAxisr r �appendZcreateVariabler )
r r" r r r! ZdimZcoordZaxisr Z cdms2_varr r r
�to_cdms2 s r$ )
�__doc__Znumpyr Zcore.dataarrayr Z conventionsr r r �setr r r$ r r r r
�<module> s |