Ejemplo n.º 1
0
def parse_options(argv, cfg):
    """ set config dictionary according to command line user arguments """

    try:
        opts, args = getopt.getopt(argv, "ghf:c:p:r:m:",
                                        ["gui", "help", "file=", "cali=", "patch=", "refo=", "meth=",
                                         "dbug", "refi", "pflu", "rota",
                                         "vgn", "awb", "con", "hot", "sat", "art", "rm"])
    except getopt.GetoptError as e:
        print(e)
        sys.exit(2)

    if opts:
        for (opt, arg) in opts:
            if opt in ("-g", "--gui"):
                PlenopticamApp().mainloop()
                sys.exit()
            if opt in ("-h", "--help"):
                usage()
                sys.exit()
            if opt in ("-f", "--file"):
                cfg.params[cfg.lfp_path] = arg.strip(" \"\'")
            if opt in ("-c", "--cali"):
                cfg.params[cfg.cal_path] = arg.strip(" \"\'")
            if opt in ("-p", "--patch"):
                cfg.params[cfg.ptc_leng] = misc.str2type(arg)
            if opt in ("-r", "--refo"):
                refo_range = misc.str2list(arg)
                cfg.params[cfg.ran_refo] = refo_range if isinstance(refo_range, list) else [0, 2]
            if opt in ("-m", "--meth"):
                cfg.params[cfg.cal_meth] = arg.strip(" \"\'") if arg.strip(" \"\'") in CALI_METH else CALI_METH[0]
            if opt == "--dbug":
                cfg.params[cfg.opt_dbug] = True
            if opt == "--refi":
                cfg.params[cfg.opt_refi] = True
            if opt == "--pflu":
                cfg.params[cfg.opt_pflu] = True
            if opt == "--rota":
                cfg.params[cfg.opt_rota] = True
            if opt == "--vgn":
                cfg.params[cfg.opt_vign] = True
            if opt == "--awb":
                cfg.params[cfg.opt_awb_] = True
            if opt == "--con":
                cfg.params[cfg.opt_cont] = True
            if opt == "--hot":
                cfg.params[cfg.opt_lier] = True
            if opt == "--sat":
                cfg.params[cfg.opt_sat_] = True
            if opt == "--art":
                cfg.params[cfg.opt_arti] = True
            if opt == "--rm":
                cfg.params[cfg.dir_remo] = True

    return cfg
Ejemplo n.º 2
0
def parse_options(argv, cfg):

    try:
        opts, args = getopt.getopt(argv, "ghf:c:p:r:", [
            "gui", "help", "file=", "cali=", "patch=", "refo=", "dbug", "refi",
            "vgn", "awb", "con", "hot", "sat", "art", "rm"
        ])
    except getopt.GetoptError as e:
        print(e)
        sys.exit(2)

    if opts:
        for (opt, arg) in opts:
            if opt in ("-g", "--gui"):
                PlenopticamApp(None).mainloop()
                sys.exit()
            if opt in ("-h", "--help"):
                usage()
                sys.exit()
            if opt in ("-f", "--file"):
                cfg.params[cfg.lfp_path] = arg
            if opt in ("-c", "--cali"):
                cfg.params[cfg.cal_path] = arg
            if opt in ("-p", "--patch"):
                cfg.params[cfg.ptc_leng] = misc.str2type(arg)
            if opt in ("-r", "--refo"):
                refo_range = misc.str2list(arg)
                cfg.params[cfg.ran_refo] = refo_range if isinstance(
                    refo_range, list) else [0, 2]
            if opt == "--dbug":
                cfg.params[cfg.opt_dbug] = True
            if opt == "--refi":
                cfg.params[cfg.opt_refi] = True
            if opt == "--vgn":
                cfg.params[cfg.opt_vign] = True
            if opt == "--awb":
                cfg.params[cfg.opt_awb_] = True
            if opt == "--con":
                cfg.params[cfg.opt_cont] = True
            if opt == "--hot":
                cfg.params[cfg.opt_lier] = True
            if opt == "--sat":
                cfg.params[cfg.opt_sat_] = True
            if opt == "--art":
                cfg.params[cfg.opt_arti] = True
            if opt == "--rm":
                cfg.params[cfg.dir_remo] = True
    # create dictionary containing all parameters for the light field
    return cfg