U md@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|jWSttfk r6YdSXdS)z& Get or set the dimension scale label ZDIMENSION_LABELSN)r _idattrs _dimensionKeyError IndexErrorselfdsetrF/home/sam/Atlas/atlas_env/lib/python3.8/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}|dkrbtt|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__c Cs(tt|j|j|jW5QRXdS)zo Attach a scale to this dimension. Provide the Dataset of the scale you would like to attach. N)rr attach_scaler idrrrrrr1XszDimensionProxy.attach_scalec Cs(tt|j|j|jW5QRXdS)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_scalec sTtFg}tdkr,tjj|jdfdd|DW5QRSQRXdS)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.itemsc Cs.t dd|DW5QRSQRXdS)z7 Get a list of names for the scales on this dimension. cSsg|] \}}|qSrr)r4key_rrrr6|sz'DimensionProxy.keys..Nrr8rrrrr"yszDimensionProxy.keysc Cs.t dd|DW5QRSQRXdS)z5 Get a list of Dataset for scales on this dimension. cSsg|] \}}|qSrr)r4r:rrrrr6sz)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__cCs|jjS)z3 Number of dimensions associated with the dataset. )r Zrankrrrrr%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_scales zDimensionManager.create_scaleN)r ) r>r?r@rArrr0r%r$r=rMrrrrrDs     rD)rArJr rZ h5py_warningsrrrrZdatasetr ZCommonStateObjectr rDrrrr s    t