Merge pull request #339 from sblauth/dev/topology_visualization

FEAT: Improve the visualization of the shape in topology optimization
sblauth · Oct 25, 2023 · 178dc56 · 178dc56
2 parents 09cb14d + 4a309b7
commit 178dc56
Show file tree

Hide file tree

Showing 14 changed files with 16 additions and 253 deletions.
diff --git a/cashocs/_optimization/topology_optimization/topology_optimization_problem.py b/cashocs/_optimization/topology_optimization/topology_optimization_problem.py
@@ -23,11 +23,12 @@
 from typing import Callable, List, Optional, TYPE_CHECKING, Union
 
 import fenics
+from matplotlib import colors
+import numpy as np
 import ufl
 
 from cashocs import _exceptions
 from cashocs import _optimization
-from cashocs import _utils
 from cashocs import io
 from cashocs._optimization import line_search as ls
 from cashocs._optimization.optimal_control import optimal_control_problem
@@ -308,8 +309,8 @@ def solve(
 
     def plot_shape(self) -> None:
         """Visualize the current shape in a plot."""
-        shape = fenics.Function(self.dg0_space)
-        _utils.interpolate_levelset_function_to_cells(
-            self.levelset_function, 1.0, 0.0, shape
+        rgbvals = np.array([[0, 107, 164], [255, 128, 14]]) / 255.0
+        cmap = colors.LinearSegmentedColormap.from_list(
+            "tab10_colorblind", rgbvals, N=256
         )
-        fenics.plot(shape)
+        fenics.plot(self.levelset_function, vmin=-1e-10, vmax=1e-10, cmap=cmap)
diff --git a/demos/documented/topology_optimization/cantilever/config.ini b/demos/documented/topology_optimization/cantilever/config.ini
@@ -48,8 +48,8 @@ inner_newton_atol = 0.0
 
 [Output]
 verbose = True
-save_results = True
-save_txt = True
+save_results = False
+save_txt = False
 save_state = False
 save_adjoint = False
 save_gradient = False

diff --git a/demos/documented/topology_optimization/cantilever/img_cantilever.png b/demos/documented/topology_optimization/cantilever/img_cantilever.png
diff --git a/demos/documented/topology_optimization/cantilever/results/history.json b/demos/documented/topology_optimization/cantilever/results/history.json
diff --git a/demos/documented/topology_optimization/cantilever/results/history.txt b/demos/documented/topology_optimization/cantilever/results/history.txt
diff --git a/demos/documented/topology_optimization/pipe_bend/config.ini b/demos/documented/topology_optimization/pipe_bend/config.ini
@@ -50,8 +50,8 @@ topological_derivative_is_identical = True
 
 [Output]
 verbose = True
-save_results = True
-save_txt = True
+save_results = False
+save_txt = False
 save_state = False
 save_adjoint = False
 save_gradient = False

diff --git a/demos/documented/topology_optimization/pipe_bend/img_pipe_bend.png b/demos/documented/topology_optimization/pipe_bend/img_pipe_bend.png
diff --git a/demos/documented/topology_optimization/pipe_bend/results/history.json b/demos/documented/topology_optimization/pipe_bend/results/history.json
diff --git a/demos/documented/topology_optimization/pipe_bend/results/history.txt b/demos/documented/topology_optimization/pipe_bend/results/history.txt
diff --git a/demos/documented/topology_optimization/poisson_clover/config.ini b/demos/documented/topology_optimization/poisson_clover/config.ini
@@ -50,8 +50,8 @@ topological_derivative_is_identical = True
 
 [Output]
 verbose = True
-save_results = True
-save_txt = True
+save_results = False
+save_txt = False
 save_state = False
 save_adjoint = False
 save_gradient = False

diff --git a/demos/documented/topology_optimization/poisson_clover/img_poisson_clover.png b/demos/documented/topology_optimization/poisson_clover/img_poisson_clover.png