o cd@stdZddlZddlmZddlmZddlmZddlmZmZdd l m Z Gd d d ej Z Gd d d ej Z dS)z3 Implements support for HDF5 dimension scales. N)h5ds)H5pyDeprecationWarning)base)phil with_phil)Datasetc@seZdZdZeeddZejeddZeddZeddZ ed d Z ed d Z ed dZ eddZ ddZddZddZddZddZeddZdS)DimensionProxyz- Represents an HDF5 "dimension". c Cs6zt|j}|jd|jWSttfyYdSw)z& Get or set the dimension scale label ZDIMENSION_LABELS)r _idattrs _dimensionKeyError IndexErrorselfdsetr-lib/python3.10/site-packages/h5py/_hl/dims.pylabels  zDimensionProxy.labelcCst|j|j||dSN)rZ set_labelr r_e)rvalrrrr)scCs||_||_dSr)r r)rZid_Z dimensionrrr__init__/s zDimensionProxy.__init__cCstt||j|jfSr)hashtyper rrrrr__hash__4szDimensionProxy.__hash__cCst|t|kSr)r)rotherrrr__eq__8zDimensionProxy.__eq__ccs|D]}|VqdSr)keys)rkrrr__iter__<s zDimensionProxy.__iter__cCst|j|jSr)rZget_num_scalesr rrrrr__len__Ar!zDimensionProxy.__len__csjttrg}tjj|jdt|Sfdd}tjj|d}|dur1tt|S)Nrcst|kr |SdS)z- Iterate over scales to find a matching name N)rget_scale_namer)ZdsiditemrrrfNsz%DimensionProxy.__getitem__..f) isinstanceintriterater rappendr r)rr(scalesr)resrr'r __getitem__Es  zDimensionProxy.__getitem__cC>tt|j|j|jWddS1swYdS)zo Attach a scale to this dimension. Provide the Dataset of the scale you would like to attach. N)rr attach_scaler idrrrrrr2X"zDimensionProxy.attach_scalecCr1)zq Remove a scale from this dimension. Provide the Dataset of the scale you would like to remove. N)rr detach_scaler r3rrrrrr5`r4zDimensionProxy.detach_scalecs`t$g}tdkrtjj|jdfdd|DWdS1s)wYdS)zX Get a list of (name, Dataset) pairs with all scales on this dimension. rcs$g|]}t|t|fqSr)Z_drr&r ).0xrrr tsz(DimensionProxy.items..N)rlenrr,r rr-)rr.rrritemshs  $zDimensionProxy.itemscC:tdd|DWdS1swYdS)z7 Get a list of names for the scales on this dimension. cSsg|]\}}|qSrr)r6key_rrrr8|z'DimensionProxy.keys..Nrr:rrrrr"y$zDimensionProxy.keyscCr;)z5 Get a list of Dataset for scales on this dimension. cSsg|]\}}|qSrr)r6r=rrrrr8r>z)DimensionProxy.values..Nr?rrrrvalues~r@zDimensionProxy.valuescCs"|jsdSd|j|jt|jfS)Nz"z)<"%s" dimension %d of HDF5 dataset at %s>)r rrr3rrrr__repr__s zDimensionProxy.__repr__N)__name__ __module__ __qualname____doc__propertyrrsetterrrr r$r%r0r2r5r:r"rArBrrrrr s6         r c@sVeZdZdZeddZeddZeddZedd Zed d Z dd dZ dS)DimensionManagerz Represents a collection of dimension associated with a dataset. Like AttributeManager, an instance of this class is returned when accessing the ".dims" property on a Dataset. cCs |j|_dS)z Private constructor. N)r3r )rparentrrrrs zDimensionManager.__init__cCs$|t|dkr tdt|j|S)z# Return a Dimension object rzIndex out of range)r9rr r )rindexrrrr0s zDimensionManager.__getitem__cCs|jjS)z3 Number of dimensions associated with the dataset. )r Zrankrrrrr%szDimensionManager.__len__ccs"tt|D]}||VqdS)z Iterate over the dimensions. N)ranger9)rirrrr$s zDimensionManager.__iter__cCs|jsdSdt|jS)Nz#z!)r r3rrrrrBszDimensionManager.__repr__r cCstjdtdd||dS)zo Create a new dimension, from an initial scale. Provide the dataset and a name for the scale. zTother_ds.dims.create_scale(ds, name) is deprecated. Use ds.make_scale(name) instead.r) stacklevelN)warningswarnrZ make_scale)rrnamerrr create_scaleszDimensionManager.create_scaleN)r ) rCrDrErFrrr0r%r$rBrRrrrrrIs     rI)rFrOr rZ h5py_warningsrrrrZdatasetr ZCommonStateObjectr rIrrrrs     t