[!ENH] All content can be plotted transformed or not

Author: Leguark

gempy/core/data/geo_model.py

@@ -208,6 +208,16 @@ def orientations_copy_transformed(self) -> OrientationsTable:
 
         og_or.grads_view = total_transform.transform_gradient(og_or.grads)
         return og_or
+    
+    @property
+    def regular_grid_coordinates(self) -> np.ndarray:
+        return self.grid.regular_grid.get_values_vtk_format(orthogonal=False)
+    
+    @property
+    def regular_grid_coordinates_transformed(self) -> np.ndarray:
+        values_transformed = self.grid.regular_grid.get_values_vtk_format(orthogonal=True)
+        values_transformed = self.input_transform.apply(values_transformed)
+        return values_transformed
 
     @property
     def orientations(self) -> OrientationsTable:
@@ -230,10 +240,6 @@ def extent_transformed(self) -> np.ndarray:
                                 transformed[:, 2].min(), transformed[:, 2].max()])
         return new_extents
 
-    @property
-    def extent(self) -> np.ndarray:
-        return self.grid.extent
-
     @property
     def interpolation_input_copy(self):
         warnings.warn("This property is deprecated. Use directly "

gempy/core/data/grid_modules/grid_types.py

@@ -199,7 +199,10 @@ def dz(self):
         return (self.extent[5] - self.extent[4]) / self.resolution[2]
 
     @property
-    def values_vtk_format(self):
+    def values_vtk_format(self) -> np.ndarray:
+        return self.get_values_vtk_format()
+    
+    def get_values_vtk_format(self, orthogonal: bool = False) -> np.ndarray:
         extent = self.extent
         resolution = self.resolution + 1
 
@@ -209,6 +212,14 @@ def values_vtk_format(self):
         xv, yv, zv = np.meshgrid(x, y, z, indexing="ij")
         g = np.vstack((xv.ravel(), yv.ravel(), zv.ravel())).T
 
+
+        # Transform the values
+        if self.transform is not None and orthogonal is False:
+            g = self.transform.apply_inverse_with_pivot(
+                points=g,
+                pivot=np.array([self.extent[0], self.extent[2], self.extent[4]])
+            )
+
         return g
 
     @staticmethod

test/test_modules/.benchmarks/test_transformed_space.py

@@ -24,7 +24,7 @@ def test_plot_transformed_data_only_transform_input():
                     'arrow_size': 10
             }
         )
-        
+
         gpv.plot_3d(
             model,
             image=False,
@@ -35,81 +35,48 @@ def test_plot_transformed_data_only_transform_input():
                     'arrow_size': .01
             }
         )
-        
-        
+
+
 def test_plot_transformed_data_including_grid_transform():
     model = gp.generate_example_model(ExampleModel.ANTICLINE, compute_model=False)
 
     # Calculate point_y_axis
     regular_grid = gp.data.grid.RegularGrid.from_corners_box(
         pivot=(200, 200),
-        point_x_axis=(800,800),
+        point_x_axis=(800, 800),
         distance_point3=1000,
-        zmin=model.extent[4],
-        zmax=model.extent[5],
+        zmin=model.grid.extent[4],
+        zmax=model.grid.extent[5],
         resolution=np.array([20, 20, 20]),
         plot=True
     )
 
     model.grid = gp.data.grid.Grid()
     model.grid.dense_grid = regular_grid
-    
+
     gp.compute_model(model)
 
-
     if PLOT:
         gpv = require_gempy_viewer()
-    
-        
+
         gpv.plot_3d(
             model,
-            image=False,
+            image=True,
             transformed_data=True,
             show_boundaries=True,
-            show_lith=False,
+            show_lith=True,
             kwargs_plot_data={
                     'arrow_size': .01
             }
         )
 
-
         gpv.plot_3d(
             model,
-            image=False,
+            image=True,
             transformed_data=False,
             show_boundaries=True,
-            show_lith=False,
+            show_lith=True,
             kwargs_plot_data={
                     'arrow_size': .01
             }
         )
-
-
-def test_transformed_data():
-    if transfromed_data := True:  # TODO: Expose this to user
-        xyz2 = surface_points.model_transform.apply_with_pivot(
-            points=xyz,
-            # pivot=np.array([5_478_256.5, 5_698_528.946534388,0]),
-            pivot=np.array([5.47825650e+06, 5.69852895e+06, -1.48920000e+03])
-
-        )
-
-        xyz = np.vstack([xyz, xyz2])
-
-    agggg = np.concatenate([mapped_array, mapped_array])
-
-    if transfromed_data := True:
-        orientations_xyz2 = orientations.model_transform.apply_with_pivot(
-            points=orientations_xyz,
-            pivot=np.array([5.47825650e+06, 5.69852895e+06, -1.48920000e+03])
-        )
-        orientations_grads2 = orientations.model_transform.transform_gradient(orientations_grads)
-        arrows_factor /= orientations.model_transform.isometric_scale
-
-        orientations_xyz = np.vstack([orientations_xyz, orientations_xyz2])
-
-        input_transform = regular_grid.input_transform
-        transformed = input_transform.apply(regular_grid.bounding_box)  # ! grid already has the grid transform applied
-        new_extents = np.array([transformed[:, 0].min(), transformed[:, 0].max(),
-                                transformed[:, 1].min(), transformed[:, 1].max(),
-                                transformed[:, 2].min(), transformed[:, 2].max()])