U Cv f(@sddlmZddlZddlZddlZddlZddlmZm Z m Z m Z m Z ddl mZmZddlmZmZddlmZmZmZddlmZddlmZdd lmZmZmZmZm Z dd l!m"Z"d d Z#d dZ$Gddde Z%ddZ&Gddde Z'Gddde Z(e(ed<dS)) annotationsN)BACKEND_ENTRYPOINTS BackendArrayBackendEntrypointWritableCFDataStore_normalize_path)CachingFileManagerDummyFileManager) ensure_lockget_write_lock)encode_nc3_attr_valueencode_nc3_variableis_valid_nc3_name)StoreBackendEntrypoint)NumpyIndexingAdapter)Frozen FrozenDictclose_on_errormodule_available'try_read_magic_number_from_file_or_path)VariablecCst|tr|ddS|S)Nzutf-8replace) isinstancebytesdecode)sr:/tmp/pip-unpacked-wheel-h316xyqg/xarray/backends/scipy_.py_decode_string#s  rcCsdd|DS)NcSs&i|]\}}||dkr|nt|qS) _FillValue)r.0kvrrr ,sz!_decode_attrs..)items)drrr _decode_attrs)sr'c@s.eZdZddZd ddZddZdd Zd S) ScipyArrayWrappercCs>||_||_|j}|j|_t|jjt|jj |_dSN) datastore variable_name get_variabledatashapenpdtypekindstritemsize)selfr+r*arrayrrr__init__0s  zScipyArrayWrapper.__init__TcCs|jj|}|j|jSr))r*_manageracquire variablesr+)r4 needs_lockdsrrrr,7szScipyArrayWrapper.get_variablecCs.t|j|}|jjj}tj||j|dS)N)r0copy) rr,r-r*r;Zuse_mmapr/r5r0)r4keyr-r<rrr __getitem__;s zScipyArrayWrapper.__getitem__c Cs^|jjL|jdd}z |||<Wn,tk rN|tkrH||dd<nYnXW5QRXdS)NF)r:)r*lockr, TypeErrorEllipsis)r4r=valuer-rrr __setitem__Cs   zScipyArrayWrapper.__setitem__N)T)__name__ __module__ __qualname__r6r,r>rCrrrrr(/s r(c Csddl}t|trz|drzz|jjt||||dWStk rx}z"|j d}d|krft dnW5d}~XYnXt|t r| drt |}z|jj||||dWStk r}z.|j d}d|krd}||7}t|nW5d}~XYnXdS)Nr.gzmodemmapversionzis not a valid NetCDF 3 filez2gzipped file loading only supports NetCDF 3 files.CDFz If this is a NetCDF4 file, you may need to install the netcdf4 library, e.g., $ pip install netcdf4 )Zscipy.iorr2endswithioZ netcdf_filegzipopenr@args ValueErrorr startswithBytesIO)filenamerIrJrKscipyeerrmsgmsgrrr_open_scipy_netcdfPs2     rZc@seZdZdZd!ddZeddZdd Zd d Zd d Z ddZ ddZ d"ddZ ddZ ddZddZd#ddZddZdd ZdS)$ScipyDataStorezStore 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 Cs|dk rtd|dks |dkr&d}n|dkr4d}ntd||dkrd|dkrdt|trdt|}t||_t|trtt|||t||dd }nt||||d } t | }||_ dS) Nz7cannot save to a group with the scipy.io.netcdf backendZ NETCDF3_64BITZNETCDF3_CLASSICz,invalid format for scipy.io.netcdf backend: r\)rJrK)rIr?kwargsrH) rRrr2r r r?rrZdictr r7) r4filename_or_objrIformatgrouprJr?rKmanagerZ scipy_datasetrrrr6s4   zScipyDataStore.__init__cCs |jSr))r7r8r4rrrr;szScipyDataStore.dscCst|jt||t|jSr))r dimensionsr(r' _attributes)r4namevarrrropen_store_variables z"ScipyDataStore.open_store_variablecstfddjjDS)Nc3s"|]\}}|||fVqdSr))rjr rerr sz/ScipyDataStore.get_variables..)rr;r9r%rerrer get_variabless  zScipyDataStore.get_variablescCstt|jjSr))rr'r;rgrerrr get_attrsszScipyDataStore.get_attrscCs t|jjSr))rr;rfrerrrget_dimensionsszScipyDataStore.get_dimensionscCsddd|jjDiS)Nunlimited_dimscSsh|]\}}|dkr|qSr)rr rrr sz.ScipyDataStore.get_encoding..)r;rfr%rerrr get_encodingszScipyDataStore.get_encodingFcCs>||jjkr tt|jd|s(|nd}|j||dS)Nz& does not support modifying dimensions)r;rfrRtyperDZcreateDimension)r4rhlengthZ is_unlimitedZ dim_lengthrrr set_dimensions   zScipyDataStore.set_dimensioncCst|stddS)NzNot a valid attribute name)rrR)r4r=rrr_validate_attr_keysz!ScipyDataStore._validate_attr_keycCs$||t|}t|j||dSr))rur setattrr;)r4r=rBrrr set_attributes zScipyDataStore.set_attributecCs t|}|Sr))r )r4variablerrrencode_variableszScipyDataStore.encode_variablec Cs|r,|jr,|jddikr,tdt|j|j}||jjkrR|j||j|j|jj|}|j D]\}}| |t |||qht ||} | |fS)Nrz'unexpected encoding for scipy backend: )encodingrRlistr-r;r9ZcreateVariabler0Zdimsattrsr%rurvr() r4rhrxZcheck_encodingror-Z scipy_varr"r#targetrrrprepare_variables$     zScipyDataStore.prepare_variablecCs|jdSr))r;syncrerrrrszScipyDataStore.synccCs|jdSr))r7closererrrrszScipyDataStore.close)r\NNNN)F)FN)rDrErF__doc__r6propertyr;rjrlrmrnrqrtrurwryr~rrrrrrr[vs. "   r[c @s2eZdZdZedZdZdZddZd d d Z dS) ScipyBackendEntrypointa Backend for netCDF files based on the scipy package. It can open ".nc", ".nc4", ".cdf" and ".gz" files but will only be selected as the default if the "netcdf4" and "h5netcdf" engines are not available. It has the advantage that is is a lightweight engine that has no system requirements (unlike netcdf4 and h5netcdf). Additionally it can open gizp compressed (".gz") files. For more information about the underlying library, visit: https://docs.scipy.org/doc/scipy/reference/generated/scipy.io.netcdf_file.html See Also -------- backends.ScipyDataStore backends.NetCDF4BackendEntrypoint backends.H5netcdfBackendEntrypoint rVzAOpen netCDF files (.nc, .nc4, .cdf and .gz) using scipy in XarrayzWhttps://docs.xarray.dev/en/stable/generated/xarray.backends.ScipyBackendEntrypoint.htmlc Cs~t|}|dk r8|dr8t|}t|}W5QRX|dk rJ|dSztj|\}}Wntk rtYdSX|dkS)NsrLF>.nc4.ncrG.cdf)rrSrOrPospathsplitextr@)r4raZ magic_numberf_extrrrguess_can_open s  z%ScipyBackendEntrypoint.guess_can_openTNr\c CsTt|}t|| | | | | d}t}t| |j||||||||d}W5QRX|S)N)rIrbrcrJr?)mask_and_scale decode_timesconcat_characters decode_coordsdrop_variables use_cftimedecode_timedelta)rr[rr open_dataset)r4rarrrrrrrrIrbrcrJr?storeZstore_entrypointr;rrrrs,  z#ScipyBackendEntrypoint.open_dataset) TTTTNNNr\NNNN) rDrErFrr available descriptionurlrrrrrrrs$rrV)) __future__rrOrNrZnumpyr/Zxarray.backends.commonrrrrrZxarray.backends.file_managerrr Zxarray.backends.locksr r Zxarray.backends.netcdf3r r rZxarray.backends.storerZxarray.core.indexingrZxarray.core.utilsrrrrrZxarray.core.variablerrr'r(rZr[rrrrrs&    !&}M