ó |£*^c@sùdZddlZddlmZmZddlZdddddd gZyeWne k rme ZnXed d d ƒd „ƒZ edd dƒd„ƒZ d„Z ed„Zd„Zdd„Zddd„Zddd„Zd„ZdS(sm ***** Pydot ***** Import and export NetworkX graphs in Graphviz dot format using pydotplus. Either this module or nx_agraph can be used to interface with graphviz. See Also -------- PyDotPlus: https://github.com/carlos-jenkins/pydotplus Graphviz: http://www.research.att.com/sw/tools/graphviz/ DOT Language: http://www.graphviz.org/doc/info/lang.html iÿÿÿÿN(t open_filetmake_strt write_dottread_dottgraphviz_layoutt pydot_layouttto_pydott from_pydotitmodetwcCs#t|ƒ}|j|jƒƒdS(sgWrite NetworkX graph G to Graphviz dot format on path. Path can be a string or a file handle. N(Rtwritet to_string(tGtpathtP((sm/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/networkx/drawing/nx_pydot.pyR%s itrcCs1ddl}|jƒ}|j|ƒ}t|ƒS(sLReturn a NetworkX MultiGraph or MultiDiGraph from a dot file on path. Parameters ---------- path : filename or file handle Returns ------- G : NetworkX multigraph A MultiGraph or MultiDiGraph. Notes ----- Use G = nx.Graph(read_dot(path)) to return a Graph instead of a MultiGraph. iÿÿÿÿN(t pydotplustreadtgraph_from_dot_dataR(R RtdataR((sm/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/networkx/drawing/nx_pydot.pyR/s  cCsˆ|jdƒrt}nt}|jƒdkrT|rEtjƒ}qutjƒ}n!|ritjƒ}n tj ƒ}|j ƒj dƒ}|dkr¢||_ nxQ|j ƒD]C}|j ƒj dƒ}|d krÜq¯n|j||jƒq¯Wx|jƒD]}|jƒ}|jƒ}|jƒ} g} g} t|tƒra| j|j dƒƒn+x(|dD]} | j| j dƒƒqlWt|tƒr´| j|j dƒƒn+x(|dD]} | j| j dƒƒq¿Wx2| D]*}x!| D]}|j||| qóWqæWqW|jƒ}|r:||jd>> 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)) tgrapht"ttnodetedgetnodesiN(RRR(t get_stricttNonetFalsetTruetget_typetnxt MultiGraphtGrapht MultiDiGraphtDiGraphtget_nametstriptnamet get_node_listtadd_nodetget_attributest get_edge_listt get_sourcetget_destinationt isinstancet basestringtappendtadd_edgeRtget_node_defaultstget_edge_defaults(Rt multiedgestNR&tptntetutvtattrtstdtunodestvnodest source_nodetdestination_nodetpattr((sm/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/networkx/drawing/nx_pydot.pyREs^           cCswddl}|jƒr!d}nd}|jƒdkoC|jƒ }|j}|jjdiƒ}|dkr‘|jdd|d||}n"|jd |d|d||}y|j|jd Wnt k rÞnXy|j |jd Wnt k r nXxa|j d t ƒD]M\}}t d „|jƒDƒƒ} |jt|ƒ| } |j| ƒqW|jƒrxõ|jd t dt ƒD]h\} } } }t d„|jƒDƒƒ}|jt| ƒt| ƒdt| ƒ|}|j|ƒq”Wnpxm|jd t ƒD]Y\} } }t d„|jƒDƒƒ}|jt| ƒt| ƒ|}|j|ƒqW|S(sReturn 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 ----- iÿÿÿÿNtdigraphRiRt graph_typetstricts"%s"RRRcss'|]\}}|t|ƒfVqdS(N(R(t.0tkR9((sm/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/networkx/drawing/nx_pydot.pys Éstkeyscss'|]\}}|t|ƒfVqdS(N(R(RERFR9((sm/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/networkx/drawing/nx_pydot.pys Ïstkeycss'|]\}}|t|ƒfVqdS(N(R(RERFR9((sm/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/networkx/drawing/nx_pydot.pys Ös(Rt is_directedtnumber_of_selfloopst is_multigraphR&RtgettDottset_node_defaultstKeyErrortset_edge_defaultst nodes_iterRtdicttitemstNodeRR(t edges_itertEdgeR0(R4RDRRCR&tgraph_defaultsRR6tnodedatat str_nodedataR5R8R9RHtedgedatat str_edgedataR((sm/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/networkx/drawing/nx_pydot.pyRžsF          +"!cCs'ddlm}|dtƒt|ƒS(s+Create a Pydot graph from a NetworkX graph.iÿÿÿÿ(twarns+pydot_from_networkx is replaced by to_pydot(twarningsR\tDeprecationWarningR(R4R\((sm/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/networkx/drawing/nx_pydot.pytpydot_from_networkxÜs cCs'ddlm}|dtƒt|ƒS(s+Create a NetworkX graph from a Pydot graph.iÿÿÿÿ(R\s-networkx_from_pydot is replaced by from_pydot(R]R\R^R(tDt create_usingR\((sm/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/networkx/drawing/nx_pydot.pytnetworkx_from_pydotâstneatocKstd|d|d||S(sICreate node positions using Pydot and Graphviz. Returns a dictionary of 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. R tprogtroot(R(R RdRetkwds((sm/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/networkx/drawing/nx_pydot.pyRéscKsNddl}t|ƒ}|dk r=|jdt|ƒƒn|jd|ƒ}|dkr…d|GHdGHdGHdGHd GHd |GHdS|j|ƒ}i}x­|jƒD]Ÿ} |jt| ƒƒj ƒ} |j | ƒ} t | t ƒró| d } n| j ƒd d!} | dkr§| jd ƒ\} }t| ƒt|ƒf|| >> G = nx.complete_graph(4) >>> pos = nx.nx_pydot.pydot_layout(G) >>> pos = nx.nx_pydot.pydot_layout(G, prog='dot') iÿÿÿÿNReRdRsGraphviz layout with %s failedsTo debug what happened try:sP=pydot_from_networkx(G)sP.write_dot("file.dot")sAnd then run %s on file.dotiit,((RRRtsetRt create_dotRRRTR$tget_nodeR-tlisttget_postsplittfloat(R RdReRfRRR`tQtnode_posR6t pydot_nodeRtpostxxtyy((sm/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/networkx/drawing/nx_pydot.pyRûs2        #cCsDddlm}yddl}Wntk r?|dƒ‚nXdS(Niÿÿÿÿ(tSkipTestspydotplus not available(tnoseRuRt ImportError(tmoduleRuR((sm/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/networkx/drawing/nx_pydot.pyt setup_module&s  (t__doc__t importlibtnetworkx.utilsRRtnetworkxRt__all__R.t NameErrortstrRRRRRR_RRbRRRy(((sm/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/networkx/drawing/nx_pydot.pyts$        Y >  +