Your IP : 216.73.216.213
�ʴWt � @ sb d d l m Z d d l m Z d d l Z d
Z Gd d � d e � Z Gd d � d e � Z d S) � )�Image)�
_imagingmorphN� � c @ sp e Z d Z d Z d d d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d S)�
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 s� | d k r | | _ n g | _ d | _ | d k r� d d d g d d g d d d g d d g d
d d g d d d
d g i } | | k r� t d | d � � | | | _ d S)NZcornerz1:(... ... ...)->0z4:(00. 01. ...)->1Z dilation4z4:(... .0. .1.)->1Z dilation8z4:(... .0. ..1)->1Zerosion4z4:(... .1. .0.)->0Zerosion8z4:(... .1. ..0)->0Zedgez4:(.0. .1. ...)->1z4:(01. .1. ...)->1zUnknown pattern �!)�patterns�lut� Exception)�selfr �op_nameZknown_patterns� r
�A/opt/alt/python35/lib64/python3.5/site-packages/PIL/ImageMorph.py�__init__2 s& zLutBuilder.__init__c C s | j | 7_ d S)N)r )r r r
r
r � add_patternsK s zLutBuilder.add_patternsc sA d d g � d � t � � f d d � t t � D� � | _ d S)Nr r � c s$ g | ] } � | � @d k � q S)r r
)�.0�i)�m�symbolsr
r �
<listcomp>Q s z0LutBuilder.build_default_lut.<locals>.<listcomp>� )� bytearray�range�LUT_SIZEr )r r
)r r r �build_default_lutN s z LutBuilder.build_default_lutc C s | j S)N)r )r r
r
r �get_lutS s zLutBuilder.get_lutc s8 t | � d k s t � d j � f d d � | D� � S)z�string_permute takes a pattern and a permutation and returns the
string permuted according to the permutation list.
r � c s g | ] } � | � q Sr
r
)r �p)�patternr
r r [ s z.LutBuilder._string_permute.<locals>.<listcomp>)�len�AssertionError�join)r r Z permutationr
)r r �_string_permuteV s zLutBuilder._string_permutec
C s� | | f g } d | k r� | d d } xV t d � D]H } | j | j | d d d d d d d d d d d
g � | f � q6 Wd | k r� t | � } xX | d | � D]F \ } } | j | j | d
d d d d d d d d g � | f � q� Wd | k rt | � } xo | d | � D]] \ } } | j d
d � j d d
� j d d � } d d t | � } | j | | f � q W| 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.�4r r r � � � � � � �M�N�0�Z�1z%d���r0 )r �appendr# r �replace�int) r Z
basic_pattern�optionsZ basic_resultr �resr �nr r
r
r �_pattern_permute] s2 zLutBuilder._pattern_permutec
C s� | j � g } x� | j D]� } t j d | j d d � � } | sX t d | d � � | j d � } | j d � } t | j d � � } | j d d � j d d � } | | j | | | � 7} q Wxg t t
| � � D]S } | | d
j d d � j d d
� } t j | � } | | | d f | | <q� Wx� t t � D]~ } t
| � d d � } d d t
| � | d d d � } x: | D]2 \ } } | j | � r�d
d g | | j | <q�Wq=W| j S)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]Nr- r r0 )r r �re�searchr2 r
�groupr3 r7 r r �compiler �bin�matchr )
r r r r r4 r �resultr Z
bitpattern�rr
r
r � build_lut� s.
& % zLutBuilder.build_lut) �__name__�
__module__� __qualname__�__doc__r r r r r# r7 rE r
r
r
r r s " %r c @ ss e Z d Z d Z d d d d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d S)�MorphOpz*A class for binary morphological operatorsNc C sX | | _ | d k r0 t d | � j � | _ n$ | d k rT t d | � j � | _ d S)z&Create a binary morphological operatorNr r )r r rE )r r r r r
r
r r � s
zMorphOp.__init__c C s� | j d k r t d � � | j d k r6 t d � � t j | j | j d � } t j t | j � | j j
| j j
� } | | f S)z�Run a single morphological operation on an image
Returns a tuple of the number of changed pixels and the
morphed imageNzNo operator loaded�Lz0Image must be binary, meaning it must use mode L) r r
�moder �new�sizer �apply�bytes�im�id)r �imageZoutimage�countr
r
r rO � s $z
MorphOp.applyc C sU | j d k r t d � � | j d k r6 t d � � t j t | j � | j j � 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.NzNo operator loadedrK z0Image must be binary, meaning it must use mode L)r r
rL r rB rP rQ rR )r rS r
r
r rB � s
z
MorphOp.matchc C s. | j d k r t d � � t j | j j � 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.rK z0Image must be binary, meaning it must use mode L)rL r
r �
get_on_pixelsrQ rR )r rS r
r
r rU � s zMorphOp.get_on_pixelsc C s\ t | d � � } t | j � � | _ Wd QRXt | j � d k rX d | _ t d � � d S)z!Load an operator from an mrl file�rbNi zWrong size operator file!)�openr �readr r r
)r �filename�fr
r
r �load_lut� s
zMorphOp.load_lutc C sH | j d k r t d � � t | d � � } | j | j � Wd QRXd S)z Save an operator to an mrl fileNzNo operator loaded�wb)r r
rW �write)r rY rZ r
r
r �save_lut� s zMorphOp.save_lutc C s
| | _ d S)z#Set the lut from an external sourceN)r )r r r
r
r �set_lut� s zMorphOp.set_lut) rF rG rH rI r rO rB rU r[ r^ r_ r
r
r
r rJ � s
rJ i )ZPILr r r= r �objectr rJ r
r
r
r �<module> s
