ó \ K]c@s4dZddlZddlmZddlmZddlmZmZm Z ddl m Z m Z m Z mZmZmZd„Zd „Zd „Zd „Zd „Zd „Zd„Ze eejejƒƒd„ƒZe ejdƒd„ƒZe ejdƒd„ƒZe ejdƒd„ƒZdS(s2 Implement slices and various slice computations. iÿÿÿÿN(tir(t zip_longest(tcgutilsttypesttypingi(t lower_builtint lower_getattrt iternext_impltimpl_ret_borrowedtimpl_ret_new_reftimpl_ret_untrackedcCsI|jd|tj|jdƒƒ}|j||ƒ}|j|||ƒS(s^ Fix negative index by adding *size* to it. Positive indices are left untouched. t=(tgetattrRtsetattrR tif_thentFalse(t bound_namet lower_replt upper_repltboundt underflowtoverflow(RRtslicetzero(s4lib/python2.7/site-packages/numba/targets/slicing.pyt fix_bound#ststarttstopN(RR Rtif_elseRt is_neg_inttstepR( RR"Rt minus_oneR$t if_neg_stept if_pos_steptlowertupper((RRR"R#s4lib/python2.7/site-packages/numba/targets/slicing.pyt fix_slices'cCs|j}|j}|j}tj|jdƒ}tj|jdƒ}tj||ƒ}|j||ƒ}|j||ƒ} |j ||ƒ} |j || | ƒ} |j ||j | |ƒƒ} |j ||j d||ƒ|j d||ƒƒ} |j | || ƒS(s Given a slice, compute the number of indices it spans, i.e. the number of iterations that for_range_slice() will execute. Pseudo-code: assert step != 0 if step > 0: if stop <= start: return 0 else: return (stop - start - 1) // step + 1 else: if stop >= start: return 0 else: return (stop - start + 1) // step + 1 (see PySlice_GetIndicesEx() in CPython) iis>=s<=( R%R&R)RR RRR(tsubRRtsdivR (Rt slicestructR%R&R)toneR#tis_step_negativetdeltat pos_dividendt neg_dividendtdividendtnominal_lengthtis_zero_length((s4lib/python2.7/site-packages/numba/targets/slicing.pytget_slice_length>s    c Cs”|j}|j}|jdƒ}|jdƒ}|jd|j|ƒ}|j||j||ƒ|ƒ}|j||j||ƒ|ƒ}||fS(s? Return the [lower, upper) indexing bounds of a slice. iiR (R%R&RR R)RR( RR2R%R&R#R3R4R-R.((s4lib/python2.7/site-packages/numba/targets/slicing.pytget_slice_boundsjs    cCs|j|j|ƒS(s4 Fix the given stride for the slice's step. (tmulR)(RR"tstride((s4lib/python2.7/site-packages/numba/targets/slicing.pyt fix_stride|scCs/|jr+tj|||jtdfƒndS(sP Guard against *slicestruct* having a zero step (and raise ValueError). sslice step cannot be zeroN(thas_stepRt guard_nullR)t ValueError(tcontextRttypR2((s4lib/python2.7/site-packages/numba/targets/slicing.pytguard_invalid_slice‚s cCs-d|jd>d}d||| ddfS(s© Get the default values for a slice's members: (start for positive step, start for negative step, stop for positive step, stop for negative step, step) ii(t address_size(RCtmaxint((s4lib/python2.7/site-packages/numba/targets/slicing.pyt get_defaults‹scs„gt|ƒD]}|jtj|ƒ^q \}}}}} dgd‰xRtt|j|ƒƒD]8\} \} } | tjkr‘dˆ| s  . $ ,   $)