1. hello vtk

import vtk


def main():
    colors = vtk.vtkNamedColors()
    # 1. 设置背景颜色
    bkg = map(lambda x: x / 255.0, [26, 51, 102, 255])
    colors.SetColor("BkgColor", *bkg)
    # 2. 创建一个带有八个圆周面的多边形圆柱体模型
    cylinder = vtk.vtkCylinderSource()
    cylinder.SetResolution(8)
    # 3. mapper 负责将几何图像推入图形库，如果定义了标量或其他属性，它可以进行颜色映射
    cylinderMapper = vtk.vtkPolyDataMapper()
    cylinderMapper.SetInputConnection(cylinder.GetOutputPort())
    # 4. actor 是一种分组机制，除几何mapper之外，还有属性，变换矩阵，纹理映射
    # 以下设置颜色，并设置旋转-22.5度
    cylinderActor = vtk.vtkActor()
    cylinderActor.SetMapper(cylinderMapper)
    cylinderActor.GetProperty().SetColor(colors.GetColor3d("Tomato"))
    cylinderActor.RotateX(30.0)
    cylinderActor.RotateY(-45.0)

    # 5. 创建图形结构。渲染器渲染到渲染窗口中，渲染窗口交互器捕获鼠标事件，并根据事件的
    # 性质执行是适当的camera 或者 actor 操作
    ren = vtk.vtkRenderer() # 5.1 创建渲染器
    renWin = vtk.vtkRenderWindow() # 5.2 创建渲染器窗口
    renWin.AddRenderer(ren) # 5.3 渲染器窗口设置渲染器
    iren = vtk.vtkRenderWindowInteractor() # 5.4 创建渲染器窗口交互器
    iren.SetRenderWindow(renWin) # 5.5 渲染器窗口交互器设置渲染器窗口

    # 6. 添加actors至渲染器，并设置背景和大小
    ren.AddActor(cylinderActor) # 6.1 渲染器添加actor
    ren.SetBackground(colors.GetColor3d("BkgColor")) # 6.2 渲染器设置背景颜色
    renWin.SetSize(300, 300) # 6.3 渲染器窗口设置大小
    renWin.SetWindowName('Cylinder') # 6.4 渲染器窗口设置窗口名称

    # 7.渲染器窗口交互器初始化，必须在事件循环之前调用它
    iren.Initialize()

    # 8. 渲染器通过访问cammera() 并调用方法Zoom()方法进行放大操作
    ren.ResetCamera()
    ren.GetActiveCamera().Zoom(1.5)
    renWin.Render()

    # 9. 启动事件循环
    iren.Start()

if __name__=='__main__':
    main()