ó ¾÷Xc@@südZddlmZddlZddlmZddlZddlm Z m Z d„Z de fd „ƒYZ d e fd „ƒYZd efd „ƒYZdefd„ƒYZdefd„ƒYZd„Zd„Zd„Zd„Zd„ZdS(s4 High-level access to HDF5 dataspace selections i(tabsolute_importN(txrangei(th5sth5rcC@s“t|tƒs|f}nt|ƒdkrç|d}t|tƒrh|j|krdtdƒ‚n|St|tjƒr’t|ƒ}|||St|t j ƒrçt j ||ƒ}||jkrÔtdƒ‚nt|d|ƒSnx‘|D]‰}t|t ƒ rî|t k rîy;t|ƒt|tjƒrJ|jdkrJtƒ‚nWqwtk rst|ƒ}|||SXqîqîWt|ƒ}|||S(s& 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. dsid A h5py.h5d.DatasetID 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 only Returns a SimpleSelection instance Indices, slices, ellipses, lists or boolean index arrays Returns a FancySelection instance. iisMismatched selection shapes,Reference shape does not match dataset shapetspaceid(i(t isinstancettupletlent Selectiontshapet TypeErrortnptndarraytPointSelectionRtRegionReferencet get_regiontslicetEllipsistintt ExceptiontFancySelectiontSimpleSelection(R targstdsidtargtseltsidta((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pytselects<     !    t _RegionProxycB@s eZdZd„Zd„ZRS(sû Thin proxy object which takes __getitem__-style index arguments and produces RegionReference objects. Example: >>> dset = myfile['dataset'] >>> myref = dset.regionref[0:100,20:30] >>> data = dset[myref] cC@s ||_dS(s& Supply a h5py.h5d.DatasetID instance N(tid(tselfR((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyt__init__mscC@s:t|jj||jƒ}tj|jdtj|jƒS(sˆ Takes arbitrary selection terms and produces a RegionReference object. Selection must be compatible with the dataset. t.(RRR RtcreatetDATASET_REGION(RRt selection((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyt __getitem__qs(t__name__t __module__t__doc__R R%(((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyRas  RcB@sheZdZdd„Zed„ƒZed„ƒZed„ƒZed„ƒZ d„Z d„Z RS( s‰ 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). cC@so|dk r$||_|j|_nGt|ƒ}||_tj|tjft|ƒƒ|_|jj ƒdS(s= Create a selection. Shape may be None if spaceid is given. N( tNonet_idR t_shapeRRt create_simplet UNLIMITEDRt select_all(RR R((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyR “s    %cC@s|jS(s SpaceID instance (R*(R((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyRžscC@s|jS(s Shape of whole dataspace (R+(R((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyR £scC@s |jjƒS(s' Number of elements currently selected (R*tget_select_npoints(R((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pytnselect¨scC@s |jfS(s0 Shape of selection (always 1-D for this class) (R0(R((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pytmshape­scc@s3tj|ƒ|jkr'tdƒ‚n|jVdS(s" Get an iterable for broadcasting s7Broadcasting is not supported for point-wise selectionsN(R tproductR0R R*(Rt target_shape((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyt broadcast²scC@stdƒ‚dS(Ns$This class does not support indexing(tNotImplementedError(RR((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyR%¸sN( R&R'R(R)R tpropertyRR R0R1R4R%(((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyRxs  R cB@s;eZdZd„Zd„Zd„Zd„Zd„ZRS(s Represents a point-wise selection. You can supply sequences of points to the three methods append(), prepend() and set(), or a single boolean array to __getitem__. cC@s¦tj|ddddƒ}t|jƒdkrId|jdf|_n|jjƒtjkrmtj}nt|ƒdkr|jj ƒn|jj ||ƒdS(s7 Internal method which actually performs the selection tordertCtdtypetu8iiN( R tasarrayRR R*tget_select_typeRt SEL_POINTSt SELECT_SETt select_nonetselect_elements(Rtpointstop((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyt_perform_selectionÃs cC@szt|tjƒo!|jjdks3tdƒ‚n|j|jksTtdƒ‚ntj|jƒƒ}|j |ƒ|S(s9 Perform point-wise selection from a NumPy boolean array tbs6PointSelection __getitem__ only works with bool arrayss-Boolean indexing array has incompatible shape( RR R R9tkindR R t transposetnonzerotset(RRRA((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyR%Ñs$ cC@s|j|tjƒdS(s@ Add the sequence of points to the end of the current selection N(RCRt SELECT_APPEND(RRA((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pytappendÜscC@s|j|tjƒdS(sF Add the sequence of points to the beginning of the current selection N(RCRtSELECT_PREPEND(RRA((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pytprependàscC@s|j|tjƒdS(s@ Replace the current selection with the given sequence of pointsN(RCRR>(RRA((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyRHäs(R&R'R(RCR%RJRLRH(((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyR »s    RcB@s8eZdZed„ƒZd„Zd„Zd„ZRS(s‰ A single "rectangular" (regular) selection composed of only slices and integer arguments. Can participate in broadcasting. cC@s|jS(s Shape of current selection (t_mshape(R((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyR1ïscO@s\tj||||Žt|jƒ}d||jd|tf|f|_|j|_dS(Nii(i(i(RR RR tFalset_selRM(RR Rtkwdstrank((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyR ôs'cC@sát|tƒs|f}n|jdkrrt|ƒdkra|dtdfkratdƒ‚n|jjƒ|St|j|ƒ\}}}}|jj |||ƒ||||f|_ td„t ||ƒDƒƒ|_ |S(Nis7Invalid index for scalar dataset (only ..., () allowed)cs@s!|]\}}|s|VqdS(N((t.0txty((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pys s((( RRR RRR R*R.t_handle_simpletselect_hyperslabROtzipRM(RRtstarttcounttsteptscalar((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyR%ús ( "cc@s|jdkrPtj|ƒdkr7td|ƒ‚n|jjƒ|jVdS|j\}}}}t|ƒ}t|ƒ}g}x—t d|dƒD]‚} t|ƒdksº|| rÊ|j dƒq—|j ƒ} | dksó|| | kr|j | ƒq—td||fƒ‚q—W|j ƒt |ƒ}t d„t||ƒDƒƒ} ttj| ƒƒ} | dkr~|jVn€|jjƒ} | jd |||ƒxWt | ƒD]I} t d„ttj| | ƒ|||ƒDƒƒ}| j|ƒ| Vq±WdS( s· Return an iterator over target dataspaces for broadcasting. Follows the standard NumPy broadcasting rules against the current selection shape (self.mshape). isCan't broadcast %s to scalarNisCan't broadcast %s -> %scs@s|]\}}||VqdS(N((RRRSRT((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pys /scs@s-|]#\}}}}||||VqdS(N((RRRSRTtzts((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pys 8s((i(R R R2R R*R.RORtlistRRJtpoptreverseRRWRtcopyRVt unravel_indext offset_simple(RR3RXRYRZR[RQttargetttshapetidxtttchunkstnchunksRtoffset((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyR4s:        1 (R&R'R(R6R1R R%R4(((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyRés   RcB@s8eZdZed„ƒZd„Zd„Zd„ZRS(s( Implements advanced NumPy-style selection operations in addition to the standard slice-and-int behavior. Indexing arguments may be ints, slices, lists of indicies, or per-axis (1D) boolean arrays. Broadcasting is not supported for these selections. cC@s|jS(N(RM(R((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyR1IscO@s&tj||||Ž|j|_dS(N(RR R RM(RR RRP((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyR Msc@st|tƒs|f}nt|t|jƒƒ}i}xút|ƒD]ì\}}t|tƒsFt|dƒr¿|jt jdƒkr¿t|jƒdkr¬t dƒ‚n|j ƒd}nyt |ƒ||hss-Indexing elements must be in increasing ordersMOnly one indexing vector or array is currently allowed for advanced selections Advanced selection inappropriatec3@s!|]}t|ƒˆkVqdS(N(R(RRRS(t vectorlength(s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pys qss3All sequence arguments must have the same length %sRBcs@s!|]}|dkr|VqdS(iN((RRRS((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pys s(RRt_expand_ellipsisRR t enumerateRthasattrR9R R RGR^RWtanytvaluestallRtsixt iteritemsRJR*R?RURVRt SELECT_ORRM(RRt sequenceargsRfRtlist_argtadjacentt argvectortentrytpositiontseqtvectorRXRYRZR[R1((Rns1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyR%QsT '  "  #   cc@s*||jkstdƒ‚n|jVdS(Ns4Broadcasting is not supported for complex selections(R1R R*(RR3((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyR4‘s(R&R'R(R6R1R R%R4(((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyR=s    @cC@sëtd„|Dƒƒ}|dkr1tdƒ‚n.|dkr_t|ƒ|kr_|tf}ng}t|ƒ}xR|D]J}|tkrµ|jtdddƒf||dƒqx|j|ƒqxWt|ƒ|krçtdƒ‚n|S(s7 Expand ellipsis objects and fill in missing axes. cs@s!|]}|tkrdVqdS(iN(R(RRR((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pys ™sisOnly one ellipsis may be used.isArgument sequence too longN( tsumt ValueErrorRRtextendRR)RJR (RRQtn_elt final_argstn_argsR((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyRo–s    +c C@s%t|t|ƒƒ}g}g}g}g}xÍt||ƒD]¼\}}t|tƒryt||ƒ\}} } t} nLy(tt|ƒ|ƒ\}} } t } Wn!t k rÄt d|ƒ‚nX|j |ƒ|j | ƒ|j | ƒ|j | ƒq=Wt |ƒt |ƒt |ƒt |ƒfS(sU Process a "simple" selection tuple, containing only slices and integer objects. Return is a 4-tuple with tuples for start, count, step, and a flag which tells if the axis is a "scalar" selection (indexed by an integer). If "args" is shorter than "shape", the remaining axes are fully selected. s.Illegal index "%s" (must be a slice or number)( RoRRWRRt_translate_sliceRNt_translate_intRtTrueR RJR( R RRXRYRZR[RtlengthRSRTR\R]((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyRU­s&       cC@s_|dkr||}nd|ko0|knsRtd||dfƒ‚n|ddfS(sk Given an integer index, return a 3-tuple (start, count, step) for hyperslab selection isIndex (%s) out of range (0-%s)i(R(texpR‰((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyR‡Îs   cC@su|j|ƒ\}}}|dkr7td|ƒ‚n||krRtdƒ‚nd||d|}|||fS(s Given a slice object, return a 3-tuple (start, count, step) for use with the hyperslab selection routines isStep must be >= 1 (got %d)s(Reverse-order selections are not allowed(tindicesR(RŠR‰RXtstopRZRY((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyR†Ûs  c@s¬ˆjƒ}ˆjƒ}|tjkr+dS|tjkrf|tjkrMdS|tjkrˆtƒSn"|tj krˆt d|ƒ‚nˆj ƒ‰t ˆj ƒ}|tjkrºd|S|tjkrЈj S|tjkræˆfS|tjkrt d|ƒ‚nˆdkrd |Sd„ˆjƒDƒ\‰}|ˆtj|fƒ‰‡‡‡fd†‰t‡‡fd†t|ƒDƒƒ}tj|ƒˆkr¨ˆfS|S( s1 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 sUnrecognized dataspace class %sis Unrecognized selection method %scs@s|]}tj|ƒVqdS(N(R tarray(RRRS((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pys sc@s‘ˆ|dkrdSˆjƒ}||cd7<ˆjƒ}||cd8<|jƒ}|jt|ƒt|ƒdtjƒ|jƒ}ˆ|S(sä 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. iRB(RaRVRRt SELECT_NOTBR/(RtaxisRXRYt masked_sidt N_leftover(tNt bottomcornertboxshape(s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyt get_n_axis#s    % c3@s|]}ˆˆ|ƒVqdS(N((RRRS(R•R(s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pys ?sN(i(i(tget_simple_extent_typeR<RtNULLR)tSCALARtSEL_NONEtSEL_ALLRtSIMPLER R/RR R=tSEL_HYPERSLABStget_select_boundsR tonesRR2(Rt sel_classtsel_typeRQt topcornerR ((R’R“R”R•Rs1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyt guess_shapeîs<     %(R(t __future__RRut six.movesRtnumpyR tRRRtobjectRRR RRRoRUR‡R†R¢(((s1/tmp/pip-build-isqEY4/h5py/h5py/_hl/selections.pyts    FC.TY  !