From 8ae6db9056b812a24a3cc598a87a41bfd2ff563f Mon Sep 17 00:00:00 2001
From: Austin Schneider <hogenshpogen@gmail.com>
Date: Tue, 23 Jan 2024 18:32:40 -0700
Subject: [PATCH] Use methods from _util to get latest detector model files

---
 python/LIController.py | 17 ++---------------
 1 file changed, 2 insertions(+), 15 deletions(-)

diff --git a/python/LIController.py b/python/LIController.py
index c79fb238..51540c39 100644
--- 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()