o hw%@sdZddlmZddlZddlmZgdZedddd d Zedd dej d dd dddZ ej dd dddZ ddZ dddZ dddZdS)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)modecCst|}||dS)zgWrite NetworkX graph G to Graphviz dot format on path. Path can be a string or a file handle. N)rwrite to_string)GpathPrm/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/networkx/drawing/nx_pydot.pyr&srrpydot_read_dotT)namegraphs returns_graphcCs&ddl}|}||}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`. rN)pydotreadgraph_from_dot_datar )rrdataP_listrrrr1s  r)rrc Cs|drd}nd}|dkr|rt}nt}n |r#t}nt}|d}|dkr5||_ | D]}|d}|dvrGq9|j |fi| q9| D]`}|}|}| } g} g} t|trw| |dn|dD] } | | dq{t|tr| |dn|dD] } | | dq| D]}| D] }|j||fi| qqqW| }|r||jd<z |d |jd <Wn ttfyYnwz |d |jd <W|SttfyY|Sw) 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)) NFTgraph")noderedgenodesrr!r") get_strictget_typenx MultiGraphGraph MultiDiGraphDiGraphget_namestripr get_node_listadd_nodeget_attributes get_edge_list get_sourceget_destination isinstancestrappendadd_edgerget_node_defaults IndexError TypeErrorget_edge_defaults)r multiedgesNrpneuvattrsdunodesvnodes source_nodedestination_nodepattrrrrr Tsh            r cCsddl}|r d}nd}t|dko| }|j}|jdi}|dkr3|jd||d|}n|jd|df||d|}z |j di|jdWn t yYYnwz |j di|jd Wn t ypYnw|j d d D]\}}d d | D} t|}|j|fi| } || qw|r|jd d dD]+\} } } }dd | D}t| t| } } |j| | fdt| i|}||q|S|jd d D]&\} } }dd | D}t| t| } } |j| | fi|}||q|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 ----- rNdigraphrr ) graph_typestrictrr!r"T)rcSi|] \}}t|t|qSrr4.0krArrr zto_pydot..)rkeyscSs&i|]\}}|dkrt|t|qS)keyrNrOrrrrRs&rUcSrMrrNrOrrrrRrS)r r)r is_directedr&number_of_selfloops is_multigraphrrgetDotset_node_defaultsKeyErrorset_edge_defaultsr#itemsr4Noder.edgesEdger6)r<rrKrLrgraph_defaultsrr>nodedata str_nodedatar=r@rArUedgedata str_edgedatar"rrrrsX      rneatocCst|||dS)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)rrhrrrrsrcCs(ddl}t|}|dur|dt||j|d}t|td}|dkrFtd|dttd td td td |d dS||}t|dksSJ|d}i} | D]4} t| } | | | } t | t rt| d} | dd} | dur| d\}}t|t|f| | <q]| 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()} rNrj)ri)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 ,)rrsetr4 create_dotrprintrlenr#get_nodequote_id_if_necessaryr3listget_possplitfloat)rrirjrrD_bytesDQ_listQnode_posr>str_nr!posxxyyrrrrs:&    r)rgN)__doc__localernetworkxr&networkx.utilsr__all__r _dispatchablerr rrrrrrrs      ! [ A!