ó ú£Õ\c@sîdZddlZddlZddlZddlZdZdddddd d gZdd „Z d „Z d „Z d„Z dddd„Z dddd„Zejjddƒdddddd„ƒZdddd„Zd„ZdS(s÷ *************** Graphviz AGraph *************** Interface to pygraphviz AGraph class. Examples -------- >>> G=nx.complete_graph(5) >>> A=nx.to_agraph(G) >>> H=nx.from_agraph(A) See Also -------- Pygraphviz: http://networkx.lanl.gov/pygraphviz iÿÿÿÿNsAric Hagberg (hagberg@lanl.gov)t from_agrapht to_agrapht write_dottread_dottgraphviz_layouttpygraphviz_layouttview_pygraphvizc Cs÷|d krl|jƒrB|jƒr3tjƒ}qitjƒ}ql|jƒr]tjƒ}qltjƒ}ntjd|ƒ}d|_ |j d k r¥|j |_ nxI|j ƒD];}t d„|j j ƒDƒƒ}|jt|ƒ|q²Wx½|jƒD]¯}t|dƒt|dƒ}}t |j ƒ}t d„|j ƒDƒƒ}|jƒs‘|j d k r{|j |d>> K5=nx.complete_graph(5) >>> A=nx.to_agraph(K5) >>> G=nx.from_agraph(A) >>> G=nx.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. itcss'|]\}}t|ƒ|fVqdS(N(tstr(t.0tktv((sR/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_agraph.pys Wsicss'|]\}}t|ƒ|fVqdS(N(R(R R R ((sR/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_agraph.pys ^stkeytgraphtnodetedgeN(tNonet is_directedt is_stricttnxtDiGrapht MultiDiGraphtGrapht MultiGrapht empty_graphtnametnodestdicttattrtitemstadd_nodeRtedgest is_multigraphtadd_edget graph_attrR t node_attrt edge_attr( tAt create_usingtNtntstr_attrtetuR R((sR/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_agraph.pyR#s8     !  c Csäyddl}Wn#tk r5tdddƒ‚nX|jƒ}|jƒdko^|jƒ }|jd|jd|d |ƒ}|jj|j j d iƒƒ|j j|j j d iƒƒ|j j|j j d iƒƒx0|j d tƒD]\}}|j||qòW|jƒr‹x¿|jd tdtƒD]M\}}} } td„| jƒDƒƒ} |j||dt| ƒ| q7WnUxR|jd tƒD]>\}}} td„| jƒDƒƒ} |j||| 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.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 s$http://networkx.lanl.gov/pygraphviz s(not available for Python3)iRtstricttdirectedR RRtdatatkeyscss'|]\}}|t|ƒfVqdS(N(R(R R R ((sR/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_agraph.pys ˜sR css'|]\}}|t|ƒfVqdS(N(R(R R R ((sR/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_agraph.pys œs(t pygraphvizt ImportErrorRtnumber_of_selfloopsR tAGraphRR"tupdateR tgetR#R$RtTrueRt edges_iterRRR!R( R'R0R-R,R%R(tnodedataR+R R tedgedatat str_edgedata((sR/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_agraph.pyRms,   ! +&"cCs]yddl}Wn#tk r5tdddƒ‚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 s$http://networkx.lanl.gov/pygraphviz s(not available for Python3)(R0R1Rtwritetclear(tGtpathR0R%((sR/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_agraph.pyR¢s      cCsRyddl}Wn#tk r5tdddƒ‚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 s$http://networkx.lanl.gov/pygraphviz s(not available for Python3)tfile(R0R1R3R(R>R0R%((sR/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_agraph.pyR·s  tneatoRcCst|d|d|d|ƒS(scCreate 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.graphviz_layout(G) >>> pos=nx.graphviz_layout(G,prog='dot') Notes ----- This is a wrapper for pygraphviz_layout. tprogtroottargs(R(R=RARBRC((sR/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_agraph.pyRÉsc Csýyddl}Wn#tk r5tdddƒ‚nX|dk rS|d|7}nt|ƒ}|jd|d|ƒi}x{|D]s}|j||ƒ}y<|jd jd ƒ\} } t| ƒt| ƒf||>> G=nx.petersen_graph() >>> pos=nx.graphviz_layout(G) >>> pos=nx.graphviz_layout(G,prog='dot') iÿÿÿÿNsrequires pygraphviz s$http://networkx.lanl.gov/pygraphviz s(not available for Python3)s -Groot=%sRARCtpost,sno position for nodeg(gg( R0R1RRtlayouttNodeRtsplittfloat( R=RARBRCR0R%tnode_posR(Rtxxtyy((sR/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_agraph.pyRçs(       itwtdotcs’tˆƒstjdƒ‚nddl}dddg}x-|D]%}|ˆjkr@iˆj|RARC(tlenRtNetworkXExceptionR0R RRthasattrR R7R6tget_edgeRRttempfiletNamedTemporaryFiletFalsetdisplay_pygraphvizR(R=R`RARCRdR>R0RZRt edge_attrst node_attrst graph_attrsR\R]R%RaR+R R R.Rtext((R=R`sR/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_agraph.pyRsT(               + "   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>tformatRARCtfilename((sR/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_agraph.pyRm”s  #  cCs:ddlm}yddl}Wn|dƒ‚nXdS(Niÿÿÿÿ(tSkipTestspygraphviz not available(tnoseR{R0(tmoduleR{R0((sR/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_agraph.pyt setup_moduleºs (t__doc__RrtsysRjtnetworkxRt __author__t__all__RRRRRRRRwt open_fileRRmR~(((sR/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_agraph.pyts*      J 5  - ~&