o h]@s UddlmZddlZddlZddlmZddlmZddlm Z m Z m Z m Z m Z ddlmZmZddlmZdd lmZejjZd ed <d ZerQdd lmZmZe eeedfefZ GdddZ!d6d7ddZ" d8d9ddZ#  d:d;d(d)Z$d) annotationsN)Sequence) ModuleType)AnyAnyStrCallableUnioncast)Image ImageColor) deprecate)Coordsz!Callable[[], Image.core._Outline]OutlineF) ImageDraw2 ImageFont.c@seZdZUdZded<dqdrd d Zdsd dZdtddZ dqduddZ  dvdwd!d"Z dqdxd%d&Z   dydzd(d)Z   dyd{d*d+Z   dyd|d.d/Z  0 d}d~d2d3Z  ddd6d7Z   dydzd8d9Zdqdd:d;Z   dyd{dsz"ImageDraw.line..rer c Ss>g|]\}}tt|d|d|d|ddqS)rr h)mathdegreesatan2)rgrHrJr2r2r3ris (coordr^anglerIrtuple[float, ...]c sZ|\}}|d8}dd}tdd||tt|f||tt|ffDS)NZrer css4|]\}}||dkrt|nt|VqdS)rN)rkfloorceil)rgpp_dr2r2r3 s  z9ImageDraw.line..coord_at_angle..)rCrkcosradianssin)rnroxydistancerKr2r3coord_at_angles z&ImageDraw.line..coord_at_anglerqr})rnr^rorIrrp) rFr* draw_linesr@listrCr rrIrangelenpiesliceline)r/rGr.rKrbr,pointsrhpointanglesr~flippedcoordsrHrJ gap_coordsr2)rKrGr3rsZ  ""&&  zImageDraw.lineshapeImage.core._OutlinecCs\||||\}}|dur|j||d|dur*||kr,|j||ddSdSdS)z(Experimental) Draw a shape.Nr r)closerFr* draw_outline)r/rr.rRr,rVr2r2r3r9szImageDraw.shapec CrS)zDraw a pieslice.Nr r)rFr* draw_pieslicerUr2r2r3rGrXzImageDraw.pieslicecCs,||\}}|dur|j||dSdS)z#Draw one or more individual pixels.N)rFr* draw_points)r/rGr.r,r2r2r3rWszImageDraw.pointc Cs|||\}}|dur|j||d|dur`||krb|dkrd|dkr/|j||d|dS|jdurftd|jj}|dd}t|} | j||d|j||d|dd|jdSdSdSdSdS)zDraw a polygon.Nr rr!re)rFr* draw_polygonrr newsizeDraw) r/rGr.rRrKr,rVmaskmask_inkr*r2r2r3polygon]s " zImageDraw.polygonbounding_circle!Sequence[Sequence[float] | float]n_sidesrotationcCs t|||}|||||dS)zDraw a regular polygon.N)!_compute_regular_polygon_verticesr)r/rrrr.rRrKrGr2r2r3regular_polygonus zImageDraw.regular_polygoncCrY)zDraw a rectangle.Nr r)rFr*draw_rectangler[r2r2r3 rectangler]zImageDraw.rectanglecornersr$tuple[bool, bool, bool, bool] | Nonec snt|dttfrtttt|\\ \ n ttt|\ kr.d}t| kr8d}t|dur>d|dtt  t  t  d\tr dkrh  dkrv rr |||Sdkst s |||St d} ||\d f d d } durT| dr҈j |d |dfdn |d|dkrj|d |d fdsTsT | g} dr| d|d7<dr| d|d8<j| d | g} dr=| d|d7<drL| d|d8<j| ddur/kr1dkr3| dsʈ dg} dr| d|d7<dr| d|d8<j| d d g} dr| d|d7<dr| d|d8<j| ds5 d g} dr| d|d7<dr| d|d8<j| d d g} dr| d|d7<dr%| d|d8<j| ddSdSdSdSdS)zDraw a rounded rectangle.rz&x1 must be greater than or equal to x0z&y1 must be greater than or equal to y0N)TTTTre)FFr rboolrrc s*r  fddf  fddff}nZr:  fddf  fddff}n=tfddt  fddf  fddf   fddf  fddffD}|D]}|rjj|dfqyjj|fqydS) Nrrjrrqc3s |] \}}|r|VqdSr?r2)rgrhpartrr2r3rvs zDImageDraw.rounded_rectangle..draw_corners..r )rC enumerater*rrM)rpartsr rdrVfull_xfull_yr,r/rKx0x1y0y1r2r3 draw_cornerss,  z1ImageDraw.rounded_rectangle..draw_cornersTrF)rrrr)r@rrCr rrIr&roundallr\anyrrLrFr*r)r/rGr_r.rRrKrr1rrleftrighttopbottomr2rr3rounded_rectangles  $ "*(              zImageDraw.rounded_rectangletextrrcCst|trdnd}||vS)N  )r@rA)r/rsplit_characterr2r2r3_multiline_checkszImageDraw._multiline_checkrdrFtuple[float, float]anchorspacingalignrA directionfeatureslist[str] | Nonelanguage stroke_width stroke_fillembedded_colorargsrkwargscsr jdvr d}t|durd r/ |||| |  Sdfd d }dd f dd }||}|duryd}| r^| dur\|| n|}|durs||| ||krq||dSdS||dSdS)z Draw text.rr3Embedded color supported only in RGB and RGBA modesNr:r.r=rrLcs*|\}}|dur|dusJ|S|Sr?)rF)r.r,rV)r/r2r3getinkQs  zImageDraw.text..getinkrr,rrIrc sj}|dkr r d}g}tdD] }|t |qt ddt ddf}z*j |gR|d||d\}}|d|d|d|dg}Wn/tyzj |||gRd|i}Wnt y }YnwYnw|dkr|| d}}t d |d} | d| |\} } jdurĈj|| | | |jd| |jdf|dSdSj|||dS) Nrrrer T)rrrr stroke_filledrr,rHrHrrh)r-rappendrLrkmodfgetmask2AttributeErrorgetmask TypeErrorgetbandstructpackfillbandrpasterr*rQ) r,rrrnrhrHroffsetcolor ink_alpharzr{ rrrrrrrrr/rrGr2r3 draw_textXsx  $  $      "z!ImageDraw.text..draw_text)r.r=rrL)r)r,rLrrIrr)rr&r<getrmultiline_text)r/rGrr.rrrrrrrrrrrrr1rrr, stroke_inkr2rr3r!sF $7   zImageDraw.text}tuple[ImageFont.ImageFont | ImageFont.FreeTypeFont | ImageFont.TransposedFont, list[tuple[tuple[float, float], str, AnyStr]]]c  s"|dur dkr dnd}nt|dkrd} t| |ddvr)dkr)d} t| dur2| |t|tr;d nd }jd d | d d| |}|d}g}dkrr|d}|D]}|||f||f||7}q]|fSg}d}|D]}j|d}||t ||}qx|ddkr|t|d|d8}n|ddkr|t|d|8}t |D]\}}|d}|||}|dvrn|dkr||d7}n|dkr||7}nd} t| |dkre|dkre|t|dkre|t|trdnd}t|dkre|ddkr||d8}n |ddkr$||8}fdd|D}d|d}|t |}t |D]\}}|||f||f||||t|d7}qB||7}q|ddkrs||d8}n |ddkr~||8}|||f||f||7}q|fS)Nttbltlarez#anchor must be a 2 character stringr tbz'anchor not supported for multiline textrr)rrA)rrrrrrrmg@r)rjustifycenterrz4align must be "left", "center", "right" or "justify"r  rc s"g|] }j|dqS)r) textlength)rgwordrrrrrr/r2r3ris z5ImageDraw._prepare_multiline_text..l) rr&r<splitr@rAtextbboxrrmaxrsum)r/rGrrrrrrrrrrr:r1lines line_spacingrrrrwidths max_width line_widthidxwidth_differencewords word_widths word_anchorrhrr2rr3_prepare_multiline_texts   L            z!ImageDraw._prepare_multiline_text)r:cCsX||||||||| | | | | \}}|D]\}}}|j||||||| | | | | d qdS)N)rrrrrr)rr)r/rGrr.rrrrrrrrrrr:rrr2r2r3rs:zImageDraw.multiline_textc Csb||r d}t||r|jdvrd}t||dur!||}|r%dn|j} ||| |||S)z@Get the length of a given string, in pixels with 1/64 precision.z&can't measure length of multiline textrrNr)rr&rr<r- getlength) r/rrrrrrr:r1rr2r2r3rSs  zImageDraw.textlength!tuple[float, float, float, float]c  Cs| r |jdvr d} t| |dur|| }||r*|||||||||| | | S| r.dn|j}|||||| | |}|d|d|d|d|d|d|d|dfS) z2Get the bounding box of a given string, in pixels.rrNrrr rer)rr&r<rmultiline_textbboxr-getbbox)r/rGrrrrrrrrrrr:r1rbboxr2r2r3rps0  <zImageDraw.textbboxc Cs|||||||||| | | | \}} d}| D]<\}}}|j||||||| | | d }|dur0|}qt|d|dt|d|dt|d|dt|d|df}q|dure|d|d|d|dfS|S)N)rrrrrrr rer)rrminr)r/rGrrrrrrrrrrr:rr r bbox_liner2r2r3rsJ zImageDraw.multiline_textbboxr?)rrrrrr)rr5)r:r;rr5)r,r=r.r=rr>)Nr ) rGrrHrIrJrIr.r=rKrLrr)rGrOrPrr.r=rr)NNr )rGrrHrIrJrIr.r=rRr=rKrLrr) rGrr.r=rRr=rKrLrr) rGr^r_rIr.r=rRr=rKrLrr)NrN) rGrr.r=rKrLrbrrrNN)rrr.r=rRr=rr)rGrr.r=rr)rNNr )rrrrLrrIr.r=rRr=rKrLrr)rGrr_rIr.r=rRr=rKrLrrrr)rrrr) NNNrdrNNNrNF) rGrrrr.r=rrrrrrIrrArrrrrrrrIrr=rrrrrrrr)rGrrrrrrrrrIrrArrrrrrrrIrrr:r;rr)rGrrrr.r=rrrrrrIrrArrrrrrrrIrr=rrr:r;rr)NNNNF)rrrrrrrrrrrrr:r;rrI) NNrdrNNNrF)rGrrrrrrrrrIrrArrrrrrrrIrrr:r;rr)__name__ __module__ __qualname__r__annotations__r4r9r<rFrNrPrWr\rarrrrrrrrrrrrrrrr2r2r2r3r=s   *        P         =&;rrrrrrcCs.zt|d|WStyt||YSw)a A simple 2D drawing interface for PIL images. :param im: The image to draw in. :param mode: Optional mode to use for color values. For RGB images, this argument can be RGB or RGBA (to blend the drawing into the image). For all other modes, this argument must be the same as the image mode. If omitted, the mode defaults to the mode of the image. getdraw)getattrrr)rrr2r2r3rs   rImage.Image | Nonehintsr)tuple[ImageDraw2.Draw | None, ModuleType]cCs<|dur tddddlm}|dur||nd}||fS)z :param im: The image to draw in. :param hints: An optional list of hints. Deprecated. :returns: A (drawing context, drawing resource factory) tuple. Nz'hints' parameter r )r)r r7rr)rrrr*r2r2r3rs   rimagerGtuple[int, int]valuefloat | tuple[int, ...]borderfloat | tuple[int, ...] | NonethreshrIrc Csn|}|dus J|\}}z|||f}t|||krWdS||||f<Wn ttfy2YdSw||fh} t} | rt} | D]j\}}|d|f|d|f||df||dffD]O\} } | | f| vsn| dksn| dkroq\z|| | f}Wn ttfyYq\w| | | f|durt|||k}n|||fv}|r||| | f<| | | fq\qB| } | } | s=dSdS)a .. warning:: This method is experimental. Fills a bounded region with a given color. :param image: Target image. :param xy: Seed position (a 2-item coordinate tuple). See :ref:`coordinate-system`. :param value: Fill color. :param border: Optional border value. If given, the region consists of pixels with a color different from the border color. If not given, the region consists of pixels having the same color as the seed pixel. :param thresh: Optional threshold value which specifies a maximum tolerable difference of a pixel value from the 'background' in order for it to be replaced. Useful for filling regions of non-homogeneous, but similar, colors. Nr r)r# _color_diffr& IndexErrorsetadd)rrGrrrpixelrzr{ backgroundedge full_edgenew_edgestrtr.r2r2r3 floodfillsH    4   r)rrrrLrlist[tuple[float, float]]cst|ts d}t||dkrd}t|t|ttfs"d}t|t|dkrDtdd|Ds7d}t|ttt t|^nPt|dkrt|d ttfrtd d|d Drgt|d tt fsmd}t|t|d dkr{d }t|ttt t|d tt |d nd }t|d krd}t|t|tt fsd}t|d#fdd d$fdd d%dd}|||}fd d!|DS)&u Generate a list of vertices for a 2D regular polygon. :param bounding_circle: The bounding circle is a sequence defined by a point and radius. The polygon is inscribed in this circle. (e.g. ``bounding_circle=(x, y, r)`` or ``((x, y), r)``) :param n_sides: Number of sides (e.g. ``n_sides=3`` for a triangle, ``6`` for a hexagon) :param rotation: Apply an arbitrary rotation to the polygon (e.g. ``rotation=90``, applies a 90 degree rotation) :return: List of regular polygon vertices (e.g. ``[(25, 50), (50, 50), (50, 25), (25, 25)]``) How are the vertices computed? 1. Compute the following variables - theta: Angle between the apothem & the nearest polygon vertex - side_length: Length of each polygon edge - centroid: Center of bounding circle (1st, 2nd elements of bounding_circle) - polygon_radius: Polygon radius (last element of bounding_circle) - angles: Location of each polygon vertex in polar grid (e.g. A square with 0 degree rotation => [225.0, 315.0, 45.0, 135.0]) 2. For each angle in angles, get the polygon vertex at that angle The vertex is computed using the equation below. X= xcos(φ) + ysin(φ) Y= −xsin(φ) + ycos(φ) Note: φ = angle in degrees x = 0 y = polygon_radius The formula above assumes rotation around the origin. In our case, we are rotating around the centroid. To account for this, we use the formula below X = xcos(φ) + ysin(φ) + centroid_x Y = −xsin(φ) + ycos(φ) + centroid_y zn_sides should be an intrzn_sides should be an int > 2z$bounding_circle should be a sequencecs|] }t|ttfVqdSr?r@rLrIrfr2r2r3rv}sz4_compute_regular_polygon_vertices..z0bounding_circle should only contain numeric datarercsr+r?r,rfr2r2r3rvs r zBbounding_circle centre should contain 2D coordinates (e.g. (x, y))z[bounding_circle should contain 2D coordinates and a radius (e.g. (x, y, r) or ((x, y), r) )z$bounding_circle radius should be > 0z"rotation should be an int or floatr list[float]rlrIrrc st|dttd||dttd|ddt|dttd||dttd|ddfS)Nrrjr re)rrkrwrxry)rrl)centroidr2r3_apply_rotations"z:_compute_regular_polygon_vertices.._apply_rotationrocsdg}||SNrr2)ro start_point)r/polygon_radiusr2r3_compute_polygon_vertexs zB_compute_regular_polygon_vertices.._compute_polygon_vertexrrLrcSsPg}d|}dd||}t|D]}||||7}|dkr%|d8}q|S)Nrjrg?)rr)rrrrl current_angle_r2r2r3 _get_angless  z6_compute_regular_polygon_vertices.._get_anglescsg|]}|qSr2r2)rgro)r3r2r3risz5_compute_regular_polygon_vertices..N)rr-rlrIrr)rorIrr)rrLrrIrr-) r@rLrr&rrCrrr rI)rrrr1r6rr2)r/r3r.r2r3rEsR +   rcolor1color2csHt|tr|n|ft|tr|n|ftfddttDS)zJ Uses 1-norm distance to calculate difference between two values. c3s$|] }t||VqdSr?)absrffirstsecondr2r3rvs"z_color_diff..)r@rCrrr)r7r8r2r:r3rs rr?)rrrrrrr )rrrrrrr0) rrrGrrrrrrrIrr)rrrrLrrIrr*)r7rr8rrrI)' __future__rrkrcollections.abcrtypesrtypingrrrrr r7r r _deprecater _typingrr)rRrr TYPE_CHECKINGrrrIrCrLrA_Inkrrrr)rrr2r2r2r3s@     -  A