VTK系列5 - 使用键盘事件调整对象的视角

舒伯特之路

已于 2024-08-06 14:56:58 修改

阅读量167

点赞数 1

分类专栏： VTK系列文章标签： python 开发语言

于 2024-08-06 14:56:37 首次发布

本文链接：https://blog.csdn.net/schuberta/article/details/140956029

版权

VTK系列专栏收录该内容

7 篇文章 0 订阅

订阅专栏

---------------本系列持续更新，喜欢的同学请收藏-----------------

需要模型文件的同学请留言，免费赠送！

一完成后的效果图

二代码分析

2.1 核心部分

# 添加键盘事件监听
    def key_press_event(obj, event):
        key = obj.GetKeySym()
        camera = renderer.GetActiveCamera()
        position = camera.GetPosition()
        focal_point = camera.GetFocalPoint()
        view_up = camera.GetViewUp()
        distance = camera.GetDistance()

        move_step = 5.0

        if key == 'Up':
            new_position = (position[0], position[1] + move_step, position[2])
            camera.SetPosition(new_position)
            camera.SetFocalPoint(focal_point)
        elif key == 'Down':
            new_position = (position[0], position[1] - move_step, position[2])
            camera.SetPosition(new_position)
            camera.SetFocalPoint(focal_point)
        elif key == 'Left':
            new_position = (position[0] - move_step, position[1], position[2])
            camera.SetPosition(new_position)
            camera.SetFocalPoint(focal_point)
        elif key == 'Right':
            new_position = (position[0] + move_step, position[1], position[2])
            camera.SetPosition(new_position)
            camera.SetFocalPoint(focal_point)

2.2 全部代

import os
import vtk


def main():
    # 创建一个VTK渲染窗口和渲染器
    renderer = vtk.vtkRenderer()
    render_window = vtk.vtkRenderWindow()
    render_window.AddRenderer(renderer)
    render_window_interactor = vtk.vtkRenderWindowInteractor()
    render_window_interactor.SetRenderWindow(render_window)

    # 读取robot文件夹内所有STL文件
    stl_files = [os.path.join('starShip/files', f) for f in os.listdir('starShip/files') if f.endswith('.stl')]
    colors = [(random.random(), random.random(), random.random()) for _ in stl_files]  # 生成随机颜色

    for file, color in zip(stl_files, colors):
        stl_reader = vtk.vtkSTLReader()
        stl_reader.SetFileName(file)
        stl_mapper = vtk.vtkPolyDataMapper()
        stl_mapper.SetInputConnection(stl_reader.GetOutputPort())
        stl_actor = vtk.vtkActor()
        stl_actor.SetMapper(stl_mapper)
        stl_actor.GetProperty().SetColor(color)  # 设置颜色
        renderer.AddActor(stl_actor)

        # 添加一些基本的渲染设置
    renderer.SetBackground(0.1, 0.2, 0.4)  # 背景颜色
    render_window.SetSize(800, 600)

    # 添加键盘事件监听
    def key_press_event(obj, event):
        key = obj.GetKeySym()
        camera = renderer.GetActiveCamera()
        position = camera.GetPosition()
        focal_point = camera.GetFocalPoint()
        view_up = camera.GetViewUp()
        distance = camera.GetDistance()

        move_step = 5.0

        if key == 'Up':
            new_position = (position[0], position[1] + move_step, position[2])
            camera.SetPosition(new_position)
            camera.SetFocalPoint(focal_point)
        elif key == 'Down':
            new_position = (position[0], position[1] - move_step, position[2])
            camera.SetPosition(new_position)
            camera.SetFocalPoint(focal_point)
        elif key == 'Left':
            new_position = (position[0] - move_step, position[1], position[2])
            camera.SetPosition(new_position)
            camera.SetFocalPoint(focal_point)
        elif key == 'Right':
            new_position = (position[0] + move_step, position[1], position[2])
            camera.SetPosition(new_position)
            camera.SetFocalPoint(focal_point)

        render_window.Render()

    render_window_interactor.AddObserver('KeyPressEvent', key_press_event)
    # 获取当前的相机并放大视图
    camera = renderer.GetActiveCamera()
    camera.Zoom(1.5)
    # 开始交互
    render_window.Render()
    render_window_interactor.Start()


if __name__ == "__main__":
    import random

    main()