a a-@s<dZddlZddlZddlZddlZddlmZgdZej ej ej ej ej ejejejejejejejejejejejejgZiZddZddZd d Zd d Zd dZ d1ddZ!d2ddZ"d3ddZ#ddZ$ddZ%ddZ&d4ddZ'dd Z(d!d"Z)d#d$Z*d%d&Z+d5d'd(Z,d)d*Z-d+d,Z.d-d.Z/d6d/d0Z0dS)7z, Utility functions for sparse matrix module N)prod) upcastgetdtypegetdata isscalarlike isintlikeisshape issequenceisdenseismatrix get_sum_dtypecGsbtt|}|dur|St|g}tD]$}t||r*|tt|<|Sq*td|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_memogethashnpZfind_common_typesupported_dtypescan_cast TypeError)argstrrd/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/scipy/sparse/sputils.pyrs    rcGs2t|}|dur|Stttj|}|t|<|S)z9Same as `upcast` but taking dtype.char as input (faster).N)r rrmaprdtype)rrrrr upcast_char6s  rcCstjdg|d|jS)z`Determine data type for binary operation between an array of type `dtype` and a scalar. rr)rarrayr)rZscalarrrr upcast_scalar@srcCsx|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_indexGs  $ r'cCstj||jddS)NZnativer)rasarrayrZ newbyteorder)Arrr to_native[sr*c Csz|durXz |j}WqvtyT}z*|dur6t|}n td|WYd}~qvd}~00nt|}|tjkrvtd|S)aQFunction 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)radefaultZnewdtypeerrrr_s  "   rFcCstj|||d}t|j|S)z This is a wrapper of `np.array(obj, dtype=dtype, copy=copy)` that will generate a warning if the result is an object array. )rcopy)rrrr)objrr1datarrrrws rrcCsttjj}ttjj}tj}|dur8||kr8tj}t|tjrJ|f}|D]l}t |}t |j tjsN|r|j dkr~qNn2t |j tjr|}|}||kr||krqNtj}qqN|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 intcZint64 isinstancendarrayr(rrr issubdtypeinteger)Zarraysr%Zcheck_contentsZint32minZint32maxrr$r&rrrget_index_dtypes,   r9cCs6|jdkrt|tjrtjSt|tjr2tjS|S)z Mimic numpy's casting for np.sumu)kindrruintint_rrrrr s r cCst|pt|o|jdkS)z8Is x either a scalar, an array scalar, or a 0-dim array?r)risscalarr ndimxrrrrsrc Cst|dkrdSzt|WnZttfyzztt||k}Wnttfy`YYdS0|rrt dt |YS0dS)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#boolintr,r-DeprecationWarning)rAZ loose_intrrrrs  rcCsnz |\}}Wnty YdS0t|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?)rAZnonnegMNrrrrs  rcCs>t|ttfr(t|dkpr6r?rrrrr s r cCs<t|ttfr&t|dkr&t|dp:t|tjo:|jdkS)Nr)r5rKrLrMr rr6r?rNrrrr s  r cCs t|tjSN)r5rr6r@rrrr sr cCsh|durdt|}|tkr tdtt|tjsFtdj|jdd|krZdksdnt ddS)NzlTuples are not accepted for the 'axis' parameter. Please pass in one of the following: {-2, -1, 0, 1, None}.z#axis must be an integer, not {name})namerJzaxis out of range) typerLrrr7rr8format__name__r#)ZaxisZ axis_typerrr validateaxissrVc Cst|dkrtdnft|dkrjzt|d}Wn"tyTt|df}Yq|0tdd|D}ntdd|D}|durt|dkrtd n |ddks|ddkrtd nt|}d d t|D}t|dkrt|}||krtd ||nt|dkr|d}t|d|||dd}t ||\} } | dkrxtdd|D} td || |d|| f||dd}ntdt|dkrtd|S)z0Imitate numpy.matrix handling of shape argumentsrz8function missing 1 required positional argument: 'shape'rJcss|]}t|VqdSrPrBrC.0argrrr zcheck_shape..css|]}t|VqdSrPrWrXrrrr["r\NrOz,shape must be a 2-tuple of positive integersz#'shape' elements cannot be negativecSsg|]\}}|dkr|qS)rr)rYirArrr /r\zcheck_shape..z-cannot reshape array of size {} into shape {}css|]}|dkrdn|VqdS)rZnewshapeNr)rYrArrrr[:r\z&can only specify one unknown dimensionz$matrix shape must be two-dimensional) rMriterrBrCrLr#r enumeraterTdivmod) rZ current_shapeZ shape_iterZ new_shapeZ current_sizeZnegative_indexesZnew_sizeskipZ specified unspecified remainderZ err_shaperrr check_shapesH          $recCs<|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. orderCr1Fz/reshape() got unexpected keywords arguments: {}z, )poprrTjoinkeys)kwargsrfr1rrrcheck_reshape_kwargsGs   rlcCs&ttjddd}|duo$t||S)zV Check whether object is pydata/sparse matrix, avoiding importing the module. sparseZ SparseArrayN)getattrsysmodulesrr5)mZbase_clsrrris_pydata_spmatrixXsrrcOstj|i|tjSrP)rrviewmatrix)rrkrrrrtgsrtcCs8t|tjr"|dus|j|kr"|Stj||dtjS)Nr)r5rrtrr(rs)r3rrrrasmatrixksru)NN)NF)rNF)F)N)N)1__doc__rorBr,ZnumpyrZscipy._lib._utilr__all__Zbool_byteZubyteshortZushortr4Zuintcr=r<ZlonglongZ ulonglongsingledoubleZ longdoubleZcsingleZcdoubleZ clongdoublerr rrrr'r*rrr9r rrrr r r rVrerlrrrtrurrrrs@ "   5   2