从vtkA更新vtkPolyData

2024-09-28 23:12:46 发布

您现在位置:Python中文网/ 问答频道 /正文

我对VTK很陌生。我尝试在python中使用VTK来编辑.stl文件,使用vtkBoxWidget。我正在读取.stl文件,用vtkBoxWidget在窗口交互器中使用鼠标/拖动来编辑它,最后我想在点击某个键时将新数据写入一个新的.stl。在

下面是一个代码来演示这个问题:

#!/usr/bin/env python

import vtk
import sys
import os

filename = "file.stl"
# render the data
reader = vtk.vtkSTLReader()
reader.SetFileName(filename)

stlMapper = vtk.vtkPolyDataMapper()
stlMapper.SetInputConnection(reader.GetOutputPort())

# create an actor for our scene

stlActor = vtk.vtkActor()
stlActor.SetMapper(stlMapper)



ren = vtk.vtkRenderer()
renWin = vtk.vtkRenderWindow()
renWin.AddRenderer(ren)
iren = vtk.vtkRenderWindowInteractor()
iren.SetRenderWindow(renWin)

# The box widget observes the events invoked by the render window
# interactor.  These events come from user interaction in the render
# window.
boxWidget = vtk.vtkBoxWidget()
boxWidget.SetInteractor(iren)
boxWidget.SetPlaceFactor(1.25)

# Add the actors to the renderer, set the background and window size.
ren.AddActor(stlActor)
ren.SetBackground(0.1, 0.2, 0.4)
renWin.SetSize(300, 300)

# As the box widget is interacted with, it produces a transformation
# matrix that is set on the actor.
t = vtk.vtkTransform()
def TransformActor(obj, event):
    global t, stlActor
    obj.GetTransform(t)
    stlActor.SetUserTransform(t)

def onKeyPressEvent(self, renderer):
    key = self.GetKeyCode()
    if(key=='l'):
        print stlActor.GetBounds()
        if(os.path.isfile("done.stl")==1):
                    os.remove("done.stl")
        print stlActor.GetMapper().GetInput().GetBounds()
        stlPolyData = stlActor.GetMapper().GetInput()
        writer = vtk.vtkSTLWriter()
        writer.SetFileName("done.stl")
        writer.AddInputDataObject(stlPolyData)
        writer.Write()
# Place the interactor initially. The actor is used to place and scale
# the interactor. An observer is added to the box widget to watch for
# interaction events. This event is captured and used to set the
# transformation matrix of the actor.
boxWidget.SetProp3D(stlActor)
boxWidget.PlaceWidget()
boxWidget.AddObserver("InteractionEvent", TransformActor)
iren.AddObserver('KeyPressEvent', onKeyPressEvent)

iren.Initialize()
renWin.Render()
iren.Start()

问题是我不知道如何更新vtkPolyData,也不知道如何将应用于vtkActor的转换传递到vtkPolyData。在上面的代码中,我用两个print函数中的.GetBounds()测试了这一点。最初对象有边界:

^{pr2}$

更新后,actor边界发生了更改,但我无法将所有这些属性传递给vtkPolyData。在

(-2.1331110248939638, 2.246371570191821, -1.5743578447908013, 2.5817210301577385, -2.1311284932847796, 2.0695034726591715)
(-2.0, 2.0, 0.0, 1.625, -1.5, 1.8125)

谢谢!在


Tags: thetoimportisoswriteractorvtk
1条回答
网友
1楼 · 发布于 2024-09-28 23:12:46

我想我觉得:

def transformPolyData(actor):
    polyData = vtk.vtkPolyData()
    polyData.DeepCopy(actor.GetMapper().GetInput())
    transform = vtk.vtkTransform()
    transform.SetMatrix(actor.GetMatrix())
    fil = vtk.vtkTransformPolyDataFilter()
    fil.SetTransform(transform)
    fil.SetInputDataObject(polyData)
    fil.Update()
    polyData.DeepCopy(fil.GetOutput())
    return polyData;

这将把polyData转换成我在保存之前所需要的。在

相关问题 更多 >