|hPxdZddlZddlZddlZddlZddlmZddl Z ddl m Z ddl mZddl mZddlmZd Zej&gd gd gd gd gd gdgdgdgdgdgdgdgdgdgdgdgd gd gd gd gd gZdZGddej,ZdZdZdZdZdZdZej<j?dej@ejBejDejFejHejJejLejNgdZ(dZ)d Z*d!Z+d"Z,d#Z-y)$z0test_watershed.py - tests the watershed functionN)ndimage)gaussian)peak_local_max)label) watershedg-q=)r r r r r r r r r r r r r r r )r r r r r r r r r r r r r r r )r r r r r r r r r r r r r r )r r r r r ogrrr r r r r r )r r r r r^H4rrrrr r r r )r r r r rr'rrrrr r r r )r r r r r rrrrrrr r r r r )r r r r r r rrrrr r r r r r )r r r r r r r rrr r r r r r r )r r r r r r r rrr r r r r r r ct|tjstj|}t|tjstj|}d|jvrd|jvryd||dk(<|j tj tjfvs,|j tj tjfvrtj|tj}tj|tj}|j|jz dzj|j|jz dzjz}n~tj|}|jtj}tj|}|jtj}||z dzj}tj|S)Nrgr) isinstancenpndarrayasarrayshapedtype complex64 complex128realsumimagastypefloat64mathsqrt)abts h/var/www/html/test/engine/venv/lib/python3.12/site-packages/skimage/segmentation/tests/test_watershed.pydiffr)+s^ a $ JJqM a $ JJqM QWW 1<Aa1fIww2<<//177  ?4 JJq"-- ( JJq"-- (ffqvvo! # ( ( *qvv1.D-I-I-K K JJqM HHRZZ  JJqM HHRZZ !e\    99Q<ceZdZejdeZdZdZdZ dZ dZ dZ dZ d Zd Zd Zd Zd ZdZy) TestWatershedr.c tjgdgdgdgdgdgdgdgdgdgdg tj}tjgdgdgdgdgdgdgdgdgdgdg tj}t |||j }tjgdgdgdgdgdgdgdgdgdgdg }t ||}|tksJy) z watershed 1rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr4r4r4r4r4r4r4r4rrrrrr4N)rarrayuint8int8reightr)eps)selfdatamarkersoutexpectederrors r(test_watershed01zTestWatershed.test_watershed01Fsxx%%%%%%%%%%  HH ((&%%%%%%%%%  GG gtzz288,,,''''',,   Xs#s{{r*ctjgdgdgdgdgdgdgdgdgdgdgdg tj}tjgdgdgdgdgdgdgdgdgdgdgdg tj}t ||}t gdgdgdgdgdgdgdgdgdgdgdg |}|j |tky ) z watershed 2r0r1r2r3r5r6)r4r4rrrr4r4r7Nrr8r9r:rr) assertTruer<r=r>r?r@rBs r(test_watershed02zTestWatershed.test_watershed02xsxx%%%%%%%%%%%  HH  ((&%%%%%%%%%%  GG  g&,,,,)'''),,     $r*c tjgdgdgdgdgdgdgdgdgdgdg tj}tjgdgdgdgdgdgdgdgdgdgdg tj}t ||}t gdgdgdgdgdgdgdgdgdgdg |}|j |tky ) z watershed 3r0r1rrrrrrrrrrrr.rrrrrrrrr4r6)r4rrrr.rr4r4rrrr.r.r4NrErGs r(test_watershed03zTestWatershed.test_watershed03sxx%%%%%%%%%%  HH ((%%%%%%%%%&  GG g&,''''',,,,     $r*c tjgdgdgdgdgdgdgdgdgdgdg tj}tjgdgdgdgdgdgdgdgdgdgdg tj}t |||j }t gdgdgdgdgdgdgdgdgdgdg |}|j|tky) z watershed 4r0r1rJrKrLr6rMN rr8r9r:rr;r)rFr<rGs r(test_watershed04zTestWatershed.test_watershed04xx%%%%%%%%%%  HH ((%%%%%%%%%&  GG gtzz2,''''',,,,     $r*c tjgdgdgdgdgdgdgdgdgdgdg tj}tjgdgdgdgdgdgdgdgdgdgdg tj}t |||j }t gdgdgdgdgdgdgdgdgdgdg |}|j|tky) z watershed 5r0r1rJ)rrr.rrrrrLr6)r4r.r.rrrr4NrPrGs r(test_watershed05zTestWatershed.test_watershed05rRr*c tjgdgdgdgdgdgdgdgdgdg tj}tjgdgdgdgdgdgdgdgdgdg tj}t |||j }t gdgdgdgdgdgdgdgdgdg |}|j|tky) z watershed 6r2r1r0r5r3r7r6NrPrGs r(test_watershed06zTestWatershed.test_watershed06Csxx%%%%%%%%%  HH  ((%%%%%%%%&  GG  gtzz2'''',,,,,      $r*cHt}|dk7}tj|jt}d|d<d|d<t |||j |}tj|dk(}tj|dk(}|jt||z dky) z3A regression test of a competitive case that failedr rrrZmaskrYN) blobrzerosrintrr;rrFabsr=r>r^r?r@size1size2s r(test_watershed07zTestWatershed.test_watershed07rss{((4::s+ gtzz= sax sax  EEM*a/0r*cvtj}d|dddf<|dk7}tj|jt }d|d<d|d <t |||j| }tj|dk(}tj|dk(}|jt||z d ky ) z2The border pixels + an edge are all the same valuer rZ r rrXrr[r]rYN) r_copyrr`rrarr;rrFrbrcs r(test_watershed08zTestWatershed.test_watershed08syy{R1W s{((4::s+ gtzz= sax sax  EEM*a/0r*ctjd}tjjdddj t }tjdt }d}|D]\}}d|||f<||||f<|dz }t |dd}t|||jtj|j tj||jy ) zTest on an image of reasonable size This is here both for timing (does it take forever?) and to ensure that the memory constraints are reasonable )rmrrm)drrreflect)sigmamodeN) rr`randomuniformr!rarrr;ndi watershed_iftuint16)r=imagecoordsr?idxxys r(test_watershed09zTestWatershed.test_watershed09s &""1dH5<?  sAE c^tjgdgdgdgdgdgdgdgdgd gd gd gd gd gdgdgdg}|dk(}t|}t||dd}t t dgdD]/\}}|j tj||k(|k(1y)zThe watershed line)r rr r r r r r r r r r r r r )rr rr r r frrrrrr r r r )rr rrr r rrMrrrr r rr)rr r rr r r rrrrr r rrr )rrr rrrr r r r r r rrr r )r rr r r rrrrrrrrr r r)r rrrr r r r r r r r r r rr)r r r rrrrrr rrrrrrr )rrr r r r rrr rrr rr r r )rrrrrr rr r rrrrr rr )r3rrrr rr rrr r r r rr )rrrrr r rr rrrr rrrr )rrrrr rrr rr rr rrrr )rrrrr rr r rrrr rrrr )rrrrr rr rrr r r rrrr )rrrrr rr rr r r r r r r r rrT) connectivitywatershed_linero)"JrrN)rr8rrziprangerFr)r=r> markerbinmarkerwslabareas r(test_watershed12zTestWatershed.test_watershed12sxx$$$$$$$$$$$$$$$_a c H AI y! tV!D IU1X'78 7IC OOBFF29-5 6 7r*cztjjjd}tjdtj}gd|gdgdf<tj |}t ||}tjj||tj||k(rJy)z.Test to ensure input markers are not modified.r)sizer)rrr.ro)rrr)rrrrN) rrs default_rngr`r9rjrtesting assert_equalr)r=rxr?original_markersresults r(!test_watershed_input_not_modifiedz/TestWatershed.test_watershed_input_not_modifieds %%'..H.=((828842>./777+5'*  0':66&G+,,,,r*N)__name__ __module__ __qualname__ronesboolr;rCrHrNrQrTrVrfrkr}rrrrr*r(r,r,Cs_ BGGFD !E0d3%j0%d0%d0%d-%^1 1"H& %@8j7X -r*r,cbtjd}d|dddf<d|ddddf<tjdt}d|d<d|d<t||d }tjgd gd gd gd gd gt}tj j ||t||}tjgd gd gd gd gd gt}tj j ||t||d d }||k7}d||dk(<tj|rJy)NrrYrr.rror)rr)rr{Gz? compactness)rrrrrrrrrrrrT)rrFr)rr`rarr8rrany)rxseedscompactrAnormal compact_wsl differences r(test_compact_watershedrs$ HHV EE!Q$KE!QR%L HHV3 'EE$KE$Ku$7Gxx        HJJGX. ue $Fxx        HJJFH-E5d4PK'J#(J{a vvj!! !!r*ctjd\}}d\}}}}d\}}||z dz||z dzz|dzk}||z dz||z dzz|dzk} tj|| } tj| } t | tj d| } | dddfxxd z cc<tj| jt } d | t| j<tj| \}}t| || }tjj!|| }|dj"dk(sJtj$ddt&j)dd}tj*j-|dj.||dj0dk(sJ|dj0dk(sJy)a Check edge case behavior reported in gh-6632 While we initially viewed the behavior described in gh-6632 [1]_ as a bug, we have reverted that decision in gh-7661. See [2]_ for an explanation. So this test now actually asserts the behavior reported in gh-6632 as correct. .. [1] https://github.com/scikit-image/scikit-image/issues/6632. .. [2] https://github.com/scikit-image/scikit-image/issues/7661#issuecomment-2645810807 )Pr)r,r)rr-) footprintlabelsNrrYrTr])intensity_imageri)startstoprr4rhi)rindices logical_orrudistance_transform_edtrrr`rrtupleTrrskimagemeasure regionpropsextentarangerreshaperrimage_intensity num_pixels)r{r|x1y1x2y2r1r2 mask_circle1 mask_circle2rxdistanceryr^r?_rpropsexpected_regions r("test_watershed_with_markers_offsetr3s ::h DAq#NBB FBFq=AFq=02q58LFq=AFq=02q58L MM, 5E ))%0H H NF 1a4LAL 88HNN$ /D Dvxx4JGQ y' 6F OO ' ' ' JE 8??a  iic?GGANOJJE!H44oF 8  " $$ $ 8  $ && &r*ctjgdgdg}tjgdgdg}tjgdgdg}t||}tjj ||tjgd }tjgd}tjgd }t|||d k7 }tjj ||y ) a Test edge case behavior when markers spill over into another basin / compete. While we initially viewed the behavior described in gh-6632 [1]_ as a bug, we have reverted that decision in gh-7661. See [2]_ for an explanation. So this test now actually asserts the behavior reported in gh-6632 as correct. .. [1] https://github.com/scikit-image/scikit-image/issues/6632. .. [2] https://github.com/scikit-image/scikit-image/issues/7661#issuecomment-2645810807 ) rYrror.rr.rrr) rrrrrrrrr) rrrrrrrrr) rrrrrrrrr) rrrrrrrrrr?)rrrrrrr.)rrrrrrr)rrrrrrrrr?r^N)rr8rrrassert_array_equalrxr?rArs r(%test_watershed_simple_basin_overspillr`sxx4467Exx4467Gxx4467Hug .FJJFH-XX+ , ,Ehh,-Gxx-.H ugEQJ ?FJJ!!&(3r*ctjgd}tjgd}tjgd}t||}tjj ||tjgd}tjgd}t||}tjj ||y)a Edge case: Basins should be distributed evenly between contesting markers. Markers should be prevented from spilling over into another basin and conquering it against other markers with the same claim, just because they get to the basin one step earlier. ) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrrr) rrrrrrrrrrNrr8rrrrs r(+test_watershed_evenly_distributed_overspillrsxx67Exx67Gxx67H ug .FJJFH-xx67Exx67H ug .FJJFH-r*ctjgdgdg}tjgdgdg}tjgdgdg}t||}tjj ||y) zCheck that markers placed at maxima don't conquer other pixels. Regression test for gh-7661 [1]_. .. [1] https://github.com/scikit-image/scikit-image/issues/7661 )rrrr.rorro)rrrr.rororo)rrrrrrrr0)rrrrrrr)rrrrrrrrNrrs r(test_markers_on_maximarsoxx..01Exx..01Gxx..01H ug .FJJFH-r*ctjd}d|ddddf<t|dd}tjgdgdgdgdgdgtj }tj j ||y) z>Test that passing just the number of seeds to watershed works.rrNr.rrrrr)rr`rr8int32rr)rxrrAs r(test_numeric_seed_watershedrsn HHV EE!QR%Lqd3Gxx       hh HJJGX.r*rctjd}tjd|}t||}|j|jk(sJy)N)rnrn)rr`rr)rrxr?r@s r(test_watershed_output_dtypersD HHZ Ehhz5)G E7 #C 99 %% %r*ctjd}tjd}tjt5t ||dddy#1swYyxYw)NrrrZrrpytestraises ValueErrorr)rxr?s r(test_incorrect_markers_shapersG GGFOEggfoG z ""%!"""s  AA#ctjd}tjd}tjt5t |d|dddy#1swYyxYw)Nrrrorr)rxr^s r(test_incorrect_mask_shapersI GGFOE 776?D z "/%.///s AA%ctt}|dk7}t|dd|}tj||dk(sJy)Nr r)rr^r)r_rrrr>r^r@s r(test_markers_in_maskrs< D 3;D D"14 8C 66#te*/ "" "r*cht}|dk7}t||}tj|dk(sJy)Nr r]r)r_rrmaxrs r(test_no_markersrs2 D 3;D Dt $C 66#;!  r*ctjd\}}d\}}}}}}}} d\} } } } ||z dz||z dzz| dzk}||z dz||z dzz| dzk}||z dz||z dzz| dzk}||z dz|| z dzz| dzk}tj||}tj||}tj||}tj|}tj |dz ddz}tj |dk(dd}td |z |dd }td |z |dd }tj|jdd k(sJtj|jdd k(sJttd gdD]"\}}tj||k(|k(r"Jttd gdD]"\}}tj||k(|k(r"Jy)z Watershed segmentation should output different result for different connectivity when markers are calculated where None is supplied. Issue = 5084 )ii)i,xrnrniTr )rn2(rr r)decimalsrr r)r^rrrYr)EiOiY+ii,)rriOib0i,N) rrrruraroundwhereruniquerrrr)r{r|rrrrx3y3x4y4rrr3r4rr mask_circle3 mask_circle4rxDummyBT DummyBT_disImg_mask labels_c1 labels_c2rrs r(test_connectivityr"s+ ::j !DAq%K"BBBB$NBBFq=AFq=02q58LFq=AFq=02q58LFq=AFq=02q58LFq=AFq=02q58L MM, 5E MM% .E MM% .E((/G))GbL15:Kxx q(!Q/H kqdI kqdI 99Y  % %a (A -- - 99Y  % %a (A -- -q#KL0 Tvvi3&'4///0q#EF0 Tvvi3&'4///0r*).__doc__r#unittestnumpyrrscipyrruskimage.measurerskimage._shared.filtersrskimage.featurerr _watershedrr<r8r_r)TestCaser,rrrrrrmark parametrizer9r:rwint16uint32ruint64int64rrrrrr"rr*r(r2s/6  ,*!" rxxaaaaaaaaaaaaaaaaaaaaa)cd00 -H%% -D+"\*'Z4@.,."/$ XXrww 288RYY"))RXXV& &"/#*0r*