d*vddlmZddlZddlZddlZddlmZddlmZm Z ddl Z ddl m Z mZmZmZmZddlmZmZddlmZmZddlmZmZmZdd lmZdd lmZdd l m!Z!m"Z"m#Z#m$Z$dd l%m&Z&erdd lm'Z'ddl m(Z(ddl)m*Z*dZ+dZ,GddeZ-dZ.GddeZ/GddeZ0de0fe d<dS)) annotationsN)Iterable) TYPE_CHECKINGAny)BACKEND_ENTRYPOINTS BackendArrayBackendEntrypointWritableCFDataStore_normalize_path)CachingFileManagerDummyFileManager) ensure_lockget_write_lock)encode_nc3_attr_valueencode_nc3_variableis_valid_nc3_name)StoreBackendEntrypoint)NumpyIndexingAdapter)Frozen FrozenDictclose_on_error'try_read_magic_number_from_file_or_path)Variable)BufferedIOBase)AbstractDataStore)Datasetc\t|tr|ddS|S)Nzutf-8replace) isinstancebytesdecode)ss 6lib/python3.11/site-packages/xarray/backends/scipy_.py_decode_stringr$*s-!U,xx+++ Hc>d|DS)NcDi|]\}}||dkr|nt|S) _FillValue)r$.0kvs r# z!_decode_attrs..3s4 U U U!QAA%%qq>!+<+< U U Ur%)items)ds r# _decode_attrsr00s! V U17799 U U UUr%c(eZdZdZddZdZdZdS)ScipyArrayWrapperc||_||_|j}|j|_t j|jjt|jj z|_dSN) datastore variable_name get_variabledatashapenpdtypekindstritemsize)selfr6r5arrays r#__init__zScipyArrayWrapper.__init__7sZ"*!!##([ Xek.U[5I1J1JJKK r%Tcd|jj|}|j|jSr4)r5_manageracquire variablesr6)r? needs_lockdss r#r7zScipyArrayWrapper.get_variable>s* ^ $ , ,Z 8 8|D.//r%ct|j|}|jjj}t j||j|S)N)r;copy) rr7r8r5rGuse_mmapr:r@r;)r?keyr8rIs r# __getitem__zScipyArrayWrapper.__getitem__BsK#D$5$5$7$7$<==cB~ )xDJT::::r%c|jj5|d} |||<n"#t$r|tur||dd<nYnwxYwddddS#1swxYwYdS)NF)rF)r5lockr7 TypeErrorEllipsis)r?rKvaluer8s r# __setitem__zScipyArrayWrapper.__setitem__Js ^  $$$66D !S    (??#DGGG                   s1A+AA A A  AA!AN)T)__name__ __module__ __qualname__rAr7rLrRr%r#r2r26sXLLL0000;;;     r%r2cTddl}t|try|drd |jt j||||S#t$r&}|j d}d|vrtdd}~wwxYwt|tr)| drt j |} |j||||S#t$r-}|j d}d|vrd}||z }t|d}~wwxYw)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 )scipy.iorr=endswithio netcdf_filegzipopenrOargs ValueErrorr startswithBytesIO)filenamerZr[r\scipyeerrmsgmsgs r#_open_scipy_netcdfrmWshOOO(C  X%6%6u%=%=  8'' (##$T7(    VAYF-77 !UVVV  (E""(x':':6'B'B(:h''x##H4dG#TTT     )V 3 3C cMFF## #  s//A B*!B  BC00 D':(D""D'ceZdZdZ ddZedZdZdZdZ d Z d Z dd Z d Z dZdZ ddZdZdZdS)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 |td||dkrd}n|dkrd}ntd||*|dkr$t|trt|}t ||_t|tr)t t|||t|| }n"t|||| } t| }||_ dS) Nz7cannot save to a group with the scipy.io.netcdf backend NETCDF3_64BITNETCDF3_CLASSICz,invalid format for scipy.io.netcdf backend: rp)r[r\)rZrNkwargsrY) rerr=rrrNr rmdictr rC) r?filename_or_objrZformatgroupr[rNr\manager scipy_datasets r#rAzScipyDataStore.__init__s   VWW W >V66GG ( ( (GGVFVVWW W z/ScipyDataStore.get_variables..sM  48AqQ((A.. /      r%)rrGrEr.r~s`r# get_variableszScipyDataStore.get_variablessL    <@Gz.ScipyDataStore.get_encoding..sUUUTQ19q999r%)rGrr.r~s r# get_encodingzScipyDataStore.get_encodings1 UUTW-?-E-E-G-GUUU  r%Fc||jjvr$tt|jd|s|nd}|j||dS)Nz& does not support modifying dimensions)rGrretyperScreateDimension)r?rlength is_unlimited dim_lengths r# set_dimensionzScipyDataStore.set_dimensionsg 47% % %::&NNN $09VVT  j11111r%cBt|stddS)NzNot a valid attribute name)rre)r?rKs r#_validate_attr_keyz!ScipyDataStore._validate_attr_keys+ %% ;9:: : ; ;r%cz||t|}t|j||dSr4)rrsetattrrG)r?rKrQs r# set_attributezScipyDataStore.set_attributes= $$$%e,,e$$$$$r%c$t|}|Sr4)r)r?variables r#encode_variablezScipyDataStore.encode_variables&x00r%c|r8|jr1|jddikr$tdt|j|j}||jjvr&|j||j|j|jj|}|j D]+\}}| |t|||,t||} | |fS)Nr(z'unexpected encoding for scipy backend: )encodingrelistr8rGrEcreateVariabler;dimsattrsr.rrr2) r?rrcheck_encodingrr8 scipy_varr+r,targets r#prepare_variablezScipyDataStore.prepare_variables  ! !lD%999S$x?P:Q:QSS } tw( ( ( G " "4X] C C CG%d+ N((** % %DAq  # #A & & & Iq! $ $ $ $"4..t|r%c8|jdSr4)rGsyncr~s r#rzScipyDataStore.syncs  r%c8|jdSr4)rCcloser~s r#rzScipyDataStore.closes r%)rpNNNN)F)FN)rSrTrU__doc__rApropertyrGrrrrrrrrrrrrrVr%r#roro}sSW     D''X'      :::***   2222;;;%%%  DH6r%roc FeZdZdZdZdZddZd d d d d d d d d d d d d ddZd S)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 zAOpen netCDF files (.nc, .nc4, .cdf and .gz) using scipy in XarrayzWhttps://docs.xarray.dev/en/stable/generated/xarray.backends.ScipyBackendEntrypoint.htmlrx;str | os.PathLike[Any] | BufferedIOBase | AbstractDataStorereturnboolct|}|P|dr;tj|5}t|}dddn #1swxYwY||dSt |t t jfr&t j |\}}|dvSdS)Nsr]>.nc.cdf.nc4rXF) rrfrbrcrr=osPathLikepathsplitext)r?rx magic_numberf_exts r#guess_can_openz%ScipyBackendEntrypoint.guess_can_opens?OO  # (?(? (L(L #?++ JqFqII  J J J J J J J J J J J J J J J  #**622 2 oR['9 : : 9W%%o66FAs88 8usAAATNrp) mask_and_scale decode_timesconcat_characters decode_coordsdrop_variables use_cftimedecode_timedeltarZryrzr[rNrstr | Iterable[str] | Nonerc t|}t|| | | | | }t}t|5|||||||||}dddn #1swxYwY|S)N)rZryrzr[rN)rrrrrrr)r rorr open_dataset)r?rxrrrrrrrrZryrzr[rNstorestore_entrypointrGs r#rz#ScipyBackendEntrypoint.open_dataset#s"*/:: $vUTX   233 E " "  !..-)"3+-%!1/  B                sA,,A03A0)rxrrr)rxrrrrr)rSrTrUr descriptionurlrrrVr%r#rrs(VK cC*59   """"""""r%rri)1 __future__rrbr`rcollections.abcrtypingrrnumpyr:xarray.backends.commonrrr r r xarray.backends.file_managerr r xarray.backends.locksrrxarray.backends.netcdf3rrrxarray.backends.storerxarray.core.indexingrxarray.core.utilsrrrrxarray.core.variablerrrxarray.core.datasetrr$r0r2rmrorrVr%r#rs"""""" $$$$$$%%%%%%%%NMMMMMMM======== 988888555555 *))))),!!!!!!888888++++++   VVV  B###Lzzzzz(zzzzKKKKK.KKK\!()?@Gr%