#!/usr/bin/env python

import vtk


def main():
    # create data manually
    # cylinder = vtk.vtkCylinderSource()
    # cylinder.SetHeight(3.0) # 设置柱体的高
    # cylinder.SetRadius(1.0) #  设置柱体横截面的半径
    # cylinder.SetResolution(6) # 设置柱体横截面的等边多边形的边数

    # Read from file
    stlreader = vtk.vtkSTLReader()
    stlreader.SetFileName("42400-IDGH.stl")

    # add a filter
    # 创建过滤器对象，该对象将输入数据集的每个单元向单元质心收缩将会导致相邻单元之间出现裂缝
    shrink = vtk.vtkShrinkPolyData()
    shrink.SetInputConnection(stlreader.GetOutputPort())
    shrink.SetShrinkFactor(0.9)

    cylinderMapper = vtk.vtkPolyDataMapper() # 渲染多边形几何数据
    cylinderMapper.SetInputConnection(shrink.GetOutputPort()) # VTK可视化管线的输入数据接口 ，对应的可视化管线输出数据的接口为GetOutputPort()；
    cylinderActor = vtk.vtkActor()
    cylinderActor.SetMapper(cylinderMapper) # 设置生成几何图元的Mapper。即连接一个Actor到可视化管线的末端(可视化管线的末端就是Mapper)。
    renderer = vtk.vtkRenderer() # 负责管理场景的渲染过程
    renderer.AddActor(cylinderActor)
    renderer.SetBackground(0.1, 0.2, 0.4)
    renWin = vtk.vtkRenderWindow() # 将操作系统与VTK渲染引擎连接到一起。
    renWin.AddRenderer(renderer)
    renWin.SetSize(300, 300)
    iren = vtk.vtkRenderWindowInteractor() # 提供平台独立的响应鼠标、键盘和时钟事件的交互机制
    iren.SetRenderWindow(renWin)

    # 交互器样式的一种，该样式下，用户是通过控制相机对物体作旋转、放大、缩小等操作
    style = vtk.vtkInteractorStyleTrackballCamera()

    iren.SetInteractorStyle(style)
    iren.Initialize()

    iren.Start()

    # Clean up
    # del cylinder
    del stlreader
    del shrink
    del cylinderMapper
    del cylinderActor
    del renderer
    del renWin
    del iren

main()