Use methods from _util to get latest detector model files

Harvard-Neutrino · Jan 24, 2024 · 8ae6db9 · 8ae6db9
1 parent 224103f
commit 8ae6db9
Showing 1 changed file with 2 additions and 15 deletions.
diff --git a/python/LIController.py b/python/LIController.py
@@ -40,21 +40,8 @@ def __init__(self, events_to_inject, experiment, seed=0):
         self.events = []
 
         # Find the density and materials files
-        materials_file = os.path.join(
-            self.resources_dir, "DetectorParams", "materials", f"{experiment}.dat"
-        )
-        if experiment in ["ATLAS", "dune", "IceCube"]:
-            detector_model_file = os.path.join(
-                self.resources_dir,
-                "DetectorParams",
-                "densities",
-                f"PREM_{experiment}.dat",
-            )
-        else:
-            detector_model_file = os.path.join(
-                self.resources_dir, "DetectorParams", "densities", f"{experiment}.dat"
-            )
-
+        materials_file = _util.get_material_model_path(experiment)
+        detector_model_file = _util.get_detector_model_path(experiment)
         print(detector_model_file)
 
         self.detector_model = _detector.DetectorModel()