Python代码,实用分享

发布于:2024-05-22 ⋅ 阅读:(168) ⋅ 点赞:(0)

①多个STL格式文件合并

【1】确保你已经安装了VTK库。如果没有安装,可以使用以下命令安装:

pip install vtk

【2】使用VTK库来合并多个STL文件,并将它们加载到一个VTK场景中,在程序中进行交互式操作,单独选中每个模型

import vtk
import os


def merge_stl_files(input_files):
    # 创建一个合并过滤器
    append_filter = vtk.vtkAppendPolyData()

    for stl_file in input_files:
        # 检查文件是否存在
        if not os.path.exists(stl_file):
            print(f"文件 {stl_file} 不存在,跳过该文件。")
            continue

        # 读取STL文件
        reader = vtk.vtkSTLReader()
        reader.SetFileName(stl_file)
        reader.Update()

        # 添加到合并过滤器中
        append_filter.AddInputData(reader.GetOutput())

    # 更新合并过滤器
    append_filter.Update()

    return append_filter.GetOutput()

def main():
    # 示例使用
    input_files = [
        "C:/pycharm-project/project1/STL/ARM_1.stl",
        "C:/pycharm-project/project1/STL/ARM_2.stl",
        "C:/pycharm-project/project1/STL/ARM_3.stl",
        "C:/pycharm-project/project1/STL/ARM_4.stl",
        "C:/pycharm-project/project1/STL/HAND.stl"
    ]

    # 合并STL文件
    merged_polydata = merge_stl_files(input_files)

    # 创建一个渲染器和渲染窗口
    renderer = vtk.vtkRenderer()
    render_window = vtk.vtkRenderWindow()
    render_window.AddRenderer(renderer)

    # 创建一个交互器
    interactor = vtk.vtkRenderWindowInteractor()
    interactor.SetRenderWindow(render_window)

    # 创建一个Mapper和Actor
    mapper = vtk.vtkPolyDataMapper()
    mapper.SetInputData(merged_polydata)
    actor = vtk.vtkActor()
    actor.SetMapper(mapper)

    # 将Actor添加到渲染器中
    renderer.AddActor(actor)
    renderer.SetBackground(0.1, 0.2, 0.4)

    # 启动交互式窗口
    render_window.Render()
    interactor.Start()

if __name__ == "__main__":
    main()

该代码义了一个merge_stl_files函数来合并多个STL文件并返回合并后的PolyData对象。然后,在main函数中,将合并后的PolyData加载到了VTK渲染器中,并添加了一个交互器,这样就可以在渲染窗口中进行交互操作

♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥♥



网站公告

今日签到

点亮在社区的每一天
去签到