diff options
author | Tom Sepez <tsepez@chromium.org> | 2015-01-21 14:38:33 -0800 |
---|---|---|
committer | Tom Sepez <tsepez@chromium.org> | 2015-01-21 14:38:33 -0800 |
commit | 9dcd7b841a6ec11b3cfff39f8d2d1f033dea9705 (patch) | |
tree | e6d363241976a3fdc85ca2f27e6035ad83e90cb8 /testing/tools/fixup_pdf_template.py | |
parent | dc43b322915c5256a5a9458e6b8cd28bcecf91ae (diff) | |
download | pdfium-9dcd7b841a6ec11b3cfff39f8d2d1f033dea9705.tar.xz |
Add tests for GetNamedDests() API.
Follow-on work from patch at https://codereview.chromium.org/845643008. This
incorporates that patch, plus adds tests for it and similar conditions. A few
changes are introduced to correct expected behaviour.
This also incoprorates Deepak's fix in https://codereview.chromium.org/845643008/
This incorporates a formerly unlanded tool to generate small valid PDF
files from template input, which is needed to cover all the error cases
here. See https://codereview.chromium.org/791993006/
BUG=450133
R=bo_xu@foxitsoftware.com
Review URL: https://codereview.chromium.org/837723009
Diffstat (limited to 'testing/tools/fixup_pdf_template.py')
-rwxr-xr-x | testing/tools/fixup_pdf_template.py | 89 |
1 files changed, 89 insertions, 0 deletions
diff --git a/testing/tools/fixup_pdf_template.py b/testing/tools/fixup_pdf_template.py new file mode 100755 index 0000000000..873caeedde --- /dev/null +++ b/testing/tools/fixup_pdf_template.py @@ -0,0 +1,89 @@ +#!/usr/bin/env python +# Copyright 2014 The PDFium Authors. All rights reserved. +# Use of this source code is governed by a BSD-style license that can be +# found in the LICENSE file. + +"""Expands a hand-written PDF testcase (template) into a valid PDF file. + +There are several places in a PDF file where byte-offsets are required. This +script replaces {{name}}-style variables in the input with calculated results + + {{header}} - expands to the header comment required for PDF files. + {{xref}} - expands to a generated xref table, noting the offset. + {{startxref} - expands to a startxref directive followed by correct offset. + {{object x y}} - expands to |x y obj| declaration, noting the offset.""" + +import optparse +import os +import re +import sys + +class TemplateProcessor: + HEADER_TOKEN = '{{header}}' + HEADER_REPLACEMENT = '%PDF-1.7\n%\xa0\xf2\xa4\xf4' + + XREF_TOKEN = '{{xref}}' + XREF_REPLACEMENT = 'xref\n%d %d\n' + XREF_REPLACEMENT_N = '%010d %05d n\n' + XREF_REPLACEMENT_F = '0000000000 65536 f\n' + + STARTXREF_TOKEN= '{{startxref}}' + STARTXREF_REPLACEMENT = 'startxref\n%d' + + OBJECT_PATTERN = r'\{\{object\s+(\d+)\s+(\d+)\}\}' + OBJECT_REPLACEMENT = r'\1 \2 obj' + + def __init__(self): + self.offset = 0 + self.xref_offset = 0 + self.max_object_number = 0 + self.objects = { } + + def insert_xref_entry(self, object_number, generation_number): + self.objects[object_number] = (self.offset, generation_number) + self.max_object_number = max(self.max_object_number, object_number) + + def generate_xref_table(self): + result = self.XREF_REPLACEMENT % (0, self.max_object_number + 1) + for i in range(0, self.max_object_number + 1): + if i in self.objects: + result += self.XREF_REPLACEMENT_N % self.objects[i] + else: + result += self.XREF_REPLACEMENT_F + return result + + def process_line(self, line): + if self.HEADER_TOKEN in line: + line = line.replace(self.HEADER_TOKEN, self.HEADER_REPLACEMENT) + if self.XREF_TOKEN in line: + self.xref_offset = self.offset + line = self.generate_xref_table() + if self.STARTXREF_TOKEN in line: + replacement = self.STARTXREF_REPLACEMENT % self.xref_offset + line = line.replace(self.STARTXREF_TOKEN, replacement) + match = re.match(self.OBJECT_PATTERN, line) + if match: + self.insert_xref_entry(int(match.group(1)), int(match.group(2))) + line = re.sub(self.OBJECT_PATTERN, self.OBJECT_REPLACEMENT, line) + self.offset += len(line) + return line + +def expand_file(input_filename): + (input_root, extension) = os.path.splitext(input_filename) + output_filename = input_root + '.pdf' + processor = TemplateProcessor() + try: + with open(input_filename, 'r') as infile: + with open(output_filename, 'w') as outfile: + for line in infile: + outfile.write(processor.process_line(line)) + except IOError: + print >> sys.stderr, 'failed to process %s' % input_filename + +def main(): + for arg in sys.argv[1:]: + expand_file(arg) + return 0 + +if __name__ == '__main__': + sys.exit(main()) |