a aat@sdZddlZddlZddlmZmZmZmZmZm Z m Z m Z m Z ddl mZmZgdZddZdd Zd d Zdd dZGdddZddZedkrddlZejddD]BZededddejeZeeedeeqdS)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)sr]/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/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'. )rinfosourcerrrr sr 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<srcCst||||||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 @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&selfrrrrowsrsz MMFile.rowscCs|jSr,)r'r-rrrcolsvsz MMFile.colscCs|jSr,)r(r-rrrentrieszszMMFile.entriescCs|jSr,)r)r-rrrformat~sz MMFile.formatcCs|jSr,)r*r-rrrr#sz MMFile.fieldcCs|jSr,)r+r-rrrr%szMMFile.symmetrycCs|j|j|j|jfvSr,)r+SYMMETRY_SYMMETRICSYMMETRY_SKEW_SYMMETRICSYMMETRY_HERMITIANr-rrr has_symmetryszMMFile.has_symmetry coordinatearraycCs ||jvrtd||jfdS)Nz)unknown format type %s, must be one of %s) FORMAT_VALUES ValueError)r.r2rrr_validate_formats zMMFile._validate_formatintegerunsigned-integerrcomplexpatterncCs ||jvrtd||jfdS)Nz(unknown field type %s, must be one of %s) FIELD_VALUESr:)r.r#rrr_validate_fields zMMFile._validate_fieldZgeneralZ symmetriczskew-symmetricZ hermitiancCs ||jvrtd||jfdS)Nz+unknown symmetry type %s, must be one of %s)SYMMETRY_VALUESr:)r.r%rrr_validate_symmetrys zMMFile._validate_symmetryintpuint64dDcCsdSr,rrrrrreadersz MMFile.readercCsdSr,rrrrrwritersz MMFile.writercCsh||\}}zB|}dd|D\}}}}} |dsHtd|dks`td||dkrt|j}n|dkr|j}|r|d d vr|}q|s|}q|} ||jkrt | d kstd | d t t | \} } | | } n0t | dkstd| d t t | \} } } | | | ||| fW|rP| Sn|rb| 0dS)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: r8r7r%%zHeader line not of length 2: asciizHeader line not of length 3: N)_openreadlinesplit startswithr:lower FORMAT_ARRAYFORMAT_COORDINATErJlenrmapintclose)r.rstreamclose_itlineZmmidrOr2r#r%Z split_liner/r0r1rrrrsP           z MMFile.inforbcCszt|}Wnty(|dfYS0|ddkrtj|stj|dr\|d}n2tj|drv|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.bz2rdT) osfspath TypeErrorpathisfileendswithgzipopenbz2BZ2File)ZfilespecmodermrarorrrrVs.         z MMFile._openc s"j\}|krtjSd}d}jjdv}tr|\}}||k||kkrftjS fdd}nfdd}|D]f\}} } |r| r|dkrd}n6|r|| krd}|r|| krd}|r|t | krd}|s|s|sqq|rtj S|rtj S|rtj StjS)NTFDc3sPD]B\\}}}||kr6||f}||dfVq||kr||dfVqdS)NFT)items)ijaijaji)r!rr symm_iteratorhs  z+MMFile._get_symmetry..symm_iteratorc3sLtD]>}t|D].}||||}}||||kfVqqdSr,)range)rurtrvrwr!nrrrxss rF)shaperSYMMETRY_GENERALdtypecharr tocooZnonzerosumZtodokrr3r4r5) r!mZissymmZisskewZishermrowcolrxrvrwZ is_diagonalrrzr _get_symmetrySs@      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|jfi|dSr,) _init_attrs)r.kwargsrrr__init__szMMFile.__init__cCsF||\}}z$||||W|r0|Sn|r@|0dS)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)rV _parse_header _parse_bodyr`)r.rrarbrrrrs  z MMFile.readrNc Cs\||d\}}z.|||||||W|r6|qX|n|rN|n|0dS)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)rV_writer`flush) r.r r!r"r#r$r%rarbrrrrs   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)rKattrrrrrMrNz&MMFile._init_attrs..zVfound %s invalid keyword arguments, please only use %srN) __class__ __slots__setkeysr:tuplesetattrr)r.rattrsZ public_attrsZ invalid_keysrrrrrszMMFile._init_attrscCs2|j|\}}}}}}|j||||||ddS)N)r/r0r1r2r#r%)rrr)r.rar/r0r1r2r#r%rrrrs   zMMFile._parse_headerc Cs|j|j|j|j|j|jf\}}}}}}zddlm}WntyNd}Yn0|j |d} |j } ||j k} ||j k} ||jk} ||jk}||jk}||jk}||jkrTt||f| d}d}d\}}|rd|||f<||dkr|d7}|r|}|r|ddvs|sq| r&t|}n2| r6t|}n"| rPttt|}nt|}||||f<| r||kr|r| |||f<n$|rt||||f<n ||||f<||dkr|d}q|d}| sd}q|}|rd|||f<||dkr|d7}q|r0|d|fvr&||dksPtdn |d|fvrH||kstdn~||jkr|durt||f| d}d}d}|r|}|r|ddvs|sq|}tt|dd\}}|d|d}}| rt|d}n>| r t|d}n*| r(ttt|dd}n t|d}||||f<| r||kr|rf| |||f<n$|r~t||||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 | r:t|dd}n t|dd}d}|D]}|rN|ddvsN|stqN|d|krtd|}tt|dd\||<||<|s| rt|d||<nJ| rt|d||<n2| rttt|dd||<nt|d||<|d7}qN||kr6td|d8}|d8}| r||k}||}||}||}t||f}t||f}|r|d9}n|r| }t||f}||||ff||f| d}nt!||S)Nr)r )r~r)rrrPz$Parse error, did not read all lines.rSzDid not read all entriesZintcZint8rDrEr>floatz5'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.rar/r0r1r2r#Zsymmr r~r6 is_integerZis_unsigned_integerZ is_complexZis_skewZis_hermZ is_patternr!rcrtrurvklIJVZ entry_numbermaskZod_IZod_JZod_Vrrrrs                                      zMMFile._parse_bodycCst|ts(t|ts(t|ts(t|dr|j}t|}t|jdkrLt d|j\}} |dur||j krt |j ds|t d|d}q||jkr|j jdvr|d}q||jkr|j jdvr|d }n&t|st d t|d }|j\}} |j j} |dur| d vrd }nd}|dur|j j} | dkrPt |j dsJt dd}n<| dkr`d}n,| dkrpd}n| dkrd}n td| |dur||}|j||j||j|d|||} || d|dD]} d| } || dq|||}||jkr d|| f} || d||j |j|j fvr<||j!krt"| D]6}t"|D]&}||||f} || dqvqjn||j#krt"| D]<}t"|d|D]&}||||f} || dqʐqnBt"| D]8}t"||D]&}||||f} || dqqn||jkr||j!krt"| D]F}t"|D]6}|||f}|t$|t%|f} || dqhq\nRt"| D]H}t"||D]6}|||f}|t$|t%|f} || dqqn"||j&krt dn td|n|'}||j!krh|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|j fvrHt-|j(d|j)d|j+D]0\}}}d"||f||} || dqnn||jkrt-|j(d|j)d|j+D]8\}}}d"||f||j$|j%f} || dqnn td|dS)#NZ __array__rSzExpected 2 dimensional arrayrDzBmmwrite does not support integer dtypes larger than native 'intp'.fdrFrrrGzunknown matrix type: %sr7ZfFrtr<frcr>ur=zunexpected dtype kind z"%%MatrixMarket matrix {0} {1} {2} r z%%%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 typekindrirrr;rArCr2rencoderXrrr}ryr4rrrrrrr dataZnnzzip)r.rar!r"r#r$r%repr/r0typecoderrrctemplaterurtrvZcooZlower_triangle_maskrerrFrrrrs                                               $ $z MMFile._write)rd)rNNN)rNNN)-__name__ __module__ __qualname__rpropertyr/r0r1r2r#r%r6r\r[r9 classmethodr;rrrrrr@rAr}r3r4r5rBrCr staticmethodrHrIrrVrrrrrrrrrrrrrrjs             K > >  #/rcCspg}zddl}||jWnty.Yn0zddl}||jWntyZYn0t|}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)rmappendGzipFilerrorprr)raZbad_clsrmrorrr_is_fromfile_compatible>s  r__main__rZReadingz... )endztook %s seconds)rNNN)__doc__rgsysZnumpyrrrrrrrr r rr r __all__rr rrrrrtimeargvfilenameprintstdoutrtrrrrs0 , Y