##

import argparse
from plotWheels.helical_wheel import helical_wheel

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Generate Helical Wheel")
    parser.add_argument("--sequence", dest="sequence", type=str)
    parser.add_argument("--seqRange", dest="seqRange", type=int, default=1)
    parser.add_argument("--t_size", dest="t_size", type=int, default=32)
    parser.add_argument("--rotation", dest="rotation", type=int, default=90)
    parser.add_argument(
        "--numbering", action="store_true", help="numbering for helical wheel"
    )
    parser.add_argument(
        "--output",
        dest="output",
        type=argparse.FileType("wb"),
        default="_helicalwheel.png",
    )  # dest="output",default="_helicalwheel.png")
    #### circle colors
    parser.add_argument("--f_A", dest="f_A", default="#ffcc33")
    parser.add_argument("--f_C", dest="f_C", default="#b5b5b5")
    parser.add_argument("--f_D", dest="f_D", default="#db270f")
    parser.add_argument("--f_E", dest="f_E", default="#db270f")
    parser.add_argument("--f_F", dest="f_F", default="#ffcc33")
    parser.add_argument("--f_G", dest="f_G", default="#b5b5b5")
    parser.add_argument("--f_H", dest="f_H", default="#12d5fc")
    parser.add_argument("--f_I", dest="f_I", default="#ffcc33")
    parser.add_argument("--f_K", dest="f_K", default="#12d5fc")
    parser.add_argument("--f_L", dest="f_L", default="#ffcc33")
    parser.add_argument("--f_M", dest="f_M", default="#ffcc33")
    parser.add_argument("--f_N", dest="f_N", default="#b5b5b5")
    parser.add_argument("--f_P", dest="f_P", default="#ffcc33")
    parser.add_argument("--f_Q", dest="f_Q", default="#b5b5b5")
    parser.add_argument("--f_R", dest="f_R", default="#12d5fc")
    parser.add_argument("--f_S", dest="f_S", default="#b5b5b5")
    parser.add_argument("--f_T", dest="f_T", default="#b5b5b5")
    parser.add_argument("--f_V", dest="f_V", default="#ffcc33")
    parser.add_argument("--f_W", dest="f_W", default="#ffcc33")
    parser.add_argument("--f_Y", dest="f_Y", default="#b5b5b5")
    ### text colors
    parser.add_argument("--t_A", dest="t_A", default="k")
    parser.add_argument("--t_C", dest="t_C", default="k")
    parser.add_argument("--t_D", dest="t_D", default="w")
    parser.add_argument("--t_E", dest="t_E", default="w")
    parser.add_argument("--t_F", dest="t_F", default="k")
    parser.add_argument("--t_G", dest="t_G", default="k")
    parser.add_argument("--t_H", dest="t_H", default="k")
    parser.add_argument("--t_I", dest="t_I", default="k")
    parser.add_argument("--t_K", dest="t_K", default="k")
    parser.add_argument("--t_L", dest="t_L", default="k")
    parser.add_argument("--t_M", dest="t_M", default="k")
    parser.add_argument("--t_N", dest="t_N", default="k")
    parser.add_argument("--t_P", dest="t_P", default="k")
    parser.add_argument("--t_Q", dest="t_Q", default="k")
    parser.add_argument("--t_R", dest="t_R", default="k")
    parser.add_argument("--t_S", dest="t_S", default="k")
    parser.add_argument("--t_T", dest="t_T", default="k")
    parser.add_argument("--t_V", dest="t_V", default="k")
    parser.add_argument("--t_W", dest="t_W", default="k")
    parser.add_argument("--t_Y", dest="t_Y", default="k")

    args = parser.parse_args()

    # print(type(args.output))

    f_colors = [
        args.f_A,
        args.f_C,
        args.f_D,
        args.f_E,
        args.f_F,
        args.f_G,
        args.f_H,
        args.f_I,
        args.f_K,
        args.f_L,
        args.f_M,
        args.f_N,
        args.f_P,
        args.f_Q,
        args.f_R,
        args.f_S,
        args.f_T,
        args.f_V,
        args.f_W,
        args.f_Y,
    ]

    t_colors = [
        args.t_A,
        args.t_C,
        args.t_D,
        args.t_E,
        args.t_F,
        args.t_G,
        args.t_H,
        args.t_I,
        args.t_K,
        args.t_L,
        args.t_M,
        args.t_N,
        args.t_P,
        args.t_Q,
        args.t_R,
        args.t_S,
        args.t_T,
        args.t_V,
        args.t_W,
        args.t_Y,
    ]

    colors = [f_colors, t_colors]

    tmp_file = "./tmp.png"

    helical_wheel(
        sequence=args.sequence,
        colorcoding=colors[0],
        text_color=colors[1],
        seqRange=args.seqRange,
        t_size=args.t_size,
        rot=args.rotation,
        numbering=args.numbering,
        filename=tmp_file,
    )

    with open("tmp.png", "rb") as f:
        for line in f:
            args.output.write(line)