Gd6dZddlZddlmZddlZddlmZgdZeddd Z edd d Z d Z d Z dZ ddZddZdS)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 N)getpreferredencoding) open_file) write_dotread_dotgraphviz_layout pydot_layoutto_pydot from_pydotw)modecd}tj|tdt|}||dS)zgWrite NetworkX graph G to Graphviz dot format on path. Path can be a string or a file handle. znx.nx_pydot.write_dot depends on the pydot package, which hasknown issues and is not actively maintained. Consider usingnx.nx_agraph.write_dot instead. See https://github.com/networkx/networkx/issues/5723 stacklevelN)warningswarnDeprecationWarningr write to_string)GpathmsgPs 9lib/python3.11/site-packages/networkx/drawing/nx_pydot.pyrr&sP ?  M#)a8888 AJJq{{}} Frcddl}d}tj|td|}||}t |dS)aOReturns 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. 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`. rNznx.nx_pydot.read_dot depends on the pydot package, which hasknown issues and is not actively maintained. Consider usingnx.nx_agraph.read_dot instead. See https://github.com/networkx/networkx/issues/5723rr)pydotrrrreadgraph_from_dot_datar )rrrdataP_lists rrr8sg.LLL ?  M#)a8888 99;;D & &t , ,F fQi  rcNd}tj|td|drd}nd}|dkr*|rt j}n=t j}n)|rt j}nt j }| d}|d kr||_ | D]N}| d}|d vr.|j|fi|O|D]>}|}|} |} g} g} t'|t(r)| | dn3|d D]*} | | d+t'| t(r)| | dn3| d D]*}| | d+| D]}| D]}|j||fi| @|}|r ||jd< |d |jd <n#t2t4f$rYnwxYw |d |jd<n#t2t4f$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)) znx.nx_pydot.from_pydot depends on the pydot package, which hasknown issues and is not actively maintained. See https://github.com/networkx/networkx/issues/5723rrNFTgraph")noder%edgenodesrr(r))rrr get_strictget_typenx MultiGraphGraph MultiDiGraphDiGraphget_namestripname get_node_listadd_nodeget_attributes get_edge_list get_sourceget_destination isinstancestrappendadd_edger%get_node_defaults IndexError TypeErrorget_edge_defaults)rr multiedgesNr4pneuvattrsdunodesvnodes source_nodedestination_nodepattrs rr r bsc0 ?  M#)a8888||D  zz||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  "     Hs$0"KK'&K'+"LL"!L"c:d|vo|ddkp |ddkS)N:rr&)rKs r_check_colon_quotesrVs( !8 514" 5rc0ddl}d}tj|td|rd}nd}t j|dko| }|j}|j di}|dkr|j d||d |}n|j d |d f||d |} |j di|j d n#t$rYnwxYw |jdi|j d n#t$rYnwxYw|d D]\}} d| D} t#|}t%|p*t'd| D} | rt)d|j|fi| } || |r|d d D]\} }}}d|D}t#| t#|}} t%| p9t%|p*t'd|D} | rt)d|j| |fdt#|i|}||n|d D]\} }}d|D}t#| t#|}} t%| p9t%|p*t'd|D} | rt)d|j| |fi|}|||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 ----- rNznx.nx_pydot.to_pydot depends on the pydot package, which hasknown issues and is not actively maintained. See https://github.com/networkx/networkx/issues/5723rrdigraphr%r') graph_typestrictr&r(r)T)r"cNi|]"\}}t|t|#SrUr<.0krIs r zto_pydot..s*DDD41aAADDDrc3XK|]%\}}t|pt|V&dSNrVr]s r zto_pydot..sPAq%Q''A+>q+A+ArzNode names and attributes should not contain ":" unless they are quoted with "". For example the string 'attribute:data1' should be written as '"attribute:data1"'. Please refer https://github.com/pydot/pydot/issues/258)r"keyscZi|](\}}|dk t|t|)S)keyr\r]s rr`zto_pydot..s0VVVtq!1PU::CFFCFF:::rc3XK|]%\}}t|pt|V&dSrbrcr^r_vals rrdzto_pydot..P"As-Q//K3Fs3K3KraNode names and attributes should not contain ":" unless they are quoted with "". For example the string 'attribute:data1' should be written as '"attribute:data1"'. Please refer https://github.com/pydot/pydot/issues/258rgcNi|]"\}}t|t|#SrUr\r]s rr`zto_pydot..(s*HHHtq!CFFCFFHHHrc3XK|]%\}}t|pt|V&dSrbrcris rrdzto_pydot...rkr)r'rU)rrrr is_directedr-number_of_selfloops is_multigraphr4r%getDotset_node_defaultsKeyErrorset_edge_defaultsr*itemsr<rVany ValueErrorNoder6edgesEdger>)rDrrrYrZr4graph_defaultsrrFnodedata str_nodedata raise_errorrErHrIrgedgedata str_edgedatar)s rr r s"LLL ?  M#)a8888 }}   #A & &! + EAOO4E4E0EF 6DW[["--N rzz EI QZ Q Q. Q Q EI KKK $.v  AO   ..agfo....       ..agfo....      wwDw)) 8DD8>>3C3CDDD FF)!,, (..00     H  EJq ) )L ) ) 1 ,#$7747#@#@   Aq#xVVx~~7G7GVVVLq663q66qA#A&&&q))&2&8&8&:&:   L 5:aAACALAAD JJt    ) . gg4g00  NAq(HHx~~7G7GHHHLq663q66qA#A&&&q))&2&8&8&:&:   L 5:a33l33D JJt     Hs$8C CC"C;; DDneatocbd}tj|tdt|||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. znx.nx_pydot.graphviz_layout depends on the pydot package, which hasknown issues and is not actively maintained. Consider usingnx.nx_agraph.graphviz_layout instead. See https://github.com/networkx/networkx/issues/5723rr)rprogroot)rrrr)rrrrs rrr?s;> ?  M#)a8888 !$T 2 2 22rc:ddl}d}tj|tdt |}|#|dt |||}t |t}|d krctd |d ttd td tdtd|ddS| |}t|dksJ|d} i} | D]} t | } t| rtd|| } | | }t%|t&r|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(G, prog='dot') G_layout = {H.nodes[n]['node_label']: p for n, p in H_layout.items()} rNznx.nx_pydot.pydot_layout depends on the pydot package, which hasknown issues and is not actively maintained. See https://github.com/networkx/networkx/issues/5723rrr)r)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 zNode names and node attributes should not contain ":" unless they are quoted with "". For example the string 'attribute:data1' should be written as '"attribute:data1"'. Please refer https://github.com/pydot/pydot/issues/258rT,)rrrrr setr< create_dotrprintr!lenr*rVrxryr2get_noder;listget_possplitfloat)rrrrrrD_bytesDQ_listQnode_posrFstr_n pydot_noder(posxxyys rrrhs$LLLL ?  M#)a8888 A  fc$ii   lll%%G G244555ABww 3d333444  +,,, +,,, '((( 0d000111 & &q ) )F v;;!     q AH WWYY11A u % % H  ZZ&&//11 zz*%% dD ! ! 7DllnnQrT" ?YYs^^FB 99eBii0HQK Or)rN)__doc__rlocalernetworkxr-networkx.utilsr__all__rrr rVr rrrUrrrs *''''''$$$$$$    13   " 13&!&!&!R^ ^ ^ B666r r r j&3&3&3&3R\\\\\\r