diff --git a/gm/rebaseline_server/imagediffdb.py b/gm/rebaseline_server/imagediffdb.py index 936301e1cd..d168c888a7 100644 --- a/gm/rebaseline_server/imagediffdb.py +++ b/gm/rebaseline_server/imagediffdb.py @@ -10,10 +10,13 @@ Calulate differences between image pairs, and store them in a database. """ import contextlib +import csv import logging import os import re import shutil +import sys +import tempfile import urllib try: from PIL import Image, ImageChops @@ -21,6 +24,15 @@ except ImportError: raise ImportError('Requires PIL to be installed; see ' + 'http://www.pythonware.com/products/pil/') +# Set the PYTHONPATH to include the tools directory. +sys.path.append( + os.path.join( + os.path.dirname(os.path.realpath(__file__)), os.pardir, os.pardir, + 'tools')) +import find_run_binary + +SKPDIFF_BINARY_NAME = 'skpdiff' + DEFAULT_IMAGE_SUFFIX = '.png' DEFAULT_IMAGES_SUBDIR = 'images' @@ -99,6 +111,27 @@ class DiffRecord(object): whitediff_image = (graydiff_image.point(lambda p: p > 0 and VALUES_PER_BAND) .convert('1', dither=Image.NONE)) + # Calculate the perceptual difference percentage. + skpdiff_csv_dir = tempfile.mkdtemp() + try: + skpdiff_csv_output = os.path.join(skpdiff_csv_dir, 'skpdiff-output.csv') + skpdiff_binary = find_run_binary.find_path_to_program(SKPDIFF_BINARY_NAME) + expected_img = os.path.join(storage_root, expected_images_subdir, + str(expected_image_locator) + image_suffix) + actual_img = os.path.join(storage_root, actual_images_subdir, + str(actual_image_locator) + image_suffix) + find_run_binary.run_command( + [skpdiff_binary, '-p', expected_img, actual_img, + '--csv', skpdiff_csv_output, '-d', 'perceptual']) + with contextlib.closing(open(skpdiff_csv_output)) as csv_file: + for row in csv.DictReader(csv_file): + perceptual_similarity = float(row[' perceptual'].strip()) + # skpdiff returns the perceptual similarity, convert it to get the + # perceptual difference percentage. + self._perceptual_difference = 100 - (perceptual_similarity * 100) + finally: + shutil.rmtree(skpdiff_csv_dir) + # Final touches on diff_image: use whitediff_image as an alpha mask. # Unchanged pixels are transparent; differing pixels are opaque. diff_image.putalpha(whitediff_image) @@ -128,6 +161,10 @@ class DiffRecord(object): return ((float(self._num_pixels_differing) * 100) / (self._width * self._height)) + def get_perceptual_difference(self): + """Returns the perceptual difference percentage.""" + return self._perceptual_difference + def get_weighted_diff_measure(self): """Returns a weighted measure of image diffs, as a float between 0 and 100 (inclusive).""" diff --git a/gm/rebaseline_server/imagediffdb_test.py b/gm/rebaseline_server/imagediffdb_test.py index 558a816a02..b1d534a5e0 100755 --- a/gm/rebaseline_server/imagediffdb_test.py +++ b/gm/rebaseline_server/imagediffdb_test.py @@ -19,7 +19,8 @@ import unittest import imagediffdb -IMG_URL_BASE = 'http://chromium-skia-gm.commondatastorage.googleapis.com/gm/bitmap-64bitMD5/' +IMG_URL_BASE = ('http://chromium-skia-gm.commondatastorage.googleapis.com/gm/' + 'bitmap-64bitMD5/') class ImageDiffDbTest(unittest.TestCase): @@ -56,21 +57,22 @@ class ImageDiffDbTest(unittest.TestCase): # 3. actual image URL # 4. expected percent_pixels_differing (as a string, to 4 decimal places) # 5. expected weighted_diff_measure (as a string, to 4 decimal places) - # 6. expected max_diff_per_channel + # 6. expected perceptual difference (as a string, to 4 decimal places) + # 7. expected max_diff_per_channel selftests = [ [ 'arcofzorro/16206093933823793653', IMG_URL_BASE + 'arcofzorro/16206093933823793653.png', 'arcofzorro/13786535001616823825', IMG_URL_BASE + 'arcofzorro/13786535001616823825.png', - '0.0662', '0.0113', [255, 255, 247], + '0.0662', '0.0113', '0.0662', [255, 255, 247], ], [ 'gradients_degenerate_2pt/10552995703607727960', IMG_URL_BASE + 'gradients_degenerate_2pt/10552995703607727960.png', 'gradients_degenerate_2pt/11198253335583713230', IMG_URL_BASE + 'gradients_degenerate_2pt/11198253335583713230.png', - '100.0000', '66.6667', [255, 0, 255], + '100.0000', '66.6667', '100.0000', [255, 0, 255], ], ] @@ -88,7 +90,8 @@ class ImageDiffDbTest(unittest.TestCase): self.assertEqual('%.4f' % record.get_percent_pixels_differing(), selftest[4]) self.assertEqual('%.4f' % record.get_weighted_diff_measure(), selftest[5]) - self.assertEqual(record.get_max_diff_per_channel(), selftest[6]) + self.assertEqual('%.4f' % record.get_perceptual_difference(), selftest[6]) + self.assertEqual(record.get_max_diff_per_channel(), selftest[7]) def main(): diff --git a/gm/rebaseline_server/results.py b/gm/rebaseline_server/results.py index fff0a941e8..3b57bc1e20 100755 --- a/gm/rebaseline_server/results.py +++ b/gm/rebaseline_server/results.py @@ -432,6 +432,7 @@ class Results(object): results_for_this_test['numDifferingPixels'] = 0 results_for_this_test['percentDifferingPixels'] = 0 results_for_this_test['weightedDiffMeasure'] = 0 + results_for_this_test['perceptualDifference'] = 0 results_for_this_test['maxDiffPerChannel'] = 0 else: try: @@ -444,6 +445,8 @@ class Results(object): diff_record.get_percent_pixels_differing()) results_for_this_test['weightedDiffMeasure'] = ( diff_record.get_weighted_diff_measure()) + results_for_this_test['perceptualDifference'] = ( + diff_record.get_perceptual_difference()) results_for_this_test['maxDiffPerChannel'] = ( diff_record.get_max_diff_per_channel()) except KeyError: diff --git a/gm/rebaseline_server/static/view.html b/gm/rebaseline_server/static/view.html index 79335ca861..9ce3fae140 100644 --- a/gm/rebaseline_server/static/view.html +++ b/gm/rebaseline_server/static/view.html @@ -263,7 +263,7 @@ value="weightedDiffMeasure" ng-checked="(sortColumn == 'weightedDiffMeasure')" ng-click="sortResultsBy('weightedDiffMeasure')"> - difference per pixel + perceptual difference
+ title="Perceptual difference measure is {{result.perceptualDifference.toFixed(4)}}%. Maximum difference per channel: R={{result.maxDiffPerChannel[0]}}, G={{result.maxDiffPerChannel[1]}}, B={{result.maxDiffPerChannel[2]}}"> - {{result.weightedDiffMeasure.toFixed(4)}}% + {{result.perceptualDifference.toFixed(4)}}% {{result.maxDiffPerChannel}}
View Image