U Cv fC @sddlmZddlZddlZddlmZddlmZdZ dddd d d d d dddddh Z dddddddZ dZ ddZ ddZddZddZd d!Zd"d#ZdS)$) annotationsN)coding)Variablez_.@+- !"#$%&\()*,:;<=>?[]^`{|}~bytecharshortZushortintZuintint64uint64floatrealdoubleboolstringZint32Zint16Zint8)r r Zuint32Zuint16Zuint8rutf-8cCsLt|j}|tkrHt|}||}||ksDtd|d||}|S)aCoerce an array to a data type that can be stored in a netCDF-3 file This function performs the dtype conversions as specified by the ``_nc3_dtype_coercions`` mapping: int64 -> int32 uint64 -> int32 uint32 -> int32 uint16 -> int16 uint8 -> int8 bool -> int8 Data is checked for equality, or equivalence (non-NaN values) using the ``(cast_array == original_array).all()``. z'could not safely cast array from dtype z to )strdtype_nc3_dtype_coercionsZastypeall ValueError)ZarrrZ new_dtypeZcast_arrr;/tmp/pip-unpacked-wheel-h316xyqg/xarray/backends/netcdf3.pycoerce_nc3_dtype/s   rcCsFt|tr n6t|tr"|t}n tt|}|jdkrBt d|S)Nz'netCDF attributes must be 1-dimensional) isinstancebytesrencodeSTRING_ENCODINGrnpZ atleast_1dndimr)valuerrrencode_nc3_attr_valueKs    r!cCsdd|DS)NcSsi|]\}}|t|qSr)r!).0kvrrr Xsz$encode_nc3_attrs..)items)attrsrrrencode_nc3_attrsWsr(cCsNtjjddtjfD]}||}qt|j}t|j}t |j |||j S)NF)Zallows_unicode) rstringsZEncodedStringCoderZCharacterArrayCoderrrdatar(r'rZdimsencoding)varZcoderr*r'rrrencode_nc3_variable[s    r-cCs|pt|ddkS)zrReturn True if the given UTF-8 encoded character is alphanumeric or multibyte. Input is not checked! rr)isalnumlenr)crrr _isalnumMUTF8fsr1cCszt|tsdSt|d}td||kox|tkox|dkoxd|kox|ddkoxt|dsh|ddkoxtd d |DS) arTest 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. FrNFCr/ _css|]}t|p|tkVqdS)N)r1 _specialchars)r"r0rrr sz$is_valid_nc3_name..) rrr/r unicodedata normalize_reserved_namesr1r)s num_bytesrrris_valid_nc3_nameos   r>) __future__rr9ZnumpyrZxarrayrZxarray.core.variablerr7r;rrrr!r(r-r1r>rrrrsB