3 `JZI5 @sddlZddlmZmZddlmZGdddeZdddZy ej j Z Wne k rbdZ YnXdd d Z dd d Zd dZdS)N)Image ImageColor) isStringTypec@seZdZd+ddZddZd,ddZd-dd Zd.d d Zd/d d Zd0ddZ 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)selfrrZblendr//usr/lib/python3/dist-packages/PIL/ImageDraw.py__init__0s0   zImageDraw.__init__cCs"|jsddlm}|j|_|jS)zH Get the current default font. :returns: An image font.r) ImageFont)rrZ load_default)rrrrrgetfontYs  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_getinkds"  zImageDraw._getinkcCs,|j|\}}|dk r(|jj||||dS)z Draw an arc.N)r$rZdraw_arc)rxystartendrrrrrarcysz ImageDraw.arccCs@|j|j|\}}|dkr"|}|dk r<|jj||j|dS)zDraw a bitmap.N)r r$r draw_bitmapr)rr%bitmaprrrrrr*s zImageDraw.bitmapcCsL|j||\}}|dk r,|jj||||d|dk rH|jj||||ddS)z Draw a chord.Nrr)r$rZ draw_chord)rr%r&r'routlinerrrrchords zImageDraw.chordcCsD|j||\}}|dk r(|jj||d|dk r@|jj||ddS)zDraw an ellipse.Nrr)r$rZ draw_ellipse)rr%rr+rrrrellipses zImageDraw.ellipsercCs*|j|\}}|dk r&|jj|||dS)z6Draw a line, or a connected sequence of line segments.N)r$rZ draw_lines)rr%rwidthrrrrlineszImageDraw.linecCsL|j|j||\}}|dk r0|jj||d|dk rH|jj||ddS)z(Experimental) Draw a shape.Nrr)closer$rZ draw_outline)rshaperr+rrrrr1s zImageDraw.shapecCsL|j||\}}|dk r,|jj||||d|dk rH|jj||||ddS)zDraw a pieslice.Nrr)r$rZ draw_pieslice)rr%r&r'rr+rrrrpieslices zImageDraw.pieslicecCs(|j|\}}|dk r$|jj||dS)z#Draw one or more individual pixels.N)r$rZ draw_points)rr%rrrrrpointszImageDraw.pointcCsD|j||\}}|dk r(|jj||d|dk r@|jj||ddS)zDraw a polygon.Nrr)r$rZ draw_polygon)rr%rr+rrrrpolygons zImageDraw.polygoncCsD|j||\}}|dk r(|jj||d|dk r@|jj||ddS)zDraw a rectangle.Nrr)r$rZdraw_rectangle)rr%rr+rrrr rectangles zImageDraw.rectanglecCst|trdnd}||kS)z Draw text.  )r!str)rtextsplit_characterrrr_multiline_checkszImageDraw._multiline_checkcCst|trdnd}|j|S)Nr6r7)r!r8split)rr9r:rrr_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) r;multiline_textr$r Zgetmask2rAttributeErrorZgetmask TypeErrorrr)) rr%r9rranchorargskwargsrmaskoffsetrrrr9s$  $zImageDraw.textleftc  Csg} d} |j|} |jd|dd|} x0| D](}|j||\}}| j|t| |} q.W|\}}xt| D]\}}|dkr~nD|dkr|| | |d7}n&|dkr|| | |7}n d std |j||f|||||| d || 7}|d}qlWdS) NrA)rrrGcenterg@rightFz)align must be "left", "center" or "right") directionfeatures)r=textsizeappendmax enumerateAssertionErrorr9)rr%r9rrrAspacingZalignrKrLZwidths max_widthlines line_spacingr/ line_width line_heightrGtopidxrrrr>s*     zImageDraw.multiline_textcCs:|j|r|j|||||S|dkr,|j}|j|||S)z*Get the size of a given string, in pixels.N)r;multiline_textsizer getsize)rr9rrRrKrLrrrrMs   zImageDraw.textsizec Csbd}|j|}|jd|dd|}x,|D]$} |j| ||||\} } t|| }q*W|t||fS)NrrH)rr)r=rMrOlen) rr9rrRrKrLrSrTrUr/rVrWrrrrZ s    zImageDraw.multiline_textsize)N)N)N)N)NN)NN)Nr)NN)NN)N)NN)NN)NNN)NNNrFrGNN)NrFNN)NrFNN)__name__ __module__ __qualname__rr r$r(r*r,r-r/r1r2r3r4r5r;r=r9r>rMrZrrrrr.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)getdrawr?r)rrrrrDraws  rac 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)rrb ImportErrorrcra)rZhintsZhandlerrrrr`-s   r`cCs|j}|\}}y.|||f}t|||kr0dS||||f<Wnttfk rVdSX||fg} |dkrx| rg} x| D]\}}x|d|f|d|f||df||dffD]V\} } y|| | f} Wntk rYqXt| ||kr||| | f<| j| | fqWq~W| } qnWnx| rg} x| D]\}}x|d|f|d|f||df||dffD]`\} } y|| | f} Wntk rYn0X| |krd| |krd||| | f<| j| | fqdWq.W| } qWdS)a (experimental) Fills a bounded region with a given color. :param image: Target image. :param xy: Seed position (a 2-item coordinate tuple). :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 IndexErrorrN)Zimager%valueZborderZthreshZpixelxyZ backgroundZedgeZnewedgestprrr floodfillEsF   6  6 rmcCs<t|d|dt|d|dt|d|dS)zN Uses 1-norm distance to calculate difference between two rgb values. rr)abs)Zrgb1Zrgb2rrrre~sre)N)NN)Nr)r"rrrZ_utilrobjectrrarr+ZOutliner?r`rmrerrrr!s h     9