[WIP] Towards a proper implementation for plotting transform data

Author: Leguark

gempy/core/data/geo_model.py

@@ -170,11 +170,15 @@ def solutions(self, value):
     def surface_points_copy(self):
         """This is a copy! Returns a SurfacePointsTable for all surface points across the structural elements"""
         surface_points_table = self.structural_frame.surface_points_copy
-        if self.input_transform is not None:
-            transform = self.input_transform + self.grid.transform
-            surface_points_table.model_transform = transform
         return surface_points_table
-
+    
+    @property
+    def surface_points_copy_transformed(self):
+        og_sp = self.surface_points_copy
+        total_transform: Transform = self.input_transform + self.grid.transform
+        og_sp.xyz_view = total_transform.apply(og_sp.xyz)
+        return og_sp
+        
     @property
     def surface_points(self):
         raise AttributeError("This property can only be set, not read. You can access the copy with `surface_points_copy` or"
@@ -192,6 +196,14 @@ def orientations_copy(self) -> OrientationsTable:
             transform = self.input_transform + self.grid.transform
             orientations_table.model_transform = transform
         return orientations_table
+    
+    @property
+    def orientations_copy_transformed(self):
+        # ! This is not done
+        og_or = self.orientations_copy
+        total_transform: Transform = self.input_transform + self.grid.transform
+        og_or.xyz_view = total_transform.apply(og_or.xyz)
+        return og_or
 
     @property
     def orientations(self) -> OrientationsTable:

gempy/core/data/surface_points.py

@@ -88,6 +88,14 @@ def id_to_name(self, id: int) -> str:
     @property
     def xyz(self) -> np.ndarray:
         return np.array([self.data['X'], self.data['Y'], self.data['Z']]).T
+    
+    @property
+    def xyz_view(self) -> np.ndarray:
+        return self.data[['X', 'Y', 'Z']]
+    
+    @xyz_view.setter
+    def xyz_view(self, value: np.ndarray):
+        self.data['X'], self.data['Y'], self.data['Z'] = value.T
 
     @property
     def nugget(self) -> np.ndarray:

test/test_modules/.benchmarks/test_transformed_space.py

@@ -1,17 +1,23 @@
 import gempy as gp
+from gempy import optional_dependencies
 from gempy.core.data.enumerators import ExampleModel
+from gempy.modules.data_manipulation.engine_factory import interpolation_input_from_structural_frame
 from gempy.optional_dependencies import require_gempy_viewer
 
 PLOT = True
 
 
 def test_plot_transformed_data():
-    model = gp.generate_example_model(ExampleModel.ANTICLINE, compute_model=True)
+    model = gp.generate_example_model(ExampleModel.ANTICLINE, compute_model=False)
     print(model.structural_frame)
 
     if PLOT:
         gpv = require_gempy_viewer()
-        gpv.plot_3d(model, image=True)
+        gpv.plot_3d(
+            model,
+            image=False,
+            transformed_data=True
+        )
 
 
 def test_transformed_data():