Skip to content

Commit 256cfa3

Browse files
committed
[ENH] Add optional Transform input to meshes_to_subsurface
Updated the `meshes_to_subsurface` method to support an optional `Transform` input, applying inverse transformations to vertex data when provided.
1 parent 43039a6 commit 256cfa3

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

gempy_engine/core/data/raw_arrays_solution.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44

55
import numpy as np
66

7+
from .transforms import Transform
78
from ..backend_tensor import BackendTensor
89
from .interp_output import InterpOutput
910
from .dual_contouring_mesh import DualContouringMesh
@@ -112,7 +113,7 @@ def _set_scalar_field_at_surface_points(self, octree_output: OctreeLevel):
112113

113114
self.scalar_field_at_surface_points = temp_list
114115

115-
def meshes_to_subsurface(self):
116+
def meshes_to_subsurface(self, input_transform: Transform | None = None):
116117
ss = require_subsurface()
117118
pd = require_pandas()
118119

@@ -129,9 +130,13 @@ def meshes_to_subsurface(self):
129130

130131
concatenated_id_array = np.concatenate(vertex_id_array)
131132
concatenated_cell_id_array = np.concatenate(cell_id_array)
133+
134+
all_vertex = np.concatenate(vertex)
135+
if input_transform:
136+
all_vertex = input_transform.apply_inverse(all_vertex)
132137

133138
meshes: ss.UnstructuredData = ss.UnstructuredData.from_array(
134-
vertex=np.concatenate(vertex),
139+
vertex=all_vertex,
135140
cells=np.concatenate(simplex_list),
136141
vertex_attr=pd.DataFrame({'id': concatenated_id_array}),
137142
cells_attr=pd.DataFrame({'id': concatenated_cell_id_array})

0 commit comments

Comments
 (0)