Add a ShapeOptimizer and an example usage

Author: FranzBangar

examples/optimization/duct.py

@@ -0,0 +1,39 @@
+# An example where a Shape is optimized *before* it is added to mesh, using ShapeOptimizer
+import os
+
+import classy_blocks as cb
+from classy_blocks.optimize.junction import ClampExistsError
+from classy_blocks.optimize.optimizer import ShapeOptimizer
+
+mesh = cb.Mesh()
+
+start_sketch = cb.SplineDisk([0, 0, 0], [2.5, 0, 0], [0, 1, 0], 0, 0)
+end_sketch = cb.SplineDisk([0, 0, 0], [1, 0, 0], [0, 2.5, 0], 0, 0).translate([0, 0, 1])
+
+shape = cb.LoftedShape(start_sketch, end_sketch)
+
+optimizer = ShapeOptimizer(shape.operations)
+
+for operation in shape.operations[:4]:
+    # remove edges because inner splines will ruin the day
+    # TODO: make edges move with points too
+    operation.top_face.remove_edges()
+    operation.bottom_face.remove_edges()
+
+    for point in operation.point_array:
+        try:
+            optimizer.add_clamp(cb.FreeClamp(point))
+        except ClampExistsError:
+            pass
+
+optimizer.optimize(tolerance=0.01)
+
+# Quick'n'dirty chopping, don't do this at home
+for operation in shape.operations:
+    for axis in range(3):
+        operation.chop(axis, count=10)
+
+mesh.add(shape)
+
+mesh.set_default_patch("walls", "wall")
+mesh.write(os.path.join("..", "case", "system", "blockMeshDict"), debug_path="debug.vtk")

src/classy_blocks/optimize/mapper.py

@@ -6,6 +6,8 @@
 from classy_blocks.util import functions as f
 from classy_blocks.util.constants import TOL
 
+ElementType = Union[Face, Operation]
+
 
 class Mapper:
     """A helper that constructs mapped sketches/shapes
@@ -32,8 +34,21 @@ def _add_point(self, point: NPPointType) -> int:
 
         return index
 
-    def add(self, element: Union[Face, Operation]):
+    def add(self, element: ElementType) -> None:
         """Add Face's or Operation's points to the map"""
         indexes = [self._add_point(point) for point in element.point_array]
         self.indexes.append(indexes)
         self.elements.append(element)
+
+    @classmethod
+    def from_map(cls, points: List[NPPointType], indexes: List[IndexType], elements: List[ElementType]) -> "Mapper":
+        """Creates a ready-made mapper from a sketch/shape that already has points/indexes defined"""
+        if len(indexes) != len(elements):
+            raise ValueError("Number of indexes and elements don't match!")
+
+        mapper = cls()
+        mapper.points = points
+        mapper.indexes = indexes
+        mapper.elements = elements
+
+        return mapper

src/classy_blocks/optimize/optimizer.py

@@ -1,18 +1,20 @@
 import abc
 import copy
 import time
-from typing import Literal
+from typing import List, Literal
 
 import numpy as np
 import scipy.optimize
 
 from classy_blocks.construct.flat.sketches.mapped import MappedSketch
+from classy_blocks.construct.operations.operation import Operation
 from classy_blocks.mesh import Mesh
 from classy_blocks.optimize.clamps.clamp import ClampBase
 from classy_blocks.optimize.clamps.surface import PlaneClamp
 from classy_blocks.optimize.grid import GridBase, HexGrid, QuadGrid
 from classy_blocks.optimize.iteration import ClampOptimizationData, IterationDriver
 from classy_blocks.optimize.links import LinkBase
+from classy_blocks.optimize.mapper import Mapper
 from classy_blocks.util.constants import TOL
 
 MinimizationMethodType = Literal["SLSQP", "L-BFGS-B", "Nelder-Mead", "Powell"]
@@ -148,6 +150,26 @@ def backport(self):
             self.mesh.vertices[i].move_to(point)
 
 
+class ShapeOptimizer(OptimizerBase):
+    def __init__(self, operations: List[Operation], report: bool = True):
+        self.mapper = Mapper()
+
+        for operation in operations:
+            self.mapper.add(operation)
+
+        grid = HexGrid(np.array(self.mapper.points), self.mapper.indexes)
+
+        super().__init__(grid, report)
+
+    def backport(self) -> None:
+        # Move every point of every operation to wherever it is now
+        for iop, indexes in enumerate(self.mapper.indexes):
+            operation = self.mapper.elements[iop]
+
+            for ipnt, i in enumerate(indexes):
+                operation.points[ipnt].move_to(self.grid.points[i])
+
+
 class SketchOptimizer(OptimizerBase):
     def __init__(self, sketch: MappedSketch, report: bool = True):
         self.sketch = sketch