a ضaF@sdZddlZddlZddlZddlZddlZddlmZddlZddl Z ddl Z ddl Z ddl m Z mZddlZddlZddlmZddlZddlmZddlmZeeZgdZd d Zd d Zd dZd0ddZ e!dj"Z#ddZ$Gddde%Z&GdddZ'Gddde'Z(Gddde'Z)ddZ*iZ+e*dd Z,d!d"Z-d#d$Z.e/d%d&Z0d'd(Z1d)d*Z2d1d,d-Z3d.d/Z4dS)2z( Utilities for comparing image results. N)Path)TemporaryDirectory TemporaryFile)Image)cbook)ImageComparisonFailure) calculate_rmscomparable_formatscompare_imagescCstj|\}}d|||fS)zQ Make a new filename by inserting *purpose* before the file's extension. z%s-%s%s)ospathsplitext)fnamepurposebaseextrj/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/matplotlib/testing/compare.pymake_test_filenamesrcCs ttd}|jddd|S)NZ test_cacheT)parentsexist_ok)rmplZ get_cachedirmkdir) cache_dirrrr_get_cache_path&srcCs ttSN)strrrrrr get_cache_dir,srcCst}t|d,}||}|s$q0||qWdn1sD0Yt|jdkrz|tt dj  dn*t|jdkr|tt dj  d| S)Nrbz.pdfgszutf-8z.svginkscape) hashlibmd5openreadupdatersuffixrr_get_executable_infoversionencode hexdigest)r block_sizer#fddatarrr get_file_hash0s  *r/s[^a-zA-Z0-9_@%+=:,./-]cCs$t|dur|Sd|dddS)N's'"'"')_find_unsafe_bytesreplace)brrr_shlex_quote_bytesGsr4c@s eZdZdS)_ConverterErrorN)__name__ __module__ __qualname__rrrrr5Lsr5c@s$eZdZddZddZddZdS) _ConvertercCsd|_t|jdSr)_procatexitregister__del__selfrrr__init__Qsz_Converter.__init__cCsN|jrJ|j|jtd|jj|jj|jjgD] }|q6d|_dSr)r:killwaitfilterstdinstdoutstderrclose)r?streamrrrr=Ys     z_Converter.__del__cCs>t}|jjd}|st||||rt|SqdS)z!Read until the prompt is reached.N) bytearrayr:rEr%r5extendendswithbytes)r? terminatorbufcrrr _read_untilcs  z_Converter._read_untilN)r6r7r8r@r=rQrrrrr9Ps r9c@seZdZddZdS) _GSConverterc Cs|jsjtjtdjdddgtjtjd|_z|dWn.tyh}zt d|WYd}~n d}~00dd }|jj d ||d ||d |jj |d }| dr|dnd}|stj|s|rt|ddnd}|jj d|t||tddS)Nr z -dNOSAFERz -dNOPAUSEz-sDEVICE=png16m)rDrEs GSzFailed to start GhostscriptcSs"t|ddddddS)N\s\\(s\()s\))r fsencoder2)namerrrencode_and_escape|s z0_GSConverter.__call__..encode_and_escapes<< /OutputFile (s) >> setpagedevice (s ) run flush )GSrY>rspop r2)r: subprocessPopenrr( executablePIPErQr5OSErrorrDwriteflushrLr r existsintrdecodesysgetfilesystemencoding)r?origdesterrrXstack stack_sizerrr__call__ps@    z_GSConverter.__call__N)r6r7r8rnrrrrrRosrRcs$eZdZddZfddZZS) _SVGConverterc Cstdjjdk}|rdnd}t|ds@t|_t|j|j |j rT|j durit j dt jd}t}tj|rgdndd gtjtj|||jjd |_ ||j _z||Wn.ty}ztd |WYd}~n d}~00t|jjt d }t|jjt d } z|t|Wn tyDt||Yn0|j j|rXdnd|j j z||WnRty}z8|j j!dt"|j j#$t%&d|WYd}~n d}~00t '|t(| |dS)Nr!rIs >s> _tmpdir)ZDISPLAYZINKSCAPE_PROFILE_DIR)r!z --without-gui--shellrr)rDrErFenvcwdz,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 rr2))rr(r)majorhasattrrrpweakreffinalizer=r:pollr environdevnullrr]r^r`rWrFrQr5rarfsdecode symlink_toresolveshutilcopyfilerDrbrcseekrr%rfrgrhremovemove) r?rirjZ old_inkscaperNrsrFrkZ inkscape_origZ inkscape_destrrrrnsh         z_SVGConverter.__call__cs"tt|dr|jdS)Nrp)superr=rvrpcleanupr> __class__rrr=s  z_SVGConverter.__del__)r6r7r8rnr= __classcell__rrrrrosArocCshztdWntjy"Yn0ttd<td<ztdWntjyXYn 0ttd<dS)Nr pdfepsr!svg)rr(ZExecutableNotFoundErrorrR converterrorrrr_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 cCs4t|}|st|d|jddtvrLddl}|d|jd|j|jd|jddd}|r| j | j kr,|rt nd}|durt t |}|||j}|rtd |t||t|Std |t|jdd|||dur,td |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.z"For %s: caching conversion result.)rrdIOErrorr'rpytestskipparentstemstatst_mtimer'_register_conversion_cache_cleaner_oncer/_logdebugrrr)filenamecacher rnewpathrZ hash_valueZ cached_pathrrrconverts. "      rcstddttjjdD}d|}ttzddtdDtdd D}t fd d d d }||kr| }||j 8}| qxWdn1s0YdS) Ncss|]}|jVqdSr)rst_size.0r rrr .sz*_clean_conversion_cache..z**/baseline_images/**/*cSsi|]}||qSr)rrrrr 6sz+_clean_conversion_cache..*css|] }|jVqdSr)r)rrrrrr8r[cs |jSr)st_atime)r Z cache_statrr:r[z)_clean_conversion_cache..T)keyreverse)sumrr__file__rglobrZ _lock_pathrvaluessortedpoprunlink)Zbaseline_images_sizeZmax_cache_sizeZ cache_sizeZpaths_by_atimer rrr_clean_conversion_cache+s"  rcCsttdSr)r;r<rrrrrrBsrc Cs|dddkr|dddkr|j\}}}|j\}}} |t|d|dt|d|dt|d|dt|d|df}||fS)Nirrr)shapere) Z actual_path actual_imageZ expected_pathexpected_imageZawZahadewehZedrrr crop_to_sameGs   &rcCs<|j|jkr td|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)rrrrrrRs rFc srt|}tj|s"td|t|jdkr>td|t|}tj|s`td||dd}|dkrt |dd }t |dd }t t | d }t t | d }t||||\}}t|d }|dkrt ||rd S|t j}|t j}t||}||krd St|||t|t|t|t||d |sngd} 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..r\rT)rRGBz failed-diffN)rmsexpectedactualdifftol)z!Error: Image files did not match.zRMS Value: {rms}zExpected: {expected}zActual: {actual}zDifference: {diff}zTolerance: {tol}z csg|]}|jfiqSrr)rlineresultsrr r[z"compare_images..)r fspathr rd Exceptionrrrsplitrrasarrayrr$rrZ array_equalrZint16rsave_diff_imagedictrjoin) rrrZ in_decorator extensionrrZ diff_imagertemplaterrrr bsD1                 r c Cs tt|d}tt|d}t||||\}}t|t}t|t}|j |j kr~t d |j |j t ||}|d9}t |ddtj}|j \}}} | dkrtj||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)ZdtypeNrr)rrrr$rrarrayrrrrrabsZclipZuint8emptyZ fromarraysave) rroutputrrZabs_diff_imageZ save_image_npheightwidthdepthZ with_alpharrrrs.   r)r)F)5__doc__r; functoolsr"loggingr pathlibrrerr]rgtempfilerrrwZnumpyrZPILrZ matplotlibrrZmatplotlib.testing.exceptionsr getLoggerr6r__all__rrrr/compilesearchr1r4rr5r9rRrorrr rr lru_cacherrrr rrrrrsR       %H*   h