0-PhFddlZddlmZddlmZddlmZddlm Z d d Z dS) N)warn) unwrap_1d) unwrap_2d) unwrap_3dFc&|jdvrtdt|tr |g|jz}nht |drIt|t s4t ||jkrtdd|D}ntd|jdkrEtj |rtd|d rtd |jd vrd|j vrtd tj |r?tj tj |tjd g}n!tj|tjd }tjtj|tjd }tj|tjd d}|jdkrt)||n=|jdkrt+|||||n|jdkrt-|||||tj |r'tj|||jS|S)ax Recover the original from a wrapped phase image. From an image wrapped to lie in the interval [-pi, pi), recover the original, unwrapped image. Parameters ---------- image : (M[, N[, P]]) ndarray or masked array of floats The values should be in the range [-pi, pi). If a masked array is provided, the masked entries will not be changed, and their values will not be used to guide the unwrapping of neighboring, unmasked values. Masked 1D arrays are not allowed, and will raise a `ValueError`. wrap_around : bool or sequence of bool, optional When an element of the sequence is `True`, the unwrapping process will regard the edges along the corresponding axis of the image to be connected and use this connectivity to guide the phase unwrapping process. If only a single boolean is given, it will apply to all axes. Wrap around is not supported for 1D arrays. rng : {`numpy.random.Generator`, int}, optional Pseudo-random number generator. By default, a PCG64 generator is used (see :func:`numpy.random.default_rng`). If `rng` is an int, it is used to seed the generator. Unwrapping relies on a random initialization. This sets the PRNG to use to achieve deterministic behavior. Returns ------- image_unwrapped : array_like, double Unwrapped image of the same shape as the input. If the input `image` was a masked array, the mask will be preserved. Raises ------ ValueError If called with a masked 1D array or called with a 1D array and ``wrap_around=True``. Examples -------- >>> c0, c1 = np.ogrid[-1:1:128j, -1:1:128j] >>> image = 12 * np.pi * np.exp(-(c0**2 + c1**2)) >>> image_wrapped = np.angle(np.exp(1j * image)) >>> image_unwrapped = unwrap_phase(image_wrapped) >>> np.std(image_unwrapped - image) < 1e-6 # A constant offset is normal True References ---------- .. [1] Miguel Arevallilo Herraez, David R. Burton, Michael J. Lalor, and Munther A. Gdeisat, "Fast two-dimensional phase-unwrapping algorithm based on sorting by reliability following a noncontinuous path", Journal Applied Optics, Vol. 41, No. 35 (2002) 7437, .. [2] Abdul-Rahman, H., Gdeisat, M., Burton, D., & Lalor, M., "Fast three-dimensional phase-unwrapping algorithm based on sorting by reliability following a non-continuous path. In W. Osten, C. Gorecki, & E. L. Novak (Eds.), Optical Metrology (2005) 32--40, International Society for Optics and Photonics. )rrz$Image must be 1, 2, or 3 dimensional __getitem__z>Length of `wrap_around` must equal the dimensionality of imagec,g|]}t|S)bool).0was Z/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/skimage/restoration/unwrap.py z unwrap_phase..Ps666BtBxx666z[`wrap_around` must be a bool or a sequence with length equal to the dimensionality of imagerz$1D masked images cannot be unwrappedrz,`wrap_around` is not supported for 1D images)rr zqImage has a length 1 dimension. Consider using an array of lower dimensionality to use a more efficient algorithmC)dtypeorderF)rrsubokrr )mask fill_value)ndim ValueError isinstancerhasattrstrlennpma isMaskedArrayshaperrequire getmaskarrayuint8 zeros_likeasarraygetdatafloat64 empty_likerrrarrayr)image wrap_aroundrngrimage_not_maskedimage_unwrappeds r unwrap_phaser2 s}z z""?@@@+t$$  "mej0 m , ,  Z S5Q5Q  {  uz ) )S 76+666  :    zQ 5  u % % ECDD D q> MKLL L zVU[ 0 0      u5!!?z"%,,U33RXuEE}U"(#>>>z"%--"6"6bjPSTTTmE3eTTTO zQ"O4444 q"D/;LLLL q"D/;LLL u5!!u{{?%BR{SSSr)FN) numpyr _shared.utilsr _unwrap_1dr _unwrap_2dr _unwrap_3drr2r rrr8s} !!!!!!!!!!!!!!!!!!jjjjjjr