如何使用Python VTK绘制线条
时间：2022-06-29 09:21:46 | 栏目：Python代码 | 点击：次
主要函数介绍：
vtk.vtkPoints() 在VTK中用于定义点的类，使用points.InsertPoint(index, x, y, z) 即可插入点集。函数中，第一个参数是点的序号，后面是三个参数是点的坐标。
vtk.vtkLineSource() 在VTK中定义直线的类，通过SetPoints(points),输入直线经过的点。
vtk.vtkParametricSpline() 在VTK中定义曲线的类，通过SetPoints(points),输入曲线经过的点。
vtk.vtkParametricFunctionSource() 曲线插值拟合函数，可以将输入的点集拟合成一条曲线。有很多生成方法。我们可以简单的看一下VTK官方文档介绍
S\CALAR_NONE - Scalars are not generated (default).
SCALAR_U - The scalar is set to the u-value.
SCALAR_V - The scalar is set to the v-value.
SCALAR_U0 - The scalar is set to 1 if u = (u_max - u_min)/2 = u_avg, 0 otherwise.
SCALAR_V0 - The scalar is set to 1 if v = (v_max - v_min)/2 = v_avg, 0 otherwise.
SCALAR_U0V0 - The scalar is set to 1 if u == u_avg, 2 if v == v_avg, 3 if u = u_avg && v = v_avg, 0 otherwise.
SCALAR_MODULUS - The scalar is set to (sqrt(uu+vv)), this is measured relative to (u_avg,v_avg).
SCALAR_PHASE - The scalar is set to (atan2(v,u)) (in degrees, 0 to 360), this is measured relative to (u_avg,v_avg).
SCALAR_QUADRANT - The scalar is set to 1, 2, 3 or 4. depending upon the quadrant of the point (u,v).
SCALAR_X - The scalar is set to the x-value.
SCALAR_Y - The scalar is set to the y-value.
SCALAR_Z - The scalar is set to the z-value.
SCALAR_DISTANCE - The scalar is set to (sqrt(xx+yy+z*z)). I.e. distance from the origin.
SCALAR_USER_DEFINED - The scalar is set to the value returned from EvaluateScalar().
actor.GetProperty().SetColor() 线条颜色配置
actor.GetProperty().SetLineWidth() 线条宽度配置
拟合曲线代码：
import vtk points = vtk.vtkPoints() # 定义一个点工具 points.InsertPoint(0, 329, 338, 45) # 使用InsertPoint可以插入点 # 注意：points.InsertPoint(a, b, c, d) # 其中a表示点的序号，(b,c,d)表示点的三维坐标 points.InsertPoint(1, 328, 319, 46) points.InsertPoint(2, 300, 329, 96) # 定义曲线工具 # 将前面的几个点插值拟合成一条曲线 spline = vtk.vtkParametricSpline() spline.SetPoints(points) splineSource = vtk.vtkParametricFunctionSource() splineSource.SetParametricFunction(spline) splineSource.Update() splineMapper = vtk.vtkPolyDataMapper() splineMapper.SetInputConnection(splineSource.GetOutputPort()) splineActor = vtk.vtkActor() splineActor.SetMapper(splineMapper) # 设置线条颜色 splineActor.GetProperty().SetColor(0.3800, 0.7000, 0.1600) # 设置线条宽度 splineActor.GetProperty().SetLineWidth(5) ren1 = vtk.vtkRenderer() renWin = vtk.vtkRenderWindow() renWin.AddRenderer(ren1) iren = vtk.vtkRenderWindowInteractor() iren.SetRenderWindow(renWin) ren1.AddActor(splineActor) ren1.SetBackground(1, 1, 1) renWin.SetSize(250, 250) renWin.Render() iren.Start()
绘制直线代码
import vtk points = vtk.vtkPoints() # 定义一个点工具 points.InsertPoint(0, 329, 338, 45) # 使用InsertPoint可以插入点 # 注意：points.InsertPoint(a, b, c, d) # 其中a表示点的序号，(b,c,d)表示点的三维坐标 points.InsertPoint(1, 328, 319, 46) points.InsertPoint(2, 300, 329, 96) # 定义直线工具 lineSource = vtk.vtkLineSource() lineSource.SetPoints(points) lineSource.Update() lineMapper = vtk.vtkPolyDataMapper() lineMapper.SetInputConnection(lineSource.GetOutputPort()) splineActor = vtk.vtkActor() splineActor.SetMapper(lineMapper) # 设置线条颜色 splineActor.GetProperty().SetColor(0.3800, 0.7000, 0.1600) # 设置线条宽度 splineActor.GetProperty().SetLineWidth(5) ren1 = vtk.vtkRenderer() renWin = vtk.vtkRenderWindow() renWin.AddRenderer(ren1) iren = vtk.vtkRenderWindowInteractor() iren.SetRenderWindow(renWin) ren1.AddActor(splineActor) ren1.SetBackground(1, 1, 1) renWin.SetSize(250, 250) renWin.Render() iren.Start()
