B ]t\D@sddZddlZddlZddlZddlZddlmZddlZddlZddl Z ddl Z ddl m Z ddl ZddlZddlmZddlmZmZddd gZd d Zed d-d dZddZd.ddZddZedjZddZGdddeZ Gddde!Z"Gddde"Z#Gddde"Z$d d!Z%iZ&e%d"d Z'd#d$Z(d%d&Z)d'd(Z*d/d*dZ+d+d,Z,dS)0zD Provides a collection of utilities for comparing (image) results. N)Path) TemporaryFile)ImageComparisonFailure)_pngcbook compare_floatcompare_imagescomparable_formatscCstj|\}}d|||fS)zU Make a new filename by inserting `purpose` before the file's extension. z%s-%s%s)ospathsplitext)fnameZpurposebaseZextr9lib/python3.7/site-packages/matplotlib/testing/compare.pymake_test_filenamesrz3.0cCs|dkr|dkrtdd}|dk r^t||}||kr^dddddg}|dd d |D7}|dk rt||}|r|t|}||krdddd d g}|dd d |D7}|pdS)z Fail if the floating point values are not close enough, with the given message. You can specify a relative tolerance, absolute tolerance, or both. Nz}You haven't specified a 'relTol' relative tolerance or a 'absTol' absolute tolerance function argument. You must specify one.zExpected: {expected}zActual: {actual}zAbs diff: {absDiff}zAbs tol: {absTol}z cSsg|]}|jftqSr)formatlocals).0linerrr :sz!compare_float..zRel diff: {relDiff}zRel tol: {relTol}cSsg|]}|jftqSr)rr)rrrrrrJs) ValueErrorabsjoin)expectedactualZrelTolZabsTolmsgZabsDifftemplateZrelDiffrrrr#s0    cCsht}|dkrtdtj|d}yt|jdddWntk rPdSXt |tj sddS|S)Nz1Could not find a suitable configuration directoryZ test_cacheT)parentsexist_ok) matplotlibZ get_cachedir RuntimeErrorr r rrmkdirIOErroraccessW_OK)Zcachedir cache_dirrrr get_cache_dirOsr(c Cst}t|d&}x||}|s&P||qWWdQRX|drlddlm}||ddn(|drddlm }||d| S) Nrbz.pdfr)checkdep_ghostscriptzutf-8z.svg)checkdep_inkscape) hashlibmd5openreadupdateendswithr!r+encoder-Z hexdigest)r Z block_sizer/fddatar+r-rrr get_file_hash]s      r7csfdd}|S)Ncs~||}tj|dtjtjd}|\}}|}tj|rD|rzdd|}|rb|d|7}|rr|d|7}t |dS)NT)Zuniversal_newlinesstdoutstderrzConversion command failed: %s  zStandard output: %s zStandard error: %s ) subprocessPopenPIPEZ communicatewaitr r existsrr$)oldnewZcmdlinepiper8r9Zerrcoder)cmdrrconvertqs    z1make_external_conversion_command..convertr)rCrDr)rCr make_external_conversion_commandps rEs[^a-zA-Z0-9_@%+=:,./-]cCs$t|dkr|Sd|dddS)N's'"'"')_find_unsafe_bytesreplace)brrr_shlex_quote_bytessrJc@s eZdZdS)_ConverterErrorN)__name__ __module__ __qualname__rrrrrKsrKc@s$eZdZddZddZddZdS) _ConvertercCsd|_t|jdS)N)_procatexitregister__del__)selfrrr__init__sz_Converter.__init__cCsR|jrN|j|jx,td|jj|jj|jjgD] }|q8Wd|_dS)N)rPkillr>filterstdinr8r9close)rTstreamrrrrSs    z_Converter.__del__cCsPt}xD|jjd}|st||||rt|dt| SqWdS)z!Read until the prompt is reached.r,N) bytearrayrPr8r1rKextendr3byteslen)rT terminatorZbufcrrr _read_untils  z_Converter._read_untilN)rLrMrNrUrSrarrrrrOs rOc@seZdZddZdS) _GSConvertercCs|jsZt|_tjtjjddgtjtjd|_y| dWnt k rXt dYnXdd}|jj d||d ||d |jj | d }| d }|stj|s|rt|d dnd}|jj d|t|d |d tddS)Nz -dNOPAUSEz-sDEVICE=png16m)rXr8s GSzFailed to start GhostscriptcSs"t|ddddddS)N\s\\(s\()s\))r fsencoderH)namerrrencode_and_escapes z0_GSConverter.__call__..encode_and_escapes<< /OutputFile (s) >> setpagedevice (s ) run flush sGS>r,rspop rH)rPrZ_stdoutr;r<r!r+ executabler=rarKOSErrorrXwriteflushr r r?intrdecodesysgetfilesystemencoding)rTorigdestrherrstack stack_sizerrr__call__s,   z_GSConverter.__call__N)rLrMrNrwrrrrrbsrbc@seZdZddZdS) _SVGConvertercCs@|jr|jdk rtj}|ddtj|d<t}tj dddgtj tj ||d|_||j_ y| dWnt k rtdYnXttttj||g\}}d |ksd |krtd d ||S|jj|d |d |jjy| dWn<t k r:|jd t|jtdYnXdS)NZDISPLAYZINKSCAPE_PROFILE_DIRinkscapez --without-guiz--shell)rXr8r9envs >z,Failed to start Inkscape in interactive mode cSsdd|d|gS)Nryz-zz --export-pngr)r@rArrrsz(_SVGConverter.__call__..s --export-png=rrH)rPZpollr environcopypopdevnullrr;r<r=r9rarKrkmaprJrfrErXrlrmZ_stderrseekrr1rorprq)rTrrrsrzr9Zorig_bZdest_brrrrws:        z_SVGConverter.__call__N)rLrMrNrwrrrrrxsrxcCs@t\}}|dk r&ttd<td<tdk rrcCs<|j|jkr td|j|jt||tdS)zHCalculate the per-pixel errors, then compute the root mean square error.z9Image sizes do not match expected size: {} actual size {}r)rrrnpZsqrtastypefloatZmean) expectedImage actualImagerrr calculate_rms_s  rFc s~tj|std|t|jdkr4td||dd}tj|sZtd||dkrvt|d}t|d }t |}t |}|d d d d d d f}|d d d d d d f}t ||||\}}t |d }|dkrt ||rd S|t j}|t j}t||}||kr"d St|||t|t|t|t||d |szddddddg} 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 If called from image_comparison decorator, this should be True. (default=False) 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.rz!Baseline image %r does not exist.rFTNz failed-diff)rmsrrZdifftolz!Error: Image files did not match.zRMS Value: {rms}zExpected: {expected}zActual: {actual}zDifference: {diff}zTolerance: {tol}z csg|]}|jfqSr)r)rr)resultsrrrsz"compare_images..)r r r? Exceptionrst_sizesplitr$rDrZ read_png_intrrrZ array_equalrZint16rsave_diff_imagedictstrr) rrrZ in_decoratorrrrZ diff_imagerrr)rrrisJ                 c Cst|dddf}t|dddf}t||||\}}t|t}t|t}|j|jkr~td |j|jt ||}|d9}t |ddtj }|j\}}} | dkrtj ||dftj d } || ddddddf<| }d|dddddf<t||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. .Nrz9Image sizes do not match expected size: {} actual size {}i r)Zdtype)rZread_pngrrZarrayrrrrrrZclipZuint8emptyZ write_png) rroutputrrZ absDiffImageZ save_image_npZheightwidthZdepthZ with_alpharrrrs(   r)NN)r))F)-__doc__rQ functoolsr.r Zpathlibrrerr;rpZtempfilerZnumpyrr!Zmatplotlib.testing.exceptionsrrr__all__rZ deprecatedrr(r7rEcompilesearchrGrJrrKobjectrOrbrxrrr rDrrrrrrrrsH      +  '6  0  W