9=e2dZddlZddlZddlZddlmZddlmZ gdZ ej ej ej ejejejejejejejejejejejejejejgZiZdZdZdZ dZ!d Z"dd Z#dd ej$fd Z%d dZ&dZ'd e(fdZ)d e(fdZ*d!d e(fdZ+d e(fdZ,d e(fdZ-d e(fdZ.d"dZ/d#dZ0dZ1d e(fdZ2dZ3d#dZ4d ej$fdZ5dS)$z, Utility functions for sparse matrix module N)prod) upcastgetdtypegetdata isscalarlike isintlikeisshape issequenceisdenseismatrix get_sum_dtypec tt|}||Stj|}t D]2}tj||r|tt|<|cS3td|)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: ) _upcast_memogethashnp result_typesupported_dtypescan_cast TypeError)argstrs 5lib/python3.11/site-packages/scipy/sparse/_sputils.pyrrs( d$$A} ^T "F  ;vq ! ! '(Ld $HHH  B$BB C CCct|}||Stttj|}|t|<|S)z9Same as `upcast` but taking dtype.char as input (faster).)rrrmaprdtype)rrs r upcast_charr8sDA}BHd##$AL Hrc@tjdg||zjS)z`Determine data type for binary operation between an array of type `dtype` and a scalar. rr)rarrayr)rscalars r upcast_scalarr#Bs# HaS & & & / 66rc|jjtjtjjkr|jdkr|tjS|}|}|tjtjjks'|tjtjjkrtd|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).) ritemsizerintpsizeastypemaxminiinfo ValueError)arrmaxvalminvals rdowncast_intp_indexr0Is yBHRW--666 8q==::bg&& & BHRW%%) ) )Vbhrw6G6G6K-K-KHII Izz"'""" Jrct|j}|jr|Stj||dS)a Ensure that the data type of the NumPy array `A` has native byte order. `A` must be a NumPy array. If the data type of `A` does not have native byte order, a copy of `A` with a native byte order is returned. Otherwise `A` is returned. nativer )risnativerasarray newbyteorder)Adts r to_nativer8]s= B { :arx88 9 9 99rc|G |j}nq#t$r1}|tj|}ntd|Yd}~n;d}~wwxYwtj|}|tjkrt d|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)rAttributeErrorrrobject_r,)radefaultnewdtypees rrrms } HwHH H H H"8G,, ?@@aG H 8E?? rz ! !B  Os A'AAFreturnc\tj|||}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)rr!rr)objrrBdatas rrrs1 8Cu4 0 0 0D TZ Krc0tjtjtjj}tjtjtjj}tjjdkr tjn tj}|&tj|}||kr tj}t|tj r|f}|D]}tj |}tj |j tjsu|re|j dkrHtj|j tjr5|}|}||kr||krtj}n|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)rint32r+r*r)intcr%int64 isinstancendarrayr4rrr' issubdtypeinteger)arraysr.check_contentsint32minint32maxrr-r/s rget_index_dtyperSsP,x**.//Hx**.//H *a//BHHRXE &!! H  HE&"*%%joo{39bh//  !8q==]39bj99! WWYYF WWYYF))f.@.@ HE E  Lrc|jdkr+tj|tjr tjStj|tjr tjS|S)z Mimic numpy's casting for np.sumu)kindrruintint_r s rr r sK zSR[88w {5"'""w Lrc^tj|pt|o |jdkS)z8Is x either a scalar, an array scalar, or a 0-dim array?r)risscalarr ndimxs rrrs& ;q>> 9gajj8QVq[9rc4tj|dkrdS tj|ng#tt f$rS t t||k}n#tt f$rYYdSwxYw|rd}t ||cYSwxYwdS)zsIs x appropriate as an index into a sparse matrix? Returns True if it can be cast safely to a machine int. rFz4Inexact indices into sparse matrices are not allowedT)rr[operatorindexrr,boolint)r] loose_intmsgs rrrs  wqzzQu q z " SVVq[))II:&   555   "HCS// ! 4s31B A$#B$A:5B9A::BBc|t|}|s|dkrdS|D]}t|sdS|r |dkrdS dS)zIs x a valid tuple of dimensions? If nonneg, also checks that the dimensions are non-negative. If allow_ndim, shapes of any dimensionality are allowed. FrT)lenr)r]nonneg allow_ndimr[ds rr r se q66D $!))u || 55  a!ee55 4rct|ttfr-t|dkpt j|dp$t|tjo |jdkS)Nr)rKlisttuplergrrZrLr[rs rr r s\ D%= ) )/ VVq[ -BK!-- : 2: & & 8AFaK;rct|ttfr(t|dkrt |dp$t|t jo |jdkS)Nrrf)rKrmrnrgr rrLr[ros rr r s] D%= ) )- VVaZ-&qt,, 8 2: & & 616Q;9rc6t|tjSN)rKrrLr\s rr r s a $ $$rc,|dSt|}|tkrtdtjtj|tjstd|jd|cxkrdksntddS)NzlTuples are not accepted for the 'axis' parameter. Please pass in one of the following: {-2, -1, 0, 1, None}.zaxis must be an integer, not rlzaxis out of range) typernrrrMrrN__name__r,)axis axis_types r validateaxisrys |T I E011 1 =),,bj 9 9NL 8JLLMMM $OOOO!OOOO,--- Orct|dkrtdt|dkr[ t|d}td|D}nD#t$rt j|df}YnwxYwtd|D}|Lt|dkrt dtd |Drt d n$t|}d t|D}|s9t|}||kr#t d ||nt|dkr|d}t|d|||dzdz}t||\} } | dkrK|]}tj|VdSrrr_r`.0args r zcheck_shape..1s,HHchnS11HHHHHHrc3>K|]}tj|VdSrrr|r}s rrzcheck_shape..3s,>>#(.-->>>>>>rNrfz,shape must be a 2-tuple of positive integersc3"K|] }|dkV dS)rNrE)r~rjs rrzcheck_shape..8s&**1Q******rz#'shape' elements cannot be negativec$g|] \}}|dk |S)rrE)r~ir]s r zcheck_shape..?s!HHH$!Q!a%%A%%%rz-cannot reshape array of size {} into shape {}c3*K|]}|dkrdn|VdS)rnewshapeNrE)r~r]s rrzcheck_shape..Js.!P!PA**1!P!P!P!P!P!Prz&can only specify one unknown dimensionz$matrix shape must be two-dimensional) rgriterrnr_r`r,anyr enumerateformatdivmod) r current_shape shape_iter new_shape current_sizenegative_indexesnew_sizeskip specified unspecified remainder err_shapes r check_shaper&s 4yyA~~"## # 4yyA~~ Id1gJHHZHHHHHII 4 4 4!Q003III 4 >>>>>>>  y>>Q  KLL L ** *** * * DBCC C DM** IH)I*>*>HHH GIH<'' !P"(&y"A"ACCC(! " "a ' '#A&DYuu- $q&''0BBCCI%+L)%D%D "KA~~!!P!Pi!P!P!PPP  "$$*F<$C$CEEE!%4%(K>9Id1fggrs  L L LHbgrx29bgHbgrw R\IryM2:rz2>K  DDDD   777( : : : 4  BJ    4444n:t:::: D,$";T;;;; 949999 %$%%%%.........b"