Dj:*dZddlZddlmZmZmZmZddlZddlZ ddl m Z ddl m ZddlmZmZgdZe j&e j(e j*e j,e j.e j0e j2eee j4e j6e j8e j:e j<e j>e j@e jBgZ"iZ#dZ$dZ%d Z&d Z'd Z(d'd Z)d(de jTfdZ+d)dZ,de jZde jZfdZ.de/fdZ0de/fdZ1d*d dde/fdZ2de/fdZ3de/fdZ4de/fdZ5d+dZ6d,d dde7e8dffdZ9dZ:de/fdZ; d,ded eed!deed"ffd#Zde jTfd&Z?y)-z, Utility functions for sparse matrix module N)AnyLiteralOptionalUnion)prod)np_longnp_ulong) upcastgetdtypegetdata isscalarlike isintlikeisshape issequenceisdenseismatrix get_sum_dtypectjt|}||Stj|}t D].}tj ||s|tt|<|cStd|)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 -------- >>> from scipy.sparse._sputils import upcast >>> upcast('int32') >>> upcast('bool') >>> upcast('int32','float32') >>> upcast('bool',complex,float) z#no supported conversion for types: ) _upcast_memogethashnp result_typesupported_dtypescan_cast TypeError)argstr s YC:\Crop_Prediction\Backend\crop-ai-system\venv\Lib\site-packages\scipy/sparse/_sputils.pyr r su( d$A} ^^T "F  ;;vq !'(Ld $H 9$B CCctj|}||Stttj |}|t|<|S)z9Same as `upcast` but taking dtype.char as input (faster).)rrr maprdtype)rrs r upcast_charr$:s@A}BHHd#$AL Hr cLtjdg||zjS)z`Determine data type for binary operation between an array of type `dtype` and a scalar. rr#)rarrayr#)r#scalars r upcast_scalarr)Ds" HHaS & / 6 66r c>|jjtjtjjkDr|jdk(r|j tjS|j }|j}|tjtjj kDs0|tjtjjkr td|j 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).) r#itemsizerintpsizeastypemaxminiinfo ValueError)arrmaxvalminvals rdowncast_intp_indexr6Ks yyBHHRWW-666 88q=::bgg& & BHHRWW%)) )Vbhhrww6G6K6K-KHI Izz"''"" Jr c|j}|jr|Stj||j 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&)r#isnativerasarray newbyteorder)Adts r to_nativer>_s5 B {{ ::arx8 99r c| |j}|Stj|}|tjk(r t d|S#t$r/}|tj|}n td|Yd}~|Sd}~wwxYw)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)r#AttributeErrorrrobject_r2)r#adefaultnewdtypees rr r os } HwwH O 88E? rzz !B  O H"88G, ?@aG O Hs A A?$A::A?Freturnc`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. )r#copy)rr'r r#)objr#rHdatas rr r s) 88Cu4 0D TZZ Kr ctjtjtjj}tjtjtjj}tj j dk(rtjntj}|*tj|}||kDrtj}t|tjr|f}|D]}tj|}tj|jtjrG|ri|jdk(rYtj|jtjr+|j }|j}||k\r||krtj}|S|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) r)rint32r1r0r/intcr+int64 isinstancendarrayr:rr#r- issubdtypeinteger)arraysr4check_contentsint32minint32maxr#r3r5s rget_index_dtyperXs=,xx*../Hxx*../H **a/BHHRXXE &! H HHE&"**%jjo{{399bhh/88q=]]399bjj9 WWYF WWYF)f.@ HHE  L#" Lr r#c|jdk(r4tj|tjrtjStj|tjrtjS|S)z Mimic numpy's casting for np.sumu)kindrruintint_r&s rrrsH zzSR[[8ww {{5"''"ww Lr chtj|xst|xr|jdk(S)z8Is x either a scalar, an array scalar, or a 0-dim array?r)risscalarrndimxs rr r s& ;;q> 9gaj8QVVq[9r c tj|dk7ry tj|y#tt f$rC t t||k(}n#tt f$rYYywxYw|r d}t ||cYSwxYw)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`operatorindexrr2boolint)rb loose_intmsgs rrrs  wwqzQ q  z " SVq[)I:&   HCS/ !s31BABA,(B+A,,BB)allow_1dcpt|}|dk7r|r|dk(sy|D]}t|sy|s|dksyy)zIs x a valid tuple of dimensions? If nonneg, also checks that the dimensions are non-negative. If allow_1d, shapes of length 1 or 2 are allowed. FrT)lenr)rbnonnegrjr`ds rrrsI q6D qy(tqy | a!e  r ct|ttfxr(t|dk(xst j |dxs+t|tj xr|jdk(S)Nrrm)rPlisttuplernrr_rQr`rs rrrsW D%= )/ Vq[ -BKK!- : 2:: & 8AFFaK;r ct|ttfxrt|dkDxrt |dxs+t|t j xr|jdk(S)Nrrl)rPrrrsrnrrrQr`rts rrrsV D%= )- VaZ-&qt, 8 2:: & 6166Q;9r c6t|tjSN)rPrrQras rrr s a $$r c*|yt|}|tk(r tdtjtj |tj std|jd|cxkrdkstdtdy)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 rmzaxis out of range) typersrrrRr#rS__name__r2)axis axis_types r validateaxisr~s |T I E01 1 ==),bjj 97 8J8J7KLMM $O!O,-- ,-- r .ct|dk(r tdt|dk(r" t|d}td|D}ntd|D}|P|rt|dvr$t dt|d k7r t d td |Drt d t|}t|Dcgc] \}}|dks |}}}|s+t|} | |k7rt d j||t|dk(rp|d} t|d| || dzdz} t|| \} } | dk7r-td|D}t d j|||d| | fz|| dzdz}n t dt|d k7r|rt|dk(s t d|S#t$rt j |df}YwxYwcc}}w)auImitate numpy.matrix handling of shape arguments Parameters ---------- args : array_like Data structures providing information about the shape of the sparse array. current_shape : tuple, optional The current shape of the sparse array or matrix. If None (default), the current shape will be inferred from args. allow_1d : bool, optional If True, then 1-D or 2-D arrays are accepted. If False (default), then only 2-D arrays are accepted and an error is raised otherwise. Returns ------- new_shape: tuple The new shape after validation. rz8function missing 1 required positional argument: 'shape'rmc3FK|]}tj|ywrwrdre.0args r zcheck_shape..EsHchnnS1H!c3FK|]}tj|ywrwrrs rrzcheck_shape..Gs>#(..->rN)rmrlz2shape must be a 1- or 2-tuple of positive integersrlz,shape must be a 2-tuple of positive integersc3&K|] }|dk yw)rN)rrps rrzcheck_shape..Ps(q1u(sz#'shape' elements cannot be negativez-cannot reshape array of size {} into shape {}c3.K|] }|dkrdn|yw)rnewshapeNr)rrbs rrzcheck_shape..bs!PA*1">>  9~V+ ",-- ^q KL L (i( (BC CM* +4I*>H$!Q!a%AHHIH<' !P"(&y"ACC ! "a '#A&DYu- $q&'0BBCI%+L)%D "KA~!!Pi!PP  "$$*F<$CEE!%4(K>9Id1fgr rQr rXr#rrfr rrrrrr~rsrgrrrrrrrrr rrs 00. LHHbggrxx299bggHHgxbllJJ BMMLL"--A  DD 7( : 4 BJJ 4nbhh:t: D,4";T; 949 %$%..DeDc3hDN"