U td @sDddlmZddlmZddlmZdZe\ZZGdddZ dS) )calculate_corner_radii)consecutive_pairs)find_matplotlib)MatplotlibPolygonDrawerc@s"eZdZdZddZdddZdS) raOClass that is used to draw polygons in matplotlib. The corner points of the polygon can be set by the C{points} property of the drawer, or passed at construction time. Most drawing methods in this class also have an extra C{points} argument that can be used to override the set of points in the C{points} property.cCs ||_dS)zConstructs a new polygon drawer that draws on the given Matplotlib axes. @param ax: the matplotlib Axes to draw on N)context)selfaxr Z/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/igraph/drawing/matplotlib/polygon.py__init__sz MatplotlibPolygonDrawer.__init__rcKsVt|dkrdS|j}|dkr8tjj|f|}||t||}g}g}||d|d|d|tj j j |d} t t |dD]\} \} } || } || | | |tj j j| | | d}| | | d}|||||| | || |tj j jgd| } qtjjtj j ||ddf|}||dS)aDraws a polygon to the associated axes. @param points: the coordinates of the corners of the polygon, in clockwise or counter-clockwise order, or C{None} if we are about to use the C{points} property of the class. @param corner_radius: if zero, an ordinary polygon will be drawn. If positive, the corners of the polygon will be rounded with the given radius. NrT)codesclosed)lenrmplZpatchesZPolygonZ add_patchrappendZtowardspathPathZMOVETO enumeraterZLINETOextendZCURVE4Z PathPatch)rZpointsZ corner_radiuskwdsr strokeZ corner_radiirruidxvwZradiusZaux1Zaux2r r r draws:    zMatplotlibPolygonDrawer.drawN)r)__name__ __module__ __qualname____doc__r rr r r r r srN) Zigraph.drawing.utilsrZ igraph.utilsrutilsr__all__rZpltrr r r r s