o ^i @s ddlZddlZddlZddlmZmZmZmZm Z ddl m Z ddl m Z ddl mZddl mZddlmZddlmZdd lmZdd lmZdd lmZejd ejejgd dZddZddZ ddZ!ddZ"ddZ#ddZ$ddZ%ddZ&ejd ej'ejej(gdd Z)ejd ej'ejej(gd!d"Z*d#d$Z+ejd%d d&gejd ej,ej-ej(gejd'gd(d)d*Z.ej/0de 1Z2e3e2Z4e5e4d+d,Z4d-d.Z6d/d0Z7ejd ej'ejej(gd1d2Z8d3d4Z9d5d6Z:ejd7d&dd8gfd dd9gfd:d;dZ;ejd7d&d?d@gfd dAd8gfdBdCdDgfgdEdFZdJZ?dKZ@dLZAdMdNZBdOdPZCejd ej'ejej(gdQdRZDdSdTZEejd ejejFej'ejej(gdUdVZGdWdXZHdYdZZId[d\ZJd]d^ZKd_d`ZLdadbZMdcddZNejjOe ejPe dekdfdgejdhdidjgdkdlZQejdmdnej,fdoejRfdpejRfdqejRfdreSfgdsdtZTdudvZUdwdxZVdydzZWejd ej'ejej(gd{d|ZXd}d~ZYddZZddZ[ddZ\ddZ]ddZ^ddZ_ddZ`ddZaddZbddZcejd ej'ejej(gddZdddZeddZfddZgddZhejd ej'ejej(gddZiddZjddZkejd ej'ejej(gddZlddZmddZnddZoddZpddZqddZrejdejsejtejugddZvddZwddZxdS)N)assert_allcloseassert_almost_equalassert_array_almost_equalassert_array_equal assert_equal)Version)data)exposure)util)rgb2gray)intensity_range) dtype_range)expected_warnings)_supported_float_typedtypecCsXtjddg|d}tjtddtj|dd\}}WddS1s%wYdS)Ndrz+Incorrect value for `source_range` argumentmatchfoobar source_range)nparraypytestraises ValueErrorr histogramrim frequencies bin_centersr#_/var/www/html/RAG/RAG_venv/lib/python3.10/site-packages/skimage/exposure/tests/test_exposure.pytest_wrong_source_ranges"r%cCsjtjddgtjd}t|\}}t|tdd|ddks"J|ddks*Jt|ddddS)Nrrrerrrint8r rraranger r!r"r#r#r$test_negative_overflow# r,cCsjtjddgtjd}t|\}}t|tdd|ddks"J|ddks*Jt|ddddS)Nirrrr'r(r+r#r#r$test_all_negative_image,r-r.cCsTtjddgtjd}t|\}}tt|t|t|ddt|dddS)N rrrr)rrr)r rrlenr+r#r#r$test_int_range_image5s r1cCsxtjddgddggtjd}tj|dd\}}t|jdD] }tt||t|qt|ddt|dddS) Nr/rfrr) channel_axisr) rrr)r rrangeshaperr0)r r!r"chr#r#r$!test_multichannel_int_range_image=s r8cCsrtjddgtjd}tj|dd\}}t|tddt|ddt|ddt|d dt|jd dS) Nr/rrrrrr'r&r9) rruint8r rrr*rr6r+r#r#r$test_peak_uint_range_dtypeFr<cCsrtjddgtjd}tj|dd\}}t|tddt|dd t|d d t|d d t|jd dS)Nr/rrrrr'rr:) rrr)r rrr*rr6r+r#r#r$test_peak_int_range_dtypePr=rCcCsHtjdddtjd}tj|dd\}}t|tddt|jddS)Nrr9rrrr:) rlinspacer;r rrr*rr6r+r#r#r$test_flat_uint_range_dtypeZrFcCsHtjdddtjd}tj|dd\}}t|tddt|jddS)Nr>r?r9rrrr:) rrEr)r rrr*rr6r+r#r#r$test_flat_int_range_dtypearGrHcCsLtjddg|d}tj|dd\}}|j|ksJt|tddddS)Nr/rrZnbins?)rrr rrrr*rr#r#r$"test_peak_float_out_of_range_imagehsrMcCsntjddg|d}d}tj||dd\}}|j|ksJtt|ddtt|ddtt |ddS) Nr/rrr)rKrg?) rrr rrrminmaxrr0)rr rKr!r"r#r#r$"test_peak_float_out_of_range_dtypeqsrRcCsvtjgdtjd}tj|ddd\}}td}d|d<d |d <t||tj|dd d\}}|d }t||dS) N)rrDrDrrF)r normalizer9r'rrTg@)rrr;r rzerosr)r r!r"expectedr#r#r$test_normalize|s   rWrimager4)rr'rc Csd}|d|d}t|\}}tjtj|||dtj|||df|d}tj|||d\}} t|tjrAt| t ||d|dd|ksKJ|dddksUJ|dddks_J|dd|ksiJdS) z-Check that all channels use the same binning.)r2r2rr'r)axis)rr4rN) r rstackfullr r issubdtypeintegerrr*) rrr4r6 channel_sizeiminimaxr r!r"r#r#r$(test_multichannel_hist_common_bins_uint8s$  rag@rcCs&tt}tjtdd}t||dS)z)Check integer bins used for uint8 images.rNrJN)r equalize_hist test_img_intr)img_eq0img_eq1r#r#r$test_equalize_uint8_approxs rfcCs.tt}t|}t|\}}t|dS)N)r img_as_ubytetest_imgr rbcumulative_distributioncheck_cdf_slope)imgimg_eqcdf bin_edgesr#r#r$test_equalize_ubytes   rocCs\ttj|dd}t|}|jt|ksJt|\}}t ||jt|ks,JdS)NFcopy) r img_as_floatrhastyper rbrrrirj)rrkrlrmrnr#r#r$test_equalize_floats  rtcCsltt}ttj}d|ddddf<tj||d}t|}t|\}}t |||k r4JdS)Nr'ri)mask) r rrrhrrUr6r rbrirjall)rkru img_mask_eqrlrmrnr#r#r$test_equalize_maskeds   rxcCsDtddt|}t||d\}}d|krdks JJdS)z=Slope of cdf which should equal 1 for an equalized histogram.rr'rO皙?N)rrEr0polyfit)rmnorm_intensityslope interceptr#r#r$rjs rjztest_input,expectedr'rD)r/r/r~cC.tjddgtjd}t||d}t||dS)Nrr'r range_values)rrr;r r test_inputrVrXoutr#r#r$test_intensity_range_uint8 r皙?皙?r)333333?皙?rrcCr)Nrrrrrrfloat64r rrr#r#r$test_intensity_range_floatrrcCs0tjddgtjd}t|ddd}t|ddS)NrrrrT)r clip_negative)rr'rrXrr#r#r$"test_intensity_range_clipped_floatsriii?icCs@tjgdtjd}t|}|jtjksJt|gddS)N3r3rrrD)rrr;r rescale_intensityrrrr#r#r$test_rescale_stretch s rcCs*tgd}t|}t|gddS)NgI@gY@g c@)rrLr'rrr rrrr#r#r$test_rescale_shrinks rcCsFtjgd|d}tj|dd}t|gddd|j|jks!JdS)NrrrrDin_range)rrg333333?)decimal)rrr rrrrrXrr#r#r$test_rescale_in_rangesrcCs.tgd}tj|dd}t|gddS)Nr)rr3r)rLr'r'rrr#r#r$test_rescale_in_range_clip"srcCsFtjgd|d}tj|dd}|jt|jksJt|gddS)a?Check that output range is correct. .. versionchanged:: 0.17 This function used to return dtype matching the input dtype. It now matches the output. .. versionchanged:: 0.19 float16 and float32 inputs now result in float32 output. Formerly they would give float64 outputs. )rr/rrr out_range)rgO@rN)rrr rrrrrr#r#r$test_rescale_out_range(srcCs<tjdttdgtjd}tj|dd}t|dttgdS)Nrrruint10r)rr uint10_maxuint16r rr uint16_maxrr#r#r$test_rescale_named_in_range<srcC4tjdtgtjd}tj|dd}t|dtgdS)Nrrrr)rrrrr rrrrr#r#r$test_rescale_named_out_rangeBrcCr)Nrruint12r)rrrrr rr uint12_maxrr#r#r$test_rescale_uint12_limitsHrrcCr)Nrruint14r)rrrrr rr uint14_maxrr#r#r$test_rescale_uint14_limitsNrrcCs<tjdtjd}t|}t|sJt||dS)NrTrTr)rrUr;r risnanrvrrr#r#r$test_rescale_all_zerosTs rcCs4tjddgtjd}tj|dd}t|ddgdS)Nrrrr)rrrr rrrr#r#r$test_rescale_constant[rrcCs6td}t|}t|sJt||dS)Nr)ronesr rrrvrrr#r#r$test_rescale_same_valuesas  rz1.25z,Older NumPy throws a few extra warnings herereasonzin_range,out_range)rXr)rrXcCs^tjdtddd}tj|d<tdgt|||WddS1s(wYdS)N rrNr)r'r'zWOne or more intensity levels are NaN\. Rescaling will broadcast NaN to the full image\.)rr*floatreshapenanrr r)rrrXr#r#r$test_rescale_nan_warninghs "rzout_range, out_dtyper;rrrrcCs4tjgdtjd}tj||d}|j|ksJdS)Nr>rrrr)rrr)r rr)r out_dtyperX output_imager#r#r$test_rescale_output_dtypezs rcCsFtjgdtjd}tj|tjd}t|gd|jtjks!JdS)Nrrrrr?rD)rrr)r rr;rrrXrr#r#r$test_rescale_no_overflowsrcCsBtjgdtjd}tj|dd}t|gd|jtksJdS)Nrrrrr)rrr)r rrrrrr#r#r$test_rescale_float_outputsrcCsRtjgdtjd}tttj|dd}WddS1s"wYdS)Nrrflatr)rrr)rrrr r)rX_r#r#r$*test_rescale_raises_on_incorrect_out_ranges "rcCsttj|dd}t|}t|||f}tj |dddd}|j |j ks)J|j t |ks2J|tj kr9dnd}tt||d |tt||d dd S) zTest a grayscale float imageFrp)9r{Gz?r?) kernel_size clip_limitrKrNrTg)\Y@ghs?N)r rrr astronautrsr rdstackr equalize_adapthistr6rrfloat16rpeak_snrnorm_brightness_err)rrkadapted snr_decimalr#r#r$test_adapthist_grayscalesrcCstt}tjdd}tdt|\}}t |dks"JWdn1s,wYtj |dd}| dks@J| dksHJ|j |j ksPJt|}tt||d d tt||d d dS) zTest an RGB color uint16 imageT)recordalwaysrNrr?x&Y[@r'{Gz?rT)r img_as_uintrrwarningscatch_warnings simplefilterr rr0rrPrQr6rrrr)rkwhistr"r full_scaler#r#r$test_adapthist_colors  rcCstt}tj|jd|jdftd}t||f}t |}|j|jks*J|ddddddf}t |}|j|jksDJt t ||ddt t||dddS) zTest an RGBA color imagerr'rNrNrrTga+e?)r rrrrrrr6rrr rrrrr)rkalpharrr#r#r$test_adapthist_alphas  rcCstt}t|}d}t|dd|dd|f}t|g|jd|}tj |ddd}tj |ddd}|j|jksAJ|j|jksIJt t |||jdddks^Jd S) z Test for n-dimensional consistency with float images Note: Currently if img.ndim == 3, img.shape[2] > 4 must hold for the image not to be interpreted as a color image by @adapt_rgb rrr2g?)rrrTrN) r rrrrr rrr6r rmeanabs)rkaimg2dimg3d adapted2d adapted3dr#r#r$test_adapthist_grayscale_Nds.rcCstd}|d7}|tj}t|d}t|t|ks!Jtd}|d7}|tj}t|d}t|t|ksBJdS)z#Test constant image, float and uintrrTrNrN) rrUrsrr rrPrQr)rkrr#r#r$test_adapthist_constants      rcCsttt}|d}d||jdd|jddf<d}tdddD]4}tj||d d }t |d d |f|d d |fd ksDJt ||d d f||d d fd ksYJq%d S) zTest border processinggY@rrrTr'rrGrLrNr) r r rrrrr6r5r rr)rk border_indexrrr#r#r$test_adapthist_borderss *,rcCsbt}t|}tj|dd}tj|dd}t||tj|dd}tj|dd}t||dS)Nrrr')rmoonr rrr rr)img_uimg_f img_clahe0 img_clahe1r#r#r$test_adapthist_clip_limit s  rcCsv|jdkrt|t|}}t|}t|}d|jt||}t |j j \}}dt ||S)zPeak signal to noise ratio of two images Parameters ---------- img1 : array-like img2 : array-like Returns ------- peak_snr : float Peak signal to noise ratio rNrr~) ndimr rqr rrsizersquaresumr rtypelog)img1img2msermax_r#r#r$rs   rcCsJ|jdkrt|t|}}t||}|t|jjd}|S)zNormalized Absolute Mean Brightness Error between two images Parameters ---------- img1 : array-like img2 : array-like Returns ------- norm_brightness_error : float Normalized absolute mean brightness error rNr')rr rrrr rr)rrambenber#r#r$r1s rcCsNtjdtd}tjtddt|dWddS1s wYdS)Nrrz Incorrect value of `kernel_size`r)rNrNrN)rrrrrrr rrkr#r#r$$test_adapthist_incorrect_kernel_sizeEs"rcCs.tddg}t|d}|j|jksJdS)"Check that the shape is maintainedr'g?N)rrr adjust_gammar6)rkresultr#r#r$test_adjust_gamma_1x1_shapeOs rcCs2tjddtjdd}t|d}t||dS)z4Same image should be returned for gamma equal to onerr9rrr'N)rr*r;rr rr)rXrr#r#r$test_adjust_gamma_oneVs rcCsTtjdddj|dd}t|d}|jj}t|t |d|j|jks(JdS)z6White image should be returned for gamma equal to zerorrDrFrpr'N) rrandomuniformrsr rrrrr )rrXrr#r#r$test_adjust_gamma_zero]s  rcCLtjddtjdd}tjgdtjdd}t|d}t||dS)z\Verifying the output with expected results for gamma correction with gamma equal to halfrr9rr(rr $'*-0257:<>@BDFrIKMNPQSTVWYrI\]^`abrr&r3gijklmopqrstuvwyz{|}~rr?rrWr@r`rfrrjrnrrrurxr{r~rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrArArBrBrrrrrrrrrrrrrrrrrrrrrrrrDrLNrr*r;rrr rrrXrVrr#r#r$test_adjust_gamma_less_oneg& rcCr)z[Verifying the output with expected results for gamma correction with gamma equal to tworr9rr(rrrrrrrrrrrrr'r'r'r'r'r'r'r'rTrTrTrTrTrTrNrNrNrNrrrrr2r2r2r2rrrrrrr rrr/r/ rrrr rrrrrrrrrr~r~rrrrrrrrrr!"#rr%&rr()rr+,r ./rr!1r"r4r#6r$8rrr%;r&=r'?r(Ar)Cr*Er+rr,Jr-Lr.r/Or0r1Rr3Ur4r5Xr6[r7r8r9_r;r<crr3r=hr>r@rArBrCrDrErGrHrIrKxrLrNrOrQrr?rrSrUrVrXrYrZr[r]r^r`rarcrdrfrgrhrirkrlrnrprqrsrtrvrxryr{r}r~rrrrrrrrrrrrrrrrrrrrrrrrrrrrrArrrrrrrrrrrrrrDrTNrrr#r#r$test_adjust_gamma_greater_onevrrcCsTtdddtjd}ttt|dWddS1s#wYdS)NrrDrrr) rr*r;rrrrr rrXr#r#r$test_adjust_gamma_negatives "rcCs6dtjdtjd}ttj|ddddksJdS)NrDrrr'ry)gammagain)rrr;rvr rr r#r#r$test_adjust_gamma_u8_overflows"rcCs@tjddg|d}t|d}|j|jksJ|j|ksJdS)rr'rN)rrr adjust_logr6rrrkrr#r#r$test_adjust_log_1x1_shapes r c Csrtdddtjd}tjgdgdgdgdgd gd gd gd gtjd }t|d}t||dS)z|Verifying the output with expected results for logarithmic correction with multiplier constant multiplier equal to unityrrDrr)rr2rrrrrr)rr!r#r%rr*r,r.)rr4rrrrrBrE)rIrLrPrRrUrYr\r`)rdrrkrnrrrvrzr})rrrrrrrr)rrrrrrrrA)rrrrrrrrrr'Nrr*r;rrr rrrr#r#r$test_adjust_logs  r c Cttdddtjd}tjgdgdgdgdgd gd gd gd gtjd }t|dd}t||dS)zVerifying the output with expected results for inverse logarithmic correction with multiplier constant multiplier equal to unityrrDrr)rrTr2rrrrr~)rrrrrrrr )r!rr$r%rrr*H)rrr2r5rIr9r<r3)r?nrFrJrMrQrrV)r@r^rbrfrjrnrsrx)r|rrrrrrr)rrrrBrrrrrr'TNr rr#r#r$test_adjust_inv_log rcCsBtjddg|d}t|dd}|j|jksJ|j|ksJdS)rr'rr2N)rrr adjust_sigmoidr6rrr#r#r$test_adjust_sigmoid_1x1_shapesrc Cr )zlVerifying the output with expected results for sigmoid correction with cutoff equal to one and gain of 5rrDrr)r'r'r'rTrTrTrTrT)rNrNrNrrrr2r2)r2rrrrrrr/)r/rrrrrrr)rr~rrrrrr)rrrrrrr"r)rrr(r*r rr0r)r6r9rrrArErJrNrr'r2Nrr*r;rrr rrrr#r#r$test_adjust_sigmoid_cutoff_onerrc Csttdddtjd}tjgdgdgdgdgd gd gd gd gtjd }t|dd }t||d S)znVerifying the output with expected results for sigmoid correction with cutoff equal to zero and gain of 10rrDrr)rrYrbrjrtr}rr)rrrrrrrBr)rrrrrrrrrrrrrrrr)rrrrrrrr)rrrrrrrrrr/Nrrr#r#r$test_adjust_sigmoid_cutoff_zerorrc Cr )znVerifying the output with expected results for sigmoid correction with cutoff equal to half and gain of 10rrDrr)r'r'rTrTrNrNrr2)r2rrrr/rrr)rrrrrrrr")rr(r r0r6rrArJ)r?r@rcrlrur~rr)rrrrrrrBr)rrrrrrrrrrrLr/Nrrr#r#r$test_adjust_sigmoid_cutoff_halfrrc Csvtdddtjd}tjgdgdgdgdgd gd gd gd gtjd }t|ddd}t||dS)zvVerifying the output with expected results for inverse sigmoid correction with cutoff equal to half and gain of 10rrDrr)rrrrrrrr)rrrrrrrr)rrrBrrrrr)rrrr|rsrirarX)rQrHr?r:r5r/r+r')r$rrrrrrr)rrrrr/rrr)r2rrrNrNrTrTr'rrLr/TNrrr#r#r$#test_adjust_inv_sigmoid_cutoff_halfs rcCstddd}t|sJd|d<t|sJtj|ddr"J|dtj}t|s1Jtj|ddr:J|tjd}t|sIJtj|ddrRJdS) Nrg{Gz?rr'r)upper_percentilerDr9)rrEr is_low_contrastrsr;rrr#r#r$test_is_low_contrast/srcCs:tjdtd}t|sJd|dd<t|rJdS)Nrrr'r2)rrUboolr rrr#r#r$test_is_low_contrast_boolean?s r exposure_funccCsTtddddtj}tt ||WddS1s#wYdS)Nrrrr)rr*rrsrrrr)rrXr#r#r$test_negative_inputKs  "rcCstjdddddlm}|jtddgddggdd}t|\}}gd }gd }t||s4Jt||ss