| |
| import sys |
| import os |
| import _paths |
| |
| OUTPUT_FORMATS = "pdf png dot".split() |
| PDF_FORMAT, PNG_FORMAT, DOT_FORMAT = OUTPUT_FORMATS |
| |
| |
| def handle_options(): |
| import argparse |
| parser = argparse.ArgumentParser(description="Opens the TFPG Viewer") |
| |
| parser.add_argument("--tfpg-file", "-t", |
| dest="tfpg_fname", |
| default=None, |
| type=str, |
| help="The input TFPG file name") |
| |
| parser.add_argument("--output-file", "-o", |
| dest="out_fname", |
| default=None, |
| type=str, |
| help="The name of the output file used to dump the TFPG") |
| |
| parser.add_argument("--output-format", "-f", |
| dest="output_format", |
| choices=OUTPUT_FORMATS, |
| default=PDF_FORMAT, |
| help="Output format (default: %(default)s)") |
| |
| args = parser.parse_args() |
| |
| if args.tfpg_fname is not None and not os.path.isfile(args.tfpg_fname): |
| parser.error("File not found: %s" % args.tfpg_fname) |
| |
| return args |
| |
| |
| def main(tfpg_fname, out_fname, output_format): |
| from tfpg_viewer import tfpgv |
| tfpgv.run(tfpg_fname, out_fname, output_format) |
| return |
| |
| |
| if __name__ == "__main__": |
| _paths.setup_path() |
| args = handle_options() |
| main(args.tfpg_fname, args.out_fname, args.output_format) |