3
���h� � @ s� d dl mZ d dlZd dlZddlmZ ddlmZ ddl m
Z
m Z m Z ddl
mZmZ ddlmZ d d
lmZ d d lmZmZmZ d d
� Zdd� ZG dd� de�ZG dd� de�ZdS )� )�BytesION� )�Variable)�
cf_encoder)� iteritems�
basestring� OrderedDict)�Frozen�FrozenOrderedDict)�NumpyIndexingAdapter� )�WritableCFDataStore)�is_valid_nc3_name�encode_nc3_attr_value�encode_nc3_variablec C s t | t�r| jdd�S | S )Nzutf-8�replace)�
isinstance�bytes�decode)�s� r �6/tmp/pip-build-5_djhm0z/xray/xarray/backends/scipy_.py�_decode_string s
r c C s t dd� t| �D ��S )Nc s s* | ]"\}}||d kr|nt |�fV qdS )Z
_FillValueN)r )�.0�k�vr r r � <genexpr> s z _decode_attrs.<locals>.<genexpr>)r r )�dr r r �
_decode_attrs s r c s< e Zd Zdd� Zedd� �Zedd� �Z� fdd�Z� ZS ) �ScipyArrayWrapperc C s || _ || _d S )N)� netcdf_file�
variable_name)�selfr r! r r r �__init__ s zScipyArrayWrapper.__init__c C s | j j| j jS )N)r � variablesr! �data)r" r r r �array# s zScipyArrayWrapper.arrayc C s t j| jjjt| jjj� �S )N)�np�dtyper& �kind�str�itemsize)r" r r r r( + s zScipyArrayWrapper.dtypec s. t t| �j|�}| jj}tj|| j|d�}|S )N)r( �copy)�superr � __getitem__r Zuse_mmapr' r& r( )r" �keyr% r, )� __class__r r r. 0 s z ScipyArrayWrapper.__getitem__) �__name__�
__module__� __qualname__r# �propertyr& r( r. �
__classcell__r r )r0 r r s r c s� e Zd ZdZd � fdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z d dd�Z � fdd�Z
dd� Zdd � Z� ZS ) �ScipyDataStorez�Store for reading and writing data via scipy.io.netcdf.
This store has the advantage of being able to be initialized with a
StringIO object, allow for serialization without writing to disk.
It only supports the NetCDF3 file-format.
�rNc s� dd l }dd l}|dkr4|jdk r4tjd|j t� |d k rDtd��|d ksT|dkrZd}n|dkrhd }n td
| ��t|t�r�|j d �r�t
|�}|j j ||||d �| _
tt| �j|� d S )
Nr r7 z0.13zscipy %s detected; the minimal recommended version is 0.13. Older version of this library do not reliably read and write files.z7cannot save to a group with the scipy.io.netcdf backendZ
NETCDF3_64BITr ZNETCDF3_CLASSICr z.invalid format for scipy.io.netcdf backend: %rZCDF)�mode�mmap�version)�scipyZscipy.io� __version__�warnings�warn�
ImportWarning�
ValueErrorr r �
startswithr �ior �dsr- r6 r# ) r" Zfilename_or_objr8 �format�group�writerr9 r; r: )r0 r r r# B s&
zScipyDataStore.__init__c C s t |jt| j|�t|j��S )N)r �
dimensionsr rC r � _attributes)r" �name�varr r r �open_store_variablec s z"ScipyDataStore.open_store_variablec s t � fdd�t� jj�D ��S )Nc 3 s" | ]\}}|� j ||�fV qd S )N)rK )r r r )r" r r r h s z/ScipyDataStore.get_variables.<locals>.<genexpr>)r
r rC r$ )r" r )r" r �
get_variablesg s z ScipyDataStore.get_variablesc C s t t| jj��S )N)r r rC rH )r" r r r � get_attrsk s zScipyDataStore.get_attrsc C s t | jj�S )N)r rC rG )r" r r r �get_dimensionsn s z ScipyDataStore.get_dimensionsc C s. || j kr tdt| �j ��| jj||� d S )Nz(%s does not support modifying dimensions)rG r@ �typer1 rC ZcreateDimension)r" rI �lengthr r r �
set_dimensionq s
z ScipyDataStore.set_dimensionc C s t |�std��d S )NzNot a valid attribute name)r r@ )r" r/ r r r �_validate_attr_keyw s z!ScipyDataStore._validate_attr_keyc C s$ | j |� t|�}t| j||� d S )N)rR r �setattrrC )r" r/ �valuer r r �
set_attribute{ s
z ScipyDataStore.set_attributeFc C s� t |�}|r$|jr$tdt|j� ��| j|� |j}| jj||j|j � | jj
| }x,t |j �D ] \}}| j
|� t|||� q`W ||fS )Nz)unexpected encoding for scipy backend: %r)r �encodingr@ �listZset_necessary_dimensionsr% rC ZcreateVariabler( Zdimsr$ r �attrsrR rS )r" rI �variableZcheck_encodingr% Z scipy_varr r r r r �prepare_variable� s
z ScipyDataStore.prepare_variablec s t t| �j� | jj� d S )N)r- r6 �syncrC �flush)r" )r0 r r r[ � s zScipyDataStore.syncc C s | j j� d S )N)rC �close)r" r r r r] � s zScipyDataStore.closec C s | j � d S )N)r] )r" rO rT �tbr r r �__exit__� s zScipyDataStore.__exit__)r7 NNNN)F)r1 r2 r3 �__doc__r# rK rL rM rN rQ rR rU rZ r[ r] r_ r5 r r )r0 r r6 : s
r6 ) rB r Znumpyr' r= � r Z conventionsr Z
core.pycompatr r r Z
core.utilsr r
Z
core.indexingr �commonr
Znetcdf3r r r r r r r6 r r r r �<module> s |