fix: put fault normal points on the trace

Loop3D · Apr 29, 2024 · 2391b30 · 2391b30
1 parent 5164c47
commit 2391b30
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/LoopStructural/modelling/features/builders/_fault_builder.py b/LoopStructural/modelling/features/builders/_fault_builder.py
@@ -192,7 +192,12 @@ def create_data_from_geometry(
                 fault_frame_data["coord"] == 0, fault_frame_data["val"] == 0
             )
             fault_center = fault_frame_data.loc[trace_mask, ["X", "Y", "Z"]].mean(axis=0).to_numpy()
-
+            dist = np.linalg.norm(
+                fault_center - fault_frame_data.loc[trace_mask, ["X", "Y", "Z"]].to_numpy(), axis=1
+            )
+            fault_center = fault_frame_data.loc[trace_mask, ["X", "Y", "Z"]].to_numpy()[
+                np.argmin(dist), :
+            ]
         self.fault_normal_vector = fault_normal_vector
         self.fault_slip_vector = fault_slip_vector