cb38a31627f152c22fb9e474c31b34da588e1134,starstack.py,,,#,192

Before Change

    if len(sys.argv) < 3:
        rgb1, rgb2 = make_test_images()
    else:
        rgb1 = read_image(sys.argv[1])
        rgb2 = read_image(sys.argv[2])

    register(rgb1, rgb2)

After Change

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Stack astronomical images")
    parser.add_argument("-t", "--test", dest="test", default=False,
                        action="store_true",
                help="Test mode: generate images instead of reading files")
    parser.add_argument("-d", action="store", default=".", dest="dir",
                        help="Directory to save files (default: .)")
    parser.add_argument("imagefiles", nargs="*", help="2 or more input images")
    args = parser.parse_args(sys.argv[1:])

    if args.test:
        register_all(make_test_images(), outdir=args.dir)
        sys.exit(0)

    if len(args.imagefiles) < 2:
        parser.print_help()
        sys.exit(1)

    register_all(args.imagefiles, outdir=args.dir)

Instances