3 hq@sdZddlZddlZddlmZmZmZmZmZm Z m Z m Z m Z ddl mZmZddlmZmZdddd gZd dZd dZdd dZGdd d ZddZedkrddlZxTejddD]BZededddejjejZ eeedeje qWdS)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)asbytesasstr) coo_matrix isspmatrixmminfommreadmmwriteMMFilecCs tj|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)sourcer./tmp/pip-build-riy7u7_k/scipy/scipy/io/mmio.pyrscCs tj|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)rrrrr8scCstj||||||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 precisionsymmetryrrrrMsc @steZdZdGZeddZed d Zed d Zed dZeddZ eddZ eddZ dZ dZ e e fZeddZdZdZdZdZdZeeeeefZeddZd Zd!Zd"Zd#ZeeeefZed$d%Zed&ed'ed(ed)ed(iZed*d+Z ed,d-Z!ed.d/Z"edHd1d2Z#ed3d4Z$ed5d6Z%d7d8Z&d9d:Z'dId=d>Z(d?d@Z)dAdBZ*dCdDZ+dJdEdFZ,dszMMFile.info..z%%MatrixMarketz%source is not in Matrix Market formatmatrixzProblem reading file header: r0r/r%%zHeader line not of length 2: zHeader line not of length 3: N)rFrG) _openreadlinesplit startswithr2lower FORMAT_ARRAYFORMAT_COORDINATErAlenmapintclose) r&rstreamclose_itlineZmmidrEr*rrr'r(r)rrrrs:              z MMFile.inforbc Csytj|}Wntk r&|dfSX|ddkrtjj|stjj|drZ|d}n2tjj|drt|d}ntjj|dr|d}|jdrddl}|j||}n*|jd rddl}|j |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.bz2rXT) osfspath TypeErrorpathisfileendswithgzipopenbz2BZ2File)ZfilespecmoderbrUrdrrrrJs.         z MMFile._openc sj\}|krtjSd}d}jjdk}tr|jj\}}||kj||kjkrftjSj fdd}nfdd}xV|D]L\}} |r|| krd}|r|| krd}|r|t | krd}|p|p|sPqW|rtj S|rtj S|rtj StjS)NTFDc3s<x6jD]*\\}}}||kr ||f}||fVq WdS)N)items)ijaijaji)rrr symm_iteratorbs z+MMFile._get_symmetry..symm_iteratorc3sRxLtD]@}x:t|dD](}||||}}||fVqWq WdS)N)range)rjrirkrl)rnrrrmksF)shaperSYMMETRY_GENERALdtypecharrtocooZnonzerosumZtodokrr+r,r-) rmZissymmZisskewZishermrowcolrmrkrlr)rrpr _get_symmetryMs<      zMMFile._get_symmetryc Cs0tjd|tjdtjdtjd||fij|dS)Nz%%.%ie z%i z%u z%%.%ie %%.%ie )r FIELD_REAL FIELD_INTEGERFIELD_UNSIGNED FIELD_COMPLEXget)rrrrr_field_templates   zMMFile._field_templatecKs|jf|dS)N) _init_attrs)r&kwargsrrr__init__szMMFile.__init__c Cs6|j|\}}z|j||j|S|r0|jXdS)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)rJ _parse_header _parse_bodyrT)r&rrUrVrrrrs   z MMFile.readrNc CsF|j|d\}}z|j||||||Wd|r8|jn|jXdS)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)rJ_writerTflush) r&rrrrrrrUrVrrrrs  z MMFile.writecKsp|jj}dd|D}t|jt|}|rBtdt||fx(|D] }t|||j|dddqHWdS)zr Initialize each attributes with the corresponding keyword arg value or a default of None cSsg|]}|ddqS)rnNr)rBattrrrrrDsz&MMFile._init_attrs..zVfound %s invalid keyword arguments, please only use %srnN) __class__ __slots__setkeysr2tuplesetattrr)r&rattrsZ public_attrsZ invalid_keysrrrrrs  zMMFile._init_attrscCs2|jj|\}}}}}}|j||||||ddS)N)r'r(r)r*rr)rrr)r&rUr'r(r)r*rrrrrrs zMMFile._parse_headerc Cs|j|j|j|j|j|jf\}}}}}}yddlm}Wntk rPd}YnX|j j |d} |j } ||j k} ||j k} ||jk} ||jk}||jk}||jk}||jkr`t||f| d}d}d\}}|rd|||f<||dkr|d7}x|r|j}| s|ddks|j r q| r0t|}n2| r@t|}n"| rZttt|j}nt|}||||f<| r||kr|r| |||f<n$|rt||||f<n ||||f<||dkr|d}q|d}| sd}q|}|rd|||f<||dkr|d7}qW|r<|d|gko.||dks\tdn |d|gkoP||kstdn||jkr|dkrt||f| d}d}d}x|r|j}| s|ddks|j rƐq|j}tt|dd\}}|d|d}}| r t|d}n>| rt|d}n*| rt"||D]0}|||f}|jt| t$|t%|fqWqWn"||j&kr t dn td|n|j'}||j!krd|j(|j)k}t*|j+||j(||j)|ff|jd}|jtd || |j,f|j||d} ||j&krx:t-|j(d|j)dD] \}}|jtd||fqWn||j |j|j fkr:xt-|j(d|j)d|j+D]*\}}}|jtd!||f| |q Wnl||jkrx^t-|j(d|j)d|j+D]2\}}}|jtd!||f| |j$|j%fqbWn td|dS)"NZ __array__rHzExpected 2 dimensional arrayr;zBmmwrite does not support integer dtypes larger than native 'intp'.fdr=rgr>zunknown matrix type: %sr/ZfFrir4frcr5uzunsigned-integerzunexpected dtype kind z"%%MatrixMarket matrix {0} {1} {2}  z%%%s z%i %i rnz*pattern type inconsisted with dense formatzUnknown field type %s)rqz %i %i %i z%i %i ). isinstancelistrrhasattrrOrrQrqr2r|r rs OverflowErrorZastyper{rtr~rtypekindr^rzrr3r8r:rr r*rLrr}rrror,rrrrurxryr dataZnnzzip)r&rUrrrrrrepr'r(typecoderrWtemplaterjrirkZcooZlower_triangle_maskrYrr=rrrrs                             ( ((             & &z MMFile._write)r r!r"r#r$r%)rX)rNNN)rNNN)-__name__ __module__ __qualname__rpropertyr'r(r)r*rrr.rPrOr1 classmethodr3r|r}r{r~rr7r8rrr+r,r-r9r:r staticmethodr?r@rrJrzrrrrrrrrrrrrrfsj             J > 9  "/cCstg}yddl}|j|jWntk r0YnXyddl}|j|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)rbappendGzipFilerrdrerr)rUZbad_clsrbrdrrr_is_fromfile_compatible+sr__main__rnZReadingz... )endztook %s seconds)rNNN)!__doc__r\sysZnumpyrrrrrrrr r Z numpy.compatr r rr r__all__rrrrrrtimeargvfilenameprintstdoutrtrrrrs0,  J