B \{@spdZddlmZddlmZddlmZddlmZmZddlm Z Gd d d ej Z Gd d d ej ej Z d S)z3 Implements support for HDF5 dimension scales. )absolute_import)h5ds)base)phil with_phil)Datasetc@seZdZdZeeddZejeddZeddZeddZ ed d Z ed d Z ed dZ eddZ ddZddZddZddZddZeddZdS)DimensionProxyz- Represents an HDF5 "dimension". c Cs>y t|j}||jd|jSttfk r8dSXdS)z& Get or set the dimension scale label ZDIMENSION_LABELSN)r _id_dZattrs _dimensionKeyError IndexError)selfdsetr,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_Z dimensionrrr__init__.szDimensionProxy.__init__cCstt||j|jfS)N)hashtyper r)rrrr__hash__3szDimensionProxy.__hash__cCst|t|kS)N)r)rotherrrr__eq__7szDimensionProxy.__eq__ccsx|D] }|Vq WdS)N)keys)rkrrr__iter__;szDimensionProxy.__iter__cCst|j|jS)N)rZget_num_scalesr r)rrrr__len__@szDimensionProxy.__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)itemrrrfMsz%DimensionProxy.__getitem__..f) isinstanceintriterater rappendr r)rr#scalesr$Zresr)r#rr __getitem__Ds  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+WszDimensionProxy.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)r rr"r ).0x)rrr tsz(DimensionProxy.items..N)rlenrr'r rr()rr)r)rritemsgs   zDimensionProxy.itemsc Cs"tdd|DSQRXdS)z7 Get a list of names for the scales on this dimension. cSsg|] \}}|qSrr)r.key_rrrr0{sz'DimensionProxy.keys..N)rr2)rrrrrxszDimensionProxy.keysc Cs"tdd|DSQRXdS)z5 Get a list of Dataset for scales on this dimension. cSsg|] \}}|qSrr)r.r4rrrrr0sz)DimensionProxy.values..N)rr2)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-r2rr5r6rrrrr 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)r1rr r )rindexrrrr*szDimensionManager.__getitem__cCstt|jjS)z3 Number of dimensions associated with the dataset. )r1r r shape)rrrrr!szDimensionManager.__len__ccs$xtt|D]}||VqWdS)z Iterate over the dimensions. N)ranger1)rirrrr szDimensionManager.__iter__cCs|js dSdt|jS)Nz#z!)r r,)rrrrr6szDimensionManager.__repr__r c Cs(tt|j||WdQRXdS)zo Create a new dimension, from an initial scale. Provide the dataset and a name for the scale. N)rrZ set_scaler,r)rrnamerrr create_scaleszDimensionManager.create_scaleN)r ) r7r8r9r:rrr*r!r r6rDrrrrr=s     r=N)r:Z __future__rr rrrrZdatasetr ZCommonStateObjectr Z MappingHDF5r=rrrr s    t