U td%@svdZddlmZmZmZmZddlmZddlm Z ddl m Z ddl m Z mZmZdZe \ZZGdd d eZd S) z:Drawers for various edge styles in Matplotlib graph plots.)atan2cospisin)AbstractEdgeDrawerAttributeCollectorBase)find_matplotlib)euclidean_distance)get_bezier_control_points_for_curved_edge!intersect_bezier_curve_and_circle)MatplotlibEdgeDrawerc@s8eZdZdZddZddZddZdd Zd d Zd S) r z0Matplotlib-specific abstract edge drawer object.cCs||_||_||_dS)zConstructs the edge drawer. @param context: a Matplotlib axes 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)selfrrrW/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/igraph/drawing/matplotlib/edge.py__init__szMatplotlibEdgeDrawer.__init__csGfdddt}|S)ztConstruct the visual edge builder that will collect the visual attributes of an edge when it is being drawn.csVeZdZdZdZdZdZdjjfZ dj fZ dZ djjfZ d Zd Zd ZdZd ZdS) zNMatplotlibEdgeDrawer._construct_visual_edge_builder..VisualEdgeBuilderzOBuilder that collects some visual properties of an edge for drawingZedge_gy&1|?gffffff?z#444gNblackg(@z sans-serif@F)__name__ __module__ __qualname____doc__Z _kwds_prefix arrow_size arrow_widthrgetcolorZ_curvature_to_floatcurvedlabelZ label_colorZ label_sizeZfontwidth backgroundZ align_labelrrrrr&s   rr)rrrr$rr"sz3MatplotlibEdgeDrawer._construct_visual_edge_builderc Cs||kr|||S|j}|j|j\}}\}}|j|j\} } \} } ggd} | d||g| dd|jrbt|||||j\}}|\}}|\}}t| | ||||| | |jd \}}t| || |}d|j }d|j }||t |t |||t |t |f||t |t |||t |t |fg}|dd|d dd|dd |d d d}}|dd|d d|dd |d d }}t| | |||j\}}||d|||d |}|tdd||d }|d|||d ||f}| d|| d|| d||g| dd gd n|j|||||j\}}t| || |}d|j }d|j }||t |t |||t |t |f||t |t |||t |t |fg}|dd|d dd|dd |d d d}}| d||g| dd tjjtjj| ddd| dDd|jd|jd}||tjj||g|d|d gd|jdd}||dS)Nverticescodesr&r'MOVETOrg.@g$@rCURVE4LINETOcSsg|]}ttjj|qSrgetattrmplpathPath.0xrrr sz;MatplotlibEdgeDrawer.draw_directed_edge..r'none edgecolor facecolor linewidthT)closedr;r:)draw_loop_edgerpositionappendr r r sizerrrrrrr extendshapeZintersection_pointr0patches PathPatchr1r2rr" add_patchZPolygon)redge src_vertex dest_vertexaxx1y1x2y2Zx_srcZy_srcZx_destZy_destr1aux1aux2Zxc1Zyc1Zxc2Zyc2ZanglerrZ aux_pointsZ x_arrow_midZ y_arrow_midZx_arrow_base_vecZy_arrow_base_vecZ offset_lengthstrokeZ arrowheadrrrdraw_directed_edge9s           z'MatplotlibEdgeDrawer.draw_directed_edgec Cs|j}|jd}|jdttd|d}|jdttd|d}tjj||f|d|ddd|j d|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)gv@r8)Ztheta1Ztheta2r<r;r:N) rrAr?rrrr0rDZArcr"rrF)rrGZvertexrJZradiusZcenter_xZcenter_yrQrrrr>s   z#MatplotlibEdgeDrawer.draw_loop_edgec Cs||kr|||S|j}ggd}|d|j|dd|jr|j|j\}}\}} t|||| |j\} } |d| |d| |d|j|ddgdn|d|j|ddtjj tj j |ddd |dDd |j d |j d } || d S)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,r-cSsg|]}ttjj|qSrr.r3rrrr6sz=MatplotlibEdgeDrawer.draw_undirected_edge..r7r8r9N)r>rr@r?r r rBr0rDrEr1r2rr"Z add_artist) rrGrHrIrJr1rKrLrMrNrOrPrQrrrdraw_undirected_edges2    z)MatplotlibEdgeDrawer.draw_undirected_edgeN) rrrrrrrRr>rTrrrrr s r N)rmathrrrrZigraph.drawing.baseclassesrZigraph.drawing.metamagicrZigraph.drawing.matplotlib.utilsr Zigraph.drawing.utilsr r r __all__r0Zpltr rrrrs