î úÙXe ã@sbddlmZddlmZddlZd ZGdd„deƒZGdd „d eƒZdS) é)Ú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 úK/home/ubuntu/projects/ifolica/lib/python3.4/site-packages/PIL/ImageMorph.pyÚ__init__2s(            zLutBuilder.__init__cCs|j|7_dS)N)r)r rr r rÚ add_patternsKszLutBuilder.add_patternscsAddg‰d‰t‡‡fdd†ttƒDƒƒ|_dS)Nrrécs$g|]}ˆ|ˆ@dk‘qS)rr )Ú.0Úi)ÚmÚsymbolsr rú Qs z0LutBuilder.build_default_lut..é)Ú bytearrayÚrangeÚLUT_SIZEr )r r )rrrÚbuild_default_lutNs zLutBuilder.build_default_lutcCs|jS)N)r )r r r rÚget_lutSszLutBuilder.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Úcsg|]}ˆ|‘qSr r )rÚp)Úpatternr rr[s z.LutBuilder._string_permute..)ÚlenÚAssertionErrorÚjoin)r rZ permutationr )rrÚ_string_permuteVszLutBuilder._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|ƒ}xr|d|…D]]\}}|jd dƒjdd ƒjddƒ}ddt|ƒ}|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Ú1z%déÿÿÿÿr0)rÚappendr#r ÚreplaceÚint) r Z basic_patternÚoptionsZ basic_resultrÚresrÚnrr r rÚ_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}qWxgt t |ƒƒD]S}||d jd d ƒjd d ƒ}tj |ƒ}|||df||\s*(\d)Ú rzSyntax error in pattern "ú"rr*r&ú rÚ.ÚXz[01]Nr-rr0)rrÚreÚsearchr2r Úgroupr3r7rr ÚcompilerÚbinÚmatchr ) r rrrr4rÚresultrZ bitpatternÚrr r rÚ build_lut‚s. &%"zLutBuilder.build_lut) Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrrr#r7rEr r r rrs "     %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 rrE)r r r rr r rr¶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Úcountr r rrOÁ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 loadedrKz0Image must be binary, meaning it must use mode L)r r rLrrBrPrQrR)r rSr r rrBÐ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.rKz0Image must be binary, meaning it must use mode L)rLr rÚ get_on_pixelsrQrR)r rSr r rrUÝszMorphOp.get_on_pixelsc Cs^t|dƒ}t|jƒƒ|_WdQXt|jƒdkrZd|_tdƒ‚ndS)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 CsJ|jdkrtdƒ‚nt|dƒ}|j|jƒWdQXdS)zSave an operator to an mrl fileNzNo operator loadedÚwb)r r rWÚwrite)r rYrZr r rÚsave_lutðszMorphOp.save_lutcCs ||_dS)z#Set the lut from an external sourceN)r )r r r r rÚset_lut÷szMorphOp.set_lut) rFrGrHrIrrOrBrUr[r^r_r r r rrJ³s   rJi)ZPILrrr=rÚobjectrrJr r r rÚs  ¤