o cd @sBdZddlZddlmZddZddZGd d d Zd d ZdS) z7 Implements a portion of the selection operations. N)h5scst|dkr }n$jdurtdtfdd|Dr!tdtfdd|D}t|d kr@|j|dd}||fS|}||fS) z Returns a 2-tuple containing: 1. Output dataset dtype 2. Dtype containing HDF5-appropriate description of destination rNz+Field names only allowed for compound typesc3s|]}|jvVqdSN)names).0x dataset_dtype4lib/python3.10/site-packages/h5py/_hl/selections2.py szread_dtypes..z#Field does not appear in this type.csg|] }|j|dfqS)r)fields)rnamerr r "szread_dtypes..)lenr ValueErroranynpZdtyper )r rZ format_dtypeZ output_dtyper rr read_dtypess   rcCsP|jdkr td|dkrd}n |tfkrd}ntd|}|||fS)z Returns a 2-tuple containing: 1. Output dataset shape 2. HDF5 dataspace containing source selection. Works for scalar datasets. r z1Illegal selection function for non-scalar datasetN-Illegal slicing argument for scalar dataspace)shape RuntimeErrorEllipsisrZ get_space select_all)ZdsidargsZ out_shapeZ source_spacer r r read_selections_scalar.s  rc@s eZdZdZddZddZdS)ScalarReadSelectionz5 Implements slicing for scalar datasets. cCsB|dkrd|_n |tfkrd|_ntdttj|_||_dS)Nr r)ZmshaperrrZcreateZSCALARmspacefspace)selfrrr r r __init__Ps  zScalarReadSelection.__init__ccs|j|j|jfVdSr)rrr)r r r r __iter__[s zScalarReadSelection.__iter__N)__name__ __module__ __qualname____doc__r!r"r r r r rJs rcCs|jdkr t||St)zn Top-level dispatch function for reading. At the moment, only supports reading from scalar datasets. r )rrNotImplementedError)rrr r r select_read_s  r() r&Znumpyrrrrrr(r r r r s