
    RfL                    r    d dl mZ d dlZddlmZmZ dZg dZg dZ G d d	          Z	 G d
 d          Z
dS )    )annotationsN   )Image_imagingmorphi   )	      r         r            )	r   r   r   r   r
   r   r   r	   r   c                  >    e Zd ZdZd
dZd Zd Zd Zd Zd Z	d	 Z
dS )
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                    ||| _         ng | _         d | _        |=ddgdgddgdgddgg dd}||vrd	|z   d
z   }t          |          ||         | _         d S d S )N1:(... ... ...)->0z4:(00. 01. ...)->1z4:(... .0. .1.)->1z4:(... .0. ..1)->1z4:(... .1. .0.)->0z4:(... .1. ..0)->0)r   z4:(.0. .1. ...)->1z4:(01. .1. ...)->1)corner	dilation4	dilation8erosion4erosion8edgezUnknown pattern !)patternslut	Exception)selfr   op_nameknown_patternsmsgs        @/var/www/html/env/lib/python3.11/site-packages/PIL/ImageMorph.py__init__zLutBuilder.__init__A   s    $DMMDM/1EF2324HI1213GH   N n,,(72S8nn$*73DMMM#     c                &    | xj         |z  c_         d S Nr   )r   r   s     r    add_patternszLutBuilder.add_patternsZ   s    !r"   c                |    ddgdt          fdt          t                    D                       | _        d S )Nr   r      c              3  6   K   | ]}|z  d k             V  dS )r   N ).0imsymbolss     r    	<genexpr>z/LutBuilder.build_default_lut.<locals>.<genexpr>`   s0      KKaWa!eq[1KKKKKKr"   )	bytearrayrangeLUT_SIZEr   )r   r-   r.   s    @@r    build_default_lutzLutBuilder.build_default_lut]   sC    a&KKKKK5??KKKKKr"   c                    | j         S r$   r   )r   s    r    get_lutzLutBuilder.get_lutb   s	    xr"   c                p    t          |          dk    sJ d                    fd|D                       S )zstring_permute takes a pattern and a permutation and returns the
        string permuted according to the permutation list.
        	    c              3  (   K   | ]}|         V  d S r$   r*   )r+   ppatterns     r    r/   z-LutBuilder._string_permute.<locals>.<genexpr>j   s'      77awqz777777r"   )lenjoin)r   r<   permutations    ` r    _string_permutezLutBuilder._string_permutee   sE     ;1$$$$ww7777;777777r"   c                   ||fg}d|v r\|d         d         }t          d          D ]>}|                    |                     |d         d         t                    |f           ?d|v rOt	          |          }|d|         D ]5\  }}|                    |                     |t
                    |f           6d|v rt	          |          }|d|         D ]l\  }}|                    d	d
                              dd	                              d
d          }dt          |          z
  }|                    ||f           m|S )zpattern_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   MNN0Z1)r1   appendr@   ROTATION_MATRIXr=   MIRROR_MATRIXreplaceint)	r   basic_patternoptionsbasic_resultr   resr,   nr<   s	            r    _pattern_permutezLutBuilder._pattern_permutel   sh    #L12 '>>2,q/C1XX  ))(2,q/?KKSQ    '>>HA (! U U!5!5g}!M!Ms STTTT '>>HA (! 0 0!//#s33;;CEEMMcSVWW#c((l#////r"   c                   |                                   g }| j        D ]}t          j        d|                    dd                    }|sd|z   dz   }t          |          |                    d          }|                    d          }t          |                    d                    }|                    d	d                              dd          }||                     |||          z  }t          |          D ]V\  }}|d
                             dd                              dd          }t          j
        |          }||d         f||<   Wt          t                    D ]i}t          |          dd         }	ddt          |	          z
  z  |	z   ddd         }	|D ],\  }}
|                    |	          rd
dg|
         | j        |<   -j| j        S )zlCompile all patterns into a morphology lut.

        TBD :Build based on (file) morphlut:modify_lut
        z(\w*):?\s*\((.+?)\)\s*->\s*(\d)
r9   zSyntax error in pattern ""r   r   r    r   .Xz[01]NrF   r8   rC   )r3   r   researchrL   r   grouprM   rS   	enumeratecompiler1   r2   binr=   matchr   )r   r   r;   r-   r   rO   r<   resultr,   
bitpatternrs              r    	build_lutzLutBuilder.build_lut   s   
 	     	H 	HA	<aiib>Q>QRRA %1A5;nn$ggajjGggajjG__F ooc2..66tR@@G--gwGGGHH $H-- 	* 	*JAw
""3,,44S&AAA
1Agaj/HQKK
 x 	, 	,AQJS__!45
BDDbDIJ  , ,177:&& ,#$a&)DHQK, xr"   )NN)__name__
__module____qualname____doc__r!   r&   r3   r6   r@   rS   rd   r*   r"   r    r   r      s        ! !F4 4 4 42" " "L L L
  8 8 8  <) ) ) ) )r"   r   c                  >    e Zd ZdZd
dZd Zd Zd Zd Zd Z	d	 Z
dS )MorphOpz*A class for binary morphological operatorsNc                    || _         |)t          |                                          | _         dS |)t          |                                          | _         dS dS )z&Create a binary morphological operatorN)r   r%   )r   r   rd   )r   r   r   r   s       r    r!   zMorphOp.__init__   sa    !'222<<>>DHHH!!8444>>@@DHHH "!r"   c                *   | j         d}t          |          |j        dk    rd}t          |          t	          j        |j        |j        d          }t          j        t          | j                   |j
        j        |j
        j                  }||fS )zRun 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counts        r    rt   zMorphOp.apply   s    
 8&CC.. :(CS//!9UZT::#E$(OOUX[(+.QQhr"   c                    | j         d}t          |          |j        dk    rd}t          |          t	          j        t          | j                   |j        j                  S )zGet 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`.Nrm   rn   ro   )	r   r   rp   rq   r   r`   ru   rv   rw   r   rx   r   s      r    r`   zMorphOp.match   sZ     8&CC.. :(CS//!"5??EHK@@@r"   c                v    |j         dk    rd}t          |          t          j        |j        j                  S )zGet 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`.rn   ro   )rp   rq   r   get_on_pixelsrv   rw   r|   s      r    r~   zMorphOp.get_on_pixels   s7     :(CS//!*58;777r"   c                   t          |d          5 }t          |                                          | _        ddd           n# 1 swxY w Y   t	          | j                  t
          k    rd| _        d}t          |          dS )z!Load an operator from an mrl filerbNzWrong size operator file!)openr0   readr   r=   r2   r   )r   filenamefr   s       r    load_lutzMorphOp.load_lut   s    (D!! 	+Q **DH	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ tx==H$$DH-CC..  %$s   'AAAc                    | j         d}t          |          t          |d          5 }|                    | j                    ddd           dS # 1 swxY w Y   dS )zSave an operator to an mrl fileNrm   wb)r   r   r   write)r   r   r   r   s       r    save_lutzMorphOp.save_lut   s    8&CC.. (D!! 	QGGDH	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	s   AAAc                    || _         dS )z#Set the lut from an external sourceNr5   )r   r   s     r    set_lutzMorphOp.set_lut   s    r"   )NNN)re   rf   rg   rh   r!   rt   r`   r~   r   r   r   r*   r"   r    rj   rj      s        44A A A A   A A A	8 	8 	8! ! !      r"   rj   )
__future__r   rZ   r9   r   r   r2   rJ   rK   r   rj   r*   r"   r    <module>r      s    # " " " " " 				 " " " " " " " "  
  V V V V V V V VrI I I I I I I I I Ir"   