o ?Hh^}@sdZddlZddlZddlmZmZmZmZmZm Z m Z m Z m Z ddl mZmZmZgdZddZdd Zd d d d ZdddZGdddZddZdS)z Matrix Market I/O in Python. See http://math.nist.gov/MatrixMarket/formats.html for information about the Matrix Market format. N) asarrayrealimagconjzerosndarray concatenateonescan_cast) coo_arrayissparse coo_matrix)mminfommreadmmwriteMMFilecCst|tr |dSt|S)Nlatin1) isinstancebytesdecodestr)srN/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/scipy/io/_mmio.pyasstrs  rcCs t|S)a Return size and storage parameters from Matrix Market file-like 'source'. Parameters ---------- source : str or file-like Matrix Market filename (extension .mtx) or open file-like object Returns ------- rows : int Number of matrix rows. cols : int Number of matrix columns. entries : int Number of non-zero entries of a sparse matrix or rows*cols for a dense matrix. format : str Either 'coordinate' or 'array'. field : str Either 'real', 'complex', 'pattern', or 'integer'. symmetry : str Either 'general', 'symmetric', 'skew-symmetric', or 'hermitian'. Examples -------- >>> from io import StringIO >>> from scipy.io import mminfo >>> text = '''%%MatrixMarket matrix coordinate real general ... 5 5 7 ... 2 3 1.0 ... 3 4 2.0 ... 3 5 3.0 ... 4 1 4.0 ... 4 2 5.0 ... 4 3 6.0 ... 4 4 7.0 ... ''' ``mminfo(source)`` returns the number of rows, number of columns, format, field type and symmetry attribute of the source file. >>> mminfo(StringIO(text)) (5, 5, 7, 'coordinate', 'real', 'general') )rinfo)sourcerrrrs 0rTspmatrixcCstj||dS)a Reads the contents of a Matrix Market file-like 'source' into a matrix. Parameters ---------- source : str or file-like Matrix Market filename (extensions .mtx, .mtz.gz) or open file-like object. spmatrix : bool, optional (default: True) If ``True``, return sparse ``coo_matrix``. Otherwise return ``coo_array``. Returns ------- a : ndarray or coo_array or coo_matrix Dense or sparse array depending on the matrix format in the Matrix Market file. Examples -------- >>> from io import StringIO >>> from scipy.io import mmread >>> text = '''%%MatrixMarket matrix coordinate real general ... 5 5 7 ... 2 3 1.0 ... 3 4 2.0 ... 3 5 3.0 ... 4 1 4.0 ... 4 2 5.0 ... 4 3 6.0 ... 4 4 7.0 ... ''' ``mmread(source)`` returns the data as sparse matrix in COO format. >>> m = mmread(StringIO(text), spmatrix=False) >>> m >>> m.toarray() array([[0., 0., 0., 0., 0.], [0., 0., 1., 0., 0.], [0., 0., 0., 2., 3.], [4., 5., 6., 7., 0.], [0., 0., 0., 0., 0.]]) r)rread)rrrrrrTs/rcCst||||||dS)a Writes the sparse or dense array `a` to Matrix Market file-like `target`. Parameters ---------- target : str or file-like Matrix Market filename (extension .mtx) or open file-like object. a : array like Sparse or dense 2-D array. comment : str, optional Comments to be prepended to the Matrix Market file. field : None or str, optional Either 'real', 'complex', 'pattern', or 'integer'. precision : None or int, optional Number of digits to display for real or complex values. symmetry : None or str, optional Either 'general', 'symmetric', 'skew-symmetric', or 'hermitian'. If symmetry is None the symmetry type of 'a' is determined by its values. Returns ------- None Examples -------- >>> from io import BytesIO >>> import numpy as np >>> from scipy.sparse import coo_array >>> from scipy.io import mmwrite Write a small NumPy array to a matrix market file. The file will be written in the ``'array'`` format. >>> a = np.array([[1.0, 0, 0, 0], [0, 2.5, 0, 6.25]]) >>> target = BytesIO() >>> mmwrite(target, a) >>> print(target.getvalue().decode('latin1')) %%MatrixMarket matrix array real general % 2 4 1 0 0 2.5 0 0 0 6.25 Add a comment to the output file, and set the precision to 3. >>> target = BytesIO() >>> mmwrite(target, a, comment='\n Some test data.\n', precision=3) >>> print(target.getvalue().decode('latin1')) %%MatrixMarket matrix array real general % % Some test data. % 2 4 1.00e+00 0.00e+00 0.00e+00 2.50e+00 0.00e+00 0.00e+00 0.00e+00 6.25e+00 Convert to a sparse matrix before calling ``mmwrite``. This will result in the output format being ``'coordinate'`` rather than ``'array'``. >>> target = BytesIO() >>> mmwrite(target, coo_array(a), precision=3) >>> print(target.getvalue().decode('latin1')) %%MatrixMarket matrix coordinate real general % 2 4 3 1 1 1.00e+00 2 2 2.50e+00 2 4 6.25e+00 Write a complex Hermitian array to a matrix market file. Note that only six values are actually written to the file; the other values are implied by the symmetry. >>> z = np.array([[3, 1+2j, 4-3j], [1-2j, 1, -5j], [4+3j, 5j, 2.5]]) >>> z array([[ 3. +0.j, 1. +2.j, 4. -3.j], [ 1. -2.j, 1. +0.j, -0. -5.j], [ 4. +3.j, 0. +5.j, 2.5+0.j]]) >>> target = BytesIO() >>> mmwrite(target, z, precision=2) >>> print(target.getvalue().decode('latin1')) %%MatrixMarket matrix array complex hermitian % 3 3 3.0e+00 0.0e+00 1.0e+00 -2.0e+00 4.0e+00 3.0e+00 1.0e+00 0.0e+00 0.0e+00 5.0e+00 2.5e+00 0.0e+00 N)rwrite)targetacommentfield precisionsymmetryrrrrslrc @seZdZdZeddZeddZeddZedd Zed d Z ed d Z eddZ dZ dZ e e fZeddZdZdZdZdZdZeeeeefZeddZdZdZdZdZeeeefZedd Zed!ed"ed#ed$ed#iZed%d&Z ed'd(Z!ed)d*Z"edDd,d-Z#ed.d/Z$ed0d1Z%d2d3Z&d4d5d6d7Z' 9 9dEd:d;Z(dd?Z*d@dAZ+ 9 9dEdBdCZ,d9S)Fr)_rows_cols_entries_format_field _symmetrycC|jSN)r(selfrrrrowsz MMFile.rowscCr.r/)r)r0rrrcolsr3z MMFile.colscCr.r/)r*r0rrrentriesr3zMMFile.entriescCr.r/)r+r0rrrformat r3z MMFile.formatcCr.r/)r,r0rrrr%r3z MMFile.fieldcCr.r/)r-r0rrrr'r3zMMFile.symmetrycCs|j|j|j|jfvSr/)r-SYMMETRY_SYMMETRICSYMMETRY_SKEW_SYMMETRICSYMMETRY_HERMITIANr0rrr has_symmetryszMMFile.has_symmetry coordinatearraycC(||jvrd|d|j}t|dS)Nzunknown format type , must be one of ) FORMAT_VALUES ValueError)r1r6msgrrr_validate_format# zMMFile._validate_formatintegerunsigned-integerrcomplexpatterncCr=)Nzunknown field type r>) FIELD_VALUESr@)r1r%rArrr_validate_field2rCzMMFile._validate_fieldgeneral symmetriczskew-symmetric hermitiancCs$||jvrtd|d|jdS)Nzunknown symmetry type r>)SYMMETRY_VALUESr@)r1r'rrr_validate_symmetry@s  zMMFile._validate_symmetryintpuint64dDcCdSr/rrrrrreaderMz MMFile.readercCrSr/rrrrrwriterRrUz MMFile.writercCsv||\}}z|}dd|D\}}}}} |ds#td|dks/td||dkr9|j}n |dkrB|j}|rX|rU|d d vrU|}nn|sD| sd|}| r\|} ||jkrt | d ks|td | d t t | \} } | | } nt | dkstd| d t t | \} } } | | | ||| fW|r|SS|r|ww)a Return size, storage parameters from Matrix Market file-like 'source'. Parameters ---------- source : str or file-like Matrix Market filename (extension .mtx) or open file-like object Returns ------- rows : int Number of matrix rows. cols : int Number of matrix columns. entries : int Number of non-zero entries of a sparse matrix or rows*cols for a dense matrix. format : str Either 'coordinate' or 'array'. field : str Either 'real', 'complex', 'pattern', or 'integer'. symmetry : str Either 'general', 'symmetric', 'skew-symmetric', or 'hermitian'. css|] }t|VqdSr/)rstrip).0partrrr yszMMFile.info..z%%MatrixMarketz%source is not in Matrix Market formatmatrixzProblem reading file header: r<r;r%%zHeader line not of length 2: asciizHeader line not of length 3: )_openreadlinesplit startswithr@lower FORMAT_ARRAYFORMAT_COORDINATElstriprWlenrmapintclose)r1rstreamclose_itlinemmidr[r6r%r' split_liner2r4r5rrrrWsT            z MMFile.inforbcCszt|}Wn ty|dfYSw|ddkrvtj|sGtj|dr.|d}ntj|dr;|d}n tj|drG|d}|drZddl}|||}|d fS|d rmddl}| |d }|d fSt||}|d fS|d ddkr|d}t||}|d fS) a Return an open file stream for reading based on source. If source is a file name, open it (after trying to find it with mtx and gzipped mtx extensions). Otherwise, just return source. Parameters ---------- filespec : str or file-like String giving file name or file-like object mode : str, optional Mode with which to open file, if `filespec` is a file name. Returns ------- fobj : file-like Open file-like object. close_it : bool True if the calling function should close this file when done, false otherwise. Frrz.mtxz.mtx.gzz.mtx.bz2z.gzNz.bz2rsT) osfspath TypeErrorpathisfileendswithgzipopenbz2BZ2File)filespecmoder|rnr~rrrrbs6            z MMFile._openc sHj\}|kr tjSd}d}jjdv}tr>\}}||k||kkr3tjS fdd}nfdd}|D]I\}} } |rX| rX|dkrXd}n1|r`|| kr`d}t j dd |rp|| krpd}Wdn1szwY|r|t | krd}|s|s|snqH|rtj S|rtjS|rtjStjS) NTFDc3sRD]!\\}}}||kr||f}||dfVq||kr&||dfVqdS)NFT)items)ijaijaji)r#rr symm_iterators  z+MMFile._get_symmetry..symm_iteratorc3sNtD]}t|D]}||||}}||||kfVq qdSr/)range)rrrrr#nrrrs rFignore)over)shaperSYMMETRY_GENERALdtypecharr tocoononzerosumtodoknperrstaterr7r8r9) r#missymmisskewishermrowcolrrr is_diagonalrrr _get_symmetrysH      zMMFile._get_symmetryc Cs0tjd|tjdtjdtjd||fi|dS)Nz%%.%ie z%i z%u z%%.%ie %%.%ie )r FIELD_REAL FIELD_INTEGERFIELD_UNSIGNED FIELD_COMPLEXget)r%r&rrr_field_template&s zMMFile._field_templatecKs|jdi|dS)Nr) _init_attrs)r1kwargsrrr__init__0szMMFile.__init__TrcCs^||\}}z||||}W|r|n|r!|ww|r-t|tr-t|}|S)aB Reads the contents of a Matrix Market file-like 'source' into a matrix. Parameters ---------- source : str or file-like Matrix Market filename (extensions .mtx, .mtz.gz) or open file object. spmatrix : bool, optional (default: True) If ``True``, return sparse ``coo_matrix``. Otherwise return ``coo_array``. Returns ------- a : ndarray or coo_array or coo_matrix Dense or sparse array depending on the matrix format in the Matrix Market file. )rb _parse_header _parse_bodyrmrr r )r1rrrnrodatarrrr4s   z MMFile.readr Nc Cs\||d\}}z|||||||W|r|dS|dS|r)|w|w)a7 Writes sparse or dense array `a` to Matrix Market file-like `target`. Parameters ---------- target : str or file-like Matrix Market filename (extension .mtx) or open file-like object. a : array like Sparse or dense 2-D array. comment : str, optional Comments to be prepended to the Matrix Market file. field : None or str, optional Either 'real', 'complex', 'pattern', or 'integer'. precision : None or int, optional Number of digits to display for real or complex values. symmetry : None or str, optional Either 'general', 'symmetric', 'skew-symmetric', or 'hermitian'. If symmetry is None the symmetry type of 'a' is determined by its values. wbN)rb_writermflush) r1r"r#r$r%r&r'rnrorrrr!Us    z MMFile.writec Ksp|jj}dd|D}t|t|}|r#tdt|d||D]}t||||dddq%dS)zr Initialize each attributes with the corresponding keyword arg value or a default of None cSsg|]}|ddqS)Nr)rXattrrrr sz&MMFile._init_attrs..zfound z, invalid keyword arguments, please only use rN) __class__ __slots__setkeysr@tuplesetattrr)r1rattrs public_attrs invalid_keysrrrrrxszMMFile._init_attrscCs2|j|\}}}}}}|j||||||ddS)N)r2r4r5r6r%r')rrr)r1rnr2r4r5r6r%r'rrrrs   zMMFile._parse_headercCs<|j|j|j|j|j|jf\}}}}}}|j|d}|j} ||j k} ||j k} ||j k} ||j k} ||j k}||jk}||jkr t||f|d}d}d\}}| rbd|||f<||dkrb|d7}|r|}|rt|ddvst|suqb| r|t|}n| rt|}n| rttt|}nt|}||||f<| r||kr| r| |||f<n|rt||||f<n||||f<||dkr|d}n|d}| sd}n|}| rd|||f<||dkr|d7}|sd| r|d|fvr||dkstd|S|d|fvr||kstd|S||jkr|dkrt||f|dSt|dd}t|dd}|r3t|dd}n$| r=t|d d}n| rGt|d d}n| rQt|d d}nt|d d}d}|D]j}|rl|ddvsl|snq[|d|krytd |}tt|dd\||<||<|s| rt|d||<n%| rt|d||<n| rttt|dd||<nt|d||<|d7}q[||krtd|d8}|d8}| r ||k}||}||}||}t||f}t||f}| r|d9}n|r|}t||f}t|||ff||f|d}|St|)N)rr)rrrr\z$Parse error, did not read all lines.intcint8rOrPrFfloatz5'entries' in header is smaller than number of entriesr_z4'entries' in header is larger than number of entries)rr) r2r4r5r6r%r'DTYPES_BY_FIELDrr:rrrr8r9 FIELD_PATTERNrgrrcrWrlrFrkrrdrr@rhr r r conjugateNotImplementedError)r1rnr2r4r5r6r%symmrr: is_integeris_unsigned_integer is_complexis_skewis_herm is_patternr#rprrrIJV entry_numberlmaskod_Iod_Jod_Vrrrrs                  "JG          zMMFile._parse_bodycCst|tst|tst|tst|drf|j}t|}t|jdkr&t d|j\}} |dure||j krDt |j ds>t d|d}n6||jkrU|j jdvrT|d}n%||jkre|j jdvre|d }nt|sst d t|d }|j\}} |j j} |dur| d vrd }nd}|dur|j j} | dkrt |j dst dd}n| dkrd}n| dkrd}n | dkrd}ntd| |dur||}|j||j||j|d|d|d|d} || d|dD]} d| d} || dq|||}||jkrd|| f} || d||j |j|jfvr||j krQt!| D]}t!|D]}||||f} || dq9q3dS||j"kr|t!| D]}t!|d|D]}||||f} || dqdq[dSt!| D]}t!||D]}||||f} || dqqdS||jkr||j krt!| D]#}t!|D]}|||f}|t#|t$|f} || dqqdSt!| D]$}t!||D]}|||f}|t#|t$|f} || dqqdS||j%kr t dtd ||&}||j kr5|j'|j(k}t)|j*||j'||j(|ff|jd!}d"|| |j+f} || d|||d}||j%krtt,|j'd|j(dD]\}}d||f} || dq^dS||j |j|jfvrt,|j'd|j(d|j*D]\}}}d#||f||} || dqdS||jkrt,|j'd|j(d|j*D]\}}}d#||f||j#|j$f} || dqdStd |)$N __array__r_zExpected 2 dimensional arrayrOzBmmwrite does not support integer dtypes larger than native 'intp'.fdrQrrRzunknown matrix type: r;fFrrDfrcrFurEzunexpected dtype kind z%%MatrixMarket matrix   rr]z%i %i rz*pattern type inconsisted with dense formatzUnknown field type )rz %i %i %i z%i %i )-rlistrrhasattrrgrrjrr@rr r OverflowErrorastyperrrr typekindrxrrrBrIrNr!encoderdrrrrr8rrrrrrr rnnzzip)r1rnr#r$r%r&r'repr2r4typecoderrrptemplaterrrcoolower_triangle_maskrtrrQrrrrs                                          $ $z MMFile._write)rsr NNN)-__name__ __module__ __qualname__rpropertyr2r4r5r6r%r'r:rhrgr? classmethodrBrrrrrrHrIrr7r8r9rMrNr staticmethodrTrVrrbrrrrr!rrrrrrrrrs             N > @  ! # rcCspg}z ddl}||jWn tyYnwz ddl}||jWn ty-Ynwt|}t|| S)z Check whether `stream` is compatible with numpy.fromfile. Passing a gzipped file object to ``fromfile/fromstring`` doesn't work with Python 3. rN)r|appendGzipFile ImportErrorr~rrr)rnbad_clsr|r~rrr_is_fromfile_compatibles   rr)__doc__rvnumpyrrrrrrrrr r scipy.sparser r r __all__rrrrrrrrrrs" ,5 4p @