ó î Xc@s\ddlmZddlmZddlZd Zdd d„ƒYZdd d „ƒYZdS( iÿÿÿÿ(tImage(t _imagingmorphNii t LutBuildercBsSeZdZddd„Zd„Zd„Zd„Zd„Zd„Z d„Z RS( s~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() cCs¾|dk r||_n g|_d|_|dk rºiddgd6dgd6ddgd6dgd 6dd gd 6dd d gd6}||krªtd|dƒ‚n|||_ndS(Ns1:(... ... ...)->0s4:(00. 01. ...)->1tcorners4:(... .0. .1.)->1t dilation4s4:(... .0. ..1)->1t dilation8s4:(... .1. .0.)->0terosion4s4:(... .1. ..0)->0terosion8s4:(.0. .1. ...)->1s4:(01. .1. ...)->1tedgesUnknown pattern t!(tNonetpatternstlutt Exception(tselfR top_nametknown_patterns((s2/usr/lib/python2.7/dist-packages/PIL/ImageMorph.pyt__init__2s(            cCs|j|7_dS(N(R (RR ((s2/usr/lib/python2.7/dist-packages/PIL/ImageMorph.pyt add_patternsKscCsLddg}d}tgttƒD]}|||@dk^q"ƒ|_dS(Niiii(t bytearraytrangetLUT_SIZER (Rtsymbolstmti((s2/usr/lib/python2.7/dist-packages/PIL/ImageMorph.pytbuild_default_lutNs cCs|jS(N(R (R((s2/usr/lib/python2.7/dist-packages/PIL/ImageMorph.pytget_lutSscCs<t|ƒdkst‚djg|D]}||^q%ƒS(s„string_permute takes a pattern and a permutation and returns the string permuted according to the permutation list. i t(tlentAssertionErrortjoin(Rtpatternt permutationtp((s2/usr/lib/python2.7/dist-packages/PIL/ImageMorph.pyt_string_permuteVsc Cs†||fg}d|kr…|dd}xYtdƒD]H}|j|j|ddddddddd d d g ƒ|fƒq6Wnd |krøt|ƒ}xX|d|!D]F\}}|j|j|d ddd ddd ddg ƒ|fƒq«Wnd |kr‚t|ƒ}xo|d|!D]]\}}|jddƒjddƒjddƒ}ddt|ƒ}|j||fƒqWn|S(sÉ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.t4iÿÿÿÿiiiiiiiiitMtNt0tZt1s%d(RtappendR"Rtreplacetint( Rt basic_patterntoptionst basic_resultR tresRtnR((s2/usr/lib/python2.7/dist-packages/PIL/ImageMorph.pyt_pattern_permute]s2            c 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)s RsSyntax error in pattern "t"iiit it.tXs[01]R&i Niÿÿÿÿ(RR tretsearchR*R tgroupR+R1RRtcompileRtbintmatchR ( RR R!RR-RtresultRt bitpatterntr((s2/usr/lib/python2.7/dist-packages/PIL/ImageMorph.pyt build_lut‚s. &%"N( t__name__t __module__t__doc__R RRRRR"R1R?(((s2/usr/lib/python2.7/dist-packages/PIL/ImageMorph.pyRs"     %tMorphOpcBsVeZdZdddd„Zd„Zd„Zd„Zd„Zd„Z d„Z RS( s*A class for binary morphological operatorscCs[||_|dk r0td|ƒjƒ|_n'|dk rWtd|ƒjƒ|_ndS(s&Create a binary morphological operatorRR N(R R RR?(RR RR ((s2/usr/lib/python2.7/dist-packages/PIL/ImageMorph.pyR¶s    cCsm|jdkrtdƒ‚ntj|j|jdƒ}tjt |jƒ|j j |j j ƒ}||fS(s‡Run a single morphological operation on an image Returns a tuple of the number of changed pixels and the morphed imagesNo operator loadedN( R R R RtnewtmodetsizeRtapplytbytestimtid(Rtimagetoutimagetcount((s2/usr/lib/python2.7/dist-packages/PIL/ImageMorph.pyRGÁs $cCs=|jdkrtdƒ‚ntjt|jƒ|jjƒS(sª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.sNo operator loadedN(R R R RR;RHRIRJ(RRK((s2/usr/lib/python2.7/dist-packages/PIL/ImageMorph.pyR;ÎscCstj|jjƒS(s‹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.(Rt get_on_pixelsRIRJ(RRK((s2/usr/lib/python2.7/dist-packages/PIL/ImageMorph.pyRNÙscCs^t|dƒ}t|jƒƒ|_WdQXt|jƒdkrZd|_tdƒ‚ndS(s!Load an operator from an mrl filetrbNi sWrong size operator file!(topenRtreadR RR R (Rtfilenametf((s2/usr/lib/python2.7/dist-packages/PIL/ImageMorph.pytload_lutás  cCsJ|jdkrtdƒ‚nt|dƒ}|j|jƒWdQXdS(sSave an operator to an mrl filesNo operator loadedtwbN(R R R RPtwrite(RRRRS((s2/usr/lib/python2.7/dist-packages/PIL/ImageMorph.pytsave_lutêscCs ||_dS(s#Set the lut from an external sourceN(R (RR ((s2/usr/lib/python2.7/dist-packages/PIL/ImageMorph.pytset_lutñsN( R@RARBR RRGR;RNRTRWRX(((s2/usr/lib/python2.7/dist-packages/PIL/ImageMorph.pyRC³s   i(((tPILRRR6RRRC(((s2/usr/lib/python2.7/dist-packages/PIL/ImageMorph.pyts  ¤