mirror of
https://github.com/pim-n/pg-rad
synced 2026-03-22 21:48:11 +01:00
81 lines
1.2 KiB
Python
81 lines
1.2 KiB
Python
from abc import ABC
|
|
from dataclasses import dataclass
|
|
|
|
|
|
@dataclass
|
|
class MetadataSpec:
|
|
name: str
|
|
|
|
|
|
@dataclass
|
|
class RuntimeSpec:
|
|
speed: float
|
|
acquisition_time: float
|
|
|
|
|
|
@dataclass
|
|
class SimulationOptionsSpec:
|
|
air_density: float
|
|
seed: int | None = None
|
|
|
|
|
|
@dataclass
|
|
class PathSpec(ABC):
|
|
pass
|
|
|
|
|
|
@dataclass
|
|
class ProceduralPathSpec(PathSpec):
|
|
segments: list[str]
|
|
angles: list[float]
|
|
lengths: list[int | None]
|
|
z: int | float
|
|
alpha: float
|
|
|
|
|
|
@dataclass
|
|
class CSVPathSpec(PathSpec):
|
|
file: str
|
|
east_col_name: str
|
|
north_col_name: str
|
|
z: int | float
|
|
|
|
|
|
@dataclass
|
|
class SourceSpec(ABC):
|
|
activity_MBq: float
|
|
isotope: str
|
|
name: str
|
|
|
|
|
|
@dataclass
|
|
class AbsolutePointSourceSpec(SourceSpec):
|
|
x: float
|
|
y: float
|
|
z: float
|
|
|
|
|
|
@dataclass
|
|
class RelativePointSourceSpec(SourceSpec):
|
|
along_path: float
|
|
dist_from_path: float
|
|
side: str
|
|
z: float
|
|
|
|
|
|
@dataclass
|
|
class DetectorSpec:
|
|
name: str
|
|
eff: float | None
|
|
is_isotropic: bool
|
|
|
|
|
|
@dataclass
|
|
class SimulationSpec:
|
|
metadata: MetadataSpec
|
|
runtime: RuntimeSpec
|
|
options: SimulationOptionsSpec
|
|
path: PathSpec
|
|
point_sources: list[SourceSpec]
|
|
detector: DetectorSpec
|