[WIP] Proper implementation of plotting lith block

Author: Leguark

gempy/core/data/geo_model.py

@@ -214,6 +214,22 @@ def orientations(self) -> OrientationsTable:
     def orientations(self, value):
         self.structural_frame.orientations = value
 
+    @property
+    def project_bounds(self) -> np.ndarray:
+        return self.grid.bounding_box
+    
+    @property
+    def extent_transformed(self) -> np.ndarray:
+        transformed = self.input_transform.apply(self.project_bounds)  # ! 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()])
+        return new_extents
+    
+    @property
+    def extent(self) -> np.ndarray:
+        return self.grid.extent
+
     @property
     def interpolation_input_copy(self):
         if self.structural_frame.is_dirty is False:

gempy/modules/data_manipulation/engine_factory.py

@@ -48,7 +48,6 @@ def interpolation_input_from_structural_frame(structural_frame: StructuralFrame,
 
 def _apply_input_transform_to_grids(grid: Grid, input_transform: Transform) -> engine_grid.EngineGrid:
     transformed = input_transform.apply(grid.bounding_box)  # ! grid already has the grid transform applied
-    grid.regular_grid.input_transform = input_transform
     new_extents = np.array([transformed[:, 0].min(), transformed[:, 0].max(),
                             transformed[:, 1].min(), transformed[:, 1].max(),
                             transformed[:, 2].min(), transformed[:, 2].max()])

test/test_modules/.benchmarks/test_transformed_space.py

@@ -8,15 +8,29 @@
 
 
 def test_plot_transformed_data():
-    model = gp.generate_example_model(ExampleModel.ANTICLINE, compute_model=False)
+    model = gp.generate_example_model(ExampleModel.ANTICLINE, compute_model=True)
     print(model.structural_frame)
 
     if PLOT:
         gpv = require_gempy_viewer()
+
+        gpv.plot_3d(
+            model,
+            image=True,
+            transformed_data=False,
+            show_boundaries=False,
+            show_lith=True,
+            kwargs_plot_data={
+                    'arrow_size': 10
+            }
+        )
+        
         gpv.plot_3d(
             model,
             image=False,
             transformed_data=True,
+            show_boundaries=False,
+            show_lith=True,
             kwargs_plot_data={
                     'arrow_size': .01
             }