B &]\p@s*dZddlmZmZmZddlZddlZddlmZm Z m Z m Z m Z m Z mZmZmZddlmZmZddlmZddlmZmZdd d d gZd dZd d Zddd ZGdd d eZddZe dkr&ddl!Z!xTej"ddD]BZ#e$de#dddej%&e!!Z'ee#e$de!!e'qWdS)z Matrix Market I/O in Python. See http://math.nist.gov/MatrixMarket/formats.html for information about the Matrix Market format. )divisionprint_functionabsolute_importN) asarrayrealimagconjzerosndarray concatenateonescan_cast)asbytesasstr) string_types) coo_matrix isspmatrixmminfommreadmmwriteMMFilecCs 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)sourcer,lib/python3.7/site-packages/scipy/io/mmio.pyrscCs 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 2D 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 precisionsymmetryrrrrPsc @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$)selfrrrrowsqsz MMFile.rowscCs|jS)N)r%)r*rrrcolsusz MMFile.colscCs|jS)N)r&)r*rrrentriesyszMMFile.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_VALUESr6)r*r!rrr_validate_fields zMMFile._validate_fieldZgeneralZ symmetriczskew-symmetricZ hermitiancCs ||jkrtd||jfdS)Nz+unknown symmetry type %s, must be one of %s)SYMMETRY_VALUESr6)r*r#rrr_validate_symmetrys zMMFile._validate_symmetryintpuint64dDcCsdS)Nrrrrrreadersz MMFile.readercCsdS)Nrrrrrwritersz MMFile.writerc Cs4||\}}z|}dd|D\}}}}} |dsHtd|dks`td||dkrt|j}n|dkr|j}x|d r|}qW|}||jkrt|d kstd |t t |\} } | | } n(t|d kstd |t t |\} } } | | | ||| fS|r.| 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: r4r3%zHeader line not of length 2: zHeader line not of length 3: N) _openreadlinesplit startswithr6lower FORMAT_ARRAYFORMAT_COORDINATElenmapintclose) r*rstreamclose_itlineZmmidrIr.r!r#r+r,r-rrrrs6              z MMFile.inforbcCsd}t|trd}|ddkrtj|svtj|drD|d}n2tj|dr^|d}ntj|drv|d}|drdd l}|||}q|d rdd l}| |d }qt||}q|d d dkr|d}t||}n|}||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. FTrrz.mtxz.mtx.gzz.mtx.bz2z.gzNz.bz2r[) isinstancerospathisfileendswithgzipopenbz2BZ2File)ZfilespecmoderYrcrXrerrrrM s.         z MMFile._openc sj\}|krtjSd}d}jjdk}tr|\}}||k||kkrftjS fdd}nfdd}xV|D]L\}} |r|| krd}|r|| krd}|r|t | krd}|s|s|sPqW|rtj S|rtj S|rtj StjS)NTFDc3s<x6D]*\\}}}||kr ||f}||fVq WdS)N)items)ijaijaji)rrr symm_iterator[s z+MMFile._get_symmetry..symm_iteratorc3sRxLtD]@}x:t|dD](}||||}}||fVqWq WdS)N)range)rkrjrlrm)rnrrrndsF)shaperSYMMETRY_GENERALdtypecharrtocooZnonzerosumZtodokrr/r0r1) rmZissymmZisskewZishermrowcolrnrlrmr)rrqr _get_symmetryFs<      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|jf|dS)N) _init_attrs)r*kwargsrrr__init__szMMFile.__init__cCs6||\}}z||||S|r0|XdS)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)rM _parse_header _parse_bodyrW)r*rrXrYrrrrs   z MMFile.readrNc CsF||d\}}z|||||||Wd|r8|n|XdS)a6 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 2D 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)rM_writerWflush) r*rrr r!r"r#rXrYrrrrs  z MMFile.writec Ksp|jj}dd|D}t|t|}|rBtdt||fx(|D] }t||||dddqHWdS)zr Initialize each attributes with the corresponding keyword arg value or a default of None cSsg|]}|ddqS)roNr)rFattrrrrrHsz&MMFile._init_attrs..zVfound %s invalid keyword arguments, please only use %sroN) __class__ __slots__setkeysr6tuplesetattrr)r*rZattrsZ public_attrsZ invalid_keysrrrrrs  zMMFile._init_attrscCs2|j|\}}}}}}|j||||||ddS)N)r+r,r-r.r!r#)rrr)r*rXr+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}||jkrRt||f| d}d}d\}}|rd|||f<||dkr|d7}x|r|}|r|drq| r"t|}n2| r2t|}n"| rLttt|}nt|}||||f<| r||kr|r| |||f<n$|rt||||f<n ||||f<||dkr|d}q|d}| sd}q|}|rd|||f<||dkr|d7}qW|r.|d|gkr$||dksNtdn |d|gkrF||kstdnp||jkr|dkrt||f| d}d}d}x |r|}|r|drq|}tt|dd\}}|d|d}}| rt|d}n>| rt|d}n*| rttt|dd}n t|d}||||f<| r||kr|r\| |||f<n$|rtt||||f<n ||||f<|d}qW||kstd n||jkr|dkr|||f| dSt|d d}t|d d}|rt|d d}nH| r t|d d}n4| rt|d d}n | r2t|dd}n t|dd}d}x|D]}|rH|drbqH|d|krxtd|}tt|dd\||<||<|s| rt|d||<nJ| rt|d||<n2| rttt|dd||<nt|d||<|d7}qHW||kr&td|d8}|d8}| r||k}||}||}||}t||f}t||f}|r|d9}n|r| }t||f}||||ff||f| d}nt!||S)Nr)r)rtro)rrrJz$Parse error, did not read all lines.rKzDid not read all entriesZintcZint8r?r@r9floatz5'entries' in header is smaller than number of entriesz4'entries' in header is larger than number of entries)rrrt)"r+r,r-r.r!r# scipy.sparser ImportErrorDTYPES_BY_FIELDrr2r}r~rr0r1 FIELD_PATTERNrRr rNrPrVr9rUrrOrr6rSr r conjugateNotImplementedError) r*rXr+r,r-r.r!Zsymmrrtr2 is_integerZis_unsigned_integerZ is_complexZis_skewZis_hermZ is_patternrrZrjrkrlklIJVZ entry_numbermaskZod_IZod_JZod_Vrrrrs                                       zMMFile._parse_bodyc Cst|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||td|||x&|dD]} |td| qW|||} ||jkr|td|| f||j |j|j fkr4||j!krxt"| D]4}x,t"|D] }|t| |||fqlWq^Wn||j#krxt"| D]:}x2t"|d|D] }|t| |||fqWqWnDxBt"| D]6}x.t"||D] }|t| |||fqWqWn||jkr||j!krxt"| D]D}xt"||D]0}|||f}|t| t$|t%|fqWqWn"||j&kr t dn td|n|'}||j!krd|j(|j)k}t*|j+||j(||j)|ff|jd}|td || |j,f|||d} ||j&krx:t-|j(d|j)dD] \}}|td||fqWn||j |j|j fkr:xt-|j(d|j)d|j+D]*\}}}|td!||f| |q Wnl||jkrx^t-|j(d|j)d|j+D]2\}}}|td!||f| |j$|j%fqbWn td|dS)"NZ __array__rKzExpected 2 dimensional arrayr?zBmmwrite does not support integer dtypes larger than native 'intp'.fdrArhrBzunknown matrix type: %sr3ZfFrjr8frcr9uzunsigned-integerzunexpected dtype kind z"%%MatrixMarket matrix {0} {1} {2}  z%%%s z%i %i roz*pattern type inconsisted with dense formatzUnknown field type %s)rrz %i %i %i z%i %i ).r^listr rhasattrrRrrTrrr6r}r rt OverflowErrorZastyper|rurrtypekind TypeErrorr{rr7r<r>rrr.rOrr~rsrpr0rrrrvryrzrdataZnnzzip)r*rXrr r!r"r#Zrepr+r,typecoderrZtemplaterkrjrlZcooZlower_triangle_maskr\rrArrrrs                             ( ((             & &z MMFile._write)r[)rNNN)rNNN)-__name__ __module__ __qualname__rpropertyr+r,r-r.r!r#r2rSrRr5 classmethodr7r}r~r|rrr;r<rsr/r0r1r=r>r staticmethodrCrDrrMr{rrrrrrrrrrrrris`             E 9 9  ",cCstjddkrdSg}yddl}||jWntk rBYnXyddl}||jWntk rpYnXt|}t || S)z Check whether `stream` is compatible with numpy.fromfile. Passing a gzipped file object to ``fromfile/fromstring`` doesn't work with Python3. rrLTN) sys version_inforcappendZGzipFilerrerfrr^)rXZbad_clsrcrerrr_is_fromfile_compatible!sr__main__roZReadingz... )endztook %s seconds)rNNN)(__doc__Z __future__rrrr_rZnumpyrrrrr r r r r Z numpy.compatrrZscipy._lib.sixrrrr__all__rrrobjectrrrZtimeargvfilenameprintstdoutrtrrrrs4,   =