3
���h� � @ s� d dl Z ddlmZ ddlmZ ddlmZ ddlmZm Z m
Z
ddl m Z m
Z
mZmZ dd lmZ dd
lmZmZmZ d d � Zd
d� Ze jeddd�ZG dd� de�ZdS )� N� )�Variable)�
cf_encoder)�indexing)�FrozenOrderedDict�close_on_error�Frozen)� iteritems�
bytes_type� unicode_type� OrderedDict� )�WritableCFDataStore)�
_nc4_group�_nc4_values_and_dtype�_extract_nc4_encodingc C s t | t�r| jd�S | S d S )Nzutf-8)�
isinstancer
�decode)�txt� r �9/tmp/pip-build-5_djhm0z/xray/xarray/backends/h5netcdf_.py�maybe_decode_bytes
s
r c C s> t � }x2| j� D ]&}| j|�}|dkr.t|�}|||< qW |S )N�
_FillValue�
missing_value)r r )r ZncattrsZ getncattrr )Z h5netcdf_var�attrs�k�vr r r �_read_attributes s
r F�h5netcdf)Zlsd_okay�backendc sp e Zd ZdZd� fdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
ddd�Z � fdd�Z dd� Z
� ZS )�
H5NetCDFStorez4Store for reading and writing data via h5netcdf
�rNc sj dd l }|dkrtd��|jj||d�}t|�� t|||�| _W d Q R X || _|| _t t
| �j |� d S )Nr �NETCDF4z#invalid format for h5netcdf backend)�mode)Nr" ) Zh5netcdf.legacyapi�
ValueErrorZ legacyapiZDatasetr r �ds�format� _filename�superr �__init__)�self�filenamer# r& �group�writerr r% )� __class__r r r) ( s
zH5NetCDFStore.__init__c C sb |j }tj|�}t|�}t|j� �}|j� }|dkr8|nd |d< | j|d< |j|d< t ||||�S )N�
contiguous�
chunksizes�sourceZoriginal_shape)
�
dimensionsr ZLazilyIndexedArrayr �dict�filters�chunkingr' �shaper )r* �varr2 �datar �encodingr5 r r r �open_store_variable4 s
z!H5NetCDFStore.open_store_variablec s t � fdd�t� jj�D ��S )Nc 3 s | ]\}}|� j |�fV qd S )N)r: )�.0r r )r* r r � <genexpr>E s z.H5NetCDFStore.get_variables.<locals>.<genexpr>)r r r% � variables)r* r )r* r �
get_variablesD s zH5NetCDFStore.get_variablesc C s t t| j��S )N)r r r% )r* r r r � get_attrsH s zH5NetCDFStore.get_attrsc C s | j jS )N)r% r2 )r* r r r �get_dimensionsK s z H5NetCDFStore.get_dimensionsc C s | j j||d� d S )N)�size)r% ZcreateDimension)r* �name�lengthr r r �
set_dimensionN s zH5NetCDFStore.set_dimensionc C s | j j||� d S )N)r% � setncattr)r* �key�valuer r r �
set_attributeQ s zH5NetCDFStore.set_attributeFc C s� dd l }|jj� }t|�\}}|tkr2|jtd�}| j|� |jdd �}|d krTd }t ||d�}i } x d
D ]}
|
|krj||
| |
< qjW | j
j |||j fd |i| ��} x t
|�D ]\} }
| j| |
� q�W | |jfS )Nr )Zvlenr � )Zraise_on_invalid�zlib� complevel�shuffler0 �
fletcher32�
fill_value)rI )rJ rK rL r0 rM )�h5pyr �copyr �strZ
special_dtyper Zset_necessary_dimensions�pop�_extract_h5nc_encodingr% ZcreateVariableZdimsr rE r8 )r* rB �variableZcheck_encodingrO r ZdtyperN r9 �kwargsrF Znc4_varr r r r r �prepare_variableT s*
z H5NetCDFStore.prepare_variablec s t t| �j� | jj� d S )N)r( r �syncr% )r* )r. r r rW r s zH5NetCDFStore.syncc C s( | j }x|jd k r|j}qW |j� d S )N)r% �parent�close)r* r% r r r rY v s
zH5NetCDFStore.close)r! NNN)F)�__name__�
__module__� __qualname__�__doc__r) r: r> r? r@ rD rH rV rW rY �
__classcell__r r )r. r r % s
r )� functools� r Z conventionsr �corer Z
core.utilsr r r Z
core.pycompatr r
r r �commonr ZnetCDF4_r r r r r �partialrS r r r r r �<module> s
|