dV dZddlmZddlZddlZddlmZddlZddl m Z ddl m Z ddlmZerddl mZd Zd Zd Zd ZiZd ddddd dd d d ZdZ d#ddd$dZ d%dZddd&d"ZdS)'zR Useful for: * users learning xarray * building tutorials in the documentation. ) annotationsN) TYPE_CHECKING) open_dataset) DataArray)Dataset)T_Enginexarray_tutorial_dataz%https://github.com/pydata/xarray-datamastercddl}t|tjrtj|}n||t }|S)Nr)pooch isinstanceosPathLikefspathos_cache_default_cache_dir_name)pathr s /lib/python3.11/site-packages/xarray/tutorial.py_construct_cache_dirrsJLLL$ $$7y ~~566 K) air_temperatureair_temperature_gradientASE_ice_velocity basin_maskersstv5rasm ROMS_exampletiny eraint_uvzcHt|}|dkr= ddl}n7#t$r* ddl}n!#t$rtd|dwxYwYnwxYw|dkr> ddl}dS#t$r* ddl}YdS#t$rtd|dwxYwwxYwdS)Nrrzopening tutorial dataset z2 requires either scipy or netCDF4 to be installed.rz5 requires either h5netcdf or netCDF4 to be installed.) file_formatsgetscipy ImportErrornetCDF4h5netcdf)nameversionr%r'r(s r_check_netcdf_engine_installedr+6s9t$$G!||  LLLL       !////  !||  OOOOO       !2222  |sA' A7AAAA%A++ B6A==BBT)enginer)strcachebool cache_dirNone | str | os.PathLiker,rreturnrc  ddl}n"#t$r}td|d}~wwxYw|}|dt |}|t vrt |}nt j|} | js)d} |t|| | } n7| jdkr,|*d} ddl } n"#t$r}td|d}~wwxYwtd td | j}||d| } t!| fd |i|} |s:| } t j| | 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//)url known_hashrr,)r r& get_loggersetLevelr external_urlspathlibPathsuffixr+ with_suffixr5base_urlr*r)retrieve _open_datasetloadunlink)r)r.r0r,kwsr eloggerr7rdefault_extensionr5filepathdss rrrQsX  0        F OOI$Y//I }D!|D!!{  % ~.t444##$566DD [G # #~!!MMMM"%T 5555$)55~~#$Y~GGH x 6 6 6# 6 6B ( WWYY X%%''' Is( &!& C C0C++C0cvt|i|5}|cdddS#1swxYwYdS)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)rrC)argskwargsrJs r load_datasetrNsJ t &v & &"wwyys .22)seedrO None | intc tj|}ttjgdgdtjdtjdddtjdd|dzd }d|jd zz|j d zzd|j z|j zz}d |jz|j d |j zz z|j z}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)xyzwrrrRrg?)dimscoordsg@g)AB)onetwothreefiverVxunitsunitsyunitszunitswunitsAunitsBunits)nprandom default_rngrzerosarangelinspacestandard_normalrSrTrUrVrattrsr\r])rOrngr\r]rJs rscatter_example_datasetrrs[ )   % %C  ! ! !1Q2&&1s**1---      A ac1f qsCx#)ac/1A qs QSAG_$qs*A qq!! " "B---BsG"BDJw"BDJw"BDJw"BDJw"BDJw"BDJw Ir)TN) r)r-r.r/r0r1r,rr2r)r2r)rOrPr2r)__doc__ __future__rrr<typingrnumpyrixarray.backends.apirrBxarray.core.dataarrayrxarray.core.datasetrrrr@r*rr;r#r+rNrrrrr{s{#"""""  ======++++++''''''-,,,,,,1 2   !     :*.U  UUUUUUp&&&&R37!!!!!!!!r