如何平滑pyvista.StructuredGrid？

Question

这是在 Python 中使用 PyVista 制作的 Hopf 环面：

import numpy as np
import pyvista as pv

A = 0.44
n = 3
def Gamma(t):
    alpha = np.pi/2 - (np.pi/2-A)*np.cos(n*t)
    beta = t + A*np.sin(2*n*t)
    return np.array([
      np.sin(alpha) * np.cos(beta),
      np.sin(alpha) * np.sin(beta),
      np.cos(alpha)
    ])

def HopfInverse(p, phi):
    return np.array([
      (1+p[2])*np.cos(phi),
      p[0]*np.sin(phi) - p[1]*np.cos(phi), 
      p[0]*np.cos(phi) + p[1]*np.sin(phi),
      (1+p[2])*np.sin(phi)
    ]) / np.sqrt(2*(1+p[2]))

def Stereo(q):
    return 2*q[0:3] / (1-q[3])

def F(t, phi):
    return Stereo(HopfInverse(Gamma(t), phi))

angle = np.linspace(0, 2 * np.pi, 300)
theta, phi = np.meshgrid(angle, angle)
x, y, z = F(theta, phi)

# Display the mesh
grid = pv.StructuredGrid(x, y, z)
grid.plot(smooth_shading=True)

颜色不完全平滑：在右下方的波瓣上，您可以看到一条分隔浅灰色和深灰色的线。如何去掉这条线？

Answer 1

我认为这里发生的事情是在结构化网格的两端相交处没有连接信息。解决此问题的一种方法是使用 extract_geometry() 方法将网格变成 PolyData，然后使用具有更大公差的 clean。这将迫使 pyvista 意识到网格中有一个点被加倍的接缝，导致点被合并并且接缝闭合：

import numpy as np
import pyvista as pv

A = 0.44
n = 3
def Gamma(t):
    alpha = np.pi/2 - (np.pi/2-A)*np.cos(n*t)
    beta = t + A*np.sin(2*n*t)
    return np.array([
      np.sin(alpha) * np.cos(beta),
      np.sin(alpha) * np.sin(beta),
      np.cos(alpha)
    ])

def HopfInverse(p, phi):
    return np.array([
      (1+p[2])*np.cos(phi),
      p[0]*np.sin(phi) - p[1]*np.cos(phi), 
      p[0]*np.cos(phi) + p[1]*np.sin(phi),
      (1+p[2])*np.sin(phi)
    ]) / np.sqrt(2*(1+p[2]))

def Stereo(q):
    return 2*q[0:3] / (1-q[3])

def F(t, phi):
    return Stereo(HopfInverse(Gamma(t), phi))

angle = np.linspace(0, 2 * np.pi, 300)
theta, phi = np.meshgrid(angle, angle)
x, y, z = F(theta, phi)

# Display the mesh, show seam
grid = pv.StructuredGrid(x, y, z)
grid.plot(smooth_shading=True)

# convert to PolyData and clean to remove the seam
cleaned_poly = grid.extract_geometry().clean(tolerance=1e-6)
cleaned_poly.plot(smooth_shading=True)

您对 tolerance 参数的了解可能会有所不同。

作为一个琐事，我们可以通过提取原始网格的特征边缘来可视化原始接缝：

grid.extract_feature_edges().plot()

这些曲线对应于原始网格中的开放边缘：

>>> grid.extract_surface().n_open_edges
1196

因为你的表面是封闭的和水密的，它应该有 0 个开放边：

>>> cleaned_poly.n_open_edges
0

如何平滑pyvista.StructuredGrid？

How to smooth pyvista.StructuredGrid?

python

3d

pyvista