ó ,µ[c@sÜdZddlZddlZddlZddddddd gZdd „Zd „Zd „Z d „Z dddd„Z dddd„Z ej jddƒdddddd„ƒZdddd„Zd„ZdS(s  *************** Graphviz AGraph *************** Interface to pygraphviz AGraph class. Examples -------- >>> G = nx.complete_graph(5) >>> A = nx.nx_agraph.to_agraph(G) >>> H = nx.nx_agraph.from_agraph(A) See Also -------- Pygraphviz: http://pygraphviz.github.io/ iÿÿÿÿNt from_agrapht to_agrapht write_dottread_dottgraphviz_layouttpygraphviz_layouttview_pygraphvizc Csé|d kr`|jƒr<|jƒr0tj}q]tj}q`|jƒrTtj}q`tj}ntjd|ƒ}|j d k r|j |_ n|j j |j ƒxC|j ƒD]5}d„|jjƒDƒ}|jt|ƒ|q°Wx·|jƒD]©}t|dƒt|dƒ}}t|jƒ}d„|jƒDƒ}|jƒsƒ|j d k rm|j |d>> K5 = nx.complete_graph(5) >>> A = nx.nx_agraph.to_agraph(K5) >>> G = nx.nx_agraph.from_agraph(A) Notes ----- The Graph G will have a dictionary G.graph_attr containing the default graphviz attributes for graphs, nodes and edges. Default node attributes will be in the dictionary G.node_attr which is keyed by node. Edge attributes will be returned as edge data in G. With edge_attr=False the edge data will be the Graphviz edge weight attribute or the value 1 if no edge weight attribute is found. icSs%i|]\}}|t|ƒ“qS((tstr(t.0tktv((s9lib/python2.7/site-packages/networkx/drawing/nx_agraph.pys Ys icSs%i|]\}}|t|ƒ“qS((R(RR R ((s9lib/python2.7/site-packages/networkx/drawing/nx_agraph.pys `s tkeytgraphtnodetedgeN(tNonet is_directedt is_stricttnxtDiGrapht MultiDiGraphtGrapht MultiGrapht empty_graphtnameR tupdatet graph_attrtnodestattrtitemstadd_nodeRtedgestdictt is_multigraphtadd_edget node_attrt edge_attr( tAt create_usingtNtntstr_attrtetuR R((s9lib/python2.7/site-packages/networkx/drawing/nx_agraph.pyR%s8        !  c Cs’yddl}Wn tk r2tddƒ‚nX|jƒ}tj|ƒdko^|jƒ }|jd|jd|d|ƒ}|jj |j j d iƒƒ|j j |j j d iƒƒ|j j |j j d iƒƒ|jj d „|j jƒDƒƒxk|jd tƒD]W\}}|j|ƒ|dk r|j|ƒ}|jj d„|jƒDƒƒqqW|jƒrx|jd tdtƒD]u\}} } } d„| jƒDƒ} |j|| dt| ƒƒ| dk r•|j|| ƒ}|jj | ƒq•q•Wn}xz|jd tƒD]f\}} } d„| jƒDƒ} |j|| ƒ| dk r$|j|| ƒ}|jj | ƒq$q$W|S(sÂReturn a pygraphviz graph from a NetworkX graph N. Parameters ---------- N : NetworkX graph A graph created with NetworkX Examples -------- >>> K5 = nx.complete_graph(5) >>> A = nx.nx_agraph.to_agraph(K5) Notes ----- If N has an dict N.graph_attr an attempt will be made first to copy properties attached to the graph (see from_agraph) and then updated with the calling arguments if any. iÿÿÿÿNsrequires pygraphviz shttp://pygraphviz.github.io/iRtstricttdirectedR R Rcss-|]#\}}|dkr||fVqdS(R R RN(R R R((RR R ((s9lib/python2.7/site-packages/networkx/drawing/nx_agraph.pys ’s tdatacSs%i|]\}}t|ƒ|“qS((R(RR R ((s9lib/python2.7/site-packages/networkx/drawing/nx_agraph.pys šs tkeyscSs1i|]'\}}|dkrt|ƒ|“qS(R (R(RR R ((s9lib/python2.7/site-packages/networkx/drawing/nx_agraph.pys Ÿs R cSs%i|]\}}t|ƒ|“qS((R(RR R ((s9lib/python2.7/site-packages/networkx/drawing/nx_agraph.pys ¨s (t pygraphvizt ImportErrorRRtnumber_of_selfloopsR!tAGraphRRRR tgetR#R$RRtTrueRRtget_nodeRRR"Rtget_edge( R'R0R-R,R%R(tnodedatataR+R R tedgedatat str_edgedata((s9lib/python2.7/site-packages/networkx/drawing/nx_agraph.pyRps>   "!#  ' + " cCsZyddl}Wn tk r2tddƒ‚nXt|ƒ}|j|ƒ|jƒdS(s¼Write NetworkX graph G to Graphviz dot format on path. Parameters ---------- G : graph A networkx graph path : filename Filename or file handle to write iÿÿÿÿNsrequires pygraphviz shttp://pygraphviz.github.io/(R0R1Rtwritetclear(tGtpathR0R%((s9lib/python2.7/site-packages/networkx/drawing/nx_agraph.pyR±s      cCsOyddl}Wn tk r2tddƒ‚nX|jd|ƒ}t|ƒS(s—Return a NetworkX graph from a dot file on path. Parameters ---------- path : file or string File name or file handle to read. iÿÿÿÿNsread_dot() requires pygraphviz shttp://pygraphviz.github.io/tfile(R0R1R3R(R?R0R%((s9lib/python2.7/site-packages/networkx/drawing/nx_agraph.pyRÆs  tneatotcCst|d|d|d|ƒS(sCreate node positions for G using Graphviz. Parameters ---------- G : NetworkX graph A graph created with NetworkX prog : string Name of Graphviz layout program root : string, optional Root node for twopi layout args : string, optional Extra arguments to Graphviz layout program Returns : dictionary Dictionary of x, y, positions keyed by node. Examples -------- >>> G = nx.petersen_graph() >>> pos = nx.nx_agraph.graphviz_layout(G) >>> pos = nx.nx_agraph.graphviz_layout(G, prog='dot') Notes ----- This is a wrapper for pygraphviz_layout. tprogtroottargs(R(R>RCRDRE((s9lib/python2.7/site-packages/networkx/drawing/nx_agraph.pyR×sc Csúyddl}Wn tk r2tddƒ‚nX|dk rP|d|7}nt|ƒ}|jd|d|ƒi}x{|D]s}|j||ƒ}y<|jdjd ƒ\} } t| ƒt| ƒf||>> G = nx.petersen_graph() >>> pos = nx.nx_agraph.graphviz_layout(G) >>> pos = nx.nx_agraph.graphviz_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_agraph.pygraphviz_layout(G, prog='dot') G_layout = {H.nodes[n]['node_label']: p for n, p in H_layout.items()} iÿÿÿÿNsrequires pygraphviz shttp://pygraphviz.github.io/s -Groot=%sRCREtpost,sno position for nodeg(gg( R0R1RRtlayouttNodeRtsplittfloat( R>RCRDRER0R%tnode_posR(R txxtyy((s9lib/python2.7/site-packages/networkx/drawing/nx_agraph.pyRös&$       isw+btdotcs’tˆƒstjdƒ‚nddl}dddg}x-|D]%}|ˆjkr@iˆj|(s9lib/python2.7/site-packages/networkx/drawing/nx_agraph.pyt update_attrsqs cs=x|D]}ˆj||=qWˆj|s9ˆj|=ndS(N(R (RZR\R(R>(s9lib/python2.7/site-packages/networkx/drawing/nx_agraph.pyt clean_attrsys  t__call__cs djdt|ˆƒdgƒS(NRBs (tjoinR(R.(t edgelabel(s9lib/python2.7/site-packages/networkx/drawing/nx_agraph.pytfuncsR/R.tlabeltpngs_%s.%ss.%stsuffixtdeleteR?RCRE(tlenRtNetworkXExceptionR0R RRthasattrR!RR5R7RRttempfiletNamedTemporaryFiletFalsetdisplay_pygraphvizR(R>RaRCREReR?R0R[Rt edge_attrst node_attrst graph_attrsR]R^R%RbR+R R R.Rtext((R>Ras9lib/python2.7/site-packages/networkx/drawing/nx_agraph.pyR/sT(               + "   cCs{|dkr8|j}tjj|ƒdjƒd}n|sGd}n|j||||ƒ|jƒtj j |ƒdS(sýInternal function to display a graph in OS dependent manner. Parameters ---------- graph : PyGraphviz graph A PyGraphviz AGraph instance. path : file object An already opened file object that will be closed. format : str, None An attempt is made to guess the output format based on the extension of the filename. If that fails, the value of `format` is used. prog : string Name of Graphviz layout program. args : str Additional arguments to pass to the Graphviz layout program. Notes ----- If this function is called in succession too quickly, sometimes the image is not displayed. So you might consider time.sleep(.5) between calls if you experience problems. iN( RRtosR?tsplitexttlowertdrawtcloseRtutilstdefault_opener(R R?tformatRCREtfilename((s9lib/python2.7/site-packages/networkx/drawing/nx_agraph.pyRm°s  #  cCs:ddlm}yddl}Wn|dƒ‚nXdS(Niÿÿÿÿ(tSkipTestspygraphviz not available(tnoseR{R0(tmoduleR{R0((s9lib/python2.7/site-packages/networkx/drawing/nx_agraph.pyt setup_module×s (t__doc__RrRjtnetworkxRt__all__RRRRRRRRwt open_fileRRmR~(((s9lib/python2.7/site-packages/networkx/drawing/nx_agraph.pyts&     K A  9 '