main.py

import argparse
from src.visualizer import Visualizer

parser = argparse.ArgumentParser(description='Run the program')
parser.add_argument('fx', metavar='fx', type=str, help='X Function')
parser.add_argument('fy', metavar='fy', type=str, help='Y Function')
parser.add_argument('--mode', metavar='mode', type=str,
                    help="Specify 'color' or 'black' to choose whether to use actual length or color", default="color",
                    nargs='?')
parser.add_argument('--skip', metavar='skip', type=int, help='Number of units to skip before each vector', default=1,
                    nargs='?')
parser.add_argument('--bound', metavar='bound', type=str,
                    help='Number of values to show. Eg. -10,10', nargs='?', default="-10,10")
parser.add_argument('--prop', metavar='prop', type=int, help='Set this value to change the cutoff for changing color',
                    default=0, nargs='?')

args = parser.parse_args()
if args.mode.upper() == "COLOR":
    v = Visualizer(f_x=args.fx, f_y=args.fy)
    v.plot_color(bound=tuple(map(int, args.bound.split(','))), skip=args.skip, prop=args.prop)
elif args.mode.upper() == "BLACK":
    v = Visualizer(f_x=args.fx, f_y=args.fy)
    v.plot(bound=tuple(map(int, args.boundx.split(','))), skip=args.skip,)