d@NddlmZddlmZddlmZmZddlZddl m Z m Z m Z m Z mZddlmZddlmZmZmZmZddlmZdd lmZdd lmZmZmZdd lmZerddl Z dd l!m"Z"dd l#m$Z$eeegZ%Gdde Z&Gdde Z'Gdde Z(de(fe d<dS)) annotations)Iterable) TYPE_CHECKINGAnyN)BACKEND_ENTRYPOINTSAbstractDataStore BackendArrayBackendEntrypoint_normalize_path)CachingFileManager) HDF5_LOCK NETCDFC_LOCK combine_locks ensure_lock)StoreBackendEntrypoint)indexing)Frozen FrozenDictclose_on_error)Variable)BufferedIOBase)Datasetc(eZdZdZddZdZdZdS)PncArrayWrapperc||_||_|}|j|_t j|j|_dSN) datastore variable_name get_arrayshapenpdtype)selfrrarrays =lib/python3.11/site-packages/xarray/backends/pseudonetcdf_.py__init__zPncArrayWrapper.__init__!s@"*  [ Xek** Tcd|jj|}|j|jSr)r_manageracquire variablesr)r# needs_lockdss r%rzPncArrayWrapper.get_array(s* ^ $ , ,Z 8 8|D.//r'cbtj||jtjj|jSr)rexplicit_indexing_adapterr IndexingSupport OUTER_1VECTOR_getitem)r#keys r% __getitem__zPncArrayWrapper.__getitem__,s*1 X5CT]   r'c|jj5|d}||cdddS#1swxYwYdS)NF)r,)rlockr)r#r3r$s r%r2zPncArrayWrapper._getitem1s ^   NNeN44E:                  s 8<<N)T)__name__ __module__ __qualname__r&rr4r2r'r%rr sU+++0000   r'rcleZdZdZed dZd dZedZdZ dZ dZ d Z d Z d ZdS)PseudoNetCDFDataStorez-Store for accessing datasets via PseudoNetCDFNc nddlm}d|i}|||d<|t}t||fd|i|}|||S)Nr)pncopenkwargsmoder6) PseudoNetCDFr> PNETCDF_LOCKr )clsfilenamer6r@ format_kwargsr>keywordsmanagers r%openzPseudoNetCDFDataStore.open:sf((((((m,  #HV  <D$WhNNTNXNNs7D!!!r'c<||_t||_dSr)r)rr6)r#rGr6s r%r&zPseudoNetCDFDataStore.__init__Is %% r'c4|jSr)r)r*r#s r%r-zPseudoNetCDFDataStore.dsMs}$$&&&r'ctjt||}fdD}t j||S)Nc2i|]}|t|Sr:)getattr).0kvars r% z=PseudoNetCDFDataStore.open_store_variable..Ss#;;;GCOO;;;r')rLazilyIndexedArrayrncattrsr dimensions)r#namerQdataattrss ` r%open_store_variablez)PseudoNetCDFDataStore.open_store_variableQsR*?4+F+FGG;;;;S[[]];;;e444r'crtfdjjDS)Nc3NK|]\}}|||fV dSr)rY)rOrPvr#s r% z6PseudoNetCDFDataStore.get_variables..WsM  48AqQ((A.. /      r')rr-r+itemsrKs`r% get_variablesz#PseudoNetCDFDataStore.get_variablesVsL    <@G.\s'III!q'$'1--IIIr')rr-rTrKs`r% get_attrszPseudoNetCDFDataStore.get_attrs[s2IIIItw7H7HIIIJJJr'c4t|jjSr)rr-rUrKs r%get_dimensionsz$PseudoNetCDFDataStore.get_dimensions^sdg()))r'c8dfdjjDiS)Nunlimited_dimsc\h|](}jj|&|)Sr:)r-rU isunlimitedrbs r% z5PseudoNetCDFDataStore.get_encoding..csD1CA1F1R1R1T1Tr')r-rUrKs`r% get_encodingz"PseudoNetCDFDataStore.get_encodingas; 7-  r'c8|jdSr)r)closerKs r%rmzPseudoNetCDFDataStore.closehs r')NNr)r7r8r9__doc__ classmethodrHr&propertyr-rYr_rcrerkrmr:r'r%r<r<7s77 " " "[ "&&&&''X'555    KKK***   r'r<c:eZdZdZdZdZdZ dddZdS)PseudoNetCDFBackendEntrypointaj Backend for netCDF-like data formats in the air quality field based on the PseudoNetCDF package. It can open: - CAMx - RACM2 box-model outputs - Kinetic Pre-Processor outputs - ICARTT Data files (ffi1001) - CMAQ Files - GEOS-Chem Binary Punch/NetCDF files - and many more This backend is not selected by default for any files, so make sure to specify ``engine="pseudonetcdf"`` in ``open_dataset``. For more information about the underlying library, visit: https://pseudonetcdf.readthedocs.io See Also -------- backends.PseudoNetCDFDataStore zGOpen many atmospheric science data formats using PseudoNetCDF in Xarrayz^https://docs.xarray.dev/en/stable/generated/xarray.backends.PseudoNetCDFBackendEntrypoint.html) filename_or_objmask_and_scale decode_timesconcat_characters decode_coordsdrop_variables use_cftimedecode_timedeltar@r6FTNrs;str | os.PathLike[Any] | BufferedIOBase | AbstractDataStorerxstr | Iterable[str] | Nonereturnrc t|}tj|f| | d| } t} t | 5| | |||||||}dddn #1swxYwY|S)N)r6r@)rtrurvrwrxryrz)r r<rHrr open_dataset)r#rsrtrurvrwrxryrzr@r6rEstorestore_entrypointr-s r%rz*PseudoNetCDFBackendEntrypoint.open_datasets*/::%*  "&T  5B  233 E " "  !..-)"3+-%!1/  B                sA,,A03A0) FTTTNNNNN)rsr{rxr|r}r)r7r8r9rn descriptionurlopen_dataset_parametersrr:r'r%rrrrlso2 R kC  59  r'rrrA pseudonetcdf)) __future__rcollections.abcrtypingrrnumpyr!xarray.backends.commonrrr r r xarray.backends.file_managerr xarray.backends.locksr rrrxarray.backends.storer xarray.corerxarray.core.utilsrrrxarray.core.variablerosiorxarray.core.datasetrrBrr<rrr:r'r%rs """"""$$$$$$%%%%%%%%<;;;;;UUUUUUUUUUUU888888 @@@@@@@@@@)))))), III!!!!!!++++++}i677 l.22222-222jLLLLL$5LLL^(67T&UN###r'