pim/pg-rad
1
0
Fork 0
mirror of https://github.com/pim-n/pg-rad synced 2026-03-11 19:58:11 +01:00
Code Issues Packages Projects Releases Wiki Activity

add plotting to test case. Add saveplot argument. Improve logging in LandscapeSlicePlotter

Browse Source
This commit is contained in:
Pim Nelissen
2026-02-17 09:46:30 +01:00
parent 3fd2eafb2a
commit 5684525d0f
2 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/pg_rad/main.py
View File

@ -2,6 +2,7 @@ import argparse
from pg_rad.logger import setup_logger from pg_rad.logger import setup_logger
from pg_rad.landscape import LandscapeDirector from pg_rad.landscape import LandscapeDirector
from pg_rad.plotting import LandscapeSlicePlotter
def main(): def main():
@ -20,13 +21,19 @@ def main():
default="INFO", default="INFO",
choices=["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"], choices=["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"],
) )
parser.add_argument(
"--saveplot",
action="store_true",
help="Save the plot or not."
)
args = parser.parse_args() args = parser.parse_args()
setup_logger(args.loglevel) setup_logger(args.loglevel)
if args.test: if args.test:
landscape = LandscapeDirector().build_test_landscape() landscape = LandscapeDirector().build_test_landscape()
print(landscape.name) plotter = LandscapeSlicePlotter()
plotter.plot(landscape, save=args.saveplot)
if __name__ == "__main__": if __name__ == "__main__":

8
src/pg_rad/plotting/landscape_plotter.py
View File

@ -36,10 +36,10 @@ class LandscapeSlicePlotter:
ax.set_aspect("equal") ax.set_aspect("equal")
if save: if save:
name = landscape.name.lower().replace(' ', '_') landscape_name = landscape.name.lower().replace(' ', '_')
plt.savefig( filename = f"{landscape_name}_z{self.z}.png"
f"{name}_z{self.z}.png" plt.savefig(filename)
) logger.info("Plot saved to file: "+filename)
if show: if show:
plt.show() plt.show()