Add boolean operation function in psDomain

ViennaTools · Oct 31, 2023 · e5f7de8 · e5f7de8
1 parent cb2eafa
commit e5f7de8
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 1 deletion.
diff --git a/Examples/OxideRegrowth/OxideRegrowth.py b/Examples/OxideRegrowth/OxideRegrowth.py
@@ -26,7 +26,7 @@
     domain=domain,
     gridDelta=params["gridDelta"],
     xExtent=params["xExtent"],
-    yExtent=0.0,
+    yExtent=params["yExtent"],
     numLayers=int(params["numLayers"]),
     layerHeight=params["layerHeight"],
     substrateHeight=params["substrateHeight"],

diff --git a/Python/pyWrap.cpp b/Python/pyWrap.cpp
@@ -125,6 +125,8 @@ PYBIND11_MODULE(VIENNAPS_MODULE_NAME, module) {
       .def("insertNextLevelSetAsMaterial",
            &psDomain<T, D>::insertNextLevelSetAsMaterial)
       .def("duplicateTopLevelSet", &psDomain<T, D>::duplicateTopLevelSet)
+      .def("removeTopLevelSet", &psDomain<T, D>::removeTopLevelSet)
+      .def("applyBooleanOperation", &psDomain<T, D>::applyBooleanOperation)
       .def("setMaterialMap", &psDomain<T, D>::setMaterialMap)
       .def("getMaterialMap", &psDomain<T, D>::getMaterialMap)
       .def("generateCellSet", &psDomain<T, D>::generateCellSet,
@@ -802,6 +804,8 @@ PYBIND11_MODULE(VIENNAPS_MODULE_NAME, module) {
       .def("insertNextLevelSetAsMaterial",
            &psDomain<T, 3>::insertNextLevelSetAsMaterial)
       .def("duplicateTopLevelSet", &psDomain<T, 3>::duplicateTopLevelSet)
+      .def("applyBooleanOperation", &psDomain<T, 3>::applyBooleanOperation)
+      .def("removeTopLevelSet", &psDomain<T, 3>::removeTopLevelSet)
       .def("setMaterialMap", &psDomain<T, 3>::setMaterialMap)
       .def("getMaterialMap", &psDomain<T, 3>::getMaterialMap)
       .def("generateCellSet", &psDomain<T, 3>::generateCellSet,

diff --git a/include/psDomain.hpp b/include/psDomain.hpp
@@ -150,6 +150,18 @@ template <class NumericType = float, int D = 3> class psDomain {
     }
   }
 
+  // Boolean Operation of all level sets in the domain with the specified LS
+  void applyBooleanOperation(lsDomainType levelSet,
+                             lsBooleanOperationEnum operation) {
+    if (levelSets->empty()) {
+      return;
+    }
+
+    for (auto ls : *levelSets) {
+      lsBooleanOperation<NumericType, D>(ls, levelSet, operation).apply();
+    }
+  }
+
   materialMapType getMaterialMap() const { return materialMap; }
 
   void generateCellSet(const NumericType depth = 0.,