ó |£*^c@sîdZddlZddlmZddlmZmZddlm Z m Z m Z m Z dj ddgƒZd d d d gZd „Zedddƒd„ƒZedƒded„ƒZedddƒded„ƒZd„ZdS(s@Sparse6 Read and write graphs in sparse6 format. Format ------ "graph6 and sparse6 are formats for storing undirected graphs in a compact manner, using only printable ASCII characters. Files in these formats have text type and contain one line per graph." See http://cs.anu.edu.au/~bdm/data/formats.txt for details. iÿÿÿÿN(t NetworkXError(t open_filetnot_implemented_for(tdata_to_graph6tgraph6_to_datat data_to_nt n_to_datas s Tomas Gavenciak s#Aric Hagberg |kr|ˆd7‰q_W‡‡fd†}d}tjƒ}|jt|ƒƒt}xŒ|ƒD]\}}|dkré|d7}n||ks||krPqÄ||kr|}qÄ|j ||ƒr5t }n|j ||ƒqÄW|satj |ƒ}n|S(sDRead 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.parse_sparse6(':A_') >>> sorted(G.edges()) [(0, 1), (0, 1), (0, 1)] See Also -------- generate_sparse6, read_sparse6, write_sparse6 References ---------- Sparse6 specification: http://cs.anu.edu.au/~bdm/data/formats.txt s >>sparse6<d@}|}x:|ˆkr¨t|ƒ}d}|d>|}|d7}qoW||ˆ?}|ˆ}||fVqWdS(s5Return stream of pairs b[i], x[i] for sparse6 format.iiiN(titertNonetnext(tchunkstdtdLentbtxtxLen(tdatatk(sn/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/networkx/readwrite/sparse6.pyt parseDataIs&       i( t startswithRRRtnxt MultiGraphtadd_nodes_fromtrangetFalsethas_edgetTruetadd_edgetGraph(tstringtnRtvtGt multigraphRR((RRsn/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/networkx/readwrite/sparse6.pyR!s4       itmodetrtcCsjg}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 multple lines then a list of graphs is returned Raises ------ NetworkXError If the string is unable to be parsed in sparse6 format Examples -------- >>> nx.write_sparse6(nx.Graph([(0,1),(0,1),(0,1)]), 'test.s6') >>> G = nx.read_sparse6('test.s6') >>> sorted(G.edges()) [(0, 1)] See Also -------- generate_sparse6, read_sparse6, parse_sparse6 References ---------- Sparse6 specification: http://cs.anu.edu.au/~bdm/data/formats.txt iiN(tstriptlentappendR(tpathtglisttline((sn/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/networkx/readwrite/sparse6.pyRws"   tdirectedcs.|jƒ}d‰xdˆ>|kr2ˆd7‰qW‡fd†}|d krct|jƒƒ‰n t|ƒ‰t‡fd†ttˆƒƒDƒƒ}g|jƒD] \}}||||f^q¤}g|D]*\}}t||ƒt ||ƒf^qÑ}|j ƒg} d} xÆ|D]¾\}}|| krY| j dƒ| j ||ƒƒq|| dkr–| d7} | j dƒ| j ||ƒƒq|} | j dƒ| j ||ƒƒ| j dƒ| j ||ƒƒqWˆdkrR|dˆ>krRt| ƒ dˆkrR| |dkrR| j dƒ| j dgt| ƒ dƒn| j dgt| ƒ dƒgtdt| ƒdƒD]h} | | dd>| | dd>| | dd >| | d d>| | dd>| | dd>^qŠ} d t t|ƒƒt | ƒ} |r&d | S| Sd S( smGenerate sparse6 format string from an undirected graph. 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<<' string to head of data Returns ------- s : string String in sparse6 format Raises ------ NetworkXError If the graph is directed Examples -------- >>> G = nx.MultiGraph([(0, 1), (0, 1), (0, 1)]) >>> nx.generate_sparse6(G) '>>sparse6<<:A_' See Also -------- read_sparse6, parse_sparse6, write_sparse6 Notes ----- The format does not support edge or node labels. References ---------- Sparse6 specification: http://cs.anu.edu.au/~bdm/data/formats.txt for details. ics9gtˆƒD](}|dˆd|>@r/dnd^q S(sBig endian k-bit encoding of xii(R(Rti(R(sn/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/networkx/readwrite/sparse6.pytencÔsc3s|]}ˆ||fVqdS(N((t.0R0(tns(sn/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/networkx/readwrite/sparse6.pys ÜsiiiiiiR s >>sparse6<>sparse6<<' string to head of data Raises ------ NetworkXError If the graph is directed Examples -------- >>> G = nx.Graph([(0, 1), (0, 1), (0, 1)]) >>> nx.write_sparse6(G, 'test.s6') See Also -------- read_sparse6, parse_sparse6, generate_sparse6 Notes ----- The format does not support edge or node labels. References ---------- Sparse6 specification: http://cs.anu.edu.au/~bdm/data/formats.txt for details. R6R=s N(twriteR (R%R,R6R=((sn/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/networkx/readwrite/sparse6.pyR s(cCs2ddl}|jjdƒr.|jdƒndS(Niÿÿÿÿstest.s6(tosR,tisfiletunlink(ttestRE((sn/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/networkx/readwrite/sparse6.pytteardown_module1s (t__doc__tnetworkxRtnetworkx.exceptionRtnetworkx.utilsRRtnetworkx.readwrite.graph6RRRRtjoint __author__t__all__RRR RR R RI(((sn/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/networkx/readwrite/sparse6.pyt s "    V- `+