import argparse

from pg_rad.logger import setup_logger
from pg_rad.landscape import LandscapeDirector
from pg_rad.plotting import LandscapeSlicePlotter


def main():
    parser = argparse.ArgumentParser(
        prog="pg-rad",
        description="Primary Gamma RADiation landscape tool"
    )

    parser.add_argument(
        "--test",
        action="store_true",
        help="Load and run the test landscape"
    )
    parser.add_argument(
        "--loglevel",
        default="INFO",
        choices=["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"],
    )
    parser.add_argument(
        "--saveplot",
        action="store_true",
        help="Save the plot or not."
    )

    args = parser.parse_args()
    setup_logger(args.loglevel)

    if args.test:
        landscape = LandscapeDirector().build_test_landscape()
        plotter = LandscapeSlicePlotter()
        plotter.plot(landscape, save=args.saveplot)


if __name__ == "__main__":
    main()