U tdz$@sdZddlmZmZmZmZddlmZddlm Z ddl m Z m Z m Z mZddlmZmZmZmZdZe ZGdd d eZd S) z:Drawers for various edge styles in Matplotlib graph plots.)atan2cospisin)AbstractEdgeDrawerAttributeCollectorBase) find_plotlyformat_path_step format_arc format_rgba)Pointeuclidean_distance)get_bezier_control_points_for_curved_edge!intersect_bezier_curve_and_circle)PlotlyEdgeDrawerc@s8eZdZdZddZddZddZdd Zd d Zd S) rz0Matplotlib-specific abstract edge drawer object.cCs||_||_||_dS)zConstructs the edge drawer. @param context: a plotly Figure object on which the edges will be drawn. @param palette: the palette that can be used to map integer color indices to colors when drawing edges N)contextpalette_construct_visual_edge_builderVisualEdgeBuilder)selfrrrS/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/igraph/drawing/plotly/edge.py__init__szPlotlyEdgeDrawer.__init__csGfdddt}|S)ztConstruct the visual edge builder that will collect the visual attributes of an edge when it is being drawn.csFeZdZdZdZdZdZdZdjfZ dZ dj j fZ d Zd Zd ZdS) zJPlotlyEdgeDrawer._construct_visual_edge_builder..VisualEdgeBuilderzOBuilder that collects some visual properties of an edge for drawingZedge_gy&1|?gffffff?z#444gNblackg(@z sans-serif@)__name__ __module__ __qualname____doc__Z _kwds_prefix arrow_size arrow_widthcolorZ_curvature_to_floatcurvedlabelrgetZ label_colorZ label_sizeZfontwidthrrrrr,s  rr)rrrr'rr(sz/PlotlyEdgeDrawer._construct_visual_edge_builderc Csz||kr|||S|j}|j|j\}}\}}|j|j\} } \} } td||gg} |jr t|||||j\}}|\}}|\}}t| | ||||| | |jd \}}t| || |}d|j }d|j }||t |t |||t |t |f||t |t |||t |t |fg}|dd|ddd|dd|ddd}}|dd|dd|dd|dd}}t| | |||j\}}||d|||d|}|tdd||d}|d|||d||f}| td||||ggnt| || |}d|j }d|j }||t |t |||t |t |f||t |t |||t |t |fg}|dd|ddd|dd|ddd}}| td t||d | } td | t|j|jd }||tjj||dd|dd|g||dd|dd|gt|jd d}||dS)NMrg.@g$@rCL pathtyper.Z line_color line_widthlines)xyZ fillcolormode)draw_loop_edgerpositionr r#rrsizerr r!rrrrappendr joindictr r"r& add_shapeplotlyZ graph_objectsZScatterZ add_trace)redge src_vertex dest_vertexfigx1y1x2y2Zx_srcZy_srcZx_destZy_destr.aux1aux2Zxc1Zyc1Zxc2Zyc2Zangler r!Z aux_pointsZ x_arrow_midZ y_arrow_midZx_arrow_base_vecZy_arrow_base_vecZ offset_lengthstrokeZ arrowheadrrrdraw_directed_edge=s            z#PlotlyEdgeDrawer.draw_directed_edgec Cs|j}|jd}|jdttd|d}|jdttd|d}tdt||f|d|dddd|j|j d }| |d S) aSDraws a loop edge. The default implementation draws a small circle. @param edge: the edge to be drawn. Visual properties of the edge are defined by the attributes of this object. @param vertex: the vertex to which the edge is attached. Visual properties are given again as attributes. g?rrr)r.gv@)Ztheta1Ztheta2r/N) rr8r7rrrr;r r"r&r<)rr>ZvertexrAZradiusZcenter_xZcenter_yrHrrrr6s"   zPlotlyEdgeDrawer.draw_loop_edgec Cs||kr|||S|j}td|jg}|jrt|j|j\}}\}} t|||| |j\} } |td| | |jgn|td|jd|}td|t |j |j d} | | dS)aDraws an undirected edge. The default implementation of this method draws undirected edges as straight lines. Loop edges are drawn as small circles. @param edge: the edge to be drawn. Visual properties of the edge are defined by the attributes of this object. @param src_vertex: the source vertex. Visual properties are given again as attributes. @param dest_vertex: the target vertex. Visual properties are given again as attributes. r(r+r,r-r.r/N) r6rr r7r#rr9r:r;r r"r&r<) rr>r?r@rAr.rBrCrDrErFrGrHrrrdraw_undirected_edges6    z%PlotlyEdgeDrawer.draw_undirected_edgeN) rrrrrrrIr6rKrrrrrs rN)rmathrrrrZigraph.drawing.baseclassesrZigraph.drawing.metamagicrZigraph.drawing.plotly.utilsr r r r Zigraph.drawing.utilsr rrr__all__r=rrrrrs