用Python在VTK7中挑选2D演员

2024-10-04 01:22:35 发布

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

我有一个用pythonvtk(7)编写的简单程序,它有三个2D磁盘,可以单击并四处移动。在

我通过创建一个自定义的交互器样式来实现这一点,在这个样式中,我将所选元素存储在LeftButtonClick事件上,更改之前所选元素在MouseMove事件上的位置,并在LeftButtonRelease事件中将所选元素设置为None。在

以下是完整的VTK代码(理想情况下)应该这样做:

import vtk

class CustomInteractor(vtk.vtkInteractorStyleTrackballCamera):

    def __init__(self, renderer, renWin):
        self.AddObserver('LeftButtonPressEvent', self.OnLeftButtonDown)
        self.AddObserver('LeftButtonReleaseEvent', self.OnLeftButtonRelease)
        self.AddObserver('MouseMoveEvent', self.OnMouseMove)

        self.renderer = renderer
        self.chosenPiece = None
        self.renWin = renWin

    def OnLeftButtonRelease(self, obj, eventType):
        self.chosenPiece = None
        vtk.vtkInteractorStyleTrackballCamera.OnLeftButtonUp(self)

    def OnLeftButtonDown(self, obj, eventType):
        clickPos = self.GetInteractor().GetEventPosition()

        picker = vtk.vtkPropPicker()
        picker.Pick(clickPos[0], clickPos[1], 0, self.renderer)
        actor = picker.GetActor2D()

        self.chosenPiece = actor

        vtk.vtkInteractorStyleTrackballCamera.OnLeftButtonDown(self)

    def OnMouseMove(self, obj, eventType):
        if self.chosenPiece is not None:

            mousePos = self.GetInteractor().GetEventPosition()

            self.chosenPiece.SetPosition(mousePos[0], mousePos[1])

            self.renWin.Render()
        else :
            vtk.vtkInteractorStyleTrackballCamera.OnMouseMove(self)


def buildDiskActor(inner_radius = 10, outer_radius = 20, position = (100, 20), color = (1, 1, 1)):
    disk = vtk.vtkDiskSource()
    disk.SetInnerRadius(inner_radius)
    disk.SetOuterRadius(outer_radius)
    disk.SetRadialResolution(100)
    disk.SetCircumferentialResolution(100)
    disk.Update()

    mapper = vtk.vtkPolyDataMapper2D()
    mapper.SetInputConnection(disk.GetOutputPort())

    actor = vtk.vtkActor2D()
    actor.SetMapper(mapper)

    actor.SetPosition(position[0], position[1])

    actor.GetProperty().SetColor(color)

    return actor

ren = vtk.vtkRenderer()
renWin = vtk.vtkRenderWindow()
renWin.AddRenderer(ren)
renWin.SetSize(600, 600)

elements = [
    buildDiskActor(position = (200, 400), color=(1, 0, 0)),
    buildDiskActor(position = (400, 400), color=(0, 1, 0)),
    buildDiskActor(position = (300, 200), color=(0, 0, 1)),
]

for actor in elements:
    ren.AddActor(actor)

iren = vtk.vtkRenderWindowInteractor()
iren.SetRenderWindow(renWin)
inStyle = CustomInteractor(ren, renWin)
iren.SetInteractorStyle(inStyle)

renWin.Render()
iren.Start()

结果如下:

enter image description here

也就是说,无论我选择哪个磁盘,只有绿色的磁盘被vtkPropPicker选中。在

由于某些原因,当我使用Python2-VTK6.3.0而不是Python3-VTK7时,一切都如预期的那样工作。这是VTK7中的错误吗?或者是实施方式有所改变?还是别的什么?在


Tags: selfnonedefpositioncolorrendereractordisk