allow setting random seed for almost all meshes

Author: nschloe

pygalmesh/main.py

@@ -29,6 +29,7 @@ def generate_mesh(
     cell_radius_edge_ratio=0.0,
     cell_size=0.0,
     verbose=True,
+    seed=0
 ):
     feature_edges = [] if feature_edges is None else feature_edges
 
@@ -51,6 +52,7 @@ def generate_mesh(
         cell_radius_edge_ratio=cell_radius_edge_ratio,
         cell_size=cell_size,
         verbose=verbose,
+        seed=seed
     )
 
     mesh = meshio.read(outfile)
@@ -73,6 +75,7 @@ def generate_with_sizing_field(
     cell_radius_edge_ratio=0.0,
     cell_size=None,
     verbose=True,
+    seed=0,
 ):
     feature_edges = [] if feature_edges is None else feature_edges
 
@@ -95,6 +98,7 @@ def generate_with_sizing_field(
         cell_radius_edge_ratio=cell_radius_edge_ratio,
         cell_size=cell_size,
         verbose=verbose,
+        seed=seed,
     )
 
     mesh = meshio.read(outfile)
@@ -117,6 +121,7 @@ def generate_periodic_mesh(
     cell_size=0.0,
     number_of_copies_in_output=1,
     verbose=True,
+    seed=0,
 ):
     fh, outfile = tempfile.mkstemp(suffix=".mesh")
     os.close(fh)
@@ -139,6 +144,7 @@ def generate_periodic_mesh(
         cell_size=cell_size,
         number_of_copies_in_output=number_of_copies_in_output,
         verbose=verbose,
+        seed=seed,
     )
 
     mesh = meshio.read(outfile)
@@ -153,6 +159,7 @@ def generate_surface_mesh(
     radius_bound=0.0,
     distance_bound=0.0,
     verbose=True,
+    seed=0,
 ):
     fh, outfile = tempfile.mkstemp(suffix=".off")
     os.close(fh)
@@ -165,6 +172,7 @@ def generate_surface_mesh(
         radius_bound=radius_bound,
         distance_bound=distance_bound,
         verbose=verbose,
+        seed=seed,
     )
 
     mesh = meshio.read(outfile)
@@ -185,6 +193,7 @@ def generate_volume_mesh_from_surface_mesh(
     cell_radius_edge_ratio=0.0,
     cell_size=0.0,
     verbose=True,
+    seed=0,
 ):
     mesh = meshio.read(filename)
 
@@ -209,6 +218,7 @@ def generate_volume_mesh_from_surface_mesh(
         cell_radius_edge_ratio=cell_radius_edge_ratio,
         cell_size=cell_size,
         verbose=verbose,
+        seed=seed,
     )
 
     mesh = meshio.read(outfile)
@@ -230,6 +240,7 @@ def generate_from_inr(
     cell_radius_edge_ratio=0.0,
     cell_size=0.0,
     verbose=True,
+    seed=0,
 ):
     fh, outfile = tempfile.mkstemp(suffix=".mesh")
     os.close(fh)
@@ -248,6 +259,7 @@ def generate_from_inr(
         cell_radius_edge_ratio=cell_radius_edge_ratio,
         cell_size=cell_size,
         verbose=verbose,
+        seed=seed,
     )
 
     mesh = meshio.read(outfile)
@@ -268,6 +280,7 @@ def generate_from_inr_with_subdomain_sizing(
     facet_distance=0.0,
     cell_radius_edge_ratio=0.0,
     verbose=True,
+    seed=0,
 ):
     fh, outfile = tempfile.mkstemp(suffix=".mesh")
     os.close(fh)
@@ -296,6 +309,7 @@ def generate_from_inr_with_subdomain_sizing(
         facet_distance=facet_distance,
         cell_radius_edge_ratio=cell_radius_edge_ratio,
         verbose=verbose,
+        seed=seed,
     )
 
     mesh = meshio.read(outfile)
@@ -310,6 +324,7 @@ def remesh_surface(
     facet_size=0.0,
     facet_distance=0.0,
     verbose=True,
+    seed=0,
 ):
     mesh = meshio.read(filename)
 
@@ -328,6 +343,7 @@ def remesh_surface(
         facet_size=facet_size,
         facet_distance=facet_distance,
         verbose=verbose,
+        seed=seed,
     )
 
     mesh = meshio.read(outfile)

setup.cfg

@@ -27,7 +27,7 @@ classifiers =
 packages = find:
 setup_requires = pybind11 >= 2.2
 install_requires =
-    importlib_metadata
+    importlib_metadata;python_version<"3.8"
     meshio >= 4.0.0, < 5.0.0
     numpy
     pybind11 >= 2.2

src/generate.cpp

