d zddlmZddlZddlmZddlmZGddeZGddZ d Z d Z d Z dS) ) annotationsN) DataArray)DatasetceZdZdZdS)AccessorRegistrationWarningz/Warning for conflicts in accessor registration.N)__name__ __module__ __qualname____doc__6lib/python3.11/site-packages/xarray/core/extensions.pyrr s9999r rceZdZdZdZdZdS)_CachedAccessorz?Custom property-like object (descriptor) for caching accessors.c"||_||_dS)N)_name _accessor)selfnameaccessors r__init__z_CachedAccessor.__init__s !r c&||jS |j}n#t$r ix}|_YnwxYw ||jS#t$rYnwxYw ||}n&#t$rt d|jdwxYw|||j<|S)Nzerror initializing z accessor.)r_cacheAttributeErrorrKeyError RuntimeError)robjclscache accessor_objs r__get__z_CachedAccessor.__get__s ;> ! $JEE $ $ $!# #ECJJJ $ $ $    D  O>>#..LL O O OMTZMMMNN N  O )djs')) : AA A!!#BN)rr r r rr!r r rrr s8II"""r rcfd}|S)Nc tr&tjd|dddtdt t ||S)Nzregistration of accessor z under name z for type z: is overriding a preexisting attribute with the same name.) stacklevel)hasattrwarningswarnrsetattrr)rrrs r decoratorz%_register_accessor..decorator2s 3    MIHIIDIIVYIII+      T?4::;;;r r )rrr*s`` r_register_accessorr+1s*       r c,t|tS)a0Register a custom accessor on xarray.DataArray objects. Parameters ---------- name : str Name under which the accessor should be registered. A warning is issued if this name conflicts with a preexisting attribute. See Also -------- register_dataset_accessor )r+rrs rregister_dataarray_accessorr.@s dI . ..r c,t|tS)aRegister a custom property on xarray.Dataset objects. Parameters ---------- name : str Name under which the accessor should be registered. A warning is issued if this name conflicts with a preexisting attribute. Examples -------- In your library code: >>> @xr.register_dataset_accessor("geo") ... class GeoAccessor: ... def __init__(self, xarray_obj): ... self._obj = xarray_obj ... ... @property ... def center(self): ... # return the geographic center point of this dataset ... lon = self._obj.latitude ... lat = self._obj.longitude ... return (float(lon.mean()), float(lat.mean())) ... ... def plot(self): ... # plot this array's data on a map, e.g., using Cartopy ... pass ... Back in an interactive IPython session: >>> ds = xr.Dataset( ... {"longitude": np.linspace(0, 10), "latitude": np.linspace(0, 20)} ... ) >>> ds.geo.center (10.0, 5.0) >>> ds.geo.plot() # plots data on a map See Also -------- register_dataarray_accessor )r+rr-s rregister_dataset_accessorr0PsV dG , ,,r ) __future__rr'xarray.core.dataarrayrxarray.core.datasetrWarningrrr+r.r0r r rr5s""""""++++++'''''':::::':::!!!!!!!!H    / / / +-+-+-+-+-r