0-Ph.PdZddlZddlmZdZdddddddZd d Zdd Z dd Z dS)z4Utility functions used in the morphology subpackage.N)ndimagecd|d}tj|rtj||}n4tj|t }|j|krtd|Itd|j Drtdtj|j dz}||fS)aConvert any valid connectivity to a footprint and offset. Parameters ---------- image_dim : int The number of dimensions of the input image. connectivity : int, array, or None The neighborhood connectivity. An integer is interpreted as in ``scipy.ndimage.generate_binary_structure``, as the maximum number of orthogonal steps to reach a neighbor. An array is directly interpreted as a footprint and its shape is validated against the input image shape. ``None`` is interpreted as a connectivity of 1. offset : tuple of int, or None The coordinates of the center of the footprint. Returns ------- c_connectivity : array of bool The footprint (structuring element) corresponding to the input `connectivity`. offset : array of int The offset corresponding to the center of the footprint. Raises ------ ValueError: If the image dimension and the connectivity or offset dimensions don't match. Nz,Connectivity dimension must be same as imagec g|] }|dzdk S)r).0xs X/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/skimage/morphology/_util.py z*_validate_connectivity..0s 999qA 999z2Connectivity array must have an unambiguous centerr) npisscalarndigenerate_binary_structurearrayboolndim ValueErroranyshape) image_dim connectivityoffsetc_connectivitys r _validate_connectivityrs<  {<  M6y,OO,55  ) + +KLL L ~ 99N$8999 : : VTUU U..//14 6 !!r rC) footprintrcenterspacingordercjt|}|tj||}|td|jD}|j|cxkrt|ksnt dtjdttj ||Dd}|dkr|dddddf}|ddd}n|d krt d |d dd z}tj |dddddd}||z d } |tj |}||z} tjtj | d zd } | tj| d} tj| d} t#dt||jDr?tj| d\}}tj|d}| |} | |} | d d} | d d} | | fS)auCompute offsets to neighboring pixels in raveled coordinate space. This function also returns the corresponding distances from the center pixel given a spacing (assumed to be 1 along each axis by default). Parameters ---------- image_shape : tuple of int The shape of the image for which the offsets are being computed. footprint : array of bool The footprint of the neighborhood, expressed as an n-dimensional array of 1s and 0s. If provided, the connectivity argument is ignored. connectivity : {1, ..., ndim} The square connectivity of the neighborhood: the number of orthogonal steps allowed to consider a pixel a neighbor. See `scipy.ndimage.generate_binary_structure`. Ignored if footprint is provided. center : tuple of int Tuple of indices to the center of the footprint. If not provided, it is assumed to be the center of the footprint, either provided or generated by the connectivity argument. spacing : tuple of float The spacing between pixels/voxels along each axis. order : 'C' or 'F' The ordering of the array, either C or Fortran ordering. Returns ------- raveled_offsets : ndarray Linear offsets to a samples neighbors in the raveled image, sorted by their distance from the center. distances : ndarray The pixel distances corresponding to each offset. Notes ----- This function will return values even if `image_shape` contains a dimension length that is smaller than `footprint`. Examples -------- >>> off, d = _raveled_offsets_and_distances( ... (4, 5), footprint=np.ones((4, 3)), center=(1, 1) ... ) >>> off array([-5, -1, 1, 5, -6, -4, 4, 6, 10, 9, 11]) >>> d[0] 1.0 >>> d[-1] # distance from (1, 1) to (3, 2) 2.236... N)rankrc3 K|] }|dzV dS)rNrr ss r z1_raveled_offsets_and_distances..xs&77!qAv777777r zQnumber of dimensions in image shape, footprint and itscenter index does not matchcg|] \}}||z  Srr)r idxcs r r z2_raveled_offsets_and_distances..s DDDvsA#'DDDr )axisFrzorder must be 'C' or 'F'rrrstable)kindc3(K|] \}}||kVdS)Nr)r r ys r r'z1_raveled_offsets_and_distances..s* ? ?TQ1q5 ? ? ? ? ? ?r T) return_index)lenrrtuplerrrrstackzipnonzerocumprodsumonessqrtargsortsortrunique) image_shaperrrr r!roffsets ravel_factorsraveled_offsetsweighted_offsets distancessorted_raveled_offsetssorted_distances_indicess r _raveled_offsets_and_distancesrJ8s|x {  D1t,WWW  ~77y77777 >T 0 0 0 0S[[ 0 0 0 0 *   hDDRZ %:%:F!C!CDDD2G ||!!!TTrT'"!$$B$' #3444 Od*MJ}TTrT233DDbD9M.333;;O'$--(/2;;;<> ? ? ???5Y5DIII 7''111!7!@+G44ABB7'+ !#3 33r c8t||||d}|S)aCompute offsets to a samples neighbors if the image would be raveled. Parameters ---------- image_shape : tuple The shape of the image for which the offsets are computed. footprint : ndarray The footprint (structuring element) determining the neighborhood expressed as an n-D array of 1's and 0's. center : tuple Tuple of indices to the center of `footprint`. order : {"C", "F"}, optional Whether the image described by `image_shape` is in row-major (C-style) or column-major (Fortran-style) order. Returns ------- raveled_offsets : ndarray Linear offsets to a samples neighbors in the raveled image, sorted by their distance from the center. Notes ----- This function will return values even if `image_shape` contains a dimension length that is smaller than `footprint`. Examples -------- >>> _offsets_to_raveled_neighbors((4, 5), np.ones((4, 3)), (1, 1)) array([-5, -1, 1, 5, -6, -4, 4, 6, 10, 9, 11]) >>> _offsets_to_raveled_neighbors((2, 3, 2), np.ones((3, 3, 3)), (1, 1, 1)) array([-6, -2, -1, 1, 2, 6, -8, -7, -5, -4, -3, 3, 4, 5, 7, 8, -9, 9]) )rrr!r)rJ)r@rrr!rCs r _offsets_to_raveled_neighborsrLs2F5yu O r Tc`|||}tj||}ntj|t}|j|krt d|r-td|jDrt dtd|jDrt d|S)aValidate or create a footprint (structuring element). Depending on the values of `connectivity` and `footprint` this function either creates a new footprint (`footprint` is None) using `connectivity` or validates the given footprint (`footprint` is not None). Parameters ---------- footprint : ndarray The footprint (structuring) element used to determine the neighborhood of each evaluated pixel (``True`` denotes a connected pixel). It must be a boolean array and have the same number of dimensions as `image`. If neither `footprint` nor `connectivity` are given, all adjacent pixels are considered as part of the neighborhood. connectivity : int 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. ndim : int Number of dimensions `footprint` ought to have. enforce_adjacency : bool A boolean that determines whether footprint must only specify direct neighbors. Returns ------- footprint : ndarray Validated or new footprint specifying the neighborhood. Examples -------- >>> _resolve_neighborhood(None, 1, 2) array([[False, True, False], [ True, True, True], [False, True, False]]) >>> _resolve_neighborhood(None, None, 3).shape (3, 3, 3) N)dtypez7number of dimensions in image and footprint do notmatchc3"K|] }|dkV dS)Nrr%s r r'z(_resolve_neighborhood..s&$E$EQ!V$E$E$E$E$E$Er z$dimension size in footprint is not 3c3(K|] }|dzdkVdS)rrNrr%s r r'z(_resolve_neighborhood..s*77!!a%1*777777r z/footprint size must be odd along all dimensions) rrrasarrayrrrrr)rrrenforce_adjacencys r _resolve_neighborhoodrTsP  L1$ EE Jy555 >T ! !L   P$E$EY_$E$E$E!E!E PCDD D 77y777 7 7 PNOO O r c^tj|r||ff|jz}n't||jkrt dt |D]\}}t|dkrt d|\}}||cxkrdkrnn=||cxkrdkr"nnt df|zdzdz}|||<l|dkr-t df|zt d|fzdz}|||<|dkr.t df|zt | dfzdz}|||<dS) aSet edge values along all axes to a constant value. Parameters ---------- image : ndarray The array to modify inplace. value : scalar The value to use. Should be compatible with `image`'s dtype. border_width : int or sequence of tuples A sequence with one 2-tuple per axis where the first and second values are the width of the border at the start and end of the axis, respectively. If an int is provided, a uniform border width along all axes is used. Examples -------- >>> image = np.zeros((4, 5), dtype=int) >>> _set_border_values(image, 1) >>> image array([[1, 1, 1, 1, 1], [1, 0, 0, 0, 1], [1, 0, 0, 0, 1], [1, 1, 1, 1, 1]]) >>> image = np.zeros((8, 8), dtype=int) >>> _set_border_values(image, 1, border_width=((1, 1), (2, 3))) >>> image array([[1, 1, 1, 1, 1, 1, 1, 1], [1, 1, 0, 0, 0, 1, 1, 1], [1, 1, 0, 0, 0, 1, 1, 1], [1, 1, 0, 0, 0, 1, 1, 1], [1, 1, 0, 0, 0, 1, 1, 1], [1, 1, 0, 0, 0, 1, 1, 1], [1, 1, 0, 0, 0, 1, 1, 1], [1, 1, 1, 1, 1, 1, 1, 1]]) z.length of `border_width` must match image.ndimrz2each sequence in `border_width` must have length 2rrN))rr+).)rrrr4r enumerateslice)imagevalue border_widthr,npadw_startw_endsls r _set_border_valuesr_sH {<  K%|46C \  ej ( (IJJJ -- d t99>>TUU U e q   " " " " " " " " "++$&3frcs:: ."."."h   l4l4l4l4l4^''''T::::z::::::r