U Cv fP@sddlmZddlZddlmZmZmZmZm Z ddl m Z ddl m Z mZmZmZddlmZddlmZddlmZmZmZmZdd lmZee egZGd d d eZGd d d eZGdddeZeed<dS)) annotationsN)BACKEND_ENTRYPOINTSAbstractDataStore BackendArrayBackendEntrypoint_normalize_path)CachingFileManager) HDF5_LOCK NETCDFC_LOCK combine_locks ensure_lock)StoreBackendEntrypoint)indexing)Frozen FrozenDictclose_on_errormodule_available)Variablec@s.eZdZddZd ddZddZdd Zd S) PncArrayWrappercCs.||_||_|}|j|_t|j|_dSN) datastore variable_name get_arrayshapenpZdtype)selfrrarrayrA/tmp/pip-unpacked-wheel-h316xyqg/xarray/backends/pseudonetcdf_.py__init__s zPncArrayWrapper.__init__TcCs|jj|}|j|jSr)r_manageracquire variablesr)r needs_lockdsrrrrszPncArrayWrapper.get_arraycCst||jtjj|jSr)rZexplicit_indexing_adapterrZIndexingSupportZ OUTER_1VECTOR_getitem)rkeyrrr __getitem__#s zPncArrayWrapper.__getitem__c Cs4|jj"|jdd}||W5QRSQRXdS)NF)r#)rlockr)rr&rrrrr%(s  zPncArrayWrapper._getitemN)T)__name__ __module__ __qualname__rrr'r%rrrrrs rc@sdeZdZdZedddZdddZeddZd d Z d d Z d dZ ddZ ddZ ddZdS)PseudoNetCDFDataStorez-Store for accessing datasets via PseudoNetCDFNcKsPddlm}d|i}|dk r$||d<|dkr0t}t||fd|i|}|||S)Nr)pncopenkwargsmoder() PseudoNetCDFr- PNETCDF_LOCKr)clsfilenamer(r/ format_kwargsr-keywordsmanagerrrropen1s zPseudoNetCDFDataStore.opencCs||_t||_dSr)r r r()rr6r(rrrr@szPseudoNetCDFDataStore.__init__cCs |jSr)r r!rrrrr$DszPseudoNetCDFDataStore.dscs4tt||}fddD}tj||S)Ncsi|]}|t|qSr)getattr.0kvarrr Jsz=PseudoNetCDFDataStore.open_store_variable..)rZLazilyIndexedArrayrncattrsr dimensions)rnamer>dataattrsrr=ropen_store_variableHsz)PseudoNetCDFDataStore.open_store_variablecstfddjjDS)Nc3s"|]\}}|||fVqdSr)rE)r;r<vr8rr Nsz6PseudoNetCDFDataStore.get_variables..)rr$r"itemsr8rr8r get_variablesMs  z#PseudoNetCDFDataStore.get_variablescstfddjDS)Ncsi|]}|tj|qSr)r9r$r:r8rrr?Ssz3PseudoNetCDFDataStore.get_attrs..)rr$r@r8rr8r get_attrsRszPseudoNetCDFDataStore.get_attrscCs t|jjSr)rr$rAr8rrrget_dimensionsUsz$PseudoNetCDFDataStore.get_dimensionscsdfddjjDiS)NZunlimited_dimscs h|]}jj|r|qSr)r$rAZ isunlimitedr:r8rr Zsz5PseudoNetCDFDataStore.get_encoding..)r$rAr8rr8r get_encodingXs  z"PseudoNetCDFDataStore.get_encodingcCs|jdSr)r closer8rrrrN_szPseudoNetCDFDataStore.close)NN)N)r)r*r+__doc__ classmethodr7rpropertyr$rErIrJrKrMrNrrrrr,.s   r,c @s.eZdZdZedZdZdZdZd d d Z dS) 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 r0zGOpen 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/r(FTNc  KsXt|}tj|f| | d| } t} t|  | j| |||||||d}W5QRX|S)N)r(r/)rTrUrVrWrXrYrZ)rr,r7r r open_dataset)rrSrTrUrVrWrXrYrZr/r(r4storeZstore_entrypointr$rrrr[s,  z*PseudoNetCDFBackendEntrypoint.open_dataset) FTTTNNNNN) r)r*r+rOr available descriptionurlZopen_dataset_parametersr[rrrrrRcs rRZ pseudonetcdf) __future__rZnumpyrZxarray.backends.commonrrrrrZxarray.backends.file_managerrZxarray.backends.locksr r r r Zxarray.backends.storer Z xarray.corerZxarray.core.utilsrrrrZxarray.core.variablerr1rr,rRrrrrs      5Q