\Mh(6dZddlZddlmZddlmZmZddlm Z m Z gdZ dZ ej dd d Zdd Ze dd ej dd dZe de dd ddZdS)aPFunctions 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: https://users.cecs.anu.edu.au/~bdm/data/formats.html N) NetworkXError) data_to_n n_to_data)not_implemented_for open_file)from_sparse6_bytes read_sparse6to_sparse6_bytes write_sparse6c#H Kt|}|dkrtd|rdVdVt|D].}tt |dzV/d d z|kr dz d z|k fd}t d|D}g d }|D]\}} ||kr4 d  || ?||dzkr9|dz } d || |} d || d  ||  d kri|d zkr`t  d z krI||dz kr@ d  dgt  d zzn* dgt  d zz fd td t d D} | D].}tt |dzV/d Vd S)a%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). l@z?sparse6 is only defined if number of nodes is less than 2 ** 36 >>sparse6<<:?c>fdtDS)zBig endian k-bit encoding of xc4g|]}ddz |z zzrdndS)rr).0ikxs Z/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/networkx/readwrite/sparse6.py z8_generate_sparse6_bytes..enc..=s4EEEqa!A **2EEE)range)rrs`rencz$_generate_sparse6_bytes..enc;s'EEEEEE!HHEEEErc3\K|]'\}}t||t||fV(dS)N)maxmin)ruvs r z*_generate_sparse6_bytes..?s;@@daC1IIs1ayy)@@@@@@rrcg|][}|dzdz|dzdzz|dzdzz|dzdzz|dzdzz|dzdzz\S)rrr)rrbitss rrz+_generate_sparse6_bytes..Zs     a!e  A;!   A;!   A;!   A;!    A;!      r N) len ValueErrorrstrencodechrsortededgesappendextendr) Gnodesheaderndrr1curvr!r datar)rs @@r_generate_sparse6_bytesr;s* AAEzz M    JJJ q\\&&jjQV%%%%%% A q&1** Q q&1**FFFFF @@aggii@@@ @ @E D D   1 99 KKNNN KKA     $(]] AID KKNNN KKA    D KKNNN KKA    KKNNN KKA    1uuqAvc$iiZ1$4#:#:tq1u~~ A QCSYYJ!+,---- QCSYYJ!+,---    q#d))Q''   D&&jjQV%%%%%% KKKKKrT)graphs returns_graphcx |dr |dd}|dstdd|ddD}t|\} d d z|kr dz d z|k fd}d }tj}|t |d }|D]U\}}|dkr|dz }||ks||krn8||kr|}'|||rd }|||V|stj |}|S) aVRead 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 r Nrz!Expected leading colon in sparse6cg|]}|dz S)rr)rcs rrz&from_sparse6_bytes..s ( ( (QV ( ( (rrc3ZKt}d}d} |dkr$ t|}n#t$rYdSwxYwd}|dz}||z dz}|d|zdz z}|}|kr7 t|}n#t$rYdSwxYwd}|dz|z}|dz }|k7||z z }|z }||fV)z6Returns stream of pairs b[i], x[i] for sparse6 format.Nrrr#)iternext StopIteration)chunksr8dLenbrxLenr:rs r parseDataz%from_sparse6_bytes..parseDatasd  axxV AA$FF AIDdaAa4i1_%AD((V AA$FF!VqL ((dQhA!8DQ$JJJ- s/ ==#A33 BBrFT) startswithrrnx MultiGraphadd_nodes_fromrhas_edgeadd_edgeGraph) stringcharsr7rJr!r4 multigraphrHrr:rs @@rrrisD((   T " "A?@@@ ( (VABBZ ( ( (EGAt A q&1** Q q&1**< A AU1XXJ  1 66 FA 66Q!VV E UUAAzz!Q "! JJq!      HQKK Hrc|||}tj|d}dt |||S)aConvert 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)) 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 Nr0orderingr)subgraphrLconvert_node_labels_to_integersjoinr;)r4r5r6s rr r sPX  JJu   *1x@@@A 88+Auf== > >>rrb)modecg}|D]H}|}t|s&|t|It|dkr|dS|S)aBRead an undirected graph in sparse6 format from path. Parameters ---------- path : file or string Filename or file handle to read. Filenames ending in .gz or .bz2 will be decompressed. 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(delete=False) 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 rr)stripr+r2r)pathglistlines rr r srj E//zz||4yy   '--.... 5zzQQx rdirectedrwbc|||}tj|d}t|||D]}||dS)aXWrite graph G to given path in sparse6 format. Parameters ---------- G : Graph (undirected) path : file or string File or filename to write. Filenames ending in .gz or .bz2 will be compressed. 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(delete=False) as f: ... nx.write_sparse6(nx.path_graph(2), f.name) ... print(f.read()) 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()) 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 Nr0rV)rXrLrYr;write)r4r_r5r6rHs rr r >sgr  JJu   *1x@@@A $Qv 6 6 1 r)NT)__doc__networkxrLnetworkx.exceptionrnetworkx.readwrite.graph6rrnetworkx.utilsrr__all__r; _dispatchablerr r r rrrrms]  ,,,,,,::::::::99999999 U U UOOOdT222^ ^ 32^ B/?/?/?/?d 14T222<<32<~Z   14;;;! ;;;r