o i|@sdZddlZddlZddlmZmZmZmZmZm Z m Z m Z m Z ddl mZmZgdZddZdd Zd d Zdd dZGdddZddZdS)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_matrixissparse)mminfommreadmmwriteMMFilecCst|tr |dSt|S)Nlatin1) isinstancebytesdecodestr)sr]/var/www/html/eduruby.in/lip-sync/lip-sync-env/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') )rinfosourcerrrr s 0r cCs 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 -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_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 precisionsymmetryrrrrslrc @s|eZdZdZeddZeddZeddZedd 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' 7 7dCd8d9Z(d:d;Z)dd?Z+ 7 7dCd@dAZ,d7S)Dr)_rows_cols_entries_format_field _symmetrycC|jSN)r&selfrrrrowsz MMFile.rowscCr,r-)r'r.rrrcolsr1z MMFile.colscCr,r-)r(r.rrrentriesr1zMMFile.entriescCr,r-)r)r.rrrformat r1z MMFile.formatcCr,r-)r*r.rrrr#r1z MMFile.fieldcCr,r-)r+r.rrrr%r1zMMFile.symmetrycCs|j|j|j|jfvSr-)r+SYMMETRY_SYMMETRICSYMMETRY_SKEW_SYMMETRICSYMMETRY_HERMITIANr.rrr has_symmetryszMMFile.has_symmetry coordinatearraycC ||jvrtd||jfdS)Nz)unknown format type %s, must be one of %s) FORMAT_VALUES ValueError)r/r4rrr_validate_format! zMMFile._validate_formatintegerunsigned-integerrcomplexpatterncCr;)Nz(unknown field type %s, must be one of %s) FIELD_VALUESr=)r/r#rrr_validate_field0r?zMMFile._validate_fieldZgeneralZ symmetriczskew-symmetricZ hermitiancCr;)Nz+unknown symmetry type %s, must be one of %s)SYMMETRY_VALUESr=)r/r%rrr_validate_symmetry>r?zMMFile._validate_symmetryintpuint64dDcCdSr-rrrrrreaderKz MMFile.readercCrLr-rrrrrwriterPrNz 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 wszMMFile.info..z%%MatrixMarketz%source is not in Matrix Market formatmatrixzProblem reading file header: r:r9r%%zHeader line not of length 2: asciizHeader line not of length 3: )_openreadlinesplit startswithr=lower FORMAT_ARRAYFORMAT_COORDINATElstriprPlenrmapintclose)r/rstreamclose_itlineZmmidrTr4r#r%Z split_liner0r2r3rrrrUsT            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.bz2rjT) osfspath TypeErrorpathisfileendswithgzipopenbz2BZ2File)Zfilespecmodersrgrurrrr[s6            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)r{rzr|r}r!nrrr~s rFignore)Zover)shaperSYMMETRY_GENERALdtypecharr tocooZnonzerosumZtodoknpZerrstaterr5r6r7) r!mZissymmZisskewZishermrowcolr~r|r}Z 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)r/kwargsrrr__init__.szMMFile.__init__cCsD||\}}z||||W|r|SS|r!|ww)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. )r[ _parse_header _parse_bodyrf)r/rrgrhrrrr2s    z MMFile.readrNc 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)r[_writerfflush) r/r r!r"r#r$r%rgrhrrrrMs    z MMFile.writec Ksl|jj}dd|D}t|t|}|r!tdt|||D]}t||||dddq#dS)zr Initialize each attributes with the corresponding keyword arg value or a default of None cSsg|]}|ddqS)Nr)rQattrrrr wsz&MMFile._init_attrs..zVfound {} invalid keyword arguments, please only use {}rN) __class__ __slots__setkeysr=r4tuplesetattrr)r/rattrsZ public_attrsZ invalid_keysrrrrrpszMMFile._init_attrscCs2|j|\}}}}}}|j||||||ddS)N)r0r2r3r4r#r%)rrr)r/rgr0r2r3r4r#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)rrrrUz$Parse error, did not read all lines.ZintcZint8rHrIrBfloatz5'entries' in header is smaller than number of entriesrXz4'entries' in header is larger than number of entries)rr) r0r2r3r4r#r%DTYPES_BY_FIELDrr8rrrr6r7 FIELD_PATTERNr`rr\rPrerBrdrr]rr=rar r r conjugateNotImplementedError)r/rgr0r2r3r4r#Zsymmrr8 is_integerZis_unsigned_integerZ is_complexZis_skewZis_hermZ is_patternr!rirzr{r|IJVZ entry_numberlmaskZod_IZod_JZod_Vrrrrs                  "JG          zMMFile._parse_bodycCst|tst|tst|tst|drf|j}t|}t|jdkr&t d|j\}} |dure||j krDt |j ds>t d|d}n5||jkrU|j jdvrT|d}n$||jkre|j jdvre|d }nt|srt 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| } || dq|||}||jkr d|| f} || d||j |j|jfvr||j krNt!| D]}t!|D]}||||f} || dq6q0dS||j"kryt!| D]}t!|d|D]}||||f} || dqaqXdSt!| D]}t!||D]}||||f} || dqq}dS||jkr||j krt!| D]#}t!|D]}|||f}|t#|t$|f} || dqqdSt!| D]$}t!||D]}|||f}|t#|t$|f} || dqݐqdS||j%krt dtd ||&}||j kr1|j'|j(k}t)|j*||j'||j(|ff|jd!}d"|| |j+f} || d|||d}||j%krpt,|j'd|j(dD]\}}d||f} || dqZdS||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 |)$NZ __array__rXzExpected 2 dimensional arrayrHzBmmwrite does not support integer dtypes larger than native 'intp'.fdrJrxrKzunknown matrix type: %sr9ZfFrzr@frcrBurAzunexpected dtype kind z%%MatrixMarket matrix   rz%%%s z%i %i rz*pattern type inconsisted with dense formatzUnknown field type %s)rz %i %i %i z%i %i )-rlistrrhasattrr`rrcrr=rr r OverflowErrorZastyperrrr typekindrorrr>rErGrencoder]rrrrr6rrrrrrr dataZnnzzip)r/rgr!r"r#r$r%repr0r2typecoderrritemplater{rzr|ZcooZlower_triangle_maskrkrrJrrrrs                                          $ $ z MMFile._write)rjrNNN)-__name__ __module__ __qualname__rpropertyr0r2r3r4r#r%r8rar`r< classmethodr>rrrrrrDrErr5r6r7rFrGr staticmethodrMrOrr[rrrrrrrrrrrrrrs             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)rsappendGzipFile ImportErrorrurvrr)rgZbad_clsrsrurrr_is_fromfile_compatibles   rr)__doc__rmnumpyrrrrrrrrr r Z scipy.sparser r __all__rr rrrrrrrrs" ,5 2p ;