[interpolatable] Merge PS/PDF code path

This commit is contained in:
Behdad Esfahbod 2023-12-13 19:25:29 -07:00
parent aad3707227
commit c13463b6da

View File

@ -1084,33 +1084,24 @@ def main(args=None):
problems = sort_problems(problems) problems = sort_problems(problems)
if args.pdf: for p in "ps", "pdf":
log.info("Writing PDF to %s", args.pdf) arg = getattr(args, p)
from .interpolatablePlot import InterpolatablePDF if arg is None:
continue
log.info("Writing %s to %s", p.upper(), arg)
from .interpolatablePlot import InterpolatablePS, InterpolatablePDF
with InterpolatablePDF( PlotterClass = InterpolatablePS if p == "ps" else InterpolatablePDF
ensure_parent_dir(args.pdf), glyphsets=glyphsets, names=names
) as pdf: with PlotterClass(
pdf.add_title_page( ensure_parent_dir(arg), glyphsets=glyphsets, names=names
) as doc:
doc.add_title_page(
original_args_inputs, tolerance=tolerance, kinkiness=kinkiness original_args_inputs, tolerance=tolerance, kinkiness=kinkiness
) )
pdf.add_problems(problems) doc.add_problems(problems)
if not problems and not args.quiet: if not problems and not args.quiet:
pdf.draw_cupcake() doc.draw_cupcake()
if args.ps:
log.info("Writing PS to %s", args.ps)
from .interpolatablePlot import InterpolatablePS
with InterpolatablePS(
ensure_parent_dir(args.ps), glyphsets=glyphsets, names=names
) as ps:
ps.add_title_page(
original_args_inputs, tolerance=tolerance, kinkiness=kinkiness
)
ps.add_problems(problems)
if not problems and not args.quiet:
ps.draw_cupcake()
if args.html: if args.html:
log.info("Writing HTML to %s", args.html) log.info("Writing HTML to %s", args.html)