E-reader implementation.

1 files changed, 70 insertions, 0 deletions

diff --git a/pdftool.py b/pdftool.py
new file mode 100644
index 0000000..62d80df
--- /dev/null
+++ b/pdftool.py
@@ -0,0 +1,70 @@
+import os
+import re
+import sys
+import subprocess
+
+pdf = sys.argv[1]
+p1 = int(sys.argv[2])
+pn = int(sys.argv[3])
+
+root = pdf.replace("." + pdf.split('.')[-1], "")
+jpg = root + ".jpg"
+h = root + ".h"
+mon = "mon.jpg"
+res = "res.jpg"
+txt = root + ".txt"
+macro = root.upper() + "_H"
+data_len = pn - p1 + 1
+
+with open(h, "w") as f:
+    f.write("#ifndef {}\n".format(macro))
+    f.write("#define {}\n\n".format(macro))
+    f.write("const size_t data_len = {};\n\n".format(data_len))
+    f.write("const unsigned char data[{0}][48000] = {{\n".format(data_len))
+    for p in range(p1, pn + 1):
+        print("Processing page {}/{}...".format(p, pn))
+        subprocess.run(["pdftoppm", "-jpeg", "-r", "300", "-thinlinemode", "solid", "-f", str(p), "-singlefile", pdf, root])
+        subprocess.run(["mogrify", "-trim", "-rotate", "-90", jpg])
+        subprocess.run(["convert", jpg, "-monochrome", mon]) 
+        subprocess.run(["convert", mon, "-resize", "800x480!", res]) 
+        subprocess.run(["convert", res, "-depth", "1", "-format", "'txt'", txt])
+
+        f.write("\t{\n\t\t")
+        total = 0
+        with open(txt, "r") as fd:
+            n = 7
+            x = 0xFF
+            count = 0
+            fd.readline()
+            for line in fd:
+                px = re.search("\([^\)]+\)", line).group()
+                if px == "(0)":                        
+                    x &= ~(1 << n)
+                n -= 1
+                if n < 0:
+                    f.write("0x{:02X}, ".format(x))
+                    count += 1
+                    total += 1
+                    if count >= 12:
+                        f.write("\n\t\t")
+                        count = 0
+                    x = 0xFF
+                    n = 7
+        f.write("\n\t},\n")
+    f.write("};\n\n")    
+    f.write("#endif /* {} */".format(macro))    
+
+size = os.path.getsize(h)
+if size < 1024:
+    print("Done! Wrote {:0.1f}B to {}".format(size, h))
+elif size < pow(1024,2):
+    print("Done! Wrote {:0.1f}KB to {}".format(round(size / 1024, 2), h))
+elif size < pow(1024,3):
+    print("Done! Wrote {:0.1f}MB to {}".format(round(size / (pow(1024, 2)), 2), h))
+elif size < pow(1024,4):
+    print("Done! Wrote {:0.1f}GB to {}".format(round(size / (pow(1024, 3)), 2), h))
+
+os.remove(txt)
+os.remove(res)
+os.remove(mon)
+os.remove(jpg)