z iӚSrSSKJr SSKJrJr SSKJr SSKr SSK r S/r SSjr SSjr SS jrSS jrSS jrSS jrSS jrg)z'Draw a rustworkx graph with matplotlib.)Iterable)islicecycle)NumberNmpl_drawc SSKJn UcUR5nOUR 5nUR S5 Uc3UR (aUR5nOURS5n[U4XX4S.UD6 UR5 UR5 UR5(aUcU$g![an[S5UeSnAff=f)aDraw a graph with Matplotlib. .. note:: Matplotlib is an optional dependency and will not be installed with rustworkx by default. If you intend to use this function make sure that you install matplotlib with either ``pip install matplotlib`` or ``pip install 'rustworkx[mpl]'`` :param graph: A rustworkx graph, either a :class:`~rustworkx.PyGraph` or a :class:`~rustworkx.PyDiGraph`. :param dict pos: An optional dictionary (or a :class:`~rustworkx.Pos2DMapping` object) with nodes as keys and positions as values. If not specified a spring layout positioning will be computed. See `layout_functions` for functions that compute node positions. :param matplotlib.Axes ax: An optional Matplotlib Axes object to draw the graph in. :param bool arrows: For :class:`~rustworkx.PyDiGraph` objects if ``True`` draw arrowheads. (defaults to ``True``) Note, that the Arrows will be the same color as edges. :param str arrowstyle: An optional string for directed graphs to choose the style of the arrowsheads. See :class:`matplotlib.patches.ArrowStyle` for more options. By default the value is set to ``'-\|>'``. :param int arrow_size: For directed graphs, choose the size of the arrow head's length and width. See :class:`matplotlib.patches.FancyArrowPatch` attribute and constructor kwarg ``mutation_scale`` for more info. Defaults to 10. :param bool with_labels: Set to ``True`` to draw labels on the nodes. Edge labels will only be drawn if the ``edge_labels`` parameter is set to a function. Defaults to ``False``. :param list node_list: An optional list of node indices in the graph to draw. If not specified all nodes will be drawn. :param list edge_list: An option list of edges in the graph to draw. If not specified all edges will be drawn :param int|list node_size: Optional size of nodes. If an array is specified it must be the same length as node_list. Defaults to 300 :param node_color: Optional node color. Can be a single color or a sequence of colors with the same length as node_list. Color can be string or rgb (or rgba) tuple of floats from 0-1. If numeric values are specified they will be mapped to colors using the ``cmap`` and ``vmin``,``vmax`` parameters. See :func:`matplotlib.scatter` for more details. Defaults to ``'#1f78b4'``) :param str node_shape: The optional shape node. The specification is the same as the :func:`matplotlib.pyplot.scatter` function's ``marker`` kwarg, valid options are one of ``['s', 'o', '^', '>', 'v', '<', 'd', 'p', 'h', '8']``. Defaults to ``'o'`` :param float alpha: Optional value for node and edge transparency :param matplotlib.colors.Colormap cmap: An optional Matplotlib colormap object for mapping intensities of nodes :param float vmin: Optional minimum value for node colormap scaling :param float vmax: Optional minimum value for node colormap scaling :param float|sequence linewidths: An optional line width for symbol borders. If a sequence is specified it must be the same length as node_list. Defaults to 1.0 :param float|sequence width: An optional width to use for edges. Can either be a float or sequence of floats. If a sequence is specified it must be the same length as node_list. Defaults to 1.0 :param str|sequence edge_color: color or array of colors (default='k') Edge color. Can be a single color or a sequence of colors with the same length as edge_list. Color can be string or rgb (or rgba) tuple of floats from 0-1. If numeric values are specified they will be mapped to colors using the ``edge_cmap`` and ``edge_vmin``, ``edge_vmax`` parameters. :param matplotlib.colors.Colormap edge_cmap: An optional Matplotlib colormap for mapping intensities of edges. :param float edge_vmin: Optional minimum value for edge colormap scaling :param float edge_vmax: Optional maximum value for node colormap scaling :param str style: An optional string to specify the edge line style. For example, ``'-'``, ``'--'``, ``'-.'``, ``':'`` or words like ``'solid'`` or ``'dashed'``. See the :class:`matplotlib.patches.FancyArrowPatch` attribute and kwarg ``linestyle`` for more details. Defaults to ``'solid'``. :param func labels: An optional callback function that will be passed a node payload and return a string label for the node. For example:: labels=str could be used to just return a string cast of the node's data payload. Or something like:: labels=lambda node: node['label'] could be used if the node payloads are dictionaries. :param func edge_labels: An optional callback function that will be passed an edge payload and return a string label for the edge. For example:: edge_labels=str could be used to just return a string cast of the edge's data payload. Or something like:: edge_labels=lambda edge: edge['label'] could be used if the edge payloads are dictionaries. If this is set edge labels will be drawn in the visualization. :param int font_size: An optional fontsize to use for text labels, By default a value of 12 is used for nodes and 10 for edges. :param str font_color: An optional font color for strings. By default ``'k'`` (ie black) is set. :param str font_weight: An optional string used to specify the font weight. By default a value of ``'normal'`` is used. :param str font_family: An optional font family to use for strings. By default ``'sans-serif'`` is used. :param str label: An optional string label to use for the graph legend. :param str connectionstyle: An optional value used to create a curved arc of rounding radius rad. For example, ``connectionstyle='arc3,rad=0.2'``. See :class:`matplotlib.patches.ConnectionStyle` and :class:`matplotlib.patches.FancyArrowPatch` for more info. By default this is set to ``"arc3"``. :returns: A matplotlib figure for the visualization if not running with an interactive backend (like in jupyter) or if ``ax`` is not set. :rtype: matplotlib.figure.Figure For Example: .. jupyter-execute:: import matplotlib.pyplot as plt import rustworkx as rx from rustworkx.visualization import mpl_draw G = rx.generators.directed_path_graph(25) mpl_draw(G) plt.draw() rNmatplotlib needs to be installed prior to running rustworkx.visualization.mpl_draw(). You can install matplotlib with: 'pip install matplotlib'w)rrr )posaxarrows with_labels) matplotlib.pyplotpyplot ImportErrorgcf get_figure set_facecoloraxesgcaadd_axes draw_graph set_axis_offdraw_if_interactive isinteractive) graphr r rrkwdspltecfs \/home/james-whalen/.local/lib/python3.13/site-packages/rustworkx/visualization/matplotlib.pyrrBsH' z WWY ]]_S z 77B\*BuU#VUPTUOO     "* #-)  9   sB== C CCc <SSKJn 1Skn1Skn1Skn 1Skn Xx-U -U -n [UV s/sHoU ;PM sn 5(a7SR UV s/sH oU ;dM U PM sn 5n [ S U 35eUR S S5nU(a,UR5Vs0sH oU"X5_M snUS 'UR S S5nU(a5UR5Vs0sHoSUS 4U"US 5_M snUS 'UR5V Vs0sHun nX;dM U U_M nn nUR5V Vs0sHun nX;dM U U_M nn n[URS5[5(aUS UR5V Vs0sHun nX;dM U U_M nn nUR5V Vs0sHun nX;dM U U_M nn nUc[R"U5n[X40UD6 [!X4SU0UD6 U(a [#X40UD6 U(a [%X40UD6 UR'5 g![an[S5UeSnAff=fs sn fs sn fs snfs snfs snn fs snn fs snn fs snn f)aDraw the graph 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 ---------- graph: A rustworkx :class:`~rustworkx.PyDiGraph` or :class:`~rustworkx.PyGraph` pos : dictionary, optional A dictionary with nodes as keys and positions as values. If not specified a spring layout positioning will be computed. See :mod:`rustworkx.drawing.layout` for functions that compute node positions. Notes ----- For directed graphs, arrows are drawn at the head end. Arrows can be turned off with keyword arrows=False. rNr > r cmapvmaxvminalphalabel node_list node_size edgecolors linewidths node_color node_shape>r r'r(stylewidth edge_cmap edge_list edge_vmax edge_vminr)r* arrow_size arrowstyle edge_colorr.connectionstylemin_source_marginmin_target_margin> r bboxr'labels font_size font_color font_family font_weightverticalalignmenthorizontalalignment> r r;r'rotater=r> edge_labelsr?r@rArBz, zReceived invalid argument(s): r<rDr r'r)rrranyjoin ValueErrorpop node_indicesweighted_edge_listitems isinstancegetlist rustworkx spring_layout draw_nodes draw_edges draw_labelsdraw_edge_labelsr)rr rrrrr valid_node_kwdsvalid_edge_kwdsvalid_label_kwdsvalid_edge_label_kwds valid_kwdsk invalid_argslabel_fnx edge_label_fnv node_kwds edge_kwds label_kwdsedge_label_kwdss r"rrs2' OO*  !25EEH]]J  .AZ  .//yyT!ITj5H!T!IJ 9,HIIxx$'H9>9K9K9MN9MAXeh//9MNXHH]D1M7<7O7O7Q 7Q!qT1Q4L-!- -7Q ]#'**,G,$!Q!2FA,IG"&**,G,$!Q!2FA,IG)--($// g #'::<I<41a13H$!Q$v ZZ3AQ3 4%"" ITP jj 1a4 1a4     ! ONN   q! e  9   $4 >5#456A=>sAC>!D!5DD!> D DDD!! E+D;;Ec ^ ^9^:^;^<SSKmRAXF5n[CU5GHun n!U!SSU!SSn#n"U!SSU!SSn%n$Sn&Sn'[$R&"U5(aFUU SS un(n)UUREU(5n*UUREU)5n+U"U*U5n&U"U+U5n'O U"UU5=n&n'U&U:aUn&U'U:aUn'[U5[U5:XaUU n,O&[U5S:XaUSn,OUU [U5-n,[$R&"U5(a0[U5[U5:XaUU n-OUU [U5-n-OUn-[![GU!55U;aS n.OS n.T', otherwise defaults to '-'. See `matplotlib.patches.ArrowStyle` for more options. arrow_size : int (default=10) For directed graphs, choose the size of the arrow head's length and width. See `matplotlib.patches.FancyArrowPatch` for attribute ``mutation_scale`` for more info. node_size : scalar or array (default=300) Size of nodes. Though the nodes are not drawn with this function, the node size is used in determining edge positioning. node_list : list, optional (default=graph.node_indices()) This provides the node order for the `node_size` array (if it is an array). node_shape : string (default='o') The marker used for nodes, used in determining edge positioning. Specification is as a `matplotlib.markers` marker, e.g. one of 'so^>v-r[r )r&r%cxUS;a[R"SU-5S- $[R"U5S- $)Nzs^>v.to_marker_edges6 X 771{?+a/ /77;'!+ +rE?z, rad = ) r6shrinkAshrinkBmutation_scalecolor linewidthr8 linestylezorderc >[R"X:H5(GaT S:XaST -OT nT RR5R U5nSU-nUS-nU[R "SU/5-U[R "Xv/5-U[R "US/5-UU[R "U*S/5-U[R "U*U/5-U[R "SU/5-/nTR RT RR U5/SQ5n U $T "X/UQ70UD6n U $)Nrg{Gzt?g??)r rrrrr)rvall transDatainverted transformrwpathPath)posAposBargskwargs selfloop_htdata_locv_shifth_shiftrretr base_connectionstyleh max_nodesizer~s r"_connectionstyle$draw_edges.._connectionstyles0 66$,  34q&%,.aK||,,.88>HK'GmG2::q'l332::w&8992::wl332::xm442::x&9::2::q'l33 D((-- 6 6t <>STC  'tCDCFCC rg?riFrj).rqmatplotlib.colorsmatplotlib.patchesmatplotlib.pathrrrrMrP PyDiGraphrr2rsrOrJdicttuplekeysrviterablerrcolorsColormapget_cmapminmax NormalizepatchesConnectionStylearraycolorConverter to_rgba_array enumerateindexreversedFancyArrowPatchappend add_patchrwaminravelamaxupdate_datalimautoscale_viewr|)=rr r2r0r7r/r'r6r5r1r4r3r rr(r*r)r.r8r9r:rqrr edge_pos_keysedge_posrg color_normalrarrow_collectionr arrow_colorsiedgex1y1x2y2 shrink_source shrink_targetsourcetargetsource_node_sizetarget_node_size arrow_color line_widthradarrow mirustworkxmaxxminymaxyr rpadxpadycornersrrrr~s= ` @@@@r"rSrSs_l   !' eY00 1 1fJJ z WWYOO%  9~ ++-.  FM >B uS1Y's1Q4y)9:;!!#H J _H - FF:>:aJq&):> ? ?  i)<)<== == I  JI  JIzz+++K :DE*Qi Q0* E ,N;;66G88I&**,L::,,:::MLX&4aT!WQZBaT!WQZB  ;;y ! !&q\"1-NFF()@A ()@A *+;ZHM*+;ZHM,:9j,Q QMM , ,-M , ,-M | H -&q/K  ! #&q/K&q3|+<'<=K ;;u  5zS]*"1X "1s5z>2 J $ H ,CC ++ H H!!!) ./xu=,   & Uq'tzz%/*H''"((8Aq!G#456K 77288HQ1W-. /D 77288HQ1W-. /D 77288HQ1W-. /D {A t AD4!8$DT!4$;/$+td{1KKGgNN      9   P?Fs#U2U: U? U7& U22U7c SSKJn U cU R5n Uc UR 5Vs0sHoU_M nn0nUR 5HWunnXunn[ U[5(d [U5nU RUUUUUUUUU U U RUU S9 nUUU'MY U RSSSSSSS9 U$![an[S5UeSnAff=fs snf)ajDraw node labels on the graph. Parameters ---------- graph: A rustworkx graph pos : dictionary A dictionary with nodes as keys and positions as values. Positions should be sequences of length 2. labels : dictionary (default={n: n for n in graph}) Node labels in a dictionary of text labels keyed by node. Node-keys in labels should appear as keys in `pos`. If needed use: `{n:lab for n,lab in labels.items() if n in pos}` font_size : int (default=12) Font size for text labels font_color : string (default='k' black) Font color string font_weight : string (default='normal') Font weight font_family : string (default='sans-serif') Font family alpha : float or None (default=None) The text transparency bbox : Matplotlib bbox, (default is Matplotlib's ax.text default) Specify text box properties (e.g. shape, color etc.) for node labels. horizontalalignment : string (default='center') Horizontal alignment {'center', 'right', 'left'} verticalalignment : string (default='center') Vertical alignment {'center', 'top', 'bottom', 'baseline', 'center_baseline'} ax : Matplotlib Axes object, optional Draw the graph in the specified Matplotlib axes. clip_on : bool (default=True) Turn on clipping of node labels at axis boundaries Returns ------- dict `dict` of labels keyed on the nodes rNr ) sizerfamilyweightr'rBrArr;clip_onriFrj) rrrrrJrLrMstrtextrr|)rr r<r=r>r?r@r'r;rBrAr rrr n text_itemsr(r^yts r"rTrT#s!D' z WWY ~ % 2 2 45 41Q$ 45JLLN5A%%%JE GG  3/ll   1 '#*NN    Y  9   6sC C C CCcJSSKJn U cUR5n Uc,UR 5VVVs0sH unnnUU4U_M nnnnOUn0nUR 5GHuunnnUUunnUUunnUU-USU- --UU-USU- --nnUU4UR 5;a0[R"UU- 5nUU:a USU--nOUSU-- nU (a[R"UU- UU- 5S[R-- S-n U S:aU S -n U S :aU S - n [R"UU45n!U RR[R"U 45U!RS 55Sn"OS n"U c [S SSS9n [!U["5(d [#U5nU R%UUUUUUUUU U U"U RU SUS9n#U#UUU4'GM U R'SSSSSSS9 U$![an[S5UeSnAff=fs snnnf)aDraw edge labels. Parameters ---------- graph: A rustworkx graph pos : dictionary A dictionary with nodes as keys and positions as values. Positions should be sequences of length 2. edge_labels : dictionary (default={}) Edge labels in a dictionary of labels keyed by edge two-tuple. Only labels for the keys in the dictionary are drawn. label_pos : float (default=0.5) Position of edge label along edge (0=head, 0.5=center, 1=tail) font_size : int (default=10) Font size for text labels font_color : string (default='k' black) Font color string font_weight : string (default='normal') Font weight font_family : string (default='sans-serif') Font family alpha : float or None (default=None) The text transparency bbox : Matplotlib bbox, optional Specify text box properties (e.g. shape, color etc.) for edge labels. Default is {boxstyle='round', ec=(1.0, 1.0, 1.0), fc=(1.0, 1.0, 1.0)}. horizontalalignment : string (default='center') Horizontal alignment {'center', 'right', 'left'} verticalalignment : string (default='center') Vertical alignment {'center', 'top', 'bottom', 'baseline', 'center_baseline'} ax : Matplotlib Axes object, optional Draw the graph in the specified Matplotlib axes. rotate : bool (default=True) Rotate edge labels to lie parallel to edges clip_on : bool (default=True) Turn on clipping of edge labels at axis boundaries Returns ------- dict `dict` of labels keyed by edge rNr ?rg@ihZi)r rErround)rrr)boxstyleecfcr ) rrrrr'rBrArotationrr;rrriFrj)rrrrrKrLrrvabsarctan2pirrtransform_anglesreshaperrMrrr|)$rr rD label_posr=r>r?r@r'r;rBrAr rCrrr ur`dr<rn1n2r(rrrrr^rdyangler trans_anglers$ r"rUrUsuT' z WWY+0+C+C+EF+E1a1a&!)+EFJ!<<>R%r7Rr7R NR3?3 3 NR3?3 3  8v{{} $RBBwTBYTBY JJrBwR0C"%%K@3FErz s{ 1a&!B,,77%8JBJJW]L^_`abKK <_QD%%%JE GG  3/ ll  " ! B8e*hNN    W  9   GsHH H HHc SSKnSSKnSSKn[ U5[ U5:XaT[ US[ 5(a<URRUS9n U RXE5 U RU5n O;[R"URRRU5/5n [ U5[ U 5:dU R [ U5:XaF[R""U [ U5S45n U SU SS2S4'U SU SS2S4'U S U SS2S 4'[%['[)U5[ U 555U SS2S 4'U $![an[S5UeSnAff=f![aS [R"UV s/sH'oRRRU 5PM) Os sn fsn 5n GN"f=f![*a XSS2S 4'U $f=f) aApply an alpha (or list of alphas) to the colors provided. Parameters ---------- colors : color string or array of floats (default='r') Color of element. Can be a single color format string, or a sequence of colors with the same length as node_list. If numeric values are specified they will be mapped to colors using the cmap and vmin,vmax parameters. See matplotlib.scatter for more details. alpha : float or array of floats Alpha values for elements. This can be a single alpha value, in which case it will be applied to all the elements of color. Otherwise, if it is an array, the elements of alpha will be applied to the colors in order (cycling through alpha multiple times if necessary). elem_list : array of rustworkx objects The list of elements which are being colored. These could be nodes, edges or labels. cmap : matplotlib colormap Color map for use if colors is a list of floats corresponding to points on a color mapping. vmin, vmax : float Minimum and maximum values for normalizing colors if a colormap is used Returns ------- rgba_colors : numpy ndarray Array containing RGBA format values for each of the node colours. rNr )r$r)rrr )rr )rrErE)rqr matplotlib.cmrrsrMrcmScalarMappableset_climto_rgbarvrrrrHrresizerOrr TypeError) rr' elem_listr$r&r%r~rqr mapper rgba_colorsrs r"rzrz0sJ    6{c)n$F1Iv)F)F&&D&1#nnV,  c((CJJ$=$=$E$Ef$M#NOK # u:K( (K,<,<I,N))K#i.!1DEK!,T!2KA !,T!2KA !,T!2KA  e c+6F!GH AqD G  9   & c((Z`#aZ`QVJJ$=$=$E$Ee$LZ`#abK c #"ArE #sH E;:E;6B%G E8' E33E8;G.G  GGG10G1)NNTF)NTF) N,z#1f78b4oNNNNNNNN)Nrr[solidNN NNNNTNrNrarc3rr) N r[ sans-serifnormalNNcenterrNT) Nrrr[rrNNrrNTT)NNN)__doc__collections.abcr itertoolsrrnumbersrnumpyrvrP__all__rrrRrSrTrUrzrr"r!sd.$#  [|@L    J     +{B     pl     WtLr