U td6@sdZdZddlmZmZddlmZmZmZm Z m Z ddl Z ddl m Z e \ZZGddded ZGd d d eZGd d d eZGdddeZGdddeZGdddeZGdddeZGdddZee jejdS)a` Shape drawing classes for igraph Vertex shapes in igraph are usually referred to by short names like C{"rect"} or C{"circle"}. This module contains the classes that implement the actual drawing routines for these shapes, and a resolver class that determines the appropriate shape drawer class given the short name. Classes that are derived from L{ShapeDrawer} in this module are automatically registered by L{ShapeDrawerDirectory}. If you implement a custom shape drawer, you must register it in L{ShapeDrawerDirectory} manually if you wish to refer to it by a name in the C{shape} attribute of vertices. )ShapeDrawerDirectory)ABCMetaabstractmethod)atan2copysigncospisinN)find_matplotlibc@s0eZdZdZeedddZedddZdS) ShapeDrawerzStatic class, the ancestor of all vertex shape drawer classes. Custom shapes must implement at least the C{draw_path} method of the class. The method I{must not} stroke or fill, it should just set up the current Cairo path appropriately.NcKstdS)a:Draws the path of the shape on the given Cairo context, without stroking or filling it. This method must be overridden in derived classes implementing custom shapes and declared as a static method using C{staticmethod(...)}. @param ctx: the context to draw on @param center_x: the X coordinate of the center of the object @param center_y: the Y coordinate of the center of the object @param width: the width of the object @param height: the height of the object. If C{None}, equals to the width. N)NotImplementedErrorctxcenter_xcenter_ywidthheightkwargsrN/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/igraph/drawing/shapes.py draw_path&szShapeDrawer.draw_pathcCs||fS)aUDetermines where the shape centered at (center_x, center_y) intersects with a line drawn from (source_x, source_y) to (center_x, center_y). Can be overridden in derived classes. Must always be defined as a static method using C{staticmethod(...)} @param width: the width of the shape @param height: the height of the shape. If C{None}, defaults to the width @return: the intersection point (the closest to (source_x, source_y) if there are more than one) or (center_x, center_y) if there is no intersection rrrsource_xsource_yrrrrrintersection_point7szShapeDrawer.intersection_point)N)N)__name__ __module__ __qualname____doc__ staticmethodrrrrrrrr s r ) metaclassc@s,eZdZdZdddddgZed dd ZdS) NullDrawerzgStatic drawer class which draws nothing. This class is used for graph vertices with unknown shapesnullnoneemptyZhiddenNcCsdS)zDraws nothing.Nr)rrrrrrrrrPszNullDrawer.draw_path)N)rrrrnamesrrrrrrr!Isr!c@s0eZdZdZdZedddZed ddZdS) RectangleDrawerz-Static class which draws rectangular verticesz'rectangle rect rectangular square box sNcKsj|p|}ttdrFt|tjrFtjj||d||df||f|S|||d||d||dS)z~Draws a rectangle-shaped path on the Cairo context without stroking or filling it. @see: ShapeDrawer.draw_pathAxesN)hasattrplt isinstancer(mplpatchesZ RectangleZ rectangler rrrr[szRectangleDrawer.draw_pathc Cs|p|}||||}}|dkr2|dkr2||fS|dkrt||krt|| krt||d}|d|} || ||fS|dkr|| kr||kr||d}|d| } || ||fS|dkr||kr|| kr||d} |d|} | || |fS|dkrD|| krD||krD||d} |d| } | || |fS|dkrx|dkrh|||dfS|||dfS|dkr|dkr||d|fS||d|fSdS)zDetermines where the rectangle centered at (center_x, center_y) having the given width and height intersects with a line drawn from (source_x, source_y) to (center_x, center_y). @see: ShapeDrawer.intersection_pointrr)Nr) rrrrrrdelta_xdelta_yZryratiorxrrrrhs8           z"RectangleDrawer.intersection_point)N)Nrrrrr&rrrrrrrr'Vs  r'c@s0eZdZdZdZedddZed ddZdS) CircleDrawerz*Static class which draws circular verticeszcircle circular oNcKsNttdr0t|tjr0tjj||f|df|S||||dddtdS)zDraws a circular path on the Cairo context without stroking or filling it. Height is ignored, it is the width that determines the diameter of the circle. @see: ShapeDrawer.draw_pathr(r)rN) r*r+r,r(r-r.ZCirclearcrr rrrrszCircleDrawer.draw_pathcCsB|p|}t||||}||dt|||dt|fS)zDetermines where the circle centered at (center_x, center_y) intersects with a line drawn from (source_x, source_y) to (center_x, center_y). @see: ShapeDrawer.intersection_pointr))rrr )rrrrrrZanglerrrrszCircleDrawer.intersection_point)N)Nr3rrrrr4s  r4c@s0eZdZdZdZedddZed ddZdS) UpTriangleDrawerz*Static class which draws upright trianglesz:triangle triangle-up up-triangle arrow arrow-up up-arrow ^NcKs|p|}ttdrtt|tjrt|d||d|g|d||d|g||d|gg}tjj|fddi|S|||d||d||||d|||d||d| dS) z{Draws an upright triangle on the Cairo context without stroking or filling it. @see: ShapeDrawer.draw_pathr(?Zd;O?MbX?closedTr)N r*r+r,r(r-r.ZPolygonmove_toZline_toZ close_pathrrrrrrZverticesrrrrszUpTriangleDrawer.draw_pathcCs|p|}||fSzDetermines where the triangle centered at (center_x, center_y) intersects with a line drawn from (source_x, source_y) to (center_x, center_y). @see: ShapeDrawer.intersection_pointrrrrrrsz#UpTriangleDrawer.intersection_point)N)Nr3rrrrr6s  r6c@s0eZdZdZdZedddZed ddZdS) DownTriangleDrawerz0Static class which draws triangles pointing downz3down-triangle triangle-down arrow-down down-arrow vNcKs|p|}ttdrtt|tjrt|d||d|g|d||d|g||d|gg}tjj|fddi|S|||d||d||||d|||d||d| dS) zrDraws a triangle on the Cairo context without stroking or filling it. @see: ShapeDrawer.draw_pathr(r7r8r9r:Tr)Nr;r=rrrrszDownTriangleDrawer.draw_pathcCs|p|}||fSr>rrrrrrsz%DownTriangleDrawer.intersection_point)N)Nr3rrrrr?s  r?c@s0eZdZdZdZedddZed ddZdS) DiamondDrawerz2Static class which draws diamonds (i.e. rhombuses)zdiamond rhombus dNcKs|p|}ttdrrt|tjrr|d||g||d|g|d||g||d|gg}tjj|fddi|S|||d|||||d|||d|||||d| dS)zqDraws a rhombus on the Cairo context without stroking or filling it. @see: ShapeDrawer.draw_pathr(r7r:Tr)Nr;r=rrrrszDiamondDrawer.draw_pathc Cs|p|}|dkr |dkr ||fS||||}}|dkr`|dkrJ||fS||t|d|fSt||}t||}|||||}|||d|d||dfS)zDetermines where the rhombus centered at (center_x, center_y) intersects with a line drawn from (source_x, source_y) to (center_x, center_y). @see: ShapeDrawer.intersection_pointrr))r) rrrrrrr/r0frrrrs  z DiamondDrawer.intersection_point)N)Nr3rrrrr@s  r@c@sHeZdZdZiZeddZeddZeddZee fdd Z d S) ra Static class that resolves shape names to their corresponding shape drawer classes. Classes that are derived from L{ShapeDrawer} in this module are automatically registered by L{ShapeDrawerDirectory} when the module is loaded for the first time. cCs0|j}t|tr|}|D]}||j|<qdS)zRegisters the given shape drawer class under the given names. @param drawer_class: the shape drawer class to be registered N)r&r,strsplit known_shapes)clsZ drawer_classr&namerrrregisterFs  zShapeDrawerDirectory.registercCsH|D]:\}}|drqt|trt|tr|tkr||qdS)zRegisters all L{ShapeDrawer} classes in the given namespace @param namespace: a Python dict mapping names to Python objects.__N)items startswithr,type issubclassr rH)rF namespacerGvaluerrrregister_namespaceSs   z'ShapeDrawerDirectory.register_namespacecCs2z |j|WStk r,td|YnXdS)zGiven a shape name, returns the corresponding shape drawer class @param shape: the name of the shape @return: the corresponding shape drawer class @raise ValueError: if the shape is unknown zunknown shape: %sN)rEKeyError ValueError)rFshaperrrresolve_s  zShapeDrawerDirectory.resolvecCs|j||S)aGiven a shape name, returns the corresponding shape drawer class or the given default shape drawer if the shape name is unknown. @param shape: the name of the shape @param default: the default shape drawer to return when the shape is unknown @return: the shape drawer class corresponding to the given name or the default shape drawer class if the name is unknown )rEget)rFrSdefaultrrrresolve_defaultms z$ShapeDrawerDirectory.resolve_defaultN) rrrrrE classmethodrHrPrTr!rWrrrrr;s   r)r__all__abcrrmathrrrrr sysZigraph.drawing.matplotlib.utilsr r-r+r r!r'r4r6r?r@rrPmodulesr__dict__rrrrs  * B%&:@