\Mh'dZddlmZddlZddlmZgdZeddd Zedd ej d dd dZ ej dd dZ dZ ddZ ddZdS)a ***** Pydot ***** Import and export NetworkX graphs in Graphviz dot format using pydot. Either this module or nx_agraph can be used to interface with graphviz. Examples -------- >>> G = nx.complete_graph(5) >>> PG = nx.nx_pydot.to_pydot(G) >>> H = nx.nx_pydot.from_pydot(PG) See Also -------- - pydot: https://github.com/erocarrera/pydot - Graphviz: https://www.graphviz.org - DOT Language: http://www.graphviz.org/doc/info/lang.html )getpreferredencodingN) open_file) write_dotread_dotgraphviz_layout pydot_layoutto_pydot from_pydotw)modecrt|}||dS)zWrite NetworkX graph G to Graphviz dot format on path. Parameters ---------- G : NetworkX graph path : string or file Filename or file handle for data output. Filenames ending in .gz or .bz2 will be compressed. N)r write to_string)GpathPs Y/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/networkx/drawing/nx_pydot.pyrr&s.  AJJq{{}} Frpydot_read_dotT)namegraphs returns_graphcddl}|}||}t|dS)aReturns a NetworkX :class:`MultiGraph` or :class:`MultiDiGraph` from the dot file with the passed path. If this file contains multiple graphs, only the first such graph is returned. All graphs _except_ the first are silently ignored. Parameters ---------- path : str or file Filename or file handle to read. Filenames ending in .gz or .bz2 will be decompressed. Returns ------- G : MultiGraph or MultiDiGraph A :class:`MultiGraph` or :class:`MultiDiGraph`. Notes ----- Use `G = nx.Graph(nx.nx_pydot.read_dot(path))` to return a :class:`Graph` instead of a :class:`MultiGraph`. rN)pydotreadgraph_from_dot_datar )rrdataP_lists rrr7sC2LLL 99;;D & &t , ,F fQi  r)rrcZ |}n%#t$r|d}YnwxYw| }|dkr*|rtj}n=tj}n)|rtj}ntj}| d}|dkr||_ | D]N}| d}|dvr.|j |fi| O|D]>}|}|} | } g} g} t#|t$r)| | dn3|dD]*} | | d+t#| t$r)| | dn3| dD]*}| | d+| D]}| D]}|j||fi| @| }|r ||jd< |d|jd<n#t.tf$rYnwxYw |d|jd <n#t.tf$rYnwxYw|S) aReturns a NetworkX graph from a Pydot graph. Parameters ---------- P : Pydot graph A graph created with Pydot Returns ------- G : NetworkX multigraph A MultiGraph or MultiDiGraph. Examples -------- >>> K5 = nx.complete_graph(5) >>> A = nx.nx_pydot.to_pydot(K5) >>> G = nx.nx_pydot.from_pydot(A) # return MultiGraph # make a Graph instead of MultiGraph >>> G = nx.Graph(nx.nx_pydot.from_pydot(A)) Ngraph")noder"edgenodesrr%r&) get_strict TypeErrorget_typenx MultiGraphGraph MultiDiGraphDiGraphget_namestripr get_node_listadd_nodeget_attributes get_edge_list get_sourceget_destination isinstancestrappendadd_edger"get_node_defaults IndexErrorget_edge_defaults)rstrict multiedgesNrpneuvattrsdunodesvnodes source_nodedestination_nodepattrs rr r [sj4$ $$$d##$Jzz||w   AA AA  !!AA A ::<<  c " "D rzz__  ,, JJLL  s # # ) ) )  1++((**++++__  BB LLNN    !!   a   , HHQWWS\\ " " " "G* , ,c**++++ a   , HHQWWS\\ " " " "G* , ,c**++++ B BK$% B B  ;(8AADAAAA B B     E !  --//2  "      --//2  "     Hs0996"KK-,K-1"LL('L(cddl}|rd}nd}tj|dko| }|j}|jdi}|dkr|jd||d|}n|jd|df||d|} |j di|jdn#t$rYnwxYw |j di|jd n#t$rYnwxYw| d D]U\}}d | D} t|}|j|fi| } || V|r|d d D]v\} } } }d| D}t| t| } } |j| | fdt| i|}||wn}|d D]f\} } }d| D}t| t| } } |j| | fi|}||g|S)aReturns a pydot graph from a NetworkX graph N. Parameters ---------- N : NetworkX graph A graph created with NetworkX Examples -------- >>> K5 = nx.complete_graph(5) >>> P = nx.nx_pydot.to_pydot(K5) Notes ----- rNdigraphr"r$) graph_typer?r#r%r&T)rcNi|]"\}}t|t|#Sr9.0krFs r zto_pydot..s*DDD41aAADDDr)rkeyscZi|](\}}|dk t|t|)S)keyrTrUs rrXzto_pydot..s0VVVtq!1PU::CFFCFF:::rr[cNi|]"\}}t|t|#SrSrTrUs rrXzto_pydot..s*HHHtq!CFFCFFHHHr)r$rS)r is_directedr+number_of_selfloops is_multigraphrr"getDotset_node_defaultsKeyErrorset_edge_defaultsr'itemsr9Noder3edgesEdger;)rArrQr?rgraph_defaultsrrCnodedata str_nodedatarBrErFr[edgedata str_edgedatar&s rr r s"LLL }}   #A & &! + EAOO4E4E0EF 6DW[["--N rzz EI QZ Q Q. Q Q EI KKK $.v  AO   ..agfo....       ..agfo....      wwDw)) 8DD8>>3C3CDDD FF EJq ) )L ) ) 1  #$7747#@#@   Aq#xVVx~~7G7GVVVLq663q66qA5:aAACALAAD JJt       gg4g00  NAq(HHx~~7G7GHHHLq663q66qA5:a33l33D JJt     Hs$B33 C?CC C*)C*neatoc&t|||S)a=Create node positions using Pydot and Graphviz. Returns a dictionary of positions keyed by node. Parameters ---------- G : NetworkX Graph The graph for which the layout is computed. prog : string (default: 'neato') The name of the GraphViz program to use for layout. Options depend on GraphViz version but may include: 'dot', 'twopi', 'fdp', 'sfdp', 'circo' root : Node from G or None (default: None) The node of G from which to start some layout algorithms. Returns ------- Dictionary of (x, y) positions keyed by node. Examples -------- >>> G = nx.complete_graph(4) >>> pos = nx.nx_pydot.graphviz_layout(G) >>> pos = nx.nx_pydot.graphviz_layout(G, prog="dot") Notes ----- This is a wrapper for pydot_layout. rprogroot)rrps rrrs< !$T 2 2 22rcddl}t|}|#|dt|||}t|t }|dkrct d|dt t d t d t d t d |d dS||}t|dksJ|d}i} | D]} t| } | | | } t| tr| d} | dd} | 9| d\}}t!|t!|f| | <| S)aCreate node positions using :mod:`pydot` and Graphviz. Parameters ---------- G : Graph NetworkX graph to be laid out. prog : string (default: 'neato') Name of the GraphViz command to use for layout. Options depend on GraphViz version but may include: 'dot', 'twopi', 'fdp', 'sfdp', 'circo' root : Node from G or None (default: None) The node of G from which to start some layout algorithms. Returns ------- dict Dictionary of positions keyed by node. Examples -------- >>> G = nx.complete_graph(4) >>> pos = nx.nx_pydot.pydot_layout(G) >>> pos = nx.nx_pydot.pydot_layout(G, prog="dot") Notes ----- If you use complex node objects, they may have the same string representation and GraphViz could treat them as the same node. The layout may assign both nodes a single location. See Issue #1568 If this occurs in your case, consider relabeling the nodes just for the layout computation using something similar to:: H = nx.convert_node_labels_to_integers(G, label_attribute="node_label") H_layout = nx.nx_pydot.pydot_layout(H, prog="dot") G_layout = {H.nodes[n]["node_label"]: p for n, p in H_layout.items()} rNrr)rq)encodingr$zGraphviz layout with z failedzTo debug what happened try:zP = nx.nx_pydot.to_pydot(G)zP.write_dot("file.dot")z And then run z on file.dotr ,)rr setr9 create_dotrprintrlenr'get_nodequote_id_if_necessaryr8listget_possplitfloat)rrqrrrrD_bytesDQ_listQnode_posrCstr_nr%posxxyys rrrsLLLL A  fc$ii   lll%%G G244555ABww 3d333444  +,,, +,,, '((( 0d000111 & &q ) )F v;;!     q AH WWYY 1 1Azz%55e<<== dD ! ! 7DllnnQrT" ?YYs^^FB 99eBii0HQK Or)rnN)__doc__localernetworkxr+networkx.utilsr__all__r _dispatchablerr r rrrSrrrsE,('''''$$$$$$    13      13'DIII!!JI!DT222Z Z 32Z z> > > B3333BNNNNNNr