swh:1:snp:91058747fd66ffbf6588e3a7feb2cbdc90d2fe46
Tip revision: 38d4379f07e6a18f41a3e325e1d2ad9d62b1b73f authored by Yiwen Ju (Owen) on 17 June 2024, 17:22:54 UTC
CSG application that takes in a csg tree and prunes away patches, cha… (#14)
CSG application that takes in a csg tree and prunes away patches, cha… (#14)
Tip revision: 38d4379
test_implicit_networks.cpp
//
// Created by Charles Du on 8/19/22.
//
#include <simplicial_arrangement/lookup_table.h>
#include <fstream>
#include "implicit_arrangement.h"
#include "material_interface.h"
#include "implicit_functions.h"
#include "csg.h"
#include <catch2/catch.hpp>
// tests:
// examples with known structure
// consistency:
// - look-up vs non-look-up
// - topo-ray-shooting vs cell-grouping
TEST_CASE("implicit arrangement on known examples", "[IA][examples]") {
bool robust_test = false;
bool use_lookup = true;
bool loaded = simplicial_arrangement::load_lookup_table();
REQUIRE(loaded);
bool use_secondary_lookup = true;
bool use_topo_ray_shooting = true;
// generate tet grid
std::vector<std::array<double, 3>> pts;
std::vector<std::array<size_t, 4>> tets;
int tet_mesh_resolution = 101;
std::array<double,3> tet_mesh_bbox_min {-1,-1,-1};
std::array<double,3> tet_mesh_bbox_max {1,1,1};
generate_tet_mesh(tet_mesh_resolution, tet_mesh_bbox_min,
tet_mesh_bbox_max, pts, tets);
// function values
Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor> funcVals;
// implicit arrangement result
std::vector<std::array<double, 3>> iso_pts;
std::vector<PolygonFace> iso_faces;
std::vector<std::vector<size_t>> patches;
std::vector<size_t> patch_function_label;
std::vector<Edge> iso_edges;
std::vector<std::vector<size_t>> chains;
std::vector<std::vector<size_t>> non_manifold_edges_of_vert;
std::vector<std::vector<size_t>> shells;
std::vector<std::vector<size_t>> arrangement_cells;
std::vector<std::vector<bool>> cell_function_label;
// record timings
std::vector<std::string> timing_labels;
std::vector<double> timings;
// record stats
std::vector<std::string> stats_labels;
std::vector<size_t> stats;
SECTION("one plane") {
// compute function values on tet grid vertices
size_t n_pts = pts.size();
size_t n_func = 1;
funcVals.resize(n_pts, n_func);
if (!load_functions(std::string(TEST_FILE) + "/1-plane.json", pts, funcVals)) {
throw std::runtime_error("ERROR: Failed to load functions.");
}
// compute implicit arrangement
bool success = implicit_arrangement(
robust_test,
use_lookup,
use_secondary_lookup,
use_topo_ray_shooting,
//
pts, tets, funcVals,
//
iso_pts,iso_faces,patches,
patch_function_label,
iso_edges,chains,
non_manifold_edges_of_vert,
shells,arrangement_cells,cell_function_label,
timing_labels,timings,
stats_labels,stats);
REQUIRE(success);
// check
REQUIRE(patches.size() == 1);
REQUIRE(patch_function_label.size() == 1);
REQUIRE(chains.size() == 0);
REQUIRE(arrangement_cells.size() == 2);
// function label check
std::vector<size_t> patch_gt = {0};
REQUIRE(patch_function_label == patch_gt);
std::vector<std::vector<bool>> cell_gt = {{1}, {0}};
REQUIRE(cell_function_label == cell_gt);
}
SECTION("one sphere") {
// compute function values on tet grid vertices
size_t n_pts = pts.size();
size_t n_func = 1;
funcVals.resize(n_pts, n_func);
if (!load_functions(std::string(TEST_FILE) + "/1-sphere.json", pts, funcVals)) {
throw std::runtime_error("ERROR: Failed to load functions.");
}
// compute implicit arrangement
bool success = implicit_arrangement(
robust_test,
use_lookup,
use_secondary_lookup,
use_topo_ray_shooting,
//
pts, tets, funcVals,
//
iso_pts,iso_faces,patches,
patch_function_label,
iso_edges,chains,
non_manifold_edges_of_vert,
shells,arrangement_cells,cell_function_label,
timing_labels,timings,
stats_labels,stats);
REQUIRE(success);
// check
REQUIRE(patches.size() == 1);
REQUIRE(patch_function_label.size() == 1);
REQUIRE(chains.size() == 0);
REQUIRE(arrangement_cells.size() == 2);
// function label check
std::vector<size_t> patch_gt = {0};
REQUIRE(patch_function_label == patch_gt);
std::vector<std::vector<bool>> cell_gt = {{1}, {0}};
REQUIRE(cell_function_label == cell_gt);
}
SECTION("plane and sphere") {
// compute function values on tet grid vertices
size_t n_pts = pts.size();
size_t n_func = 2;
funcVals.resize(n_pts, n_func);
size_t func_id;
if (!load_functions(std::string(TEST_FILE) + "/2-planesphere.json", pts, funcVals)) {
throw std::runtime_error("ERROR: Failed to load functions.");
}
// compute implicit arrangement
bool success = implicit_arrangement(
robust_test,
use_lookup,
use_secondary_lookup,
use_topo_ray_shooting,
//
pts, tets, funcVals,
//
iso_pts,iso_faces,patches,
patch_function_label,
iso_edges,chains,
non_manifold_edges_of_vert,
shells,arrangement_cells,cell_function_label,
timing_labels,timings,
stats_labels,stats);
REQUIRE(success);
// check
REQUIRE(patches.size() == 4);
REQUIRE(patch_function_label.size() == 4);
REQUIRE(chains.size() == 1);
REQUIRE(arrangement_cells.size() == 4);
// function label check
std::vector<size_t> patch_gt = {1,0,0,1};
REQUIRE(patch_function_label == patch_gt);
std::vector<std::vector<bool>> cell_gt = {{0,1}, {0,0},{1,0},{1,1}};
REQUIRE(cell_function_label == cell_gt);
}
SECTION("two spheres") {
// compute function values on tet grid vertices
size_t n_pts = pts.size();
size_t n_func = 2;
funcVals.resize(n_pts, n_func);
size_t func_id;
if (!load_functions(std::string(TEST_FILE) + "/2-sphere.json", pts, funcVals)) {
throw std::runtime_error("ERROR: Failed to load functions.");
}
// compute implicit arrangement
bool success = implicit_arrangement(
robust_test,
use_lookup,
use_secondary_lookup,
use_topo_ray_shooting,
//
pts, tets, funcVals,
//
iso_pts,iso_faces,patches,
patch_function_label,
iso_edges,chains,
non_manifold_edges_of_vert,
shells,arrangement_cells,cell_function_label,
timing_labels,timings,
stats_labels,stats);
REQUIRE(success);
// check
REQUIRE(patches.size() == 4);
REQUIRE(patch_function_label.size() == 4);
REQUIRE(chains.size() == 1);
REQUIRE(arrangement_cells.size() == 4);
// function label check
std::vector<size_t> patch_gt = {0,1,1,0};
REQUIRE(patch_function_label == patch_gt);
std::vector<std::vector<bool>> cell_gt = {{1,0}, {0,0},{1,1},{0,1}};
REQUIRE(cell_function_label == cell_gt);
}
SECTION("three spheres config 1") {
// compute function values on tet grid vertices
size_t n_pts = pts.size();
size_t n_func = 3;
funcVals.resize(n_pts, n_func);
size_t func_id;
if (!load_functions(std::string(TEST_FILE) + "/3-sphere-1.json", pts, funcVals)) {
throw std::runtime_error("ERROR: Failed to load functions.");
}
// compute implicit arrangement
bool success = implicit_arrangement(
robust_test,
use_lookup,
use_secondary_lookup,
use_topo_ray_shooting,
//
pts, tets, funcVals,
//
iso_pts,iso_faces,patches,
patch_function_label,
iso_edges,chains,
non_manifold_edges_of_vert,
shells,arrangement_cells,cell_function_label,
timing_labels,timings,
stats_labels,stats);
REQUIRE(success);
// check
REQUIRE(patches.size() == 3);
REQUIRE(patch_function_label.size() == 3);
REQUIRE(chains.size() == 0);
REQUIRE(arrangement_cells.size() == 4);
// function label check
std::vector<size_t> patch_gt = {0,2,1};
REQUIRE(patch_function_label == patch_gt);
std::vector<std::vector<bool>> cell_gt = {{1,0,0}, {0,0,0},{0,0,1},{0,1,0}};
REQUIRE(cell_function_label == cell_gt);
}
SECTION("three spheres config 2") {
// compute function values on tet grid vertices
size_t n_pts = pts.size();
size_t n_func = 3;
funcVals.resize(n_pts, n_func);
size_t func_id;
if (!load_functions(std::string(TEST_FILE) + "/3-sphere-2.json", pts, funcVals)) {
throw std::runtime_error("ERROR: Failed to load functions.");
}
// compute implicit arrangement
bool success = implicit_arrangement(
robust_test,
use_lookup,
use_secondary_lookup,
use_topo_ray_shooting,
//
pts, tets, funcVals,
//
iso_pts,iso_faces,patches,
patch_function_label,
iso_edges,chains,
non_manifold_edges_of_vert,
shells,arrangement_cells,cell_function_label,
timing_labels,timings,
stats_labels,stats);
REQUIRE(success);
// check
REQUIRE(patches.size() == 7);
REQUIRE(patch_function_label.size() == 7);
REQUIRE(chains.size() == 2);
REQUIRE(arrangement_cells.size() == 6);
// function label check
std::vector<size_t> patch_gt = {0, 2, 0, 2, 1, 1, 2};
REQUIRE(patch_function_label == patch_gt);
std::vector<std::vector<bool>> cell_gt = {{1, 0, 0}, {0, 0, 0}, {1, 0, 1}, {0, 0, 1}, {0, 1, 1}, {0, 1, 0}};
REQUIRE(cell_function_label == cell_gt);
}
SECTION("three spheres config 3") {
// compute function values on tet grid vertices
size_t n_pts = pts.size();
size_t n_func = 3;
funcVals.resize(n_pts, n_func);
size_t func_id;
if (!load_functions(std::string(TEST_FILE) + "/3-sphere-3.json", pts, funcVals)) {
throw std::runtime_error("ERROR: Failed to load functions.");
}
// compute implicit arrangement
bool success = implicit_arrangement(
robust_test,
use_lookup,
use_secondary_lookup,
use_topo_ray_shooting,
//
pts, tets, funcVals,
//
iso_pts,iso_faces,patches,
patch_function_label,
iso_edges,chains,
non_manifold_edges_of_vert,
shells,arrangement_cells,cell_function_label,
timing_labels,timings,
stats_labels,stats);
REQUIRE(success);
// check
REQUIRE(patches.size() == 12);
REQUIRE(patch_function_label.size() == 12);
REQUIRE(chains.size() == 6);
REQUIRE(arrangement_cells.size() == 8);
// function label check
std::vector<size_t> patch_gt = {0, 2, 0, 2, 1, 1, 2, 1, 0, 1, 2, 0};
REQUIRE(patch_function_label == patch_gt);
std::vector<std::vector<bool>> cell_gt = {{1, 0, 0},{0, 0, 0}, {0, 0, 1}, {1, 0, 1}, {1, 1, 0}, {1, 1, 1}, {0, 1, 1}, {0, 1, 0}};
REQUIRE(cell_function_label == cell_gt);
}
SECTION("three spheres config 4") {
// compute function values on tet grid vertices
size_t n_pts = pts.size();
size_t n_func = 3;
funcVals.resize(n_pts, n_func);
size_t func_id;
if (!load_functions(std::string(TEST_FILE) + "/3-sphere-4.json", pts, funcVals)) {
throw std::runtime_error("ERROR: Failed to load functions.");
}
// compute implicit arrangement
bool success = implicit_arrangement(
robust_test,
use_lookup,
use_secondary_lookup,
use_topo_ray_shooting,
//
pts, tets, funcVals,
//
iso_pts,iso_faces,patches,
patch_function_label,
iso_edges,chains,
non_manifold_edges_of_vert,
shells,arrangement_cells,cell_function_label,
timing_labels,timings,
stats_labels,stats);
REQUIRE(success);
// check
REQUIRE(patches.size() == 3);
REQUIRE(patch_function_label.size() == 3);
REQUIRE(chains.size() == 0);
REQUIRE(arrangement_cells.size() == 4);
// function label check
std::vector<size_t> patch_gt = {2, 1, 0};
REQUIRE(patch_function_label == patch_gt);
std::vector<std::vector<bool>> cell_gt = {{0, 0, 1}, {0, 0, 0}, {0, 1, 1}, {1, 1, 1}};
REQUIRE(cell_function_label == cell_gt);
}
SECTION("three spheres config 5") {
// compute function values on tet grid vertices
size_t n_pts = pts.size();
size_t n_func = 3;
funcVals.resize(n_pts, n_func);
size_t func_id;
if (!load_functions(std::string(TEST_FILE) + "/3-sphere-5.json", pts, funcVals)) {
throw std::runtime_error("ERROR: Failed to load functions.");
}
// the third sphere is negative inside
for (Eigen::Index i = 0; i < n_pts; i++) {
funcVals(i, 2) = -funcVals(i, 2);
}
// compute implicit arrangement
bool success = implicit_arrangement(
robust_test,
use_lookup,
use_secondary_lookup,
use_topo_ray_shooting,
//
pts, tets, funcVals,
//
iso_pts,iso_faces,patches,
patch_function_label,
iso_edges,chains,
non_manifold_edges_of_vert,
shells,arrangement_cells,cell_function_label,
timing_labels,timings,
stats_labels,stats);
REQUIRE(success);
// check
REQUIRE(patches.size() == 4);
REQUIRE(patch_function_label.size() == 4);
REQUIRE(chains.size() == 1);
REQUIRE(arrangement_cells.size() == 4);
// function label check
std::vector<size_t> patch_gt = {2, 1, 1, 2};
REQUIRE(patch_function_label == patch_gt);
std::vector<std::vector<bool>> cell_gt = {{1, 0, 1}, {1, 0, 0}, {1, 1, 1}, {1, 1, 0}};
REQUIRE(cell_function_label == cell_gt);
}
}
TEST_CASE("material interface on known examples", "[MI][examples]") {
bool robust_test = false;
bool use_lookup = true;
bool loaded = load_lookup_table(simplicial_arrangement::MATERIAL_INTERFACE);
REQUIRE(loaded);
bool use_secondary_lookup = true;
bool use_topo_ray_shooting = true;
// generate tet grid
std::vector<std::array<double, 3>> pts;
std::vector<std::array<size_t, 4>> tets;
int tet_mesh_resolution = 101;
std::array<double,3> tet_mesh_bbox_min {-1,-1,-1};
std::array<double,3> tet_mesh_bbox_max {1,1,1};
generate_tet_mesh(tet_mesh_resolution, tet_mesh_bbox_min,
tet_mesh_bbox_max, pts, tets);
// function values
Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor> funcVals;
// implicit arrangement result
std::vector<std::array<double, 3>> MI_pts;
std::vector<PolygonFace> MI_faces;
std::vector<std::vector<size_t>> patches;
std::vector<std::pair<size_t, size_t>> patch_function_label;
std::vector<Edge> MI_edges;
std::vector<std::vector<size_t>> chains;
std::vector<std::vector<size_t>> non_manifold_edges_of_vert;
std::vector<std::vector<size_t>> shells;
std::vector<std::vector<size_t>> arrangement_cells;
std::vector<size_t> cell_function_label;
// record timings
std::vector<std::string> timing_labels;
std::vector<double> timings;
// record stats
std::vector<std::string> stats_labels;
std::vector<size_t> stats;
SECTION("one sphere") {
// compute function values on tet grid vertices
size_t n_pts = pts.size();
size_t n_func = 3;
funcVals.resize(n_pts, n_func);
size_t func_id;
if (!load_functions(std::string(TEST_FILE) + "/1-sphere.json", pts, funcVals)) {
throw std::runtime_error("ERROR: Failed to load functions.");
}
// compute material interface
bool success = material_interface(
robust_test,
use_lookup,
use_secondary_lookup,
use_topo_ray_shooting,
//
pts, tets, funcVals,
//
MI_pts,MI_faces,patches,
patch_function_label,
MI_edges,chains,
non_manifold_edges_of_vert,
shells,arrangement_cells,cell_function_label,
timing_labels,timings,
stats_labels,stats);
REQUIRE(success);
// check
REQUIRE(patches.size() == 0);
REQUIRE(patch_function_label.size() == 0);
REQUIRE(chains.size() == 0);
REQUIRE(arrangement_cells.size() == 1);
// function label check
std::vector<std::pair<size_t, size_t>> patch_gt = {};
REQUIRE(patch_function_label == patch_gt);
std::vector<size_t> cell_gt = {0};
REQUIRE(cell_function_label == cell_gt);
}
SECTION("plane and sphere") {
// compute function values on tet grid vertices
size_t n_pts = pts.size();
size_t n_func = 2;
funcVals.resize(n_pts, n_func);
size_t func_id;
if (!load_functions(std::string(TEST_FILE) + "/2-planesphere.json", pts, funcVals)) {
throw std::runtime_error("ERROR: Failed to load functions.");
}
// compute material interface
bool success = material_interface(
robust_test,
use_lookup,
use_secondary_lookup,
use_topo_ray_shooting,
//
pts, tets, funcVals,
//
MI_pts,MI_faces,patches,
patch_function_label,
MI_edges,chains,
non_manifold_edges_of_vert,
shells,arrangement_cells,cell_function_label,
timing_labels,timings,
stats_labels,stats);
REQUIRE(success);
// check
REQUIRE(patches.size() == 1);
REQUIRE(patch_function_label.size() == 1);
REQUIRE(chains.size() == 0);
REQUIRE(arrangement_cells.size() == 2);
// function label check
std::vector<std::pair<size_t, size_t>> patch_gt = {{1, 0}};
REQUIRE(patch_function_label == patch_gt);
std::vector<size_t> cell_gt = {1, 0};
REQUIRE(cell_function_label == cell_gt);
}
SECTION("two spheres") {
// compute function values on tet grid vertices
size_t n_pts = pts.size();
size_t n_func = 2;
funcVals.resize(n_pts, n_func);
size_t func_id;
if (!load_functions(std::string(TEST_FILE) + "/2-sphere.json", pts, funcVals)) {
throw std::runtime_error("ERROR: Failed to load functions.");
}
// compute material interface
bool success = material_interface(
robust_test,
use_lookup,
use_secondary_lookup,
use_topo_ray_shooting,
//
pts, tets, funcVals,
//
MI_pts,MI_faces,patches,
patch_function_label,
MI_edges,chains,
non_manifold_edges_of_vert,
shells,arrangement_cells,cell_function_label,
timing_labels,timings,
stats_labels,stats);
REQUIRE(success);
// check
REQUIRE(patches.size() == 1);
REQUIRE(patch_function_label.size() == 1);
REQUIRE(chains.size() == 0);
REQUIRE(arrangement_cells.size() == 2);
// function label check
std::vector<std::pair<size_t, size_t>> patch_gt = {{1, 0}};
REQUIRE(patch_function_label == patch_gt);
std::vector<size_t> cell_gt = {1, 0};
REQUIRE(cell_function_label == cell_gt);
}
SECTION("three spheres config 1") {
// compute function values on tet grid vertices
size_t n_pts = pts.size();
size_t n_func = 3;
funcVals.resize(n_pts, n_func);
size_t func_id;
if (!load_functions(std::string(TEST_FILE) + "/3-sphere-1.json", pts, funcVals)) {
throw std::runtime_error("ERROR: Failed to load functions.");
}
// compute material interface
bool success = material_interface(
robust_test,
use_lookup,
use_secondary_lookup,
use_topo_ray_shooting,
//
pts, tets, funcVals,
//
MI_pts,MI_faces,patches,
patch_function_label,
MI_edges,chains,
non_manifold_edges_of_vert,
shells,arrangement_cells,cell_function_label,
timing_labels,timings,
stats_labels,stats);
REQUIRE(success);
// check
REQUIRE(patches.size() == 3);
REQUIRE(patch_function_label.size() == 3);
REQUIRE(chains.size() == 1);
REQUIRE(arrangement_cells.size() == 3);
// function label check
std::vector<std::pair<size_t, size_t>> patch_gt = {{2, 0}, {2, 1}, {1, 0}};
REQUIRE(patch_function_label == patch_gt);
std::vector<size_t> cell_gt = {2, 0, 1};
REQUIRE(cell_function_label == cell_gt);
}
SECTION("three sphere config 4") {
// compute function values on tet grid vertices
size_t n_pts = pts.size();
size_t n_func = 3;
funcVals.resize(n_pts, n_func);
size_t func_id;
if (!load_functions(std::string(TEST_FILE) + "/3-sphere-4.json", pts, funcVals)) {
throw std::runtime_error("ERROR: Failed to load functions.");
}
// compute material interface
bool success = material_interface(
robust_test,
use_lookup,
use_secondary_lookup,
use_topo_ray_shooting,
//
pts, tets, funcVals,
//
MI_pts,MI_faces,patches,
patch_function_label,
MI_edges,chains,
non_manifold_edges_of_vert,
shells,arrangement_cells,cell_function_label,
timing_labels,timings,
stats_labels,stats);
REQUIRE(success);
// check
REQUIRE(patches.size() == 0);
REQUIRE(patch_function_label.size() == 0);
REQUIRE(chains.size() == 0);
REQUIRE(arrangement_cells.size() == 1);
// function label check
std::vector<std::pair<size_t, size_t>> patch_gt = {};
REQUIRE(patch_function_label == patch_gt);
std::vector<size_t> cell_gt = {2};
REQUIRE(cell_function_label == cell_gt);
}
SECTION("eight sphere") {
load_tet_mesh(std::string(TEST_FILE) + "/mesh.json", pts, tets);
// compute function values on tet grid vertices
size_t n_pts = pts.size();
size_t n_func = 8;
funcVals.resize(n_pts, n_func);
size_t func_id;
if (!load_functions(std::string(TEST_FILE) + "/8-sphere.json", pts, funcVals)) {
throw std::runtime_error("ERROR: Failed to load functions.");
}
// compute material interface
bool success = material_interface(
robust_test,
use_lookup,
use_secondary_lookup,
use_topo_ray_shooting,
//
pts, tets, funcVals,
//
MI_pts,MI_faces,patches,
patch_function_label,
MI_edges,chains,
non_manifold_edges_of_vert,
shells,arrangement_cells,cell_function_label,
timing_labels,timings,
stats_labels,stats);
REQUIRE(success);
size_t corners_count = 0;
for (size_t i = 0; i < non_manifold_edges_of_vert.size(); i++) {
if (non_manifold_edges_of_vert[i].size() > 2) {
corners_count++;
}
}
std::cout << "corners: " << corners_count << std::endl;
std::cout << "cell labels: " << std::endl;
// check
REQUIRE(shells.size() == 8);
REQUIRE(arrangement_cells.size() == 8);
REQUIRE(corners_count == 6);
// function label check
std::vector<std::pair<size_t, size_t>> patch_gt = {{3, 1}, {7, 3}, {7, 6}, {3, 2}, {6, 2}, {6, 4}, {5, 4}, {5, 1}, {7, 5}, {4, 0}, {2, 0}, {1, 0}, {7, 1}, {7, 4}, {5, 0}, {6, 0}, {3, 0}, {7, 2}, {7, 0}};
REQUIRE(patch_function_label == patch_gt);
std::vector<size_t> cell_gt = {3, 1, 7, 6, 2, 4, 5, 0};
REQUIRE(cell_function_label == cell_gt);
}
}
TEST_CASE("material interface on a failed example", "[MI][examples][!shouldfail]") {
bool robust_test = false;
bool use_lookup = true;
bool loaded = load_lookup_table(simplicial_arrangement::MATERIAL_INTERFACE);
REQUIRE(loaded);
bool use_secondary_lookup = true;
bool use_topo_ray_shooting = true;
// generate tet grid
std::vector<std::array<double, 3>> pts;
std::vector<std::array<size_t, 4>> tets;
// function values
Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor> funcVals;
// implicit arrangement result
std::vector<std::array<double, 3>> MI_pts;
std::vector<PolygonFace> MI_faces;
std::vector<std::vector<size_t>> patches;
std::vector<std::pair<size_t, size_t>> patch_function_label;
std::vector<Edge> MI_edges;
std::vector<std::vector<size_t>> chains;
std::vector<std::vector<size_t>> non_manifold_edges_of_vert;
std::vector<std::vector<size_t>> shells;
std::vector<std::vector<size_t>> arrangement_cells;
std::vector<size_t> cell_function_label;
// record timings
std::vector<std::string> timing_labels;
std::vector<double> timings;
// record stats
std::vector<std::string> stats_labels;
std::vector<size_t> stats;
SECTION("eight sphere") {
load_tet_mesh(std::string(TEST_FILE) + "/mesh_fail.json", pts, tets);
// compute function values on tet grid vertices
size_t n_pts = pts.size();
size_t n_func = 8;
funcVals.resize(n_pts, n_func);
size_t func_id;
if (!load_functions(std::string(TEST_FILE) + "/8-sphere.json", pts, funcVals)) {
throw std::runtime_error("ERROR: Failed to load functions.");
}
// compute material interface
bool success = material_interface(
robust_test,
use_lookup,
use_secondary_lookup,
use_topo_ray_shooting,
//
pts, tets, funcVals,
//
MI_pts,MI_faces,patches,
patch_function_label,
MI_edges,chains,
non_manifold_edges_of_vert,
shells,arrangement_cells,cell_function_label,
timing_labels,timings,
stats_labels,stats);
REQUIRE(success);
size_t corners_count = 0;
for (size_t i = 0; i < non_manifold_edges_of_vert.size(); i++) {
if (non_manifold_edges_of_vert[i].size() > 2) {
corners_count++;
}
}
std::cout << "corners: " << corners_count << std::endl;
std::cout << "cell labels: " << std::endl;
// check
REQUIRE(shells.size() == 8);
REQUIRE(arrangement_cells.size() == 8);
REQUIRE(corners_count == 6);
// function label check
std::vector<std::pair<size_t, size_t>> patch_gt = {{3, 1}, {7, 3}, {7, 6}, {3, 2}, {6, 2}, {6, 4}, {5, 4}, {5, 1}, {7, 5}, {4, 0}, {2, 0}, {1, 0}, {7, 1}, {7, 4}, {5, 0}, {6, 0}, {3, 0}, {7, 2}, {7, 0}};
REQUIRE(patch_function_label == patch_gt);
std::vector<size_t> cell_gt = {3, 1, 7, 6, 2, 4, 5, 0};
REQUIRE(cell_function_label == cell_gt);
}
}
TEST_CASE("CSG on known examples", "[CSG][examples]") {
bool robust_test = false;
bool use_lookup = true;
bool loaded = simplicial_arrangement::load_lookup_table();
REQUIRE(loaded);
bool use_secondary_lookup = true;
bool use_topo_ray_shooting = true;
bool positive_inside = true;
// generate tet grid
std::vector<std::array<double, 3>> pts;
std::vector<std::array<size_t, 4>> tets;
int tet_mesh_resolution = 101;
std::array<double,3> tet_mesh_bbox_min {-1,-1,-1};
std::array<double,3> tet_mesh_bbox_max {1,1,1};
generate_tet_mesh(tet_mesh_resolution, tet_mesh_bbox_min,
tet_mesh_bbox_max, pts, tets);
// function values
Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor> funcVals;
// implicit arrangement result
std::vector<std::array<double, 3>> iso_pts;
std::vector<PolygonFace> iso_faces;
std::vector<std::vector<size_t>> patches;
std::vector<size_t> patch_function_label;
std::vector<bool> patch_sign_label;
std::vector<Edge> iso_edges;
std::vector<std::vector<size_t>> chains;
std::vector<std::vector<size_t>> non_manifold_edges_of_vert;
std::vector<std::vector<size_t>> shells;
std::vector<std::vector<size_t>> arrangement_cells;
std::vector<std::vector<bool>> cell_function_label;
// record timings
std::vector<std::string> timing_labels;
std::vector<double> timings;
// record stats
std::vector<std::string> stats_labels;
std::vector<size_t> stats;
std::string csg_file = "";
SECTION("one sphere intersection its negation") {
// compute function values on tet grid vertices
size_t n_pts = pts.size();
size_t n_func = 1;
funcVals.resize(n_pts, n_func);
size_t func_id;
if (!load_functions(std::string(TEST_FILE) + "/1-sphere.json", pts, funcVals)) {
throw std::runtime_error("ERROR: Failed to load functions.");
}
// compute lambda function and CSG
auto lambda = [&](std::vector<bool> cells_label){
return (cells_label[0] && !cells_label[0]);
};
bool success = csg(robust_test,
use_lookup,
use_secondary_lookup,
use_topo_ray_shooting,
positive_inside,
//
pts, tets, funcVals, lambda,
//
iso_pts,iso_faces,patches,
patch_function_label, patch_sign_label,
iso_edges,chains,
non_manifold_edges_of_vert,
shells,arrangement_cells,cell_function_label,
timing_labels,timings,
stats_labels,stats);
REQUIRE(success);
size_t corners_count = 0;
for (size_t i = 0; i < non_manifold_edges_of_vert.size(); i++) {
if (non_manifold_edges_of_vert[i].size() > 2) {
corners_count++;
}
}
std::cout << "corners: " << corners_count << std::endl;
std::cout << "cell labels: " << std::endl;
// check
REQUIRE(patches.size() == 0);
REQUIRE(chains.size() == 0);
REQUIRE(corners_count == 0);
std::vector<bool> gt_sign_label = {};
REQUIRE(patch_sign_label == gt_sign_label);
}
SECTION("three spheres config 2") {
// compute function values on tet grid vertices
size_t n_pts = pts.size();
size_t n_func = 3;
funcVals.resize(n_pts, n_func);
size_t func_id;
if (!load_functions(std::string(TEST_FILE) + "/3-sphere-2.json", pts, funcVals)) {
throw std::runtime_error("ERROR: Failed to load functions.");
}
// compute lambda function and CSG
auto lambda = [&](std::vector<bool> cells_label){
return (cells_label[0] || (cells_label[1] && cells_label[2]));
};
bool success = csg(robust_test,
use_lookup,
use_secondary_lookup,
use_topo_ray_shooting,
positive_inside,
//
pts, tets, funcVals, lambda,
//
iso_pts,iso_faces,patches,
patch_function_label, patch_sign_label,
iso_edges,chains,
non_manifold_edges_of_vert,
shells,arrangement_cells,cell_function_label,
timing_labels,timings,
stats_labels,stats);
REQUIRE(success);
size_t corners_count = 0;
for (size_t i = 0; i < non_manifold_edges_of_vert.size(); i++) {
if (non_manifold_edges_of_vert[i].size() > 2) {
corners_count++;
}
}
std::cout << "corners: " << corners_count << std::endl;
std::cout << "cell labels: " << std::endl;
// check
REQUIRE(patches.size() == 4);
REQUIRE(chains.size() == 2);
REQUIRE(corners_count == 0);
std::vector<bool> gt_sign_label = {1,1,1,1};
REQUIRE(patch_sign_label == gt_sign_label);
}
SECTION("three spheres config 3") {
// compute function values on tet grid vertices
size_t n_pts = pts.size();
size_t n_func = 3;
funcVals.resize(n_pts, n_func);
size_t func_id;
if (!load_functions(std::string(TEST_FILE) + "/3-sphere-3.json", pts, funcVals)) {
throw std::runtime_error("ERROR: Failed to load functions.");
}
// compute lambda function and CSG
auto lambda = [&](std::vector<bool> cells_label){
return (cells_label[0] && !(cells_label[1] && cells_label[2]));
};
bool success = csg(robust_test,
use_lookup,
use_secondary_lookup,
use_topo_ray_shooting,
positive_inside,
//
pts, tets, funcVals, lambda,
//
iso_pts,iso_faces,patches,
patch_function_label, patch_sign_label,
iso_edges,chains,
non_manifold_edges_of_vert,
shells,arrangement_cells,cell_function_label,
timing_labels,timings,
stats_labels,stats);
REQUIRE(success);
size_t corners_count = 0;
for (size_t i = 0; i < non_manifold_edges_of_vert.size(); i++) {
if (non_manifold_edges_of_vert[i].size() > 2) {
corners_count++;
}
}
std::cout << "corners: " << corners_count << std::endl;
std::cout << "cell labels: " << std::endl;
// check
REQUIRE(patches.size() == 5);
REQUIRE(chains.size() == 5);
REQUIRE(corners_count == 2);
std::vector<bool> gt_sign_label = {1,1,0,0,1};
REQUIRE(patch_sign_label == gt_sign_label);
}
SECTION("a plane and 2 spheres") {
// compute function values on tet grid vertices
size_t n_pts = pts.size();
size_t n_func = 3;
funcVals.resize(n_pts, n_func);
size_t func_id;
if (!load_functions(std::string(TEST_FILE) + "/3-planesphere.json", pts, funcVals)) {
throw std::runtime_error("ERROR: Failed to load functions.");
}
// compute lambda function and CSG
auto lambda = [&](std::vector<bool> cells_label){
return (cells_label[1] || cells_label[2]);
};
bool success = csg(robust_test,
use_lookup,
use_secondary_lookup,
use_topo_ray_shooting,
positive_inside,
//
pts, tets, funcVals, lambda,
//
iso_pts,iso_faces,patches,
patch_function_label, patch_sign_label,
iso_edges,chains,
non_manifold_edges_of_vert,
shells,arrangement_cells,cell_function_label,
timing_labels,timings,
stats_labels,stats);
REQUIRE(success);
size_t corners_count = 0;
for (size_t i = 0; i < non_manifold_edges_of_vert.size(); i++) {
if (non_manifold_edges_of_vert[i].size() > 2) {
corners_count++;
}
}
std::cout << "corners: " << corners_count << std::endl;
std::cout << "cell labels: " << std::endl;
// check
REQUIRE(patches.size() == 4);
REQUIRE(chains.size() == 2);
REQUIRE(corners_count == 0);
std::vector<bool> gt_sign_label = {1,1,1,1};
REQUIRE(patch_sign_label == gt_sign_label);
}
}