OwlCyberSecurity - MANAGER
Edit File: ImageMorph.cpython-38.pyc
U g� ^� � @ sh d dl Z ddlmZmZ dZddd dddd d d g Zd dd d ddd ddg ZG d d � d �ZG dd� d�ZdS )� N� )�Image� _imagingmorphi � � � � � � � c @ sJ e Zd ZdZddd�Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )� LutBuildera~ 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 sr |d k r|| _ ng | _ d | _|d k rnddgdgddgdgddgdddgd �}||krdtd | d ��|| | _ d S ) Nz1:(... ... ...)->0z4:(00. 01. ...)->1z4:(... .0. .1.)->1z4:(... .0. ..1)->1z4:(... .1. .0.)->0z4:(... .1. ..0)->0z4:(.0. .1. ...)->1z4:(01. .1. ...)->1)ZcornerZ dilation4Z dilation8Zerosion4Zerosion8ZedgezUnknown pattern �!)�patterns�lut� Exception)�selfr �op_nameZknown_patterns� r �0/usr/lib/python3/dist-packages/PIL/ImageMorph.py�__init__@ s$ �� zLutBuilder.__init__c C s | j |7 _ d S �N�r )r r r r r � add_patternsX 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��mZsymbolsr 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_lut` s zLutBuilder.get_lutc s( t |�dkst�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 h s z-LutBuilder._string_permute.<locals>.<genexpr>)�len�AssertionError�join)r r* Z permutationr r) r �_string_permutec s zLutBuilder._string_permutec C s� ||fg}d|krJ|d d }t d�D ]"}|�| �|d d t�|f� q&d|kr�t|�}|d|� D ] \}}|�| �|t�|f� qfd|kr�t|�}|d|� D ]>\}}|�dd ��d d��d d �}dt|� }|�||f� q�|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 �M�N�0�Z�1)r �appendr. �ROTATION_MATRIXr+ � MIRROR_MATRIX�replace�int) r Z basic_pattern�optionsZ basic_resultr �resr �nr* r r r �_pattern_permutej s$ � zLutBuilder._pattern_permutec C s< | � � g }| jD ]t}t�d|�dd��}|s>td| d ��|�d�}|�d�}t|�d��}|�d d��dd�}|| �|||�7 }qt |�D ]:\}}|d �d d ��d d �}t� |�}||d f||< q�t t �D ]`}t |�dd� }ddt|� | ddd� }|D ](\}} |�|��r d dg| | j|< �q q�| jS )zlCompile all patterns into a morphology lut. TBD :Build based on (file) morphlut:modify_lut z (\w*):?\s*\((.+?)\)\s*->\s*(\d)� r' zSyntax error in pattern "�"r r r � r �.�Xz[01]Nr3 r& r0 )r# r �re�searchr9 r �groupr: r> � enumerate�compiler r! �binr+ �matchr ) r r r( r r; r* �resultr Z bitpattern�rr r r � build_lut� s, zLutBuilder.build_lut)NN) �__name__� __module__� __qualname__�__doc__r r r# r% r. r> rM r r r r r s # r c @ sJ e Zd ZdZddd�Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )�MorphOpz*A class for binary morphological operatorsNc C s< || _ |dk r t|d��� | _ n|dk r8t|d��� | _ dS )z&Create a binary morphological operatorN)r r )r r rM )r r r r r r r r � s zMorphOp.__init__c C sZ | j dkrtd��|jdkr$td��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�0Image must be binary, meaning it must use mode L) r r �moder �new�sizer �apply�bytes�im�id)r �imageZoutimage�countr r r rY � s z MorphOp.applyc C s: | j dkrtd��|jdkr$td��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`.NrS rT rU )r r rV r rJ rZ r[ r\ �r r] r r r rJ � s z MorphOp.matchc C s |j dkrtd��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`.rT rU )rV r r � get_on_pixelsr[ r\ r_ r r r r` � s zMorphOp.get_on_pixelsc C sD t |d��}t|�� �| _W 5 Q R X t| j�tkr@d| _td��dS )z!Load an operator from an mrl file�rbNzWrong size operator file!)�openr �readr r+ r! r �r �filename�fr r r �load_lut� s zMorphOp.load_lutc C s8 | j dkrtd��t|d��}|�| j � W 5 Q R X dS )z Save an operator to an mrl fileNrS �wb)r r rb �writerd 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) rN rO rP rQ r rY rJ r` rg rj rk r r r r rR � s rR ) rD r' r r r! r7 r8 r rR r r r r �<module> s2 � �