a v=ic.@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 Cs:zt|j}|jd|jWSttfy4YdS0dS)z& Get or set the dimension scale label ZDIMENSION_LABELSN)r _idattrs _dimensionKeyError IndexErrorselfdsetr^/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/h5py/_hl/dims.pylabels  zDimensionProxy.labelcCst|j|j||dSN)rZ set_labelr r_e)rvalrrrr)scCs||_||_dSr)r r)rZid_ dimensionrrr__init__/szDimensionProxy.__init__cCstt||j|jfSr)hashtyper rrrrr__hash__4szDimensionProxy.__hash__cCst|t|kSr)r)rotherrrr__eq__8szDimensionProxy.__eq__ccs|D] }|VqdSr)keys)rkrrr__iter__<s zDimensionProxy.__iter__cCst|j|jSr)rZget_num_scalesr rrrrr__len__AszDimensionProxy.__len__csnttr0g}tjj|jdt|Sfdd}tjj|d}|durbtt|SdS)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__cCs<t$t|j|j|jWdn1s.0YdS)zo Attach a scale to this dimension. Provide the Dataset of the scale you would like to attach. N)rr attach_scaler idrrrrrr1XszDimensionProxy.attach_scalecCs<t$t|j|j|jWdn1s.0YdS)zq Remove a scale from this dimension. Provide the Dataset of the scale you would like to remove. N)rr detach_scaler r2rrrrrr3`szDimensionProxy.detach_scalecs`tHg}tdkr,tjj|jdfdd|DWdS1sR0YdS)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.itemscCs:t"dd|DWdS1s,0YdS)z7 Get a list of names for the scales on this dimension. cSsg|] \}}|qSrr)r4key_rrrr6|z'DimensionProxy.keys..Nrr8rrrrr"yszDimensionProxy.keyscCs:t"dd|DWdS1s,0YdS)z5 Get a list of Dataset for scales on this dimension. cSsg|] \}}|qSrr)r4r:rrrrr6r;z)DimensionProxy.values..Nr<rrrrvalues~szDimensionProxy.valuescCs"|js dSd|j|jt|jfS)Nz"z)<"%s" dimension %d of HDF5 dataset at %s>)r rrr2rrrr__repr__s zDimensionProxy.__repr__N)__name__ __module__ __qualname____doc__propertyrrsetterrrr!r$r%r0r1r3r8r"r=r>rrrrr s4        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)r2r )rparentrrrrszDimensionManager.__init__cCs$|t|dkrtdt|j|S)z# Return a Dimension object rzIndex out of range)r7rr r )rindexrrrr0szDimensionManager.__getitem__cCstt|jjS)z3 Number of dimensions associated with the dataset. )r7r r shaperrrrr%szDimensionManager.__len__ccs tt|D]}||Vq dS)z Iterate over the dimensions. N)ranger7)rirrrr$szDimensionManager.__iter__cCs|js dSdt|jS)Nz#z!)r r2rrrrr>szDimensionManager.__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 ) r?r@rArBrrr0r%r$r>rOrrrrrEs     rE)rBrLr rZ h5py_warningsrrrrZdatasetr ZCommonStateObjectr rErrrr s    t