U tdE1@sdZddlmZmZmZmZddlmZddlm Z ddl m Z ddl m Z mZmZddlmZd ZeZGd d d eZGd d d eZGdddeZGdddeZGdddeZGdddeZdS)z1 Drawers for various edge styles in graph plots. )atan2cospisin)AbstractEdgeDrawer)clampAttributeCollectorBase)euclidean_distance)get_bezier_control_points_for_curved_edge!intersect_bezier_curve_and_circle) find_cairo)AbstractCairoEdgeDrawerAlphaVaryingEdgeDrawerCairoArrowEdgeDrawerDarkToLightEdgeDrawerLightToDarkEdgeDrawerTaperedEdgeDrawerc@s0eZdZdZddZddZddZdd Zd S) rz+Cairo-specific abstract edge drawer object.cCs||_||_||_dS)zConstructs the edge drawer. @param context: a Cairo context 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_builderVisualEdgeBuilderselfrrrR/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/igraph/drawing/cairo/edge.py__init__!sz AbstractCairoEdgeDrawer.__init__csGfdddt}|S)ztConstruct the visual edge builder that will collect the visual attributes of an edge when it is being drawn.csNeZdZdZdZdZdZdjjfZ dj fZ dZ djjfZ dZd ZdZdS) zQAbstractCairoEdgeDrawer._construct_visual_edge_builder..VisualEdgeBuilderzOBuilder that collects some visual properties of an edge for drawingZedge_?z#444Nblackg(@z sans-serif)__name__ __module__ __qualname____doc__Z _kwds_prefix arrow_size arrow_widthrgetcolorZ_curvature_to_floatcurvedlabelZ label_colorZ label_sizeZfontwidthrrrrr0s   rr)rrrr,rr,sz6AbstractCairoEdgeDrawer._construct_visual_edge_buildercCs|j}|j|j||j|jd}|jdttd|d}|jdt td|d}| |||ddtd| dS)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?r@r N) rset_source_rgbar(set_line_widthr+sizepositionrrrarcstroke)redgeZvertexctxZradiusZcenter_xZcenter_yrrrdraw_loop_edgeAs    z&AbstractCairoEdgeDrawer.draw_loop_edgec Cs||kr|||S|j}|j|j||j|j|j|jr|j|j\}}\}}t |||||j\} } |j | d| d| d| df|jn |j |j| 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. rr N) r8rr0r(r1r+move_tor3r)r curve_toline_tor5) rr6 src_vertex dest_vertexr7x1y1x2y2aux1aux2rrrdraw_undirected_edgeTs$     * z,AbstractCairoEdgeDrawer.draw_undirected_edgeN)r!r"r#r$rrr8rDrrrrrs  rc@seZdZdZddZdS)rzoEdge drawer implementation that draws undirected edges as straight lines and directed edges as arrows. c Cs||kr|||S|j}|j|j\}}\}}|j|j\} } \} } |j|j||j||||jrt |||||j\} }| \}}|\}}t | | ||||| | |j d \}}t | || |}d|j }d|j}||t|t|||t|t|f||t|t|||t|t|fg}|dd|ddd|dd|ddd}}|dd|dd|dd|dd}}d| |d|jd|| d| |d|jd|| f} | d|d|jd|| | d|d|jd|| f}||d|||d|}|tdd||d}|d|||d||f}|| d| d|d|d||n|j|||||j \}}t | || |}d|j }d|j}||t|t|||t|t|f||t|t|||t|t|fg}|dd|ddd|dd|ddd}}||||||||j|d|j|d||||dS) Nr.g.@g$@rr r/g@g?)r8rr3r0r(r1r+r9r)r r r2rr%r&rrrr r:shapeintersection_pointr;r5fill)rr6r<r=r7r>r?r@rAZx_srcZy_srcZx_destZy_destrBrCZxc1Zyc1Zxc2Zyc2angler%r& aux_pointsZ x_arrow_midZ y_arrow_midZx_arrow_base_vecZy_arrow_base_vecZ offset_lengthrrrdraw_directed_edgezs         &      z'CairoArrowEdgeDrawer.draw_directed_edgeNr!r"r#r$rJrrrrrusrc@seZdZdZddZdS)rzEdge drawer implementation that draws undirected edges as straight lines and directed edges as tapered lines that are wider at the source and narrow at the destination. c Cs>||kr|||S|j|j}}|j|d|d|d|d|j}|j}|j|j||j t |d|d|d|d}|jd}|d|t |t d|d|t |t df|d|t |t d|d|t |t dfg} |j||j| d|j| d|j||dS)Nrr g@r/)r8r3rErFr2rr0r(r1r+rrrrr9r;rG) rr6r<r=src_posdest_posr7rHr%rIrrrrJs6   "   z$TaperedEdgeDrawer.draw_directed_edgeNrKrrrrrsrcs(eZdZdZfddZddZZS)rzEdge drawer implementation that draws undirected edges as straight lines and directed edges by varying the alpha value of the specified edge color between the source and the destination. cs:t||tt|ddf|_tt|ddf|_dSNrr)superrrfloat alpha_at_src alpha_at_dest)rrrrQrR __class__rrr$szAlphaVaryingEdgeDrawer.__init__c Cs||kr|||S|j|j}}|j}t|d|d|d|d}|jdd|j}|dd|j} |jd||jd| | || |j |j ||j ||dS)Nrr )r)r )r8r3rcairoZLinearGradientr(rQrRZadd_color_stop_rgbaZ set_sourcer1r+r9r;r5) rr6r<r=rLrMr7ZlgZ edge_colorZedge_color_endrrrrJ)s      z)AlphaVaryingEdgeDrawer.draw_directed_edge)r!r"r#r$rrJ __classcell__rrrSrrs rcs eZdZdZfddZZS)raEdge drawer implementation that draws undirected edges as straight lines and directed edges by using an alpha value of zero (total transparency) at the source and an alpha value of one (full opacity) at the destination. The alpha value is interpolated in-between. cst||dddSrNrOrrrSrrrGszLightToDarkEdgeDrawer.__init__r!r"r#r$rrWrrrSrr?srcs eZdZdZfddZZS)raEdge drawer implementation that draws undirected edges as straight lines and directed edges by using an alpha value of one (full opacity) at the source and an alpha value of zero (total transparency) at the destination. The alpha value is interpolated in-between. cst||dddS)NrrrXrrSrrrSszDarkToLightEdgeDrawer.__init__rYrrrSrrKsrN)r$mathrrrrZigraph.drawing.baseclassesrZigraph.drawing.colorsrZigraph.drawing.metamagicr Zigraph.drawing.utilsr r r utilsr__all__rVrrrrrrrrrrs     W)!