fix: Use fault names if present before labelling fault Fault_XXX

Loop3D · May 31, 2023 · 7bf34fc · 7bf34fc
1 parent ed61243
commit 7bf34fc
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/LoopStructural/modelling/input/project_file.py b/LoopStructural/modelling/input/project_file.py
@@ -29,12 +29,12 @@ def __init__(self, projectfile, use_thickness=None):
         contacts.rename(columns=column_map, inplace=True)
         fault_locations.rename(columns=column_map, inplace=True)
         fault_orientations.rename(columns=column_map, inplace=True)
-        fault_locations["fault_name"] = [
-            f"Fault_{eventid}" for eventid in fault_locations["eventId"]
-        ]
-        fault_orientations["fault_name"] = [
-            f"Fault_{eventid}" for eventid in fault_orientations["eventId"]
-        ]
+        # fault_locations["fault_name"] = [
+        #     f"Fault_{eventid}" for eventid in fault_locations["eventId"]
+        # ]
+        # fault_orientations["fault_name"] = [
+        #     f"Fault_{eventid}" for eventid in fault_orientations["eventId"]
+        # ]
         thicknesses = dict(
             zip(
                 projectfile["stratigraphicLog"].name,
@@ -52,6 +52,10 @@ def __init__(self, projectfile, use_thickness=None):
             },
             inplace=True,
         )
+        fault_locations = fault_properties.reset_index()[["name", "eventId"]].merge(fault_locations, on="eventId")
+        fault_locations["fault_name"] = fault_locations["name"]
+        fault_orientations = fault_properties.reset_index()[["name", "eventId"]].merge(fault_orientations, on="eventId")
+        fault_orientations["fault_name"] = fault_orientations["name"]
         colours = dict(
             zip(
                 self.projectfile.stratigraphicLog.name,