In DIY book maker, fix horizontal shifts.

[misc] / bookmaker.py

diff --git a/bookmaker.py b/bookmaker.py
index dd5b2e741fc78fbe2789ee53def885d0db0d9db8..0e35820654fb55677c7f341d6db1f3905871cea4 100755 (executable)
--- a/bookmaker.py
+++ b/bookmaker.py
@@ -12,6 +12,9 @@ parser.add_argument("-p", "--pages", dest="page_range", help="page range, e.g.,
 parser.add_argument("-c", "--crop", dest="crop_range", help="crops left, bottom, right, top – e.g., '10,10,10,10'")
 parser.add_argument("-n", "--nup4", dest="nup4", action='store_true', help="puts 4 input pages onto 1 output page")
 parser.add_argument("-a", "--analyze", dest="analyze", action="store_true", help="print lines identifying spine, page borders")
+parser.add_argument("-t", "--symmetry", dest="symmetry", action="store_true", help="alternate horizontal crops between odd and even pages")
+parser.add_argument("-s", "--second", dest="second", help="append second file as input to append")
+parser.add_argument("-r", "--rotate", dest="rotate", help="rotate page of number by 90°")
 args = parser.parse_args()
 
 with open(args.input_file, 'rb') as file:
@@ -32,9 +35,27 @@ with open(args.input_file, 'rb') as file:
         pages_to_add += [page]
         print("read in page number", page_num+1)
 
-    # normalize all pages to A4
-    for page_num in range(start_page, end_page):
-        page = reader.pages[page_num]
+    # add pages of second PDF
+    if args.second:
+        file2 = open(args.second, 'rb')
+        reader2 = pypdf.PdfReader(file2)
+        page_num = 1
+        for page in reader2.pages:
+            pages_to_add += [page]
+            print("read second PDF's page number", page_num)
+            page_num += 1
+
+    # rotate page canvas
+    if args.rotate:
+        page = pages_to_add[int(args.rotate) - 1]
+        page.add_transformation(pypdf.Transformation().translate(tx=-a4_width/2, ty=-a4_height/2))
+        page.add_transformation(pypdf.Transformation().rotate(-90))
+        page.add_transformation(pypdf.Transformation().translate(tx=a4_width/2, ty=a4_height/2))
+
+    # normalize all pages to portrait A4
+    for page in pages_to_add:
+        if "/Rotate" in page:
+            page.rotate(360 - page["/Rotate"])
         page.mediabox.left = 0
         page.mediabox.bottom = 0
         page.mediabox.top = a4_height 
@@ -61,12 +82,17 @@ with open(args.input_file, 'rb') as file:
 
     writer = pypdf.PdfWriter()
     if not args.nup4:
+        odd_page = True
         for page in pages_to_add:
-            page.add_transformation(pypdf.Transformation().translate(tx=-crop_left, ty=-crop_bottom))
+            if args.symmetry and odd_page:
+                page.add_transformation(pypdf.Transformation().translate(tx=-crop_right, ty=-crop_bottom))
+            else:
+                page.add_transformation(pypdf.Transformation().translate(tx=-crop_left, ty=-crop_bottom))
             page.add_transformation(pypdf.Transformation().scale(zoom, zoom))
             page.mediabox.right = cropped_width * zoom
             page.mediabox.top = cropped_height * zoom
             writer.add_page(page)
+            odd_page = not odd_page
     else:
         n_pages_per_axis = 2
         points_per_mm = 2.83465
@@ -111,9 +137,15 @@ with open(args.input_file, 'rb') as file:
             # in-section transformations: align pages on top, left-hand pages to left, right-hand to right
             page.add_transformation(pypdf.Transformation().translate(ty=(a4_height / zoom - (a4_height - crop_top))))
             if i == 0 or i == 2:
-                page.add_transformation(pypdf.Transformation().translate(tx=-crop_left))
+                if args.symmetry:
+                    page.add_transformation(pypdf.Transformation().translate(tx=-crop_left))
+                else:
+                    page.add_transformation(pypdf.Transformation().translate(tx=-crop_right))
             elif i == 1 or i == 3:
-                page.add_transformation(pypdf.Transformation().translate(tx=(a4_width / zoom - (a4_width - crop_right))))
+                if args.symmetry:
+                    page.add_transformation(pypdf.Transformation().translate(tx=(a4_width / zoom - (a4_width - crop_left))))
+                else:
+                    page.add_transformation(pypdf.Transformation().translate(tx=(a4_width / zoom - (a4_width - crop_right))))
             page.add_transformation(pypdf.Transformation().scale(zoom * bonus_shrink_factor, zoom * bonus_shrink_factor))
             if i == 2 or i == 3:
                 page.add_transformation(pypdf.Transformation().translate(ty=-2*printable_margin/printable_scale))
@@ -206,3 +238,5 @@ with open(args.input_file, 'rb') as file:
 
     with open(args.output_file, 'wb') as output_file:
         writer.write(output_file)
+    if args.second:
+        file2.close()