ó ˽÷Xc@`s´dZddlmZmZmZddlZddlZddlmZm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZddlmZmZddlmZddlmZmZdd d d gZd „Zd „Z de!e!e!d„Z"d e#fd„ƒYZ$d„Z%e&dkr°ddl'Z'xfej(dD]WZ)e*de)dddƒej+j,ƒe'j'ƒZ-e e)ƒe*de'j'ƒe-ƒqRWndS(s‡ Matrix Market I/O in Python. See http://math.nist.gov/MatrixMarket/formats.html for information about the Matrix Market format. i(tdivisiontprint_functiontabsolute_importN(tasarraytrealtimagtconjtzerostndarrayt concatenatetonestascontiguousarraytvstacktsavetxttfromfilet fromstringtcan_cast(tasbytestasstr(t string_types(t coo_matrixt isspmatrixtmminfotmmreadtmmwritetMMFilecC`s tj|ƒS(s¢ 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'. (Rtinfo(tsource((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pyRscC`stƒj|ƒS(s‚ 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. (Rtread(R((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pyR<stcC`s#tƒj||||||ƒdS(sò 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(Rtwrite(ttargettatcommenttfieldt precisiontsymmetry((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pyRQscB`sæeZd,Zed„ƒZed„ƒZed„ƒZed „ƒZed „ƒZed „ƒZ ed „ƒZ d Z dZ e e fZ ed„ƒZdZdZdZdZeeeefZed„ƒZdZdZdZdZeeeefZed„ƒZide6de6de6de6Zed„ƒZed„ƒZed„ƒZ ed d!„ƒZ!ed"„ƒZ"ed#„ƒZ#d$„Z$d%„Z%d&d-d-d-d'„Z'd(„Z(d)„Z)d*„Z*d&d-d-d-d+„Z+RS(.t_rowst_colst_entriest_formatt_fieldt _symmetrycC`s|jS(N(R%(tself((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pytrowsrscC`s|jS(N(R&(R+((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pytcolsvscC`s|jS(N(R'(R+((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pytentrieszscC`s|jS(N(R((R+((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pytformat~scC`s|jS(N(R)(R+((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pyR"‚scC`s|jS(N(R*(R+((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pyR$†scC`s|j|j|j|jfkS(N(R*tSYMMETRY_SYMMETRICtSYMMETRY_SKEW_SYMMETRICtSYMMETRY_HERMITIAN(R+((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pyt has_symmetryŠs t coordinatetarraycC`s/||jkr+td||jfƒ‚ndS(Ns)unknown format type %s, must be one of %s(t FORMAT_VALUESt ValueError(R+R/((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pyt_validate_format•stintegerRtcomplextpatterncC`s/||jkr+td||jfƒ‚ndS(Ns(unknown field type %s, must be one of %s(t FIELD_VALUESR7(R+R"((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pyt_validate_field¢stgeneralt symmetricsskew-symmetrict hermitiancC`s/||jkr+td||jfƒ‚ndS(Ns+unknown symmetry type %s, must be one of %s(tSYMMETRY_VALUESR7(R+R$((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pyt_validate_symmetry°stintptdtDcC`sdS(N((((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pytreader¼scC`sdS(N((((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pytwriterÁscC`sÜ|j|ƒ\}}z¬|jƒ}g|jƒD]}t|jƒƒ^q1\}}}} } |jdƒs|tdƒ‚n|jƒdks¡td|ƒ‚n|jƒdkr¿|j}n|jƒdkrÝ|j }nx|jdƒrþ|jƒ}qàW|jƒ}||jkrat |ƒdks?td |ƒ‚nt t |ƒ\} } | | } n=t |ƒd ks†td |ƒ‚nt t |ƒ\} } } | | | || jƒ| jƒfSWd |r×|j ƒnXd S( só 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'. s%%MatrixMarkets%source is not in Matrix Market formattmatrixsProblem reading file header: R5R4t%isHeader line not of length 2: isHeader line not of length 3: N(t_opentreadlinetsplitRtstript startswithR7tlowert FORMAT_ARRAYtFORMAT_COORDINATEtlentmaptinttclose(R+Rtstreamtclose_ittlinetparttmmidRHR/R"R$R,R-R.((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pyRÆs6 :    trbcC`sZt}t|tƒrJt}|ddkrtjj|ƒs©tjj|dƒr`|d}q©tjj|dƒrƒ|d}q©tjj|dƒr©|d}q©n|jdƒrÙddl}|j ||ƒ}qG|jdƒr ddl }|j |d ƒ}qGt ||ƒ}qP|d dkr8|d}nt ||ƒ}n|}||fS( s˜ 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. itrs.mtxs.mtx.gzs.mtx.bz2s.gzNs.bz2R[iüÿÿÿ( tFalset isinstanceRtTruetostpathtisfiletendswithtgziptopentbz2tBZ2File(tfilespectmodeRWRdRVRf((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pyRJ s.     c `s{ˆj\}‰|ˆkr"tjSt}t}ˆjjdk}tˆƒr³ˆjƒ‰ˆjƒ\}}||kj ƒ||kj ƒkr•tjSˆj ƒ‰‡fd†}n‡‡fd†}x…|ƒD]z\}} |rö|| kröt }n|r|| krt }n|r3|t | ƒkr3t }n|pB|pB|sÏPqÏqÏW|rZtj S|rgtjS|rttjStjS(NtFDc3`sQxJˆjƒD]<\\}}}||kr ˆ||f}||fVq q WdS(N(titems(titjtaijtaji(R (s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pyt symm_iteratorZs c3`saxZtˆƒD]L}xCt|dˆƒD].}ˆ||ˆ||}}||fVq'Wq WdS(Ni(trange(RmRlRnRo(R tn(s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pyRpcs(tshapeRtSYMMETRY_GENERALR_tdtypetcharRttocootnonzerotsumttodokR]RR0R1R2( R tmtissymmtisskewtishermtrowtcolRpRnRo((R Rrs,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pyt _get_symmetryEs<   $     cC`s<id|tj6dtj6d||ftj6j|dƒS(Ns%%.%ie s%i s%%.%ie %%.%ie (Rt FIELD_REALt FIELD_INTEGERt FIELD_COMPLEXtgettNone(R"R#((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pyt_field_template~s  cK`s|j|dS(N(t _init_attrs(R+tkwargs((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pyt__init__‡scC`sN|j|ƒ\}}z|j|ƒ|j|ƒSWd|rI|jƒnXdS(s­ 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(RJt _parse_headert _parse_bodyRU(R+RRVRW((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pyR‹s  Rc C`s]|j|dƒ\}}z |j||||||ƒWd|rN|jƒn |jƒXdS(s6 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. twbN(RJt_writeRUtflush( R+RR R!R"R#R$RVRW((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pyR¦s   cK`sŸ|jj}g|D]}|d^q}t|jƒƒt|ƒ}|rjtdt|ƒ|fƒ‚nx.|D]&}t|||j|ddƒƒqqWdS(sr Initialize each attributes with the corresponding keyword arg value or a default of None isVfound %s invalid keyword arguments, please only use %sN( t __class__t __slots__tsettkeysR7ttupletsetattrR…R†(R+R‰tattrstattrt public_attrst invalid_keys((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pyRˆÉs   c C`sV|jj|ƒ\}}}}}}|jd|d|d|d|d|d|ƒdS(NR,R-R.R/R"R$(RRRˆ(R+RVR,R-R.R/R"R$((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pyR‹Üs$!cC`s‹|j|j|j|j|j|jf\}}}}}}yddlm}Wntk rid}nX|j j |dƒ} |j } ||j k} ||jk} ||jk} ||jk}||jk}||jkr`t||fd| ƒ}d}d\}}x!|r/|jƒ}| s|jdƒr=qn| rRt|ƒ}n0| rvttt|jƒƒŒ}n t|ƒ}||||f<| rð||krð| r¾| |||f|jdƒ}q>qA||jkrA|j jdkr>|jd ƒ}q>qAq~n:t|ƒsit d t|ƒƒ‚nd }|j\}} |j j} |dkr´| d kr«d }q´d}n|dkr?|j j} | dkrt |j dƒsùtdƒ‚nd}q?| dkrd}q?| dkr,d}q?td| ƒ‚n|dkr]|j|ƒ}n|jj|ƒ|jj|ƒ|jj|ƒ|jtdj|||ƒƒƒx.|jdƒD]} |jtd| ƒƒq¿W|j ||ƒ} ||jkr|jtd|| fƒƒ||j |jfkrì||j!kr—x¡t"| ƒD]>}x5t"|ƒD]'}|jt| |||fƒƒqeWqRWqxut"| ƒD]A}x8t"||ƒD]'}|jt| |||fƒƒqºWq¤Wq||jkrá||j!krtxÑt"| ƒD]V}xMt"|ƒD]?}|||f}|jt| t#|ƒt$|ƒfƒƒq*WqWqx˜t"| ƒD]Y}xPt"||ƒD]?}|||f}|jt| t#|ƒt$|ƒfƒƒq—WqWq||j%krÿt dƒ‚qtd|ƒ‚ný|j&ƒ}||j!krx|j'|j(k}t)|j*||j'||j(|ffd|jƒ}n|jtd|| |j+fƒƒ|j ||dƒ} ||j%krxLt,|j'd|j(dƒD])\}}|jtd||fƒƒqÞWn||j |jfkr…xæt,|j'd|j(d|j*ƒD]4\}}}|jtd||f| |ƒƒqJWnŠ||jkrÿxxt,|j'd|j(d|j*ƒD]@\}}}|jtd||f| |j#|j$fƒƒq¸Wntd|ƒ‚dS( Nt __array__isExpected 2 dimensional arrayRCsBmmwrite does not support integer dtypes larger than native 'intp'.tfdRDRjREsunknown matrix type: %sR4tfFiiRlR9tfRtcR:sunexpected dtype kind s"%%MatrixMarket matrix {0} {1} {2} s s%%%s s%i %i s*pattern type inconsisted with dense formatsUnknown field type %sRss %i %i %i is%i %i (-R^tlistRR”thasattrRPRRRRsR7R†RƒRRut OverflowErrortastypeR‚RvR„Rttypetkindt TypeErrorRRR8R=RBRRR/RLR‡RtRqRRR RwRR€Rtdatatnnztzip(R+RVR R!R"R#R$trepR,R-ttypecodeR½RXttemplateRmRlRntcootlower_triangle_maskR\R·RD((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pyRŽxs¼                  ,,   #*$33"(s_rowss_colss_entriess_formats_fields _symmetryN(,t__name__t __module__R‘tpropertyR,R-R.R/R"R$R3RQRPR6t classmethodR8RƒR‚R„R R<R=RtR0R1R2RARBRŸt staticmethodRFRGRRJRR‡RŠRR†RRˆR‹RŒRŽ(((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pyRjsd    E99    "   • cC`sŸtjddkrtSg}y ddl}|j|jƒWntk rPnXy ddl}|j|jƒWntk r„nXt |ƒ}t ||ƒ S(s  Check whether `stream` is compatible with numpy.fromfile. Passing a gzipped file object to ``fromfile/fromstring`` doesn't work with Python3. iiN( tsyst version_infoR_RdtappendtGzipFileRžRfRgR”R^(RVtbad_clsRdRf((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pyt_is_fromfile_compatibles     t__main__itReadings...tendt stook %s seconds(.t__doc__t __future__RRRR`RÌtnumpyRRRRRRR R R R R RRRt numpy.compatRRtscipy._lib.sixRRRRt__all__RRR†RtobjectRRÑRÇttimetargvtfilenametprinttstdoutRtt(((s,/tmp/pip-build-7oUkmx/scipy/scipy/io/mmio.pyts.  ^  ÿÿž