U td^@sdZddlmZddlmZmZmZmZddlm Z ddl m Z dZ GdddZ Gd d d e ZGd d d ZGd dde ddefdefgZddZddZddZddZd$ddZd%ddZd&d!d"Zd#S)'z' Utility classes for drawing routines. ) defaultdict)atan2coshypotsin) NamedTuple)consecutive_pairs) BoundingBoxPoint Rectanglecalculate_corner_radiieuclidean_distanceevaluate_cubic_bezier)get_bezier_control_points_for_curved_edge!intersect_bezier_curve_and_circlestr_to_orientation autocurvec@seZdZdZdZddZeddZejddZedd Z e jd d Z ed d Z e jd d Z eddZ e jddZ eddZ e jddZ eddZ e jddZ eddZejddZeddZejddZeddZejddZed d!Zejd"d!Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,ZeZd-d.Zd/d0ZeZd1d2Zd3d4Zd5d6Zd7d8Zd9d:Zd;d<Z d=S)>r zClass representing a rectangle._left_top_right_bottomcGsd}t|dkr~t|dtr*|dj}qt|ddkrPt|ddd}qt|ddkrdd|dd|ddf}n6t|dkrt|}n t|dkrdd|d|df}|dkrtdztdd|D}Wntk rtd YnX||_dS) a>Creates a rectangle. The corners of the rectangle can be specified by either a tuple (four items, two for each corner, respectively), four separate numbers (X and Y coordinates for each corner) or two separate numbers (width and height, the upper left corner is assumed to be at (0,0))Nrzinvalid coordinate formatcss|]}t|VqdSNfloat).0ZcoordrM/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/igraph/drawing/utils.py 7sz%Rectangle.__init__..z+invalid coordinate format, numbers expected)len isinstancer coordstuple ValueError)selfargsr$rrr __init__ s&     zRectangle.__init__cCs|j|j|j|jfS)zThe coordinates of the corners. The coordinates are returned as a 4-tuple in the following order: left edge, top edge, right edge, bottom edge. rr'rrr r$=szRectangle.coordscCsT|\|_|_|_|_|j|jkr2|j|j|_|_|j|jkrP|j|j|_|_dS)zsSets the coordinates of the corners. @param coords: a 4-tuple with the coordinates of the corners Nr)r'r$rrr r$Fs   cCs |j|jS)zThe width of the rectangle)rrr*rrr widthRszRectangle.widthcCs|j||_dS)zSets the height of the rectangle by adjusting the bottom edge.Nrrr-rrr r/ascCs|jS)z,The X coordinate of the left side of the box)rr*rrr leftfszRectangle.leftcCst||_t|j|j|_dS)z1Sets the X coordinate of the left side of the boxN)rrmaxrr-rrr r1ks cCs|jS)z-The X coordinate of the right side of the box)rr*rrr rightqszRectangle.rightcCst||_t|j|j|_dS)z2Sets the X coordinate of the right side of the boxN)rrminrr-rrr r3vs cCs|jS)z+The Y coordinate of the top edge of the box)rr*rrr top|sz Rectangle.topcCs||_t|j|j|_dS)z0Sets the Y coordinate of the top edge of the boxN)rr2rr-rrr r5scCs|jS)z.The Y coordinate of the bottom edge of the box)rr*rrr bottomszRectangle.bottomcCs||_t|j|j|_dS)z3Sets the Y coordinate of the bottom edge of the boxN)rr4rr-rrr r6scCs|j|jdS)z)The X coordinate of the center of the box@r,r*rrr midxszRectangle.midxcCs4||j|jd}|j|7_|j|7_dS)z5Moves the center of the box to the given X coordinater7Nr,)r'r.dxrrr r8scCs|j|jdS)z)The Y coordinate of the center of the boxr7r0r*rrr midyszRectangle.midycCs4||j|jd}|j|7_|j|7_dS)z5Moves the center of the box to the given Y coordinater7Nr0)r'r.dyrrr r:scCs|j|j|j|jfS)z*The shape of the rectangle (width, height))rrrrr*rrr shapeszRectangle.shapecCs|\|_|_dS)z0Sets the shape of the rectangle (width, height).N)r+r/)r'r<rrr r<scCst|tst|tr"t|gd}t|dkr6td|j|d|j|d}}|j|d|j|d}}||kr||d}|}||kr||d}|}| ||||S)zcContracts the rectangle by the given margins. @return: a new L{Rectangle} object. rz,margins must be a 4-tuple or a single numberrrrr7) r#intrr"r&rrrr __class__)r'marginsZnx1Zny1Znx2Zny2rrr contracts   zRectangle.contractcCs8t|tst|tr$|t| S|dd|DS)zaExpands the rectangle by the given margins. @return: a new L{Rectangle} object. cSsg|]}t| qSrr)rmarginrrr sz$Rectangle.expand..)r#r>rrA)r'r@rrr expandszRectangle.expandcCs0|j|jkp.|j|jkp.|j|jkp.|j|jkS)aReturns C{True} if the two rectangles have no intersection. Example:: >>> r1 = Rectangle(10, 10, 30, 30) >>> r2 = Rectangle(20, 20, 50, 50) >>> r3 = Rectangle(70, 70, 90, 90) >>> r1.isdisjoint(r2) False >>> r2.isdisjoint(r1) False >>> r1.isdisjoint(r3) True >>> r3.isdisjoint(r1) True rrrrr'otherrrr isdisjoints    zRectangle.isdisjointcCs|j|jko|j|jkS)anReturns C{True} if the rectangle is empty (i.e. it has zero width and height). Example:: >>> r1 = Rectangle(10, 10, 30, 30) >>> r2 = Rectangle(70, 70, 90, 90) >>> r1.isempty() False >>> r2.isempty() False >>> r1.intersection(r2).isempty() True rEr*rrr isemptyszRectangle.isemptycCsN||rtddddStt|j|jt|j|jt|j|jt|j|jS)aReturns the intersection of this rectangle with another. Example:: >>> r1 = Rectangle(10, 10, 30, 30) >>> r2 = Rectangle(20, 20, 50, 50) >>> r3 = Rectangle(70, 70, 90, 90) >>> r1.intersection(r2) Rectangle(20.0, 20.0, 30.0, 30.0) >>> r2 & r1 Rectangle(20.0, 20.0, 30.0, 30.0) >>> r2.intersection(r1) == r1.intersection(r2) True >>> r1.intersection(r3) Rectangle(0.0, 0.0, 0.0, 0.0) r)rHr r2rrr4rrrFrrr intersections     zRectangle.intersectioncCs<|j|7_|j|7_|j|7_|j|7_dS)aNTranslates the rectangle in-place. Example: >>> r = Rectangle(10, 20, 50, 70) >>> r.translate(30, -10) >>> r Rectangle(40.0, 10.0, 80.0, 60.0) @param dx: the X coordinate of the translation vector @param dy: the Y coordinate of the translation vector NrE)r'r9r;rrr translates zRectangle.translatecCs6tt|j|jt|j|jt|j|jt|j|jS)aWReturns the union of this rectangle with another. The resulting rectangle is the smallest rectangle that contains both rectangles. Example:: >>> r1 = Rectangle(10, 10, 30, 30) >>> r2 = Rectangle(20, 20, 50, 50) >>> r3 = Rectangle(70, 70, 90, 90) >>> r1.union(r2) Rectangle(10.0, 10.0, 50.0, 50.0) >>> r2 | r1 Rectangle(10.0, 10.0, 50.0, 50.0) >>> r2.union(r1) == r1.union(r2) True >>> r1.union(r3) Rectangle(10.0, 10.0, 90.0, 90.0) )r r4rrr2rrrFrrr union's     zRectangle.unioncCsDt|j|j|_t|j|j|_t|j|j|_t|j|j|_|S)aExpands this rectangle to include itself and another completely while still being as small as possible. Example:: >>> r1 = Rectangle(10, 10, 30, 30) >>> r2 = Rectangle(20, 20, 50, 50) >>> r3 = Rectangle(70, 70, 90, 90) >>> r1 |= r2 >>> r1 Rectangle(10.0, 10.0, 50.0, 50.0) >>> r1 |= r3 >>> r1 Rectangle(10.0, 10.0, 90.0, 90.0) r4rrr2rrrFrrr __ior__Ds zRectangle.__ior__cCsd|jj|j|j|j|jfS)Nz%s(%s, %s, %s, %s))r?__name__rrrrr*rrr __repr__ZszRectangle.__repr__cCs |j|jkSrr$rFrrr __eq__cszRectangle.__eq__cCs |j|jkSrrQrFrrr __ne__fszRectangle.__ne__cCs|j|jkp|j|jkSrrEr*rrr __bool__iszRectangle.__bool__cCs t|jSr)hashr$r*rrr __hash__lszRectangle.__hash__N)!rO __module__ __qualname____doc__ __slots__r)propertyr$setterr+r/r1r3r5r6r8r:r<rArDrHrIrJ__and__rKrL__or__rNrPrRrSrTrVrrrr r st                      r c@s eZdZdZddZddZdS)r zVClass representing a bounding box (a rectangular area) that encloses some objects.cCsDt|j|j|_t|j|j|_t|j|j|_t|j|j|_|S)a6Replaces this bounding box with the union of itself and another. Example:: >>> box1 = BoundingBox(10, 20, 50, 60) >>> box2 = BoundingBox(70, 40, 100, 90) >>> box1 |= box2 >>> print(box1) BoundingBox(10.0, 20.0, 100.0, 90.0) rMrFrrr rNws zBoundingBox.__ior__cCs8|t|j|jt|j|jt|j|jt|j|jS)aUTakes the union of this bounding box with another. The result is a bounding box which encloses both bounding boxes. Example:: >>> box1 = BoundingBox(10, 20, 50, 60) >>> box2 = BoundingBox(70, 40, 100, 90) >>> box1 | box2 BoundingBox(10.0, 20.0, 100.0, 90.0) )r?r4rrr2rrrFrrr r^s     zBoundingBox.__or__N)rOrWrXrYrNr^rrrr r ssr cs8eZdZdZddZddZddZfdd ZZS) FakeModulez3Fake module that raises an exception for everythingcCs ||_dS)zeConstructor. @param message: message to print in exceptions raised from this module N)_message)r'messagerrr r)szFakeModule.__init__cCst|jdSr)AttributeErrorr`r'_rrr __getattr__szFakeModule.__getattr__cCst|jdSr) TypeErrorr`rcrrr __call__szFakeModule.__call__cs&|dkrt||n t|jdS)Nr`)super __setattr__rbr`)r'keyr.r?rr riszFakeModule.__setattr__) rOrWrXrYr)rergri __classcell__rrrkr r_s r_c@s|eZdZdZddZddZddZeZdd Zd d Z d d Z dddZ ddZ ddZ ddZdddZeddZdS)r z+Class representing a point on the 2D plane.cCs|j|j|j|j|jdS)z.Adds the coordinates of a point to another onexyr?rnrorFrrr __add__sz Point.__add__cCs|j|j|j|j|jdS)z3Subtracts the coordinates of a point to another onermrprFrrr __sub__sz Point.__sub__cCs|j|j||j|dS)z&Multiplies the coordinates by a scalarrmrpr'Zscalarrrr __mul__sz Point.__mul__cCs|j|j||j|dS)z#Divides the coordinates by a scalarrmrprsrrr __div__sz Point.__div__cCst|t|j|jfS)zyReturns the polar coordinate representation of the point. @return: the radius and the angle in a tuple. )r"rrornr*rrr as_polarszPoint.as_polarcCs.|j|j|j|j}}||||dS)zReturns the distance of the point from another one. Example: >>> p1 = Point(5, 7) >>> p2 = Point(8, 3) >>> p1.distance(p2) 5.0 ?rm)r'rGr9r;rrr distances zPoint.distancerwcCs>t|}|j|jd||j||jd||j|dS)a Linearly interpolates between the coordinates of this point and another one. @param other: the other point @param ratio: the interpolation ratio between 0 and 1. Zero will return this point, 1 will return the other point. ?rm)rr?rnro)r'rGratiorrr interpolates zPoint.interpolatecCs|jd|jddS)zPReturns the length of the vector pointing from the origin to this point.rrwrmr*rrr lengthsz Point.lengthcCs<|}|dkr"|j|j|jdS|j|j||j|dS)z|Normalizes the coordinates of the point s.t. its length will be 1 after normalization. Returns the normalized point.rrm)r|r?rnro)r'r"rrr normalizedszPoint.normalizedcCs|jd|jdS)zXReturns the squared length of the vector pointing from the origin to this point.rrmr*rrr sq_lengthszPoint.sq_lengthrcCsJ|s|St|j|j|j|j}||j|t||j|t|S)zZReturns the point that is at a given distance from this point towards another one.)rrornr?rr)r'rGrxanglerrr towardssz Point.towardscCs||t||t|S)zConstructs a point from polar coordinates. C{radius} is the distance of the point from the origin; C{angle} is the angle between the X axis and the vector pointing to the point from the origin. )rr)clsradiusrrrr FromPolarszPoint.FromPolarN)rw)r)rOrWrXrYrqrrrt__rmul__rurvrxr{r|r}r~r classmethodrrrrr r s  r Z_PointrnrocCsdd|D}ddt|ddD}dd|D}|gt|}tt|D]6}|dkr\dn|d }|||||g}t|||<qL|S) aGiven a list of points and a desired corner radius, returns a list containing proposed corner radii for each of the points such that it is ensured that the corner radius at a point is never larger than half of the minimum distance between the point and its neighbors. cSsg|] }t|qSr)r )rpointrrr rC sz*calculate_corner_radii..cSsg|]\}}||qSrr)ruvrrr rC!sT)ZcircularcSsg|]}|dqS)r)r|)rZsiderrr rC"srr)rr"ranger4)ZpointsZ corner_radiusZ side_vecsZhalf_side_lengthsZ corner_radiiidxZprev_idxZradiirrr r sr cCst||||S)zCComputes the Euclidean distance between points (x1,y1) and (x2,y2).)r)x1y1x2y2rrr r +sr c Csd|d|d|d|d|d|dd|||d|} d|d|d|d|d|d|dd|||d|} | | fS)zEvaluates the Bezier curve from point (x0,y0) to (x3,y3) via control points (x1,y1) and (x2,y2) at t. t is typically in the range [0; 1] such that 0 returns (x0, y0) and 1 returns (x3, y3). ryr=@rr) x0y0rrrrx3y3tZxtZytrrr r0s"  rcCsd||d|d||d||d|d||f}|d|d|d|||d|d|d||f}||fS)zyHelper function that calculates the Bezier control points for a curved edge that goes from (x1, y1) to (x2, y2). rrrwr)rrrrZ curvatureZaux1Zaux2rrr rDs  r c  CsN|d} t||||} t|}d} d|| } t||||||||| \}}d}t||||}d}t||| kr6|| kr6||dk||dkkr| | d}n6t||t||kr| | | d}n | | | }|dkrdn|dkrdn|}| |} } |}t||||||||| \}}t||||}|d7}q`t||||||||| S)zBinary search solver for finding the intersection of a Bezier curve and a circle centered at the curve's end point. Returns the x, y coordinates of the intersection point. g4@ryrr7r)r rrabs)rrrrrrrrrZmax_iter precisionZsource_target_distancet0t1Zxt1Zyt1Z distance_t0Z distance_t1counterZt_newrrr rSs,    rFc Csvddddddddddd }ddg|}|j|||dddg|}|j|||d|||}|dkrrtd ||S) aTries to interpret a string as an orientation value. The following basic values are understood: ``left-right``, ``bottom-top``, ``right-left``, ``top-bottom``. Possible aliases are: - ``horizontal``, ``horiz``, ``h`` and ``lr`` for ``left-right`` - ``vertical``, ``vert``, ``v`` and ``tb`` for top-bottom. - ``lr`` for ``left-right``. - ``rl`` for ``right-left``. ``reversed_horizontal`` reverses the meaning of ``horizontal``, ``horiz`` and ``h`` to ``rl`` (instead of ``lr``); similarly, ``reversed_vertical`` reverses the meaning of ``vertical``, ``vert`` and ``v`` to ``bt`` (instead of ``tb``). Returns one of ``lr``, ``rl``, ``tb`` or ``bt``, or throws ``ValueError`` if the string cannot be interpreted as an orientation. lrrltbbt) z left-rightz right-left top-bottom bottom-topztop-downz bottom-uprrtdZbu) horizontalZhorizh)verticalZvertr)rrrrzunknown orientation: %s)updategetr&)r.Zreversed_horizontalZreversed_verticalaliasesdirresultrrr rs&    rcurvedcCs.tt}|jD]@}|j\}}||kr:|||f|jq|||f|jq|g|}|D]}t|dkrxqft|ddkrd|| <dt|d} | d} } t |D]X\} } |j| }|j |j kr| | || <n | | || <| ddkr| | 7} | d9} qqf|dkr |S||j|<dS)aCalculates curvature values for each of the edges in the graph to make sure that multiple edges are shown properly on a graph plot. This function checks the multiplicity of each edge in the graph and assigns curvature values (numbers between -1 and 1, corresponding to CCW (-1), straight (0) and CW (1) curved edges) to them. The assigned values are either stored in an edge attribute or returned as a list, depending on the value of the I{attribute} argument. @param graph: the graph on which the calculation will be run @param attribute: the name of the edge attribute to save the curvature values to. The default value is C{curved}, which is the name of the edge attribute the default graph plotter checks to decide whether an edge should be curved on the plot or not. If I{attribute} is C{None}, the result will not be stored. @param default: the default curvature for single edges. Zero means that single edges will be straight. If you want single edges to be curved as well, try passing 0.5 or -0.5 here. @return: the list of curvature values if I{attribute} is C{None}, otherwise C{None}. rrrr7rN) rlistesr%appendindexZecountvaluesr"pop enumeratesourcetarget)graph attributedefaultZmultiplicitiesedgerrrZeidsZcurveZdcurvesignrZeidrrr rs2           rN)r)FF)rr)rY collectionsrmathrrrrtypingrZ igraph.utilsr__all__r r r_rr r r rrrrrrrrr s&   Z."_ - 0