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
|
"""
Converts PDFs to bitmaps for e-paper displays using Poppler and ImageMagick.
Usage: python -m doctoebm -o [path] [document]
o: output directory
document: path to PDF
"""
import io
import os
import re
import sys
import getopt
import subprocess
from pathlib import Path
dpi = "300"
screen_size = "480x800!"
doc = sys.argv[-1]
argv = sys.argv[1:-1]
opts, args = getopt.getopt(argv, "o:")
output_path = "."
for o, a in opts:
if o == "-o":
output_path = a.rstrip("/")
if not os.path.exists(output_path):
os.makedirs(output_path)
root = output_path + "/doc"
print("Converting PDF to JPEGs...")
subprocess.run(["pdftoppm", "-jpeg", "-progress", "-r", dpi, "-thinlinemode", "solid", doc, root])
print("Finished converting PDF to JPEGs.")
paths = list(Path(output_path).glob('*.jpg'))
print("Determining page size...")
w = h = 0
dx = dy = sys.maxsize
for p in paths:
rv = subprocess.run(
["magick", p, "-trim", "-format", "%[fx:w] %[fx:h] %[fx:page.x] %[fx:page.y]", "info:"],
capture_output=True,
text=True
)
area = [int(x) for x in rv.stdout.split()]
if w < area[0]:
w = area[0]
if h < area[1]:
h = area[1]
if dx > area[2]:
dx = area[2]
if dy > area[3]:
dy = area[3]
crop = "{}x{}+{}+{}".format(w, h, dx, dy)
print("Crop area: {}".format(crop))
for i, p in enumerate(paths):
print("Processing page {}/{}...".format(i+1, len(paths)))
jpg = str(p)
txt = jpg.replace(".jpg", ".txt")
ebm = jpg.replace(".jpg", ".ebm")
subprocess.run(["magick", jpg, "-crop", crop, jpg])
subprocess.run(["convert", jpg, "-resize", screen_size, jpg])
subprocess.run(["convert", jpg, "-threshold", "80%", jpg])
subprocess.run(["mogrify", "-rotate", "-90", jpg])
subprocess.run(["convert", jpg, "-depth", "1", "-format", "'txt'", txt])
with open(txt, "r") as src, open(ebm, "wb") as dst:
total = 0
n = 7
x = 0xFF
count = 0
src.readline()
for line in src:
px = re.search("\([^\)]+\)", line).group()
if px == "(0)":
x &= ~(1 << n)
n -= 1
if n < 0:
dst.write(x.to_bytes(1))
count += 1
total += 1
if count >= 12:
count = 0
n = 7
x = 0xFF
os.remove(txt)
os.remove(jpg)
|
