o 6Wjt(@sXddlmZddlZddlmZmZdZgdZgdZGdd d Z Gd d d Z dS) ) annotationsN)Image _imagingmorphi) rr) r rrr r rr rrc@sZeZdZdZ d#d$d d Zd%d d Zd&ddZd'ddZd(ddZd)dd Z d&d!d"Z dS)* LutBuilderaTA 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() Npatternslist[str] | Noneop_name str | NonereturnNonecCszd|_|dur/ddgdgddgdgddggdd }||vr(d |d }t||||_dS|dur8||_dSg|_dS) a :param patterns: A list of input patterns, or None. :param op_name: The name of a known pattern. One of "corner", "dilation4", "dilation8", "erosion4", "erosion8" or "edge". :exception Exception: If the op_name is not recognized. N1:(... ... ...)->0z4:(00. 01. ...)->1z4:(... .0. .1.)->1z4:(... .0. ..1)->1z4:(... .1. .0.)->0z4:(... .1. ..0)->0)rz4:(.0. .1. ...)->1z4:(01. .1. ...)->1)corner dilation4 dilation8erosion4erosion8edgezUnknown pattern !)lut Exceptionr)selfrrknown_patternsmsgr!]/home/esfera/Documents/content_generation/venv/lib/python3.10/site-packages/PIL/ImageMorph.py__init__As     zLutBuilder.__init__ list[str]cCs|j|7_dS)z\ Append to list of patterns. :param patterns: Additional patterns. N)r)rrr!r!r" add_patternsbszLutBuilder.add_patterns bytearraycs0ddgdtfddttD|_|jS)z Set the current LUT, and return it. This is the default LUT that patterns will be applied against when building. rrc3s |] }|@dkVqdS)rNr!).0imsymbolsr!r" rsz/LutBuilder.build_default_lut..)r&rangeLUT_SIZErrr!r*r"build_default_lutjszLutBuilder.build_default_lutbytearray | NonecCs|jS)z) Returns the current LUT rr0r!r!r"get_lutuszLutBuilder.get_lutpatternstr permutation list[int]cs(t|dksJdfdd|DS)zuTakes a pattern and a permutation and returns the string permuted according to the permutation list. c3s|]}|VqdS)Nr!)r(pr5r!r"r-sz-LutBuilder._string_permute..)lenjoin)rr5r7r!r<r"_string_permute{szLutBuilder._string_permute basic_patternoptions basic_resultintlist[tuple[str, int]]c Cs||fg}d|vr%|dd}tdD]}|||ddt|fqd|vrCt|}|d|D]\}}|||t|fq3d|vrqt|}|d|D]\}}|d d d d d d }dt|}|||fqQ|S) zTakes 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.4rr rMNN0Z1)r.appendr?ROTATION_MATRIXr= MIRROR_MATRIXreplacerC) rr@rArBrresr)nr5r!r!r"_pattern_permutes$    zLutBuilder._pattern_permutec CsD||jdus Jg}|jD]<}td|dd}|s(d|d}t||d}|d}t|d }|d ddd}|| |||7}qg}|D]}|d d d d d}| t ||dfqQt t D].} t| dd} ddt| | ddd} |D]\}} || rd dg| |j| <qqp|jS)zoCompile all patterns into a morphology LUT, and return it. This is the data to be passed into MorphOp.Nz(\w):?\s*\((.+?)\)\s*->\s*(\d) r:zSyntax error in pattern ""rr r r.Xz[01]rIr9rF)r1rrresearchrOrgrouprCrRrLcompiler.r/binr=match) rrr;r+r rAr5resultcompiled_patternsr) bitpatternrr!r!r" build_luts4       zLutBuilder.build_lut)NN)rrrrrr)rr$rr)rr&)rr2)r5r6r7r8rr6)r@r6rAr6rBrCrrD) __name__ __module__ __qualname____doc__r#r%r1r4r?rRrbr!r!r!r"r s$ !    r c@s^eZdZdZ   dd d d Zd!ddZd"ddZd"ddZd#ddZd#ddZ d$ddZ dS)%MorphOpz*A class for binary morphological operatorsNrr2rrrrrrcCs.|dur |dur ||_dSt|||_dS)aCreate a binary morphological operator. If the LUT is not provided, then it is built using LutBuilder from the op_name or the patterns. :param lut: The LUT data. :param patterns: A list of input patterns, or None. :param op_name: The name of a known pattern. One of "corner", "dilation4", "dilation8", "erosion4", "erosion8", "edge". :exception Exception: If the op_name is not recognized. N)rr rb)rrrrr!r!r"r#s zMorphOp.__init__image Image.Imagetuple[int, Image.Image]cCs`|jdur d}t||jdvrd}t|t|j|j}tt |j| | }||fS)a9Run a single morphological operation on an image. Returns a tuple of the number of changed pixels and the morphed image. :param image: A 1-mode or L-mode image. :exception Exception: If the current operator is None. :exception ValueError: If the image is not 1 or L mode.NNo operator loadedrKLImage mode must be 1 or L) rrmode ValueErrorrnewsizerapplybytesgetim)rrhr outimagecountr!r!r"rss  z MorphOp.applylist[tuple[int, int]]cCsB|jdur d}t||jdvrd}t|tt|j|S)axGet 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`. :param image: A 1-mode or L-mode image. :exception Exception: If the current operator is None. :exception ValueError: If the image is not 1 or L mode.Nrkrlrn)rrrorprr]rtrurrhr r!r!r"r]s  z MorphOp.matchcCs$|jdvr d}t|t|S)a!Get a list of all turned on pixels in a 1 or L mode image. Returns a list of tuples of (x,y) coordinates of all non-empty pixels. See :ref:`coordinate-system`. :param image: A 1-mode or L-mode image. :exception ValueError: If the image is not 1 or L mode.rlrn)rorpr get_on_pixelsruryr!r!r"rz s zMorphOp.get_on_pixelsfilenamer6cCs\t|d}t||_Wdn1swYt|jtkr,d|_d}t|dS)z Load an operator from an mrl file :param filename: The file to read from. :exception Exception: If the length of the file data is not 512. rbNzWrong size operator file!)openr&readrr=r/r)rr{fr r!r!r"load_luts zMorphOp.load_lutcCsR|jdur d}t|t|d}||jWddS1s"wYdS)z Save an operator to an mrl file. :param filename: The destination file. :exception Exception: If the current operator is None. Nrkwb)rrr}write)rr{r rr!r!r"save_lut*s  "zMorphOp.save_lutcCs ||_dS)zU Set the LUT from an external source :param lut: A new LUT. Nr3)rrr!r!r"set_lut7s zMorphOp.set_lut)NNN)rr2rrrrrr)rhrirrj)rhrirrx)r{r6rr)rr2rr) rcrdrerfr#rsr]rzrrrr!r!r!r"rgs      rg) __future__rrXr:rrr/rMrNr rgr!r!r!r"s 1