a İa@sdZddlZddlZddlZddlmZddlm Z ddl m Z ddl mZdZd Zd Zd d ZiZd ddZddddddZddZdddddZd ddZ ddZddZdS)!zR Useful for: * users learning xarray * building tutorials in the documentation. N) open_dataset) open_rasterio) DataArray)DatasetZxarray_tutorial_dataz%https://github.com/pydata/xarray-dataZmastercCs6ddl}t|tjr t|}n|dur2|t}|S)Nr)pooch isinstanceosPathLikefspathZos_cache_default_cache_dir_name)pathrr.lib/python3.9/site-packages/xarray/tutorial.py_construct_cache_dirs    rzDhttps://github.com/mapbox/rasterio/raw/1.2.1/tests/data/RGB.byte.tifzAhttps://github.com/mapbox/rasterio/raw/1.2.1/tests/data/shade.tif)zRGB.byteZshade)Zair_temperatureZrasmZ 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_installed0s*           rT)enginec Ksz ddl}Wn.ty:}ztd|WYd}~n d}~00|}|dt|}|tvrht|}n\t|} | jsd} |durt || | } n| jdkr|durd}t dt 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 * ``"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 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 -------- xarray.open_dataset rNzhtutorial.open_dataset depends on pooch to download and manage datasets. To proceed please install pooch.WARNINGz.ncz.gribZcfgribz/raw//urlZ known_hashr r)rr get_loggersetLevelr external_urlspathlibPathsuffixr with_suffixbase_urlrrretrieve _open_datasetloadunlink) rcache cache_dirrkwsreloggerr r Zdefault_extensionfilepathdsrrrrKs:&      rc Ksz ddl}Wn.ty:}ztd|WYd}~n d}~00|}|dt|}t|}|durvtd||j|d|d} t | fi|} |s| } t |  | S)a Open a rasterio dataset from the online repository (requires internet). If a local copy is found then always use that to avoid network traffic. Available datasets: * ``"RGB.byte"``: TIFF file derived from USGS Landsat 7 ETM imagery. * ``"shade"``: TIFF file derived from from USGS SRTM 90 data ``RGB.byte`` and ``shade`` are downloaded from the ``rasterio`` repository [1]_. Parameters ---------- name : str Name of the file containing the dataset. e.g. 'RGB.byte' 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_rasterio See Also -------- xarray.open_rasterio References ---------- .. [1] https://github.com/mapbox/rasterio rNzitutorial.open_rasterio depends on pooch to download and manage datasets. To proceed please install pooch.rzunknown rasterio dataset: r)rrr!r"rexternal_rasterio_urlsr ValueErrorr)_open_rasterior+r$r%r,) rrr-r.r/rr0r1r r2Zarrrrrrs('   rcOs:t|i|}|WdS1s,0YdS)z Open, load into memory, and close a dataset from the online repository (requires internet). See Also -------- open_dataset N)rr+)argskwargsr3rrr load_datasets r9c Csttgdgdtdtdddtddtjdgd }d|jd |jd d|j |j }d |j|jd |j |j }t ||d}gd|d<d|jj d<d|jj d<d|j j d<d|j j d<d|j j d<d|jj d<|S)N)r rr)xyzwrrrr:rg?)ZdimsZcoordsg@g)AB)ZoneZtwoZthreeZfiver>ZxunitsZunitsZyunitsZzunitsZwunitsZAunitsZBunits)rnpZzerosZarangeZlinspaceZrandomZrandnr;r<r=r>rattrsrArB)rArBr3rrrscatter_example_datasets(   (        rE)TN)NTN)__doc__r r$ZnumpyrCZ backends.apirr*Zbackends.rasterio_rr6Zcore.dataarrayrZ core.datasetrr r(rrr#r4rrr9rErrrrsB       N A