Skip to content

Commit 0f4b1f0

Browse files
committed
[WIP]
1 parent 61dd15f commit 0f4b1f0

File tree

1 file changed

+15
-15
lines changed

1 file changed

+15
-15
lines changed

tests/test_common/test_modules/test_dual.py

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -245,7 +245,7 @@ def test_compute_dual_contouring_complex(unconformity_complex_one_layer, n_oct_l
245245
dc_data = solutions.dc_meshes[0].dc_data
246246

247247
if plot_pyvista or False:
248-
output_corners: InterpOutput = solutions.octrees_output[-1].outputs_corners[-1]
248+
output_corners: InterpOutput = solutions.octrees_output[-1].outputs_centers[-1]
249249
vertices = output_corners.grid.values
250250

251251
intersection_xyz = dc_data.xyz_on_edge
@@ -278,11 +278,11 @@ def test_compute_dual_contouring_several_meshes(simple_model_3_layers, simple_gr
278278

279279
last_octree_level: OctreeLevel = octree_list[-1]
280280

281-
corners = last_octree_level.outputs_corners[0]
281+
corners = last_octree_level.outputs_centers[0]
282282
# First find xyz on edges:
283283
xyz, edges = find_intersection_on_edge(
284-
_xyz_corners=last_octree_level.grid_corners.values,
285-
scalar_field_on_corners=corners.exported_fields.scalar_field,
284+
_xyz_corners=last_octree_level.grid_centers.corners_grid.values,
285+
scalar_field_on_corners=corners.exported_fields.scalar_field[corners.grid.corners_grid_slice],
286286
scalar_at_sp=corners.scalar_field_at_sp,
287287
masking=None
288288
)
@@ -295,7 +295,7 @@ def test_compute_dual_contouring_several_meshes(simple_model_3_layers, simple_gr
295295
dc_data = DualContouringData(
296296
xyz_on_edge=intersection_xyz,
297297
valid_edges=valid_edges,
298-
xyz_on_centers=last_octree_level.grid_centers.values,
298+
xyz_on_centers=last_octree_level.grid_centers.octree_grid.values,
299299
dxdydz=last_octree_level.grid_centers.octree_dxdydz,
300300
exported_fields_on_edges=output_on_edges.exported_fields,
301301
n_surfaces_to_export=data_shape.tensors_structure.n_surfaces
@@ -336,9 +336,9 @@ def test_find_edges_intersection_step_by_step(simple_model, simple_grid_3d_octre
336336

337337
last_octree_level: OctreeLevel = octree_list[-1]
338338

339-
sfsp = last_octree_level.last_output_corners.scalar_field_at_sp
339+
sfsp = last_octree_level.last_output_center.scalar_field_at_sp
340340

341-
xyz_on_edge, valid_edges = find_intersection_on_edge(last_octree_level.grid_corners.values, last_octree_level.output_corners.exported_fields.scalar_field, sfsp, )
341+
xyz_on_edge, valid_edges = find_intersection_on_edge(last_octree_level.grid_centers.corners_grid.values, last_octree_level.last_output_center.exported_fields.scalar_field[last_octree_level.last_output_center.grid.corners_grid_slice], sfsp, )
342342

343343
# endregion
344344

@@ -450,9 +450,9 @@ def test_find_edges_intersection_pro(simple_model, simple_grid_3d_octree):
450450

451451
last_octree_level: OctreeLevel = octree_list[-1]
452452

453-
sfsp = last_octree_level.output_corners.scalar_field_at_sp
453+
sfsp = last_octree_level.last_output_center.scalar_field_at_sp
454454
# sfsp = np.append(sfsp, -0.1)
455-
xyz_on_edge, valid_edges = find_intersection_on_edge(last_octree_level.grid_corners.values, last_octree_level.output_corners.exported_fields.scalar_field, sfsp, )
455+
xyz_on_edge, valid_edges = find_intersection_on_edge(last_octree_level.grid_centers.corners_grid.values, last_octree_level.last_output_center.exported_fields.scalar_field[last_octree_level.last_output_center.grid.corners_grid_slice], sfsp, )
456456
# endregion
457457

458458
# region Get Normals
@@ -560,8 +560,8 @@ def test_find_edges_intersection_bias_on_center_of_the_cell(simple_model, simple
560560

561561
last_octree_level: OctreeLevel = octree_list[-1]
562562

563-
sfsp = last_octree_level.output_corners.scalar_field_at_sp
564-
xyz_on_edge, valid_edges = find_intersection_on_edge(last_octree_level.grid_corners.values, last_octree_level.output_corners.exported_fields.scalar_field, sfsp, )
563+
sfsp = last_octree_level.last_output_center.scalar_field_at_sp
564+
xyz_on_edge, valid_edges = find_intersection_on_edge(last_octree_level.grid_centers.corners_grid.values, last_octree_level.last_output_center.exported_fields.scalar_field[last_octree_level.last_output_center.grid.corners_grid_slice], sfsp, )
565565
valid_voxels = valid_edges.sum(axis=1, dtype=bool)
566566

567567
# endregion
@@ -591,7 +591,7 @@ def test_find_edges_intersection_bias_on_center_of_the_cell(simple_model, simple
591591

592592
BIAS_STRENGTH = 1
593593

594-
mass_points = last_octree_level.grid_centers.values
594+
mass_points = last_octree_level.grid_centers.octree_grid.values
595595

596596
xyz[:, 12] = mass_points
597597
xyz[:, 13] = mass_points
@@ -622,13 +622,13 @@ def test_find_edges_intersection_bias_on_center_of_the_cell(simple_model, simple
622622
# region triangulate
623623
grid_centers = last_octree_level.grid_centers
624624

625-
temp_ids = octree_list[-1].output_centers.ids_block # ! I need this because setters in python sucks
625+
temp_ids = octree_list[-1].last_output_center.ids_block # ! I need this because setters in python sucks
626626
temp_ids[valid_voxels] = 5
627-
octree_list[-1].output_centers.ids_block = temp_ids # paint valid voxels
627+
octree_list[-1].last_output_center.ids_block = temp_ids # paint valid voxels
628628

629629
dc_data = DualContouringData(
630630
xyz_on_edge=xyz_on_edge,
631-
xyz_on_centers=grid_centers.values,
631+
xyz_on_centers=grid_centers.octree_grid.values,
632632
dxdydz=grid_centers.octree_dxdydz,
633633
valid_edges=valid_edges,
634634
exported_fields_on_edges=None,

0 commit comments

Comments
 (0)