0-Ph)pdZddlZddlmZddlmZmZddlm Z m Z m Z ddl m Z dddd d d Zdddd d ZdS)zflood_fill.py - in place flood fill algorithm This module provides a function to fill all equal (or within tolerance) values connected to a given seed point with a different value. N)crop)_flood_fill_equal_flood_fill_tolerance)_offsets_to_raveled_neighbors_resolve_neighborhood_set_border_values)numeric_dtype_min_maxF) footprint connectivity tolerancein_placecdt|||||}|s|}|||<|S)a Perform flood filling on an image. Starting at a specific `seed_point`, connected points equal or within `tolerance` of the seed value are found, then set to `new_value`. Parameters ---------- image : ndarray An n-dimensional array. seed_point : tuple or int The point in `image` used as the starting point for the flood fill. If the image is 1D, this point may be given as an integer. new_value : `image` type New value to set the entire fill. This must be chosen in agreement with the dtype of `image`. footprint : ndarray, optional The footprint (structuring element) used to determine the neighborhood of each evaluated pixel. It must contain only 1's and 0's, have the same number of dimensions as `image`. If not given, all adjacent pixels are considered as part of the neighborhood (fully connected). connectivity : int, optional A number used to determine the neighborhood of each evaluated pixel. Adjacent pixels whose squared distance from the center is less than or equal to `connectivity` are considered neighbors. Ignored if `footprint` is not None. tolerance : float or int, optional If None (default), adjacent values must be strictly equal to the value of `image` at `seed_point` to be filled. This is fastest. If a tolerance is provided, adjacent points with values within plus or minus tolerance from the seed point are filled (inclusive). in_place : bool, optional If True, flood filling is applied to `image` in place. If False, the flood filled result is returned without modifying the input `image` (default). Returns ------- filled : ndarray An array with the same shape as `image` is returned, with values in areas connected to and equal (or within tolerance of) the seed point replaced with `new_value`. Notes ----- The conceptual analogy of this operation is the 'paint bucket' tool in many raster graphics programs. Examples -------- >>> from skimage.morphology import flood_fill >>> image = np.zeros((4, 7), dtype=int) >>> image[1:3, 1:3] = 1 >>> image[3, 0] = 1 >>> image[1:3, 4:6] = 2 >>> image[3, 6] = 3 >>> image array([[0, 0, 0, 0, 0, 0, 0], [0, 1, 1, 0, 2, 2, 0], [0, 1, 1, 0, 2, 2, 0], [1, 0, 0, 0, 0, 0, 3]]) Fill connected ones with 5, with full connectivity (diagonals included): >>> flood_fill(image, (1, 1), 5) array([[0, 0, 0, 0, 0, 0, 0], [0, 5, 5, 0, 2, 2, 0], [0, 5, 5, 0, 2, 2, 0], [5, 0, 0, 0, 0, 0, 3]]) Fill connected ones with 5, excluding diagonal points (connectivity 1): >>> flood_fill(image, (1, 1), 5, connectivity=1) array([[0, 0, 0, 0, 0, 0, 0], [0, 5, 5, 0, 2, 2, 0], [0, 5, 5, 0, 2, 2, 0], [1, 0, 0, 0, 0, 0, 3]]) Fill with a tolerance: >>> flood_fill(image, (0, 0), 5, tolerance=1) array([[5, 5, 5, 5, 5, 5, 5], [5, 5, 5, 5, 2, 2, 5], [5, 5, 5, 5, 2, 2, 5], [5, 5, 5, 5, 5, 5, 3]]) r r r)floodcopy)image seed_point new_valuer r rrmasks ^/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/skimage/morphology/_flood_fill.py flood_fillrsM~  !    D  E$K Lrc tj|}|jjdurd}n'|jjdurd}ntj|}d}d|jvr tj|jtS t|n#t$r|f}YnwxYw||}ttj||jz}t|||j d}td|jD}d ttj||D}tj||d | } tjd t||D| j| } t'| j|||} tj| jtj|} t+| d| |t-|}t/|j\} }t3| ||z }t#|||z}t7| || || | |||n9t;| || || | |n3#t$r&| jtjkrtdwxYwt?| |d tS)a/ Mask corresponding to a flood fill. Starting at a specific `seed_point`, connected points equal or within `tolerance` of the seed value are found. Parameters ---------- image : ndarray An n-dimensional array. seed_point : tuple or int The point in `image` used as the starting point for the flood fill. If the image is 1D, this point may be given as an integer. footprint : ndarray, optional The footprint (structuring element) used to determine the neighborhood of each evaluated pixel. It must contain only 1's and 0's, have the same number of dimensions as `image`. If not given, all adjacent pixels are considered as part of the neighborhood (fully connected). connectivity : int, optional A number used to determine the neighborhood of each evaluated pixel. Adjacent pixels whose squared distance from the center is less than or equal to `connectivity` are considered neighbors. Ignored if `footprint` is not None. tolerance : float or int, optional If None (default), adjacent values must be strictly equal to the initial value of `image` at `seed_point`. This is fastest. If a value is given, a comparison will be done at every point and if within tolerance of the initial value will also be filled (inclusive). Returns ------- mask : ndarray A Boolean array with the same shape as `image` is returned, with True values for areas connected to and equal (or within tolerance of) the seed point. All other values are False. Notes ----- The conceptual analogy of this operation is the 'paint bucket' tool in many raster graphics programs. This function returns just the mask representing the fill. If indices are desired rather than masks for memory reasons, the user can simply run `numpy.nonzero` on the result, save the indices, and discard this mask. Examples -------- >>> from skimage.morphology import flood >>> image = np.zeros((4, 7), dtype=int) >>> image[1:3, 1:3] = 1 >>> image[3, 0] = 1 >>> image[1:3, 4:6] = 2 >>> image[3, 6] = 3 >>> image array([[0, 0, 0, 0, 0, 0, 0], [0, 1, 1, 0, 2, 2, 0], [0, 1, 1, 0, 2, 2, 0], [1, 0, 0, 0, 0, 0, 3]]) Fill connected ones with 5, with full connectivity (diagonals included): >>> mask = flood(image, (1, 1)) >>> image_flooded = image.copy() >>> image_flooded[mask] = 5 >>> image_flooded array([[0, 0, 0, 0, 0, 0, 0], [0, 5, 5, 0, 2, 2, 0], [0, 5, 5, 0, 2, 2, 0], [5, 0, 0, 0, 0, 0, 3]]) Fill connected ones with 5, excluding diagonal points (connectivity 1): >>> mask = flood(image, (1, 1), connectivity=1) >>> image_flooded = image.copy() >>> image_flooded[mask] = 5 >>> image_flooded array([[0, 0, 0, 0, 0, 0, 0], [0, 5, 5, 0, 2, 2, 0], [0, 5, 5, 0, 2, 2, 0], [1, 0, 0, 0, 0, 0, 3]]) Fill with a tolerance: >>> mask = flood(image, (0, 0), tolerance=1) >>> image_flooded = image.copy() >>> image_flooded[mask] = 5 >>> image_flooded array([[5, 5, 5, 5, 5, 5, 5], [5, 5, 5, 5, 2, 2, 5], [5, 5, 5, 5, 2, 2, 5], [5, 5, 5, 5, 5, 5, 3]]) TFCr)dtypeF)enforce_adjacencyc3 K|] }|dzV dS)rN).0ss r zflood..s&33a16333333rcng|]2\}}tjtj||z fdz3S)r)npmaxabs)r"idxcs r zflood..sG+13sQw "Q&rconstant)modeconstant_valuesc$g|] \}\}}||zSr!r!)r"i pad_startpad_ends rr+zflood..s&QQQ212y'YQQQr)order)centerr3)rr3r)value border_widthNzLdtype of `image` is float16 which is not supported, try upcasting to float32)r)!r&asarrayflags f_contiguous c_contiguousascontiguousarrayshapezerosbooliter TypeErrortupler ndimzipnonzeropadminravel_multi_indexruint8r r(r rr'itemrravelrfloat16rview)rrr r rr3 seed_valuer4 pad_width working_imageravelled_seed_idxneighbor_offsetsr8 min_value max_valuelow_tolhigh_tols rrrs| Ju  E {4''  !T ) )$U++ EKx 40000# Z ### ] #z"Jrz*-- ;<>>>"  II$99I#J#J Iy)..**JOO,=,= ,IJJG9>>++Z__->->-JKKH !##E** E"" !     ##E** E"" !       "* , ,6   yu - - - 2 24 8 88s7B BBD K((0L)__doc__numpyr&utilr_flood_fill_cyrr_utilrr r _shared.dtyper rrr!rrr\s DDDDDDDD 211111 kkkkk\+/TTu9u9u9u9u9u9u9r