From 56ea545ad862594a2ea9ee9276498ddfba98a357 Mon Sep 17 00:00:00 2001
From: "Erin M. Gunn" <ikonone@gmail.com>
Date: Tue, 15 Feb 2022 14:48:51 -0700
Subject: [PATCH] Added porosity estimation of surfaces.

---
 implicit_modeller.py | 33 ++++++++++++++++++++++++++++++++-
 1 file changed, 32 insertions(+), 1 deletion(-)

diff --git a/implicit_modeller.py b/implicit_modeller.py
index 2ab94fb..b788d5e 100644
--- a/implicit_modeller.py
+++ b/implicit_modeller.py
@@ -117,6 +117,8 @@ def __init__(self, plotter, fn, periods=1.0, dimensions=50.0, roadWidth=0.5, roa
         self._roadWidth = roadWidth
         self._roadCount = roadCount
         self._dimensions = dimensions
+        self._porosity = 0.0
+        self.cb_updated = None
         self._update()
     
     def _update(self):
@@ -161,15 +163,31 @@ def _update(self):
             cols = colorMap([0,1], 'rainbow', vmin=0, vmax=1)
         else:
             cols = colorMap(range(n), 'rainbow')
+        
+        V = -np.ones_like(x)
+        min_offset = np.min(offsets)
+        max_offset = np.max(offsets)
         for i in range(n):
             U = self._fn(x,y,z, offsets[i])
+            if offsets[i] == max_offset:
+                V = np.maximum(V, U)
+            elif offsets[i] == min_offset:
+                V = np.maximum(V, -U)
+
             isos = Volume(U, dims=U.shape).isosurface(threshold=0)
             isos.scale(1/scale * self._dimensions)
             isos.c(cols[i])
             self._plotter += isos
             self._isos.append(isos)
         
+        empty_voxels = np.count_nonzero(V > 0)
+        total_voxels = V.shape[0]*V.shape[1]*V.shape[2]
+        self._porosity = empty_voxels / total_voxels
+        
         self._plotter.window.Render()
+
+        if self.cb_updated:
+            self.cb_updated()
              
     def setTPMSfunc(self, fn):
         self._fn = fn
@@ -279,9 +297,17 @@ def export_tpms(e):
         )
         vedo.io.write(merge(tpms._plotter.actors), filename)
     export_btn.clicked.connect(export_tpms)
-
     editor_layout.addWidget(export_btn)
 
+    row = 7
+    editor_layout.setRowStretch(row, 0)
+    editor_layout.setRowMinimumHeight(row, 20)
+    editor_layout.addWidget(QLabel("Porosity:"), row, 0)
+    porosity_text = QLineEdit("0.0")
+    porosity_text.setReadOnly(True)
+    editor_layout.addWidget(porosity_text)
+
+
     vtk_widget = QVTKRenderWindowInteractor(window)
     main_layout.addWidget(vtk_widget, stretch=1)
 
@@ -299,6 +325,11 @@ def export_tpms(e):
     road_count_text.textChanged.connect(tpms.setRoadCount)
     resoltuion_text.textChanged.connect(tpms.setResolution)
 
+    def handle_tpms_update():
+        porosity_text.setText(f'%{tpms._porosity*100:2f}')
+    tpms.cb_updated = handle_tpms_update
+    tpms._update()
+
     vp.show(bg='blackboard', axes=8)
 
     # show the widget