o iUB@sZdZddlZddlZddlZddlmZddlZddlZddlZddl Z ddl m Z m Z ddl ZddlmZddlZddlmZddlmZdd gZd d Zd d Zd,ddZedddZedjZddZGddde Z!GdddZ"Gddde"Z#Gddde"Z$dd Z%iZ&e%d!d Z'd"d#Z(d$d%Z)d&d'Z*d-d)dZ+d*d+Z,dS).z( Utilities for comparing image results. N)Path)TemporaryDirectory TemporaryFile)Image)cbook)ImageComparisonFailurecompare_imagescomparable_formatscCstj|\}}d|||fS)zQ Make a new filename by inserting *purpose* before the file's extension. z%s-%s%s)ospathsplitext)fnamepurposebaseextrU/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/matplotlib/testing/compare.pymake_test_filenamesrcCs$ttd}|jdddt|S)N test_cacheT)parentsexist_ok)rmpl get_cachedirmkdirstr) cache_dirrrr get_cache_dir!srcCst}t|d} ||}|sn||q Wdn1s#wYt|jdkrA|tt dj  d| St|jdkrV|tt dj  d| S)NrbTz.pdfgszutf-8z.svginkscape) hashlibmd5openreadupdatersuffixrr_get_executable_infoversionencode hexdigest)r block_sizer"fddatarrr get_file_hash's(   r.z3.3csfdd}|S)Ncs~||}tj|dtjtjd}|\}}|}tj|r"|r=dd|}|r1|d|7}|r9|d|7}t |dS)NT)universal_newlinesstdoutstderrzConversion command failed: %s  zStandard output: %s zStandard error: %s ) subprocessPopenPIPE communicatewaitr r existsjoinIOError)oldnewcmdlinepiper0r1errcodemsgcmdrrconvert<s    z1make_external_conversion_command..convertr)rBrCrrAr make_external_conversion_command:s rDs[^a-zA-Z0-9_@%+=:,./-]cCs$t|dur|Sd|dddS)N's'"'"')_find_unsafe_bytesreplace)brrr_shlex_quote_bytesQsrIc@s eZdZdS)_ConverterErrorN)__name__ __module__ __qualname__rrrrrJVsrJc@s$eZdZddZddZddZdS) _ConvertercCsd|_t|jdSN)_procatexitregister__del__selfrrr__init__[sz_Converter.__init__cCsR|jr'|j|jtd|jj|jj|jjgD]}|qd|_dSdSrO)rPkillr7filterstdinr0r1close)rUstreamrrrrScs     z_Converter.__del__cCsJt} |jjd}|st||||r$t|dt| Sq)z!Read until the prompt is reached.TN) bytearrayrPr0r$rJextendendswithbyteslen)rU terminatorbufcrrr _read_untilms  z_Converter._read_untilN)rKrLrMrVrSrerrrrrNZs rNc@seZdZddZdS) _GSConverterc Cs |js/tjtdjdddgtjtjd|_z|dWnty.}zt d|d}~wwdd }|jj d ||d ||d |jj |d }|d}|s^t j|s|rht|ddnd}|jj d|t|d |dtddS)Nrz -dNOSAFERz -dNOPAUSEz-sDEVICE=png16m)rYr0s GSzFailed to start GhostscriptcSs"t|ddddddS)N\s\\(s\()s\))r fsencoderG)namerrrencode_and_escapes z0_GSConverter.__call__..encode_and_escapes<< /OutputFile (s) >> setpagedevice (s ) run flush sGS>r\rspop rG)rPr3r4rr' executabler5rerJOSErrorrYwriteflushr r r8intrdecodesysgetfilesystemencoding)rUorigdesterrrlstack stack_sizerrr__call__zsH      z_GSConverter.__call__N)rKrLrMr{rrrrrfys rfcs$eZdZddZfddZZS) _SVGConverterc Cstdjdk}|r dnd}t|dst|_|jr!|jdurditj dtj d}t }t j |r7gdndd gt jt j|||jjd |_||j_z||Wntyc}ztd |d}~wwt|jjtd }t|jjtd } z |t|Wntyt||Ynw|jj|rdnd|jjz||Wn"ty}z|jjdt|jj t!"d|d}~wwt#|t$| |dS)Nr 1s >s> _tmpdir)DISPLAYINKSCAPE_PROFILE_DIR)r z --without-gui--shellr)rYr0r1envcwdz,Failed to start Inkscape in interactive modesf.svgsf.pngsf.svg --export-png=f.png s;file-open:f.svg;export-filename:f.png;export-do;file-close rrG)%rr'r(hasattrrr~rPpollr environdevnullrr3r4r5rkr1rerJrorfsdecode symlink_toresolveshutilcopyfilerYrprqseekrr$rsrtruremovemove) rUrvrw old_inkscaperbrr1rx inkscape_orig inkscape_destrrrr{sn        z_SVGConverter.__call__cs&tt|dr|jdSdS)Nr~)superrSrr~cleanuprT __class__rrrSs  z_SVGConverter.__del__)rKrLrMr{rS __classcell__rrrrr|s>r|cCsjztdWn tjyYn wttd<td<ztdWn tjy-YdSwttd<dS)Nrpdfepsr svg)rr'ExecutableNotFoundErrorrf converterr|rrrr_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 cCs t|}|st|d|jddtvr&ddl}|d|jd|j|jd|jddd}|rE| j | j kr|rLtt nd}|durkt |}|||j}|rkt ||t|St|jdd|||durt ||t|S) at 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. There is no limit on the size of the cache, so it may need to be manually cleared periodically. z does not existr\NrzDon't know how to convert z files to png_z.png)rr8r:r&rpytestskipparentstemstatst_mtimerr.rrr)filenamecacher rnewpathr hash_value cached_pathrrrrC s& "  rCc Cs|dddkrF|dddkrF|j\}}}|j\}}} |t|d|dt|d|dt|d|dt|d|df}||fS)Nirr)shaperr) actual_path actual_image expected_pathexpected_imageawahadewehedrrr crop_to_same.s   &rcCs<|j|jkrtd|j|jt||tdS)zR Calculate the per-pixel errors, then compute the root mean square error. 9Image sizes do not match expected size: {} actual size {}r)rrformatnpsqrtastypefloatmean)rrrrr calculate_rms9s  rFc snt|}tj|std|t|jdkrtd|t|}tj|s0td||dd}|dkrGt |dd }t |d d }t t | d }t t | d }t||||\}}t|d }|dkrwt ||rwd 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) zOutput image %s does not exist.rzOutput image file %s is empty.z!Baseline image %r does not exist..rF)rTRGBz failed-diffN)rmsexpectedactualdifftol)z!Error: Image files did not match.zRMS Value: {rms}zExpected: {expected}zActual: {actual}zDifference: {diff}zTolerance: {tol}z csg|] }|jdiqS)rr).0lineresultsrr sz"compare_images..)r fspathr r8 Exceptionrst_sizer:splitrCrasarrayrr#rr array_equalrint16rsave_diff_imagedictrr9) rrr in_decorator extensionrr diff_imagertemplaterrrrIsD 1               c Cs tt|d}tt|d}t||||\}}t|t}t|t}|j |j kr?t d |j |j t ||}|d9}t |ddtj}|j \}}} | dkrytj||dftjd} || d d d d ddf<| }d|d d d d df<t|j|d d d S) 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. rri r)dtypeNrr)rrrr#rCrarrayrrrrrabsclipuint8empty fromarraysave) rroutputrrabs_diff_image save_image_npheightwidthdepth with_alpharrrrs,    r)r)F)-__doc__rQr!r pathlibrrerr3rttempfilerrnumpyrPILr matplotlibrrmatplotlib.testing.exceptionsr__all__rrr. deprecatedrDcompilesearchrFrIrrJrNrfr|rrr rCrrrrrrrrsH       &E%  h