]> git.cworth.org Git - apitrace/blobdiff - scripts/snapdiff.py
Update docs / scripts for tracedump -> apitrace dump change.
[apitrace] / scripts / snapdiff.py
index 965044068a1d3a55cf401c4f76e791dd2a31d79d..0ed3937a8f8e669accf577c4b06d6472784e3c57 100755 (executable)
@@ -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: