a Sic8@sdZddlZddlmZddlmZmZmZdddZ Gd d d Z Gd d d e Z Gd dde Z Gddde Z ddZdS)z4 High-level access to HDF5 dataspace selections N)product)h5sh5r _selectorcCst|ts|f}t|dkr|d}t|trD|j|kr@td|St|tjrx|jj dkrx|j|krntdt |St|t j r|durtdt ||j}||jkrtdt||d S|dur|j}nt|}t|}||S) a1 High-level routine to generate a selection from arbitrary arguments to __getitem__. The arguments should be the following: shape Shape of the "source" dataspace. args Either a single argument or a tuple of arguments. See below for supported classes of argument. dataset A h5py.Dataset instance representing the source dataset. Argument classes: Single Selection instance Returns the argument. numpy.ndarray Must be a boolean mask. Returns a PointSelection instance. RegionReference Returns a Selection instance. Indices, slices, ellipses, MultiBlockSlices only Returns a SimpleSelection instance Indices, slices, ellipses, lists or boolean index arrays Returns a FancySelection instance. rrzMismatched selection shapebz-Boolean indexing array has incompatible shapeNz1Cannot apply a region reference without a datasetz,Reference shape does not match dataset shape)spaceid) isinstancetuplelen Selectionshape TypeErrornpndarraydtypekindPointSelection from_maskrRegionReference get_regionidrr create_simpleSelectormake_selection)rargsdatasetargsidselectorspacer"O/var/www/html/django/DPS/env/lib/python3.9/site-packages/h5py/_hl/selections.pyselects0           r$c@sneZdZdZdddZeddZeddZed d Zed d Z ed dZ ddZ ddZ ddZ dS)r a Base class for HDF5 dataspace selections. Subclasses support the "selection protocol", which means they have at least the following members: __init__(shape) => Create a new selection on "shape"-tuple __getitem__(args) => Perform a selection with the range specified. What args are allowed depends on the particular subclass in use. id (read-only) => h5py.h5s.SpaceID instance shape (read-only) => The shape of the dataspace. mshape (read-only) => The shape of the selection region. Not guaranteed to fit within "shape", although the total number of points is less than product(shape). nselect (read-only) => Number of selected points. Always equal to product(mshape). broadcast(target_shape) => Return an iterable which yields dataspaces for read, based on target_shape. The base class represents "unshaped" selections (1-D). NcCsN|dur||_|j|_n2t|}||_t|tjft||_|jdS)z= Create a selection. Shape may be None if spaceid is given. N) _idr_shaper rr UNLIMITEDr select_all)selfrr r"r"r#__init__ps zSelection.__init__cCs|jS)z SpaceID instance )r%r)r"r"r#r{sz Selection.idcCs|jS)z Shape of whole dataspace )r&r+r"r"r#rszSelection.shapecCs |jS)z' Number of elements currently selected )r%get_select_npointsr+r"r"r#nselectszSelection.nselectcCs|jfS)z0 Shape of selection (always 1-D for this class) )r-r+r"r"r#mshapeszSelection.mshapecCs|jS)z8Shape of array to read/write (always 1-D for this class))r.r+r"r"r# array_shapeszSelection.array_shapecCst||jkrtd|S)N7Broadcasting is not supported for point-wise selections)rr-rr) source_shaper"r"r# expand_shapeszSelection.expand_shapeccs"t||jkrtd|jVdS)z" Get an iterable for broadcasting r0N)rr-rr%r1r"r"r# broadcastszSelection.broadcastcCs tddS)Nz$This class does not support indexing)NotImplementedError)r)rr"r"r# __getitem__szSelection.__getitem__)N)__name__ __module__ __qualname____doc__r*propertyrrr-r.r/r3r4r6r"r"r"r#r Us      r csPeZdZdZdfdd ZddZedddZd d Zd d Z d dZ Z S)rz Represents a point-wise selection. You can supply sequences of points to the three methods append(), prepend() and set(), or instantiate it with a single boolean array using from_mask(). Ncs(t|||dur$||tjdSN)superr*_perform_selectionr SELECT_SET)r)rr points __class__r"r#r*szPointSelection.__init__cCsntj|ddd}t|jdkr.d|jdf|_|jtjkrDtj}t|dkr\|j n|j ||dS)z7 Internal method which actually performs the selection Cu8)orderrrrN) rasarrayr rr%get_select_typer SEL_POINTSr? select_noneselect_elements)r)r@opr"r"r#r>s  z!PointSelection._perform_selectioncCs>t|tjr|jjdks tdt|}||j||dS)z9Create a point-wise selection from a NumPy boolean array rz4PointSelection.from_mask only works with bool arrays)r@) r rrrrr transposenonzeror)clsmaskr r@r"r"r#rszPointSelection.from_maskcCs||tjdS)z@ Add the sequence of points to the end of the current selection N)r>r SELECT_APPENDr)r@r"r"r#appendszPointSelection.appendcCs||tjdS)zF Add the sequence of points to the beginning of the current selection N)r>rSELECT_PREPENDrQr"r"r#prependszPointSelection.prependcCs||tjdS)z@ Replace the current selection with the given sequence of pointsN)r>rr?rQr"r"r#setszPointSelection.set)NN)N) r7r8r9r:r*r> classmethodrrRrTrU __classcell__r"r"rAr#rs rcsJeZdZdZeddZeddZd fdd Zd d Zd d Z Z S)SimpleSelectionz A single "rectangular" (regular) selection composed of only slices and integer arguments. Can participate in broadcasting. cCs |jdS)z Shape of current selection r)_selr+r"r"r#r.szSimpleSelection.mshapecCs$|jd}tddt|j|DS)Ncss|]\}}|s|VqdSr<r").0xsr"r"r# z.SimpleSelection.array_shape..)rYr zipr.)r)scalarr"r"r#r/s zSimpleSelection.array_shapeNcsHt|||dur||_n&t|j}d||jd|d|f|_dS)Nrr)F)r=r*rYr r)r)rr Z hyperslabrankrAr"r#r*s  zSimpleSelection.__init__c Cs|j\}}}}t|}t|}g}td|dD]b} t|dksJ|| rV|dq0|} | dkst|| | kr|| q0td||jfq0tdd|Drtd||jft |dddS)aMatch the dimensions of an array to be broadcast to the selection The returned shape describes an array of the same size as the input shape, but its dimensions E.g. with a dataset shape (10, 5, 4, 2), writing like this:: ds[..., 0] = np.ones((5, 4)) The source shape (5, 4) will expand to (1, 5, 4, 1). Then the broadcast method below repeats that chunk 10 times to write to an effective shape of (10, 5, 4, 1). rrzCan't broadcast %s -> %scSsg|] }|dkqSrcr")r[nr"r"r# r_z0SimpleSelection.expand_shape..N) rYr listrangerRpoprr/anyr ) r)r2startcountsteprardZremaining_src_dimsZeshapeidxtr"r"r#r3s  zSimpleSelection.expand_shapec cs|jdkr8t|dkr"td||j|jVdS|j\}}}}t|}||}tddt ||D}t|} | dkr|jVn^|j } | d|||t | D]8} tddt t | ||||D} | | | VqdS) z Return an iterator over target dataspaces for broadcasting. Follows the standard NumPy broadcasting rules against the current selection shape (self.mshape). r"rzCan't broadcast %s to scalarNcss|]\}}||VqdSr<r")r[r\yr"r"r#r^%r_z,SimpleSelection.broadcast..rbcss&|]\}}}}||||VqdSr<r")r[r\rqzr]r"r"r#r^.r_)rrrr%r(rYr r3r r`copyselect_hyperslabrir unravel_index offset_simple) r)r2rlrmrnrardtshapechunksZnchunksrrooffsetr"r"r#r4s&        $ zSimpleSelection.broadcast)NN r7r8r9r:r;r.r/r*r3r4rWr"r"rAr#rXs   'rXcsJeZdZdZeddZeddZd fdd Zd d Zd d Z Z S)FancySelectiona' Implements advanced NumPy-style selection operations in addition to the standard slice-and-int behavior. Indexing arguments may be ints, slices, lists of indices, or per-axis (1D) boolean arrays. Broadcasting is not supported for these selections. cCs|jSr<)_mshaper+r"r"r#r.?szFancySelection.mshapecCs|jSr<) _array_shaper+r"r"r#r/CszFancySelection.array_shapeNcs8t|||dur|j}|dur(|}||_||_dSr<)r=r*rr|r})r)rr r.r/rAr"r#r*GszFancySelection.__init__cCs||jkstd|SNz4Broadcasting is not supported for complex selections)r/rr1r"r"r#r3Ps zFancySelection.expand_shapeccs||jkstd|jVdSr~)r/rr%r1r"r"r#r4Us zFancySelection.broadcast)NNNrzr"r"rAr#r{3s    r{cs8}}|tjkrdS|tjkrH|tjkr6dS|tjkr^tSn|tjkr^t d| t j }|tjkrd|S|tjkrj S|tj krfS|tjkrt d|dkrd|SddD\}|t|ffdd tfd dt|D}t|kr4fS|S) a0 Given a dataspace, try to deduce the shape of the selection. Returns one of: * A tuple with the selection shape, same length as the dataspace * A 1D selection shape for point-based and multiple-hyperslab selections * None, for unselected scalars and for NULL dataspaces NzUnrecognized dataspace class %srbz Unrecognized selection method %srcss|]}t|VqdSr<)rarrayr[r\r"r"r#r^r_zguess_shape..csr|dkrdS}||d7<}||d8<|}|jt|t|tjd|}|S)a Determine the number of elements selected along a particular axis. To do this, we "mask off" the axis by making a hyperslab selection which leaves only the first point along the axis. For a 2D dataset with selection box shape (X, Y), for axis 1, this would leave a selection of shape (X, 1). We count the number of points N_leftover remaining in the selection and compute the axis selection length by N_axis = N/N_leftover. r)rK)rsrtr r SELECT_NOTBr,)raxisrlrmZ masked_sidZ N_leftover)N bottomcornerboxshaper"r# get_n_axiss zguess_shape..get_n_axisc3s|]}|VqdSr<r"r)rrr"r#r^r_)get_simple_extent_typerGrNULLSCALARSEL_NONESEL_ALLr SIMPLErr,r rrHSEL_HYPERSLABSget_select_boundsronesrir)rZ sel_classZsel_typerdZ topcornerrr")rrrrrr# guess_shape[s8           r)N)r:numpyrbaserrrrr$r rrXr{rr"r"r"r# s  BN0`(