[BUG] Triangulation looking good

Author: Leguark

gempy_engine/API/dual_contouring/_dual_contouring.py

@@ -46,7 +46,6 @@ def compute_dual_contouring(dc_data_per_stack: DualContouringData, left_right_co
                 dc_data_per_stack,
                 debug, 
                 i, 
-                last_surface_edge_idx, 
                 left_right_codes, 
                 valid_edges_per_surface,
                 compute_indices=False 
@@ -57,7 +56,6 @@ def compute_dual_contouring(dc_data_per_stack: DualContouringData, left_right_co
                 dc_data_per_stack,
                 debug, 
                 i, 
-                last_surface_edge_idx, 
                 left_right_codes, 
                 valid_edges_per_surface,
                 compute_indices=True

gempy_engine/core/data/dual_contouring_data.py

@@ -21,6 +21,7 @@ class DualContouringData:
     _gradients: np.ndarray = None
 
     tree_depth: int = -1
+    last_surface_edge_idx: int = 0
     # Water tight 
     mask: np.ndarray = None
 

gempy_engine/modules/dual_contouring/_sequential_triangulation.py

@@ -11,16 +11,19 @@
 
 
 def _sequential_triangulation(dc_data_per_stack: DualContouringData,
-                              debug: bool
-                              , i: int, last_surface_edge_idx: int, 
+                              debug: bool , 
+                              i: int,
                               left_right_codes, 
                               valid_edges_per_surface,
                               compute_indices=True
                               ) -> tuple[Any, Any]:
     valid_edges: np.ndarray = valid_edges_per_surface[i]
-    next_surface_edge_idx: int = valid_edges.sum() + last_surface_edge_idx
+    next_surface_edge_idx: int = valid_edges_per_surface[:i+1].sum()
+    if i == 0:
+        last_surface_edge_idx = 0
+    else:
+        last_surface_edge_idx: int = valid_edges_per_surface[:i].sum() 
     slice_object: slice = slice(last_surface_edge_idx, next_surface_edge_idx)
-    last_surface_edge_idx: int = next_surface_edge_idx
 
     dc_data_per_surface = DualContouringData(
         xyz_on_edge=dc_data_per_stack.xyz_on_edge,