61 lines
1.3 KiB
Python
61 lines
1.3 KiB
Python
# Add the igl library to the modules search path
|
|
import sys, os
|
|
sys.path.insert(0, os.getcwd() + "/../")
|
|
|
|
import igl
|
|
|
|
V = igl.eigen.MatrixXd()
|
|
U = igl.eigen.MatrixXd()
|
|
F = igl.eigen.MatrixXi()
|
|
|
|
c = 0
|
|
bbd = 1.0
|
|
twod = False
|
|
|
|
if not igl.read_triangle_mesh("../../tutorial/shared/beetle.off",V,F):
|
|
print("failed to load mesh")
|
|
|
|
twod = V.col(2).minCoeff() == V.col(2).maxCoeff()
|
|
bbd = (V.colwiseMaxCoeff() - V.colwiseMinCoeff()).norm()
|
|
|
|
L = igl.eigen.SparseMatrixd()
|
|
M = igl.eigen.SparseMatrixd()
|
|
|
|
igl.cotmatrix(V,F,L)
|
|
L = -L
|
|
igl.massmatrix(V,F,igl.MASSMATRIX_TYPE_DEFAULT,M)
|
|
k = 5
|
|
|
|
D = igl.eigen.MatrixXd()
|
|
if not igl.eigs(L,M,k+1,igl.EIGS_TYPE_SM,U,D):
|
|
print("Eigs failed.")
|
|
|
|
U = (U-U.minCoeff())/(U.maxCoeff()-U.minCoeff());
|
|
|
|
viewer = igl.viewer.Viewer()
|
|
|
|
def key_down(viewer,key,mod):
|
|
global U, c
|
|
|
|
if key == ord(' '):
|
|
U = U.rightCols(k)
|
|
|
|
# Rescale eigen vectors for visualization
|
|
Z = bbd*0.5*U.col(c)
|
|
C = igl.eigen.MatrixXd()
|
|
igl.parula(U.col(c),False,C)
|
|
c = (c+1)%U.cols()
|
|
|
|
if twod:
|
|
V.setcol(2,Z)
|
|
|
|
viewer.data.set_mesh(V,F)
|
|
viewer.data.compute_normals()
|
|
viewer.data.set_colors(C)
|
|
return True
|
|
|
|
viewer.callback_key_down = key_down
|
|
viewer.callback_key_down(viewer,ord(' '),0);
|
|
viewer.core.show_lines = False
|
|
viewer.launch()
|