ó ,µ[c@sõdZddlmZddlZddlZddlZddlmZddl m Z m Z ddl m Z mZddd d gZd „Zd „Zded „Ze dddƒd„ƒZe dƒe dddƒded„ƒƒZdS(sOFunctions for reading and writing graphs in the *sparse6* format. The *sparse6* file format is a space-efficient format for large sparse graphs. For small graphs or large dense graphs, use the *graph6* file format. For more information, see the `sparse6`_ homepage. .. _sparse6: http://users.cecs.anu.edu.au/~bdm/data/formats.html iÿÿÿÿ(tchainN(t NetworkXError(t open_filetnot_implemented_for(t data_to_nt n_to_datatfrom_sparse6_bytest read_sparse6tto_sparse6_bytest write_sparse6c #sÚt|ƒ}|dkr'tdƒ‚n|r5dVndVx,t|ƒD]}tjt|dƒƒVqGWd‰xdˆ>|krˆd7‰qrW‡fd†}td „|jƒDƒƒ}g}d }xÆ|D]¾\} } | |kr |jd ƒ|j || ƒƒqÎ| |dkrF|d7}|jdƒ|j || ƒƒqÎ| }|jdƒ|j || ƒƒ|jd ƒ|j || ƒƒqÎWˆd kr|dˆ>krt|ƒ d ˆkr||dkr|jd ƒ|j dgt|ƒ d ƒn|j dgt|ƒ d ƒgt d t|ƒd ƒD]h} || d d >|| dd >|| dd>|| dd>|| d d>|| d d >^q:} x&| D]}tjt|dƒƒVq¯WdVdS(s%Yield bytes in the sparse6 encoding of a graph. `G` is an undirected simple graph. `nodes` is the list of nodes for which the node-induced subgraph will be encoded; if `nodes` is the list of all nodes in the graph, the entire graph will be encoded. `header` is a Boolean that specifies whether to generate the header ``b'>>sparse6<<'`` before the remaining data. This function generates `bytes` objects in the following order: 1. the header (if requested), 2. the encoding of the number of nodes, 3. each character, one-at-a-time, in the encoding of the requested node-induced subgraph, 4. a newline character. This function raises :exc:`ValueError` if the graph is too large for the graph6 format (that is, greater than ``2 ** 36`` nodes). ii$s?sparse6 is only defined if number of nodes is less than 2 ** 36s >>sparse6<@r/dnd^q S(sBig endian k-bit encoding of xii(trange(txti(tk(s9lib/python2.7/site-packages/networkx/readwrite/sparse6.pytencHscss3|])\}}t||ƒt||ƒfVqdS(N(tmaxtmin(t.0tutv((s9lib/python2.7/site-packages/networkx/readwrite/sparse6.pys Lsiiiiis NI( tlent ValueErrorRtstrtencodetchrtsortedtedgestappendtextendR ( tGtnodestheadertntdRRtbitstcurvRRR tdata((Rs9lib/python2.7/site-packages/networkx/readwrite/sparse6.pyt_generate_sparse6_bytes%sJ        C "„ c sµ|jdƒr|d}n|jdƒs:tdƒ‚ntjd krsg|dD]}t|ƒd^qT}n!g|dD]}|d^q~}t|ƒ\}‰d‰xdˆ>|kr̈d7‰q¯W‡‡fd†}d }tjƒ}|jt |ƒƒt }xŒ|ƒD]\}} |dkr9|d7}n| |ksQ||krUPq| |krj| }q|j | |ƒr…t }n|j | |ƒqW|s±tj|ƒ}n|S( sURead an undirected graph in sparse6 format from string. Parameters ---------- string : string Data in sparse6 format Returns ------- G : Graph Raises ------ NetworkXError If the string is unable to be parsed in sparse6 format Examples -------- >>> G = nx.from_sparse6_bytes(b':A_') >>> sorted(G.edges()) [(0, 1), (0, 1), (0, 1)] See Also -------- read_sparse6, write_sparse6 References ---------- .. [1] Sparse6 specification s >>sparse6<d@}|}xS|ˆkrÚyt|ƒ}Wntk r¸dSXd}|d>|}|d7}qˆW||ˆ?}|ˆ}||fVqWdS(s5Return stream of pairs b[i], x[i] for sparse6 format.iiNi(titertNonetnextt StopIteration(tchunksR"tdLentbR txLen(R%R(s9lib/python2.7/site-packages/networkx/readwrite/sparse6.pyt parseDatažs2       i(i(t startswithRtsyst version_infotordRtnxt MultiGraphtadd_nodes_fromR tFalsethas_edgetTruetadd_edgetGraph( tstringtctcharsR!R/RRt multigraphR-R ((R%Rs9lib/python2.7/site-packages/networkx/readwrite/sparse6.pyRos:! *!      cCsL|dk r|j|ƒ}ntj|ddƒ}djt|||ƒƒS(sConvert an undirected graph to bytes in sparse6 format. Parameters ---------- G : Graph (undirected) nodes: list or iterable Nodes are labeled 0...n-1 in the order provided. If None the ordering given by ``G.nodes()`` is used. header: bool If True add '>>sparse6<<' bytes to head of data. Raises ------ NetworkXNotImplemented If the graph is directed. ValueError If the graph has at least ``2 ** 36`` nodes; the sparse6 format is only defined for graphs of order less than ``2 ** 36``. Examples -------- >>> nx.to_sparse6_bytes(nx.path_graph(2)) # doctest: +SKIP b'>>sparse6<<:An\n' See Also -------- to_sparse6_bytes, read_sparse6, write_sparse6_bytes Notes ----- The returned bytes end with a newline character. The format does not support edge or node labels. References ---------- .. [1] Graph6 specification torderingRtN(R(tsubgraphR4tconvert_node_labels_to_integerstjoinR&(RRR ((s9lib/python2.7/site-packages/networkx/readwrite/sparse6.pyRÓs, itmodetrbcCsjg}x?|D]7}|jƒ}t|ƒs1q n|jt|ƒƒq Wt|ƒdkrb|dS|SdS(sòRead an undirected graph in sparse6 format from path. Parameters ---------- path : file or string File or filename to write. Returns ------- G : Graph/Multigraph or list of Graphs/MultiGraphs If the file contains multiple lines then a list of graphs is returned Raises ------ NetworkXError If the string is unable to be parsed in sparse6 format Examples -------- You can read a sparse6 file by giving the path to the file:: >>> import tempfile >>> with tempfile.NamedTemporaryFile() as f: ... _ = f.write(b'>>sparse6<<:An\n') ... _ = f.seek(0) ... G = nx.read_sparse6(f.name) >>> list(G.edges()) [(0, 1)] You can also read a sparse6 file by giving an open file-like object:: >>> import tempfile >>> with tempfile.NamedTemporaryFile() as f: ... _ = f.write(b'>>sparse6<<:An\n') ... _ = f.seek(0) ... G = nx.read_sparse6(f) >>> list(G.edges()) [(0, 1)] See Also -------- read_sparse6, from_sparse6_bytes References ---------- .. [1] Sparse6 specification iiN(tstripRRR(tpathtglisttline((s9lib/python2.7/site-packages/networkx/readwrite/sparse6.pyRs3   tdirecteditwbcCsa|dk r|j|ƒ}ntj|ddƒ}x't|||ƒD]}|j|ƒqFWdS(s3Write graph G to given path in sparse6 format. Parameters ---------- G : Graph (undirected) path : file or string File or filename to write nodes: list or iterable Nodes are labeled 0...n-1 in the order provided. If None the ordering given by G.nodes() is used. header: bool If True add '>>sparse6<<' string to head of data Raises ------ NetworkXError If the graph is directed Examples -------- You can write a sparse6 file by giving the path to the file:: >>> import tempfile >>> with tempfile.NamedTemporaryFile() as f: ... nx.write_sparse6(nx.path_graph(2), f.name) ... print(f.read()) # doctest: +SKIP b'>>sparse6<<:An\n' You can also write a sparse6 file by giving an open file-like object:: >>> with tempfile.NamedTemporaryFile() as f: ... nx.write_sparse6(nx.path_graph(2), f) ... _ = f.seek(0) ... print(f.read()) # doctest: +SKIP b'>>sparse6<<:An\n' See Also -------- read_sparse6, from_sparse6_bytes Notes ----- The format does not support edge or node labels. References ---------- .. [1] Sparse6 specification R@RN(R(RBR4RCR&twrite(RRHRR R-((s9lib/python2.7/site-packages/networkx/readwrite/sparse6.pyR Ds 8 (t__doc__t itertoolsRtmathR1tnetworkxR4tnetworkx.exceptionRtnetworkx.utilsRRtnetworkx.readwrite.graph6RRt__all__R&RR(R9RRR (((s9lib/python2.7/site-packages/networkx/readwrite/sparse6.pyts       J d2?