vtk 绘制等高线

使用 VTK 也可以实现等高线绘制，在地图、气象、地质等领域中，经常需要将数据以多个等高线的形式进行展示。在VTK(Visualization Tookt) 中，有专门的AP何以支持简便地实现这一功能。

我们还可以根据算法操作的数据类型或生成的数据类型对算法进行分类。通过类型，我们通常是指属性数据的类型，例如标量或向量。

•标量算法对标量数据进行操作。例如，在天气图上生成温度等高线。

•向量算法对向量数据进行运算。显示气流方向箭头(方向和大小)是矢量可视化的一个例子。

对颜色映射的自然延伸是轮廓。当我们看到用数据值着色的表面时，眼睛通常会将类似颜色的区域分割成不同的区域。当我们等高线数据时，
我们有效地构建了这些区域之间的边界。这些边界对应于常数标量值的等高线(2D)或曲面(3D)。

import vtk

def main():
    # import data
    reader = vtk.vtkPolyDataReader()
    #reader = vtk.vtkDataSetReader()
    #reader.SetFileName('G:/myproject/2023/03/OCT_2d-3d/1.vtk')
  
    reader.ReadAllVectorsOn()
    reader.ReadAllScalarsOn()
    reader.Update()


    # create the filter
    contours = vtk.vtkContourFilter()
    contours.SetInputData(reader.GetOutput())
    contours.GenerateValues(5, 0., 5.)
    contours.Update()

    # create the mapper
    mapper = vtk.vtkPolyDataMapper()
    mapper.SetInputData(contours.GetOutput())
    mapper.ScalarVisibilityOff()
    mapper.SetScalarRange(0., 5.)

    # create the actor
    actor = vtk.vtkActor()
    actor.SetMapper(mapper)

    # create a rendering window and renderer
    ren = vtk.vtkRenderer()
    ren.SetBackground(0.329412, 0.34902, 0.427451)  # Paraview blue

    # Assign actor to the renderer
    ren.AddActor(actor)

    renWin = vtk.vtkRenderWindow()
    renWin.AddRenderer(ren)
    renWin.SetSize(750, 750)

    # create a renderwindowinteractor
    iren = vtk.vtkRenderWindowInteractor()
    iren.SetRenderWindow(renWin)

    ren.ResetCamera()
    # render
    renWin.Render()

    # screenshot
    w2if = vtk.vtkWindowToImageFilter()
    w2if.SetInput(renWin)
    w2if.Update()
    #w2if.SetMagnification(5.)

    writer = vtk.vtkPNGWriter()
    writer.SetFileName("screenshot.png")
    writer.SetInputData(w2if.GetOutput())
    writer.Write()

    # Enable user interface interactor
    iren.Initialize()
    iren.Start()


main()