ó î&]\c@`sˆdZddlmZmZmZddlZddlZddlZdddddd d d d g Z d ddddddddddddddgZ ge D]Z ej e ^q¡Z iZ d„Zd„Zd„Zd„Zd „Zddd!„Zd/ded"„Zd#„Zd$„Zd%„Zed&„Zd'„Zd(„Zd)„Zd*„Zdd+„Zd,„Z d-e!fd.„ƒYZ"dS(0s, 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((s3lib/python2.7/site-packages/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%((s3lib/python2.7/site-packages/scipy/sparse/sputils.pyt upcast_char7s   cC`stjdgd|ƒ|jS(s`Determine data type for binary operation between an array of type `dtype` and a scalar. iR'(RtarrayR'(R'tscalar((s3lib/python2.7/site-packages/scipy/sparse/sputils.pyt upcast_scalarAscC`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((s3lib/python2.7/site-packages/scipy/sparse/sputils.pytdowncast_intp_indexHs!  6cC`stj|d|jjdƒƒS(NR'tnative(RtasarrayR't newbyteorder(tA((s3lib/python2.7/site-packages/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((s3lib/python2.7/site-packages/scipy/sparse/sputils.pyR`s    cC`s0tjtjƒj}tjtjƒj}tj}|dk rZ||krZtj}qZnt|tj ƒrx|f}nx±|D]©}tj |ƒ}tj |j tjƒs|r|j dkrÇqqtj|j tjƒr|jƒ}|jƒ}||kr||krqqqntj}PqqW|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(RR2tint32R1R0RRtint64t isinstancetndarrayR9R"R'R.t issubdtypetinteger(tarraysR5tcheck_contentstint32mintint32maxR'R4R6((s3lib/python2.7/site-packages/scipy/sparse/sputils.pytget_index_dtypexs,         cC`sK|jdkr+tj|tjƒr+tjStj|tjƒrGtjS|S(s Mimic numpy's casting for np.sumtu(tkindRR"tuinttint_(R'((s3lib/python2.7/site-packages/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((s3lib/python2.7/site-packages/scipy/sparse/sputils.pyR¶scC`s™tj|ƒdkrtSytj|ƒWnhttfk r”ytt|ƒ|kƒ}Wnttfk rvtSX|rt j dt ƒn|SXt S(ssIs x appropriate as an index into a sparse matrix? Returns True if it can be cast safely to a machine int. is3Inexact indices into sparse matrices are deprecated( RRTtFalsetoperatortindexR#R3R tintR?R@tDeprecationWarningtTrue(RUt loose_int((s3lib/python2.7/site-packages/scipy/sparse/sputils.pyR»s  cC`s˜y|\}}Wntk r$tSXt|ƒrt|ƒrtj|ƒdkrtj|ƒdkr| s†|dkr|dkrtSqntSdS(sjIs x a valid 2-tuple of dimensions? If nonneg, also checks that the dimensions are non-negative. iN(t ExceptionRVRRRTR[(RUtnonnegtMtN((s3lib/python2.7/site-packages/scipy/sparse/sputils.pyRÑs * cC`sYt|ttfƒr:t|ƒdkpXtj|dƒpXt|tjƒoX|jdkS(Nii(RFtlistttupletlenRRSRGRT(R%((s3lib/python2.7/site-packages/scipy/sparse/sputils.pyRãs%cC`sVt|ttfƒr7t|ƒdkr7t|dƒpUt|tjƒoU|jdkS(Nii(RFRaRbRcRRRGRT(R%((s3lib/python2.7/site-packages/scipy/sparse/sputils.pyR és"cC`st|tjƒS(N(RFRRG(RU((s3lib/python2.7/site-packages/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( RttypeRbR#RRHR'RItformatt__name__R3(taxist axis_type((s3lib/python2.7/site-packages/scipy/sparse/sputils.pyt validateaxisós    cC`såt|ƒdkr!tdƒ‚nt|ƒdkrŠyt|dƒ}Wn'tk rptj|dƒf}q Xtd„|Dƒƒ}ntd„|Dƒƒ}|dkrÿt|ƒdkrÍtdƒ‚qÀ|ddksí|ddkrÀtdƒ‚qÀnÁtj |d t ƒ}gt |ƒD]\}}|dkr!|^q!}t|ƒdkr–tj |d t ƒ}||krUtd j ||ƒƒ‚qUn¿t|ƒdkrI|d} tj |d| !|| dƒ} t || ƒ\} } | dkr&td „|Dƒƒ} td j || ƒƒ‚n|d| !| f|| d}n td ƒ‚t|ƒdkrÀtd „|Dƒƒ}t|ƒdkr˜d}qÀt|ƒdkrÀd|df}qÀnt|ƒdkrátdƒ‚n|S(s0Imitate numpy.matrix handling of shape argumentsis8function missing 1 required positional argument: 'shape'ics`s|]}tj|ƒVqdS(N(RWRX(t.0targ((s3lib/python2.7/site-packages/scipy/sparse/sputils.pys scs`s|]}tj|ƒVqdS(N(RWRX(RkRl((s3lib/python2.7/site-packages/scipy/sparse/sputils.pys sis,shape must be a 2-tuple of positive integerss#'shape' elements cannot be negativeR's-cannot reshape array of size {} into shape {}cs`s'|]}|dkrdn|VqdS(itnewshapeN((RkRU((s3lib/python2.7/site-packages/scipy/sparse/sputils.pys /ss&can only specify one unknown dimensioncs`s!|]}|dkr|VqdS(iN((RkRl((s3lib/python2.7/site-packages/scipy/sparse/sputils.pys 8ssshape too large to be a matrixN(ii(RcR#titerRWRXRbRR3RtprodRYt enumerateRftdivmod(R$t current_shapet shape_itert new_shapet current_sizetiRUtnegative_indexestnew_sizetskipt specifiedt unspecifiedt remaindert err_shape((s3lib/python2.7/site-packages/scipy/sparse/sputils.pyt check_shape sP   1   "  #  cC`s[|jddƒ}|jdtƒ}|rQtdjdj|jƒƒƒƒ‚n||fS(s?Unpack keyword arguments for reshape function. This is useful because keyword arguments after star arguments are not allowed in Python 2, but star keyword arguments are. This function unpacks 'order' and 'copy' from the star keyword arguments (with defaults) and throws an error for any remaining. tordertCtcopys/reshape() got unexpected keywords arguments: {}s, (tpopRVR#Rftjointkeys(tkwargsRR((s3lib/python2.7/site-packages/scipy/sparse/sputils.pytcheck_reshape_kwargsEs  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((s3lib/python2.7/site-packages/scipy/sparse/sputils.pyt_slicetoarangeZscC`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(tbaseR‘RFRRGRTR'RPtnonzerot_check_ellipsisRbRctsliceRt IndexErrort_check_boolean(RŠRXR‘trowtcol((s3lib/python2.7/site-packages/scipy/sparse/sputils.pyt _unpack_indexas! 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((tEllipsisR–RRFRbRpRcR0(RŠRXR‹tvtfirst_ellipsisttailtndtnslice((s3lib/python2.7/site-packages/scipy/sparse/sputils.pyR•}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.R’( R“R¢R—RFRRGR'RPt_boolean_index_to_array(RŠR™RšR¢((s3lib/python2.7/site-packages/scipy/sparse/sputils.pyR˜£s $$cC`s,|jdkrtdƒ‚n|jƒdS(Nisinvalid index shapei(RTR—R”(RŠRv((s3lib/python2.7/site-packages/scipy/sparse/sputils.pyR£²scC`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( R˜RFR–RRŒRRt atleast_1dRTR—Rtbroadcast_arrays(RŠRvR‹ti_slice((s3lib/python2.7/site-packages/scipy/sparse/sputils.pyt_index_to_arrays·s8,)  ( Rgt __module__t__doc__RR›R•R˜R£R§(((s3lib/python2.7/site-packages/scipy/sparse/sputils.pyR‡Vs   &  ((#R©t __future__RRRRWR?tnumpyRt__all__R!RUttypeDictRRR(R+R7R<RRRVRNR RRRRR R RjR~R†tobjectR‡(((s3lib/python2.7/site-packages/scipy/sparse/sputils.pyts:      "   5        ;