mirror of
https://github.com/pim-n/pg-rad
synced 2026-03-23 21:58:12 +01:00
41 lines
964 B
Python
41 lines
964 B
Python
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()
|