diff --git a/src/pg_rad/dataloader/dataloader.py b/src/pg_rad/dataloader/dataloader.py index fd0a28a..dfe94a6 100644 --- a/src/pg_rad/dataloader/dataloader.py +++ b/src/pg_rad/dataloader/dataloader.py @@ -2,7 +2,7 @@ import logging import pandas as pd -from pg_rad.exceptions import DataLoadError, InvalidCSVError +from pg_rad.exceptions.exceptions import DataLoadError, InvalidCSVError logger = logging.getLogger(__name__) diff --git a/src/pg_rad/isotopes/isotope.py b/src/pg_rad/isotopes/isotope.py index 78ac54e..5706acd 100644 --- a/src/pg_rad/isotopes/isotope.py +++ b/src/pg_rad/isotopes/isotope.py @@ -1,4 +1,4 @@ -from pg_rad.physics import get_mass_attenuation_coeff +from pg_rad.physics.attenuation import get_mass_attenuation_coeff class Isotope: diff --git a/src/pg_rad/landscape/director.py b/src/pg_rad/landscape/director.py index edddd7f..4484092 100644 --- a/src/pg_rad/landscape/director.py +++ b/src/pg_rad/landscape/director.py @@ -2,9 +2,9 @@ from importlib.resources import files import logging from pg_rad.configs.filepaths import TEST_EXP_DATA -from pg_rad.isotopes import CS137 +from pg_rad.isotopes.presets import CS137 from pg_rad.landscape.landscape import LandscapeBuilder -from pg_rad.objects import PointSource +from pg_rad.objects.sources import PointSource class LandscapeDirector: diff --git a/src/pg_rad/landscape/landscape.py b/src/pg_rad/landscape/landscape.py index 805081e..3af4733 100644 --- a/src/pg_rad/landscape/landscape.py +++ b/src/pg_rad/landscape/landscape.py @@ -1,11 +1,10 @@ import logging from typing import Self -from pg_rad.dataloader import load_data -from pg_rad.exceptions import OutOfBoundsError -from pg_rad.objects import PointSource -from pg_rad.path import Path, path_from_RT90 -from pg_rad.physics.fluence import phi_single_source +from pg_rad.dataloader.dataloader import load_data +from pg_rad.exceptions.exceptions import OutOfBoundsError +from pg_rad.objects.sources import PointSource +from pg_rad.path.path import Path, path_from_RT90 logger = logging.getLogger(__name__) @@ -44,23 +43,6 @@ class Landscape: logger.debug(f"Landscape created: {self.name}") - def calculate_fluence_at(self, pos: tuple): - total_phi = 0. - for source in self.point_sources: - r = source.distance_to(pos) - phi_source = phi_single_source( - r=r, - activity=source.activity, - branching_ratio=source.isotope.b, - mu_mass_air=source.isotope.mu_mass_air, - air_density=self.air_density - ) - total_phi += phi_source - return total_phi - - def calculate_fluence_along_path(self): - pass - class LandscapeBuilder: def __init__(self, name: str = "Unnamed landscape"): diff --git a/src/pg_rad/objects/sources.py b/src/pg_rad/objects/sources.py index 9361f1f..7659704 100644 --- a/src/pg_rad/objects/sources.py +++ b/src/pg_rad/objects/sources.py @@ -1,7 +1,7 @@ import logging from .objects import BaseObject -from pg_rad.isotopes import Isotope +from pg_rad.isotopes.isotope import Isotope logger = logging.getLogger(__name__) diff --git a/src/pg_rad/plotting/landscape_plotter.py b/src/pg_rad/plotting/landscape_plotter.py index d54ca9a..9c6b589 100644 --- a/src/pg_rad/plotting/landscape_plotter.py +++ b/src/pg_rad/plotting/landscape_plotter.py @@ -3,7 +3,7 @@ import logging from matplotlib import pyplot as plt from matplotlib.patches import Circle -from pg_rad.landscape import Landscape +from pg_rad.landscape.landscape import Landscape logger = logging.getLogger(__name__)