具有多个组合STL文件的VTK

时间:2019-02-12 14:56:27

标签: python stl vtk

我正在使用python和VTK开发一个项目。我成功地将三个STL文件与一个python STL libraray结合在一起。之后,我可以像这样在VTK中可视化此Combine.stl文件:

stl_reader = vtk.vtkSTLReader()
stl_reader.SetFileName(PATH)

stl_mapper = vtk.vtkPolyDataMapper()
stl_mapper.SetInputConnection(stl_reader.GetOutputPort())

stl_actor = vtk.vtkActor()
stl_actor.SetMapper(stl_mapper)

...add actor on renderer, renderer on window etc.

到目前为止,一切都很好。但是现在我想给这些STL对象之一上色/突出显示。 (所有STL一起形成一个更大的对象,因此定位很重要,并且必须保持不变。因此,整个过程最后需要保留一个vtkActor。)

我认为不可能在组合的STL文件中进行着色,因为组合的文件无法跟踪原始对象(也许这里也有解决方案,如果我错了,请纠正我)。所以我想我需要一个不同的解决方案。

我已经通过CompositePolyDataMapper示例(https://lorensen.github.io/VTKExamples/site/Python/CompositeData/CompositePolyDataMapper/)进行了尝试,但是我不知道如何从vtkSTLReader转换为可以放入vtkMultiBlockDataSet的对象。

有人知道我的问题的解决方案吗?

谢谢。

1 个答案:

答案 0 :(得分:0)

好的,我可以自己解决问题。 如果有人遇到类似问题,则解决方案非常简单:

assembly = vtk.vtkAssembly()
filenames = ["example1.stl","example2.stl","example3.stl"]
for filename in filenames:
    stl_reader = vtk.vtkSTLReader()
    stl_reader.SetFileName(filename)

    stl_mapper = vtk.vtkPolyDataMapper()
    stl_mapper.SetInputConnection(stl_reader.GetOutputPort())

    stl_actor = vtk.vtkActor()
    if filename == "example1.stl":
        stl_actor.GetProperty().SetColor(0.86,0.08,0.24)
    else:
        stl_actor.GetProperty().SetOpacity(0.4)
    stl_actor.SetMapper(stl_mapper)

    assembly.AddPart(stl_actor)

    scene_renderer.AddActor(assembly)

vtkAssembly类使您可以向其添加角色,并且如果您与之交互,则它就像一个角色一样工作。