[WIP] Caching pivot in Transforms when makes sense

Author: Leguark

gempy/core/data/geo_model.py

@@ -157,6 +157,7 @@ def solutions(self, value):
 
             # TODO: These meshes are in the order of the scalar field 
             world_coord_vertices = self.input_transform.apply_inverse(dc_mesh.vertices)
+            world_coord_vertices = self.grid.transform.apply_inverse_with_cached_pivot(world_coord_vertices)
 
             element.vertices = world_coord_vertices
             element.edges = (dc_mesh.edges if dc_mesh is not None else None)
@@ -176,11 +177,8 @@ def surface_points_copy(self):
     @property
     def surface_points_copy_transformed(self) -> SurfacePointsTable:
         og_sp = self.surface_points_copy
-        og_sp.xyz_view = self.grid.transform.apply_with_pivot(
-            points=og_sp.xyz,
-            pivot=self.grid.corner_min
-        )
-
+        
+        og_sp.xyz_view = self.grid.transform.apply_with_cached_pivot(og_sp.xyz)
         og_sp.xyz_view = self.input_transform.apply(og_sp.xyz)
         return og_sp
 
@@ -205,13 +203,9 @@ def orientations_copy_transformed(self) -> OrientationsTable:
         og_or = self.orientations_copy
         total_transform: Transform = self.input_transform + self.grid.transform
 
-        og_or.xyz_view = self.grid.transform.apply_with_pivot(
-            points=og_or.xyz,
-            pivot=self.grid.corner_min
-        )
-
+        og_or.xyz_view = self.grid.transform.apply_with_cached_pivot(og_or.xyz)
         og_or.xyz_view = self.input_transform.apply(og_or.xyz)
-        # og_or.xyz_view = total_transform.apply(og_or.xyz)
+        
         og_or.grads_view = total_transform.transform_gradient(og_or.grads)
         return og_or
 
@@ -244,7 +238,7 @@ def extent(self) -> np.ndarray:
     def interpolation_input_copy(self):
         warnings.warn("This property is deprecated. Use directly "
                       "`interpolation_input_from_structural_frame` instead.", DeprecationWarning)
-        
+
         if self.structural_frame.is_dirty is False:
             return self._interpolationInput
 

gempy/core/data/grid_modules/grid_types.py

@@ -127,21 +127,21 @@ def _calculate_rotated_box_val(v1, v2):
         ])
 
         inverted_rotation_matrix = np.linalg.inv(rotation_matrix)
-        transform = Transform.from_matrix(inverted_rotation_matrix)
-
         # Calculate the extents in the new coordinate system
         extent_x = np.linalg.norm(v1)
         extent_y = np.linalg.norm(v2)
 
         # We transform the origin point1 to the new coordinates
-        # [[ 5.47925650e+06  5.70152895e+06 -2.39200000e+02]]
         origin_ = [x1, y1, zmin]
 
         xmin, ymin, zmin = origin_
         xmax, ymax, zmax = xmin + extent_x, ymin + extent_y, zmin + zmax - zmin
 
         extent = np.array([xmin, xmax, ymin, ymax, zmin, zmax], dtype='float64')
 
+
+        transform = Transform.from_matrix(inverted_rotation_matrix)
+        transform.cached_pivot = origin_
         grid = cls(extent=extent, resolution=resolution, transform=transform)
 
         if plot:

test/test_modules/.benchmarks/test_transformed_space.py

@@ -47,7 +47,7 @@ def test_plot_transformed_data_including_grid_transform():
         distance_point3=1000,
         zmin=model.extent[4],
         zmax=model.extent[5],
-        resolution=np.array([50, 50, 50]),
+        resolution=np.array([20, 20, 20]),
         plot=True
     )
 
@@ -60,12 +60,25 @@ def test_plot_transformed_data_including_grid_transform():
     if PLOT:
         gpv = require_gempy_viewer()
 
+        
         gpv.plot_3d(
             model,
             image=False,
             transformed_data=True,
             show_boundaries=True,
-            show_lith=True,
+            show_lith=False,
+            kwargs_plot_data={
+                    'arrow_size': .01
+            }
+        )
+
+
+        gpv.plot_3d(
+            model,
+            image=False,
+            transformed_data=False,
+            show_boundaries=True,
+            show_lith=False,
             kwargs_plot_data={
                     'arrow_size': .01
             }