chromium/tools/site_compare/operators/equals_with_mask.py

# Copyright 2011 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Compare two images for equality, subject to a mask."""

from PIL import Image
from PIL import ImageChops

import os.path


def Compare(file1, file2, **kwargs):
  """Compares two images to see if they're identical subject to a mask.

  An optional directory containing masks is supplied. If a mask exists
  which matches file1's name, areas under the mask where it's black
  are ignored.

  Args:
    file1: path to first image to compare
    file2: path to second image to compare
    kwargs: ["maskdir"] contains the directory holding the masks

  Returns:
    None if the images are identical
    A tuple of (errorstring, image) if they're not
  """

  maskdir = None
  if "maskdir" in kwargs:
    maskdir = kwargs["maskdir"]

  im1 = Image.open(file1)
  im2 = Image.open(file2)

  if im1.size != im2.size:
    return ("The images are of different size (%r vs %r)" %
            (im1.size, im2.size), im1)

  diff = ImageChops.difference(im1, im2)

  if maskdir:
    maskfile = os.path.join(maskdir, os.path.basename(file1))
    if os.path.exists(maskfile):
      mask = Image.open(maskfile)

      if mask.size != im1.size:
        return ("The mask is of a different size than the images (%r vs %r)" %
                (mask.size, im1.size), mask)

      diff = ImageChops.multiply(diff, mask.convert(diff.mode))

  if max(diff.getextrema()) != (0, 0):
    return ("The images differ", diff)
  else:
    return None