o h @sTddlmZddlmZmZddlZerddlmZ ddd d Z ddddZ dS)) annotations) TYPE_CHECKINGAnyN)CoordinateArrayFshapetuple[int, int] want_maskboolreturnVtuple[CoordinateArray, CoordinateArray, CoordinateArray | np.ma.MaskedArray[Any, Any]]c Csr|\}}tj|tjd}tj|tjd}t||\}}|d}|d}tgd}tddgddgdd gd dgd d gg} tgd } t|} tt|D]*} | || t||| | d fd||| | dfd | | d7} qQ|rt ||ddd||ddddk||ddd||ddddk} tj j | | d} ||| fS)aiReturn simple test data consisting of the sum of two gaussians. Args: shape (tuple(int, int)): 2D shape of data to return. want_mask (bool, optional): Whether test data should be masked or not, default ``False``. Return: Tuple of 3 arrays: ``x``, ``y``, ``z`` test data, ``z`` will be masked if ``want_mask=True``. dtype?)rg皙?gffffff?皙?皙?g333333?rg?g?rg?)rrrr皙?rrg{Gz?g?g{Gz?mask) nparangefloat64meshgridasarray zeros_likerangelenexp logical_ormaarray)rrnynxxyxscaleyscaleampmidwidthzirr0g/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/contourpy/util/data.pysimple s& ( R** r2rseedint mask_fractionfloatc Cs|\}}tj|tjd}tj|tjd}t||\}}tj|}|j|d}|dkrAt|d}|j|d|k} tjj || d}|||fS)aReturn random test data in the range 0 to 1. Args: shape (tuple(int, int)): 2D shape of data to return. seed (int, optional): Seed for random number generator, default 2187. mask_fraction (float, optional): Fraction of elements to mask, default 0. Return: Tuple of 3 arrays: ``x``, ``y``, ``z`` test data, ``z`` will be masked if ``mask_fraction`` is greater than zero. r )sizergGz?r) rrrrrandom default_rnguniformminr#r$) rr4r6r%r&r'r(rngr.rr0r0r1r93s    r9)F)rrrr r r )r3r)rrr4r5r6r7r r ) __future__rtypingrrnumpyrcontourpy._contourpyrr2r9r0r0r0r1s   )