3 }\B @sddlZddlZddlmZmZddlmZGdddeZdddZ y ej j Z Wne k rjdZ YnXdd d Zdd d Zd dZdS)N)Image ImageColor) isStringTypec@seZdZd+ddZddZd,ddZd-d d Zd.d d Zd/d dZd0ddZ d1ddZ d2ddZ d3ddZ d4ddZ d5ddZd6ddZddZdd Zd7d!d"Zd8d%d&Zd9d'd(Zd:d)d*ZdS); ImageDrawNcCs|j|jr|jd}|dkr(|j}||jkrR|dkrJ|jdkrJd}ntd|dkrd|j|_nd|_|j|_tjj |j||_ ||_|d kr|j j d||_ n|j j d ||_ |dkrd |_ nd |_ d|_ d|_dS)a Create a drawing instance. :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. rNZRGBAZRGBrz mode mismatchPIF1L)rr )r rrr )loadreadonlyZ_copymode ValueErrorpaletteimrcoredrawdraw_inkinkfontmodefillfont)selfrrZblendre/private/var/folders/pf/wv4htv3x0qs2c2mp0dnn0kchsvlck3/T/pip-install-i584jbuk/Pillow/PIL/ImageDraw.py__init__1s0   zImageDraw.__init__cCs"|jsddlm}|j|_|jS)zH Get the current default font. :returns: An image font.r) ImageFont)rrZ load_default)rrrrrgetfontZs  zImageDraw.getfontcCs|dkr&|dkr&|jr|j}q|j}n|dk rtt|rDtj||j}|jrdt|tj  rd|jj|}|j j ||j}|dk rt|rtj||j}|jrt|tj  r|jj|}|j j ||j}||fS)N) rrrrZgetcolorrr isinstancenumbersNumberrr)rrrrrr_getinkes"  zImageDraw._getinkrcCs.|j|\}}|dk r*|jj|||||dS)z Draw an arc.N)r$rZdraw_arc)rxystartendrwidthrrrrarczsz ImageDraw.arccCs@|j|j|\}}|dkr"|}|dk r<|jj||j|dS)zDraw a bitmap.N)r r$r draw_bitmapr)rr%bitmaprrrrrr+s zImageDraw.bitmapcCsV|j||\}}|dk r,|jj||||d|dk rR||krR|jj||||d|dS)z Draw a chord.Nrr)r$rZ draw_chord)rr%r&r'routliner(rrrrchords zImageDraw.chordcCsN|j||\}}|dk r(|jj||d|dk rJ||krJ|jj||d|dS)zDraw an ellipse.Nrr)r$rZ draw_ellipse)rr%rr,r(rrrrellipses zImageDraw.ellipsecs|j|d}|dk r|jj|||dko6dkrxtdt|dD]}||}dd||d|f|||dffD}|d|dkrqPfdd } |d|dkr|dd |dkp|d|dko|dd |dk} |dd d|dd df|dd d|dd dfg} | r`|dd |dd } } n|dd |dd } } |j| | d | d |d krP| r| ||dd || ||dd g}n&| ||dd || ||dd g}|j||ddqPWdS)z6Draw a line, or a connected sequence of line segments.rNZcurvercSs>g|]6\}}tjtj|d|d|d|ddqS)rrih)mathdegreesatan2).0r&r'rrr sz"ImageDraw.line..csZ|\}}|d8}dd}tdd||tjtj|f||tjtj|ffDS)NZrcSs0g|](\}}||dkr tj|ntj|qS)r)r0floorceil)r3pZp_drrrr4sz:ImageDraw.line..coord_at_angle..)tupler0cosradianssin)ZcoordZanglexyZdistance)r(rrcoord_at_angles z&ImageDraw.line..coord_at_angler6r5)r()r$rZ draw_linesrangelenpiesliceline)rr%rr(ZjointripointZanglesr@ZflippedZcoordsr&r'Z gapCoordsr)r(rrGs< $ &*zImageDraw.linecCsT|j|j||\}}|dk r0|jj||d|dk rP||krP|jj||ddS)z(Experimental) Draw a shape.Nrr)closer$rZ draw_outline)rshaperr,rrrrrKs zImageDraw.shapecCsV|j||\}}|dk r,|jj||||d|dk rR||krR|jj||||d|dS)zDraw a pieslice.Nrr)r$rZ draw_pieslice)rr%r&r'rr,r(rrrrrFs zImageDraw.pieslicecCs(|j|\}}|dk r$|jj||dS)z#Draw one or more individual pixels.N)r$rZ draw_points)rr%rrrrrrIszImageDraw.pointcCsL|j||\}}|dk r(|jj||d|dk rH||krH|jj||ddS)zDraw a polygon.Nrr)r$rZ draw_polygon)rr%rr,rrrrpolygons zImageDraw.polygoncCsN|j||\}}|dk r(|jj||d|dk rJ||krJ|jj||d|dS)zDraw a rectangle.Nrr)r$rZdraw_rectangle)rr%rr,r(rrrr rectangles zImageDraw.rectanglecCst|trdnd}||kS)z Draw text.  )r!str)rtextsplit_characterrrr_multiline_checkszImageDraw._multiline_checkcCst|trdnd}|j|S)NrNrO)r!rPsplit)rrQrRrrr_multiline_splitszImageDraw._multiline_splitc Os|j|r$|j|||||f||S|j|\}}|dkrB|j}|dkrN|}|dk ry>|j||jf||\} } |d| d|d| df}WnNtk ry|j||jf||} Wntk r|j|} YnXYnX|j j || |dS)Nrr) rSmultiline_textr$r Zgetmask2rAttributeErrorZgetmask TypeErrorrr*) rr%rQrranchorargskwargsrmaskoffsetrrrrQs&   $zImageDraw.textr/leftc  Csg} d} |j|} |jd|dd|}x8| D]0}|j|||| | d\}}| j|t| |} q.W|\}}xt| D]\}}|dkrn@|dkr|| | |d7}n"|d kr|| | |7}ntd |j||f|||||| | d||7}|d}qtWdS) NrA)rr) directionfeatureslanguager^centerg@rightz)align must be "left", "center" or "right")rUtextsizeappendmax enumeraterrQ)rr%rQrrrYspacingZalignr`rarbZwidths max_widthlines line_spacingrG line_width line_heightr^topidxrrrrVs0     zImageDraw.multiline_textcCs>|j|r|j||||||S|dkr.|j}|j||||S)z*Get the size of a given string, in pixels.N)rSmultiline_textsizer getsize)rrQrrir`rarbrrrre9s    zImageDraw.textsizec Cshd}|j|}|jd|dd|} x.|D]&} |j| |||||\} } t|| }q*W|t|| |fS)Nrr_)rr)rUrergrE) rrQrrir`rarbrjrkrlrGrmrnrrrrqDs    zImageDraw.multiline_textsize)N)N)Nr)N)NNr)NNr)NrN)NN)NNr)N)NN)NNr)NNN)NNNr/r^NNN)Nr/NNN)Nr/NNN)__name__ __module__ __qualname__rr r$r)r+r-r.rGrKrFrIrLrMrSrUrQrVrerqrrrrr/s. )     ;       rc Cs*y |j|Stk r$t||SXdS)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. N)getdrawrWr)rrrrrDrawQs  rwc Csbd}| sd|kr8yddlm}Wntk r6YnX|dkrLddlm}|rZ|j|}||fS)a (Experimental) A more advanced 2D drawing interface for PIL images, based on the WCK interface. :param im: The image to draw in. :param hints: An optional list of hints. :returns: A (drawing context, drawing resource factory) tuple. NZnicestr) _imagingagg) ImageDraw2)rrx ImportErrorryrw)rZhintshandlerrrrrvis   rvcCsb|j}|\}}y.|||f}t|||kr0dS||||f<Wnttfk rVdSX||fh} t} x| r\t} x| D]\}}x|d|f|d|f||df||dffD]\} } | | f| krqy|| | f}Wnttfk rYqX| j| | f|dkrt|||k}n||ko*||k}|r||| | f<| j| | fqWq|W| } | } qjWdS)a (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 _color_diffr IndexErrorsetadd)imager%valueZborderZthreshZpixelr>r?Z backgroundZedgeZ full_edgeZnew_edgestr9rrrr floodfills<  6   rcs<ttr,tfddtdtDStSdS)zJ Uses 1-norm distance to calculate difference between two values. cs g|]}t||qSr)abs)r3rH)color1color2rrr4sz_color_diff..rN)r!r:sumrDrEr)rrr)rrrr|s "r|)N)NN)Nr)r0r"rrrZ_utilrobjectrrwrr,ZOutlinerWrvrr|rrrr!s $     7