better plots

Author: teseoch

tests/bending.py

@@ -1,7 +1,7 @@
 import unittest
 
 import polyfempy as pf
-import utils
+from .utils import plot
 import os
 
 
@@ -42,7 +42,7 @@ def test_run(self):
         vertices = pts + disp
         mises, _ = solver.get_sampled_mises_avg()
 
-        utils.plot(vertices, tets, mises)
+        plot(vertices, tets, mises)
 
 
 

tests/inflation.py

@@ -3,7 +3,7 @@
 import polyfempy as pf
 import numpy as np
 
-import utils
+from .utils import plot
 
 import os
 
@@ -16,7 +16,6 @@ def test_run(self):
 		dir_path = os.path.dirname(os.path.realpath(__file__))
 		mesh_path = os.path.join(dir_path, "../3rdparty/data/circle2.msh")
 		print(mesh_path)
-		output = "inflation.obj"
 
 		settings = pf.Settings()
 		settings.discr_order = 2
@@ -72,7 +71,7 @@ def test_run(self):
 		for i in range(len(vv)):
 			vv[i] = np.array([vertices[i][0], vertices[i][1], sol[resi[i]][0]])
 
-		utils.plot(vv, faces, None)
+		plot(vv, faces, None)
 
 		# #save obj
 		# with open(output, "w") as file:

tests/plane_hole.py

@@ -1,7 +1,7 @@
 import unittest
 
 import polyfempy as pf
-import utils
+from .utils import plot
 import os
 
 
@@ -42,7 +42,7 @@ def test_run(self):
         vertices = pts + disp
         mises, _ = solver.get_sampled_mises_avg()
 
-        utils.plot(vertices, tets, mises)
+        plot(vertices, tets, mises)
 
 
 

tests/torsion.py

@@ -1,7 +1,7 @@
 import unittest
 
 import polyfempy as pf
-import utils
+from .utils import plot
 
 import os
 
@@ -47,7 +47,7 @@ def test_run(self):
         vertices = pts + disp
         mises, _ = solver.get_sampled_mises_avg()
 
-        utils.plot(vertices, tets, mises)
+        plot(vertices, tets, mises)
 
 
 if __name__ == '__main__':

tests/utils.py

@@ -26,7 +26,9 @@ def plot(vertices, connectivity, function):
     mesh = go.Mesh3d(x=x, y=y, z=z,
                      i=f[:,0], j=f[:,1], k=f[:,2],
                      intensity=function, flatshading=function is not None,
-                     hoverinfo="none")
+                     colorscale='Viridis',
+                     contour=dict(show=True, color='#fff'),
+                     hoverinfo="all")
     layout = go.Layout(scene=go.layout.Scene(aspectmode='data'))
     fig = go.Figure(data=[mesh], layout=layout)