X-Git-Url: https://git.notmuchmail.org/git?a=blobdiff_plain;f=scripts%2Fsnapdiff.py;h=7ae3a433fa125724f83c82548bbff4da9e7886c8;hb=d2d7112462378739bf96775f547837431375ff4d;hp=fa634fcf49c9a444d891f2dee0436cb542f4a9ba;hpb=0c881f7e9fd139ed13f9e81d761b9393bb08ed15;p=apitrace diff --git a/scripts/snapdiff.py b/scripts/snapdiff.py index fa634fc..7ae3a43 100755 --- a/scripts/snapdiff.py +++ b/scripts/snapdiff.py @@ -1,6 +1,7 @@ #!/usr/bin/env python ########################################################################## # +# Copyright 2011 Jose Fonseca # Copyright 2008-2009 VMware, Inc. # All Rights Reserved. # @@ -25,23 +26,89 @@ ##########################################################################/ +'''Snapshot (image) comparison script. +''' + + import sys import os.path -import subprocess import optparse +import math +import operator -import Image +from PIL import Image +from PIL import ImageChops +from PIL import ImageEnhance +from PIL import ImageFilter thumb_size = 320, 320 +gaussian_kernel = ImageFilter.Kernel((3, 3), [1, 2, 1, 2, 4, 2, 1, 2, 1], 16) + +class Comparer: + '''Image comparer.''' + + def __init__(self, ref_image, src_image, alpha = False): + if isinstance(ref_image, basestring): + self.ref_im = Image.open(ref_image) + else: + self.ref_im = ref_image + + if isinstance(src_image, basestring): + self.src_im = Image.open(src_image) + else: + self.src_im = src_image + + # Ignore + if not alpha: + self.ref_im = self.ref_im.convert('RGB') + self.src_im = self.src_im.convert('RGB') + + self.diff = ImageChops.difference(self.src_im, self.ref_im) + + def size_mismatch(self): + return self.ref_im.size != self.src_im.size + + def write_diff(self, diff_image, fuzz = 0.05): + # make a difference image similar to ImageMagick's compare utility + mask = ImageEnhance.Brightness(self.diff).enhance(1.0/fuzz) + mask = mask.convert('L') + + lowlight = Image.new('RGB', self.src_im.size, (0xff, 0xff, 0xff)) + highlight = Image.new('RGB', self.src_im.size, (0xf1, 0x00, 0x1e)) + diff_im = Image.composite(highlight, lowlight, mask) + + diff_im = Image.blend(self.src_im, diff_im, 0xcc/255.0) + diff_im.save(diff_image) -def compare(im, ref): - import ImageMath - # See http://www.pythonware.com/library/pil/handbook/imagemath.htm - mask = ImageMath.eval("min(abs(a - b), 1)", a=im, b=ref) - gray = ref.convert('L') - # TODO + def precision(self, filter=False): + if self.size_mismatch(): + return 0.0 + + diff = self.diff + if filter: + diff = diff.filter(gaussian_kernel) + + # See also http://effbot.org/zone/pil-comparing-images.htm + h = diff.histogram() + square_error = 0 + for i in range(1, 256): + square_error += sum(h[i : 3*256: 256])*i*i + rel_error = float(square_error*2 + 1) / float(self.diff.size[0]*self.diff.size[1]*3*255*255*2) + bits = -math.log(rel_error)/math.log(2.0) + return bits + + def ae(self, fuzz = 0.05): + # Compute absolute error + + if self.size_mismatch(): + return sys.maxint + + # TODO: this is approximate due to the grayscale conversion + h = self.diff.convert('L').histogram() + ae = sum(h[int(255 * fuzz) + 1 : 256]) + return ae def surface(html, image): @@ -60,14 +127,13 @@ def surface(html, image): def is_image(path): - return \ - path.endswith('.png') \ - and not path.endswith('.diff.png') \ - and not path.endswith('.thumb.png') + name = os.path.basename(path) + name, ext1 = os.path.splitext(name) + name, ext2 = os.path.splitext(name) + return ext1 in ('.png', '.bmp') and ext2 not in ('.diff', '.thumb') def find_images(prefix): - prefix = os.path.abspath(prefix) if os.path.isdir(prefix): prefix_dir = prefix else: @@ -84,17 +150,30 @@ def find_images(prefix): def main(): + global options + optparser = optparse.OptionParser( - usage="\n\t%prog [options] ", - version="%%prog") + usage="\n\t%prog [options] ") optparser.add_option( '-o', '--output', metavar='FILE', - type="string", dest="output", - help="output filename [stdout]") + type="string", dest="output", default='index.html', + help="output filename [default: %default]") optparser.add_option( '-f', '--fuzz', - type="string", dest="fuzz", default='5%', - help="fuzz [default: %default]") + type="float", dest="fuzz", default=0.05, + help="fuzz ratio [default: %default]") + optparser.add_option( + '-a', '--alpha', + action="store_true", dest="alpha", default=False, + help="take alpha channel in consideration") + optparser.add_option( + '--overwrite', + action="store_true", dest="overwrite", default=False, + help="overwrite images") + optparser.add_option( + '--show-all', + action="store_true", dest="show_all", default=False, + help="show all images, including similar ones") (options, args) = optparser.parse_args(sys.argv[1:]) @@ -116,28 +195,41 @@ def main(): html.write('\n') html.write(' \n') html.write(' \n') - html.write(' \n' % (ref_prefix, src_prefix)) + html.write(' \n' % (ref_prefix, src_prefix)) + failures = 0 for image in images: ref_image = ref_prefix + image src_image = src_prefix + image root, ext = os.path.splitext(src_image) delta_image = "%s.diff.png" % (root, ) if os.path.exists(ref_image) and os.path.exists(src_image): - if not os.path.exists(delta_image) \ - or (os.path.getmtime(delta_image) < os.path.getmtime(ref_image) \ - and os.path.getmtime(delta_image) < os.path.getmtime(src_image)): - subprocess.call(["compare", '-metric', 'AE', '-fuzz', options.fuzz, ref_image, src_image, delta_image]) - + print "Comparing %s and %s\n" % (ref_image, src_image) + comparer = Comparer(ref_image, src_image, options.alpha) + match = comparer.ae(fuzz=options.fuzz) == 0 + if match: + bgcolor = '#20ff20' + else: + failures += 1 + bgcolor = '#ff2020' html.write(' \n') - surface(html, ref_image) - surface(html, src_image) - surface(html, delta_image) + html.write(' \n' % (bgcolor, ref_image, image)) + if not match or options.show_all: + if options.overwrite \ + or not os.path.exists(delta_image) \ + or (os.path.getmtime(delta_image) < os.path.getmtime(ref_image) \ + and os.path.getmtime(delta_image) < os.path.getmtime(src_image)): + comparer.write_diff(delta_image, fuzz=options.fuzz) + surface(html, ref_image) + surface(html, src_image) + surface(html, delta_image) html.write(' \n') html.flush() html.write('
%s%sΔ
File%s%sΔ
%s
\n') html.write(' \n') html.write('\n') + if failures: + sys.exit(1) if __name__ == '__main__': main()