Aluode/PerceptionLabPortable
0
1o
2 ��Yiˑ � @ s� U d dl mZ d dlZd dlZd dlmZ d dlmZ ddlm Z m3Z4m Z dZ e rMd dlm
Z
d d lmZ d d5lmZmZ dd lmZmZ dd lmZmZ e jjZd
ed< G dd� d�Zd4d5dd�Z d4d6dd�Z d7d8d&d'�Z d9d.d/�Z d:d2d3�Z dS );� )� annotationsN)�Sequence)�cast� )�Image�6ImageColor� ImageTextF)�Callable)�7ModuleType)�Any�AnyStr)�8ImageDraw2� ImageFont)�Coords�_Inkz!Callable[[], Image.core._Outline]�Outlinec @ s� e Zd ZU dZded< dmdnd9d �Zdod
d�Zdpdd�Z dmdqdd�Z drdsd!d"�Z dmdtd%d&�Z10 dudvd(d)�Z dudwd*d+�Z dudxd.d/�Z
0 dydzd2d3�Z d{d|d6d7�Z dudvd8d9�Zdmd}d:d;�Z dudwd<d=�Z 0 d~ddBdC�Z dudwdDdE�Z 0 d~ddF�d�dIdJ�Z K L 0 Md�d�d`da�Z K L 0 Md�ddb�d�ddde�Z Md�ddb�d�dfdg�Z K L 0 Md�ddb�d�didj�Z K L 0 Md�ddb�d�dkdl�ZdS )�� ImageDrawN�NImageFont.ImageFont | ImageFont.FreeTypeFont | ImageFont.TransposedFont | None�font�im� Image.Image�mode�11str | None�return�Nonec C s� |� � d}|du r
|j}||jkr$|dkr |jdkr d}nd}t|��|dkr-|j| _nd| _|| _|j| _tj�| j|�| _|| _|dv rO| j� d�| _12n| j� d �| _13|d14v r^d | _ nd | _ d
| _ dS )a� 15 Create a drawing instance.16 17 :param im: The image to draw in.18 :param mode: Optional mode to use for color values. For RGB19 images, this argument can be RGB or RGBA (to blend the20 drawing into the image). For all other modes, this argument21 must be the same as the image mode. If omitted, the mode22 defaults to the mode of the image.23 r N�RGBA�RGBr z
mode mismatch�P)�I�F�����)�1r r r r! �LF)
Z_ensure_mutabler �24ValueError�palette�_imager r �core�draw�draw_ink�ink�fontmode�fill)�selfr r �blend�msg� r/ �wE:\DocsHouse\542 percep lab latest\PerceptionLab\PerceptionLab_Portable\python_embed\Lib\site-packages\PIL/ImageDraw.py�__init__B s. 252627zImageDraw.__init__�GImageFont.ImageFont | ImageFont.FreeTypeFont | ImageFont.TransposedFontc C s" | j sddlm} |�� | _ | j S )a= 28 Get the current default font.29 30 To set the default font for this ImageDraw instance::31 32 from PIL import ImageDraw, ImageFont33 draw.font = ImageFont.truetype("Tests/fonts/FreeMono.ttf")34 35 To set the default font for all future ImageDraw instances::36 37 from PIL import ImageDraw, ImageFont38 ImageDraw.ImageDraw.font = ImageFont.truetype("Tests/fonts/FreeMono.ttf")39 40 If the current default font is ``None``,41 it is initialized with ``ImageFont.load_default()``.42 43 :returns: An image font.r �r )r � r � load_default)r, r r/ r/ r0 �getfontj s 44zImageDraw.getfont� font_size� float | Nonec C s& |d urddl m} |�|�S | �� S )Nr r3 )r4 r r5 r6 )r, r7 r r/ r/ r0 �_getfont� s 45zImageDraw._getfontr) � _Ink | Noner+ � tuple[int | None, int | None]c C s� d }d }|d u r |d u r | j r| j}||fS | j}||fS |d urCt|t�r-t�|| j�}| jr=t|t�r=| j�|| j �}| j46� |�}|d urit|t�rSt�|| j�}| jrct|t�rc| j�|| j �}| j47� |�}||fS �N) r+ r) �48isinstance�strr Zgetcolorr r$ �tupler% r' r( )r, r) r+ Z49result_inkZ result_fillr/ r/ r0 �_getink� s* ��50 51 zImageDraw._getinkr �xyr �start�float�end�width�intc C s2 | � |�\}}|dur| j�|||||� dS dS )z Draw an arc.N)r@ r' �draw_arc)r, rA rB rD r+ rE r) r/ r/ r0 �arc� s �z
ImageDraw.arc�
Sequence[int]�bitmapc C sD |� � | �|�\}}|du r|}|dur | j�||j|� dS dS )zDraw a bitmap.N)�loadr@ r' � draw_bitmapr )r, rA rJ r+ r) r/ r/ r0 rJ � s �zImageDraw.bitmap�outlinec C �j | � ||�\}}|dur| j�||||d� |dur/||kr1|dkr3| j�||||d|� dS dS dS dS )z
Draw a chord.Nr r )r@ r' Z52draw_chord� r, rA rB rD r+ rM rE r) �fill_inkr/ r/ r0 �chord� � 53 �zImageDraw.chordc C �b | � ||�\}}|dur| j�||d� |dur+||kr-|dkr/| j�||d|� dS dS dS dS )zDraw an ellipse.Nr r )r@ r' Z draw_ellipse�r, rA r+ rM rE r) rP r/ r/ r0 �ellipse� � �zImageDraw.ellipse�Sequence[float]�radiusc C s@ |d | |d | |d | |d | f}| � ||||� dS )z4Draw a circle given center coordinates and a radius.r r N)rU )r, rA rX r+ rM rE Z54ellipse_xyr/ r/ r0 �circle� s , zImageDraw.circler �jointc sR | � |�d }|du�r!| j��|� � |dk�r#� dk�r%t�d ttf�r1tttt ��}n�fdd�t dt55��d�D �}t dt56|�d �D ]�}|| }d d� ||d |f|||d ffD �} | d | d krmqId� fdd� }57| d | d ko�| d d | d kp�| d | d k o�| d d | d k} |d � d d |d � d d f|d � d d |d � d d fg} | r�| d d | d d }
}n
| d d | d d }
}| � | |
d |d |� � dk�r | �r|58|| d d �||59|| d d �g}n|60|| d d �||61|| d d �g}| j ||dd� qIdS dS dS dS )z6Draw a line, or a connected sequence of line segments.r NZcurve� c s* g | ]}t tt t� ||d � ���qS )� )r r rC r? ��.0�i)rA r/ r0 �62<listcomp>� s ��z"ImageDraw.line.<locals>.<listcomp>r\ r c 63 S s> g | ]\}}t �t �|d |d |d |d ��d �qS )r r �h )�math�degrees�atan2)r^ rB rD r/ r/ r0 r` s (����coordrW �anglerC r �tuple[float, ...]c 64 sZ | \}}|d8 }� d d }t dd� ||t�t�|�� f||t�t�|�� ffD ��S )N�Z r\ r c s s4 � | ]\}}||d krt �|�nt �|� V qdS )r N)rb �floor�ceil)r^ �pZp_dr/ r/ r0 � <genexpr> s65 � �66�z9ImageDraw.line.<locals>.coord_at_angle.<locals>.<genexpr>)r? rb �cos�radians�sin)re rf �x�yZdistance�rE r/ r0 �coord_at_angle s ��z&ImageDraw.line.<locals>.coord_at_angle� rh � � rr )re rW rf rC r rg )
r@ r' Z67draw_linesr= �listr? r r rC �range�len�pieslice�line)r, rA r+ rE rZ r) Zpointsr_ �point�anglesrs Zflipped�coordsrB rD Z68gap_coordsr/ )rE rA r0 r{ � sZ 6970���"�"�&&� 71���� zImageDraw.line�shape�Image.core._Outlinec C s\ |� � | �||�\}}|dur| j�||d� |dur*||kr,| j�||d� dS dS dS )z (Experimental) Draw a shape.Nr r )�closer@ r' Z draw_outline)r, r r+ rM r) rP r/ r/ r0 r 7 s �zImageDraw.shapec C rN )zDraw a pieslice.Nr r )r@ r' �
draw_pieslicerO r/ r/ r0 rz E rR zImageDraw.pieslicec C s, | � |�\}}|dur| j�||� dS dS )z#Draw one or more individual pixels.N)r@ r' Z draw_points)r, rA r+ r) r/ r/ r0 r| U s �zImageDraw.pointc 72 C s� | � ||�\}}|dur| j�||d� |dur`||krb|dkrd|dkr/| j�||d|� dS | jdurft�d| jj�}| � d�d }t|�} | j�||d� | j�||d|d d |j� dS dS dS dS dS )zDraw a polygon.Nr r r! r\ )r@ r' Z draw_polygonr r �new�size�Draw)73r, rA r+ rM rE r) rP �maskZmask_inkr' r/ r/ r0 �polygon[ s 74" �zImageDraw.polygon�bounding_circle�!Sequence[Sequence[float] | float]�n_sides�rotationc C s t |||�}| �||||� dS )zDraw a regular polygon.N)�!_compute_regular_polygon_verticesr� )r, r� r� r� r+ rM rE rA r/ r/ r0 �regular_polygons s 75zImageDraw.regular_polygonc C rS )zDraw a rectangle.Nr r )r@ r' �draw_rectanglerT r/ r/ r0 � rectangle� rV zImageDraw.rectangle��cornersr� �$tuple[bool, bool, bool, bool] | Nonec
sn t |d ttf�rtttt |�\\��76\� � n ttt |�\��77� � � �k r.d}t|��� �78k r8d}t|��� du r>d� |d �t���t�79��80t� �� t� �� d\��t� �r��� � d k��rh� � ��� �81 d k��rv� �82 ��r��r��� |||��S �dks�t83� �s��� |||��S t �d �}��
||�\��d� ��������� �84� f d
d� } �du�rT| d� �r҈j���85| d � � | d f�d� n � | d �| d kr�j��| d �86� | d � f�d� ��sT��sT��87�| � g}88� d �r|89d |d 7 < � d �r |90d |d 8 < �j�|91�d� � | �92� � g} � d �r=| d |d 7 < � d �rL| d |d 8 < �j�| �d� �du�r/��k�r1�dk�r3| d� ��sʈ�93� �94� d g} � d �r�| d |d 7 < � d �r�| d |d 8 < �j�| �d� �� � d � � g}
� d �r�|
d |d 7 < � d �r�|
d |d 8 < �j�|
�d� ��s5��95�� d � g}96� d �r�|97d |d 7 < � d �r�|98d |d 8 < �j�|99�d� � � d �100� � g} � d �r| d |d 7 < � d �r%| d |d 8 < �j�| �d� dS dS dS dS dS )zDraw a rounded rectangle.r z&x1 must be greater than or equal to x0z&y1 must be greater than or equal to y0N)TTTTr\ )FFr rz �boolr r c 101 s* �r ��102�� �103� fddf�� � �� � fddff}nZ�r:��104�� �105� fddf� � �106� �107� fddff}n=t � fdd�t��108�� �109� fddf� � �110� �111� fddf� � � � � � fddf�� � �� � fddff�D ��}|D ]}| r��jj|�df � qy�jj|��f � qyd S ) Nrt ra r rh � c 3 s � | ] \}}� | r|V qd S r< r/ )r^ r_ �partr� r/ r0 rl � s � �112�zDImageDraw.rounded_rectangle.<locals>.draw_corners.<locals>.<genexpr>r )r? � enumerater' r� rG )rz �partsr� � r� �drP Zfull_xZfull_yr) r, rE Zx0�x1Zy0�y1r/ r0 � draw_corners� s, �� ��� �z1ImageDraw.rounded_rectangle.<locals>.draw_cornersTrv F)rz r� r r )r= rw r? r r rC r# �round�allrU �anyr� rF r@ r' r� )r, rA rX r+ rM rE r� r. �rr� �left�right�top�bottomr/ r� r0 �rounded_rectangle� s� $113"*( 114115116117 118119120121122123124125 �zImageDraw.rounded_rectangler[ r� F�tuple[float, float]�text�AnyStr | ImageText.Text�anchor�spacing�alignr> � direction�features�list[str] | None�language� stroke_width� stroke_fill�embedded_colorr� �argsr �kwargsc s t |tj�r |�n&|du r� ���d��}t�||� j|������r'��� | r/��| | � d� fdd� }||�}|du r@dS d}�jrQ�j durO|�j �n|}��126�127� |�D ].\�128� �dd� ��������� �129f dd�
}|dur�||�j� ||kr�||� qX||� qXdS )z130Draw text.Nr7 r+ r: r rF c s* � � | �\}}|d u r|d usJ �|S |S r< )r@ )r+ r) rP )r, r/ r0 �getink? s131 z ImageDraw.text.<locals>.getinkr r) r� rC r c s� � j }|dkr �r d}g }td�D ] }|�t�132| �� qt��133d �d t��134d �d f}z+�jj�|g��R ���|d� | |d����\}}|d |d |d |d g}W n1 ty� z�jj �|���|� | g��R d|i���}W n t135y� �j� ��}Y nw Y nw |dkr�||� d�}}t �
d | �d } |�d| � |\}136} � jd urLj j�||137| |138|jd | |jd f|� d S d S � j�||| � d S )139Nr r r\ r T)r� r� r� r� Z
stroke_filledr� r) rB rB rv r_ )r* rx �appendrF rb �modfr Zgetmask2�AttributeErrorZgetmask� TypeErrorZgetband�struct�packZfillbandr Zpaster� r' rL ) r) r� r re r_ rB r� �offset�colorZ ink_alpharp rq � r� r� r� r� r� �140image_textr� r� r{ r, rA r/ r0 � draw_textT sx $� �� 141�$ �142� � 143�
��� 144"��z!ImageDraw.text.<locals>.draw_text)r+ r: r rF )r )r) rF r� rC r r ) r= r �Textr9 �getr � embed_color�stroker� r� �_split)r, rA r� r+ r r� r� r� r� r� r� r� r� r� r� r� r� r) Z145stroke_inkr� r/ r� r0 r� s< � 146147��$8 �148�zImageDraw.text�r7 r c C s& | j ||||||||| |149| | |
|d�S �Nr� )r� )r, rA r� r+ r r� r� r� r� r� r� r� r� r� r7 r/ r/ r0 �multiline_text� s �zImageDraw.multiline_textc C s> |du r | � |�}tj||| j|||d�}|r|�� |�� S )z@Get the length of a given string, in pixels with 1/64 precision.N)r� r� r� )r9 r r� r r� Z150get_length) r, r� r r� r� r� r� r7 r� r/ r/ r0 �151textlength� s 152�zImageDraw.textlength�!tuple[float, float, float, float]c C sR |du r | � | �}t�||| j|||| �}
| r|
�� |153r"|
�|154� |
�|||�S )z2Get the bounding box of a given string, in pixels.N)r9 r r� r r� r� Zget_bbox)r, rA r� r r� r� r� r� r� r� r� r� r7 r� r/ r/ r0 �textbbox� s 155�156zImageDraw.textbboxc
C s" | j ||||||||| |157| | d� S r� )r� )
r, rA r� r r� r� r� r� r� r� r� r� r7 r/ r/ r0 �multiline_textbbox� s �z ImageDraw.multiline_textbboxr< )r r r r r r )r r2 )r7 r8 r r2 )r) r: r+ r: r r; )Nr ) rA r rB rC rD rC r+ r: rE rF r r )rA rI rJ r r+ r: r r )NNr )rA r rB rC rD rC r+ r: rM r: rE rF r r )158rA r r+ r: rM r: rE rF r r ) rA rW rX rC r+ r: rM r: rE rF r r )Nr N)159rA r r+ r: rE rF rZ r r r )NN)r r� r+ r: rM r: r r )rA r r+ r: r r )r NNr )r� r� r� rF r� rC r+ r: rM r: rE rF r r )rA r rX rC r+ r: rM r: rE rF r� r� r r ) NNNr[ r� NNNr NF) rA r� r� r� r+ r: r r r� r r� rC r� r> r� r r� r� r� r r� rC r� r: r� r� r� r r� r r r ) rA r� r� r r+ r: r r r� r r� rC r� r> r� r r� r� r� r r� rC r� r: r� r� r7 r8 r r )NNNNF)r� r r r r� r r� r� r� r r� r� r7 r8 r rC ) NNr[ r� NNNr F)rA r� r� r r r r� r r� rC r� r> r� r r� r� r� r r� rC r� r� r7 r8 r r� )�__name__�160__module__� __qualname__r �__annotations__r1 r6 r9 r@ rH rJ rQ rU rY r{ r rz r| r� r� r� r� r� r� r� r� r� r/ r/ r/ r0 r = s� 161 162� 163(164 � � � � � � �P � � � � ��� � ��0��(��*��r r r r r r c C s. zt | d�|�W S ty t| |� Y S w )a� 165 A simple 2D drawing interface for PIL images.166 167 :param im: The image to draw in.168 :param mode: Optional mode to use for color values. For RGB169 images, this argument can be RGB or RGBA (to blend the170 drawing into the image). For all other modes, this argument171 must be the same as the image mode. If omitted, the mode172 defaults to the mode of the image.173 �getdraw)�getattrr� r )r r r/ r/ r0 r� $ s174 �r� �Image.Image | None�)tuple[ImageDraw2.Draw | None, ModuleType]c C s* ddl m} | dur|�| �nd}||fS )zm175 :param im: The image to draw in.176 :returns: A (drawing context, drawing resource factory) tuple.177 r )r
N)r4 r
r� )r r
r' r/ r/ r0 r� 5 s r� �imagerA �tuple[int, int]�value�float | tuple[int, ...]�border� float | tuple[int, ...] | None�threshrC r c C sn | � � }|dus178J �|\}}z|||f }t||�|kr W dS ||||f< W n ttfy2 Y dS w ||fh} t� }179| r�t� } | D ]j\}}|d |f|d |f||d f||d ffD ]O\} }
| |
f|180v sn| dk sn|
dk roq\z|| |
f }W n ttfy� Y q\w |181�| |
f� |du r�t||�|k}n|||fv}|r�||| |
f< | �| |
f� q\qB| }182| } | s=dS dS )a� 183 .. warning:: This method is experimental.184 185 Fills a bounded region with a given color.186 187 :param image: Target image.188 :param xy: Seed position (a 2-item coordinate tuple). See189 :ref:`coordinate-system`.190 :param value: Fill color.191 :param border: Optional border value. If given, the region consists of192 pixels with a color different from the border color. If not given,193 the region consists of pixels having the same color as the seed194 pixel.195 :param thresh: Optional threshold value which specifies a maximum196 tolerable difference of a pixel value from the 'background' in197 order for it to be replaced. Useful for filling regions of198 non-homogeneous, but similar, colors.199 Nr r )rK � _color_diffr# �200IndexError�set�add)r� rA r� r� r� Zpixelrp rq �201background�edgeZ full_edgeZnew_edge�s�trk r+ r/ r/ r0 � floodfill@ sH �202 4 � �� �r� r� r� r� rF r� �list[tuple[float, float]]c s� t |t�s d}t|��|dk rd}t|��t | ttf�s"d}t|��t| �dkrDtdd� | D ��s7d}t|��ttt t| ���^ ��nPt| �dkr�t | d ttf�r�td203d� | d D ��rgt | d tt f�smd}t|��t| d �dkr{d }t|��ttt t| d ���tt | d ��nd
}t|���d kr�d}t|��t |tt f�s�d}t|��d#�fdd� � d$� �fdd� �d%d d �}|||�}�fd d!�|D �S )&u� 204 Generate a list of vertices for a 2D regular polygon.205 206 :param bounding_circle: The bounding circle is a sequence defined207 by a point and radius. The polygon is inscribed in this circle.208 (e.g. ``bounding_circle=(x, y, r)`` or ``((x, y), r)``)209 :param n_sides: Number of sides210 (e.g. ``n_sides=3`` for a triangle, ``6`` for a hexagon)211 :param rotation: Apply an arbitrary rotation to the polygon212 (e.g. ``rotation=90``, applies a 90 degree rotation)213 :return: List of regular polygon vertices214 (e.g. ``[(25, 50), (50, 50), (50, 25), (25, 25)]``)215 216 How are the vertices computed?217 1. Compute the following variables218 - theta: Angle between the apothem & the nearest polygon vertex219 - side_length: Length of each polygon edge220 - centroid: Center of bounding circle (1st, 2nd elements of bounding_circle)221 - polygon_radius: Polygon radius (last element of bounding_circle)222 - angles: Location of each polygon vertex in polar grid223 (e.g. A square with 0 degree rotation => [225.0, 315.0, 45.0, 135.0])224 225 2. For each angle in angles, get the polygon vertex at that angle226 The vertex is computed using the equation below.227 X= xcos(φ) + ysin(φ)228 Y= −xsin(φ) + ycos(φ)229 230 Note:231 φ = angle in degrees232 x = 0233 y = polygon_radius234 235 The formula above assumes rotation around the origin.236 In our case, we are rotating around the centroid.237 To account for this, we use the formula below238 X = xcos(φ) + ysin(φ) + centroid_x239 Y = −xsin(φ) + ycos(φ) + centroid_y240 zn_sides should be an intrv z n_sides should be an int > 2z$bounding_circle should be a sequencec s � � | ]241}t |ttf�V qd S r< �r= rF rC r] r/ r/ r0 rl � s � z4_compute_regular_polygon_vertices.<locals>.<genexpr>z0bounding_circle should only contain numeric datar\ r c s r� r< r� r] r/ r/ r0 rl � s � 242�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]rc rC r r� c 243 s� t | d t�t�d| �� | d t�t�d| �� � d d�t | d t�t�d| �� | d t�t�d| �� � d d�fS )Nr ra r r\ )r� rb rm rn ro )r| rc )�centroidr/ r0 �_apply_rotation� s" �������z:_compute_regular_polygon_vertices.<locals>._apply_rotationrf c s �dg}� || �S �Nr r/ )rf Z start_point)r� �polygon_radiusr/ r0 �_compute_polygon_vertex� s 244zB_compute_regular_polygon_vertices.<locals>._compute_polygon_vertexr� rF r� c S sP g }d| }dd| | }t | �D ]}|�|� ||7 }|dkr%|d8 }q|S )Nra r� g �?)rx r� )r� r� r} rc Z
current_angle�_r/ r/ r0 � _get_angles� s 245�z6_compute_regular_polygon_vertices.<locals>._get_anglesc s g | ]}� |��qS r/ r/ )r^ rf )r� r/ r0 r` s z5_compute_regular_polygon_vertices.<locals>.<listcomp>N)r| r� rc rC r r� )rf rC r r� )r� rF r� rC r r� )246r= rF r� r# rw r? ry r� r rC )r� r� r� r. r� r} r/ )r� r� r� r� r0 r� � sR 247+ ���248249
r� �color1�color2c sH t | t�r| n| f� t |t�r|n|f�t� �fdd�tt���D ��S )zJ250 Uses 1-norm distance to calculate difference between two values.251 c 3 s$ � | ]
}t � | �| �V qd S r< )�absr] ��first�secondr/ r0 rl s �" z _color_diff.<locals>.<genexpr>)r= r? �sumrx ry )r� r� r/ r� r0 r� s r� r< )r r r r r r )r r� r r� r� ) r� r rA r� r� r� r� r� r� rC r r )r� r� r� rF r� rC r r� )r� r� r� r� r rC )!�252__future__r rb r� �collections.abcr �typingr r4 r r r �
TYPE_CHECKINGr �typesr253 r r r
r Z_typingr r r& rM r r� r r� r� r� r� r� r/ r/ r/ r0 �<module> s: l �254A