blob: 72f51250836a935cb98ee133eef29cc873152184 (
about) (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
"""
pdf2image custom buffer parsers
"""
from io import BytesIO
from typing import List
from PIL import Image
def parse_buffer_to_ppm(data: bytes) -> List[Image.Image]:
"""Parse PPM file bytes to Pillow Image
:param data: pdftoppm/pdftocairo output bytes
:type data: bytes
:return: List of PPM images parsed from the output
:rtype: List[Image.Image]
"""
images = []
index = 0
while index < len(data):
code, size, rgb = tuple(data[index : index + 40].split(b"\n")[0:3])
size_x, size_y = tuple(size.split(b" "))
file_size = len(code) + len(size) + len(rgb) + 3 + int(size_x) * int(size_y) * 3
images.append(Image.open(BytesIO(data[index : index + file_size])))
index += file_size
return images
def parse_buffer_to_pgm(data: bytes) -> List[Image.Image]:
"""Parse PGM file bytes to Pillow Image
:param data: pdftoppm/pdftocairo output bytes
:type data: bytes
:return: List of PGM images parsed from the output
:rtype: List[Image.Image]
"""
images = []
index = 0
while index < len(data):
code, size, maxval = tuple(data[index : index + 40].split(b"\n")[0:3])
size_x, size_y = tuple(size.split(b" "))
file_size = len(code) + len(size) + len(maxval) + 3 + int(size_x) * int(size_y)
images.append(Image.open(BytesIO(data[index : index + file_size])))
index += file_size
return images
def parse_buffer_to_jpeg(data: bytes) -> List[Image.Image]:
"""Parse JPEG file bytes to Pillow Image
:param data: pdftoppm/pdftocairo output bytes
:type data: bytes
:return: List of JPEG images parsed from the output
:rtype: List[Image.Image]
"""
return [
Image.open(BytesIO(image_data + b"\xff\xd9"))
for image_data in data.split(b"\xff\xd9")[
:-1
] # Last element is obviously empty
]
def parse_buffer_to_png(data: bytes) -> List[Image.Image]:
"""Parse PNG file bytes to Pillow Image
:param data: pdftoppm/pdftocairo output bytes
:type data: bytes
:return: List of PNG images parsed from the output
:rtype: List[Image.Image]
"""
images = []
c1 = 0
c2 = 0
data_len = len(data)
while c1 < data_len:
# IEND can appear in a PNG without being the actual end
if data[c2 : c2 + 4] == b"IEND" and (
c2 + 8 == data_len or data[c2 + 9 : c2 + 12] == b"PNG"
):
images.append(Image.open(BytesIO(data[c1 : c2 + 8])))
c1 = c2 + 8
c2 = c1
c2 += 1
return images
|
