X-Git-Url: https://git.cworth.org/git?a=blobdiff_plain;f=scripts%2Fsnapdiff.py;h=0ed3937a8f8e669accf577c4b06d6472784e3c57;hb=ae2b4d32ed56e3ac193cc7205aeb58082c448ce8;hp=965044068a1d3a55cf401c4f76e791dd2a31d79d;hpb=0b956fd72fee30dccb7c011450066bc8e47c2d66;p=apitrace

diff --git a/scripts/snapdiff.py b/scripts/snapdiff.py
index 9650440..0ed3937 100755
--- a/scripts/snapdiff.py
+++ b/scripts/snapdiff.py
@@ -36,9 +36,9 @@ import optparse
 import math
 import operator
 
-import Image
-import ImageChops
-import ImageEnhance
+from PIL import Image
+from PIL import ImageChops
+from PIL import ImageEnhance
 
 
 thumb_size = 320, 320
@@ -48,8 +48,15 @@ class Comparer:
     '''Image comparer.'''
 
     def __init__(self, ref_image, src_image, alpha = False):
-        self.ref_im = Image.open(ref_image)
-        self.src_im = Image.open(src_image)
+        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: