\c @ssdZddlZddlmZmZmZmZmZdZ ddddd d d d d dddg Z ddddZ de dZdddddddddddd Zddddddddde dd ZddddddddZddd dddddde d! Zd"Zd#Zd$Zd%Zd&Zd'd(Zd)Zd*ZdS(+s ********** Matplotlib ********** Draw networks with matplotlib (pylab). See Also -------- matplotlib: http://matplotlib.sourceforge.net/ pygraphviz: http://networkx.lanl.gov/pygraphviz/ iN(t shell_layouttcircular_layouttspectral_layoutt spring_layoutt random_layoutsAric Hagberg (hagberg@lanl.gov)tdrawt draw_networkxtdraw_networkx_nodestdraw_networkx_edgestdraw_networkx_labelstdraw_networkx_edge_labelst draw_circulart draw_randomt draw_spectralt draw_springt draw_shellt draw_graphvizc Ks9yddlj}Wn5tk r2tdntk rJdGHnX|j}|jd|dkr|jdkr|jd }q|j }n|j }|j dd}|dk r|j |ny1t |d |d |||j|jWn|j |nX|j |dS( sDraw the graph G with Matplotlib (pylab). Draw the graph as a simple representation with no node labels or edge labels and using the full Matplotlib figure area and no axis labels by default. See draw_networkx() for more full-featured drawing that allows title, axis labels etc. Parameters ---------- G : graph A networkx graph pos : dictionary, optional A dictionary with nodes as keys and positions as values. If not specified a spring layout positioning will be computed. See networkx.layout for functions that compute node positions. ax : Matplotlib Axes object, optional Draw the graph in specified Matplotlib axes. hold : bool, optional Set the Matplotlib hold state. If True subsequent draw commands will be added to the current axes. **kwds : optional keywords See networkx.draw_networkx() for a description of optional keywords. Examples -------- >>> G=nx.dodecahedral_graph() >>> nx.draw(G) >>> nx.draw(G,pos=nx.spring_layout(G)) # use spring layout See Also -------- draw_networkx() draw_networkx_nodes() draw_networkx_edges() draw_networkx_labels() draw_networkx_edge_labels() Notes ----- This function has the same name as pylab.draw and pyplot.draw so beware when using >>> from networkx import * since you might overwrite the pylab.draw function. Good alternatives are: With pylab: >>> import pylab as P # >>> import networkx as nx >>> G=nx.dodecahedral_graph() >>> nx.draw(G) # networkx draw() >>> P.draw() # pylab draw() With pyplot >>> import matplotlib.pyplot as plt >>> import networkx as nx >>> G=nx.dodecahedral_graph() >>> nx.draw(G) # networkx draw() >>> plt.draw() # pyplot draw() Also see the NetworkX drawing examples at http://networkx.lanl.gov/gallery.html iNsMatplotlib required for draw()s!Matplotlib unable to open displaytwiitholdtpostax(iiii(tmatplotlib.pylabtpylabt ImportErrort RuntimeErrortgcft set_facecolortNonet_axstacktadd_axestgcatisholdtpopRRt set_axis_offtdraw_if_interactive( tGRRRtkwdsRtcftbth((sQ/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_pylab.pyR's4H          cKsyddlj}Wn5tk r2tdntk rJdGHnX|dkrltjj|}nt|||}t |||}|rt |||n|j dS(s Draw the graph G using Matplotlib. Draw the graph with Matplotlib with options for node positions, labeling, titles, and many other drawing features. See draw() for simple drawing without labels or axes. Parameters ---------- G : graph A networkx graph pos : dictionary, optional A dictionary with nodes as keys and positions as values. If not specified a spring layout positioning will be computed. See networkx.layout for functions that compute node positions. with_labels : bool, optional (default=True) Set to True to draw labels on the nodes. ax : Matplotlib Axes object, optional Draw the graph in the specified Matplotlib axes. nodelist : list, optional (default G.nodes()) Draw only specified nodes edgelist : list, optional (default=G.edges()) Draw only specified edges node_size : scalar or array, optional (default=300) Size of nodes. If an array is specified it must be the same length as nodelist. node_color : color string, or array of floats, (default='r') Node color. Can be a single color format string, or a sequence of colors with the same length as nodelist. If numeric values are specified they will be mapped to colors using the cmap and vmin,vmax parameters. See matplotlib.scatter for more details. node_shape : string, optional (default='o') The shape of the node. Specification is as matplotlib.scatter marker, one of 'so^>v>> G=nx.dodecahedral_graph() >>> nx.draw(G) >>> nx.draw(G,pos=nx.spring_layout(G)) # use spring layout >>> import pylab >>> limits=pylab.axis('off') # turn of axis Also see the NetworkX drawing examples at http://networkx.lanl.gov/gallery.html See Also -------- draw() draw_networkx_nodes() draw_networkx_edges() draw_networkx_labels() draw_networkx_edge_labels() iNsMatplotlib required for draw()s!Matplotlib unable to open display( RRRRRtnxtdrawingRRRR R"(R#Rt with_labelsR$Rtnode_collectiontedge_collection((sQ/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_pylab.pyRsp   i,trtog?c Ksyddlj}ddl}Wn5tk r>tdntk rVdGHnX| dkrr|j} n|dkr|j}n| st|dkrdSy*|j g|D]}||^q}WnEt k r}t j d|n t k rt j dnX| j|dddf|dddfd |d |d |d |d |d| d|d| d|  }|j||jd|S(sDraw the nodes of the graph G. This draws only the nodes of the graph G. Parameters ---------- G : graph A networkx graph pos : dictionary A dictionary with nodes as keys and positions as values. If not specified a spring layout positioning will be computed. See networkx.layout for functions that compute node positions. ax : Matplotlib Axes object, optional Draw the graph in the specified Matplotlib axes. nodelist : list, optional Draw only specified nodes (default G.nodes()) node_size : scalar or array Size of nodes (default=300). If an array is specified it must be the same length as nodelist. node_color : color string, or array of floats Node color. Can be a single color format string (default='r'), or a sequence of colors with the same length as nodelist. If numeric values are specified they will be mapped to colors using the cmap and vmin,vmax parameters. See matplotlib.scatter for more details. node_shape : string The shape of the node. Specification is as matplotlib.scatter marker, one of 'so^>v>> G=nx.dodecahedral_graph() >>> nodes=nx.draw_networkx_nodes(G,pos=nx.spring_layout(G)) Also see the NetworkX drawing examples at http://networkx.lanl.gov/gallery.html See Also -------- draw() draw_networkx() draw_networkx_edges() draw_networkx_labels() draw_networkx_edge_labels() iNsMatplotlib required for draw()s!Matplotlib unable to open displayisNode %s has no position.sBad value in node positions.itstctmarkertcmaptvmintvmaxtalphat linewidthstlabeli(RRtnumpyRRRRtnodestlentasarraytKeyErrorR(t NetworkXErrort ValueErrortscattertscit set_zorder(R#Rtnodelistt node_sizet node_colort node_shapeR5R2R3R4RR6R7R$RR8tvtxyteR+((sQ/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_pylab.pyRsBO    * /   tktsolidc 5 Ksy`ddl}ddlj}ddlj}ddlm}m}ddlm }ddl }Wn5t k rt dnt k rdGHnX| dkr|j} n|dkr|j}n| st|dkrdS|jg|D]"}||d||df^q}|j|s>|f}n|}|j| rd|j|rdt|t|krd|jg|D]}|j|^qrtg|D]}|j||^q}q|jg|D]}|j| ^qrU|jg|D]'}|j|o1t|dk^q rLt|}qad}qtd nE|j|st|dkr|j||f}n td ||d |d|ddd|d| j}|jd|j| | j||j|r|j|n|dkr|dk rLt||sLtn|j |j||j!||dk s| dk r|j"|| n |j#|j$|nd}|j%ri| ri|}g}d}x |D]\}} |\}!}"| \}#}$|#|!}%|$|"}&|j&t'|%d|&d}'|'dkrPqn|%dkrs|#}(|&||"})n|&dkr|$})|%||!}(nH|j(|&|%}*||'|j)|*|!}(||'|j*|*|"})|j+|(|)f|#|$ffqW||d |dg|D]}+d |+^qddd| j}|jd|j| | j|n|j,|j-|dddddf},|j.|j-|dddddf}-|j,|j-|dddddf}.|j.|j-|dddddf}/|-|,}0|/|.}1d|0d|1}2}3|,|2|.|3f|-|2|/|3ff}4| j/|4| j0|S(sDraw the edges of the graph G. This draws only the edges of the graph G. Parameters ---------- G : graph A networkx graph pos : dictionary A dictionary with nodes as keys and positions as values. If not specified a spring layout positioning will be computed. See networkx.layout for functions that compute node positions. edgelist : collection of edge tuples Draw only specified edges(default=G.edges()) width : float Line width of edges (default =1.0) edge_color : color string, or array of floats Edge color. Can be a single color format string (default='r'), or a sequence of colors with the same length as edgelist. If numeric values are specified they will be mapped to colors using the edge_cmap and edge_vmin,edge_vmax parameters. style : string Edge line style (default='solid') (solid|dashed|dotted,dashdot) alpha : float The edge transparency (default=1.0) edge_ cmap : Matplotlib colormap Colormap for mapping intensities of edges (default=None) edge_vmin,edge_vmax : floats Minimum and maximum for edge colormap scaling (default=None) ax : Matplotlib Axes object, optional Draw the graph in the specified Matplotlib axes. arrows : bool, optional (default=True) For directed graphs, if True draw arrowheads. label : [None| string] Label for legend Notes ----- For directed graphs, "arrows" (actually just thicker stubs) are drawn at the head end. Arrows can be turned off with keyword arrows=False. Yes, it is ugly but drawing proper arrows with Matplotlib this way is tricky. Examples -------- >>> G=nx.dodecahedral_graph() >>> edges=nx.draw_networkx_edges(G,pos=nx.spring_layout(G)) Also see the NetworkX drawing examples at http://networkx.lanl.gov/gallery.html See Also -------- draw() draw_networkx() draw_networkx_nodes() draw_networkx_labels() draw_networkx_edge_labels() iN(tcolorConvertertColormap(tLineCollectionsMatplotlib required for draw()s!Matplotlib unable to open displayiiiis8edge_color must consist of either color names or numberss\edge_color must be a single color or list of exactly m colors where m is the number or edgestcolorsR6t antialiasedst linestylet transOffsetg?g?ig?(ii(ig?(i(1t matplotlibRRtmatplotlib.cbooktcbooktmatplotlib.colorsRKRLtmatplotlib.collectionsRMR8RRRRtedgesR:R;titerabletis_string_liketalltruettupletto_rgbaR>t transDataRAt set_labeltadd_collectiont is_numliket set_alphat isinstancetAssertionErrort set_arraytset_cmaptset_climt autoscaleR@t is_directedtsqrttfloattarctan2tcostsintappendtamintraveltamaxtupdate_datalimtautoscale_view(5R#Rtedgelisttwidtht edge_colortstyleR5t edge_cmapt edge_vmint edge_vmaxRtarrowsR7R$RRRtcbRKRLRMR8RHtedge_postlwR0t edge_colorsR,tarrow_collectiont arrow_colorsta_postptsrctdsttx1ty1tx2ty2tdxtdytdtxatyatthetatwwtminxtmaxxtminytmaxyRR'tpadxtpadytcorners((sQ/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_pylab.pyRsS     8  "( # 4 !              !   #    1111  (  i s sans-seriftnormalc Ksky"ddlj} ddlj} Wn5tk rAtdntk rYdGHnX|dkru| j}n|dkrtd|j D}n| j dd} | j dd} i}x|j D]\}}||\}}| j |st |}n|j|||d |d |d |d |d| d| d |jdt}|||>> G=nx.dodecahedral_graph() >>> labels=nx.draw_networkx_labels(G,pos=nx.spring_layout(G)) Also see the NetworkX drawing examples at http://networkx.lanl.gov/gallery.html See Also -------- draw() draw_networkx() draw_networkx_nodes() draw_networkx_edges() draw_networkx_edge_labels() iNsMatplotlib required for draw()s!Matplotlib unable to open displaycss|]}||fVqdS(N((t.0tn((sQ/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_pylab.pys sthorizontalalignmenttcentertverticalalignmenttsizetcolortfamilytweightt transformtclip_on(RRRSRTRRRRtdictR9tgettitemsRYtstrttextR]tTrue(R#Rtlabelst font_sizet font_colort font_familyt font_weightR5RR$RR|RRt text_itemsRR7txtytt((sQ/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_pylab.pyR ts>9       g?i c !Ksy.ddlj} ddlj}ddl}Wn5tk rMtdntk redGHnX| dkr| j} n|dkrt d|j dt D}n|}i}x|j D]\\}}}||\}}||\}}|||d||||d|}}| r|j ||||d|jd }|d krr|d 8}n|d kr|d 7}n|j||f}| jj|j|f|jd!d}nd}| dkrt dddd"dd#} n|j|st|}n| jdd}| jdd}| j|||d|d|d|d|d|d|d|d| jd| dd d t  } | |||f>> G=nx.dodecahedral_graph() >>> edge_labels=nx.draw_networkx_edge_labels(G,pos=nx.spring_layout(G)) Also see the NetworkX drawing examples at http://networkx.lanl.gov/gallery.html See Also -------- draw() draw_networkx() draw_networkx_nodes() draw_networkx_edges() draw_networkx_labels() iNsMatplotlib required for draw()s!Matplotlib unable to open displaycss*|] \}}}||f|fVqdS(N((RtuRFR((sQ/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_pylab.pys )stdatag?g@ihiZiiiiigtboxstyletroundtectfcRRRRRRRtrotationRtbboxtzorderR(ii(g?g?g?(g?g?g?(RRRSRTR8RRRRRRWRRRktpitarrayR]ttransform_anglestreshapeRYRRR(!R#Rt edge_labelst label_posRRRRR5RRtrotateR$RR|R8RRtn1tn2R7RRRRRRtangleRGt trans_angleRRR((sQ/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_pylab.pyR sjH    %)          cKst|t||dS(s(Draw the graph G with a circular layout.N(RR(R#tkwargs((sQ/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_pylab.pyR _scKst|t||dS(s&Draw the graph G with a random layout.N(RR(R#R((sQ/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_pylab.pyR cscKst|t||dS(s(Draw the graph G with a spectral layout.N(RR(R#R((sQ/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_pylab.pyR gscKst|t||dS(s&Draw the graph G with a spring layout.N(RR(R#R((sQ/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_pylab.pyRkscKsH|jdd}|dkr(|d=nt|t|d||dS(s&Draw networkx graph with shell layout.tnlistN(RRRR(R#RR((sQ/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_pylab.pyRos  tneatocKs)tjj||}t|||dS(s)Draw networkx graph with graphviz layout.N(R(R)tgraphviz_layoutR(R#tprogRR((sQ/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_pylab.pyRvscKst|||dS(s6For backward compatibility; use draw or draw_networkx.N(R(R#RR$((sQ/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_pylab.pytdraw_nx{scCsYddlm}y/ddl}|jddtddl}Wn|dnXdS(Ni(tSkipTesttPStwarnsmatplotlib not available(tnoseRRRtusetFalseR(tmoduleRtmplR((sQ/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_pylab.pyt setup_modules (t__doc__tnetworkxR(tnetworkx.drawing.layoutRRRRRt __author__t__all__RRRRRRR R R R R RRRRR(((sQ/Users/dxp/prism/prism-games/prism-examples/smgs/car/networkx/drawing/nx_pylab.pyts ( h r  Z