5 from reportlab.lib.pagesizes import A4
6 a4_width, a4_height = A4
7 points_per_cm = 10 * 72 / 25.4
8 cut_depth = 1.95 * points_per_cm
9 cut_width = 1.05 * points_per_cm
10 middle_point_depth = 0.4 * points_per_cm
12 parser = argparse.ArgumentParser(description="build print-ready book PDF")
13 parser.add_argument("-i", "--input_file", action="append", required=True, help="input PDF file")
14 parser.add_argument("-o", "--output_file", required=True, help="output PDF file")
15 parser.add_argument("-p", "--page_range", action="append", help="page range, e.g., '3-end'")
16 parser.add_argument("-c", "--crop_range", action="append", help="cm crops left, bottom, right, top – e.g., '10,10,10,10'; prefix with ':'-delimited page range to limit effect")
17 parser.add_argument("-t", "--symmetry", action="store_true", help="alternate horizontal crops between odd and even pages")
18 parser.add_argument("-r", "--rotate", dest="rotate", type=int, action="append", help="rotate page of number by 90° (usable multiple times on same page!)")
19 parser.add_argument("-n", "--nup4", action='store_true', help="puts 4 input pages onto 1 output page")
20 parser.add_argument("-a", "--analyze", action="store_true", help="in --nup4, print lines identifying spine, page borders")
21 parser.add_argument("-m", "--margin", type=float, default=0.43, help="print margin for --nup4 in cm (default 0.43)")
22 parser.add_argument("-s", "--spine", type=float, default=1, help="on --nup4, cm width of margin hidden in spine (default 1)")
23 args = parser.parse_args()
26 # select pages from input files
27 def parse_page_range(range_string, pages):
31 start, end = range_string.split('-')
32 if not (len(start) == 0 or start == "start"):
33 start_page = int(start) - 1
34 if not (len(end) == 0 or end == "end"):
36 return start_page, end_page
39 for i, input_file in enumerate(args.input_file):
40 file = open(input_file, 'rb')
41 opened_files += [file]
42 reader = pypdf.PdfReader(file)
44 if args.page_range and len(args.page_range) > i:
45 range_string = args.page_range[i]
46 start_page, end_page = parse_page_range(range_string, reader.pages)
48 for old_page_num in range(start_page, end_page):
50 page = reader.pages[new_page_num]
51 pages_to_add += [page]
52 print("-i, -p: read in %s page number %d as new page %d" % (input_file, new_page_num+1, new_page_num))
56 for rotate in args.rotate:
57 page = pages_to_add[rotate - 1]
58 page.add_transformation(pypdf.Transformation().translate(tx=-a4_width/2, ty=-a4_height/2))
59 page.add_transformation(pypdf.Transformation().rotate(-90))
60 page.add_transformation(pypdf.Transformation().translate(tx=a4_width/2, ty=a4_height/2))
61 print("-r: rotating (by 90°) page", rotate)
63 # normalize all pages to portrait A4
64 for page in pages_to_add:
66 page.rotate(360 - page["/Rotate"])
67 page.mediabox.left = 0
68 page.mediabox.bottom = 0
69 page.mediabox.top = a4_height
70 page.mediabox.right = a4_width
71 page.cropbox = page.mediabox
73 # determine page crops, zooms
74 crops_at_page = [(0,0,0,0)]*len(pages_to_add)
75 zoom_at_page = [1]*len(pages_to_add)
77 for crop_range in args.crop_range:
78 initial_split = crop_range.split(':')
79 if len(initial_split) > 1:
80 page_range = initial_split[0]
81 crops = initial_split[1]
84 crops = initial_split[0]
85 start_page, end_page = parse_page_range(page_range, pages_to_add)
86 crop_left, crop_bottom, crop_right, crop_top = [float(x) * points_per_cm for x in crops.split(',')]
87 print("-c: to pages %d to %d applying crops: left %dcm, bottom %dcm, right %dcm, top %dcm" % (start_page + 1, end_page, crop_left, crop_bottom, crop_right, crop_top))
88 cropped_width = a4_width - crop_left - crop_right
89 cropped_height = a4_height - crop_bottom - crop_top
91 zoom_horizontal = a4_width / (a4_width - crop_left - crop_right)
92 zoom_vertical = a4_height / (a4_height - crop_bottom - crop_top)
93 if (zoom_horizontal > 1 and zoom_vertical < 1) or (zoom_horizontal < 1 and zoom_vertical > 1):
94 print("Error: opposing zooms.")
96 elif zoom_horizontal + zoom_vertical > 2:
97 zoom = min(zoom_horizontal, zoom_vertical)
99 zoom = max(zoom_horizontal, zoom_vertical)
100 for page_num in range(start_page, end_page):
101 crops_at_page[page_num] = (crop_left, crop_bottom, crop_right, crop_top)
102 zoom_at_page[page_num] = zoom
104 writer = pypdf.PdfWriter()
107 for i, page in enumerate(pages_to_add):
108 crop_left, crop_bottom, crop_right, crop_top = crops_at_page[i]
109 zoom = zoom_at_page[i]
110 if args.symmetry and odd_page:
111 page.add_transformation(pypdf.Transformation().translate(tx=-crop_right, ty=-crop_bottom))
113 page.add_transformation(pypdf.Transformation().translate(tx=-crop_left, ty=-crop_bottom))
114 page.add_transformation(pypdf.Transformation().scale(zoom, zoom))
115 cropped_width = a4_width - crop_left - crop_right
116 cropped_height = a4_height - crop_bottom - crop_top
117 page.mediabox.right = cropped_width * zoom
118 page.mediabox.top = cropped_height * zoom
119 writer.add_page(page)
120 odd_page = not odd_page
124 spine_limit = args.spine * points_per_cm
125 printable_margin = args.margin * points_per_cm
126 printable_scale = (a4_width - 2*printable_margin)/a4_width
127 half_width = a4_width / n_pages_per_axis
128 half_height = a4_height / n_pages_per_axis
129 section_scale_factor = 1 / n_pages_per_axis
130 spine_part_of_page = (spine_limit / half_width) / printable_scale
131 bonus_shrink_factor = 1 - spine_part_of_page
135 mod_to_8 = len(pages_to_add) % 8
137 for _ in range(8 - mod_to_8):
138 new_page = pypdf.PageObject.create_blank_page(width=a4_width, height=a4_height)
139 pages_to_add += [new_page]
142 for page in pages_to_add:
149 new_i_order += [8 * n_eights + 3,
158 new_page_order += [eight_pack[3]] # page front, upper left
159 new_page_order += [eight_pack[0]] # page front, upper right
160 new_page_order += [eight_pack[7]] # page front, lower left
161 new_page_order += [eight_pack[4]] # page front, lower right
162 new_page_order += [eight_pack[1]] # page back, upper left
163 new_page_order += [eight_pack[2]] # page back, upper right
164 new_page_order += [eight_pack[5]] # page back, lower left
165 new_page_order += [eight_pack[6]] # page back, lower right
169 for j, page in enumerate(new_page_order):
171 new_page = pypdf.PageObject.create_blank_page(width=a4_width, height=a4_height)
173 # in-section transformations: align pages on top, left-hand pages to left, right-hand to right
174 new_i = new_i_order[j]
175 crop_left, crop_bottom, crop_right, crop_top = crops_at_page[new_i]
176 zoom = zoom_at_page[new_i]
177 page.add_transformation(pypdf.Transformation().translate(ty=(a4_height / zoom - (a4_height - crop_top))))
180 page.add_transformation(pypdf.Transformation().translate(tx=-crop_left))
182 page.add_transformation(pypdf.Transformation().translate(tx=-crop_right))
183 elif i == 1 or i == 3:
185 page.add_transformation(pypdf.Transformation().translate(tx=(a4_width / zoom - (a4_width - crop_left))))
187 page.add_transformation(pypdf.Transformation().translate(tx=(a4_width / zoom - (a4_width - crop_right))))
188 page.add_transformation(pypdf.Transformation().scale(zoom * bonus_shrink_factor, zoom * bonus_shrink_factor))
190 page.add_transformation(pypdf.Transformation().translate(ty=-2*printable_margin/printable_scale))
192 # outer section transformations
193 page.add_transformation(pypdf.Transformation().translate(ty=(1-bonus_shrink_factor)*a4_height))
195 y_section = a4_height
196 page.mediabox.bottom = half_height
197 page.mediabox.top = a4_height
200 page.mediabox.bottom = 0
201 page.mediabox.top = half_height
204 page.mediabox.left = 0
205 page.mediabox.right = half_width
207 page.add_transformation(pypdf.Transformation().translate(tx=(1-bonus_shrink_factor)*a4_width))
209 page.mediabox.left = half_width
210 page.mediabox.right = a4_width
211 page.add_transformation(pypdf.Transformation().translate(tx=x_section, ty=y_section))
212 page.add_transformation(pypdf.Transformation().scale(section_scale_factor, section_scale_factor))
213 new_page.merge_page(page)
215 print("merged page number", page_count)
218 from reportlab.pdfgen import canvas
221 packet = io.BytesIO()
222 c = canvas.Canvas(packet, pagesize=A4)
224 c.line(0, a4_height, a4_width, a4_height)
225 c.line(0, half_height, a4_width, half_height)
226 c.line(0, 0, a4_width, 0)
227 c.line(0, a4_height, 0, 0)
228 c.line(half_width, a4_height, half_width, 0)
229 c.line(a4_width, a4_height, a4_width, 0)
231 new_pdf = pypdf.PdfReader(packet)
232 new_page.merge_page(new_pdf.pages[0])
233 printable_offset_x = printable_margin
234 printable_offset_y = printable_margin * a4_height / a4_width
235 new_page.add_transformation(pypdf.Transformation().scale(printable_scale, printable_scale))
236 new_page.add_transformation(pypdf.Transformation().translate(tx=printable_offset_x, ty=printable_offset_y))
237 x_left_spine_limit = half_width * bonus_shrink_factor
238 x_right_spine_limit = a4_width - x_left_spine_limit
239 if args.analyze or front_page:
240 packet = io.BytesIO()
241 c = canvas.Canvas(packet, pagesize=A4)
245 c.line(x_left_spine_limit, a4_height, x_left_spine_limit, 0)
246 c.line(x_right_spine_limit, a4_height, x_right_spine_limit, 0)
250 start_up_left_left_x = x_left_spine_limit - 0.5 * cut_width
251 start_up_left_right_x = x_left_spine_limit + 0.5 * cut_width
252 middle_point_up_left_y = half_height + middle_point_depth
253 end_point_up_left_y = half_height + cut_depth
254 c.line(start_up_left_right_x, half_height, x_left_spine_limit, end_point_up_left_y)
255 c.line(x_left_spine_limit, end_point_up_left_y, x_left_spine_limit, middle_point_up_left_y)
256 c.line(x_left_spine_limit, middle_point_up_left_y, start_up_left_left_x, half_height)
258 start_down_right_left_x = x_right_spine_limit - 0.5 * cut_width
259 start_down_right_right_x = x_right_spine_limit + 0.5 * cut_width
260 middle_point_down_right_y = half_height - middle_point_depth
261 end_point_down_right_y = half_height - cut_depth
262 c.line(start_down_right_left_x, half_height, x_right_spine_limit, end_point_down_right_y)
263 c.line(x_right_spine_limit, end_point_down_right_y, x_right_spine_limit, middle_point_down_right_y)
264 c.line(x_right_spine_limit, middle_point_down_right_y, start_down_right_right_x, half_height)
266 if args.analyze or front_page:
268 new_pdf = pypdf.PdfReader(packet)
269 new_page.merge_page(new_pdf.pages[0])
270 writer.add_page(new_page)
272 front_page = not front_page
275 for file in opened_files:
277 with open(args.output_file, 'wb') as output_file:
278 writer.write(output_file)