File: //home/arjun/projects/env/lib/python3.10/site-packages/PIL/__pycache__/ImageMorph.cpython-310.pyc
o
we) � @ sL d dl Z ddlmZmZ dZg d�Zg d�ZG dd� d�ZG d d
� d
�ZdS ) � N� )�Image�
_imagingmorphi ) � � r � � r � � � ) r r r r
r r r r r c @ �J e Zd ZdZddd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�
LutBuilderaT A class for building a MorphLut from a descriptive language
The input patterns is a list of a strings sequences like these::
4:(...
.1.
111)->1
(whitespaces including linebreaks are ignored). The option 4
describes a series of symmetry operations (in this case a
4-rotation), the pattern is described by:
- . or X - Ignore
- 1 - Pixel is on
- 0 - Pixel is off
The result of the operation is described after "->" string.
The default is to return the current pixel value, which is
returned if no other match is found.
Operations:
- 4 - 4 way rotation
- N - Negate
- 1 - Dummy op for no other operation (an op must always be given)
- M - Mirroring
Example::
lb = LutBuilder(patterns = ["4:(... .1. 111)->1"])
lut = lb.build_lut()
Nc C sx |d ur|| _ ng | _ d | _|d ur:ddgdgddgdgddgg d�d�}||vr3d | d
}t|��|| | _ d S d S ) N�1:(... ... ...)->0z4:(00. 01. ...)->1z4:(... .0. .1.)->1z4:(... .0. ..1)->1z4:(... .1. .0.)->0z4:(... .1. ..0)->0)r z4:(.0. .1. ...)->1z4:(01. .1. ...)->1)�corner� dilation4� dilation8�erosion4�erosion8�edgezUnknown pattern �!)�patterns�lut� Exception)�selfr �op_name�known_patterns�msg� r �G/home/arjun/projects/env/lib/python3.10/site-packages/PIL/ImageMorph.py�__init__@ s" � �zLutBuilder.__init__c C s | j |7 _ d S �N�r )r r r r r � add_patternsY s zLutBuilder.add_patternsc s. ddg�d� t � �fdd�tt�D ��| _d S )Nr r � c 3 s � | ] }�|� @ d k V qdS )r Nr )�.0�i��m�symbolsr r � <genexpr>_ s � z/LutBuilder.build_default_lut.<locals>.<genexpr>)� bytearray�range�LUT_SIZEr �r r r& r �build_default_lut\ s "z LutBuilder.build_default_lutc C s | j S r �r r- r r r �get_luta s zLutBuilder.get_lutc s( t |�dksJ �d�� fdd�|D ��S )z�string_permute takes a pattern and a permutation and returns the
string permuted according to the permutation list.
� � c 3 s � | ]}� | V qd S r r )r$ �p��patternr r r) i s � z-LutBuilder._string_permute.<locals>.<genexpr>)�len�join)r r5 � permutationr r4 r �_string_permuted s zLutBuilder._string_permutec C s� ||fg}d|v r%|d d }t d�D ]}|�| �|d d t�|f� qd|v rCt|�}|d|� D ]\}}|�| �|t�|f� q3d|v rqt|�}|d|� D ] \}}|�d d
��d d ��d
d �}dt|� }|�||f� qQ|S ) z�pattern_permute takes a basic pattern and its result and clones
the pattern according to the modifications described in the $options
parameter. It returns a list of all cloned patterns.�4���r r r �MN�N�0�Z�1)r+ �appendr9 �ROTATION_MATRIXr6 �
MIRROR_MATRIX�replace�int) r �
basic_pattern�options� basic_resultr �resr% �nr5 r r r �_pattern_permutek s$
� zLutBuilder._pattern_permutec C s< | � � g }| jD ]<}t�d|�dd��}|s!d| d }t|��|�d�}|�d�}t|�d��}|�d d��dd�}|| �|||�7 }q t |�D ] \}}|d
�d d ��d d
�}t�
|�}||d f||< qJt t �D ].}t
|�dd� } ddt| � | ddd� } |D ]\}}
|�| �r�d
dg|
| j|< q�ql| jS )zlCompile all patterns into a morphology lut.
TBD :Build based on (file) morphlut:modify_lut
z (\w*):?\s*\((.+?)\)\s*->\s*(\d)�
r2 zSyntax error in pattern "�"r r r � r �.�Xz[01]Nr> r1 r; )r. r �re�searchrD r �grouprE rK � enumerate�compiler+ r, �binr6 �matchr ) r r r3 r' r rG r5 �resultr% �
bitpattern�rr r r � build_lut� s2
��zLutBuilder.build_lut)NN) �__name__�
__module__� __qualname__�__doc__r r" r. r0 r9 rK r[ r r r r r
s
# r
c @ r )�MorphOpz*A class for binary morphological operatorsNc C sB || _ |durt|d��� | _ dS |dur t|d��� | _ dS dS )z&Create a binary morphological operatorN)r r! )r r
r[ )r r r r r r r r � s �zMorphOp.__init__c C sb | j du r d}t|��|jdkrd}t|��t�|j|jd�}t�t | j �|j
j |j
j �}||fS )z�Run a single morphological operation on an image
Returns a tuple of the number of changed pixels and the
morphed imageN�No operator loaded�L�Image mode must be L) r r �mode�
ValueErrorr �new�sizer �apply�bytes�im�id)r �imager �outimage�countr r r rh � s
z
MorphOp.applyc C sB | j du r d}t|��|jdkrd}t|��t�t| j �|jj�S )z�Get a list of coordinates matching the morphological operation on
an image.
Returns a list of tuples of (x,y) coordinates
of all matching pixels. See :ref:`coordinate-system`.Nra rb rc ) r r rd re r rW ri rj rk �r rl r r r r rW � s
z
MorphOp.matchc C s$ |j dkr d}t|��t�|jj�S )z�Get a list of all turned on pixels in a binary image
Returns a list of tuples of (x,y) coordinates
of all matching pixels. See :ref:`coordinate-system`.rb rc )rd re r �
get_on_pixelsrj rk ro r r r rp � s
zMorphOp.get_on_pixelsc C s\ t |d��}t|�� �| _W d � n1 sw Y t| j�tkr,d| _d}t|��dS )z!Load an operator from an mrl file�rbNzWrong size operator file!)�openr* �readr r6 r, r )r �filename�fr r r r �load_lut� s ��zMorphOp.load_lutc C sR | j du r d}t|��t|d��}|�| j � W d � dS 1 s"w Y dS )z Save an operator to an mrl fileNra �wb)r r rr �write)r rt r ru r r r �save_lut� s
"�zMorphOp.save_lutc C s
|| _ dS )z#Set the lut from an external sourceNr/ )r r r r r �set_lut� s
zMorphOp.set_lut)NNN) r\ r] r^ r_ r rh rW rp rv ry rz r r r r r` � s
r` ) rQ r2 r r r, rB rC r
r` r r r r �<module> s