diff --git a/foliator.py b/foliator.py index 9c09a75..d380859 100755 --- a/foliator.py +++ b/foliator.py @@ -138,9 +138,15 @@ parser.add_argument("-a5", help="output as a5, reverse on short edge", action="store_true") parser.add_argument("-fs", help="maximum folio size, in sheets", action="store", default=-1, type=int) +parser.add_argument("--strategy", help="greedy packing of pages into folios", + action="store", default="balanced") + +stratnames_dict = {"balanced": find_partition_mink, "greedy": find_partition_greedy} + args = parser.parse_args() input_pdf = PdfReader(args.infile) +strategy = stratnames_dict[args.strategy] a5 = foliate_A6(input_pdf, find_partition_greedy, args.fs) if (args.a5): a5.write(args.outfile)