From f51044f9873a900b1aa8e25ae90cccaa41d22fb8 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Mon, 3 Apr 2023 09:29:25 +0200 Subject: Merging upstream version 0.6.0. Signed-off-by: Daniel Baumann --- pydyf/__init__.py | 210 +++++++++++++++++++++++++++++++++++------------------- 1 file changed, 136 insertions(+), 74 deletions(-) (limited to 'pydyf') diff --git a/pydyf/__init__.py b/pydyf/__init__.py index 85811f4..2e589fe 100755 --- a/pydyf/__init__.py +++ b/pydyf/__init__.py @@ -7,8 +7,9 @@ import re import zlib from codecs import BOM_UTF16_BE from hashlib import md5 +from math import ceil, log -VERSION = __version__ = '0.5.0' +VERSION = __version__ = '0.6.0' def _to_bytes(item): @@ -21,7 +22,7 @@ def _to_bytes(item): if item.is_integer(): return f'{int(item):d}'.encode('ascii') else: - return f'{item:f}'.encode('ascii') + return f'{item:f}'.rstrip('0').encode('ascii') elif isinstance(item, int): return f'{item:d}'.encode('ascii') return str(item).encode('ascii') @@ -43,51 +44,41 @@ class Object: @property def indirect(self): """Indirect representation of an object.""" - return b'\n'.join(( - str(self.number).encode() + b' ' + - str(self.generation).encode() + b' obj', - self.data, - b'endobj', - )) + header = f'{self.number} {self.generation} obj\n'.encode() + return header + self.data + b'\nendobj' @property def reference(self): """Object identifier.""" - return ( - str(self.number).encode() + b' ' + - str(self.generation).encode() + b' R') + return f'{self.number} {self.generation} R'.encode() @property def data(self): """Data contained in the object. Shall be defined in each subclass.""" raise NotImplementedError() + @property + def compressible(self): + """Whether the object can be included in an object stream.""" + return not self.generation and not isinstance(self, Stream) -class Dictionary(Object, dict): - """PDF Dictionary object. - - Inherits from :class:`Object` and Python :obj:`dict`. - """ +class Dictionary(Object, dict): + """PDF Dictionary object.""" def __init__(self, values=None): Object.__init__(self) dict.__init__(self, values or {}) @property def data(self): - result = [b'<<'] - for key, value in self.items(): - result.append(b'/' + _to_bytes(key) + b' ' + _to_bytes(value)) - result.append(b'>>') - return b'\n'.join(result) + result = [ + b'/' + _to_bytes(key) + b' ' + _to_bytes(value) + for key, value in self.items()] + return b'<<' + b''.join(result) + b'>>' class Stream(Object): - """PDF Stream object. - - Inherits from :class:`Object`. - - """ + """PDF Stream object.""" def __init__(self, stream=None, extra=None, compress=False): super().__init__() #: Python array of data composing stream. @@ -221,6 +212,10 @@ class Stream(Object): """Begin new subpath by moving current point to ``(x, y)``.""" self.stream.append(b' '.join((_to_bytes(x), _to_bytes(y), b'm'))) + def move_text_to(self, x, y): + """Move text to next line at ``(x, y)`` distance from previous line.""" + self.stream.append(b' '.join((_to_bytes(x), _to_bytes(y), b'Td'))) + def shading(self, name): """Paint shape and color shading using shading dictionary ``name``.""" self.stream.append(b'/' + _to_bytes(name) + b' sh') @@ -310,9 +305,13 @@ class Stream(Object): self.stream.append(b'/' + _to_bytes(state_name) + b' gs') def show_text(self, text): - """Show text.""" + """Show text strings with individual glyph positioning.""" self.stream.append(b'[' + _to_bytes(text) + b'] TJ') + def show_text_string(self, text): + """Show single text string.""" + self.stream.append(String(text).data + b' Tj') + def stroke(self): """Stroke path.""" self.stream.append(b'S') @@ -369,7 +368,7 @@ class Stream(Object): extra = Dictionary(self.extra.copy()) if self.compress: extra['Filter'] = '/FlateDecode' - compressobj = zlib.compressobj() + compressobj = zlib.compressobj(level=9) stream = compressobj.compress(stream) stream += compressobj.flush() extra['Length'] = len(stream) @@ -377,11 +376,7 @@ class Stream(Object): class String(Object): - """PDF String object. - - Inherits from :class:`Object`. - - """ + """PDF String object.""" def __init__(self, string=''): super().__init__() #: Unicode string. @@ -402,22 +397,14 @@ class String(Object): class Array(Object, list): - """PDF Array object. - - Inherits from :class:`Object` and Python :obj:`list`. - - """ + """PDF Array object.""" def __init__(self, array=None): Object.__init__(self) list.__init__(self, array or []) @property def data(self): - result = [b'['] - for child in self: - result.append(_to_bytes(child)) - result.append(b']') - return b' '.join(result) + return b'[' + b' '.join(_to_bytes(child) for child in self) + b']' class PDF: @@ -500,13 +487,14 @@ class PDF: self.current_position += len(content) + 1 output.write(content + b'\n') - def write(self, output, version=None, identifier=None): + def write(self, output, version=None, identifier=None, compress=False): """Write PDF to output. :param output: Output stream. :type output: binary :term:`file object` :param bytes version: PDF version. :param bytes identifier: PDF file identifier. + :param bool compress: whether the PDF uses a compressed object stream. """ version = self.version if version is None else _to_bytes(version) @@ -516,36 +504,110 @@ class PDF: self.write_line(b'%PDF-' + version, output) self.write_line(b'%\xf0\x9f\x96\xa4', output) - # Write all non-free PDF objects - for object_ in self.objects: - if object_.free == 'f': - continue - object_.offset = self.current_position - self.write_line(object_.indirect, output) - - # Write cross reference table - self.xref_position = self.current_position - self.write_line(b'xref', output) - self.write_line(f'0 {len(self.objects)}'.encode(), output) - for object_ in self.objects: - self.write_line( - (f'{object_.offset:010} {object_.generation:05} ' - f'{object_.free} ').encode(), output) - - # Write trailer - self.write_line(b'trailer', output) - self.write_line(b'<<', output) - self.write_line(f'/Size {len(self.objects)}'.encode(), output) - self.write_line(b'/Root ' + self.catalog.reference, output) - self.write_line(b'/Info ' + self.info.reference, output) - if identifier is not None: - data = b''.join( - obj.data for obj in self.objects if obj.free != 'f') - data_hash = md5(data).hexdigest().encode() - self.write_line( - b'/ID [' + String(identifier).data + b' ' + - String(data_hash).data + b']', output) - self.write_line(b'>>', output) + if version >= b'1.5' and compress: + # Store compressed objects for later and write other ones in PDF + compressed_objects = [] + for object_ in self.objects: + if object_.free == 'f': + continue + if object_.compressible: + compressed_objects.append(object_) + else: + object_.offset = self.current_position + self.write_line(object_.indirect, output) + + # Write compressed objects in object stream + stream = [[]] + position = 0 + for i, object_ in enumerate(compressed_objects): + data = object_.data + stream.append(data) + stream[0].append(object_.number) + stream[0].append(position) + position += len(data) + 1 + stream[0] = ' '.join(str(i) for i in stream[0]) + extra = { + 'Type': '/ObjStm', + 'N': len(compressed_objects), + 'First': len(stream[0]) + 1, + } + object_stream = Stream(stream, extra, compress) + object_stream.offset = self.current_position + self.add_object(object_stream) + self.write_line(object_stream.indirect, output) + + # Write cross-reference stream + xref = [] + dict_index = 0 + for object_ in self.objects: + if object_.compressible: + xref.append((2, object_stream.number, dict_index)) + dict_index += 1 + else: + xref.append(( + bool(object_.number), object_.offset, + object_.generation)) + xref.append((1, self.current_position, 0)) + + field2_size = ceil(log(self.current_position, 8)) + max_generation = max( + object_.generation for object_ in self.objects) + field3_size = ceil(log( + max(max_generation, len(compressed_objects)), 8)) + xref_lengths = (1, field2_size, field3_size) + xref_stream = b''.join( + value.to_bytes(length, 'big') + for line in xref for length, value in zip(xref_lengths, line)) + extra = { + 'Type': '/XRef', + 'Index': Array((0, len(self.objects) + 1)), + 'W': Array(xref_lengths), + 'Size': len(self.objects) + 1, + 'Root': self.catalog.reference, + 'Info': self.info.reference, + } + if identifier is not None: + data = b''.join( + obj.data for obj in self.objects if obj.free != 'f') + data_hash = md5(data).hexdigest().encode() + extra['ID'] = Array(( + String(identifier).data, String(data_hash).data)) + dict_stream = Stream([xref_stream], extra, compress) + self.xref_position = dict_stream.offset = self.current_position + self.add_object(dict_stream) + self.write_line(dict_stream.indirect, output) + else: + # Write all non-free PDF objects + for object_ in self.objects: + if object_.free == 'f': + continue + object_.offset = self.current_position + self.write_line(object_.indirect, output) + + # Write cross-reference table + self.xref_position = self.current_position + self.write_line(b'xref', output) + self.write_line(f'0 {len(self.objects)}'.encode(), output) + for object_ in self.objects: + self.write_line( + (f'{object_.offset:010} {object_.generation:05} ' + f'{object_.free} ').encode(), output) + + # Write trailer + self.write_line(b'trailer', output) + self.write_line(b'<<', output) + self.write_line(f'/Size {len(self.objects)}'.encode(), output) + self.write_line(b'/Root ' + self.catalog.reference, output) + self.write_line(b'/Info ' + self.info.reference, output) + if identifier is not None: + data = b''.join( + obj.data for obj in self.objects if obj.free != 'f') + data_hash = md5(data).hexdigest().encode() + self.write_line( + b'/ID [' + String(identifier).data + b' ' + + String(data_hash).data + b']', output) + self.write_line(b'>>', output) + self.write_line(b'startxref', output) self.write_line(f'{self.xref_position}'.encode(), output) self.write_line(b'%%EOF', output) -- cgit v1.2.3