B 0dat@sdZddlZddlZddlmZmZmZmZmZm Z m Z m Z m Z ddl mZmZddddgZd d Zd dZd dZdddZGdddZddZedkrddlZxTejddD]BZededddejeZeeedeeqWdS)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)sr?/var/www/html/venv/lib/python3.7/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'. )rinfo)sourcerrrr scCs 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. )rread)rrrrr<scCst||||||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 precisionsymmetryrrrrQsc @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|jS)N)r%)selfrrrrowsrsz MMFile.rowscCs|jS)N)r&)r+rrrcolsvsz MMFile.colscCs|jS)N)r')r+rrrentrieszszMMFile.entriescCs|jS)N)r()r+rrrformat~sz MMFile.formatcCs|jS)N)r))r+rrrr"sz MMFile.fieldcCs|jS)N)r*)r+rrrr$szMMFile.symmetrycCs|j|j|j|jfkS)N)r*SYMMETRY_SYMMETRICSYMMETRY_SKEW_SYMMETRICSYMMETRY_HERMITIAN)r+rrr has_symmetryszMMFile.has_symmetry coordinatearraycCs ||jkrtd||jfdS)Nz)unknown format type %s, must be one of %s) FORMAT_VALUES ValueError)r+r/rrr_validate_formats zMMFile._validate_formatintegerzunsigned-integerrcomplexpatterncCs ||jkrtd||jfdS)Nz(unknown field type %s, must be one of %s) FIELD_VALUESr7)r+r"rrr_validate_fields zMMFile._validate_fieldZgeneralZ symmetriczskew-symmetricZ hermitiancCs ||jkrtd||jfdS)Nz+unknown symmetry type %s, must be one of %s)SYMMETRY_VALUESr7)r+r$rrr_validate_symmetrys zMMFile._validate_symmetryintpuint64dDcCsdS)Nrrrrrreadersz MMFile.readercCsdS)Nrrrrrwritersz MMFile.writercCs`||\}}z:|}dd|D\}}}}} |dsHtd|dks`td||dkrt|j}n|dkr|j}x|r|d d kr|}qWx|s|}qW|} ||jkrt | d kstd | d t t | \} } | | } n0t | dks$td| d t t | \} } } | | | ||| fS|rZ| 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 szMMFile.info..z%%MatrixMarketz%source is not in Matrix Market formatmatrixzProblem reading file header: r5r4r)%%zHeader line not of length 2: asciizHeader line not of length 3: N)_openreadlinesplit startswithr7lower FORMAT_ARRAYFORMAT_COORDINATErFlenrmapintclose)r+rstreamclose_itlineZmmidrJr/r"r$Z split_liner,r-r.rrrrs>            z MMFile.inforbcCsyt|}Wntk r&|dfSX|ddkrtj|stj|drZ|d}n2tj|drt|d}ntj|dr|d}|drddl}|||}q|d rddl}| |d }qt||}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.bz2r^T) osfspath TypeErrorpathisfileendswithgzipopenbz2BZ2File)Zfilespecmodergr[rirrrrPs.         z MMFile._openc s$j\}|krtjSd}d}jjdk}tr|\}}||k||kkrftjS fdd}nfdd}xn|D]d\}} } |r| r|dkrd}n6|r|| krd}|r|| krd}|r|t | krd}|s|s|sPqW|rtj S|rtj S|rtj StjS)NTFDc3sTxND]B\\}}}||kr8||f}||dfVq ||kr ||dfVq WdS)NFT)items)ijaijaji)r rr symm_iteratorhs  z+MMFile._get_symmetry..symm_iteratorc3sTxNtD]B}x.zVfound %s invalid keyword arguments, please only use %srN) __class__ __slots__setkeysr7tuplesetattrr)r+rattrsZ public_attrsZ invalid_keysrrrrrs  zMMFile._init_attrscCs2|j|\}}}}}}|j||||||ddS)N)r,r-r.r/r"r$)rrr)r+r[r,r-r.r/r"r$rrrrs zMMFile._parse_headerc Cs|j|j|j|j|j|jf\}}}}}}yddlm}Wntk rPd}YnX|j |d} |j } ||j k} ||j k} ||jk} ||jk}||jk}||jk}||jkr\t||f| d}d}d\}}|rd|||f<||dkr|d7}x|r |}|r|ddks|sq| r,t|}n2| r| rt|d}n*| r4ttt|dd}n t|d}||||f<| r||kr|rr| |||f<n$|rt||||f<n ||||f<|d}qW||kstd n*||jkr|dkr|||f| dSt|d d}t|d d}|r t|d d}nH| r t|d d}n4| r4t|d d}n | rHt|dd}n t|dd}d}x|D]}|r^|ddks^|sq^|d|krtd|}tt|dd\||<||<|s(| rt|d||<nJ| rt|d||<n2| rttt|dd||<nt|d||<|d7}q^W||krHtd|d8}|d8}| r||k}||}||}||}t||f}t||f}|r|d9}n|r| }t||f}||||ff||f| d}nt!||S)Nr)r )rwr)rr)rKrLz$Parse error, did not read all lines.rMzDid not read all entriesZintcZint8r@rAr:floatz5'entries' in header is smaller than number of entriesz4'entries' in header is larger than number of entries)rurw)"r,r-r.r/r"r$ scipy.sparser ImportErrorDTYPES_BY_FIELDrr3rrrr1r2 FIELD_PATTERNrUrrQrFrYr:rXrrRrr7rVr r conjugateNotImplementedError) r+r[r,r-r.r/r"Zsymmr rwr3 is_integerZis_unsigned_integerZ is_complexZis_skewZis_hermZ is_patternr r]rnrorpklIJVZ 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|||} || dx,|dD]} d| } || dqW|||}||jkrLd|| f} || d||j |j|j fkrX||j!krxt"| D]:}x2t"|D]&}||||f} || dq~WqpWn||j#kr xt"| D]@}x8t"|d|D]&}||||f} || dqWqWnJxHt"| D]<}x4t"||D]&}||||f} || dq&WqWn||jkr&||j!krxt"| D]J}xBt"|D]6}|||f}|t$|t%|f} || dqWqzWnZx|t"| D]L}xDt"||D]6}|||f}|t$|t%|f} || dqWqWn"||j&krr?r staticmethodrDrErrPr~rrrrrrrrrrrrrjs`             L > ?  "/cCstg}yddl}||jWntk r0YnXyddl}||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)rgappendGzipFilerrirjrr)r[Zbad_clsrgrirrr_is_fromfile_compatible>sr__main__rZReadingz... )endztook %s seconds)rNNN)__doc__rasysnumpyrrrrrrrr r rr r __all__rr rrrrrtimeargvfilenameprintstdoutrtrrrrs0,  Y