3 6^=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%j&e!j!Z'ee#e$de!j!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 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-vw4w4j08/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)rrrrr;scCstj||||||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 @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: r4r3%zHeader line not of length 2: zHeader line not of length 3: N) _openreadlinesplit startswithr6lower FORMAT_ARRAYFORMAT_COORDINATErElenmapintclose) r*rstreamclose_itlineZmmidrIr.r!r#r+r,r-rrrrs:               z MMFile.inforbcCsd}t|trd}|ddkrtjj|svtjj|drD|d}n2tjj|dr^|d}ntjj|drv|d}|jdrdd l}|j||}q|jd rdd l}|j |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)ZfilespecmoderYrdrXrfrrrrMs.         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_iterator_s z+MMFile._get_symmetry..symm_iteratorc3sRxLtD]@}x:t|dD](}||||}}||fVqWq WdS)N)range)rlrkrmrn)rnrrrohsF)shaperSYMMETRY_GENERALdtypecharrtocooZnonzerosumZtodokrr/r0r1) rmZissymmZisskewZishermrowcolrormrnr)rrrr _get_symmetryJs<      zMMFile._get_symmetryc Cs0tjd|tjdtjdtjd||fij|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__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)rM _parse_header _parse_bodyrW)r*rrXrYrrrrs   z MMFile.readrNc CsF|j|d\}}z|j||||||Wd|r8|jn|jXdS)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.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)rpNr)rFattrrrrrHsz&MMFile._init_attrs..zVfound %s invalid keyword arguments, please only use %srpN) __class__ __slots__setkeysr6tuplesetattrr)r*rattrsZ public_attrsZ invalid_keysrrrrrs  zMMFile._init_attrscCs2|jj|\}}}}}}|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 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|jds|j rq| r.t|}n2| r>t|}n"| rXttt|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,||dksZtdn |d|gkoN||kstdn||jkr|dkrt||f| d}d}d}x|r|j}| s|jds|j rq|j}tt|dd\}}|d|d}}| rt|d}n>| rt|d}n*| r8ttt|dd}n t|d}||||f<| r||kr|rv| |||f<n$|rt||||f<n ||||f<|d}qW||kstdn,||jkr|dkr|||f| dSt|d d}t|d d}|rt|d d}nH| r$t|d d}n4| r8t|d d}n | rLt|d d}n t|dd}d}x|D]}| sb|jdsb|j rqb|d|krtd|j}tt|dd\||<||<|s.| rt|d||<nJ| rt|d||<n2| rttt|dd||<nt|d||<|d7}qbW||krNtd|d8}|d8}| r||k}||}||}||}t ||f}t ||f}|r|d9}n|r|j!}t ||f}||||ff||f| d}nt"||S)Nr)r)rurprJz$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)rsru)rr)#r+r,r-r.r!r# scipy.sparser ImportErrorDTYPES_BY_FIELDrr2r~rrr0r1 FIELD_PATTERNrRr rNrPrErVr9rUrrOrr6rSr r conjugateNotImplementedError) r*rXr+r,r-r.r!Zsymmrrur2 is_integerZis_unsigned_integerZ is_complexZis_skewZis_hermZ is_patternrrZrkrlrmklIJVZ 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|jd}q||jkr|j jdkr|jd}q||jkr|j jdkr|jd }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|}|jj||jj||jj||jtdj|||x&|jdD]} |jtd| qW|j||} ||jkr|jtd|| f||j |j|j fkr4||j!krxt"| D]4}x,t"|D] }|jt| |||fqlWq^Wn||j#krxt"| D]:}x2t"|d|D] }|jt| |||fqWqWnDxBt"| D]6}x.t"||D] }|jt| |||fqWqWn||jkr||j!krxt"| D]D}xt"||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__rKzExpected 2 dimensional arrayr?zBmmwrite does not support integer dtypes larger than native 'intp'.fdrArirBzunknown matrix type: %sr3ZfFrkr8frcr9uzunsigned-integerzunexpected dtype kind z"%%MatrixMarket matrix {0} {1} {2}  z%%%s z%i %i rpz*pattern type inconsisted with dense formatzUnknown field type %s)rsz %i %i %i z%i %i ).r_listr rhasattrrRrrTrsr6r~r ru OverflowErrorZastyper}rvrrtypekind TypeErrorr|rr7r<r>rrr.rOrrrtrqr0rrrrwrzr{rdataZnnzzip)r*rXrr r!r"r#repr+r,typecoderrZtemplaterlrkrmZcooZlower_triangle_maskr\rrArrrrs                             ( ((             & &z MMFile._write)r$r%r&r'r(r))r[)rNNN)rNNN)-__name__ __module__ __qualname__rpropertyr+r,r-r.r!r#r2rSrRr5 classmethodr7r~rr}rrr;r<rtr/r0r1r=r>r staticmethodrCrDrrMr|rrrrrrrrrrrrrisj             I 9 9  ",cCstjddkrdSg}yddl}|j|jWntk rBYnXyddl}|j|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_infordappendGzipFilerrfrgrr_)rXZbad_clsrdrfrrr_is_fromfile_compatible%sr__main__rpZReadingz... )endztook %s seconds)rNNN)(__doc__ __future__rrrr`rZnumpyrrrrr r r r r Z numpy.compatrrZscipy._lib.sixrrrr__all__rrrobjectrrrtimeargvfilenameprintstdoutrtrrrrs4,   A