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)
47 for page_num in range(start_page, end_page):
48 page = reader.pages[page_num]
49 pages_to_add += [page]
50 print("read in %s page number %d" % (input_file, page_num+1))
54 for rotate in args.rotate:
55 page = pages_to_add[rotate - 1]
56 page.add_transformation(pypdf.Transformation().translate(tx=-a4_width/2, ty=-a4_height/2))
57 page.add_transformation(pypdf.Transformation().rotate(-90))
58 page.add_transformation(pypdf.Transformation().translate(tx=a4_width/2, ty=a4_height/2))
60 # normalize all pages to portrait A4
61 for page in pages_to_add:
63 page.rotate(360 - page["/Rotate"])
64 page.mediabox.left = 0
65 page.mediabox.bottom = 0
66 page.mediabox.top = a4_height
67 page.mediabox.right = a4_width
68 page.cropbox = page.mediabox
70 # determine page crops, zooms
71 crops_at_page = [(0,0,0,0)]*len(pages_to_add)
72 zoom_at_page = [1]*len(pages_to_add)
74 for crop_range in args.crop_range:
75 initial_split = crop_range.split(':')
76 if len(initial_split) > 1:
77 page_range = initial_split[0]
78 crops = initial_split[1]
81 crops = initial_split[0]
82 start_page, end_page = parse_page_range(page_range, pages_to_add)
83 crop_left, crop_bottom, crop_right, crop_top = [float(x) * points_per_cm for x in crops.split(',')]
84 cropped_width = a4_width - crop_left - crop_right
85 cropped_height = a4_height - crop_bottom - crop_top
87 zoom_horizontal = a4_width / (a4_width - crop_left - crop_right)
88 zoom_vertical = a4_height / (a4_height - crop_bottom - crop_top)
89 if (zoom_horizontal > 1 and zoom_vertical < 1) or (zoom_horizontal < 1 and zoom_vertical > 1):
90 print("Error: opposing zooms.")
92 elif zoom_horizontal + zoom_vertical > 2:
93 zoom = min(zoom_horizontal, zoom_vertical)
95 zoom = max(zoom_horizontal, zoom_vertical)
96 for page_num in range(start_page, end_page):
97 crops_at_page[page_num] = (crop_left, crop_bottom, crop_right, crop_top)
98 zoom_at_page[page_num] = zoom
100 writer = pypdf.PdfWriter()
103 for i, page in enumerate(pages_to_add):
104 crop_left, crop_bottom, crop_right, crop_top = crops_at_page[i]
105 zoom = zoom_at_page[i]
106 if args.symmetry and odd_page:
107 page.add_transformation(pypdf.Transformation().translate(tx=-crop_right, ty=-crop_bottom))
109 page.add_transformation(pypdf.Transformation().translate(tx=-crop_left, ty=-crop_bottom))
110 page.add_transformation(pypdf.Transformation().scale(zoom, zoom))
111 cropped_width = a4_width - crop_left - crop_right
112 cropped_height = a4_height - crop_bottom - crop_top
113 page.mediabox.right = cropped_width * zoom
114 page.mediabox.top = cropped_height * zoom
115 writer.add_page(page)
116 odd_page = not odd_page
120 spine_limit = args.spine * points_per_cm
121 printable_margin = args.margin * points_per_cm
122 printable_scale = (a4_width - 2*printable_margin)/a4_width
123 half_width = a4_width / n_pages_per_axis
124 half_height = a4_height / n_pages_per_axis
125 section_scale_factor = 1 / n_pages_per_axis
126 spine_part_of_page = (spine_limit / half_width) / printable_scale
127 bonus_shrink_factor = 1 - spine_part_of_page
131 mod_to_8 = len(pages_to_add) % 8
133 for _ in range(8 - mod_to_8):
134 new_page = pypdf.PageObject.create_blank_page(width=a4_width, height=a4_height)
135 pages_to_add += [new_page]
138 for page in pages_to_add:
145 new_i_order += [8 * n_eights + 3,
154 new_page_order += [eight_pack[3]] # page front, upper left
155 new_page_order += [eight_pack[0]] # page front, upper right
156 new_page_order += [eight_pack[7]] # page front, lower left
157 new_page_order += [eight_pack[4]] # page front, lower right
158 new_page_order += [eight_pack[1]] # page back, upper left
159 new_page_order += [eight_pack[2]] # page back, upper right
160 new_page_order += [eight_pack[5]] # page back, lower left
161 new_page_order += [eight_pack[6]] # page back, lower right
165 for j, page in enumerate(new_page_order):
167 new_page = pypdf.PageObject.create_blank_page(width=a4_width, height=a4_height)
169 # in-section transformations: align pages on top, left-hand pages to left, right-hand to right
170 new_i = new_i_order[j]
171 crop_left, crop_bottom, crop_right, crop_top = crops_at_page[new_i]
172 zoom = zoom_at_page[new_i]
173 page.add_transformation(pypdf.Transformation().translate(ty=(a4_height / zoom - (a4_height - crop_top))))
176 page.add_transformation(pypdf.Transformation().translate(tx=-crop_left))
178 page.add_transformation(pypdf.Transformation().translate(tx=-crop_right))
179 elif i == 1 or i == 3:
181 page.add_transformation(pypdf.Transformation().translate(tx=(a4_width / zoom - (a4_width - crop_left))))
183 page.add_transformation(pypdf.Transformation().translate(tx=(a4_width / zoom - (a4_width - crop_right))))
184 page.add_transformation(pypdf.Transformation().scale(zoom * bonus_shrink_factor, zoom * bonus_shrink_factor))
186 page.add_transformation(pypdf.Transformation().translate(ty=-2*printable_margin/printable_scale))
188 # outer section transformations
189 page.add_transformation(pypdf.Transformation().translate(ty=(1-bonus_shrink_factor)*a4_height))
191 y_section = a4_height
192 page.mediabox.bottom = half_height
193 page.mediabox.top = a4_height
196 page.mediabox.bottom = 0
197 page.mediabox.top = half_height
200 page.mediabox.left = 0
201 page.mediabox.right = half_width
203 page.add_transformation(pypdf.Transformation().translate(tx=(1-bonus_shrink_factor)*a4_width))
205 page.mediabox.left = half_width
206 page.mediabox.right = a4_width
207 page.add_transformation(pypdf.Transformation().translate(tx=x_section, ty=y_section))
208 page.add_transformation(pypdf.Transformation().scale(section_scale_factor, section_scale_factor))
209 new_page.merge_page(page)
211 print("merged page number", page_count)
214 from reportlab.pdfgen import canvas
217 packet = io.BytesIO()
218 c = canvas.Canvas(packet, pagesize=A4)
220 c.line(0, a4_height, a4_width, a4_height)
221 c.line(0, half_height, a4_width, half_height)
222 c.line(0, 0, a4_width, 0)
223 c.line(0, a4_height, 0, 0)
224 c.line(half_width, a4_height, half_width, 0)
225 c.line(a4_width, a4_height, a4_width, 0)
227 new_pdf = pypdf.PdfReader(packet)
228 new_page.merge_page(new_pdf.pages[0])
229 printable_offset_x = printable_margin
230 printable_offset_y = printable_margin * a4_height / a4_width
231 new_page.add_transformation(pypdf.Transformation().scale(printable_scale, printable_scale))
232 new_page.add_transformation(pypdf.Transformation().translate(tx=printable_offset_x, ty=printable_offset_y))
233 x_left_spine_limit = half_width * bonus_shrink_factor
234 x_right_spine_limit = a4_width - x_left_spine_limit
235 if args.analyze or front_page:
236 packet = io.BytesIO()
237 c = canvas.Canvas(packet, pagesize=A4)
241 c.line(x_left_spine_limit, a4_height, x_left_spine_limit, 0)
242 c.line(x_right_spine_limit, a4_height, x_right_spine_limit, 0)
246 start_up_left_left_x = x_left_spine_limit - 0.5 * cut_width
247 start_up_left_right_x = x_left_spine_limit + 0.5 * cut_width
248 middle_point_up_left_y = half_height + middle_point_depth
249 end_point_up_left_y = half_height + cut_depth
250 c.line(start_up_left_right_x, half_height, x_left_spine_limit, end_point_up_left_y)
251 c.line(x_left_spine_limit, end_point_up_left_y, x_left_spine_limit, middle_point_up_left_y)
252 c.line(x_left_spine_limit, middle_point_up_left_y, start_up_left_left_x, half_height)
254 start_down_right_left_x = x_right_spine_limit - 0.5 * cut_width
255 start_down_right_right_x = x_right_spine_limit + 0.5 * cut_width
256 middle_point_down_right_y = half_height - middle_point_depth
257 end_point_down_right_y = half_height - cut_depth
258 c.line(start_down_right_left_x, half_height, x_right_spine_limit, end_point_down_right_y)
259 c.line(x_right_spine_limit, end_point_down_right_y, x_right_spine_limit, middle_point_down_right_y)
260 c.line(x_right_spine_limit, middle_point_down_right_y, start_down_right_right_x, half_height)
262 if args.analyze or front_page:
264 new_pdf = pypdf.PdfReader(packet)
265 new_page.merge_page(new_pdf.pages[0])
266 writer.add_page(new_page)
268 front_page = not front_page
269 for file in opened_files:
272 with open(args.output_file, 'wb') as output_file:
273 writer.write(output_file)