o hDM@sdZddlZddlZddlZddlZddlZddlmZddlZddl Z ddl Z ddl m Z m Z ddlZddlZddlZddlmZddlZddlmZddlmZeeZgdZd d Zd d Zd dZd1ddZ Gddde!Z"GdddZ#Gddde#Z$Gddde#Z%Gddde%Z&ddZ'iZ(e'e&Z)ddZ*d d!Z+d"d#Z,ej-d$d%Z.d&d'Z/d(d)Z0d*d+Z1d2d-d.Z2d/d0Z3dS)3z( Utilities for comparing image results. N)Path)TemporaryDirectory TemporaryFile)Image)cbook)ImageComparisonFailure) calculate_rmscomparable_formatscompare_imagescCs"tj|\}}|d||S)zQ Make a new filename by inserting *purpose* before the file's extension. -)ospathsplitext)fnamepurposebaseextrn/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/matplotlib/testing/compare.pymake_test_filenamesrcCs ttd}|jddd|S)N test_cacheT)parentsexist_ok)rmpl get_cachedirmkdir) cache_dirrrr_get_cache_path&srcCs ttSN)strrrrrr get_cache_dir,s r cCstjdd}t|d} ||}|sn||q Wdn1s%wYt|jdkrC|tt dj  d| St|jdkrX|tt d j  d| S) NF)usedforsecurityrbTz.pdfgszutf-8.svginkscape) hashlibsha256openreadupdatersuffixrr_get_executable_infoversionencode hexdigest)r block_sizer(fddatarrr get_file_hash0s    r4c@s eZdZdS)_ConverterErrorN)__name__ __module__ __qualname__rrrrr5Asr5c@s$eZdZddZddZddZdS) _ConvertercCsd|_t|jdSr)_procatexitregister__del__selfrrr__init__Fsz_Converter.__init__cCsR|jr'|j|jtd|jj|jj|jjgD]}|qd|_dSdSr)r:killwaitfilterstdinstdoutstderrclose)r?streamrrrr=Ns     z_Converter.__del__cCsJt} |jjd}|sttt|||| |r$t|Sq)z!Read until the prompt is reached.T) bytearrayr:rEr*r5r fsdecodebytesextendendswith)r? terminatorbufcrrr _read_untilXs  z_Converter._read_untilN)r6r7r8r@r=rRrrrrr9Es r9c@seZdZddZdS) _GSConverterc Cs|js6tjtdjddddgtjtjd|_z|dWnty5}z t d|j d dd}~wwd d }|jj d ||d ||d|jj |d}|drb|dnd}|sltj|s|rvt|ddnd }|jj d|t||tddS)Nr$z -dNOSAFERz -dNOPAUSEz -dEPSCropz-sDEVICE=png16m)rDrEs GSzFailed to start Ghostscript: rcSs"t|ddddddS)N\s\\(s\()s\))r fsencodereplace)namerrrencode_and_escapeqs z0_GSConverter.__call__..encode_and_escapes<< /OutputFile (s) >> setpagedevice (s ) run flush )GSr[>spop rX)r: subprocessPopenrr- executablePIPErRr5OSErrorargsrDwriteflushrNr r existsintrdecodesysgetfilesystemencoding)r?origdesterZerrstack stack_sizerrr__call__esF   z_GSConverter.__call__N)r6r7r8rrrrrrrSds rScs$eZdZddZfddZZS) _SVGConverterc Cstdjjdk}|r dnd}t|ds t|_t|j|j |j r*|j dur|j durK|j durKt d|j j |j j|j jgD]}|qDitjd|jjd}t}tj|rbgdndd gtjtj|||jjd |_ ||j _z||Wnty}z td |jd |d}~wwt|jjtd } t|jjtd} z | t|Wntyt !|| Ynw|j j "|rdnd|j j #z||Wn"ty}z|j j$d t%|j j&'t()d|d}~wwt*| t +| |dS)Nr&rIs >s> _tmpdir)DISPLAYINKSCAPE_PROFILE_DIR)r&z --without-gui--shellrx)rDrErFenvcwdz/Failed to start Inkscape in interactive mode: rsf.svgsf.pngsf.svg --export-png=f.png s;file-open:f.svg;export-filename:f.png;export-do;file-close rX),rr-r.majorhasattrrrtweakreffinalizer=r:pollrCrDrErFrGr environrYrr_r`rbrRr5rcrdrrK symlink_toresolveshutilcopyfilererfseekrr*rirjrkremovemove) r?rlrm old_inkscaperOrHryrFro inkscape_orig inkscape_destrrrrrs          z_SVGConverter.__call__cs&tt|dr|jdSdS)Nrt)superr=r|rtcleanupr> __class__rrr=s  z_SVGConverter.__del__)r6r7r8rrr= __classcell__rrrrrssGrscs eZdZdZfddZZS) _SVGWithMatplotlibFontsConverterz A SVG converter which explicitly adds the fonts shipped by Matplotlib to Inkspace's font search path, to better support `svg.fonttype = "none"` (which is in particular used by certain mathtext tests). cs<t|dst|_ttdt|jjdt ||S)Nrtz fonts/ttffonts) r|rrtrcopytreer_get_data_pathrrYrrr)r?rlrmrrrrrs   z)_SVGWithMatplotlibFontsConverter.__call__)r6r7r8__doc__rrrrrrrrsrcCsjztdWn tjyYn wttd<td<ztdWn tjy-YdSwttd<dS)Nr$pdfepsr&svg)rr-ExecutableNotFoundErrorrS converterrsrrrr_update_convertersrcCs dgtS)z Return the list of file formats that `.compare_images` can compare on this system. Returns ------- list of str E.g. ``['png', 'pdf', 'svg', 'eps']``. png)rrrrrr s r c CsVt|}|st|d|jddtvr&ddl}|d|jd|j|jd|jddd}|rE| j | j kr|rJt nd}|durrt t |}|||j}|rrtd |t||t|Std |t|jdd}|jd kr|} td | rt}||||durtd |t||t|S)a Convert the named file to png; return the name of the created file. If *cache* is True, the result of the conversion is cached in `matplotlib.get_cachedir() + '/test_cache/'`. The caching is based on a hash of the exact contents of the input file. Old cache entries are automatically deleted as needed to keep the size of the cache capped to twice the size of all baseline images. z does not existrINrzDon't know how to convert z files to png_z.pngz"For %s: reusing cached conversion.zFor %s: converting to png.r%z9style="[^"]*font(|-size|-weight|-family|-variant|-style):z"For %s: caching conversion result.)rrgrcr,rpytestskipparentstemstatst_mtimer'_register_conversion_cache_cleaner_oncer4_logdebugrrr read_textresearch$_svg_with_matplotlib_fonts_converter) filenamecacher rnewpathr hash_value cached_pathconvertcontentsrrrr s> "       rcstddttjjdD}d|}ttIddtdDtdd D}t fd d d d }||kr[| }||j 8}| ||ks@WddSWddS1sfwYdS) Ncss|]}|jVqdSr)rst_size.0r rrr Ks  z*_clean_conversion_cache..z**/baseline_images/**/*cSsi|]}||qSr)rrrrr Ss z+_clean_conversion_cache..*css|]}|jVqdSr)r)rrrrrrUscs |jSr)st_atime)r  cache_statrrWs z)_clean_conversion_cache..T)keyreverse)sumrr__file__rglobr _lock_pathrvaluessortedpoprunlink)baseline_images_sizemax_cache_size cache_sizepaths_by_atimer rrr_clean_conversion_cacheHs*   "rcCsttdSr)r;r<rrrrrr_src Cs|dddkrF|dddkrF|j\}}}|j\}}} |t|d|dt|d|dt|d|dt|d|df}||fS)Nirrr)shaperh) actual_path actual_image expected_pathexpected_imageawahadewehedrrr crop_to_sameds   &rcCs@|j|jkrtd|jd|jt||tdS)zR Calculate the per-pixel errors, then compute the root mean square error. (Image sizes do not match expected size: actual size r)rrnpsqrtastypefloatmean)rrrrrros  rcCs<t|}|jdks|dddkr|d}t|S)NRGBArRGB)rr)mode getextremarrasarray)r imgrrr _load_images   rFc s^t|}tj|std|dt|jdkr#td|dt|}tj|s6td|d|dd}|d krMt |d d }t |d d }t |}t |}t ||||\}}t |d }|dkrot ||rod S|t j}|t j}t||}||krd St|||t|t|t|t||d|sgd} dfdd| DS)a+ Compare two "image" files checking differences within a tolerance. The two given filenames may point to files which are convertible to PNG via the `.converter` dictionary. The underlying RMS is calculated with the `.calculate_rms` function. Parameters ---------- expected : str The filename of the expected image. actual : str The filename of the actual image. tol : float The tolerance (a color value difference, where 255 is the maximal difference). The test fails if the average pixel difference is greater than this value. in_decorator : bool Determines the output format. If called from image_comparison decorator, this should be True. (default=False) Returns ------- None or dict or str Return *None* if the images are equal within the given tolerance. If the images differ, the return value depends on *in_decorator*. If *in_decorator* is true, a dict with the following entries is returned: - *rms*: The RMS of the image difference. - *expected*: The filename of the expected image. - *actual*: The filename of the actual image. - *diff_image*: The filename of the difference image. - *tol*: The comparison tolerance. Otherwise, a human-readable multi-line string representation of this information is returned. Examples -------- :: img1 = "./baseline/plot.png" img2 = "./output/plot.png" compare_images(img1, img2, 0.001) z Output image z does not exist.rzOutput image file z is empty.zBaseline image .r^rT)rz failed-diffN)rmsexpectedactualdifftol)z!Error: Image files did not match.zRMS Value: {rms}zExpected: {expected}zActual: {actual}zDifference: {diff}zTolerance: {tol}z csg|] }|jdiqS)rformat)rlineresultsrr sz"compare_images..)r fspathr rg Exceptionrrrcsplitrrrrr array_equalrint16rsave_diff_imagedictrjoin) rrr in_decorator extensionrr diff_imagertemplaterrrr sD 1            r cCst|}t|}t||||\}}t|t}t|t}|j|jkr/td|jd|jt||}|d9}t|dd tj }|jddkrWd|ddddd f<t |j |d d dS) z Parameters ---------- expected : str File path of expected image. actual : str File path of actual image. output : str File path to save difference image to. rr rrrNrrr)rrrarrayrrrabsclipruint8r fromarraysave)rroutputrrabs_diffrrrrs&     r)r!)F)4rr; functoolsr'loggingr pathlibrrr_rjtempfilerrr}rnumpyrPILr matplotlibrrmatplotlib.testing.exceptionsr getLoggerr6r__all__rrr r4rr5r9rSrsrrrrr rrrrrrrr rrrrrsV      %N>   h