3
���hR �
@ s� d dl Z d dlZddlmZmZ ddlmZ ddlm Z m
Z
m Z dZ e
ddd d
d d d
dddddg �Zddd�Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� ZdS )$� N� )� conventions�Variable)�ops)�
basestring� unicode_type� OrderedDictz _.@+- !"#$%&\()*,:;<=>?[]^`{|}~Zbyte�char�shortZushort�intZuint�int64Zuint64Z floatreal�double�bool�stringZint32Zint8)r r c C s� t | j�}|tkrnt| }| j|�}d|ks4d|krB|| kj� sXd|krhtj|| � rhtd||f ��|} n | jjdkr�t j
j j | d�} | S )ae Coerce an array to a data type that can be stored in a netCDF-3 file
This function performs the following dtype conversions:
int64 -> int32
float64 -> float32
bool -> int8
unicode -> string
Data is checked for equality, or equivalence (non-NaN values) with
`np.allclose` with the default keyword arguments.
r �U�floatz/could not safely cast array from dtype %s to %szutf-8)
�str�dtype�_nc3_dtype_coercionsZastype�allr Zallclose_or_equiv�
ValueError�kind�np�coreZ defchararray�encode)Zarrr Z new_dtypeZcast_arr� r �7/tmp/pip-build-5_djhm0z/xray/xarray/backends/netcdf3.py�coerce_nc3_dtype s
r c C s> | j jdkr6| j jdkr6tj| �} |d| jd f }| |fS )N�S� zstring%s���)r r �itemsizer Zstring_to_char�shape)�data�dimsr r r �maybe_convert_to_char_array6 s
r% c C sD t | t�r t | t�s@| jd�} n ttj| ��} | jdkr@td��| S )Nzutf-8r z'netCDF attributes must be 1-dimensional) �
isinstancer r �decoder r Z
atleast_1d�ndimr )�valuer r r �encode_nc3_attr_value= s
r* c C s t dd� | j� D ��S )Nc S s g | ]\}}|t |�f�qS r )r* )�.0�k�vr r r �
<listcomp>I s z$encode_nc3_attrs.<locals>.<listcomp>)r �items)�attrsr r r �encode_nc3_attrsH s r1 c C s4 t | j�}t|| j�\}}t| j�}t|||| j�S )N)r r# r% r$ r1 r0 r �encoding)�varr# r$ r0 r r r �encode_nc3_variableM s
r4 c C s | j � pt| jd��dkS )zrReturn True if the given UTF-8 encoded character is alphanumeric
or multibyte.
Input is not checked!
zutf-8r )�isalnum�lenr )�cr r r �
_isalnumMUTF8T s r8 c C s� t | t�sdS t | t�s"| jd�} t| jd��}tjd| �| ko�| tko�|dko�d| ko�| d dko�t | d �s|| d dko�t
d d
� | D ��S ) ar Test whether an object can be validly converted to a netCDF-3
dimension, variable or attribute name
Earlier versions of the netCDF C-library reference implementation
enforced a more restricted set of characters in creating new names,
but permitted reading names containing arbitrary bytes. This
specification extends the permitted characters in names to include
multi-byte UTF-8 encoded Unicode and additional printing characters
from the US-ASCII alphabet. The first character of a name must be
alphanumeric, a multi-byte UTF-8 character, or '_' (reserved for
special names with meaning to implementations, such as the
"_FillValue" attribute). Subsequent characters may also include
printing special characters, except for '/' which is not allowed in
names. Names that have trailing space characters are also not
permitted.
Fzutf-8�NFCr �/r � �_c s s | ]}t |�p|tkV qd S )N)r8 �
_specialchars)r+ r7 r r r � <genexpr>y s z$is_valid_nc3_name.<locals>.<genexpr>r ) r& r r r' r6 r � unicodedata� normalize�_reserved_namesr8 r )�s� num_bytesr r r �is_valid_nc3_name] s
rD )r? Znumpyr � r r r r Z
core.pycompatr r r r= �setrA r r r% r* r1 r4 r8 rD r r r r �<module> s
|