ó ÚÆ÷Xc@`sddZddlmZmZmZddlZddlZdddddd d d d g Zd ddddddddddddddgZ ge D]Z ej e ^q•Z iZ d„Z d„Zd„Zd„Zd „Zddd!„Zd-ded"„Zd#„Zd$„Zd%„Zd&„Zd'„Zd(„Zd)„Zd*„Zd+efd,„ƒYZdS(.s, Utility functions for sparse matrix module i(tdivisiontprint_functiontabsolute_importNtupcasttgetdtypet isscalarliket isintliketisshapet issequencetisdensetismatrixt get_sum_dtypetbooltint8tuint8tshorttushorttintctuintctlonglongt ulonglongtsingletdoublet longdoubletcsingletcdoublet clongdoublecG`s…tjt|ƒƒ}|dk r%|Stj|gƒ}x4tD],}tj||ƒr>|tt|ƒ<|Sq>Wtd|fƒ‚dS(s›Returns the nearest supported sparse dtype for the combination of one or more types. upcast(t0, t1, ..., tn) -> T where T is a supported dtype Examples -------- >>> upcast('int32') >>> upcast('bool') >>> upcast('int32','float32') >>> upcast('bool',complex,float) s%no supported conversion for types: %rN( t _upcast_memotgetthashtNonetnptfind_common_typetsupported_dtypestcan_castt TypeError(targsttR((s3/tmp/pip-build-X4mzal/scipy/scipy/sparse/sputils.pyRs  cG`sEtj|ƒ}|dk r|Stttj|ƒŒ}|t|<|S(s9Same as `upcast` but taking dtype.char as input (faster).N(RRRRtmapRtdtype(R$R%((s3/tmp/pip-build-X4mzal/scipy/scipy/sparse/sputils.pyt upcast_char6s   cC`stjdgd|ƒ|jS(s`Determine data type for binary operation between an array of type `dtype` and a scalar. iR'(RtarrayR'(R'tscalar((s3/tmp/pip-build-X4mzal/scipy/scipy/sparse/sputils.pyt upcast_scalar@scC`s±|jjtjtjƒjkr­|jdkr@|jtjƒS|jƒ}|jƒ}|tjtjƒjksŽ|tjtjƒjkrt dƒ‚n|jtjƒS|S(sž Down-cast index array to np.intp dtype if it is of a larger dtype. Raise an error if the array contains a value that is too large for intp. iszCannot deal with arrays with indices larger than the machine maximum address size (e.g. 64-bit indices on 32-bit machine).( R'titemsizeRtintptsizetastypetmaxtmintiinfot ValueError(tarrtmaxvaltminval((s3/tmp/pip-build-X4mzal/scipy/scipy/sparse/sputils.pytdowncast_intp_indexGs!  6cC`stj|d|jjdƒƒS(NR'tnative(RtasarrayR't newbyteorder(tA((s3/tmp/pip-build-X4mzal/scipy/scipy/sparse/sputils.pyt to_native[scC`sŒ|dkrZy |j}Wqˆtk rV|dk rGtj|ƒ}qWtdƒ‚qˆXn.tj|ƒ}|tjkrˆtjdƒn|S(sSFunction used to simplify argument processing. If 'dtype' is not specified (is None), returns a.dtype; otherwise returns a np.dtype object created from the specified dtype argument. If 'dtype' and 'a' are both None, construct a data type out of the 'default' parameter. Furthermore, 'dtype' must be in 'allowed' set. scould not interpret data types0object dtype is not supported by sparse matricesN(RR'tAttributeErrorRR#tobject_twarningstwarn(R'tatdefaulttnewdtype((s3/tmp/pip-build-X4mzal/scipy/scipy/sparse/sputils.pyR_s    cC`s3tjtjƒj}tj}|dk rE||krEtj}qEnt|tjƒrc|f}nxÉ|D]Á}tj |ƒ}|j tjkrj|r|j dkr¬qjqtj |j tj ƒr|jƒ}|jƒ}|tjtjƒjkr|tjtjƒjkrqjqqntj}PqjqjW|S(s& Based on input (integer) arrays `a`, determine a suitable index data type that can hold the data in the arrays. Parameters ---------- arrays : tuple of array_like Input arrays whose types/contents to check maxval : float, optional Maximum value needed check_contents : bool, optional Whether to check the values in the arrays and not just their types. Default: False (check only the types) Returns ------- dtype : dtype Suitable index data type (int32 or int64) iN(RR2tint32R0RRtint64t isinstancetndarrayR9R'R.t issubdtypetintegerR1(tarraysR5tcheck_contentstint32maxR'R4R6((s3/tmp/pip-build-X4mzal/scipy/scipy/sparse/sputils.pytget_index_dtypews,         cC`sgtj|tjƒrtjS|jdkrGtj|tjƒrGtjStj|tjƒrctjS|S(s Mimic numpy's casting for np.sumtu(RRHtfloat_tkindR"tuinttint_(R'((s3/tmp/pip-build-X4mzal/scipy/scipy/sparse/sputils.pyR ¬s$cC`s(tj|ƒp't|ƒo'|jdkS(s8Is x either a scalar, an array scalar, or a 0-dim array?i(RtisscalarR tndim(tx((s3/tmp/pip-build-X4mzal/scipy/scipy/sparse/sputils.pyR·scC`sIt|ƒstSytt|ƒ|kƒSWnttfk rDtSXdS(ssIs x appropriate as an index into a sparse matrix? Returns True if it can be cast safely to a machine int. N(RtFalseR tintR#R3(RU((s3/tmp/pip-build-X4mzal/scipy/scipy/sparse/sputils.pyR¼s  cC`sly|\}}WntSXt|ƒrdt|ƒrdtj|ƒdkrdtj|ƒdkrdtSntSdS(s(Is x a valid 2-tuple of dimensions? iN(RVRRRTtTrue(RUtMtN((s3/tmp/pip-build-X4mzal/scipy/scipy/sparse/sputils.pyRÈs*cC`sYt|ttfƒr:t|ƒdkpXtj|dƒpXt|tjƒoX|jdkS(Nii(RFtlistttupletlenRRSRGRT(R%((s3/tmp/pip-build-X4mzal/scipy/scipy/sparse/sputils.pyR×s%cC`sVt|ttfƒr7t|ƒdkr7t|dƒpUt|tjƒoU|jdkS(Nii(RFR[R\R]RRRGRT(R%((s3/tmp/pip-build-X4mzal/scipy/scipy/sparse/sputils.pyR Ýs"cC`st|tjƒS(N(RFRRG(RU((s3/tmp/pip-build-X4mzal/scipy/scipy/sparse/sputils.pyR ãscC`s¡|dk rt|ƒ}|tkr3tdƒ‚ntjtj|ƒtjƒsotdjd|j ƒƒ‚nd|ko†dknst dƒ‚qndS(NslTuples are not accepted for the 'axis' parameter. Please pass in one of the following: {-2, -1, 0, 1, None}.s#axis must be an integer, not {name}tnameiþÿÿÿisaxis out of range( RttypeR\R#RRHR'RItformatt__name__R3(taxist axis_type((s3/tmp/pip-build-X4mzal/scipy/scipy/sparse/sputils.pyt validateaxisçs    t IndexMixincB`sDeZdZd„Zd„Zd„Zd„Zd„Zd„ZRS(sT This class simply exists to hold the methods necessary for fancy indexing. cC`s+|j|ƒ\}}}tj|||ƒS(sT Given a slice object, use numpy arange to change it to a 1D array. (tindicesRtarange(tselftjtshapetstarttstoptstep((s3/tmp/pip-build-X4mzal/scipy/scipy/sparse/sputils.pyt_slicetoarangescC`sddlm}t||tjfƒrS|jdkrS|jjdkrS|jƒS|j |ƒ}t|t ƒrÍt |ƒdkr’|\}}qàt |ƒdkr¾|dt dƒ}}qàtdƒ‚n|t dƒ}}|j||ƒ\}}||fS(s… Parse index. Always return a tuple of the form (row, col). Where row/col is a integer, slice, or array of integers. i(tspmatrixitbisinvalid number of indicesN(tbaseRoRFRRGRTR'RPtnonzerot_check_ellipsisR\R]tsliceRt IndexErrort_check_boolean(RhtindexRotrowtcol((s3/tmp/pip-build-X4mzal/scipy/scipy/sparse/sputils.pyt _unpack_index s! cC`s’|tkr"tdƒtdƒfSt|tƒrŽx6t|ƒD]"\}}|tkr>|}Pq>q>Wd}|dk rŽt|ƒdkržtdƒtdƒfSt|ƒdkr|dkrù|dtkrâtdƒtdƒfStdƒ|dfSq|dtdƒfSnd}x2||dD]"}|tk r%||f}q%q%W|t|ƒ}tdd|ƒ}|| tdƒf||Sn|S(s6Process indices with Ellipsis. Returns modified index.iiiN((tEllipsisRtRRFR\t enumerateR]R0(RhRwRitvtfirst_ellipsisttailtndtnslice((s3/tmp/pip-build-X4mzal/scipy/scipy/sparse/sputils.pyRs%s2      cC`s­ddlm}||ƒs(||ƒr7tdƒ‚nt|tjƒrm|jjdkrm|j|ƒ}nt|tjƒr£|jjdkr£|j|ƒ}n||fS(Ni(t isspmatrixsoIndexing with sparse matrices is not supported except boolean indexing where matrix and index are equal shapes.Rp( RqR‚RuRFRRGR'RPt_boolean_index_to_array(RhRxRyR‚((s3/tmp/pip-build-X4mzal/scipy/scipy/sparse/sputils.pyRvKs $$cC`s,|jdkrtdƒ‚n|jƒdS(Nisinvalid index shapei(RTRuRr(Rhti((s3/tmp/pip-build-X4mzal/scipy/scipy/sparse/sputils.pyRƒZscC`s |j||ƒ\}}t|tƒ}|rY|j||jdƒdd…df}ntj|ƒ}t|tƒrà|j||jdƒddd…f}|jdkrÈ|dd…df}qŽ|sŽt dƒ‚qŽn®t |ƒr[tj|ƒ}|jdkrŽtj ||ƒ\}}|dd…df}|dd…df}||fSn3tj|ƒ}|rŽ|jdkrŽt dƒ‚ntj ||ƒ\}}|jdkrä|ddd…f}|ddd…f}n|jdkrt dƒ‚n||fS(Niisindex returns 3-dim structureisIndex dimension must be <= 2( RvRFRtRnRjRRt atleast_1dRTRuRtbroadcast_arrays(RhR„Riti_slice((s3/tmp/pip-build-X4mzal/scipy/scipy/sparse/sputils.pyt_index_to_arrays_s8,)  ( Rat __module__t__doc__RnRzRsRvRƒRˆ(((s3/tmp/pip-build-X4mzal/scipy/scipy/sparse/sputils.pyReþs   &  (( RŠt __future__RRRR?tnumpyRt__all__R!RUttypeDictRRR(R+R7R<RRRVRMR RRRRR R RdtobjectRe(((s3/tmp/pip-build-X4mzal/scipy/scipy/sparse/sputils.pyts4     "   5