"c)fddlZddlmZmZdZgdZgdZGddZGd d ZdS) N)Image _imagingmorphi) rr) r rrr r rr rrc>eZdZdZd dZdZdZdZdZdZ 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() Nc|||_ng|_d|_|=ddgdgddgdgddggdd}||vrd |zd z}t||||_dSdS) 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 !)patternslut Exception)selfrop_nameknown_patternsmsgs 0/usr/lib/python3/dist-packages/PIL/ImageMorph.py__init__zLutBuilder.__init__@s  $DMMDM  /1EF2324HI1213GH  Nn,,(72S8nn$*73DMMM#  c&|xj|z c_dSNr)rrs r add_patternszLutBuilder.add_patternsYs ! r!c|ddgdtfdttD|_dS)Nrrc36K|]}|zdkVdS)rN).0imsymbolss r z/LutBuilder.build_default_lut.._s0KKaWa!eq[1KKKKKKr!) bytearrayrangeLUT_SIZEr)rr,r-s @@rbuild_default_lutzLutBuilder.build_default_lut\sCa& KKKKK5??KKKKKr!c|jSr#r)rs rget_lutzLutBuilder.get_lutas xr!cpt|dksJdfd|DS)zstring_permute takes a pattern and a permutation and returns the string permuted according to the permutation list. c3(K|] }|V dSr#r))r*ppatterns rr.z-LutBuilder._string_permute..is'77awqz777777r!)lenjoin)rr; permutations ` r_string_permutezLutBuilder._string_permutedsE;1$$$$ww7777;777777r!c||fg}d|vr\|dd}tdD]>}|||ddt|f?d|vrOt |}|d|D]5\}}|||t |f6d|vrt |}|d|D]l\}}|d d d d d d }dt|z }|||fm|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.4rr rMNN0Z1)r0appendr?ROTATION_MATRIXr< MIRROR_MATRIXreplaceint) r basic_patternoptions basic_resultrresr+nr;s r_pattern_permutezLutBuilder._pattern_permuteksh#L12 '>>2,q/C1XX  ))(2,q/?KKSQ '>>H A (!  U U !5!5g}!M!Ms STTTT '>>H A (!  0 0 !//#s33;;CEEMMcSVWW#c((l#////r!c|g}|jD]}tjd|dd}|sd|zdz}t ||d}|d}t|d}|d ddd}|||||z }t|D]V\}}|d d d d d }tj |}||df||<WttD]i}t|dd} ddt| z z| zddd} |D],\}} || rd dg| |j|<-j|jS)zlCompile all patterns into a morphology lut. TBD :Build based on (file) morphlut:modify_lut z(\w*):?\s*\((.+?)\)\s*->\s*(\d) r8zSyntax error in pattern ""rr r r.Xz[01]NrEr7rB)r2rresearchrKrgrouprLrR enumeratecompiler0r1binr<matchr) rrr:r,rrNr;resultr+ bitpatternrs r build_lutzLutBuilder.build_luts     H HA Q>QRRA %1A5;nn$ggajjGggajjG__Fooc2..66tR@@G --gwGG GHH$H-- * *JAw ""3,,44S&AAA 1 Agaj/HQKK x , ,AQJS__!45 BDDbDIJ  , ,177:&&,#$a&)DHQK ,xr!)NN) __name__ __module__ __qualname____doc__r r%r2r5r?rRrcr)r!rrrs!!F44442"""LLL 888<)))))r!rc>eZdZdZd dZdZdZdZdZdZ d Z dS) MorphOpz*A class for binary morphological operatorsNc||_|)t||_dS|)t||_dSdS)z&Create a binary morphological operatorN)rr$)rrrc)rrrrs rr zMorphOp.__init__sa  !'222<<>>DHHH  !!8444>>@@DHHH" !r!c*|jd}t||jdkrd}t|t j|j|jd}tjt|j|j j |j j }||fS)zRun a single morphological operation on an image Returns a tuple of the number of changed pixels and the morphed imageNNo operator loadedLImage mode must be L) rrmode ValueErrorrnewsizerapplybytesimid)rimageroutimagecounts rrsz MorphOp.applys 8 &CC.. :  (CS// !9UZT::#E$(OOUX[(+.QQhr!c|jd}t||jdkrd}t|t jt |j|jjS)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`.Nrlrmrn) rrrorprr_rtrurvrrwrs rr_z MorphOp.matchsZ 8 &CC.. :  (CS// !"5??EHK@@@r!cv|jdkrd}t|tj|jjS)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`.rmrn)rorpr get_on_pixelsrurvr{s rr}zMorphOp.get_on_pixelss7 :  (CS// !*58;777r!c t|d5}t||_dddn #1swxYwYt |jt krd|_d}t |dS)z!Load an operator from an mrl filerbNzWrong size operator file!)openr/readrr<r1r)rfilenamefrs rload_lutzMorphOp.load_luts (D ! ! +Q **DH + + + + + + + + + + + + + + + tx==H $ $DH-CC..  % $s'AA Ac|jd}t|t|d5}||jddddS#1swxYwYdS)zSave an operator to an mrl fileNrlwb)rrrwrite)rrrrs rsave_lutzMorphOp.save_luts 8 &CC.. (D ! ! Q GGDH                     sAAAc||_dS)z#Set the lut from an external sourceNr4)rrs rset_lutzMorphOp.set_luts r!)NNN) rdrerfrgr rsr_r}rrrr)r!rriris44AAAA A A A 8 8 8!!!r!ri) rYr8rrr1rIrJrrir)r!rrs """"""""   VVVVVVVVrIIIIIIIIIIr!