fix: add fault pitch

Author: lachlangrose

LoopStructural/modelling/core/geological_model.py

@@ -1293,6 +1293,7 @@ def create_and_add_fault(
         fault_trace_anisotropy=0.0,
         fault_dip=90,
         fault_dip_anisotropy=0.0,
+        fault_pitch=None,
         **kwargs,
     ):
         """
@@ -1393,6 +1394,7 @@ def create_and_add_fault(
             fault_trace_anisotropy=fault_trace_anisotropy,
             fault_dip=fault_dip,
             fault_dip_anisotropy=fault_dip_anisotropy,
+            fault_pitch=fault_pitch,
         )
         if "force_mesh_geometry" not in kwargs:
             fault_frame_builder.set_mesh_geometry(kwargs.get("fault_buffer", 0.2), 0)

LoopStructural/modelling/features/builders/_fault_builder.py

@@ -91,6 +91,7 @@ def create_data_from_geometry(
         fault_trace_anisotropy=1.0,
         fault_dip=90,
         fault_dip_anisotropy=1.0,
+        fault_pitch=None,
     ):
         """Generate the required data for building a fault frame for a fault with the
         specified parameters
@@ -204,13 +205,19 @@ def create_data_from_geometry(
                 fault_frame_data["coord"] == 1, ~np.isnan(fault_frame_data["gz"])
             )
             fault_slip_data = fault_frame_data.loc[slip_mask, ["gx", "gy", "gz"]]
+
             if len(fault_slip_data) == 0:
                 logger.warning(
                     "There is no slip vector data for the fault, using vertical slip vector\n\
                           projected onto fault surface estimating from fault normal"
                 )
                 strike_vector, dip_vector = get_vectors(fault_normal_vector[None, :])
                 fault_slip_vector = dip_vector[:, 0]
+                if fault_pitch is not None:
+                    print('using pitch')
+                    rotm = rotation(fault_normal_vector[None,:],[fault_pitch])
+                    print(rotm.shape,fault_slip_vector.shape)
+                    fault_slip_vector = np.einsum("ijk,k->ij", rotm, fault_slip_vector)[0,:]
                 logger.info(f"Estimated fault slip vector: {fault_slip_vector}")
             else:
                 fault_slip_vector = fault_slip_data.mean(axis=0).to_numpy()