|
@@ -0,0 +1,94 @@
|
|
|
+import sys, os
|
|
|
+
|
|
|
+
|
|
|
+sys.path.insert(0, os.getcwd() + "/../")
|
|
|
+import pyigl as igl
|
|
|
+
|
|
|
+from shared import TUTORIAL_SHARED_PATH, check_dependencies, print_usage
|
|
|
+
|
|
|
+dependencies = ["copyleft"]
|
|
|
+check_dependencies(dependencies)
|
|
|
+
|
|
|
+
|
|
|
+def key_down(viewer, key, modifier):
|
|
|
+ if key == ord('1'):
|
|
|
+ viewer.data.clear()
|
|
|
+ viewer.data.set_mesh(V, F)
|
|
|
+ elif key == ord('2'):
|
|
|
+ viewer.data.clear()
|
|
|
+ viewer.data.set_mesh(SV, SF)
|
|
|
+ elif key == ord('3'):
|
|
|
+ viewer.data.clear()
|
|
|
+ viewer.data.set_mesh(BV, BF)
|
|
|
+
|
|
|
+ return True
|
|
|
+
|
|
|
+
|
|
|
+if __name__ == "__main__":
|
|
|
+ keys = {"1": "show original mesh",
|
|
|
+ "2": "show marching cubes contour of signed distance",
|
|
|
+ "3": "show marching cubes contour of indicator function"}
|
|
|
+
|
|
|
+ print_usage(keys)
|
|
|
+
|
|
|
+ V = igl.eigen.MatrixXd()
|
|
|
+ F = igl.eigen.MatrixXi()
|
|
|
+
|
|
|
+
|
|
|
+ igl.read_triangle_mesh(TUTORIAL_SHARED_PATH + "armadillo.obj", V, F)
|
|
|
+
|
|
|
+
|
|
|
+ s = 50
|
|
|
+ Vmin = V.colwiseMinCoeff()
|
|
|
+ Vmax = V.colwiseMaxCoeff()
|
|
|
+ h = (Vmax - Vmin).maxCoeff() / s
|
|
|
+ res = (s * ((Vmax - Vmin) / (Vmax - Vmin).maxCoeff())).castint()
|
|
|
+
|
|
|
+ def lerp(res, Vmin, Vmax, di, d):
|
|
|
+ return Vmin[d] + di / (res[d] - 1) * (Vmax[d] - Vmin[d])
|
|
|
+
|
|
|
+
|
|
|
+ print("Creating grid...")
|
|
|
+ GV = igl.eigen.MatrixXd(res[0] * res[1] * res[2], 3)
|
|
|
+ for zi in range(res[2]):
|
|
|
+ z = lerp(res, Vmin, Vmax, zi, 2)
|
|
|
+ for yi in range(res[1]):
|
|
|
+ y = lerp(res, Vmin, Vmax, yi, 1)
|
|
|
+ for xi in range(res[0]):
|
|
|
+ x = lerp(res, Vmin, Vmax, xi, 0)
|
|
|
+ GV.setRow(xi + res[0] * (yi + res[1] * zi), igl.eigen.MatrixXd([[x, y, z]]))
|
|
|
+
|
|
|
+
|
|
|
+ print("Computing distances...")
|
|
|
+ S = igl.eigen.MatrixXd()
|
|
|
+ B = igl.eigen.MatrixXd()
|
|
|
+ I = igl.eigen.MatrixXi()
|
|
|
+ C = igl.eigen.MatrixXd()
|
|
|
+ N = igl.eigen.MatrixXd()
|
|
|
+
|
|
|
+ igl.signed_distance(GV, V, F, igl.SIGNED_DISTANCE_TYPE_PSEUDONORMAL, S, I, C, N)
|
|
|
+
|
|
|
+ B = S.copy()
|
|
|
+ for e in range(B.rows()):
|
|
|
+ if B[e] > 0:
|
|
|
+ B[e] = 1
|
|
|
+ else:
|
|
|
+ if B[e] < 0:
|
|
|
+ B[e] = -1
|
|
|
+ else:
|
|
|
+ B[e] = 0
|
|
|
+
|
|
|
+ print("Marching cubes...")
|
|
|
+ SV = igl.eigen.MatrixXd()
|
|
|
+ BV = igl.eigen.MatrixXd()
|
|
|
+ SF = igl.eigen.MatrixXi()
|
|
|
+ BF = igl.eigen.MatrixXi()
|
|
|
+
|
|
|
+ igl.copyleft.marching_cubes(S, GV, res[0], res[1], res[2], SV, SF)
|
|
|
+ igl.copyleft.marching_cubes(B, GV, res[0], res[1], res[2], BV, BF)
|
|
|
+
|
|
|
+
|
|
|
+ viewer = igl.viewer.Viewer()
|
|
|
+ viewer.data.set_mesh(SV, SF)
|
|
|
+ viewer.callback_key_down = key_down
|
|
|
+ viewer.launch()
|