U @f@sdZddlZddlZddlmZmZmZmZmZm Z m Z m Z m Z ddl mZmZddddgZd d Zd dZd dZdddZGdddZddZdS)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_matrix isspmatrixmminfommreadmmwriteMMFilecCst|tr|dSt|S)Nlatin1) isinstancebytesdecodestr)sr2/tmp/pip-unpacked-wheel-w7avvj8p/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') )rinfosourcerrrr s0cCs t|S)aG 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. Returns ------- a : ndarray or coo_matrix Dense or sparse matrix 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)) >>> m <5x5 sparse matrix of type '' with 7 stored elements in COOrdinate format> >>> m.A 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.]]) )rreadrrrrrTs-cCst||||||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_matrix >>> 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.0000000000000000e+00 0.0000000000000000e+00 0.0000000000000000e+00 2.5000000000000000e+00 0.0000000000000000e+00 0.0000000000000000e+00 0.0000000000000000e+00 6.2500000000000000e+00 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.000e+00 0.000e+00 0.000e+00 2.500e+00 0.000e+00 0.000e+00 0.000e+00 6.250e+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_matrix(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.00e+00 0.00e+00 1.00e+00 -2.00e+00 4.00e+00 3.00e+00 1.00e+00 0.00e+00 0.00e+00 5.00e+00 2.50e+00 0.00e+00 N)rwrite)targetacommentfield precisionsymmetryrrrrslc @steZdZdZeddZeddZeddZedd 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"edBd,d-Z#ed.d/Z$ed0d1Z%d2d3Z&d4d5Z'dCd8d9Z(d:d;Z)dd?Z+dDd@dAZ,d7S)Er)_rows_cols_entries_format_field _symmetrycCs|jSN)r&selfrrrrowssz MMFile.rowscCs|jSr,)r'r-rrrcolssz MMFile.colscCs|jSr,)r(r-rrrentriesszMMFile.entriescCs|jSr,)r)r-rrrformat sz MMFile.formatcCs|jSr,)r*r-rrrr#sz MMFile.fieldcCs|jSr,)r+r-rrrr%szMMFile.symmetrycCs|j|j|j|jfkSr,)r+SYMMETRY_SYMMETRICSYMMETRY_SKEW_SYMMETRICSYMMETRY_HERMITIANr-rrr has_symmetryszMMFile.has_symmetry coordinatearraycCs ||jkrtd||jfdS)Nz)unknown format type %s, must be one of %s) FORMAT_VALUES ValueError)r.r2rrr_validate_format!s zMMFile._validate_formatintegerunsigned-integerrcomplexpatterncCs ||jkrtd||jfdS)Nz(unknown field type %s, must be one of %s) FIELD_VALUESr:)r.r#rrr_validate_field0s zMMFile._validate_fieldZgeneralZ symmetriczskew-symmetricZ hermitiancCs ||jkrtd||jfdS)Nz+unknown symmetry type %s, must be one of %s)SYMMETRY_VALUESr:)r.r%rrr_validate_symmetry>s zMMFile._validate_symmetryintpuint64dDcCsdSr,rrrrrreaderKsz MMFile.readercCsdSr,rrrrrwriterPsz MMFile.writercCsZ||\}}z4|}dd|D\}}}}} |dsHtd|dks`td||dkrt|j}n|dkr|j}|r|d d kr|}q| s|}q|} ||jkrt | d kstd | d t t | \} } | | } n0t | dkstd| d t t | \} } } | | | ||| fWS|rT|XdS)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'. cSsg|]}t|qSr)rstrip).0partrrr wszMMFile.info..z%%MatrixMarketz%source is not in Matrix Market formatmatrixzProblem reading file header: r8r7r%%zHeader line not of length 2: asciizHeader line not of length 3: N)_openclosereadlinesplit startswithr:lower FORMAT_ARRAYFORMAT_COORDINATErJlenrmapint)r.rstreamclose_itlineZmmidrNr2r#r%Z split_liner/r0r1rrrrUsF           z MMFile.inforbcCszt|}Wntk r*|dfYSX|ddkrtj|stj|dr^|d}n2tj|drx|d}ntj|dr|d}|drddl}|||}n*|d rddl}| |d }n t||}n"|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.bz2rcT) osfspath TypeErrorpathisfileendswithgzipopenbz2BZ2File)Zfilespecmoderlr`rnrrrrUs.        z MMFile._openc s<j\}|krtjSd}d}jjdk}tr|\}}||k||kkrftjS fdd}nfdd}|D]\}} } |r| r|dkrd}nN|r|| krd}t j dd |r|| krd}W5QRX|r|t | krd}|s|s|sqq|rtj S|r*tjS|r6tjStjS) NTFDc3sPD]B\\}}}||kr6||f}||dfVq||kr||dfVqdS)NFT)items)ijaijaji)r!rr symm_iterators  z+MMFile._get_symmetry..symm_iteratorc3sLtD]>}t|D].}||||}}||||kfVqqdSr,)range)rtrsrurvr!nrrrws rFignore)Zover)shaperSYMMETRY_GENERALdtypecharr tocooZnonzerosumZtodoknpZerrstaterr3r4r5) r!mZissymmZisskewZishermrowcolrwrurvZ is_diagonalrryr _get_symmetrysB      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|jf|dSr,) _init_attrs)r.kwargsrrr__init__+szMMFile.__init__cCs:||\}}z||||WS|r4|XdS)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 object. Returns ------- a : ndarray or coo_matrix Dense or sparse matrix depending on the matrix format in the Matrix Market file. N)rUrV _parse_header _parse_body)r.rr`rarrrr/s  z MMFile.readrNc CsF||d\}}z|||||||W5|r8|n|XdS)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)rUrVflush_write) r.r r!r"r#r$r%r`rarrrrJs  z MMFile.writec Ksl|jj}dd|D}t|t|}|rBtdt||f|D] }t||||dddqFdS)zr Initialize each attributes with the corresponding keyword arg value or a default of None cSsg|]}|ddqS)Nr)rKattrrrrrMtsz&MMFile._init_attrs..zVfound %s invalid keyword arguments, please only use %srN) __class__ __slots__setkeysr:tuplesetattrr)r.rattrsZ public_attrsZ invalid_keysrrrrrmszMMFile._init_attrscCs2|j|\}}}}}}|j||||||ddS)N)r/r0r1r2r#r%)rrr)r.r`r/r0r1r2r#r%rrrrs   zMMFile._parse_headerc Cs|j|j|j|j|j|jf\}}}}}}zddlm}Wntk rPd}YnX|j |d} |j } ||j k} ||j k} ||jk} ||jk}||jk}||jk}||jkrVt||f| d}d}d\}}|rd|||f<||dkr|d7}|r|}|r|ddks|sq| r(t|}n2| r8t|}n"| rRttt|}nt|}||||f<| r||kr|r| |||f<n$|rt||||f<n ||||f<||dkr|d}q|d}| sd}q|}|rd|||f<||dkr|d7}q|r2|d|fkr(||dksRtdn |d|fkrJ||kstdn~||jkr|dkrt||f| d}d}d}|r|}|r|ddks|sq|}tt|dd\}}|d|d}}| rt|d}n>| r t|d}n*| r*ttt|dd}n t|d}||||f<| r||kr|rh| |||f<n$|rt||||f<n ||||f<|d}q||kstd n&||jkr|dkr|||f| dSt|d d}t|d d}|rt|d d}nH| rt|d d}n4| r(t|d d}n | rfloatz5'entries' in header is smaller than number of entriesz4'entries' in header is larger than number of entries)r|r~)"r/r0r1r2r#r% scipy.sparser ImportErrorDTYPES_BY_FIELDrr6rrrr4r5 FIELD_PATTERNr[rrWrJr_r>r^rrXrr:r\r r conjugateNotImplementedError) r.r`r/r0r1r2r#Zsymmr r~r6 is_integerZis_unsigned_integerZ is_complexZis_skewZis_hermZ is_patternr!rbrsrtruklIJVZ entry_numbermaskZod_IZod_JZod_Vrrrrs                                     zMMFile._parse_bodycCst|ts(t|ts(t|ts(t|dr|j}t|}t|jdkrLt d|j\}} |dk r||j krt |j ds|t d|d}q||jkr|j jdkr|d}q||jkr|j jdkr|d }n&t|st d t|d }|j\}} |j j} |dkr| d krd }nd}|dkr|j j} | dkrPt |j dsJt dd}n<| dkr`d}n,| dkrpd}n| dkrd}n td| |dkr||}|j||j||j|d|d|d|d} || d|dD]} d| } || dq|||}||jkr*d|| f} || d||j |j|jfkrF||j krt!| D]6}t!|D]&}||||f} || dqqtn||j"krt!| D]<}t!|d|D]&}||||f} || dqԐqnBt!| D]8}t!||D]&}||||f} || dqq n||jkr||j krt!| D]F}t!|D]6}|||f}|t#|t$|f} || dqrqfnRt!| D]H}t!||D]6}|||f}|t#|t$|f} || dqƐqn"||j%krt dn td |n|&}||j krr|j'|j(k}t)|j*||j'||j(|ff|jd!}d"|| |j+f} || d|||d}||j%krt,|j'd|j(dD]&\}}d||f} || dqn||j |j|jfkrRt,|j'd|j(d|j*D]0\}}}d#||f||} || dqnn||jkrt,|j'd|j(d|j*D]8\}}}d#||f||j#|j$f} || dqxn td |dS)$NZ __array__rRzExpected 2 dimensional arrayrDzBmmwrite does not support integer dtypes larger than native 'intp'.fdrFrqrGzunknown matrix type: %sr7ZfFrsr<frcr>ur=zunexpected dtype kind z%%MatrixMarket matrix   rz%%%s z%i %i rz*pattern type inconsisted with dense formatzUnknown field type %s)r|z %i %i %i z%i %i )-rlistrrhasattrr[rr]r|r:rr r~ OverflowErrorZastyperrrr typekindrhrrr;rArCrencoderXrrr}rxr4rrrrrrr dataZnnzzip)r.r`r!r"r#r$r%repr/r0typecoderrrbtemplatertrsruZcooZlower_triangle_maskrdrrFrrrr5s                                                 $ $z MMFile._write)rc)rNNN)rNNN)-__name__ __module__ __qualname__rpropertyr/r0r1r2r#r%r6r\r[r9 classmethodr;rrrrrr@rAr}r3r4r5rBrCr staticmethodrHrIrrUrrrrrrrrrrrrrrs             K > @  #/cCstg}zddl}||jWntk r0YnXzddl}||jWntk r^YnXt|}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)rlappendGzipFilerrnrorr)r`Zbad_clsrlrnrrr_is_fromfile_compatiblesr)rNNN)__doc__rfZnumpyrrrrrrrrr r rr r __all__rr rrrrrrrrs  , 52 p^