diff --git a/src/pg_rad/simulator/engine.py b/src/pg_rad/simulator/engine.py index 7562206..7b8c950 100644 --- a/src/pg_rad/simulator/engine.py +++ b/src/pg_rad/simulator/engine.py @@ -48,6 +48,8 @@ class SimulationEngine: ) return CountRateOutput( + self.landscape.path.x_list, + self.landscape.path.y_list, acq_points, sub_points, cps, diff --git a/src/pg_rad/simulator/outputs.py b/src/pg_rad/simulator/outputs.py index 89c597c..9aca77e 100644 --- a/src/pg_rad/simulator/outputs.py +++ b/src/pg_rad/simulator/outputs.py @@ -5,7 +5,9 @@ from dataclasses import dataclass @dataclass class CountRateOutput: - acquisition_points: List[float] + x: List[float] + y: List[float] + distance: List[float] sub_points: List[float] cps: List[float] integrated_counts: List[float]