@@ -60,9 +60,12 @@ generate_mesh(
     const double facet_distance,
     const double cell_radius_edge_ratio,
     const double cell_size,
-    const bool verbose
+    const bool verbose,
+    const int seed
     )
 {
+  CGAL::get_default_random() = CGAL::Random(seed);
+
   const double bounding_sphere_radius2 = bounding_sphere_radius > 0 ?
     bounding_sphere_radius*bounding_sphere_radius :
     // some wiggle room
@@ -134,9 +137,12 @@ generate_with_sizing_field(
     const double facet_distance,
     const double cell_radius_edge_ratio,
     const std::shared_ptr<pygalmesh::SizingFieldBase> & cell_size,
-    const bool verbose
+    const bool verbose,
+    const int seed
     )
 {
+  CGAL::get_default_random() = CGAL::Random(seed);
+
   const double bounding_sphere_radius2 = bounding_sphere_radius > 0 ?
     bounding_sphere_radius*bounding_sphere_radius :
     // some wiggle room

src/generate.hpp

@@ -25,7 +25,8 @@ void generate_mesh(
     const double facet_distance = 0.0,
     const double cell_radius_edge_ratio = 0.0,
     const double cell_size = 0.0,
-    const bool verbose = true
+    const bool verbose = true,
+    const int seed = 0
     );
 
 void generate_with_sizing_field(
@@ -43,7 +44,8 @@ void generate_with_sizing_field(
     const double facet_distance = 0.0,
     const double cell_radius_edge_ratio = 0.0,
     const std::shared_ptr<pygalmesh::SizingFieldBase> & cell_size = nullptr,
-    const bool verbose = true
+    const bool verbose = true,
+    const int seed = 0
     );
 
 } // namespace pygalmesh

src/generate_from_inr.cpp

@@ -95,9 +95,9 @@ void
 generate_from_inr_with_subdomain_sizing(
     const std::string & inr_filename,
     const std::string & outfile,
-	const double default_cell_size,
+    const double default_cell_size,
     const std::vector<double> & cell_sizes,
-	const std::vector<int> & cell_labels,
+    const std::vector<int> & cell_labels,
     const bool lloyd,
     const bool odt,
     const bool perturb,
@@ -107,9 +107,12 @@ generate_from_inr_with_subdomain_sizing(
     const double facet_size,
     const double facet_distance,
     const double cell_radius_edge_ratio,
-    const bool verbose
+    const bool verbose,
+    const int seed
     )
 {
+  CGAL::get_default_random() = CGAL::Random(seed);
+
   CGAL::Image_3 image;
   const bool success = image.read(inr_filename.c_str());
   if (!success) {
@@ -120,7 +123,7 @@ generate_from_inr_with_subdomain_sizing(
   Sizing_field_cell cell_size(default_cell_size);
   const int ndimensions = 3;
   for(std::vector<double>::size_type i(0); i < cell_sizes.size(); ++i)
-	  cell_size.set_size(cell_sizes[i], ndimensions, cgal_domain.index_from_subdomain_index(cell_labels[i]));
+    cell_size.set_size(cell_sizes[i], ndimensions, cgal_domain.index_from_subdomain_index(cell_labels[i]));
 
   Mesh_criteria criteria(
       CGAL::parameters::edge_size=edge_size,

src/generate_from_inr.hpp

@@ -19,16 +19,17 @@ void generate_from_inr(
     const double facet_distance = 0.0,
     const double cell_radius_edge_ratio = 0.0,
     const double cell_size = 0.0,
-    const bool verbose = true
+    const bool verbose = true,
+    const int seed = 0
     );
 
 void
 generate_from_inr_with_subdomain_sizing(
     const std::string & inr_filename,
     const std::string & outfile,
-	const double default_cell_size,
+    const double default_cell_size,
     const std::vector<double> & cell_sizes,
-	const std::vector<int> & cell_labels,
+    const std::vector<int> & cell_labels,
     const bool lloyd = false,
     const bool odt = false,
     const bool perturb  = true,

src/generate_from_off.cpp

@@ -29,13 +29,24 @@ typedef CGAL::Mesh_criteria_3<Tr> Mesh_criteria;
 // To avoid verbose function and named parameters call
 using namespace CGAL::parameters;
 
-void generate_from_off(const std::string& infile, const std::string& outfile,
-                       const bool lloyd, const bool odt, const bool perturb,
-                       const bool exude, const double edge_size,
-                       const double facet_angle, const double facet_size,
-                       const double facet_distance,
-                       const double cell_radius_edge_ratio,
-                       const double cell_size, const bool verbose) {
+void generate_from_off(
+    const std::string& infile,
+    const std::string& outfile,
+    const bool lloyd,
+    const bool odt,
+    const bool perturb,
+    const bool exude,
+    const double edge_size,
+    const double facet_angle,
+    const double facet_size,
+    const double facet_distance,
+    const double cell_radius_edge_ratio,
+    const double cell_size,
+    const bool verbose,
+    const int seed
+) {
+  CGAL::get_default_random() = CGAL::Random(seed);
+
   // Create input polyhedron
   Polyhedron polyhedron;
   std::ifstream input(infile);

src/generate_from_off.hpp

@@ -20,7 +20,8 @@ generate_from_off(
     const double facet_distance = 0.0,
     const double cell_radius_edge_ratio = 0.0,
     const double cell_size = 0.0,
-    const bool verbose = true
+    const bool verbose = true,
+    const int seed = 0
     );
 
 } // namespace pygalmesh

src/generate_periodic.cpp

@@ -64,9 +64,12 @@ generate_periodic_mesh(
     const double cell_radius_edge_ratio,
     const double cell_size,
     const int number_of_copies_in_output,
-    const bool verbose
+    const bool verbose,
+    const int seed
     )
 {
+  CGAL::get_default_random() = CGAL::Random(seed);
+
   K::Iso_cuboid_3 cuboid(
       bounding_cuboid[0],
       bounding_cuboid[1],

src/generate_periodic.hpp

@@ -24,7 +24,8 @@ void generate_periodic_mesh(
     const double cell_radius_edge_ratio = 0.0,
     const double cell_size = 0.0,
     const int number_of_copies_in_output = 1,
-    const bool verbose = true
+    const bool verbose = true,
+    const int seed = 0
     );
 
 } // namespace pygalmesh