I.e7STddlmZddlZddlZddlZddlm Z ddlm Z m Z m Z ddl mZddlmZddlmZmZddlmZd d lmZd d lmZd d lmZmZejr$eejd dsejddde fdZddZ ddZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)dZ*dZ+dS)) defaultdictN) HoverTool)GraphLabelsdim)Storedimension_sanitizer) GraphPlot LabelsPlot)markers)_transfer_opts_cur_backend) process_crs)saveshow_loadedFbokeh)logoc  tt|D]\}}t|j||D]=\}}t |ttfr"||>t |trt|}t |trt|}d|d|tfdD} d| D} tfdddg| zD} j j dd\ } } |rx| |j djtfd D\}}d tt|D}ng}d}tt|D]@\}}|j|}|#|\}} j|| j||D]C\}}t |tttfr(||Dt%|D],\}}||||-t |trt|}| j|Bt fd D} j| j| jg|zt|z}tfd |D}g}|D]_}t |t&rt|}n#|||jvr||}n|}||` || }| |f| S)aF Generate a Graph element from a networkx.Graph object and networkx layout function or dictionary of node positions. Any keyword arguments will be passed to the layout function. By default it will extract all node and edge attributes from the networkx.Graph but explicit node information may also be supplied. Any non-scalar attributes, such as lists or dictionaries will be ignored. Parameters ---------- G : networkx.Graph Graph to convert to Graph element positions : dict or callable Node positions defined as a dictionary mapping from node id to (x, y) tuple or networkx layout function which computes a positions dictionary. kwargs : dict Keyword arguments for the element Returns ------- graph : holoviews.Graph Graph element startendc3|K|]6}|dvr0t|tdk2|V7dS))rrrN)len).0kedgess /lib/python3.11/site-packages/hvplot/networkx.py z!_from_networkx..?s^BBQ14D+D+DaMMSw-@-@@@@@@@BBcZg|](}t|trt|n|)S) isinstanceintstr)rcols r z"_from_networkx..As1QQQjc22;#c(((QQQr c3(K|] }|V dSNr")rr&rs rrz!_from_networkx..Bs'GGSeCjGGGGGGr Nrc32K|]\}}|k ||fVdSr)r")rrvidx_dims rrz!_from_networkx..Is0!Z!ZTQQRY\\1a&\\\\!Z!Zr ci|]\}}|| Sr"r")rivalss r z"_from_networkx..JsUUUDQUUUr c3K|]E}|jjvr5t|tjkA|VFdSr)) node_typekdimsrname)rrcls node_columnsxdims rrz!_from_networkx..asgRRQ!3=;N2N2N Q00C TY8O4P4PPPPPPPRRr c3(K|] }|V dSr)r")rr&r7s rrz!_from_networkx..ds(;;Cl3';;;;;;r )vdims)rlistrsortedadjitemsr#dictappendtupler%r3r4columnsr5zipnodesget enumerater$r: get_dimension)!G positionsrDr6kwargsrrattrvalue edge_cols edge_vdims edge_dataydimidim info_colsvalues node_infoidxposnodexyr/r& node_colsrB node_datar:rrr-r7r8s! ` @@@@r_from_networkxr\s6   Eggii ! ! s!!%,s"3"9"9";";<< & &KD%%$..  $K  u % % % % eU # # JJE c5 ! ! c((C ge$$$ e C    BBBB%BBBBBIQQyQQQJGGGGWe,t'jt&j| j} j| njddd d } t ddddt|tjdddddddddd | d } | ddd| fdtDd| vrPt| dt6rt9| d| d<t'j| d| d<d| vr| d| d <d!| vr| d!| d"<d#vrcd#}|t<vr@t<|}|d$}d%|vr't?j td&gd'|d%| d&<|| d(<d)vrd)}|| d*<|| d+<d,vrd,| d-<d.vrd.| d/<d0vrd0| d1<d2D]}t| |t&j!ttDfrY#j j $|t%j j%| |df|| |<d3D]q}t| |t&j!ttDfr5$|t%j%| |d|| |<r| d j &vrfd4dd5df}|d6kr4j '| d }||_"| d"&vrad7dd8df}|d6kr/'| d"}||_"| d kr.fd9j j(d:dj j%zD}nfd;j(j%zD}fd<|D}tS|=d>g| d?<j*dOi| d@d'idAdBdrhfdCtD}dDvrdD|dD<dEvrdE|dE<dFvrdF|dG<dHvrdH|dI<dJvrdJ|dK<dBj j(d:ttrPj j j(}fdL|D}||j j(dd:dMfi|n"|j j j(dd:fi|j*dOi|d@d'izdNvr(+dNtYS)PaV Draw the graph G using hvPlot. Draw the graph with hvPlot with options for node positions, labeling, titles, and many other drawing features. 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 :py:mod:`networkx.drawing.layout` for functions that compute node positions. arrows : bool, optional (default=True) For directed graphs, if True draw arrowheads. Note: Arrows will be the same color as edges. arrowhead_length : float, optional (default=0.025) The length of the arrows as fraction of the overall extent of the graph with_labels : bool, optional (default=True) Set to True to draw labels on the nodes. 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, node attribute, or array of floats, (default='r') Can be a single color, the name of an attribute on the nodes or sequence of colors with the same length as nodelist. If the node_color references an attribute on the nodes or is a list of values they will be colormapped using the cmap and vmin, vmax parameters. node_shape : string, optional (default='o') The shape of the node. Specification is as valid bokeh marker. alpha : float, optional (default=1.0) The node and edge transparency cmap : Colormap, optional (default=None) Colormap for mapping intensities of nodes vmin,vmax : float, optional (default=None) Minimum and maximum for node colormap scaling linewidths : [None | scalar | sequence] Line width of symbol border (default =1.0) edge_width : float, optional (default=1.0) Line width of edges edge_color : color string, or array of floats (default='r') Can be a single color, the name of an attribute on the edges or sequence of colors with the same length as the edges. If the edge_color references an attribute on the edges or is a list of values they will be colormapped using the edge_cmap and edge_vmin, edge_vmax parameters. edge_cmap : Matplotlib colormap, optional (default=None) Colormap for mapping intensities of edges edge_vmin,edge_vmax : floats, optional (default=None) Minimum and maximum for edge colormap scaling style : string, optional (default='solid') Edge line style (solid|dashed|dotted,dashdot) labels : dictionary or string, optional (default=None) Node labels in a dictionary keyed by node of text labels or a string referencing a node attribute font_size : int, optional (default=12) Font size for text labels font_color : string, optional (default='black') Font color string font_family : string, optional (default='sans-serif') Font family label : string, optional Label for graph legend selection_policy : string, optional (default='nodes') Whether to select 'nodes', 'edges' or None on tap and selection events. inspection_policy : string, optional (default='nodes') Whether to select 'nodes', 'edges' or None on tap and selection events. geo : boolean, optional (default=False) Whether to return a GeoViews graph crs : cartopy.crs.CRS A cartopy coordinate reference system (enables a geographic plot) height : int, optional (default=400) The height of the plot in pixels width : int, optional (default=400) The width of the plot in pixels N layout_kwargsgeoFcrsrzsIn order to use geo-related features the geoviews library must be available. It can be installed with pip or conda.r6nodelistedgelistrinspection_policyrDTarrowhead_lengthg?arrowscolorbarpaddingg?widthiheightselection_policyred) axiswiserddirectedrfrgrhrirjrcnode_fill_color_axis_defaults)xaxisyaxis show_framecXi|]&}|tjv||'Sr")r style_optspoprrrJs rr1zdraw..s2UUUa1 @T;T;TFJJqMM;T;T;Tr node_size node_colorrn edge_coloredge_line_color node_shapemarkerangle node_angler node_markeralpha node_alpha edge_alpha linewidthsnode_line_width edge_widthedge_line_widthstyleedge_line_dash)rnrwrr)rzedge_line_alpharrvminvmax)NN edge_vmin edge_vmaxcLg|] }|j|jjvrdn|jf!S) index_hover)labelrDr4r5rdgs rr'zdraw..:sJDDD1 3E3E--16RDDDr cRg|]#}|j|jvr |jdzn|jf$S)_values)rr4r5rs rr'zdraw..=sH444a17ll!&"2"2O444r cLg|] \}}|zv |dt|zf!S)z@{%s}r )rrr5 edge_styles node_styless rr'zdraw..?sLZZZE44{[?X3X3X"5d";";;<3X3X3Xr )tooltipstaptoolsbackend with_labelslabelscXi|]&}|tjv||'Sr")r rtrurvs rr1zdraw..Gs2[[[1ZEZ@Z@ZaA@Z@Z@Zr xoffsetyoffset font_sizetext_font_size font_color text_color font_family text_fontc6g|]\}}}|v |||fSr"r")rrXrYr/rs rr'zdraw..Us/MMM)1afQ6!9%r textrr")-nxdrawing spring_layoutr#r?rErgeoviews ImportErrorrrr\rDdatailocr;arrayr@rnp logical_and logical_orreduceTruDiGraphupdater%rsqrtr radd_style_optsndarrayrangeclone add_dimensionr: dimensionsrGr4roptsrelabelr)rHrVrJparams label_params label_elementrr`r comparisonsedgeselectorrcrr| marker_optsr node_style edge_stylelims dimension tooltip_dimsr label_optsrSrrrrrs ` @@@@rdrawrvsn n {j& c4 8c!77vzz/26677rLFM zz% "5F?? H OOOO H H HGHH H H&**U++,,   u u ! U q#((((AVw|(fZ.@)A)AB VA :& . .D   u} - - - - {   ~r}';';K'H'H'JKHx AArr A #6@@ $6>>HjBJ&?&?@@J.. 9c**jj#&&zz(C(($6@@+   D> $du ===KKUUUU4<<UUUVVVd d;' - - 7 #D$5 6 6D GD$566[t"&((<"8"8 t"&((<"8"8 vL)) W  !&/K *F+%%$U\NGDDD%0%9\"$]& 7##"\"\v"(**\":": v"(**\":": &!'G!4!4 SK!** dhhz**RZu,E F F *!6!6z3qw}CUCUW[\fWgim!n!noppA)D YK!** dhhz**RZu,E F F * CLL$z:JDQQA)D  xx!""ag&8&8&:&::: 64((&**VT*B*BC <  --dhh7H.I.IJJI"IO xx!""allnn44 ;--vzz+t/L/LM <  1B(C(CDDI"IOG##DDDD!"qrr!2QW]!BDDD 4444!"17!2444 ZZZZZ#/ZZZH11159DM AF##T##7####zz-He! ? ??r c 4t|fdtji|SaDraw networkx graph with circular layout. Parameters ---------- G : graph A networkx graph kwargs : optional keywords See hvplot.networkx.draw() for a description of optional keywords, with the exception of the pos parameter which is not used by this function. Returns ------- graph : holoviews.Graph or holoviews.Overlay Graph element or Graph and Labels rV)rrcircular_layoutrHrJs r draw_circularrs""  4 4r) 4V 4 44r c 4t|fdtji|Sr)rrkamada_kawai_layoutrs rdraw_kamada_kawairs""  8 8r- 8 8 88r c 2t|tjfi|S)aDraw networkx graph with random layout. Parameters ---------- G : graph A networkx graph kwargs : optional keywords See hvplot.networkx.draw() for a description of optional keywords, with the exception of the pos parameter which is not used by this function. Returns ------- graph : holoviews.Graph or holoviews.Overlay Graph element )rr random_layoutrs r draw_randomr" 2# . .v . ..r c p|dd}|d|i|d<t|tjfi|S)aDraw networkx graph with shell layout. Parameters ---------- G : graph A networkx graph kwargs : optional keywords See hvplot.networkx.draw() for a description of optional keywords, with the exception of the pos parameter which is not used by this function. Returns ------- graph : holoviews.Graph or holoviews.Overlay Graph element or Graph and Labels nlistNr^)rurr shell_layout)rHrJrs r draw_shellrsF" JJw % %E #*E"2 2? - -f - --r c 2t|tjfi|S)aDraw networkx graph with spectral layout. Parameters ---------- G : graph A networkx graph kwargs : optional keywords See hvplot.networkx.draw() for a description of optional keywords, with the exception of the pos parameter which is not used by this function. Returns ------- graph : holoviews.Graph or holoviews.Overlay Graph element or Graph and Labels )rrspectral_layoutrs r draw_spectralr&s" 2% 0 0 0 00r c 2t|tjfi|S)aDraw networkx graph with spring layout. Parameters ---------- G : graph A networkx graph kwargs : optional keywords See hvplot.networkx.draw() for a description of optional keywords, with the exception of the pos parameter which is not used by this function. Returns ------- graph : holoviews.Graph or holoviews.Overlay Graph element or Graph and Labels )rrrrs r draw_springr:rr c 2t|tjfi|S)aDraw networkx graph with planar layout. Parameters ---------- G : graph A networkx graph kwargs : optional keywords See hvplot.networkx.draw() for a description of optional keywords, with the exception of the pos parameter which is not used by this function. Returns ------- graph : holoviews.Graph or holoviews.Overlay Graph element or Graph and Labels )rr planar_layoutrs r draw_planarrMrr r)), collectionsrnumpyrnetworkxr holoviews_hv bokeh.modelsrrrrholoviews.core.optionsrholoviews.core.utilr holoviews.plotting.bokehr r holoviews.plotting.bokeh.stylesr backend_transformsrutilr utilitiesrr extensiongetattrr\rrrrrrrrrrrrr"r rrs######""""""((((((((((((((((333333::::::::333333::::::!!!!!!!!=' 5AA'CM'&&&&(,]5]5]5]5@l l l l ^"""">"""6""">@@@*555(999(///(....111(///&/////r