o Ebat@sdZddlZddlZddlmZmZmZmZmZm Z m Z m Z m Z ddl mZmZgdZddZdd Zd d Zdd dZGdddZddZedkryddlZejddD]#ZededddejeZeeedeeqWdSdS)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 isspmatrix)mminfommreadmmwriteMMFilecCst|tr |dSt|S)Nlatin1) isinstancebytesdecodestr)sr0/usr/lib/python3/dist-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'. )rinfosourcerrrr s r cCs t|S)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. Returns ------- a : ndarray or coo_matrix Dense or sparse matrix depending on the matrix format in the Matrix Market file. )rreadrrrrr<s 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. N)rwrite)targetacommentfield precisionsymmetryrrrrQsrc @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&selfrrrrowsrz MMFile.rowscCr,r-)r'r.rrrcolsvr1z MMFile.colscCr,r-)r(r.rrrentrieszr1zMMFile.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_fieldr?zMMFile._validate_fieldZgeneralZ symmetriczskew-symmetricZ hermitiancCr;)Nz+unknown symmetry type %s, must be one of %s)SYMMETRY_VALUESr=)r/r%rrr_validate_symmetryr?zMMFile._validate_symmetryintpuint64dDcCdSr-rrrrrreaderz MMFile.readercCrLr-rrrrrwriterrNz MMFile.writercCsr||\}}z|}dd|D\}}}}} |ds#td|dks/td||dkr9|j}n |dkrB|j}|rV|d d vrV|}|rV|d d vsJ|sb|}|rZ|} ||jkrt | d ksztd | 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'. cSsg|]}t|qSr)rstrip).0partrrr zMMFile.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_COORDINATErPlenrmapintclose)r/rstreamclose_itlineZmmidrUr4r#r%Z split_liner0r2r3rrrrsP           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 sj\}|kr tjSd}d}jjdv}tr>\}}||k||kkr3tjS fdd}nfdd}|D]3\}} } |rX| rX|dkrXd}n|r`|| kr`d}|ri|| krid}|rs|t | krsd}|s{|s{|s{nqH|rtj S|rtj S|rtj StjS)NTFDc3sRD]!\\}}}||kr||f}||dfVq||kr&||dfVqdS)NFT)items)ijaijaji)r!rr symm_iteratorhs  z+MMFile._get_symmetry..symm_iteratorc3sNtD]}t|D]}||||}}||||kfVq qdSr-)range)r{rzr|r}r!nrrr~ss rF)shaperSYMMETRY_GENERALdtypecharr tocooZnonzerosumZtodokrr5r6r7) r!mZissymmZisskewZishermrowcolr~r|r}Z is_diagonalrrr _get_symmetrySsB      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_templates 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/rrgrhrrrrs    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%rgrhrrrrs    z MMFile.writec Ksl|jj}dd|D}t|t|}|r!tdt||f|D]}t||||dddq#dS)zr Initialize each attributes with the corresponding keyword arg value or a default of None cSsg|]}|ddqS)Nr)rQattrrrrrSrTz&MMFile._init_attrs..zVfound %s invalid keyword arguments, please only use %srN) __class__ __slots__setkeysr=tuplesetattrr)r/rZattrsZ public_attrsZ invalid_keysrrrrrszMMFile._init_attrscCs2|j|\}}}}}}|j||||||ddS)N)r0r2r3r4r#r%)rrr)r/rgr0r2r3r4r#r%rrrrs   zMMFile._parse_headerc Cs|j|j|j|j|j|jf\}}}}}}zddlm}Wn ty'd}Ynw|j |d} |j } ||j k} ||j k} ||jk} ||jk}||jk}||jk}||jkr!t||f| d}d}d\}}|rvd|||f<||dkrv|d7}|r|}|r|ddvs|sqv| rt|}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}|sx|r|d|fvr ||dks td|S|d|fvr||kstd|S||jkr|durt||f| d}d}d}|r|}|rN|ddvsN|sPq8|}tt|dd\}}|d|d}}| rrt|d}n| r|t|d}n| rttt|dd}nt|d}||||f<| r||kr|r| |||f<n|rt||||f<n||||f<|d}|s;||kstd |S||jkr|dkr|||f| dSt|d d}t|d d}|rt|d d}n$| rt|d d}n| r t|d d}n| rt|dd}nt|dd}d}|D]j}|r0|ddvs0|s2q|d|kr=td|}tt|dd\||<||<|s| r_t|d||<n%| rkt|d||<n| r|ttt|dd||<nt|d||<|d7}q||krtd|d8}|d8}| r||k}||}||}||}t||f}t||f}|r|d9}n|r| }t||f}||||ff||f| d}|St!|)Nr)r )rr)rrrVz$Parse error, did not read all lines.rYzDid not read all entriesZintcZint8rHrIrBfloatz5'entries' in header is smaller than number of entriesz4'entries' in header is larger than number of entries)rr)"r0r2r3r4r#r% scipy.sparser ImportErrorDTYPES_BY_FIELDrr8rrrr6r7 FIELD_PATTERNrarr]rPrerBrdrr^rr=rbr r conjugateNotImplementedError) r/rgr0r2r3r4r#Zsymmr rr8 is_integerZis_unsigned_integerZ is_complexZis_skewZis_hermZ is_patternr!rirzr{r|klIJVZ entry_numbermaskZod_IZod_JZod_Vrrrrs                   "kh    G          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|dD]} d| } || dq|||}||jkrd|| f} || d||j |j|j fvr||j!krIt"| D]}t"|D]}||||f} || dq1q+dS||j#krtt"| D]}t"|d|D]}||||f} || dq\qSdSt"| D]}t"||D]}||||f} || dqqxdS||jkr||j!krt"| D]#}t"|D]}|||f}|t$|t%|f} || dqqdSt"| D]$}t"||D]}|||f}|t$|t%|f} || dqؐqdS||j&krt dtd||'}||j!kr,|j(|j)k}t*|j+||j(||j)|ff|jd }d!|| |j,f} || d|||d}||j&krkt-|j(d|j)dD]\}}d||f} || dqUdS||j |j|j fvrt-|j(d|j)d|j+D]\}}}d"||f||} || dqdS||jkrt-|j(d|j)d|j+D]\}}}d"||f||j$|j%f} || dqdStd|)#NZ __array__rYzExpected 2 dimensional arrayrHzBmmwrite does not support integer dtypes larger than native 'intp'.fdrJrxrKzunknown matrix type: %sr9ZfFrzr@frcrBurAzunexpected dtype kind z"%%MatrixMarket matrix {0} {1} {2} r z%%%s z%i %i rz*pattern type inconsisted with dense formatzUnknown field type %s)rz %i %i %i z%i %i ).rlistrrhasattrrarrcrr=rr r OverflowErrorZastyperrrr typekindrorrr>rErGr4rencoder^rrrrr6rrrrrrr dataZnnzzip)r/rgr!r"r#r$r%Zrepr0r2typecoderrritemplater{rzr|ZcooZlower_triangle_maskrkrrJrrrrs                                        $ $ z MMFile._write)rjrNNN)-__name__ __module__ __qualname__rpropertyr0r2r3r4r#r%r8rbrar< classmethodr>rrrrrrDrErr5r6r7rFrGr staticmethodrMrOrr\rrrrrrrrrrrrrrjs             K > >   #/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)rsappendZGzipFilerrurvrr)rgZbad_clsrsrurrr_is_fromfile_compatible>s   r__main__rZReadingz... )endztook %s secondsr)__doc__rmsysZnumpyrrrrrrrr r rr r __all__rr rrrrrtimeargvfilenameprintstdoutrtrrrrs6 , Y