4-2 STL Viewer

import vtk

def main():
    mx = vtk.vtkSTLReader() #STL Reader
    mx.SetFileName("./Resources/MX.stl") #STL File

    mxMapper = vtk.vtkPolyDataMapper() #vtkPolyData를 Geometry로 매핑
    mxMapper.SetInputConnection(mx.GetOutputPort()) #stl file의 Output과 Mapper의 Input을 연결

    mxActor = vtk.vtkActor() #Actor 생성
    mxActor.SetMapper(mxMapper) #Mapper 할당
    mxActor.GetProperty().SetColor(1, 1, 1) #Color 설정

    md = vtk.vtkSTLReader() #STL Reader
    md.SetFileName("./Resources/MD.stl") #STL File

    mdMapper = vtk.vtkPolyDataMapper() #vtkPolyData를 Geometry로 매핑
    mdMapper.SetInputConnection(md.GetOutputPort()) #stl file의 Output과 Mapper의 Input을 연결

    mdActor = vtk.vtkActor() #Actor 생성
    mdActor.SetMapper(mdMapper) #Mapper 할당
    mdActor.GetProperty().SetColor(1, 1, 1) #Color 설정

    renderer = vtk.vtkRenderer() #Renderer 생성
    renderer.AddActor(mxActor) #Renderer에 mx Actor 할당
    renderer.AddActor(mdActor) #Renderer에 md Actor 할당
    renderer.SetBackground(0.3, 0.5, 0.7) #배경화면 색상지정

    window = vtk.vtkRenderWindow() #Window 생성
    window.AddRenderer(renderer) #Window에 Renderer 할당
    window.SetSize(800, 600) #Window 크기 지정

    windowInteractor = vtk.vtkRenderWindowInteractor() #입력수단
    windowInteractor.SetRenderWindow(window) #입력수단 설정

    window.Render()
    windowInteractor.Start()

if __name__ == "__main__":
    main()