B }]=q@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)sourcerY/home/fristb/BIRCH/lib-linux-x86_64/python/lib64/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_fieldZgeneral symmetriczskew-symmetric 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 CsL||\}}z&|}dd|D\}}}}} |dsHtd|dks`td||dkrt|j}n|dkr|j}x|d r|}qWx|s|}qW|}||jkrt |d kstd |t t |\} } | | } n*t |d kstd |t t |\} } } | | | ||| fS|rF| 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_COORDINATErGlenmapintclose) r*rstreamclose_itlineZmmidrKr.r!r#r+r,r-rrrrs:               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)Zfilespecmoder[rerZrgrrrrOs.         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)rmrlrnro)rnrrrphsF)shaperSYMMETRY_GENERALdtypecharrtocoononzerosumZtodokrr/r0r1) rmZissymmZisskewZishermrowcolrprnror)rrsr _get_symmetryJs<      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)rO _parse_header _parse_bodyrY)r*rrZr[rrrrs   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)rO_writerYflush) r*rrr r!r"r#rZr[rrrrs  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)rqNr)rHattrrrrrJsz&MMFile._init_attrs..zVfound %s invalid keyword arguments, please only use %srqN) __class__ __slots__setkeysr6tuplesetattrr)r*rattrsZ public_attrsZ invalid_keysrrrrrs  zMMFile._init_attrscCs2|j|\}}}}}}|j||||||ddS)N)r+r,r-r.r!r#)rrr)r*rZr+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}||jkrZt||f| d}d}d\}}|rd|||f<||dkr|d7}x|r|}|r|ds|sq| r*t|}n2| r:t|}n"| rTttt|}nt|}||||f<| r||kr|r| |||f<n$|rt||||f<n ||||f<||dkr|d}q|d}| sd}q|}|rd|||f<||dkr|d7}qW|r6|d|gkr,||dksVtdn |d|gkrN||kstdn||jkr|dkrt||f| d}d}d}x|r|}|r|ds|sq|}tt|dd\}}|d|d}}| rt|d}n>| rt|d}n*| r0ttt|dd}n t|d}||||f<| r||kr|rn| |||f<n$|rt||||f<n ||||f<|d}qW||kstd n(||jkr|dkr|||f| dSt|d d}t|d d}|rt|d d}nH| rt|d d}n4| r0t|d d}n | rDt|dd}n t|dd}d}x|D]}|rZ|dsZ|s~qZ|d|krtd|}tt|dd\||<||<|s"| rt|d||<nJ| rt|d||<n2| rttt|dd||<nt|d||<|d7}qZW||krBtd|d8}|d8}| r||k}||}||}||}t ||f}t ||f}|r|d9}n|r|!}t ||f}||||ff||f| d}nt"||S)Nr)r)rvrq)rrrLz$Parse error, did not read all lines.rMzDid not read all entriesintcint8rArBr9floatz5'entries' in header is smaller than number of entriesz4'entries' in header is larger than number of entries)rtrv)#r+r,r-r.r!r# scipy.sparser ImportErrorDTYPES_BY_FIELDrr2rrrr0r1 FIELD_PATTERNrTr rPrRrGrXr9rWrrQrr6rUr r conjugateNotImplementedError) r*rZr+r,r-r.r!Zsymmrrvr2 is_integerZis_unsigned_integer is_complexZis_skewZis_hermZ is_patternrr\rlrmrnklIJVZ 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)"N __array__rMzExpected 2 dimensional arrayrAzBmmwrite does not support integer dtypes larger than native 'intp'.fdrCrjrDzunknown matrix type: %sr3ZfFrlr8frcr9uzunsigned-integerzunexpected dtype kind z"%%MatrixMarket matrix {0} {1} {2}  z%%%s z%i %i rqz*pattern type inconsisted with dense formatzUnknown field type %s)rtz %i %i %i z%i %i ).r`listr rhasattrrTrrVrtr6rr rv OverflowErrorastyperrwrrtypekind TypeErrorr~rr7r<r@rrr.rQrrrurrr0rrrrxr|r}rdataZnnzzip)r*rZrr r!r"r#repr+r,typecoderr\templatermrlrnZcooZlower_triangle_maskr^rrCrrrrs                             ( ((             & &z MMFile._write)r])rNNN)rNNN)-__name__ __module__ __qualname__rpropertyr+r,r-r.r!r#r2rUrTr5 classmethodr7rrrrrr;r<rur/r0r1r?r@r staticmethodrErFrrOr~rrrrrrrrrrrrris`             I 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. rrNTN) sys version_inforeappendGzipFilerrgrhrr`)rZZbad_clsrergrrr_is_fromfile_compatible%sr__main__rqZReadingz... )endztook %s seconds)rNNN)(__doc__ __future__rrrrarnumpyrrrrr r r r r numpy.compatrrscipy._lib.sixrrrr__all__rrrobjectrrrtimeargvfilenameprintstdoutrtrrrrs4,   A