a iDfW @sdZddlmZddlZddlZddlmZddlZddl m Z ddl m Z ddlmZerhddl mZd Zd Zd Zd d ZiZdddddddddd ZddZd%ddddddddddZ ddddZdd d!dd"d#d$ZdS)&zR Useful for: * users learning xarray * building tutorials in the documentation. ) annotationsN) TYPE_CHECKING) open_dataset) DataArray)Dataset)T_EngineZxarray_tutorial_dataz%https://github.com/pydata/xarray-dataZmastercCs6ddl}t|tjr t|}n|dur2|t}|S)Nr)pooch isinstanceosPathLikefspathZos_cache_default_cache_dir_name)pathrr\/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/xarray/tutorial.py_construct_cache_dirs    r) Zair_temperatureZair_temperature_gradientZASE_ice_velocityZ basin_maskZersstv5ZrasmZ ROMS_exampleZtinyZ eraint_uvzc Cst|}|dkrbz ddl}WnBty`z ddl}Wn"tyZtd|dYn0Yn0|dkrz ddl}WnBtyz ddl}Wn"tytd|dYn0Yn0dS)Nrrzopening tutorial dataset z2 requires either scipy or netCDF4 to be installed.rz5 requires either h5netcdf or netCDF4 to be installed.) file_formatsgetscipy ImportErrornetCDF4h5netcdf)nameversionrrrrrr_check_netcdf_engine_installed7s*           rT)enginestrboolzNone | str | os.PathLikerr)rcache cache_dirrreturnc KsDz ddl}Wn.ty:}ztd|WYd}~n d}~00|}|dt|}|tvrht|}nt|} | jsd} |durt || | } nR| jdkr|durd}z ddl } Wn.ty}ztd|WYd}~n d}~00t d t d | j}|j|d|d } t| fd |i|} |s@| } t| | S) a Open a dataset from the online repository (requires internet). If a local copy is found then always use that to avoid network traffic. Available datasets: * ``"air_temperature"``: NCEP reanalysis subset * ``"air_temperature_gradient"``: NCEP reanalysis subset with approximate x,y gradients * ``"basin_mask"``: Dataset with ocean basins marked using integers * ``"ASE_ice_velocity"``: MEaSUREs InSAR-Based Ice Velocity of the Amundsen Sea Embayment, Antarctica, Version 1 * ``"rasm"``: Output of the Regional Arctic System Model (RASM) * ``"ROMS_example"``: Regional Ocean Model System (ROMS) output * ``"tiny"``: small synthetic dataset with a 1D data variable * ``"era5-2mt-2019-03-uk.grib"``: ERA5 temperature data over the UK * ``"eraint_uvz"``: data from ERA-Interim reanalysis, monthly averages of upper level data * ``"ersstv5"``: NOAA's Extended Reconstructed Sea Surface Temperature monthly averages Parameters ---------- name : str Name of the file containing the dataset. e.g. 'air_temperature' cache_dir : path-like, optional The directory in which to search for and write cached data. cache : bool, optional If True, then cache data locally for use on subsequent calls **kws : dict, optional Passed to xarray.open_dataset See Also -------- tutorial.load_dataset open_dataset load_dataset rNzhtutorial.open_dataset depends on pooch to download and manage datasets. To proceed please install pooch.WARNINGz.ncz.gribcfgribz:Reading this tutorial dataset requires the cfgrib package.z/raw//)urlZ known_hashrr)rrZ get_loggersetLevelr external_urlspathlibPathsuffixr with_suffixr$base_urlrrretrieve _open_datasetloadunlink)rr r!rkwsreloggerr&rZdefault_extensionr$filepathdsrrrrRsJ,       r)r"cOs:t|i|}|WdS1s,0YdS)aS Open, load into memory, and close a dataset from the online repository (requires internet). If a local copy is found then always use that to avoid network traffic. Available datasets: * ``"air_temperature"``: NCEP reanalysis subset * ``"air_temperature_gradient"``: NCEP reanalysis subset with approximate x,y gradients * ``"basin_mask"``: Dataset with ocean basins marked using integers * ``"rasm"``: Output of the Regional Arctic System Model (RASM) * ``"ROMS_example"``: Regional Ocean Model System (ROMS) output * ``"tiny"``: small synthetic dataset with a 1D data variable * ``"era5-2mt-2019-03-uk.grib"``: ERA5 temperature data over the UK * ``"eraint_uvz"``: data from ERA-Interim reanalysis, monthly averages of upper level data * ``"ersstv5"``: NOAA's Extended Reconstructed Sea Surface Temperature monthly averages Parameters ---------- name : str Name of the file containing the dataset. e.g. 'air_temperature' cache_dir : path-like, optional The directory in which to search for and write cached data. cache : bool, optional If True, then cache data locally for use on subsequent calls **kws : dict, optional Passed to xarray.open_dataset See Also -------- tutorial.open_dataset open_dataset load_dataset N)rr0)argskwargsr6rrr load_datasets%r9)seedz None | int)r:r"c Cstj|}ttgdgdtdtdddtdd|ddd }d|jd |j d d|j |j }d |j|j d |j |j }t ||d}gd|d<d|jj d<d|j j d<d|j j d<d|j j d<d|jj d<d|jj d<|S)z Create an example dataset. Parameters ---------- seed : int, optional Seed for the random number generation. )r rr)xyzwrrr;rg?)dimsZcoordsg@g)AB)ZoneZtwoZthreeZfiver?ZxunitsZunitsZyunitsZzunitsZwunitsZAunitsZBunits)nprandomZ default_rngrzerosZarangeZlinspaceZstandard_normalr<r=r>r?rattrsrDrE)r:rngrDrEr6rrrscatter_example_datasets*     (        rK)TN)__doc__ __future__rr r)typingrnumpyrFZxarray.backends.apirr/Zxarray.core.dataarrayrZxarray.core.datasetrrr r-rrr(rrr9rKrrrrsB        X)