mapnik/tests/visual_tests/compare.py

import math, operator
import Image
import sys

COMPUTE_THRESHOLD = 16

errors = []

# returns true if pixels are not identical
def compare_pixels(pixel1, pixel2):
    r_diff = abs(pixel1[0] - pixel2[0])
    g_diff = abs(pixel1[1] - pixel2[1])
    b_diff = abs(pixel1[2] - pixel2[2])
    if(r_diff > COMPUTE_THRESHOLD or g_diff > COMPUTE_THRESHOLD or b_diff > COMPUTE_THRESHOLD):
        return True
    else:
        return False

# compare tow images and return number of different pixels
def compare(fn1, fn2):
    global errors
    im1 = Image.open(fn1)
    try:
        im2 = Image.open(fn2)
    except IOError:
        errors.append((fn1, None))
        return -1
    diff = 0
    pixels = im1.size[0] * im1.size[1]
    im1 = im1.getdata()
    im2 = im2.getdata()
    for i in range(3, pixels - 1, 3):
        if(compare_pixels(im1[i], im2[i])):
            diff = diff + 1
    if diff != 0:
        errors.append((fn1, diff))
    return diff

def summary():
    global errors
    if len(errors) != 0:
        print "-"*80
        print "Summary:"
        for error in errors:
            if (error[1] is None):
                print "Could not verify %s: No reference image found!" % error[0]
            else:
                print "%s failed: %d different pixels" % error
        print "-"*80
        sys.exit(1)