î hÐR\v ã@shddlmZddlmZmZddlZd ZGdd„deƒZGdd „d eƒZ dS) é)Úprint_functioné)ÚImageÚ _imagingmorphNé c@speZdZdZdddd„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() NcCs¾|dk r||_n g|_d|_|dk rºiddgd6dgd6ddgd6dgd 6dd gd 6dd d gd6}||krªtd|dƒ‚n|||_ndS)Nz1:(... ... ...)->0z4:(00. 01. ...)->1Zcornerz4:(... .0. .1.)->1Z dilation4z4:(... .0. ..1)->1Z dilation8z4:(... .1. .0.)->0Zerosion4z4:(... .1. ..0)->0Zerosion8z4:(.0. .1. ...)->1z4:(01. .1. ...)->1ZedgezUnknown pattern ú!)ÚpatternsÚlutÚ Exception)Úselfr Úop_nameZknown_patterns©rúC/var/www/dbchiro/venv/lib/python3.4/site-packages/PIL/ImageMorph.pyÚ__init__3s(            zLutBuilder.__init__cCs|j|7_dS)N)r )r r rrrÚ add_patternsLszLutBuilder.add_patternscsAddg‰d‰t‡‡fdd†ttƒDƒƒ|_dS)Nrréc3s#|]}ˆ|ˆ@dkVqdS)rNr)Ú.0Úi)ÚmÚsymbolsrrú Rsz/LutBuilder.build_default_lut..é)Ú bytearrayÚrangeÚLUT_SIZEr )r r)rrrÚbuild_default_lutOs zLutBuilder.build_default_lutcCs|jS)N)r )r rrrÚget_lutTszLutBuilder.get_lutcs8t|ƒdkst‚dj‡fdd†|DƒƒS)z„string_permute takes a pattern and a permutation and returns the string permuted according to the permutation list. rÚc3s|]}ˆ|VqdS)Nr)rÚp)Úpatternrrr\sz-LutBuilder._string_permute..)ÚlenÚAssertionErrorÚjoin)r r Z permutationr)r rÚ_string_permuteWszLutBuilder._string_permutec Csˆ||fg}d|kr…|dd}xYtdƒD]H}|j|j|dddddddddd d g ƒ|fƒq6Wnd |krût|ƒ}x[|d|…D]F\}}|j|j|d ddd dddddg ƒ|fƒq®Wnd |kr„t|ƒ}xn|d|…D]Y\}}|jd dƒjdd ƒjddƒ}dt|ƒ}|j||fƒq$Wn|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.Ú4rrrééééééÚMÚNÚ0ÚZÚ1éÿÿÿÿr1)rÚappendr$r!ÚreplaceÚint) r Z basic_patternÚoptionsZ basic_resultr ÚresrÚnr rrrÚ_pattern_permute^s2            zLutBuilder._pattern_permutec CsÄ|jƒg}x¶|jD]«}tjd|jddƒƒ}|s[td|dƒ‚n|jdƒ}|jdƒ}t|jdƒƒ}|jd dƒjddƒ}||j|||ƒ7}qWx_t |ƒD]Q\}}|d jd d ƒjd d ƒ}tj |ƒ}||df||\s*(\d)Ú rzSyntax error in pattern "ú"rr+r'ú rÚ.ÚXz[01]Nr.rr1)rr ÚreÚsearchr3r Úgroupr4r8Ú enumerateÚcompilerrÚbinr!Úmatchr ) r r rrr5r ÚresultrZ bitpatternÚrrrrÚ build_lutƒs. "%"zLutBuilder.build_lut) Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrrr$r8rGrrrrrs "     %rc@sseZdZdZddddd„Zdd„Zdd„Zd d „Zd d „Zd d„Z dd„Z dS)ÚMorphOpz*A class for binary morphological operatorsNcCs[||_|dk r0td|ƒjƒ|_n'|dk rWtd|ƒjƒ|_ndS)z&Create a binary morphological operatorNr r )r rrG)r r r r rrrr·s    zMorphOp.__init__cCs‹|jdkrtdƒ‚n|jdkr<tdƒ‚ntj|j|jdƒ}tjt|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 imageNzNo operator loadedÚLz0Image must be binary, meaning it must use mode L) r r ÚmoderÚnewÚsizerÚapplyÚbytesÚimÚid)r ÚimageZoutimageÚcountrrrrQÂs$z MorphOp.applycCs[|jdkrtdƒ‚n|jdkr<tdƒ‚ntjt|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.NzNo operator loadedrMz0Image must be binary, meaning it must use mode L)r r rNrrDrRrSrT)r rUrrrrDÑs z MorphOp.matchcCs1|jdkrtdƒ‚ntj|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.rMz0Image must be binary, meaning it must use mode L)rNr rÚ get_on_pixelsrSrT)r rUrrrrWÞszMorphOp.get_on_pixelsc Cs^t|dƒ}t|jƒƒ|_WdQXt|jƒtkrZd|_tdƒ‚ndS)z!Load an operator from an mrl fileÚrbNzWrong size operator file!)ÚopenrÚreadr r!rr )r ÚfilenameÚfrrrÚload_lutès  zMorphOp.load_lutc CsJ|jdkrtdƒ‚nt|dƒ}|j|jƒWdQXdS)zSave an operator to an mrl fileNzNo operator loadedÚwb)r r rYÚwrite)r r[r\rrrÚsave_lutñszMorphOp.save_lutcCs ||_dS)z#Set the lut from an external sourceN)r )r r rrrÚset_lutøszMorphOp.set_lut) rHrIrJrKrrQrDrWr]r`rarrrrrL´s   rLi) Ú __future__rrrrr>rÚobjectrrLrrrrÚs  ¤