B d@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 Cs8yt|j}|jd|jSttfk r2dSXdS)z& Get or set the dimension scale label ZDIMENSION_LABELSN)r _idattrs _dimensionKeyError IndexError)selfdsetr?/var/www/html/venv/lib/python3.7/site-packages/h5py/_hl/dims.pylabels  zDimensionProxy.labelcCst|j|j||dS)N)rZ set_labelr r_e)rvalrrrr)scCs||_||_dS)N)r r)rZid_ dimensionrrr__init__/szDimensionProxy.__init__cCstt||j|jfS)N)hashtyper r)rrrr__hash__4szDimensionProxy.__hash__cCst|t|kS)N)r)rotherrrr__eq__8szDimensionProxy.__eq__ccsx|D] }|Vq WdS)N)keys)rkrrr__iter__<szDimensionProxy.__iter__cCst|j|jS)N)rZget_num_scalesr r)rrrr__len__AszDimensionProxy.__len__csnttr0g}tjj|jdt|Sfdd}tjj|d}|dkrbtt|SdS)Nrcst|kr|SdS)z- Iterate over scales to find a matching name N)rget_scale_namer)Zdsid)itemrrrfNsz%DimensionProxy.__getitem__..f) isinstanceintriterater rappendr r)rr$scalesr%resr)r$rr __getitem__Es  zDimensionProxy.__getitem__c Cs(tt|j|j|jWdQRXdS)zo Attach a scale to this dimension. Provide the Dataset of the scale you would like to attach. N)rr attach_scaler idr)rrrrrr-XszDimensionProxy.attach_scalec Cs(tt|j|j|jWdQRXdS)zq Remove a scale from this dimension. Provide the Dataset of the scale you would like to remove. N)rr detach_scaler r.r)rrrrrr/`szDimensionProxy.detach_scalec sHt:g}tdkr,tjj|jdfdd|DSQRXdS)zX Get a list of (name, Dataset) pairs with all scales on this dimension. rcs$g|]}t|t|fqSr)Z_drr#r ).0x)rrr usz(DimensionProxy.items..N)rlenrr(r rr))rr*r)rritemshs   zDimensionProxy.itemsc Cs"tdd|DSQRXdS)z7 Get a list of names for the scales on this dimension. cSsg|] \}}|qSrr)r0key_rrrr2|sz'DimensionProxy.keys..N)rr4)rrrrryszDimensionProxy.keysc Cs"tdd|DSQRXdS)z5 Get a list of Dataset for scales on this dimension. cSsg|] \}}|qSrr)r0r6rrrrr2sz)DimensionProxy.values..N)rr4)rrrrvalues~szDimensionProxy.valuescCs"|js dSd|j|jt|jfS)Nz"z)<"%s" dimension %d of HDF5 dataset at %s>)r rrr.)rrrr__repr__szDimensionProxy.__repr__N)__name__ __module__ __qualname____doc__propertyrrsetterrrrr!r"r,r-r/r4rr7r8rrrrr s"       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)r.r )rparentrrrrszDimensionManager.__init__cCs$|t|dkrtdt|j|S)z# Return a Dimension object rzIndex out of range)r3rr r )rindexrrrr,szDimensionManager.__getitem__cCs|jjS)z3 Number of dimensions associated with the dataset. )r Zrank)rrrrr"szDimensionManager.__len__ccs$xtt|D]}||VqWdS)z Iterate over the dimensions. N)ranger3)rirrrr!szDimensionManager.__iter__cCs|js dSdt|jS)Nz#z!)r r.)rrrrr8szDimensionManager.__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_scales zDimensionManager.create_scaleN)r ) r9r:r;r<rrr,r"r!r8rHrrrrr?s     r?)r<rEr rZ h5py_warningsrrrrZdatasetr ZCommonStateObjectr r?rrrr s    t