B &]\=@sdZddlmZmZmZddlZddlZddlZdddddd d d d g Z d ddddddddddddddgZ dde DZ iZ ddZ dd Z d!d"Zd#d$Zd%d&Zd;d'dZdd5d6Zd7d8ZGd9d:d:eZdS)?z, Utility functions for sparse matrix module )divisionprint_functionabsolute_importNupcastgetdtype isscalarlike isintlikeisshape issequenceisdenseismatrix get_sum_dtypeboolZint8Zuint8ZshortZushortintcZuintcZlonglongZ ulonglongZsingleZdoubleZ longdoubleZcsingleZcdoubleZ clongdoublecCsg|]}tj|qS)npZtypeDict).0xrr3lib/python3.7/site-packages/scipy/sparse/sputils.py srcGsbtt|}|dk r|St|g}x(tD] }t||r,|tt|<|Sq,Wtd|fdS)aReturns 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) Nz%no supported conversion for types: %r) _upcast_memogethashrZfind_common_typesupported_dtypescan_cast TypeError)argstrrrrrs    cGs2t|}|dk r|Stttj|}|t|<|S)z9Same as `upcast` but taking dtype.char as input (faster).N)rrrmaprdtype)rrrrr upcast_char7s  r cCstjdg|d|jS)z`Determine data type for binary operation between an array of type `dtype` and a scalar. r)r)rZarrayr)rZscalarrrr upcast_scalarAsr!cCsx|jjttjjkrt|jdkr,|tjS|}|}|ttjjks`|ttjjkrht d|tjS|S)z 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. rzzCannot deal with arrays with indices larger than the machine maximum address size (e.g. 64-bit indices on 32-bit machine).) ritemsizerZintpsizeZastypemaxminiinfo ValueError)arrmaxvalminvalrrrdowncast_intp_indexHs  $ r+cCstj||jddS)NZnative)r)rasarrayrZ newbyteorder)Arrr to_native\sr.cCsh|dkrFy |j}Wqdtk rB|dk r6t|}ntdYqdXnt|}|tjkrdtd|S)aSFunction 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. Nzcould not interpret data typez0object dtype is not supported by sparse matrices)rAttributeErrorrrZobject_warningswarn)radefaultZnewdtyperrrr`s     rFcCsttjj}ttjj}tj}|dk r8||kr8tj}t|tjrJ|f}xr|D]j}t |}t |j tjsP|r|j dkrqPn2t |j tjr|}|}||kr||krqPtj}PqPW|S)a& 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) Nr)rr&Zint32r%r$rZint64 isinstancendarrayr,rrr# issubdtypeinteger)Zarraysr)Zcheck_contentsZint32minZint32maxrr(r*rrrget_index_dtypexs,    r8cCs6|jdkrt|tjrtjSt|tjr2tjS|S)z Mimic numpy's casting for np.sumu)kindrrZuintZint_)rrrrr s cCst|pt|o|jdkS)z8Is x either a scalar, an array scalar, or a 0-dim array?r)risscalarr ndim)rrrrrsc Cs|t|dkrdSyt|WnVttfk rvytt||k}Wnttfk r`dSX|rrt dt |SXdS)zsIs x appropriate as an index into a sparse matrix? Returns True if it can be cast safely to a machine int. rFz3Inexact indices into sparse matrices are deprecatedT) rr<operatorindexrr'rintr0r1DeprecationWarning)rZ loose_intrrrrscCsny |\}}Wntk r dSXt|rft|rft|dkrft|dkrf|rb|dkrf|dkrfdSdSdS)zjIs x a valid 2-tuple of dimensions? If nonneg, also checks that the dimensions are non-negative. FrTN) Exceptionrrr<)rZnonnegMNrrrr s cCs>t|ttfr(t|dkp)rargrrr szcheck_shape..css|]}t|VqdS)N)r=r>)rrOrrrrPsNrHz,shape must be a 2-tuple of positive integersz#'shape' elements cannot be negative)rcSsg|]\}}|dkr|qS)rr)rirrrrr$szcheck_shape..z-cannot reshape array of size {} into shape {}css|]}|dkrdn|VqdS)rZnewshapeNr)rrrrrrP/sz&can only specify one unknown dimensioncss|]}|dkr|VqdS)rDNr)rrOrrrrP8s)rDrDzshape too large to be a matrix) rGriterr=r>rFr'rZprodr? enumeraterLdivmod) rZ current_shapeZ shape_iterZ new_shapeZ current_sizeZnegative_indexesZnew_sizeskipZ specifiedZ unspecifiedZ remainderZ err_shaperrr check_shape sP        "  $ rVcCs<|dd}|dd}|r4tdd|||fS)a?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. orderCcopyFz/reshape() got unexpected keywords arguments: {}z, )poprrLjoinkeys)kwargsrWrYrrrcheck_reshape_kwargsEs  r^c@s@eZdZdZddZddZddZdd Zd d Zd d Z dS) IndexMixinzT This class simply exists to hold the methods necessary for fancy indexing. cCs||\}}}t|||S)zT Given a slice object, use numpy arange to change it to a 1D array. )indicesrZarange)selfjshapestartstopsteprrr_slicetoarangeZszIndexMixin._slicetoarangecCsddlm}t||tjfr:|jdkr:|jjdkr:|S| |}t|t rt |dkrd|\}}qt |dkr|dt d}}qt dn|t d}}|||\}}||fS)z Parse index. Always return a tuple of the form (row, col). Where row/col is a integer, slice, or array of integers. rD)spmatrixrHbrNzinvalid number of indices)baserhr4rr5r<rr:nonzero_check_ellipsisrFrGslice IndexError_check_boolean)rar>rhrowcolrrr _unpack_indexas       zIndexMixin._unpack_indexcCs0|tkrtdtdfSt|tr,x&t|D]\}}|tkr.|}Pq.Wd}|dk r,t|dkrrtdtdfSt|dkr|dkr|dtkrtdtdfStd|dfSn|dtdfSd}x*||ddD]}|tk r||f}qW|t|}tdd|}|d|tdf||S|S)z6Process indices with Ellipsis. Returns modified index.NrDrHrr)Ellipsisrmr4rFrSrGr$)rar>rbvZfirst_ellipsistailZndZnslicerrrrl}s2      zIndexMixin._check_ellipsiscCspddlm}||s||r$tdt|tjrF|jjdkrF||}t|tjrh|jjdkrh||}||fS)NrD) isspmatrixzoIndexing with sparse matrices is not supported except boolean indexing where matrix and index are equal shapes.ri) rjrvrnr4rr5rr:_boolean_index_to_array)rarprqrvrrrros   zIndexMixin._check_booleancCs|jdkrtd|dS)NrDzinvalid index shaper)r<rnrk)rarQrrrrws z"IndexMixin._boolean_index_to_arraycCsn|||\}}t|t}|r>|||jddddf}n t|}t|tr|||jddddf}|jdkr|dddf}n |stdnzt |rt|}|jdkrt ||\}}|dddf}|dddf}||fSn$t|}|r|jdkrtdt ||\}}|jdkrR|dddf}|dddf}n|jdkrftd||fS)NrrDzindex returns 3-dim structurerHzIndex dimension must be <= 2) ror4rmrgrcrZ atleast_1dr<rnrZbroadcast_arrays)rarQrbZi_slicerrr_index_to_arrayss8            zIndexMixin._index_to_arraysN) rM __module__ __qualname____doc__rgrrrlrorwrxrrrrr_Vs&r_)NN)rNF)F)N)r{Z __future__rrrr=r0Znumpyr__all__rrrr r!r+r.rr8r rrr r r r rNrVr^objectr_rrrrs:   "   5   ;