Refactor PDFium python test utilities.
[pdfium.git] / testing / tools / pngdiffer.py
1 #!/usr/bin/env python
2 # Copyright 2015 The PDFium Authors. All rights reserved.
3 # Use of this source code is governed by a BSD-style license that can be
4 # found in the LICENSE file.
5
6 import os
7 import subprocess
8 import sys
9
10 class PNGDiffer():
11   ACTUAL_TEMPLATE = '.pdf.%d.png'
12   EXPECTED_TEMPLATE = '_expected' + ACTUAL_TEMPLATE
13
14   def __init__(self, finder):
15     self.pdfium_diff_path = finder.ExecutablePath('pdfium_diff')
16
17   def HasDifferences(self, input_filename, source_dir, working_dir):
18     input_root, _ = os.path.splitext(input_filename)
19     actual_path_template = os.path.join(
20         working_dir, input_root + self.ACTUAL_TEMPLATE)
21     expected_path_template = os.path.join(
22         source_dir, input_root + self.EXPECTED_TEMPLATE)
23     i = 0
24     try:
25       while True:
26         actual_path = actual_path_template % i;
27         expected_path = expected_path_template % i;
28         if not os.path.exists(expected_path):
29           if i == 0:
30             print "WARNING: no expected results files for " + input_filename
31           break
32         print "Checking " + actual_path
33         sys.stdout.flush()
34         subprocess.check_call(
35             [self.pdfium_diff_path, expected_path, actual_path])
36         i += 1
37     except subprocess.CalledProcessError as e:
38       print "FAILURE: " + input_filename + "; " + str(e)
39       return True
40     return False